iatoolkit 2.12.0__tar.gz → 2.12.1__tar.gz

{iatoolkit-2.12.0/src/iatoolkit.egg-info → iatoolkit-2.12.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.12.0
+Version: 2.12.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "2.12.0"
+__version__ = "2.12.1"
 
 # Expose main classes and functions at the top level of the package
 

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/repositories/llm_query_repo.py

@@ -151,7 +151,10 @@ class LLMQueryRepo:
             return self.session.query(Prompt).filter(
                 Prompt.company_id == company.id,
                 Prompt.active.is_(True),
-                Prompt.agent_role == PromptAgentRole.WORKSPACE_CHAT.value,
+                Prompt.agent_role.in_([
+                    PromptAgentRole.WORKSPACE_CHAT.value,
+                    PromptAgentRole.WORKSPACE_AGENT.value,
+                ]),
             ).all()
 
     def create_or_update_prompt(self, new_prompt: Prompt):
@@ -169,9 +172,9 @@ class LLMQueryRepo:
                 or PromptAgentRole.WORKSPACE_CHAT.value
             )
             prompt.execution_mode = (
-                getattr(new_prompt, "execution_mode", None)
-                or prompt.execution_mode
-                or PromptExecutionMode.CONVERSATIONAL.value
+                PromptExecutionMode.CONVERSATIONAL.value
+                if prompt.agent_role == PromptAgentRole.WORKSPACE_CHAT.value
+                else PromptExecutionMode.AGENTIC.value
             )
             prompt.filename = new_prompt.filename
             prompt.custom_fields = new_prompt.custom_fields
@@ -195,7 +198,11 @@ class LLMQueryRepo:
             if not new_prompt.agent_role:
                 new_prompt.agent_role = PromptAgentRole.WORKSPACE_CHAT.value
             if not new_prompt.execution_mode:
-                new_prompt.execution_mode = PromptExecutionMode.CONVERSATIONAL.value
+                new_prompt.execution_mode = (
+                    PromptExecutionMode.CONVERSATIONAL.value
+                    if new_prompt.agent_role == PromptAgentRole.WORKSPACE_CHAT.value
+                    else PromptExecutionMode.AGENTIC.value
+                )
             if not new_prompt.output_schema_mode:
                 new_prompt.output_schema_mode = "best_effort"
             if not new_prompt.output_response_mode:

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/repositories/models.py

@@ -38,6 +38,7 @@ class PromptExecutionMode(str, enum.Enum):
 
 class PromptAgentRole(str, enum.Enum):
     WORKSPACE_CHAT = "workspace_chat"
+    WORKSPACE_AGENT = "workspace_agent"
     CHANNELS = "channels"
     OPERATIONS = "operations"
 

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/services/configuration_service.py

@@ -634,6 +634,7 @@ class ConfigurationService:
         allowed_execution_modes = {"conversational", "agentic"}
         allowed_agent_roles = {
             "workspace_chat",
+            "workspace_agent",
             "channels",
             "operations",
         }

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/services/context_builder_service.py

@@ -123,7 +123,7 @@ class ContextBuilderService:
         raw_agent_role = str(getattr(prompt_obj, "agent_role", "") or "").strip().lower()
         agent_role = (
             raw_agent_role
-            if raw_agent_role in {"workspace_chat", "channels", "operations"}
+            if raw_agent_role in {"workspace_chat", "workspace_agent", "channels", "operations"}
             else "workspace_chat"
         )
         execution_mode = "conversational" if agent_role == "workspace_chat" else "agentic"

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/services/prompt_service.py

@@ -50,6 +50,7 @@ class PromptService:
     EXECUTION_MODE_CONVERSATIONAL = PromptExecutionMode.CONVERSATIONAL.value
     EXECUTION_MODE_AGENTIC = PromptExecutionMode.AGENTIC.value
     AGENT_ROLE_WORKSPACE_CHAT = PromptAgentRole.WORKSPACE_CHAT.value
+    AGENT_ROLE_WORKSPACE_AGENT = PromptAgentRole.WORKSPACE_AGENT.value
     AGENT_ROLE_CHANNELS = PromptAgentRole.CHANNELS.value
     AGENT_ROLE_OPERATIONS = PromptAgentRole.OPERATIONS.value
     DEFAULT_CATEGORY_LABEL = "General"
@@ -96,6 +97,7 @@ class PromptService:
         candidate = str(agent_role or "").strip().lower()
         allowed = {
             self.AGENT_ROLE_WORKSPACE_CHAT,
+            self.AGENT_ROLE_WORKSPACE_AGENT,
             self.AGENT_ROLE_CHANNELS,
             self.AGENT_ROLE_OPERATIONS,
         }
@@ -120,7 +122,10 @@ class PromptService:
     @classmethod
     def is_workspace_chat_role(cls, agent_role: str | None) -> bool:
         normalized = str(agent_role or "").strip().lower()
-        return normalized == cls.AGENT_ROLE_WORKSPACE_CHAT
+        return normalized in {
+            cls.AGENT_ROLE_WORKSPACE_CHAT,
+            cls.AGENT_ROLE_WORKSPACE_AGENT,
+        }
 
     @classmethod
     def is_channel_role(cls, agent_role: str | None) -> bool:

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/services/query_service.py

@@ -286,7 +286,7 @@ class QueryService:
                 str(raw_agent_role).strip().lower()
                 if raw_agent_role is not None else "workspace_chat"
             )
-            if normalized_agent_role not in {"workspace_chat", "channels", "operations"}:
+            if normalized_agent_role not in {"workspace_chat", "workspace_agent", "channels", "operations"}:
                 normalized_agent_role = "workspace_chat"
             contract["agent_role"] = normalized_agent_role
             contract["execution_mode"] = (
@@ -480,7 +480,7 @@ class QueryService:
     @staticmethod
     def _is_workspace_chat_role(agent_role: str | None) -> bool:
         normalized = str(agent_role or "").strip().lower()
-        return normalized == "workspace_chat"
+        return normalized in {"workspace_chat", "workspace_agent"}
 
     @staticmethod
     def _is_channel_role(agent_role: str | None) -> bool:

{iatoolkit-2.12.0 → iatoolkit-2.12.1}/src/iatoolkit/views/categories_api_view.py

@@ -135,5 +135,6 @@ class CategoriesApiView(MethodView):
     SUPPORTED_PROMPT_AGENT_ROLES = [
         "workspace_chat",
         "operations",
+        "workspace_agent",
         "channels",
     ]

{iatoolkit-2.12.0 → iatoolkit-2.12.1/src/iatoolkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.12.0
+Version: 2.12.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT