cwyodmodules 0.3.43__tar.gz → 0.3.44__tar.gz

{cwyodmodules-0.3.43 → cwyodmodules-0.3.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cwyodmodules
-Version: 0.3.43
+Version: 0.3.44
 Summary: Add your description here
 Author-email: Patrik <patrikhartl@gmail.com>
 Classifier: Operating System :: OS Independent

cwyodmodules-0.3.44/cwyodmodules/batch/utilities/helpers/config/agent_mode.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class AgentMode(Enum):
+    OFF = "off"
+    ON = "on" 

{cwyodmodules-0.3.43 → cwyodmodules-0.3.44}/cwyodmodules/batch/utilities/helpers/config/config_helper.py

@@ -13,6 +13,7 @@ from ..env_helper import EnvHelper
 from .assistant_strategy import AssistantStrategy
 from .conversation_flow import ConversationFlow
 from .database_type import DatabaseType
+from .agent_mode import AgentMode
 
 CONFIG_CONTAINER_NAME = "config"
 CONFIG_FILE_NAME = "active.json"
@@ -59,6 +60,7 @@ class Config:
         self.conversational_flow = config.get(
             "conversational_flow", self.env_helper.CONVERSATION_FLOW
         )
+        self.agent_mode = config.get("agent_mode", AgentMode.NORMAL.value)
 
     @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def get_available_document_types(self) -> list[str]:
@@ -103,6 +105,10 @@ class Config:
     def get_available_conversational_flows(self):
         return [c.value for c in ConversationFlow]
 
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_agent_modes(self):
+        return [c.value for c in AgentMode]
+
 
 # TODO: Change to AnsweringChain or something, Prompts is not a good name
 class Prompts:
@@ -198,6 +204,9 @@ class ConfigHelper:
             ]
         if config.get("enable_chat_history") is None:
             config["enable_chat_history"] = default_config["enable_chat_history"]
+            
+        if config.get("agent_mode") is None:
+            config["agent_mode"] = default_config["agent_mode"]
 
     @staticmethod
     @functools.cache

{cwyodmodules-0.3.43 → cwyodmodules-0.3.44}/cwyodmodules/batch/utilities/helpers/config/default.json

@@ -143,5 +143,6 @@
     "strategy": "${ORCHESTRATION_STRATEGY}"
   },
   "enable_chat_history": true,
-  "database_type": "${DATABASE_TYPE}"
+  "database_type": "${DATABASE_TYPE}",
+  "agent_mode": "off"
 }

{cwyodmodules-0.3.43 → cwyodmodules-0.3.44}/cwyodmodules.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cwyodmodules
-Version: 0.3.43
+Version: 0.3.44
 Summary: Add your description here
 Author-email: Patrik <patrikhartl@gmail.com>
 Classifier: Operating System :: OS Independent

{cwyodmodules-0.3.43 → cwyodmodules-0.3.44}/cwyodmodules.egg-info/SOURCES.txt

@@ -49,6 +49,7 @@ cwyodmodules/batch/utilities/helpers/lightrag_helper.py
 cwyodmodules/batch/utilities/helpers/llm_helper.py
 cwyodmodules/batch/utilities/helpers/orchestrator_helper.py
 cwyodmodules/batch/utilities/helpers/secret_helper.py
+cwyodmodules/batch/utilities/helpers/config/agent_mode.py
 cwyodmodules/batch/utilities/helpers/config/assistant_strategy.py
 cwyodmodules/batch/utilities/helpers/config/config_helper.py
 cwyodmodules/batch/utilities/helpers/config/conversation_flow.py

{cwyodmodules-0.3.43 → cwyodmodules-0.3.44}/pyproject.toml

@@ -3,7 +3,7 @@ name = "cwyodmodules"
 authors = [
   { name="Patrik", email="patrikhartl@gmail.com" },
 ]
-version = '0.3.43'
+version = '0.3.44'
 description = "Add your description here"
 readme = "README.md"
 requires-python = "==3.11.11"