Merge pull request #12 from zcsadmin/11-modifica-zcs-python-core-per-follia-module-in-requeststate-refs-ap-2057

add follia module in RequestState

Author: marchettizcs

app/src/zcs/core/session/__init__.py

@@ -1,3 +1,4 @@
 from .request_context import request_context
 from .request_state import RequestState
-from .auth_info import AuthInfo
+from .auth_info import AuthInfo
+from .follia_module import FolliaModule

app/src/zcs/core/session/follia_module.py

@@ -0,0 +1,9 @@
+
+from enum import StrEnum
+
+class FolliaModule(StrEnum):
+    """
+    Data class to hold Follia module information for a user request.
+    """
+    VIRTUAL_ASSISTANT = "virtual_assistant"
+    DATA_EXTRACTION = "data_extraction"

app/src/zcs/core/session/request_state.py

@@ -3,6 +3,7 @@
 from typing import Optional
 
 from zcs.core.session.auth_info import AuthInfo
+from zcs.core.session.follia_module import FolliaModule
 
 
 class RequestState():
@@ -12,13 +13,29 @@ def __init__(
             request_id: Optional[str] = None,
             prefix: Optional[str] = None,
             op_code: Optional[str] = None,
-            auth_info: Optional[AuthInfo] = None):
+            auth_info: Optional[AuthInfo] = None,
+            follia_module: Optional[FolliaModule] = None):
 
         self.__request_id = request_id if request_id else RequestState.generate_op_code(prefix=prefix)
         self.__op_code = op_code if op_code else self.__request_id
         self.__request_start_ns = time.perf_counter_ns()
         self.__checkpoint_ns = self.__request_start_ns
         self.__auth_info = auth_info
+        self.__follia_module = follia_module
+
+    def getFolliaModule(self) -> Optional[FolliaModule]:
+        """
+        Get Follia module information.
+        """
+
+        return self.__follia_module
+    
+    def setFolliaModule(self, follia_module: FolliaModule):
+        """
+        Set Follia module information.
+        """
+
+        self.__follia_module = follia_module
 
     def getAuthInfo(self) -> Optional[AuthInfo]:
         """