iatoolkit 0.4.1__py3-none-any.whl → 0.4.4__py3-none-any.whl

iatoolkit/base_company.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 # companies/base_company.py
 from abc import ABC, abstractmethod

iatoolkit/cli_commands.py

@@ -1,3 +1,8 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
 import click
 import logging
 from iatoolkit import IAToolkit

iatoolkit/company_registry.py

@@ -1,5 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from typing import Dict, Type, Any
 from .base_company import BaseCompany
@@ -19,19 +21,17 @@ class CompanyRegistry:
 
     def instantiate_companies(self, injector) -> Dict[str, BaseCompany]:
         """
-        Instancia todas las empresas registradas con inyección de dependencias.
-
-        Returns:
-            Dict con instancias de empresas {name: instance}
+        intantiate all registered companies using the toolkit injector
         """
-
         for company_key, company_class in self._company_classes.items():
             if company_key not in self._company_instances:
                 try:
                     # use de injector to create the instance
                     company_instance = injector.get(company_class)
+
+                    # save the created instance in the registry
                     self._company_instances[company_key] = company_instance
-                    logging.info(f"company '{company_key}' created in dispatcher")
+                    logging.info(f"company '{company_key}' instantiated")
 
                 except Exception as e:
                     logging.error(f"Error instanciando empresa {company_key}: {e}")
@@ -48,7 +48,6 @@ class CompanyRegistry:
         return self._company_classes.copy()
 
     def clear(self) -> None:
-        """Limpia el registro (útil para tests)"""
         self._company_classes.clear()
         self._company_instances.clear()
 

iatoolkit/iatoolkit.py

@@ -1,6 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit Core
-# Framework opensource para chatbots empresariales con IA
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from flask import Flask, url_for, current_app
 from flask_session import Session
@@ -90,7 +91,10 @@ class IAToolkit:
         # and other integrations, as views are handled manually.
         FlaskInjector(app=self.app, injector=self._injector)
 
-        # Step 6: Finalize setup within the application context
+        # Step 6: initialize dispatcher and registered compaies
+        self._init_dispatcher_and_company_instances()
+
+        # Step 7: Finalize setup within the application context
         self._setup_redis_sessions()
         self._setup_cors()
         self._setup_additional_services()
@@ -107,7 +111,7 @@ class IAToolkit:
         return self.app
 
     def _get_config_value(self, key: str, default=None):
-        """Obtiene un valor de configuración, primero del dict config, luego de env vars"""
+        # get a value from the config dict or the environment variable
         return self.config.get(key, os.getenv(key, default))
 
     def _setup_logging(self):
@@ -231,7 +235,7 @@ class IAToolkit:
         """⚙️ Configures all system dependencies."""
         try:
             # Core dependencies
-            binder.bind(Flask, to=self.app, scope=singleton)
+            binder.bind(Flask, to=self.app)
             binder.bind(DatabaseManager, to=self.db_manager, scope=singleton)
 
             # Bind all application components by calling the specific methods
@@ -285,7 +289,7 @@ class IAToolkit:
         binder.bind(LoadDocumentsService, to=LoadDocumentsService)
         binder.bind(ProfileService, to=ProfileService)
         binder.bind(JWTService, to=JWTService)
-        binder.bind(Dispatcher, to=Dispatcher, scope=singleton)
+        binder.bind(Dispatcher, to=Dispatcher)
 
     def _bind_infrastructure(self, binder: Binder):
         from infra.llm_client import llmClient
@@ -317,9 +321,19 @@ class IAToolkit:
     def _setup_additional_services(self):
         Bcrypt(self.app)
 
+    def _init_dispatcher_and_company_instances(self):
+        from iatoolkit.company_registry import get_company_registry
+        from services.dispatcher_service import Dispatcher
+
+        # instantiate all the registered companies
+        get_company_registry().instantiate_companies(self._injector)
+
+        # use the dispatcher to start the execution of every company
+        dispatcher = self._injector.get(Dispatcher)
+        dispatcher.start_execution()
+
     def _setup_cli_commands(self):
         from iatoolkit.cli_commands import register_core_commands
-        from services.dispatcher_service import Dispatcher
         from iatoolkit.company_registry import get_company_registry
 
         # 1. Register core commands
