iatoolkit 0.4.2__py3-none-any.whl → 0.66.2__py3-none-any.whl

iatoolkit/__init__.py

@@ -4,33 +4,22 @@ IAToolkit Package
 
 # Expose main classes and functions at the top level of the package
 
-# Assuming 'toolkit.py' contains the IAToolkit class
-from .iatoolkit import IAToolkit, create_app
-from .iatoolkit import current_iatoolkit
+# main IAToolkit class
+from .iatoolkit import IAToolkit, current_iatoolkit, create_app
 
-# Assuming 'app_factory.py' contains create_app and register_company
+# for registering the client companies
 from .company_registry import register_company
-
-# Assuming 'base_company.py' contains BaseCompany
 from .base_company import BaseCompany
+from iatoolkit.repositories.database_manager import DatabaseManager
 
 # --- Services ---
-# Assuming they are in a 'services' sub-package
-from services.sql_service import SqlService
-from services.excel_service import ExcelService
-from services.dispatcher_service import Dispatcher
-from services.document_service import DocumentService
-from services.search_service import SearchService
-from services.load_documents_service import LoadDocumentsService
-from repositories.profile_repo import ProfileRepo
-from repositories.llm_query_repo import LLMQueryRepo
-from services.query_service import QueryService
-from services.prompt_manager_service import PromptService
-from repositories.database_manager import DatabaseManager
-from infra.call_service import CallServiceClient
-from common.util import Utility
-from repositories.models import Base, Company, Function, TaskType, Prompt, PromptCategory
-
+from iatoolkit.services.query_service import QueryService
+from iatoolkit.services.sql_service import SqlService
+from iatoolkit.services.document_service import DocumentService
+from iatoolkit.services.search_service import SearchService
+from iatoolkit.services.load_documents_service import LoadDocumentsService
+from iatoolkit.services.excel_service import ExcelService
+from iatoolkit.infra.call_service import CallServiceClient
 
 __all__ = [
     'IAToolkit',
@@ -38,23 +27,12 @@ __all__ = [
     'current_iatoolkit',
     'register_company',
     'BaseCompany',
+    'DatabaseManager',
+    'QueryService',
     'SqlService',
     'ExcelService',
-    'Dispatcher',
     'DocumentService',
     'SearchService',
-    'QueryService',
     'LoadDocumentsService',
-    'ProfileRepo',
-    'LLMQueryRepo',
-    'PromptService',
-    'DatabaseManager',
     'CallServiceClient',
-    'Utility',
-    'Company',
-    'Function',
-    'TaskType',
-    'Base',
-    'Prompt',
-    'PromptCategory'
 ]

iatoolkit/base_company.py

@@ -1,17 +1,82 @@
 # 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
-from typing import Any
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
+
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.repositories.models import Company, Function, PromptCategory
+from iatoolkit import IAToolkit
 
 
 class BaseCompany(ABC):
-    def __init__(self, profile_repo: Any = None, llm_query_repo: Any = None):
-        self.profile_repo = profile_repo
-        self.llm_query_repo = llm_query_repo
+    def __init__(self):
+        # Obtener el inyector global y resolver las dependencias internamente
+        injector = IAToolkit.get_instance().get_injector()
+        self.profile_repo: ProfileRepo = injector.get(ProfileRepo)
+        self.llm_query_repo: LLMQueryRepo = injector.get(LLMQueryRepo)
+        self.prompt_service: PromptService = injector.get(PromptService)
+        self.company: Company | None = None
+
+    def _load_company_by_short_name(self, short_name: str) -> Company:
+        self.company = self.profile_repo.get_company_by_short_name(short_name)
+        return self.company
+
+    def _create_company(self,
+                        short_name: str,
+                        name: str,
+                        parameters: dict | None = None,
+                        branding: dict | None = None,
+                        onboarding_cards: dict | None = None,
+                        ) -> Company:
+        company_obj = Company(short_name=short_name,
+                              name=name,
+                              parameters=parameters,
+                              branding=branding,
+                              onboarding_cards=onboarding_cards)
+        self.company = self.profile_repo.create_company(company_obj)
+        return self.company
+
+    def _create_function(self, function_name: str, description: str, params: dict, **kwargs):
+        if not self.company:
+            raise ValueError("La compañía debe estar definida antes de crear una función.")
+
+        self.llm_query_repo.create_or_update_function(
+            Function(
+                company_id=self.company.id,
+                name=function_name,
+                description=description,
+                parameters=params,
+                system_function=False,
+                **kwargs
+            )
+        )
+
+    def _create_prompt_category(self, name: str, order: int) -> PromptCategory:
+        if not self.company:
+            raise ValueError("La compañía debe estar definida antes de crear una categoría.")
+
+        return self.llm_query_repo.create_or_update_prompt_category(
+            PromptCategory(name=name, order=order, company_id=self.company.id)
+        )
+
+    def _create_prompt(self, prompt_name: str, description: str, category: PromptCategory, order: int, **kwargs):
+        if not self.company:
+            raise ValueError("La compañía debe estar definida antes de crear un prompt.")
+
+        self.prompt_service.create_prompt(
+            prompt_name=prompt_name,
+            description=description,
+            order=order,
+            company=self.company,
+            category=category,
+            **kwargs
+        )
+
 
     @abstractmethod
     # initialize all the database tables  needed
@@ -25,7 +90,7 @@ class BaseCompany(ABC):
 
     @abstractmethod
     # get context specific for this company
-    def get_user_info(self, **kwargs) -> str:
+    def get_user_info(self, user_identifier: str) -> dict:
         raise NotImplementedError("La subclase debe implementar el método get_user_info()")
 
     @abstractmethod
@@ -49,5 +114,6 @@ class BaseCompany(ABC):
         """
         pass
 
+
     def unsupported_operation(self, tag):
         raise NotImplementedError(f"La operación '{tag}' no está soportada por esta empresa.")

iatoolkit/cli_commands.py

@@ -1,31 +1,21 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
 import click
 import logging
 from iatoolkit import IAToolkit
-from services.dispatcher_service import Dispatcher
-from services.profile_service import ProfileService
+from iatoolkit.services.profile_service import ProfileService
 
 def register_core_commands(app):
     """Registra los comandos CLI del núcleo de IAToolkit."""
-    
-    @app.cli.command("setup-all-companies")
-    def setup_all_companies():
-        """🗄️ Inicializa todas las compañías registradas en la base de datos."""
-        try:
-            dispatcher = IAToolkit.get_instance().get_injector().get(Dispatcher)
-            click.echo("🚀 Inicializando base de datos y compañías...")
-            dispatcher.setup_all_companies()
-            click.echo("✅ Base de datos y compañías inicializadas correctamente.")
-        except Exception as e:
-            logging.exception(e)
-            click.echo(f"❌ Error: {e}")
 
-    @app.cli.command("setup-company")
+    @app.cli.command("api-key")
     @click.argument("company_short_name")
-    def setup_company(company_short_name: str):
+    def api_key(company_short_name: str):
         """⚙️ Genera una nueva API key para una compañía ya registrada."""
         try:
-            dispatcher = IAToolkit.get_instance().get_injector().get(Dispatcher)
-            dispatcher.setup_all_companies()
             profile_service = IAToolkit.get_instance().get_injector().get(ProfileService)
             click.echo(f"🔑 Generando API key para '{company_short_name}'...")
             result = profile_service.new_api_key(company_short_name)
@@ -35,20 +25,22 @@ def register_core_commands(app):
                 click.echo("👉 Asegúrate de que el nombre de la compañía es correcto y está registrada.")
             else:
                 click.echo("✅ ¡Configuración lista! Agrega esta variable a tu entorno:")
-                click.echo(f"IATOOLKIT_API_KEY={result['api-key']}")
+                click.echo(f"IATOOLKIT_API_KEY='{result['api-key']}'")
         except Exception as e:
             logging.exception(e)
             click.echo(f"❌ Ocurrió un error inesperado durante la configuración: {e}")
 
     @app.cli.command("encrypt-key")
     @click.argument("key")
-    def api_key(key: str):
-        from common.util import Utility
+    def encrypt_llm_api_key(key: str):
+        from iatoolkit.common.util import Utility
+
+
 
         util = IAToolkit.get_instance().get_injector().get(Utility)
         try:
             encrypt_key = util.encrypt_key(key)
-            click.echo(f'la clave encriptada es: {encrypt_key} \n')
+            click.echo(f'la api-key del LLM encriptada es: {encrypt_key} \n')
         except Exception as e:
             logging.exception(e)
             click.echo(f"Error: {str(e)}")
@@ -56,7 +48,7 @@ def register_core_commands(app):
     @app.cli.command("exec-tasks")
     @click.argument("company_short_name")
     def exec_pending_tasks(company_short_name: str):
-        from services.tasks_service import TaskService
+        from iatoolkit.services.tasks_service import TaskService
         task_service = IAToolkit.get_instance().get_injector().get(TaskService)
 
         try:

iatoolkit/common/exceptions.py

@@ -0,0 +1,46 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from enum import Enum
+
+
+class IAToolkitException(Exception):
+
+    class ErrorType(Enum):
+        SYSTEM_ERROR = 0
+        DATABASE_ERROR = 1
+        LLM_ERROR = 2
+        CLOUD_STORAGE_ERROR = 3
+        DOCUMENT_NOT_FOUND = 4
+        INVALID_PARAMETER = 5
+        MISSING_PARAMETER = 6
+        PARAM_NOT_FILLED = 7
+        PERMISSION = 8
+        EXIST = 9
+        API_KEY = 10
+        CALL_ERROR = 11
+        PROMPT_ERROR = 12
+        FILE_FORMAT_ERROR = 13
+        FILE_IO_ERROR = 14
+        TEMPLATE_ERROR = 15
+        EXTERNAL_SOURCE_ERROR = 16
+        MAIL_ERROR = 17
+        CONFIG_ERROR = 18
+        INVALID_NAME = 19
+        REQUEST_ERROR = 20
+        TASK_EXECUTION_ERROR = 21
+        TASK_NOT_FOUND = 22
+        INVALID_STATE = 23
+        CRYPT_ERROR = 24
+        LOAD_DOCUMENT_ERROR = 25
+        INVALID_USER = 26
+        VECTOR_STORE_ERROR = 27
+
+
+
+    def __init__(self, error_type: ErrorType = ErrorType.SYSTEM_ERROR, message=None):
+        self.error_type = error_type
+        self.message = message
+        super().__init__(self.message)

iatoolkit/common/routes.py

@@ -0,0 +1,141 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import render_template, redirect, url_for,send_from_directory, current_app, abort
+from flask import jsonify
+
+
+# this function register all the views
+def register_views(injector, app):
+
+    from iatoolkit.views.index_view import IndexView
+    from iatoolkit.views.init_context_api_view import InitContextApiView
+    from iatoolkit.views.llmquery_api_view import LLMQueryApiView
+    from iatoolkit.views.tasks_api_view import TaskApiView
+    from iatoolkit.views.tasks_review_api_view import TaskReviewApiView
+    from iatoolkit.views.login_simulation_view import LoginSimulationView
+    from iatoolkit.views.signup_view import SignupView
+    from iatoolkit.views.verify_user_view import VerifyAccountView
+    from iatoolkit.views.forgot_password_view import ForgotPasswordView
+    from iatoolkit.views.change_password_view import ChangePasswordView
+    from iatoolkit.views.file_store_api_view import FileStoreApiView
+    from iatoolkit.views.user_feedback_api_view import UserFeedbackApiView
+    from iatoolkit.views.prompt_api_view import PromptApiView
+    from iatoolkit.views.history_api_view import HistoryApiView
+    from iatoolkit.views.help_content_api_view import HelpContentApiView
+    from iatoolkit.views.profile_api_view import UserLanguageApiView  # <-- Importa la nueva vista
+
+    from iatoolkit.views.login_view import LoginView, FinalizeContextView
+    from iatoolkit.views.external_login_view import ExternalLoginView, RedeemTokenApiView
+    from iatoolkit.views.logout_api_view import LogoutApiView
+    from iatoolkit.views.home_view import HomeView
+
+    # iatoolkit home page
+    app.add_url_rule('/', view_func=IndexView.as_view('index'))
+
+    # company home view
+    app.add_url_rule('/<company_short_name>/home', view_func=HomeView.as_view('home'))
+
+    # login for the iatoolkit integrated frontend
+    app.add_url_rule('/<company_short_name>/login', view_func=LoginView.as_view('login'))
+
+    # this is the login for external users
+    app.add_url_rule('/<company_short_name>/external_login',
+                     view_func=ExternalLoginView.as_view('external_login'))
+
+    # this endpoint is called when onboarding_shell finish the context load
+    app.add_url_rule(
+        '/<company_short_name>/finalize',
+        view_func=FinalizeContextView.as_view('finalize_no_token')
+    )
+
+    app.add_url_rule(
+        '/<company_short_name>/finalize/<token>',
+        view_func=FinalizeContextView.as_view('finalize_with_token')
+    )
+
+    app.add_url_rule(
+        '/api/profile/language',
+        view_func=UserLanguageApiView.as_view('user_language_api')
+    )
+
+    # logout
+    app.add_url_rule('/<company_short_name>/api/logout',
+                     view_func=LogoutApiView.as_view('logout'))
+
+    # this endpoint is called by the JS for changing the token for a session
+    app.add_url_rule('/<string:company_short_name>/api/redeem_token',
+                     view_func = RedeemTokenApiView.as_view('redeem_token'))
+
+    # init (reset) the company context
+    app.add_url_rule('/<company_short_name>/api/init-context',
+                     view_func=InitContextApiView.as_view('init-context'),
+                     methods=['POST', 'OPTIONS'])
+
+    # register new user, account verification and forgot password
+    app.add_url_rule('/<company_short_name>/signup',view_func=SignupView.as_view('signup'))
+    app.add_url_rule('/<company_short_name>/verify/<token>', view_func=VerifyAccountView.as_view('verify_account'))
+    app.add_url_rule('/<company_short_name>/forgot-password', view_func=ForgotPasswordView.as_view('forgot_password'))
+    app.add_url_rule('/<company_short_name>/change-password/<token>', view_func=ChangePasswordView.as_view('change_password'))
+
+    # main chat query, used by the JS in the browser (with credentials)
+    # can be used also for executing iatoolkit prompts
+    app.add_url_rule('/<company_short_name>/api/llm_query', view_func=LLMQueryApiView.as_view('llm_query_api'))
+
+    # open the promt directory
+    app.add_url_rule('/<company_short_name>/api/prompts', view_func=PromptApiView.as_view('prompt'))
+
+    # toolbar buttons
+    app.add_url_rule('/<company_short_name>/api/feedback', view_func=UserFeedbackApiView.as_view('feedback'))
+    app.add_url_rule('/<company_short_name>/api/history', view_func=HistoryApiView.as_view('history'))
+    app.add_url_rule('/<company_short_name>/api/help-content', view_func=HelpContentApiView.as_view('help-content'))
+
+    # tasks management endpoints: create task, and review answer
+    app.add_url_rule('/tasks', view_func=TaskApiView.as_view('tasks'))
+    app.add_url_rule('/tasks/review/<int:task_id>', view_func=TaskReviewApiView.as_view('tasks-review'))
+
+    # this endpoint is for upload documents into the vector store (api-key)
+    app.add_url_rule('/api/load', view_func=FileStoreApiView.as_view('load_api'))
+
+
+    @app.route('/download/<path:filename>')
+    def download_file(filename):
+        """
+        Esta vista sirve un archivo previamente generado desde el directorio
+        configurado en IATOOLKIT_DOWNLOAD_DIR.
+        """
+        # Valida que la configuración exista
+        if 'IATOOLKIT_DOWNLOAD_DIR' not in current_app.config:
+            abort(500, "Error de configuración: IATOOLKIT_DOWNLOAD_DIR no está definido.")
+
+        download_dir = current_app.config['IATOOLKIT_DOWNLOAD_DIR']
+
+        try:
+            return send_from_directory(
+                download_dir,
+                filename,
+                as_attachment=True  # Fuerza la descarga en lugar de la visualización
+            )
+        except FileNotFoundError:
+            abort(404)
+
+    # login testing
+    app.add_url_rule('/<company_short_name>/login_test',
+                     view_func=LoginSimulationView.as_view('login_test'))
+
+    app.add_url_rule(
+        '/about',  # URL de la ruta
+        view_func=lambda: render_template('about.html'))
+
+    app.add_url_rule('/version', 'version',
+                     lambda: jsonify({"iatoolkit_version": current_app.config.get('VERSION', 'N/A')}))
+
+
+    # hacer que la raíz '/' vaya al home de iatoolkit
+    @app.route('/')
+    def root_redirect():
+        return redirect(url_for('index'))
+
+

iatoolkit/common/session_manager.py

@@ -0,0 +1,24 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import session
+
+class SessionManager:
+    @staticmethod
+    def set(key, value):
+        session[key] = value
+
+    @staticmethod
+    def get(key, default=None):
+        return session.get(key, default)
+
+    @staticmethod
+    def remove(key):
+        if key in session:
+            session.pop(key)
+
+    @staticmethod
+    def clear():
+        session.clear()