iatoolkit 2.3.1__tar.gz → 2.3.2__tar.gz

{iatoolkit-2.3.1/src/iatoolkit.egg-info → iatoolkit-2.3.2}/PKG-INFO

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

{iatoolkit-2.3.1 → iatoolkit-2.3.2}/src/iatoolkit/__init__.py

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

iatoolkit-2.3.2/src/iatoolkit/system_prompts/memory_usage.prompt

@@ -0,0 +1,13 @@
+### Memoria personal
+
+- El usuario puede tener paginas de memoria personal construidas desde notas,
+  links, archivos y mensajes de chat guardados.
+- Usa `iat_memory_search` cuando el usuario pregunte explicitamente por notas
+  guardadas, elementos recordados o continuidad personal entre sesiones.
+- Si `iat_memory_search` devuelve un resultado con `has_native_files=true`,
+  llama a `iat_memory_get_page` antes de responder cuando el contenido de los
+  archivos adjuntos pueda importar.
+- Usa `iat_memory_get_page` para leer una pagina especifica antes de responder
+  cuando sea necesario.
+- Prefiere las paginas de memoria como contexto compilado; no asumas que todo
+  el contenido guardado es globalmente relevante.

iatoolkit-2.3.2/src/iatoolkit/views/chat_context_preview_api_view.py

@@ -0,0 +1,61 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from __future__ import annotations
+
+import logging
+
+from flask import jsonify, request
+from flask.views import MethodView
+from injector import inject
+
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.query_service import QueryService
+
+
+class ChatContextPreviewApiView(MethodView):
+    @inject
+    def __init__(
+        self,
+        auth_service: AuthService,
+        query_service: QueryService,
+    ):
+        self.auth_service = auth_service
+        self.query_service = query_service
+
+    def _require_admin_auth(self, company_short_name: str) -> dict | tuple:
+        auth_result = self.auth_service.verify_for_company(company_short_name)
+        if not auth_result.get("success"):
+            status_code = auth_result.get("status_code", 401)
+            if status_code == 403:
+                return jsonify({"error": "Forbidden"}), 403
+            return jsonify(auth_result), status_code
+
+        role = (auth_result.get("user_role") or "").lower()
+        if role not in {"admin", "owner"}:
+            return jsonify({"error": "Forbidden"}), 403
+
+        return auth_result
+
+    def get(self, company_short_name: str):
+        auth = self._require_admin_auth(company_short_name)
+        if isinstance(auth, tuple):
+            return auth
+
+        try:
+            question = str(request.args.get("question") or "")
+            result = self.query_service.preview_chat_context(
+                company_short_name=company_short_name,
+                user_identifier=auth.get("user_identifier"),
+                question=question,
+            )
+            if result.get("error"):
+                status_code = int(result.get("status_code") or 400)
+                return jsonify({"error": result.get("error_message") or "Preview failed"}), status_code
+
+            return jsonify(result), 200
+        except Exception as exc:
+            logging.exception("Fatal error building chat context preview")
+            return jsonify({"error": str(exc)}), 500

iatoolkit-2.3.2/src/iatoolkit/views/prompt_context_preview_api_view.py

@@ -0,0 +1,66 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from __future__ import annotations
+
+import logging
+
+from flask import jsonify, request
+from flask.views import MethodView
+from injector import inject
+
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.query_service import QueryService
+
+
+class PromptContextPreviewApiView(MethodView):
+    @inject
+    def __init__(
+        self,
+        auth_service: AuthService,
+        query_service: QueryService,
+    ):
+        self.auth_service = auth_service
+        self.query_service = query_service
+
+    def _require_admin_auth(self, company_short_name: str) -> dict | tuple:
+        auth_result = self.auth_service.verify_for_company(company_short_name)
+        if not auth_result.get("success"):
+            status_code = auth_result.get("status_code", 401)
+            if status_code == 403:
+                return jsonify({"error": "Forbidden"}), 403
+            return jsonify(auth_result), status_code
+
+        role = (auth_result.get("user_role") or "").lower()
+        if role not in {"admin", "owner"}:
+            return jsonify({"error": "Forbidden"}), 403
+
+        return auth_result
+
+    def post(self, company_short_name: str, prompt_name: str):
+        auth = self._require_admin_auth(company_short_name)
+        if isinstance(auth, tuple):
+            return auth
+
+        try:
+            payload = request.get_json(silent=True) or {}
+            client_data = payload.get("client_data")
+            question = payload.get("question") or ""
+
+            result = self.query_service.preview_prompt_context(
+                company_short_name=company_short_name,
+                user_identifier=auth.get("user_identifier"),
+                prompt_name=prompt_name,
+                client_data=client_data if isinstance(client_data, dict) else {},
+                question=str(question or ""),
+            )
+            if result.get("error"):
+                status_code = int(result.get("status_code") or 400)
+                return jsonify({"error": result.get("error_message") or "Preview failed"}), status_code
+
+            return jsonify(result), 200
+        except Exception as exc:
+            logging.exception("Fatal error building prompt context preview")
+            return jsonify({"error": str(exc)}), 500

{iatoolkit-2.3.1 → iatoolkit-2.3.2/src/iatoolkit.egg-info}/PKG-INFO

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

{iatoolkit-2.3.1 → iatoolkit-2.3.2}/src/iatoolkit.egg-info/SOURCES.txt

@@ -167,6 +167,7 @@ src/iatoolkit/system_prompts/file_download_output.prompt
 src/iatoolkit/system_prompts/format_styles.prompt
 src/iatoolkit/system_prompts/html_structures.prompt
 src/iatoolkit/system_prompts/links_documents.prompt
+src/iatoolkit/system_prompts/memory_usage.prompt
 src/iatoolkit/system_prompts/multimodal_basics.prompt
 src/iatoolkit/system_prompts/output_basics.prompt
 src/iatoolkit/system_prompts/presentation_formatting.prompt
@@ -199,6 +200,7 @@ src/iatoolkit/views/api_key_api_view.py
 src/iatoolkit/views/base_login_view.py
 src/iatoolkit/views/categories_api_view.py
 src/iatoolkit/views/change_password_view.py
+src/iatoolkit/views/chat_context_preview_api_view.py
 src/iatoolkit/views/chat_view.py
 src/iatoolkit/views/configuration_api_view.py
 src/iatoolkit/views/connectors_api_view.py
@@ -215,6 +217,7 @@ src/iatoolkit/views/logout_api_view.py
 src/iatoolkit/views/memory_api_view.py
 src/iatoolkit/views/profile_api_view.py
 src/iatoolkit/views/prompt_api_view.py
+src/iatoolkit/views/prompt_context_preview_api_view.py
 src/iatoolkit/views/prompt_resource_api_view.py
 src/iatoolkit/views/rag_api_view.py
 src/iatoolkit/views/root_redirect_view.py