@@ -328,16 +342,10 @@ class IAToolkit:
 
         # 2. Register company-specific commands
         try:
-            # Get the dispatcher, which holds the company instances
-            dispatcher = self.get_injector().get(Dispatcher)
-            registry = get_company_registry()
-
             # Iterate through the registered company names
-            for company_name in registry.get_registered_companies():
-                company_instance = dispatcher.get_company_instance(company_name)
-                if company_instance:
-                    company_instance.register_cli_commands(self.app)
-                    logging.info(f"✅ Comandos CLI para la compañía '{company_name}' registrados.")
+            all_company_instances = get_company_registry().get_all_company_instances()
+            for company_name, company_instance in all_company_instances.items():
+                company_instance.register_cli_commands(self.app)
 
         except Exception as e:
             logging.error(f"❌ Error durante el registro de comandos de compañías: {e}")

{iatoolkit-0.4.1.dist-info → iatoolkit-0.4.4.dist-info}/METADATA

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

iatoolkit-0.4.4.dist-info/RECORD

@@ -0,0 +1,31 @@
+iatoolkit/__init__.py,sha256=GkFxAQHKPifz4Kd8M73Rc8TWRVIxjxkl1N0nsPvb_sU,1743
+iatoolkit/base_company.py,sha256=LyTnNqxfKKks0eu2DaisWFyMF11PetWNy5QiZe1nYRA,1938
+iatoolkit/cli_commands.py,sha256=XApFY4TrLjyyEt8L1ckY-Ro7AWFed8_tHh-aVsim3ww,3011
+iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
+iatoolkit/iatoolkit.py,sha256=WgBm34nAeVBCpfDDZD13Pos9xIoRx5viVwxjyYCXAmY,15790
+iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6VCHQq1p37rpu5wwk,3576
+iatoolkit/system_prompts/query_main.prompt,sha256=Eu5VOQzUygJ45Ct1WKYGbi0JMltgI6FQIZWlGmN1bdk,3214
+iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC9grYoOhRnrJo,59174
+services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
+services/api_service.py,sha256=7FxO9_g-gxuEmYIl5PhgJb7lHVgXJ1btHdH9fDrdhvY,2840
+services/benchmark_service.py,sha256=_ruKh9YzrTLtR0ZKrRNxqJQW0HdbwWuFz1gfLzJ9owA,5850
+services/dispatcher_service.py,sha256=pc_ksXM96c876LJ38c2kCLGIZlKRQcLahC66UruZN9c,14520
+services/document_service.py,sha256=r9EPjkhkjbAdfdZ0LCmUcvQbmGZR7BWgOsBiaeSmuoU,5890
+services/excel_service.py,sha256=ATPaeAvkLwQAkPZ3AKIUpO73RVyRg0D8c6i37_mcql0,3559
+services/file_processor_service.py,sha256=98yWYF7nIq1nm7nh6IzMmTKaOMTIeqCFWYwXVtV-ZJI,4102
+services/history_service.py,sha256=j0QCqcIIyw7DBy3GrZrEZNk0I4m-uuRoG5g0Z2RCcOE,1586
+services/jwt_service.py,sha256=YoZ9h7_o9xBko-arNQv4MbcwnxoSWVNj4VbZmMo_QGY,3908
+services/load_documents_service.py,sha256=XQ_Ck1pvj6TfJRIC1cNrWRbLmC0u7E0VJkNWC6L2gI4,7027
+services/mail_service.py,sha256=_67pctxZO46DHnWBip51ayuYtWd4bEoS1kg29ACO7_I,2162
+services/profile_service.py,sha256=vVRGNVJqByM-Drl17hmcDRnb-usQKJpiwlb733FfHXs,17851
+services/prompt_manager_service.py,sha256=wz5hhthLKt_tsj7WhsbuerXfpTCJhY4aIfjvCXREjs4,7690
+services/query_service.py,sha256=yITLcopc1stgzTkFdrBuF84k_XYsZyoKGue1GfXsb-U,15614
+services/search_service.py,sha256=bB3FWFxJi1iYsOdBxyu9tzIO406nQxcyeQzEowpmpjY,1803
+services/sql_service.py,sha256=s84K1ADlvMtum949wgMh8jsmqlOUeL-m_SWfAM4Wsv4,2141
+services/tasks_service.py,sha256=1DdbERlAxIkCpGEylnHDKC-KAsXRJugbaRSzRbPfL58,6790
+services/user_feedback_service.py,sha256=_LeNBYz4hHFapXfYTQVfkkD34gE8j2UeKnyOZ8H0nWo,2442
+services/user_session_context_service.py,sha256=GluNSgqP6W_hFke4oslSnfGnU_b-ph28BHH6jf3EIm0,3797
+iatoolkit-0.4.4.dist-info/METADATA,sha256=h3whrSrA15ia8BjgxLrhVSH9EqX24u9IFkbDgfgoWxc,9300
+iatoolkit-0.4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.4.4.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
+iatoolkit-0.4.4.dist-info/RECORD,,

