iatoolkit 0.91.1__py3-none-any.whl → 1.4.2__py3-none-any.whl

iatoolkit/views/load_document_api_view.py

@@ -5,21 +5,22 @@
 
 from flask.views import MethodView
 from flask import request, jsonify
-from iatoolkit.services.load_documents_service import LoadDocumentsService
-from iatoolkit.services.auth_service import AuthService
-from iatoolkit.repositories.profile_repo import ProfileRepo
 from injector import inject
 import base64
 
+from iatoolkit.services.knowledge_base_service import KnowledgeBaseService
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.repositories.profile_repo import ProfileRepo
+
 
 class LoadDocumentApiView(MethodView):
     @inject
     def __init__(self,
                  auth_service: AuthService,
-                 doc_service: LoadDocumentsService,
-                 profile_repo: ProfileRepo,):
+                 knowledge_base_service: KnowledgeBaseService,
+                 profile_repo: ProfileRepo):
         self.auth_service = auth_service
-        self.doc_service = doc_service
+        self.knowledge_base_service = knowledge_base_service
         self.profile_repo = profile_repo
 
     def post(self):
@@ -48,18 +49,21 @@ class LoadDocumentApiView(MethodView):
             # get the file content from base64
             content = base64.b64decode(base64_content)
 
-            new_document = self.doc_service._file_processing_callback(
+            # Use KnowledgeBaseService for ingestion
+            new_document = self.knowledge_base_service.ingest_document_sync(
+                company=company,
                 filename=filename,
                 content=content,
-                company=company,
-                context={'metadata': metadata})
+                metadata=metadata
+            )
 
             return jsonify({
                 "document_id": new_document.id,
+                "status": "active" # ingest_document_sync returns ACTIVE on success
             }), 200
 
         except Exception as e:
             response = jsonify({"error": str(e)})
             response.status_code = 500
 
-            return response
+            return response

iatoolkit/views/login_view.py

@@ -5,7 +5,7 @@
 
 from flask.views import MethodView
 from flask import (request, redirect, render_template, url_for,
-                   render_template_string, flash)
+                   render_template_string, flash, make_response)
 from injector import inject
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.jwt_service import JWTService
@@ -133,6 +133,8 @@ class FinalizeContextView(MethodView):
                             message="Empresa no encontrada"), 404
             branding_data = self.branding_service.get_company_branding(company_short_name)
 
+            default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
+
             # 2. Finalize the context rebuild (the heavy task).
             self.query_service.set_context_for_llm(
                 company_short_name=company_short_name,
@@ -146,6 +148,8 @@ class FinalizeContextView(MethodView):
             # Get the entire 'js_messages' block in the correct language.
             js_translations = self.i18n_service.get_translation_block('js_messages')
 
+            # Importante: no envolver con make_response; dejar que Flask gestione
+            # tanto strings como tuplas (string, status) que pueda devolver render_template
             return render_template(
                 "chat.html",
                 company_short_name=company_short_name,
@@ -154,7 +158,9 @@ class FinalizeContextView(MethodView):
                 prompts=prompts,
                 onboarding_cards=onboarding_cards,
                 js_translations=js_translations,
-                redeem_token=token
+                redeem_token=token,
+                llm_default_model=default_llm_model,
+                llm_available_models=available_llm_models,
             )
 
         except Exception as e:
@@ -162,4 +168,3 @@ class FinalizeContextView(MethodView):
                                    company_short_name=company_short_name,
                                    branding=branding_data,
                                    message=f"An unexpected error occurred during context loading: {str(e)}"), 500
-

iatoolkit/views/rag_api_view.py

