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

iatoolkit/__init__.py

@@ -1,14 +1,17 @@
-"""
-IAToolkit Package
-"""
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+__version__ = "0.91.1"
 
 # Expose main classes and functions at the top level of the package
 
 # main IAToolkit class
-from .iatoolkit import IAToolkit, create_app, current_iatoolkit
+from iatoolkit.core import IAToolkit, create_app, current_iatoolkit
 
 # for registering the client companies
-from .company_registry import register_company
+from .company_registry import register_company, set_company_registry
 from .base_company import BaseCompany
 
 # --- Services ---
@@ -18,12 +21,16 @@ from iatoolkit.services.search_service import SearchService
 from iatoolkit.services.sql_service import SqlService
 from iatoolkit.services.load_documents_service import LoadDocumentsService
 from iatoolkit.infra.call_service import CallServiceClient
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.mail_service import MailService
+from iatoolkit.repositories.models import Base as OrmModel
 
 __all__ = [
     'IAToolkit',
     'create_app',
     'current_iatoolkit',
     'register_company',
+    'set_company_registry',
     'BaseCompany',
     'QueryService',
     'SqlService',
@@ -31,4 +38,7 @@ __all__ = [
     'SearchService',
     'LoadDocumentsService',
     'CallServiceClient',
+    'ProfileService',
+    'MailService',
+    'OrmModel'
 ]

iatoolkit/base_company.py

@@ -7,9 +7,8 @@
 from abc import ABC, abstractmethod
 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
+from iatoolkit.repositories.models import Company
+from iatoolkit.core import IAToolkit
 
 
 class BaseCompany(ABC):
@@ -18,74 +17,21 @@ class BaseCompany(ABC):
         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
-        self.company_short_name: str
+        self.company_short_name = ''
 
-    def _create_company(self,
-                        short_name: str,
-                        name: str,
-                        parameters: dict | None = None,
-                        ) -> Company:
-        company_obj = Company(short_name=short_name,
-                              name=name,
-                              parameters=parameters)
-        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
-    # get context specific for this company
-    def get_user_info(self, user_identifier: str) -> dict:
-        raise NotImplementedError("La subclase debe implementar el método get_user_info()")
 
     @abstractmethod
     # execute the specific action configured in the intent table
     def handle_request(self, tag: str, params: dict) -> dict:
         raise NotImplementedError("La subclase debe implementar el método handle_request()")
 
-
+    @abstractmethod
     def register_cli_commands(self, app):
         """
         optional method for a company definition of it's cli commands
         """
         pass
 
-
     def unsupported_operation(self, tag):
         raise NotImplementedError(f"La operación '{tag}' no está soportada por esta empresa.")

iatoolkit/cli_commands.py

@@ -5,9 +5,10 @@
 
 import click
 import logging
-from .iatoolkit import IAToolkit
+from iatoolkit.core import IAToolkit
 from iatoolkit.services.profile_service import ProfileService
 
+
 def register_core_commands(app):
     """Registra los comandos CLI del núcleo de IAToolkit."""
 
@@ -17,26 +18,24 @@ def register_core_commands(app):
         """⚙️ Genera una nueva API key para una compañía ya registrada."""
         try:
             profile_service = IAToolkit.get_instance().get_injector().get(ProfileService)
-            click.echo(f"🔑 Generando API key para '{company_short_name}'...")
+            click.echo(f"🔑 Generating API-KEY for company: '{company_short_name}'...")
             result = profile_service.new_api_key(company_short_name)
 
             if 'error' in result:
                 click.echo(f"❌ Error: {result['error']}")
-                click.echo("👉 Asegúrate de que el nombre de la compañía es correcto y está registrada.")
+                click.echo("👉 Make sure the company is registered and valid.")
             else:
-                click.echo("✅ ¡Configuración lista! Agrega esta variable a tu entorno:")
+                click.echo("✅ ¡Api-key is ready! add this variable to your environment:")
                 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}")
+            click.echo(f"❌ unexpectd error during the configuration: {e}")
 
     @app.cli.command("encrypt-key")
     @click.argument("key")
     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)

iatoolkit/common/exceptions.py

@@ -37,6 +37,7 @@ class IAToolkitException(Exception):
         LOAD_DOCUMENT_ERROR = 25
         INVALID_USER = 26
         VECTOR_STORE_ERROR = 27
+        EMBEDDING_ERROR = 28
 
 
 

iatoolkit/common/routes.py

@@ -8,32 +8,31 @@ from flask import jsonify
 
 
 # this function register all the views