services/__init__.py

@@ -1,5 +1,5 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 

services/api_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from infra.call_service import CallServiceClient
 from injector import inject

services/benchmark_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 import pandas as pd
 import time

services/dispatcher_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from common.exceptions import IAToolkitException
 from services.prompt_manager_service import PromptService
@@ -39,12 +39,6 @@ class Dispatcher:
         self._company_registry = None
         self._company_instances = None
 
-        # load into the dispatcher the configured companies
-        self.initialize_companies()
-
-        # run the statrtup logic for all companies
-        self.start_execution()
-
         self.tool_handlers = {
             "iat_generate_excel": self.excel_service.excel_generator,
             "iat_send_email": self.mail_service.send_mail,
@@ -66,25 +60,9 @@ class Dispatcher:
             self._company_instances = self.company_registry.get_all_company_instances()
         return self._company_instances
 
-    def initialize_companies(self):
-        from iatoolkit import current_iatoolkit
-        """
-        Initializes and instantiates all registered company classes.
-        This method should be called *after* the main injector is fully configured
-        and the company registry is populated.
-        """
-        if self.company_registry.get_all_company_instances():  # Check if already instantiated
-            return
-
-        # ✅ NOW it is safe to get the injector and instantiate companies.
-        injector = current_iatoolkit().get_injector()
-        self.company_registry.instantiate_companies(injector)
-
-
     def start_execution(self):
         """Runs the startup logic for all registered companies."""
         for company_name, company_instance in self.company_instances.items():
-            logging.info(f'Starting execution for company: {company_name}')
             company_instance.start_execution()
 
         return True
@@ -109,7 +87,7 @@ class Dispatcher:
                     prompt_name=prompt['name'],
                     description=prompt['description'],
                     order=1,
-                    is_system_prompt=True
+                    is_system_prompt=True,
                 )
                 i += 1
 
@@ -117,7 +95,7 @@ class Dispatcher:
         for company in self.company_instances.values():
             company.register_company()
 
-    def dispatch(self, company_name: str, action: str, **kwargs) -> str:
+    def dispatch(self, company_name: str, action: str, **kwargs) -> dict:
         company_key = company_name.lower()
 
         if company_key not in self.company_instances:
@@ -219,7 +197,7 @@ class Dispatcher:
         """
         Asegura que los datos del usuario siempre tengan una estructura consistente.
         """
-        # Valores por defecto para un perfil robusto
+        # default values
         normalized_user = {
             "id": raw_data.get("id", 0),
             "user_email": raw_data.get("email", ""),

services/document_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from docx import Document
 import fitz  # PyMuPDF
@@ -16,7 +16,7 @@ class DocumentService:
     @inject
     def __init__(self):
         # max number of pages to load
-        self.max_doc_pages = int(os.getenv("MAX_DOC_PAGES", "10"))
+        self.max_doc_pages = int(os.getenv("MAX_DOC_PAGES", "200"))
 
     def file_to_txt(self, filename, file_content):
         try:

services/excel_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from common.util import Utility
 import pandas as pd

services/file_processor_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from infra.connectors.file_connector import FileConnector
 import logging

services/history_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from injector import inject
 from repositories.llm_query_repo import LLMQueryRepo

services/jwt_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 import jwt
 import time

services/load_documents_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from repositories.vs_repo import VSRepo
 from repositories.document_repo import DocumentRepo
@@ -52,103 +52,16 @@ class LoadDocumentsService:
             separators=["\n\n", "\n", "."]
         )
 