@@ -0,0 +1,216 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+
+from iatoolkit.common.exceptions import IAToolkitException
+from iatoolkit.services.knowledge_base_service import KnowledgeBaseService
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.common.util import Utility
+from iatoolkit.services.i18n_service import I18nService
+from flask import request, jsonify, send_file
+from flask.views import MethodView
+from injector import inject
+from datetime import datetime
+import io
+import mimetypes
+
+class RagApiView(MethodView):
+    """
+    API Endpoints for managing the RAG Knowledge Base.
+    """
+
+    @inject
+    def __init__(self,
+                 knowledge_base_service: KnowledgeBaseService,
+                 auth_service: AuthService,
+                 i18n_service: I18nService,
+                 utility: Utility):
+        self.knowledge_base_service = knowledge_base_service
+        self.auth_service = auth_service
+        self.utility = utility
+        self.i18n_service = i18n_service
+
+    def dispatch_request(self, *args, **kwargs):
+        """
+        Sobreescribimos el dispatch para soportar el mapeo de acciones personalizadas
+        pasadas a través de 'defaults' en add_url_rule (ej: action='list_files').
+        """
+        action = kwargs.pop('action', None)
+        if action:
+            method = getattr(self, action, None)
+            if method:
+                return method(*args, **kwargs)
+            else:
+                raise AttributeError(self.i18n_service.t('rag.management.action_not_found', action=action))
+
+        return super().dispatch_request(*args, **kwargs)
+
+    def list_files(self, company_short_name):
+        """
+        POST /api/rag/<company_short_name>/files
+        Returns a paginated list of documents based on filters provided in the JSON body.
+        """
+        try:
+            # 1. Authenticate the user from the current session.
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            # 2. Parse Input
+            data = request.get_json() or {}
+
+            status = data.get('status', [])
+            user_identifier = data.get('user_identifier')
+            keyword = data.get('filename_keyword')
+            from_date_str = data.get('from_date')
+            to_date_str = data.get('to_date')
+            collection = data.get('collection', '')
+            limit = int(data.get('limit', 100))
+            offset = int(data.get('offset', 0))
+
+            from_date = datetime.fromisoformat(from_date_str) if from_date_str else None
+            to_date = datetime.fromisoformat(to_date_str) if to_date_str else None
+
+            # 3. Call Service
+            documents = self.knowledge_base_service.list_documents(
+                company_short_name=company_short_name,
+                status=status,
+                collection=collection,
+                filename_keyword=keyword,
+                user_identifier=user_identifier,
+                from_date=from_date,
+                to_date=to_date,
+                limit=limit,
+                offset=offset
+            )
+
+            # 4. Format Response
+            response_list = []
+            for doc in documents:
+                response_list.append({
+                    'id': doc.id,
+                    'filename': doc.filename,
+                    'user_identifier': doc.user_identifier,
+                    'status': doc.status.value if hasattr(doc.status, 'value') else str(doc.status),
+                    'created_at': doc.created_at.isoformat() if doc.created_at else None,
+                    'metadata': doc.meta,
+                    'error_message': doc.error_message,
+                    'collection': doc.collection_type.name if doc.collection_type else None,
+                })
+
+            return jsonify({
+                'result': 'success',
+                'count': len(response_list),
+                'documents': response_list
+            }), 200
+
+        except IAToolkitException as e:
+            return jsonify({'result': 'error', 'message': e.message}), e.http_code
+        except Exception as e:
+            return jsonify({'result': 'error', 'message': str(e)}), 500
+
+    def get_file_content(self, company_short_name, document_id):
+        """
+        GET /api/rag/<company_short_name>/files/<document_id>/content
+        Streams the file content to the browser (inline view preferred).
+        """
+        try:
+            # 1. Authenticate
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            # 2. Get content from service
+            file_bytes, filename = self.knowledge_base_service.get_document_content(document_id)
+
+            if not file_bytes:
+                msg = self.i18n_service.t('rag.management.not_found')
+                return jsonify({'result': 'error', 'message': msg}), 404
+
+            # 3. Determine MIME type
+            mime_type, _ = mimetypes.guess_type(filename)
+            if not mime_type:
+                mime_type = 'application/octet-stream'
+
+            # 4. Stream response
+            return send_file(
+                io.BytesIO(file_bytes),
+                mimetype=mime_type,
+                as_attachment=False,  # Inline view
+                download_name=filename
+            )
+
+        except IAToolkitException as e:
+            return jsonify({'result': 'error', 'message': e.message}), e.http_code
+        except Exception as e:
+            return jsonify({'result': 'error', 'message': str(e)}), 500
+
+    def delete_file(self, company_short_name, document_id):
+        """
+        DELETE /api/rag/<company_short_name>/files/<document_id>
+        Deletes a document and its vectors.
+        """
+        try:
+            # 1. Authenticate
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            # 2. Call Service
+            success = self.knowledge_base_service.delete_document(document_id)
+
+            if success:
+                msg = self.i18n_service.t('rag.management.delete_success')
+                return jsonify({'result': 'success', 'message': msg}), 200
+            else:
+                msg = self.i18n_service.t('rag.management.not_found')
+                return jsonify({'result': 'error', 'message': msg}), 404
+
+        except IAToolkitException as e:
+            return jsonify({'result': 'error', 'message': e.message}), e.http_code
+        except Exception as e:
+            return jsonify({'result': 'error', 'message': str(e)}), 500
+
+    def search(self, company_short_name):
+        """
+        POST /api/rag/<company_short_name>/search
+        Synchronous semantic search for the "Search Lab" UI.
+        Returns detailed chunks with text and metadata using search_raw.
+        """
+        try:
+            # 1. Authenticate
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            # 2. Parse Input
+            data = request.get_json() or {}
+            query = data.get('query')
+            n_results = int(data.get('k', 5))
+            collection = data.get('collection')
+            metadata_filter = data.get('metadata_filter')
+
+            if not query:
+                msg = self.i18n_service.t('rag.search.query_required')
+                return jsonify({'result': 'error', 'message': msg}), 400
+
+            # 3. Call Service
+            chunks = self.knowledge_base_service.search_raw(
+                company_short_name=company_short_name,
+                query=query,
+                n_results=n_results,
+                collection=collection,
+                metadata_filter=metadata_filter,
+            )
+
+            return jsonify({
+                "result": "success",
+                "chunks": chunks
+            }), 200
+
+        except IAToolkitException as e:
+            return jsonify({'result': 'error', 'error_message': e.message}), 501
+        except Exception as e:
+            return jsonify({'result': 'error', 'error_message': str(e)}), 500

iatoolkit/views/users_api_view.py

@@ -0,0 +1,33 @@
+from flask.views import MethodView
+from flask import jsonify
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+import logging
+
+
+class UsersApiView(MethodView):
+    """
+    list company users and their roles
+    """
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 profile_service: ProfileService):
+        self.auth_service = auth_service
+        self.profile_service = profile_service
+
+    def get(self, company_short_name: str):
+        try:
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            # get users of the company
+            users = self.profile_service.get_company_users(company_short_name)
+
+            return jsonify(users), 200
+
+        except Exception as e:
+            logging.exception(f"Error fetching users for {company_short_name}: {e}")
+            return jsonify({"error": "Unexpected error", "details": str(e)}), 500

