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

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.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.91.1
+Version: 1.7.0
 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.7.0.dist-info}/RECORD

@@ -1,67 +1,72 @@
-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=lBC4IhNqHSyQzU5dQEHh49nkOgc97JfQq2E27-VVPa0,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=iVZawjKfGt6iK9tZ8t3y6dofP4KJpIQex48OVVvlRYo,7540
 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=NfSuDbzrGVS59fIgqJETYUlD5otMLwAjyp7OVhaMoo8,16310
+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=7ayYuEYRunkm4udoOGumoX9PfdgohBcrj0JDPfyfM00,1156
 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
-iatoolkit/infra/connectors/file_connector_factory.py,sha256=3qvyfH4ZHKuiMxJFkawOxhW2-TGKKtsBYHgoPpZMuKU,2118
+iatoolkit/infra/connectors/file_connector_factory.py,sha256=iCOhklUaYG3YxtJGBYOrLN8spF5mjKu6vyA6nj71FHI,2167
 iatoolkit/infra/connectors/google_cloud_storage_connector.py,sha256=IXpL3HTo7Ft4EQsYiQq5wXRRQK854jzOEB7ZdWjLa4U,2050
 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/connectors/s3_connector.py,sha256=r8bgye8ydMzayAAg5RS22oakpKd-vNw-cSrOCvlwG9g,1209
+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=NX1LMzpBNtgRl0q5RqD9wwTbF9IKzdVheTOaGzI1P7E,13571
+iatoolkit/locales/es.yaml,sha256=ote9JSzWq8lGwbfX7wKKIp7ccdZ76-sbXIDnTp_szQQ,14443
 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=F6oYrmNp29SrYHJeoe7UIzQuXdZ6fVyRMgdWhGo6oCs,5905
+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=Oqldfdo8MSGv_KB8-4EztUVcqycoabrVLSbfky2n5s4,10435
+iatoolkit/services/configuration_service.py,sha256=TsrTzou2Z8rC50favIlPgAloHBcy2p1jlZjRNtg3cq8,24184
+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=UuFWdjobFqwsBuLwAqi4FRNaVuYabSIjSj4vPKxZzLk,17098
 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=WEaPMRxginbsxOXopofPj6FeH-iIrSTlipFFRV20t4I,7069
 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=nafxpDdv1AbOYMeHzoiD6G8V64TPMniUlLhrJ0J_KOU,7802
+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,37 @@ 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=8tLQlpCS7Nd1-07kefIZzv4S5Hcd1UIhQmC5P_Z760s,15685
 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/configuration_api_view.py,sha256=K8aclK6oKkoTZoYwck-mozLchEOFCkrXFtraHh7tI00,6370
 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 +113,21 @@ 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_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.7.0.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
+iatoolkit-1.7.0.dist-info/licenses/LICENSE_COMMUNITY.md,sha256=9lVNcggPNUnleMF3_h3zu9PbNTeaRqB1tHAbiBLQJZU,566
+iatoolkit-1.7.0.dist-info/METADATA,sha256=V0ZDSh08B6jbggfwSY10wfIt7oDiFVWHuPTg-Dbkg30,8395
+iatoolkit-1.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-1.7.0.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-1.7.0.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

iatoolkit/services/search_service.py

@@ -1,55 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-from iatoolkit.repositories.vs_repo import VSRepo
-from iatoolkit.repositories.document_repo import DocumentRepo
-from iatoolkit.repositories.profile_repo import ProfileRepo
-from iatoolkit.repositories.models import Company
-from injector import inject
-
-
-class SearchService:
-    @inject
-    def __init__(self,
-                 profile_repo: ProfileRepo,
-                 doc_repo: DocumentRepo,
-                 vs_repo: VSRepo):
-        self.profile_repo = profile_repo
-        self.vs_repo = vs_repo
-        self.doc_repo = doc_repo
-
-    def search(self, company_short_name: str, query: str, metadata_filter: dict = None) -> str:
-        """
-        Performs a semantic search for a given query within a company's documents.
-
-        This method queries the vector store for relevant documents based on the
-        provided query text. It then constructs a formatted string containing the
-        content of the retrieved documents, which can be used as context for an LLM.
-
-        Args:
-            company_short_name: The  company to search within.
-            query: The text query to search for.
-            metadata_filter: An optional dictionary to filter documents by their metadata.
-
-        Returns:
-            A string containing the concatenated content of the found documents,
-            formatted to be used as a context.
-        """
-        company = self.profile_repo.get_company_by_short_name(company_short_name)
-        if not company:
-            return f"error: company {company_short_name} not found"
-
-        document_list = self.vs_repo.query(company_short_name=company_short_name,
-                                           query_text=query,
-                                           metadata_filter=metadata_filter)
-
-        search_context = ''
-        for doc in document_list:
-            search_context += f'documento "{doc.filename}"'
-            if doc.meta and 'document_type' in doc.meta:
-                search_context += f' tipo: {doc.meta.get('document_type', '')}'
-            search_context += f': {doc.content}\n'
-
-        return search_context