-    # load the files for all of the companies.
-    def load(self, doc_type: str = None):
+    def load_company_files(self,
+                         company: Company,
+                         connector_config: Dict,
+                         predefined_metadata: Dict = None,
+                         filters: Dict = None):
         """
-                Loads documents for all companies based on their configuration.
-                It can load all document types or a specific one if provided.
-
-                Args:
-                    doc_type (str, optional): A specific document type to load.
-                                              If None, all configured types are loaded.
-
-                Returns:
-                    Dict: A dictionary with a summary message.
-                """
-        # doc_type: an optional document_type for loading
-        files_loaded = 0
-        companies = self.profile_repo.get_companies()
-
-        for company in companies:
-            load_config = company.parameters.get('load', {})
-            if not load_config:
-                continue
-
-            print(f"Cargando datos de ** {company.short_name} **")
-
-            # Si hay configuraciones de tipos de documento específicos
-            doc_types_config = load_config.get('document_types', {})
-
-            if doc_types_config and len(doc_types_config) > 0:
-                # Si se especificó un tipo de documento, cargar solo ese tipo
-                if doc_type and doc_type in doc_types_config:
-                    files_loaded += self._load_document_type(company, doc_type, doc_types_config[doc_type])
-                # Si no se especificó, cargar todos los tipos configurados
-                elif not doc_type:
-                    for type_name, type_config in doc_types_config.items():
-                        files_loaded += self._load_document_type(company, type_name, type_config)
-            else:
-                # Comportamiento anterior: usar la configuración general
-                connector = load_config.get('connector', {})
-                if not connector:
-                    raise IAToolkitException(IAToolkitException.ErrorType.MISSING_PARAMETER,
-                                       f"Falta configurar conector en empresa {company.short_name}")
-
-                files_loaded += self.load_data_source(company=company,
-                                                      connector_config=connector)
-
-        return {'message': f'{files_loaded} files processed'}
-
-    def load_company_files(self, company: Company,
-                           connector: dict,
-                           predefined_metadata: Dict = None,
-                           filters: Dict = None):
-        """
-        Loads all files for a specific company using a given connector.
+        Loads all the company files from a connector
 
         Args:
             company (Company): The company to load files for.
-            connector (dict): The connector configuration.
-
-        Returns:
-            Dict: A dictionary with a summary message.
-        """
-        if not connector:
-            raise IAToolkitException(IAToolkitException.ErrorType.MISSING_PARAMETER,
-                        f"Falta configurar conector")
-
-        files_loaded = self.load_data_source(
-                            company=company,
-                            connector_config=connector,
-                            predefined_metadata=predefined_metadata,
-                            filters=filters)
-
-        return {'message': f'{files_loaded} files processed'}
-
-    def _load_document_type(self, company: Company, doc_type_name: str, type_config: Dict) -> int:
-        # load specific document_types for a company
-        connector = type_config.get('connector')
-        if not connector:
-            logging.warning(f"Falta configurar conector para tipo {doc_type_name} en empresa {company.short_name}")
-            raise IAToolkitException(IAToolkitException.ErrorType.MISSING_PARAMETER,
-                               f"Falta configurar conector para tipo {doc_type_name} en empresa {company.short_name}")
-
-        # get the metadata for this connector
-        predefined_metadata = type_config.get('metadata', {})
-
-        # config specific filters
-        filters = type_config.get('filters', {"filename_contains": ".pdf"})
-
-        return self.load_data_source(company=company,
-                                     connector_config=connector,
-                                     predefined_metadata=predefined_metadata,
-                                     filters=filters)
-
-    def load_data_source(self, company: Company, connector_config: Dict, predefined_metadata: Dict = None, filters: Dict = None):
-        """
-        Loads files from a data source using a connector and a FileProcessor.
-
-        Args:
             connector_config (Dict): The configuration for the file connector.
             predefined_metadata (Dict, optional): Metadata to be added to all documents from this source.
             filters (Dict, optional): Filters to apply to the files.
@@ -156,6 +69,10 @@ class LoadDocumentsService:
         Returns:
             int: The number of processed files.
         """