{iatoolkit-0.91.1.dist-info → iatoolkit-1.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.91.1
+Version: 1.4.2
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
@@ -207,9 +207,9 @@ Perfect for developers, small teams, single-business use cases, and experimentat
 
 ### 🟥 Enterprise Edition (Commercial License)
 - Unlimited Companies (multi-tenant)  
-- Advanced RAG pipelines  
-- External connectors (S3, APIs, email)  
-- SSO, auditing, access controls  
+- Payment services integration 
+- Enterprise Agent Workflows
+- SSO integration
 - Priority support & continuous updates  
 - Activation via **License Key**  
 

{iatoolkit-0.91.1.dist-info → iatoolkit-1.4.2.dist-info}/RECORD

@@ -1,22 +1,23 @@
-iatoolkit/__init__.py,sha256=U1z2Ra1_zh3nri1jxmA3UnM7X89FNXUv9aovMHxTikc,1337
-iatoolkit/base_company.py,sha256=t3h4oFye9NQMpa4GwEQwHEz6Wd0TF31Vux0xY8-q6Bg,1334
-iatoolkit/cli_commands.py,sha256=1lSDE_6K3p_psGnHrgaGT6MWaz27aRDn10dzb2wf9_E,2271
-iatoolkit/company_registry.py,sha256=rZxuAMO-5ehdaYltRANLhBk7tp7H_Tfxs--XKiY_PE4,4015
-iatoolkit/core.py,sha256=EcwRiKyn5oCRCBMbvOVCNOjgOXeT55vP613pwIjoXbg,19433
+iatoolkit/__init__.py,sha256=4kPCwhdZff5rt-FhnHfafMCPvD5hDFGEdRwINPm8ATc,1425
+iatoolkit/base_company.py,sha256=TzGvdseDpJMi9fl5nbR2pMARgU6pN6ISatFoMJKtNxw,648
+iatoolkit/cli_commands.py,sha256=EDky1QpvHnatzXsd1evYW0OQt-k3JIbPGMc13vgDvy4,1814
+iatoolkit/company_registry.py,sha256=XopFzqIpUxBtQJzdZwu2-c0v98MLT7SP_xLPxn_iENM,4176
+iatoolkit/core.py,sha256=QKqu-QDrNWunDU9uX96a86kFNv3a_aAxonkpxJptzoE,20885
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-iatoolkit/common/exceptions.py,sha256=kZyyTL9P72SlwljddmrFFvARRVdzWv7uQW3H2bM4GmU,1183
-iatoolkit/common/routes.py,sha256=uAFwVt_hmirDQbi-scZA3ZFmHOsYWxLh-vXJE-4tK0c,5888
+iatoolkit/common/exceptions.py,sha256=xmNi5Et13pCvCM1so79z36I-7Kdie8nVUAqW9acGs9k,1202
+iatoolkit/common/model_registry.py,sha256=HiFr1FtRKGjpp3YurkiF4l6sGuirigDVL-b_3_R-vlM,5561
+iatoolkit/common/routes.py,sha256=cJCFLgSqrTXPOC9CraXLQtoOhVCphWXHyFadYiTXQGw,7292
 iatoolkit/common/session_manager.py,sha256=OUYMzT8hN1U-NCUidR5tUAXB1drd8nYTOpo60rUNYeY,532
-iatoolkit/common/util.py,sha256=38G_7TUm_d1pIcJNNV-poZZjlvFVSXnuBPgagSi-9LM,15299
+iatoolkit/common/util.py,sha256=e000sUjUXeY4BJrJ3HtnsVMMhn8guclRZkl0f2Dmfy0,15270
+iatoolkit/common/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+iatoolkit/common/interfaces/asset_storage.py,sha256=IYXyc73gmwFT7T8RQhC4MymQDW0vGDSEzuKk43w7Ct4,989
+iatoolkit/common/interfaces/database_provider.py,sha256=Cf-33SZ7HPxTv2b_ahew1TcqDmFH442OPwIv6YEiubk,1097
 iatoolkit/infra/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/infra/brevo_mail_app.py,sha256=xvy3KxEEgjFpHlIAogP6SU5KXmg3w7lC16nnNmOYU8Y,5323
 iatoolkit/infra/call_service.py,sha256=iRk9VxbXaAwlLIl8fUzGDWIAdzwfsbs1MtP84YeENxg,4929
-iatoolkit/infra/gemini_adapter.py,sha256=kXV-t5i9GmWBafUPX2kAyiqvcT7GPoxHylcCUWG_c_U,15051
 iatoolkit/infra/google_chat_app.py,sha256=_uKWxeacHH6C5a4FVx0YZjBn1tL-x_MIQV9gqgWGAjo,1937
-iatoolkit/infra/llm_client.py,sha256=bVrPYooJbkKYZsZKZ8VeyeC7ewLnKAVBTkvJtfRpbgE,18522
-iatoolkit/infra/llm_proxy.py,sha256=TJti_mkdeFfsl4oGd5gTB7gaxF6bNG_rwqchkDiubw0,6933
-iatoolkit/infra/llm_response.py,sha256=YUUQPBHzmP3Ce6-t0kKMRIpowvh7de1odSoefEByIvI,904
-iatoolkit/infra/openai_adapter.py,sha256=tbzd8aPAH5cQOJT-sD4ypqq2fWB6WiEIGuGesUDnQNk,3550
+iatoolkit/infra/llm_proxy.py,sha256=6PngfQ4Q-5tLiqdroNCn58kBJh5uZZb_6S98HKH4pQY,10639
+iatoolkit/infra/llm_response.py,sha256=6hSmI909m0Pdgs7UnJHrv_nokbCZLzHpdwBUED6VNck,1062
 iatoolkit/infra/redis_session_manager.py,sha256=EPr3E_g7LHxn6U4SV5lT_L8WQsAwg8VzA_WIEZ3TwOw,3667
 iatoolkit/infra/connectors/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/infra/connectors/file_connector.py,sha256=HOjRTFd-WfDOcFyvHncAhnGNZuFgChIwC-P6osPo9ZM,352
@@ -25,43 +26,47 @@ iatoolkit/infra/connectors/google_cloud_storage_connector.py,sha256=IXpL3HTo7Ft4
 iatoolkit/infra/connectors/google_drive_connector.py,sha256=WR1AlO5-Bl3W89opdja0kKgHTJzVOjTsy3H4SlIvwVg,2537
 iatoolkit/infra/connectors/local_file_connector.py,sha256=hrzIgpMJOTuwTqzlQeTIU_50ZbZ6yl8lcWPv6hMnoqI,1739
 iatoolkit/infra/connectors/s3_connector.py,sha256=Nj4_YaLobjfcnbZewJf21_K2EXohgcc3mJll1Pzn4zg,1123
-iatoolkit/locales/en.yaml,sha256=e8tTV8k4qk0ESSj3OlFqX7iz0pq8Bii-io_q6kw3Aeg,7989
-iatoolkit/locales/es.yaml,sha256=xqryHAeRSGBhhrdt0FnK-OxOtfWYsbVU2sEuxsrnjTw,8459
+iatoolkit/infra/llm_providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+iatoolkit/infra/llm_providers/deepseek_adapter.py,sha256=ui4V05xplUvLWHu7y_dUMKpP4Af-BkZi5tvdgZY16dU,10719
+iatoolkit/infra/llm_providers/gemini_adapter.py,sha256=xjYAMgS4bh7qfpCEv6n92tzpKdjnNBVFfr8GkSrcatM,14555
+iatoolkit/infra/llm_providers/openai_adapter.py,sha256=9SuHYtPJcadWSV_GHfx8DDj9BqkUS3RvLSyDd_ZVB2M,4848
+iatoolkit/locales/en.yaml,sha256=cKPJQB2sV_AuFd80eOK4r-Y08cKlmG9eG7b34iuBqNo,12210
+iatoolkit/locales/es.yaml,sha256=1LT81MPQAW0mhHOVxWtgiOTxYs6B9Jyc6wgQGz04xGo,13027
 iatoolkit/repositories/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
-iatoolkit/repositories/database_manager.py,sha256=bpVSLAE-ltDB4vAfNlf7q6QfAWWJmRthmzscrv1q3K8,5832
-iatoolkit/repositories/document_repo.py,sha256=vhFc0hu9GK6yoKJHs2dLaAoQ9ZJaf9GEOsD2yWuVuNw,1130
-iatoolkit/repositories/llm_query_repo.py,sha256=AuelF-LfMoJScgwVEYIN1JoOLk9UKANoRsnzNtKA7CI,3839
-iatoolkit/repositories/models.py,sha256=78oHVVxt6xQlizth0GDmEcMCMuvCOi6mFM2dRaPuoLE,14343
-iatoolkit/repositories/profile_repo.py,sha256=ETSnEwDVni5w-5hSHFxnf-zJq5QUqP6DoR7qfJeqfBA,3845
-iatoolkit/repositories/tasks_repo.py,sha256=icVO_r2oPagGnnBhwVFzznnvEEU2EAx-2dlWuWvoDC4,1745
-iatoolkit/repositories/vs_repo.py,sha256=jVyzcK4l7T4n7O0KVLDIfOiZkPEA-rV1OtgYCD2M_G4,6110
+iatoolkit/repositories/database_manager.py,sha256=JAi-A_siEZlyQeV7xk1NIYFZjfBCCo12RVSV_pLYYhI,6631
+iatoolkit/repositories/document_repo.py,sha256=5oCrq8pjcyOCBrOc4XYeUvza9nUd0VJ4eNL0g1MQb68,1441
+iatoolkit/repositories/filesystem_asset_repository.py,sha256=ZFr_FpHQD5nszZnJSki6nc-W3PKvmhTxKmVZMd2DUbM,1825
+iatoolkit/repositories/llm_query_repo.py,sha256=8rGSVx7WZlGbUfeVw_uGygTUuJXRK5vGfFHj1znc1ss,3981
+iatoolkit/repositories/models.py,sha256=PH8HiMxKDzZ7khDl9dod2XbRrkWlWCThOrRF9BeChPE,13460
+iatoolkit/repositories/profile_repo.py,sha256=zhjTTizhkTaCZvWMCZtTzTTTmx0fsjtao2fKurwP1xE,5838
+iatoolkit/repositories/vs_repo.py,sha256=CzT77B7_kWN5p3QhXgdf5X3JkS_02CwFVhxmp0K3rrc,6140
 iatoolkit/services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/services/auth_service.py,sha256=B0gTUHDf56WiTnA5viBEvbmf2A6y_6bkAoFlMvu0IWY,7756
 iatoolkit/services/benchmark_service.py,sha256=CdbFYyS3FHFhNzWQEa9ZNjUlmON10DT1nKNbZQ1EUi8,5880
 iatoolkit/services/branding_service.py,sha256=Fsjcos7hGLx2qhoZ9hTaFCgb7x5eYHMDGvXg4mXx8D0,8120
-iatoolkit/services/company_context_service.py,sha256=LhscfhcB1w6ycjNkd075cZSknIAJQQB_Aty5CWM5hfY,7381
-iatoolkit/services/configuration_service.py,sha256=1HPTrfsqF1VyDS1zZc5tI9bo22OMeZwuD1BGVDnXsD8,13764
-iatoolkit/services/dispatcher_service.py,sha256=e-y8T5Zvy1_XUc1uDttImnhwbtI3pblEFGOxIb9vnOQ,4516
+iatoolkit/services/company_context_service.py,sha256=MV07jvef_9u_55n92yHCDwThBykn5rQT1tKg51Ebxoo,9339
+iatoolkit/services/configuration_service.py,sha256=bYjhR5cZcJunFr1OtX2ReOJaHbYrxBlxizEdoo52zaw,17933
+iatoolkit/services/dispatcher_service.py,sha256=f3aPbKU9UkNS61dNVZjT-J9hg3x-H2Nb_n-cZRnZ9aI,5247
 iatoolkit/services/document_service.py,sha256=XrAwbNPyhmP94bGdDwHy76Mg_PCg6O2QyZwii1T6kZ8,6469
 iatoolkit/services/embedding_service.py,sha256=ngRnFXEpko-rqpicqbJbVdr91IjSe1jXSYVWprOVlE0,6296
 iatoolkit/services/excel_service.py,sha256=de4fj3klWFWSTdgej8yk0FBuNr1lunyy9cuOyZwl0rI,6221
-iatoolkit/services/file_processor_service.py,sha256=h8Tg2Xwvjv_HOIUo51Rnu5H4tUmkJ8W9tKvmFfgKPms,3792
-iatoolkit/services/history_manager_service.py,sha256=PMDj35XR96IbwLl0VnD6QE-1mSpl_eouTlf6MdkLXQs,8502
+iatoolkit/services/file_processor_service.py,sha256=5xS7zrdL_kpCFXQ5seqNl9tiseI7otD8NRIQENuafjE,3628
+iatoolkit/services/history_manager_service.py,sha256=nq1hk39WDbyHG6s8irpIdCS98EBDagF0djH6JHq9WuY,8734
 iatoolkit/services/i18n_service.py,sha256=mR4pS0z56NgZLeSnEvDXiMvVBeOCl5CkUWdYBTVEhyM,3941
 iatoolkit/services/jwt_service.py,sha256=pVoH1rzRwWixpvT3AhFdiE1BDmdbo4KQMyOF0P7tz-Y,2939
+iatoolkit/services/knowledge_base_service.py,sha256=kjVI04WjOf5q_fYEDNkg-lRN1hqbQrS0URzUGy3dkuo,16500
 iatoolkit/services/language_service.py,sha256=ktNzj0hrCPm_VMIpWTxFEpEHKQ1lOHOfuhzCbZud2Dc,3428
 iatoolkit/services/license_service.py,sha256=pdl9szSEPbIu2ISl6g-WIR1p4HQu2gSGdAjViKxY6Hg,2751
-iatoolkit/services/load_documents_service.py,sha256=04XZmn6AWOK0wiRU7QDu2VYRekOK7fD19H-mC4FiMjA,7908
+iatoolkit/services/llm_client_service.py,sha256=YSvoNKeBQRGL3znMDIyLH2PGZt4ImmYKD_aBlBeS91k,19024
+iatoolkit/services/load_documents_service.py,sha256=zK1mXsj9aLDDKoV9WKgNuFlg02ROnlf9umcTm_Y4OIs,6761
 iatoolkit/services/mail_service.py,sha256=6Kx1CIbXzAr_ucoqwqTlhhwE6y2Jw64eNDjXysaZP-c,8297
-iatoolkit/services/profile_service.py,sha256=6MxFfaQeDf-VXcs8ijQNsiYfDXfN_YyEWETHlrdYuUQ,22581
-iatoolkit/services/prompt_service.py,sha256=HFCAhhtrdhOZmf5f-DjUoYlFQob3MHTPrGGJQ48u_4o,13005
-iatoolkit/services/query_service.py,sha256=eR2YrDX5-fJE3IuUFSwqvc_UfEtHwmH3wi2QWEEQKR8,19423
-iatoolkit/services/search_service.py,sha256=Omx7tXit5V9o_HdIEFs24Hmaz382CxuphfFzvige4Uo,2188
-iatoolkit/services/sql_service.py,sha256=JMDCCn43e2woYsGPhMfFvAc-c-XuBQ6_oGlxWSVZtyE,5522
-iatoolkit/services/tasks_service.py,sha256=itREO5rDnUIgsqtyCOBKDtH30QL5v1egs4qPTiBK8xU,6865
-iatoolkit/services/tool_service.py,sha256=K1CesxoovZT9wGqaE6QQoVPtMU4bNno4Dc0uXpHDcIc,9458
+iatoolkit/services/profile_service.py,sha256=mUXW5dwFF4SBH04adQuoSj-E0FmuLDt_VcPo9oxdr1M,23680
+iatoolkit/services/prompt_service.py,sha256=mr3szmOThNi6MtqCbh4rwJJOlk-Be35bkIJXqh7_hvM,12898
+iatoolkit/services/query_service.py,sha256=aS2-NtEMdu-QViw58d-g9iZI1j3KZIJb0xrN8N_c__U,20406
+iatoolkit/services/sql_service.py,sha256=kgLEE7iiw9WjnyvOBfm4VJ60-F6JAg26cAKF6LlF-zU,7087
+iatoolkit/services/tool_service.py,sha256=FQuplPsgs1AdyOk2pxUBkPIvAvUqJU1ElOpOWz5wkr8,10002
 iatoolkit/services/user_feedback_service.py,sha256=KbjMjl0CDF7WpPHioJ3sqQVwocjfq_cisrpeqYUqtas,5361
-iatoolkit/services/user_session_context_service.py,sha256=O9yX2i9B2OEo0pzMvkRKRrb9EV7kdlpYkX4GXUCKTpM,7670
+iatoolkit/services/user_session_context_service.py,sha256=RSv_NxOSTVPA6nFLiZKSIuUWa59cmZWBwiN_TWSljBo,9074
 iatoolkit/static/images/fernando.jpeg,sha256=W68TYMuo5hZVpbP-evwH6Nu4xWFv2bc8pJzSKDoLTeQ,100612
 iatoolkit/static/images/iatoolkit_core.png,sha256=EsmhI8ucevGtzWUPtRDUBfoU-1jqXK7ZmQtrZ-cbJO8,42358
 iatoolkit/static/images/iatoolkit_logo.png,sha256=cGSO-06Y8hCwxsnEGbUBjqFmlSA4pm-OOnNypOEAZVs,7563
@@ -70,34 +75,36 @@ iatoolkit/static/js/chat_filepond.js,sha256=mzXafm7a506EpM37KATTK3zvAswO1E0KSUY1
 iatoolkit/static/js/chat_help_content.js,sha256=VwJ4esnrQ2xCP09IhObJyg-sRIJYMGGiAIRF8fX_lE0,5201
 iatoolkit/static/js/chat_history_button.js,sha256=i9EBAWWW2XyQnAuj1b-c8102EG_gDkQ2ElbA8_Nu0yo,3491
 iatoolkit/static/js/chat_logout_button.js,sha256=Of9H6IbAboSBmeqRaurEVW6_dL752L0UeDcDLNBD5Z0,1335
-iatoolkit/static/js/chat_main.js,sha256=DllrDCL1bwR3gA6mHHckGSzttmJw0K8li2f5atL6Pro,12960
-iatoolkit/static/js/chat_onboarding_button.js,sha256=nlm7dmMs-KTq3f2Ve5LhAkCbn-wx_3zDReivX8Ch7DU,3364
+iatoolkit/static/js/chat_main.js,sha256=9qlBiiR1Xf1iW0kn6W4gkKtk_9mr4ZT5Ib8-Y0GkXr4,14964
+iatoolkit/static/js/chat_model_selector.js,sha256=EJC3GAZK0AXZaYdyYL4oA0NbxAVwK-bV2EyFbtU01RA,7175
+iatoolkit/static/js/chat_onboarding_button.js,sha256=o1lovwq38bENAzSnCi8c0bUgEszq_C49NdzeaAwXthU,3359
 iatoolkit/static/js/chat_prompt_manager.js,sha256=QYki28CpyM2Chn82dnOP2eH6FObxH8eChGFyUxukv1M,3319
-iatoolkit/static/js/chat_reload_button.js,sha256=FHMm5sLdn2HuWDq66wphNuRcltjLG-aADp0yTwEbnQw,1256
-iatoolkit/static/styles/chat_iatoolkit.css,sha256=_3tqIydkfbLBoo_mASdSD-jmjIsvh5wePhGgMwPFWS4,13808
+iatoolkit/static/js/chat_reload_button.js,sha256=MFZgtBCZdNpWnx-L1V1M_dUFx0p8lWs_G0LdXFol3gY,1361
+iatoolkit/static/styles/chat_iatoolkit.css,sha256=cGD9riX-4s978XJJKPcyO6qHq1ngKMzNEhsRpimySVs,15926
 iatoolkit/static/styles/chat_modal.css,sha256=9rwWrzL4Vq7AsdiGb3qczyOUf2PJEgLjSLCkSns8dQA,3031
 iatoolkit/static/styles/chat_public.css,sha256=2SDz0gGyTnTn3BUwD--QX-0qEc3eYKXeTTC6h730UCU,4486
 iatoolkit/static/styles/documents.css,sha256=6Yoprc9JHlKdtqUBEGr4r-JCQsK3Bg1ivoxfJyGuKvg,11893
 iatoolkit/static/styles/landing_page.css,sha256=HVYbo9ZqbqsImvMhGBq2dDFkRiRWOtVaZ0IP9QqDquA,8914
-iatoolkit/static/styles/llm_output.css,sha256=AlxgRSOleeCk2dLAqFWVaQ-jwZiJjcpC5rHuUv3T6VU,2312
+iatoolkit/static/styles/llm_output.css,sha256=EQjfETdViLjIsyUYp3NvlR1fGICUCK531OsRx43dYvk,3131
 iatoolkit/static/styles/onboarding.css,sha256=fNiqT_MMJ6gGhNzfbSZhJwvcCZ_8gibL-MWZwxyAgAs,3749
 iatoolkit/system_prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6VCHQq1p37rpu5wwk,3576
-iatoolkit/system_prompts/query_main.prompt,sha256=FXYjE4ICiEHiu7oTgK7lAEytUlk30WVacIf5P_yRm2c,2155
+iatoolkit/system_prompts/query_main.prompt,sha256=ZiuiRQRkfZn7Su8SilV57LwCdsIyRASrKxdAWi5h27w,2991
 iatoolkit/system_prompts/sql_rules.prompt,sha256=U1Z5BObI-fgkMatcYfx89q2-YmuRNK6OV6q69p-wAFI,60585
 iatoolkit/templates/_company_header.html,sha256=L1QtrMpqYXALqtinCa3WhG4CrgHKLObl_d-jfQYNjhg,1903
 iatoolkit/templates/_login_widget.html,sha256=qziV70-n7SwC11rL4cU1LenJM83jYvuyw2xSz6OORJA,2063
-iatoolkit/templates/base.html,sha256=xOox6JJEd6dHBlw6DBrFDJTXtAKCaXZc3Ffrufa-GDk,3042
+iatoolkit/templates/base.html,sha256=sRHUnHaKIycLRKZ_PQOTb88kSBC6kigzxlm60vpXj1I,3434
 iatoolkit/templates/change_password.html,sha256=p0GWZ9gldluYQM8OPSz2Q4CYhU8UJmb-72iz_Sl_6Ho,3485
-iatoolkit/templates/chat.html,sha256=Ey4iJkyYpNkOnOrwx7Qngy7UDwk8GOq3eB4HyYaC9tA,13300
+iatoolkit/templates/chat.html,sha256=HtmTBbbBPvVHscEymSWqAFjCnxE4hlMckzRgdi-1MXs,15642
 iatoolkit/templates/chat_modals.html,sha256=x0Do7I7X9cfgZFHSAwWreIN55nYPCpZK7PTBxGXjrYY,8501
 iatoolkit/templates/error.html,sha256=4aT6cIYQUILciekAceHEiUuZxcAqkmNPUMnB9CD4BNA,1778
 iatoolkit/templates/forgot_password.html,sha256=XLVMmLJK5SAXmPqIH_OyTlSbu5QTAdekO6ggzfcNup0,2376
-iatoolkit/templates/onboarding_shell.html,sha256=fKNT3aIhXcO95Q8C092axRxSGYGKOTD00vetM-sMSt0,4648
+iatoolkit/templates/onboarding_shell.html,sha256=GODHz4JrNyZyf9sPq7GNKMoRrLmKn6kZZQBU4HLgyiU,4604
 iatoolkit/templates/signup.html,sha256=kvV9B8M99zMDj4y9up-K5ChzOznSxzZTX4jPShGRBvI,4316
 iatoolkit/views/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
-iatoolkit/views/base_login_view.py,sha256=S53PSIoML1yMCfP2kr8-csNGe0Ig-wNaBilzpdjjMyo,3683
+iatoolkit/views/base_login_view.py,sha256=l45msQUrLGxxxZrSb5l7tnTLLQyIoPTu4TgbkHBvVlc,3981
 iatoolkit/views/change_password_view.py,sha256=TXFvvtNrR37yrNAHYzcMSSIkV4-MIzmIpwbPffaJJyE,4957
+iatoolkit/views/chat_view.py,sha256=QNlC0n1hxjWbI3wyLhV6D-U2kJZkIqClNg94iQP4PrA,3534
 iatoolkit/views/embedding_api_view.py,sha256=9n5utihq8zHiG0V2sVCOYQ72wOKa4denqE951gqaJWo,2371
 iatoolkit/views/forgot_password_view.py,sha256=C4eJe9GmGrl6M3bTt_anwuPQib9ouXZtKYp6kB2xBeU,3380
 iatoolkit/views/help_content_api_view.py,sha256=8EUw2iSE6X-Lm2dGlYZxSp-29HzaAGwD3CpL4Otqp5k,2011
@@ -105,21 +112,22 @@ iatoolkit/views/history_api_view.py,sha256=oNYfo-fZbz4sY6VpE_goaCFQyH2k9IX85VhPo
 iatoolkit/views/home_view.py,sha256=GaU2-qyTnE8jE1ejdNjHW3azo72X_Awhp1y1Zx9BXRg,2662
 iatoolkit/views/init_context_api_view.py,sha256=axfNno8E1v4dCNuXNkdV1XpDZtypICj48hW3cRyl_eA,2961
 iatoolkit/views/llmquery_api_view.py,sha256=ohlYmFcvhvZL2BTMeCzd7ZPeTBzIPcIZbC9CRWqQM70,2290
-iatoolkit/views/load_document_api_view.py,sha256=RFkVrzOy6eG_aoQYdxb3EpInTnomZpyDaHRW3_n4dro,2310
-iatoolkit/views/login_view.py,sha256=9EilnksI6M46FsqVAaiADVrtekZDtJyBpQgoRQQqTgE,7099
+iatoolkit/views/load_company_configuration_api_view.py,sha256=ufX_jrSO7NTUsgNJNrAXxFKmyp2JyNYoplMUm4biULs,1902
+iatoolkit/views/load_document_api_view.py,sha256=RaO4KRz19uSOv3LL7Fn8FJSLzHtRkPAIYRPU-yT1nxQ,2485
+iatoolkit/views/login_view.py,sha256=NKPjR9cCOsw7FhOfXCM32m6veKxRdLhsc0H7DLc5_ug,7517
 iatoolkit/views/logout_api_view.py,sha256=7c0rL5sLTuoRPqQs73wlH_7eb3--S6GLcr7Yu4hNOYU,2035
 iatoolkit/views/profile_api_view.py,sha256=qhlmhyygIs5h-OyNDwq1uGUS8S-GxB2zOYY51Hs5prY,1645
 iatoolkit/views/prompt_api_view.py,sha256=WAv4I6WmGfT1TTZ-lml9LaR2wDsSx9mshYcxFAV_wtk,1255
+iatoolkit/views/rag_api_view.py,sha256=zsEdsSMoxBrX3jGFoKWB7oUm_Sp1x6lh2tAy-g_CUiI,8389
 iatoolkit/views/root_redirect_view.py,sha256=t7tqiyJAPffzIuQVwnfgIw3Bo51qkegyMmBTfChsrJk,833
 iatoolkit/views/signup_view.py,sha256=83jgbAuOAETID88TC8h-wCszkkEaFygCH6tTj7XyZYI,4261
 iatoolkit/views/static_page_view.py,sha256=V0oTybQ5lFo3hp5xOKau-yMO8vK8NCM6zWiwVI8lpBQ,881
-iatoolkit/views/tasks_api_view.py,sha256=wGnuwuuL83ByQ1Yre6ytRVztA0OGQjGrwMjB1_G830U,2630
-iatoolkit/views/tasks_review_api_view.py,sha256=wsCpzqyRyUdCXWAhyGlBe3eNZZ6A1DQG7TblN_GZNfM,1894
 iatoolkit/views/user_feedback_api_view.py,sha256=QOTBtpNqYAmewXDgVAoaIprnVjvf1xM0ExWxSFp3ICI,2098
+iatoolkit/views/users_api_view.py,sha256=hVR6rxLNyhGiXhCIHvttUeSinH071Pyk-2_Fcm-v2TM,1139
 iatoolkit/views/verify_user_view.py,sha256=Uc9P2wiRjZ7pMRTjWUn8Y-p0DFqw_AwNYQsdbAwhbg8,2659
-iatoolkit-0.91.1.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
-iatoolkit-0.91.1.dist-info/licenses/LICENSE_COMMUNITY.md,sha256=9lVNcggPNUnleMF3_h3zu9PbNTeaRqB1tHAbiBLQJZU,566
-iatoolkit-0.91.1.dist-info/METADATA,sha256=no3mDlRTL0FwdP007a2cmmvwJIbgeeFQRAcB3TTPoCc,8421
-iatoolkit-0.91.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.91.1.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-0.91.1.dist-info/RECORD,,
+iatoolkit-1.4.2.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
+iatoolkit-1.4.2.dist-info/licenses/LICENSE_COMMUNITY.md,sha256=9lVNcggPNUnleMF3_h3zu9PbNTeaRqB1tHAbiBLQJZU,566
+iatoolkit-1.4.2.dist-info/METADATA,sha256=j6HullcLZAkwHSYnpcXew6VVsL6nzFkyoXwpuSBDy1E,8395
+iatoolkit-1.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-1.4.2.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-1.4.2.dist-info/RECORD,,

iatoolkit/repositories/tasks_repo.py

@@ -1,52 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-from injector import inject
-from datetime import datetime
-from iatoolkit.repositories.models import Task, TaskStatus, TaskType
-from iatoolkit.repositories.database_manager import DatabaseManager
-from sqlalchemy import or_
-
-
-class TaskRepo:
-    @inject
-    def __init__(self, db_manager: DatabaseManager):
-        self.session = db_manager.get_session()
-
-    def create_task(self, new_task: Task) -> Task:
-        self.session.add(new_task)
-        self.session.commit()
-        return new_task
-
-    def update_task(self, task: Task) -> Task:
-        self.session.commit()
-        return task
-
-    def get_task_by_id(self, task_id: int):
-        return self.session.query(Task).filter_by(id=task_id).first()
-
-    def create_or_update_task_type(self, new_task_type: TaskType):
-        task_type = self.session.query(TaskType).filter_by(name=new_task_type.name).first()
-        if task_type:
-            task_type.prompt_template = new_task_type.prompt_template
-            task_type.template_args = new_task_type.template_args
-        else:
-            self.session.add(new_task_type)
-            task_type = new_task_type
-        self.session.commit()
-        return task_type
-
-    def get_task_type(self, name: str):
-        task_type = self.session.query(TaskType).filter_by(name=name).first()
-        return task_type
-
-    def get_pending_tasks(self, company_id: int):
-        now = datetime.now()
-        tasks = self.session.query(Task).filter(
-            Task.company_id == company_id,
-            Task.status == TaskStatus.pendiente,
-            or_(Task.execute_at == None, Task.execute_at <= now)
-        ).all()
-        return tasks