-def register_views(injector, app):
+def register_views(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.load_document_api_view import LoadDocumentApiView
     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.profile_api_view import UserLanguageApiView
     from iatoolkit.views.embedding_api_view import EmbeddingApiView
     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
+    from iatoolkit.views.static_page_view import StaticPageView
+    from iatoolkit.views.root_redirect_view import RootRedirectView
 
-    # iatoolkit home page
-    app.add_url_rule('/', view_func=IndexView.as_view('index'))
+    # assign root '/' to our new redirect logic
+    app.add_url_rule('/home', view_func=RootRedirectView.as_view('root_redirect'))
 
     # company home view
     app.add_url_rule('/<company_short_name>/home', view_func=HomeView.as_view('home'))
@@ -41,10 +40,6 @@ def register_views(injector, app):
     # 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',
@@ -65,10 +60,6 @@ def register_views(injector, app):
     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'),
@@ -97,12 +88,16 @@ def register_views(injector, app):
     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.add_url_rule('/api/load-document', view_func=LoadDocumentApiView.as_view('load-document'), methods=['POST'])
 
     # this endpoint is for generating embeddings for a given text
     app.add_url_rule('/<company_short_name>/api/embedding',
                      view_func=EmbeddingApiView.as_view('embedding_api'))
 
+    # static pages
+    # url: /pages/foundation o /pages/implementation_plan
+    static_view = StaticPageView.as_view('static_pages')
+    app.add_url_rule('/pages/<page_name>', view_func=static_view, methods=['GET'])
 
     @app.route('/download/<path:filename>')
     def download_file(filename):
@@ -125,21 +120,10 @@ def register_views(injector, app):
         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/util.py

@@ -6,6 +6,7 @@
 import logging
 from typing import List
 from iatoolkit.common.exceptions import IAToolkitException
+from flask import request
 from injector import inject
 import os
 from jinja2 import Environment, FileSystemLoader
@@ -16,6 +17,7 @@ from cryptography.fernet import Fernet
 import base64
 
 
+
 class Utility:
     @inject
     def __init__(self):
@@ -97,6 +99,10 @@ class Utility:
             logging.exception(e)
             return None
 
+    def get_template_by_language(self, template_name: str, default_langueage: str = 'en') -> str:
+        # english is default
+        lang = request.args.get("lang", default_langueage)
+        return f'{template_name}_{lang}.html'
 
     def serialize(self, obj):
         if isinstance(obj, datetime) or isinstance(obj, date):
@@ -343,6 +349,6 @@ class Utility:
 
     def is_gemini_model(self, model: str) -> bool:
         gemini_models = [
-            'gemini', 'gemini-2.5-pro'
+            'gemini', 'gemini-2.5-pro', 'gemini-3'
         ]
         return any(gemini_model in model.lower() for gemini_model in gemini_models)

iatoolkit/company_registry.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from typing import Dict, Type, Any
+from typing import Dict, Type, Any, Optional
 from .base_company import BaseCompany
 import logging
 
@@ -18,6 +18,30 @@ class CompanyRegistry:
         self._company_classes: Dict[str, Type[BaseCompany]] = {}
         self._company_instances: Dict[str, BaseCompany] = {}
 
+    def register(self, name: str, company_class: Type[BaseCompany]) -> None:
+        """
+        Registers a company in the registry.
+
+        COMMUNITY EDITION LIMITATION:
+        This base implementation enforces a strict single-tenant limit.
+        It raises a RuntimeError if a second company is registered.
+        """
+        if not issubclass(company_class, BaseCompany):
+            raise ValueError(f"The class {company_class.__name__} must be a subclass of BaseCompany")
+
+        company_key = name.lower()
+
+        # --- STRICT SINGLE-TENANT ENFORCEMENT ---
+        # If a company is already registered (and it's not an update to the same key)
+        if len(self._company_classes) > 0 and company_key not in self._company_classes:
+            logging.error(f"❌ Community Edition Restriction: Cannot register '{name}'. Limit reached (1).")
+            raise RuntimeError(
+                "IAToolkit Community Edition allows only one company instance. "
+                "Upgrade to IAToolkit Enterprise to enable multi-tenancy."
+            )
+
+        self._company_classes[company_key] = company_class
+        logging.info(f"Company registered: {name}")
 
     def instantiate_companies(self, injector) -> Dict[str, BaseCompany]:
         """
@@ -34,15 +58,16 @@ class CompanyRegistry:
 
                 except Exception as e:
                     logging.error(f"Error while creating company instance for {company_key}: {e}")
-                    logging.exception(e)
-                    raise
+                    raise e
 
         return self._company_instances.copy()
 
     def get_all_company_instances(self) -> Dict[str, BaseCompany]:
-        """Devuelve un diccionario con todas las instancias de empresas creadas."""
         return self._company_instances.copy()
 
+    def get_company_instance(self, company_name: str) -> Optional[BaseCompany]:
+        return self._company_instances.get(company_name.lower())
+
     def get_registered_companies(self) -> Dict[str, Type[BaseCompany]]:
         return self._company_classes.copy()
 
@@ -50,11 +75,30 @@ class CompanyRegistry:
         self._company_classes.clear()
         self._company_instances.clear()
 
+# --- Singleton Management ---
 
-# global instance of the company registry
+# Global instance (Default: Community Edition)
 _company_registry = CompanyRegistry()
 
 
+def get_company_registry() -> CompanyRegistry:
+    """Get the global company registry instance."""
+    return _company_registry
+
+
+def set_company_registry(registry: CompanyRegistry) -> None:
+    """
+    Sets the global company registry instance.
+    Use this to inject an Enterprise-compatible registry implementation.
+    """
+    global _company_registry
+    if not isinstance(registry, CompanyRegistry):
+        raise ValueError("Registry must inherit from CompanyRegistry")
+
+    _company_registry = registry
+    logging.info(f"✅ Company Registry implementation swapped: {type(registry).__name__}")
+
+
 def register_company(name: str, company_class: Type[BaseCompany]) -> None:
     """
     Public function to register a company.
@@ -63,13 +107,5 @@ def register_company(name: str, company_class: Type[BaseCompany]) -> None:
         name: Name of the company
         company_class: Class that inherits from BaseCompany
     """
-    if not issubclass(company_class, BaseCompany):
-        raise ValueError(f"La clase {company_class.__name__} debe heredar de BaseCompany")
-
-    company_key = name.lower()
-    _company_registry._company_classes[company_key] = company_class
+    _company_registry.register(name, company_class)
 
-
-def get_company_registry() -> CompanyRegistry:
-    """get the global company registry instance"""
-    return _company_registry