+        if not connector_config:
+            raise IAToolkitException(IAToolkitException.ErrorType.MISSING_PARAMETER,
+                        f"Falta configurar conector")
+
         try:
             if not filters:
                 filters = {"filename_contains": ".pdf"}

services/mail_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from infra.mail_app import MailApp
 from injector import inject

services/profile_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from injector import inject
 from repositories.profile_repo import ProfileRepo

services/prompt_manager_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from injector import inject
 from repositories.llm_query_repo import LLMQueryRepo

services/query_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from infra.llm_client import llmClient
 from repositories.document_repo import DocumentRepo
@@ -48,7 +48,7 @@ class QueryService:
         self.session_context = session_context
         self.llm_client = llm_client
 
-        # Obtener el modelo de las variables de entorno
+        # get the model from the environment variable
         self.model = os.getenv("LLM_MODEL", "")
         if not self.model:
             raise IAToolkitException(IAToolkitException.ErrorType.API_KEY,

services/search_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from repositories.vs_repo import VSRepo
 from repositories.document_repo import DocumentRepo
@@ -18,6 +18,22 @@ class SearchService:
         self.doc_repo = doc_repo
 
     def search(self, company_id:  int, 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_id: The ID of 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.
+        """
         document_list = self.vs_repo.query(company_id=company_id,
                                            query_text=query,
                                            metadata_filter=metadata_filter)

services/sql_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from repositories.database_manager import DatabaseManager
 from common.util import Utility
@@ -17,6 +17,23 @@ class SqlService:
         self.util = util
 
     def exec_sql(self, db_manager: DatabaseManager, sql_statement: str) -> str:
+        """
+        Executes a raw SQL statement and returns the result as a JSON string.
+
+        This method takes a DatabaseManager instance and a SQL query, executes it
+        against the database, and fetches all results. The results are converted
+        into a list of dictionaries, where each dictionary represents a row.
+        This list is then serialized to a JSON string.
+        If an exception occurs during execution, the transaction is rolled back,
+        and a custom IAToolkitException is raised.
+
+        Args:
+            db_manager: The DatabaseManager instance to get the database session from.
+            sql_statement: The raw SQL statement to be executed.
+
+        Returns:
+            A JSON string representing the list of rows returned by the query.
+        """
         try:
             # here the SQL is executed
             result = db_manager.get_session().execute(text(sql_statement))

services/tasks_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from injector import inject
 from repositories.models import Task, TaskStatus

services/user_feedback_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from repositories.models import UserFeedback
 from injector import inject

services/user_session_context_service.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
 
 from infra.redis_session_manager import RedisSessionManager
 from typing import List, Dict, Optional

iatoolkit/system_prompts/arquitectura.prompt

@@ -1,32 +0,0 @@
-Mi app es un chatbot multiempresa, que se conecta
-con datos empresariales por sql y llamadas a endpoints.
-
-Es principalmente un backend, pero tambien puede actuar como frontend.
-Cuando actua como backend, los usuarios se identifican como external_user_id
-y el front lo informa en la vista external_login_view.
-
-Cuando actua como front y back,
-existe un acceso que simula un usuario externo a través de
-public_chat_view.
-
-El otro mecanismo de acceso es con el sistema de usuarios
-integrado a mi app, con tablas propias: user, company, user_company
-y el acceso se hace a través de profile_service.
-
-Estos accesos sirven para autentificar las consultas.
-Cada vez que hay un acceso en estos puntos, se llama a query_service
-para que construya el contexto de la empresa y se lo envie al
-llm.
-
-Se guarda en session de redis, el response.id generado
-para linkearlo con las consultas futuras (responses api de openai).
-
-Por otra parte, las queries ingresan a través de llm_query_view,
-el que autentifica al solicitante y la dirige a query_service.
-
-Query_service gestiona los prompts que pueden ser solicitados
-y llama a invoke en openai_client, quien se encarga
-de interactuar con el llm.
-
-Una vez terminada la interacción, se graba un registro en la
-tabla llm_queries.

iatoolkit-0.4.1.dist-info/RECORD

@@ -1,32 +0,0 @@
-iatoolkit/__init__.py,sha256=GkFxAQHKPifz4Kd8M73Rc8TWRVIxjxkl1N0nsPvb_sU,1743
-iatoolkit/base_company.py,sha256=FlB-HFYH8FoTl4nbtsYgfKjkdZtizJbKwXqaosxmRqc,2009
-iatoolkit/cli_commands.py,sha256=CyaabHA3HdKd-eIqrJD8IQFT7Tqn_BEdi4jb1utisMo,2909
-iatoolkit/company_registry.py,sha256=KOUzJHLYgzMAV6BxkTiDPlN_ME4fktp7yRzKLlXZ5-w,2597
-iatoolkit/iatoolkit.py,sha256=OwlGujwtNLBYtfZuCpcX_yzrgB8BVo9Jfh72owM8FFc,15651
-iatoolkit/system_prompts/arquitectura.prompt,sha256=2W-7NWy6P6y1Gh5_-zD1iK-BWq1Siu8TuvGCouP67bQ,1267
-iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6VCHQq1p37rpu5wwk,3576
-iatoolkit/system_prompts/query_main.prompt,sha256=Eu5VOQzUygJ45Ct1WKYGbi0JMltgI6FQIZWlGmN1bdk,3214
-iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC9grYoOhRnrJo,59174
-services/__init__.py,sha256=fSvSfIcPW1dHwTBY1hQ5dBEhaoorzk_GzR4G46gD8tY,173
-services/api_service.py,sha256=InIKTc64BWcp4U4tYKHz28x4ErPxIfvR9x3ZlxJZlXs,2911
-services/benchmark_service.py,sha256=0Vgsx_FaUZL7igoBYbe1AZkIWOiEUx1FSCV_0Ut0mtk,5921
-services/dispatcher_service.py,sha256=AOLsv9haFOxW5hV-6LOP8tryDdlfvjEGhNNGjgphH2c,15489
-services/document_service.py,sha256=sm5QtbrKs2dF9hpLuSLMB-IMWYNBD7yWHv3rd80aD0o,5960
-services/excel_service.py,sha256=wE9Udbyb96kGRSnZZ6KM2mbE484rKjTEhta9GKKpy-8,3630
-services/file_processor_service.py,sha256=0CM4CQu6KKfcLVGkxs4hYxgdz8kKRWfkV5rDH9UoccM,4173
-services/history_service.py,sha256=6fGSSWxy60nxtkwp_fodwDHoVKhpIUbHnzAzUSiNi-Y,1657
-services/jwt_service.py,sha256=dC45Sn6FyzdzRiQJnzgkjN3Hy21V1imRxB0hTyWRvlA,3979
-services/load_documents_service.py,sha256=AtB5VnSjOEILG3wcQm6YiXkRrVvL87JWeRPgon81hf4,11064
-services/mail_service.py,sha256=ystFit1LuYUC4ekYYebyiy1rqYQmxeL6K8h58MxEkOY,2233
-services/profile_service.py,sha256=vZV0cregZQiPKYcNLaD7xjez2y6-3Mq97cDndC8NL8w,17922
-services/prompt_manager_service.py,sha256=bWG4SIgt0u45PVUfm0xRLbLfKC7bk6uozVHRdkdgCmc,7761
-services/query_service.py,sha256=zpaDzjzh2HqAG1F2Ap8WHBpfAMVtzZ_6v1JGVEguwvs,15687
-services/search_service.py,sha256=oJD6WRXCJBD7WUVHWWKxexRkhR8nQSrFtcPV3pFO2KQ,1153
-services/sql_service.py,sha256=H7CIPpXTcxLXLojD2fBFr_mIAD0PW1vEJhKHLfJi4Hk,1418
-services/tasks_service.py,sha256=hHJDlcsSOPtEleD6_Vv3pocfxWNmthIhmZSdnoWFpEM,6861
-services/user_feedback_service.py,sha256=YtCndRBekDEWYEbac431Ksn2gMO5iBrI3WqKK0xtShE,2513
-services/user_session_context_service.py,sha256=5qn7fqpuiU8KgMpU4M5-iRUsETumz1raBw-EeZLuE1A,3868
-iatoolkit-0.4.1.dist-info/METADATA,sha256=rzRfT16CeRWOOnJorFb5kNJ3W0U98yKb1FZiB_L6p8M,9300
-iatoolkit-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.4.1.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
-iatoolkit-0.4.1.dist-info/RECORD,,