iatoolkit 2.9.0__tar.gz → 2.10.0__tar.gz

{iatoolkit-2.9.0/src/iatoolkit.egg-info → iatoolkit-2.10.0}/PKG-INFO

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

{iatoolkit-2.9.0 → iatoolkit-2.10.0}/src/iatoolkit/__init__.py

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

{iatoolkit-2.9.0 → iatoolkit-2.10.0}/src/iatoolkit/locales/en.yaml

@@ -1322,6 +1322,9 @@ ui:
     reasoning_card_helper: "Applies when the selected model and provider support this reasoning level. Prompts can still override it."
     reasoning_unavailable: "Not available for this model"
     reasoning_unavailable_help: "This model/provider does not support a company-level reasoning default."
+    telemetry_toggle_label: "LLM telemetry"
+    telemetry_card_helper: "When enabled, requests executed through QueryService send LLM runs to the company telemetry backend."
+    telemetry_requires_setup: "Configure llm.telemetry provider, project, and api_key before enabling this switch."
     reasoning_effort_minimal: "Minimal"
     reasoning_effort_low: "Low"
     reasoning_effort_medium: "Medium"

{iatoolkit-2.9.0 → iatoolkit-2.10.0}/src/iatoolkit/locales/es.yaml

@@ -1326,6 +1326,9 @@ ui:
     reasoning_card_helper: "Se aplica cuando el modelo y el provider seleccionados soportan este nivel de razonamiento. Los prompts igual pueden sobreescribirlo."
     reasoning_unavailable: "No disponible para este modelo"
     reasoning_unavailable_help: "Este modelo o provider no soporta un valor por defecto de razonamiento a nivel company."
+    telemetry_toggle_label: "Telemetría LLM"
+    telemetry_card_helper: "Cuando está habilitada, las requests ejecutadas a través de QueryService envían las ejecuciones LLM al backend de telemetría de la empresa."
+    telemetry_requires_setup: "Configura llm.telemetry con provider, project y api_key antes de habilitar este switch."
     reasoning_effort_minimal: "Minimal"
     reasoning_effort_low: "Bajo"
     reasoning_effort_medium: "Medio"

{iatoolkit-2.9.0 → iatoolkit-2.10.0}/src/iatoolkit/services/query_service.py

@@ -200,12 +200,14 @@ class QueryService:
                        company_short_name: str,
                        model: Optional[str],
                        prompt_output_contract: dict | None = None) -> str:
-        # Priority: 1. Explicit model -> 2. Prompt default -> 3. Company config
-        effective_model = model
-        if not effective_model and isinstance(prompt_output_contract, dict):
+        # Priority: 1. Prompt model -> 2. Explicit model -> 3. Company config
+        effective_model = None
+        if isinstance(prompt_output_contract, dict):
             prompt_model = str(prompt_output_contract.get("llm_model") or "").strip()
             if prompt_model:
                 effective_model = prompt_model
+        if not effective_model:
+            effective_model = model
         if not effective_model:
             llm_config = self.configuration_service.get_configuration(company_short_name, 'llm')
             if llm_config and llm_config.get('model'):

{iatoolkit-2.9.0 → iatoolkit-2.10.0}/src/iatoolkit/services/telemetry_service.py

@@ -310,15 +310,32 @@ class TelemetryService:
         execution_metadata: dict | None = None,
         request_metadata: dict | None = None,
     ) -> dict[str, Any]:
+        telemetry_config = self.configuration_service.get_llm_telemetry_config(company_short_name) or {}
+        company_telemetry_enabled = bool(telemetry_config.get("enabled"))
         prompt_options = dict((prompt_output_contract or {}).get("llm_request_options") or {})
-        if prompt_options.get("telemetry_enabled") is not True:
+        prompt_telemetry_requested = prompt_options.get("telemetry_enabled") is True
+        request_source = str((execution_metadata or {}).get("request_source") or "").strip().lower()
+        chat_request = request_source == "chat_ui"
+
+        telemetry_requested = company_telemetry_enabled or prompt_telemetry_requested
+        if not telemetry_requested:
             return {}
 
-        telemetry_config = self.configuration_service.get_llm_telemetry_config(company_short_name) or {}
         provider_name = str(telemetry_config.get("provider") or "").strip().lower()
-        enabled = bool(telemetry_config.get("enabled"))
+        enabled = company_telemetry_enabled
         prompt_name = str((prompt_output_contract or {}).get("prompt_name") or "").strip()
         execution_mode = str((prompt_output_contract or {}).get("execution_mode") or "").strip().lower() or None
+        if chat_request and not execution_mode:
+            execution_mode = "chat"
+
+        agent_name = prompt_name or ("chat" if chat_request else None)
+        execution_name = prompt_name or ("iatoolkit.chat" if chat_request else "iatoolkit.llm_execution")
+        if chat_request:
+            telemetry_scope = "chat"
+        elif prompt_name:
+            telemetry_scope = "prompt"
+        else:
+            telemetry_scope = "query_service"
         request: dict[str, Any] = {
             "requested": True,
             "record_stats": True,
@@ -326,14 +343,16 @@ class TelemetryService:
             "provider": provider_name or None,
             "metadata": {
                 "company": str(company_short_name or "").strip() or None,
-                "agent_name": prompt_name or None,
+                "agent_name": agent_name,
                 "provider": provider,
                 "model": model,
                 "task_id": task_id,
                 "user_identifier": user_identifier,
                 "execution_mode": execution_mode,
+                "request_source": request_source or None,
+                "telemetry_scope": telemetry_scope,
             },
-            "execution_name": prompt_name or "iatoolkit.llm_execution",
+            "execution_name": execution_name,
         }
 
         if isinstance(request_metadata, dict):

{iatoolkit-2.9.0 → iatoolkit-2.10.0}/src/iatoolkit/views/llmquery_api_view.py

@@ -37,6 +37,13 @@ class LLMQueryApiView(MethodView):
             if not data:
                 return jsonify({"error": "Invalid JSON body"}), 400
 
+            client_data = data.get('client_data') or {}
+            if not isinstance(client_data, dict):
+                client_data = {}
+            else:
+                client_data = dict(client_data)
+            client_data.setdefault("source", "chat_ui")
+
             # 4. Call the unified query service method.
             result = self.query_service.llm_query(
                 company_short_name=company_short_name,
@@ -47,7 +54,7 @@ class LLMQueryApiView(MethodView):
                 },
                 question=data.get('question', ''),
                 prompt_name=data.get('prompt_name'),
-                client_data=data.get('client_data', {}),
+                client_data=client_data,
                 ignore_history=data.get('ignore_history', False),
                 files=data.get('files', [])
             )

{iatoolkit-2.9.0 → iatoolkit-2.10.0/src/iatoolkit.egg-info}/PKG-INFO

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