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

iatoolkit/{iatoolkit.py → core.py}

@@ -3,28 +3,26 @@
 #
 # IAToolkit is open source software.
 
-from flask import Flask, url_for, get_flashed_messages
+from flask import Flask, url_for, get_flashed_messages, request
 from flask_session import Session
 from flask_injector import FlaskInjector
 from flask_bcrypt import Bcrypt
 from flask_cors import CORS
 from iatoolkit.common.exceptions import IAToolkitException
-from urllib.parse import urlparse
-import redis
-import logging
-import os
 from typing import Optional, Dict, Any
 from iatoolkit.repositories.database_manager import DatabaseManager
 from werkzeug.middleware.proxy_fix import ProxyFix
 from injector import Binder, Injector, singleton
-from importlib.metadata import version as _pkg_version, PackageNotFoundError
+from urllib.parse import urlparse
+import redis
+import logging
+import os
 
-IATOOLKIT_VERSION = "0.71.4"
+from iatoolkit import __version__ as IATOOLKIT_VERSION
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
 
-
 class IAToolkit:
     """
     IAToolkit main class
@@ -53,6 +51,7 @@ class IAToolkit:
         self.db_manager = None
         self._injector = None
         self.version = IATOOLKIT_VERSION    # default version
+        self.license = "Community Edition"
 
     @classmethod
     def get_instance(cls) -> 'IAToolkit':
@@ -88,13 +87,16 @@ class IAToolkit:
         # and other integrations, as views are handled manually.
         FlaskInjector(app=self.app, injector=self._injector)
 
-        # Step 6: initialize dispatcher and registered companies
-        self._init_dispatcher_and_company_instances()
+        # Step 6: initialize registered companies
+        self._instantiate_company_instances()
 
         # Re-apply logging configuration in case it was modified by company-specific code
         self._setup_logging()
 
-        # Step 7: Finalize setup within the application context
+        # Step 7: load company configuration file
+        self._load_company_configuration()
+
+        # Step 8: Finalize setup within the application context
         self._setup_redis_sessions()
         self._setup_cors()
         self._setup_additional_services()
@@ -102,11 +104,12 @@ class IAToolkit:
         self._setup_request_globals()
         self._setup_context_processors()
 
-        # Step 8: define the download_dir for excel's
+        # Step 9: define the download_dir
         self._setup_download_dir()
 
-        logging.info(f"🎉 IAToolkit v{self.version} inicializado correctamente")
+        logging.info(f"🎉 IAToolkit {self.license} version {self.version} correctly initialized.")
         self._initialized = True
+
         return self.app
 
     def _get_config_value(self, key: str, default=None):
@@ -141,14 +144,15 @@ class IAToolkit:
             force=True
         )
 
+        logging.getLogger("httpx").setLevel(logging.WARNING)
+
     def _register_routes(self):
         """Registers routes by passing the configured injector."""
         from iatoolkit.common.routes import register_views
 
         # Pass the injector to the view registration function
-        register_views(self._injector, self.app)
-
-        logging.info("✅ Routes registered.")
+        register_views(self.app)
+        logging.info("✅ Community routes registered.")
 
     def _create_flask_instance(self):
         static_folder = self._get_config_value('STATIC_FOLDER') or self._get_default_static_folder()
@@ -158,12 +162,6 @@ class IAToolkit:
                          static_folder=static_folder,
                          template_folder=template_folder)
 
-        # get the IATOOLKIT_VERSION from the package metadata
-        try:
-            self.version = _pkg_version("iatoolkit")
-        except PackageNotFoundError:
-            pass
-
         self.app.config.update({
             'VERSION': self.version,
             'SECRET_KEY': self._get_config_value('FLASK_SECRET_KEY', 'iatoolkit-default-secret'),
@@ -171,15 +169,10 @@ class IAToolkit:
             'SESSION_COOKIE_SECURE': True,
             'SESSION_PERMANENT': False,
             'SESSION_USE_SIGNER': True,
-            'JWT_SECRET_KEY': self._get_config_value('JWT_SECRET_KEY', 'iatoolkit-jwt-secret'),
+            'IATOOLKIT_SECRET_KEY': self._get_config_value('IATOOLKIT_SECRET_KEY', 'iatoolkit-jwt-secret'),
             'JWT_ALGORITHM': 'HS256',
-            'JWT_EXPIRATION_SECONDS_CHAT': int(self._get_config_value('JWT_EXPIRATION_SECONDS_CHAT', 3600))
         })
 
-        parsed_url = urlparse(os.getenv('IATOOLKIT_BASE_URL'))
-        if parsed_url.scheme == 'https':
-            self.app.config['PREFERRED_URL_SCHEME'] = 'https'
-
         # 2. ProxyFix para no tener problemas con iframes y rutas
         self.app.wsgi_app = ProxyFix(self.app.wsgi_app, x_proto=1)
 
@@ -192,12 +185,12 @@ class IAToolkit:
         if not database_uri:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,
-                "DATABASE_URI es requerida (config dict o variable de entorno)"
+                "DATABASE_URI is requires (config dict or env. variable)"
             )
 
-        self.db_manager = DatabaseManager(database_uri)
+        self.db_manager = DatabaseManager(database_url=database_uri, schema='iatoolkit')
         self.db_manager.create_all()
-        logging.info("✅ Base de datos configurada correctamente")
+        logging.info("✅ Database configured successfully")
 
         @self.app.teardown_appcontext
         def remove_session(exception=None):
@@ -211,7 +204,7 @@ class IAToolkit:
     def _setup_redis_sessions(self):
         redis_url = self._get_config_value('REDIS_URL')
         if not redis_url:
-            logging.warning("⚠️ REDIS_URL no configurada, usando sesiones en memoria")
+            logging.warning("⚠️ REDIS_URL not configured, will use memory sessions")
             return
 
         try:
@@ -230,20 +223,18 @@ class IAToolkit:
             })
 
             Session(self.app)
-            logging.info("✅ Redis y sesiones configurados correctamente")
+            logging.info("✅ Redis and sessions configured successfully")
 
         except Exception as e:
-            logging.error(f"❌ Error configurando Redis: {e}")
-            logging.warning("⚠️ Continuando sin Redis")
+            logging.error(f"❌ Error configuring Redis: {e}")
+            raise e
 
     def _setup_cors(self):
         """🌐 Configura CORS"""
         from iatoolkit.company_registry import get_company_registry
 
         # default CORS origin
-        default_origins = [
-            os.getenv('IATOOLKIT_BASE_URL')
-        ]
+        default_origins = []
 
         # Iterate through the registered company names
         extra_origins = []
@@ -263,7 +254,7 @@ class IAToolkit:
              ],
              methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"])
 
-        logging.info(f"✅ CORS configurado para: {all_origins}")
+        logging.info(f"✅ CORS configured for: {all_origins}")
 
     def _configure_core_dependencies(self, binder: Binder):
         """⚙️ Configures all system dependencies."""
@@ -277,13 +268,13 @@ class IAToolkit:
             self._bind_services(binder)
             self._bind_infrastructure(binder)
 
-            logging.info("✅ Dependencias configuradas correctamente")
+            logging.info("✅ Dependencies configured successfully")
 
         except Exception as e:
-            logging.error(f"❌ Error configurando dependencias: {e}")
+            logging.error(f"❌ Error configuring dependencies: {e}")
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,
-                f"❌ Error configurando dependencias: {e}"
+                f"❌ Error configuring dependencies: {e}"
             )
 
     def _bind_repositories(self, binder: Binder):
@@ -304,7 +295,7 @@ class IAToolkit:
         from iatoolkit.services.tasks_service import TaskService
         from iatoolkit.services.benchmark_service import BenchmarkService
         from iatoolkit.services.document_service import DocumentService
-        from iatoolkit.services.prompt_manager_service import PromptService
+        from iatoolkit.services.prompt_service import PromptService
         from iatoolkit.services.excel_service import ExcelService
         from iatoolkit.services.mail_service import MailService
         from iatoolkit.services.load_documents_service import LoadDocumentsService
@@ -316,6 +307,8 @@ class IAToolkit:
         from iatoolkit.services.language_service import LanguageService
         from iatoolkit.services.configuration_service import ConfigurationService
         from iatoolkit.services.embedding_service import EmbeddingService
+        from iatoolkit.services.history_manager_service import HistoryManagerService
+        from iatoolkit.services.tool_service import ToolService
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(TaskService, to=TaskService)
@@ -333,32 +326,36 @@ class IAToolkit:
         binder.bind(LanguageService, to=LanguageService)
         binder.bind(ConfigurationService, to=ConfigurationService)
         binder.bind(EmbeddingService, to=EmbeddingService)
+        binder.bind(HistoryManagerService, to=HistoryManagerService)
+        binder.bind(ToolService, to=ToolService)
 
     def _bind_infrastructure(self, binder: Binder):
         from iatoolkit.infra.llm_client import llmClient
         from iatoolkit.infra.llm_proxy import LLMProxy
         from iatoolkit.infra.google_chat_app import GoogleChatApp
-        from iatoolkit.infra.mail_app import MailApp
+        from iatoolkit.infra.brevo_mail_app import BrevoMailApp
         from iatoolkit.services.auth_service import AuthService
         from iatoolkit.common.util import Utility
 
         binder.bind(LLMProxy, to=LLMProxy)
         binder.bind(llmClient, to=llmClient)
         binder.bind(GoogleChatApp, to=GoogleChatApp)
-        binder.bind(MailApp, to=MailApp)
+        binder.bind(BrevoMailApp, to=BrevoMailApp)
         binder.bind(AuthService, to=AuthService)
         binder.bind(Utility, to=Utility)
 
     def _setup_additional_services(self):
         Bcrypt(self.app)
 
-    def _init_dispatcher_and_company_instances(self):
+    def _instantiate_company_instances(self):
         from iatoolkit.company_registry import get_company_registry
-        from iatoolkit.services.dispatcher_service import Dispatcher
 
         # instantiate all the registered companies
         get_company_registry().instantiate_companies(self._injector)
 
+    def _load_company_configuration(self):
+        from iatoolkit.services.dispatcher_service import Dispatcher
+
         # use the dispatcher to load the company config.yaml file and prepare the execution
         dispatcher = self._injector.get(Dispatcher)
         dispatcher.load_company_configs()
@@ -369,17 +366,18 @@ class IAToolkit:
 
         # 1. Register core commands
         register_core_commands(self.app)
-        logging.info("✅ Comandos CLI del núcleo registrados.")
+        logging.info("✅ Core CLI commands registered.")
 
         # 2. Register company-specific commands
         try:
             # Iterate through the registered company names
             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)
+                if hasattr(company_instance, "register_cli_commands"):
+                    company_instance.register_cli_commands(self.app)
 
         except Exception as e:
-            logging.error(f"❌ Error durante el registro de comandos de compañías: {e}")
+            logging.error(f"❌ error while registering company commands: {e}")
 
     def _setup_context_processors(self):
         # Configura context processors para templates
@@ -403,13 +401,14 @@ class IAToolkit:
 
             return {
                 'url_for': url_for,
-                'iatoolkit_version': self.version,
+                'iatoolkit_version': f'{self.version}',
+                'license': self.license,
                 'app_name': 'IAToolkit',
                 'user_identifier': SessionManager.get('user_identifier'),
                 'company_short_name': SessionManager.get('company_short_name'),
                 'user_is_local': user_profile.get('user_is_local'),
                 'user_email': user_profile.get('user_email'),
-                'iatoolkit_base_url': os.environ.get('IATOOLKIT_BASE_URL', ''),
+                'iatoolkit_base_url': request.url_root,
                 'flashed_messages': get_flashed_messages(with_categories=True),
                 't': translate_for_template
             }
@@ -442,7 +441,7 @@ class IAToolkit:
         if not self._injector:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,
-                "App no inicializada. Llame a create_app() primero"
+                "App no initialized. Call create_app() first"
             )
         return self._injector.get(Dispatcher)
 
@@ -450,7 +449,7 @@ class IAToolkit:
         if not self.db_manager:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,
-                "Database manager no inicializado"
+                "Database manager not initialized."
             )
         return self.db_manager
 
@@ -459,7 +458,7 @@ class IAToolkit:
         default_download_dir = os.path.join(os.getcwd(), 'iatoolkit-downloads')
 
         # 3. if user specified one, use it
-        download_dir = self.app.config.get('IATOOLKIT_DOWNLOAD_DIR', default_download_dir)
+        download_dir = self._get_config_value('IATOOLKIT_DOWNLOAD_DIR', default_download_dir)
 
         # 3. save it in the app config
         self.app.config['IATOOLKIT_DOWNLOAD_DIR'] = download_dir

iatoolkit/infra/{mail_app.py → brevo_mail_app.py}

@@ -13,12 +13,13 @@ import logging
 MAX_ATTACH_BYTES = int(os.getenv("BREVO_MAX_ATTACH_BYTES", str(5 * 1024 * 1024)))  # 5MB seguro
 
 
-class MailApp:
-    def __init__(self,):
+class BrevoMailApp:
+    def _init_brevo(self, provider_config: dict, sender: dict = None):
+        # config and init the brevo client
         self.configuration = sib_api_v3_sdk.Configuration()
-        self.configuration.api_key['api-key'] = os.getenv('BREVO_API_KEY')
+        self.configuration.api_key['api-key'] = provider_config.get("api_key")
         self.mail_api = sib_api_v3_sdk.TransactionalEmailsApi(sib_api_v3_sdk.ApiClient(self.configuration))
-        self.sender = {"email": "ia@iatoolkit.com", "name": "IA Toolkit"}
+
 
     @staticmethod
     def _strip_data_url_prefix(b64: str) -> str:
@@ -73,13 +74,20 @@ class MailApp:
 
 
     def send_email(self,
+                   provider_config: dict,
                    to: str,
                    subject: str,
                    body: str,
-                   sender: dict = None,
+                   sender: dict,
                    attachments: list[dict] = None):
-        if not sender:
-            sender = self.sender
+
+        if not provider_config.get("api_key"):
+            logging.error(f'Try to send brevo_mail without api_key in provider_config')
+            raise IAToolkitException(IAToolkitException.ErrorType.MAIL_ERROR,
+                                     f"Invalid mail configuration for Brevo: missing api-key")
+
+        # init the Brevo client
+        self._init_brevo(provider_config)
 
         try:
             sdk_attachments = self._normalize_attachments(attachments)
@@ -112,34 +120,4 @@ class MailApp:
             logging.exception("MAIL ERROR: %s", str(e))
             raise IAToolkitException(IAToolkitException.ErrorType.MAIL_ERROR,
                                f"No se pudo enviar correo: {str(e)}") from e
-''''
-    def send_template_email(self,
-                   subject: str,
-                   recipients: list,
-                   template_name: str,
-                   context: dict,
-                   sender=None):
-        try:
-            # Renderiza el template con el contexto proporcionado
-            with self.app.app_context():
-                html_message = render_template(template_name, **context)
 
-            # Crea el mensaje
-            msg = Message(
-                subject=subject,
-                recipients=recipients,
-                html=html_message,
-                sender=sender or self.app.config.get('MAIL_DEFAULT_SENDER')
-            )
-
-            # Envía el correo
-            # self.send_brevo_email(msg)
-            pass
-        except jinja2.exceptions.TemplateNotFound:
-            raise IAToolkitException(IAToolkitException.ErrorType.MAIL_ERROR,
-                               f"Error: No se encontró el template '{template_name}'.")
-        except Exception as e:
-            raise IAToolkitException(IAToolkitException.ErrorType.MAIL_ERROR,
-                               f'No se pudo enviar correo: {str(e)}') from e
-    
-    '''

iatoolkit/infra/llm_client.py

@@ -75,9 +75,12 @@ class llmClient:
         force_tool_name = None
         reasoning = {}
 
-        if 'gpt-5' in model:
+        if model in ('gpt-5', 'gpt-5-mini'):
             text['verbosity'] = "low"
             reasoning = {"effort": 'minimal'}
+        elif model == 'gpt-5.1':
+            text['verbosity'] = "low"
+            reasoning = {"effort": 'low'}
 
         try:
             start_time = time.time()
@@ -86,7 +89,7 @@ class llmClient:
             # get the proxy for the company
             llm_proxy = self.llm_proxy_factory.create_for_company(company)
 
-            # here is the first call to the LLM
+            # this is the first call to the LLM on the iteration
             try:
                 input_messages = [{
                     "role": "user",
@@ -132,7 +135,7 @@ class llmClient:
                     try:
                         result = self.dispatcher.dispatch(
                             company_short_name=company.short_name,
-                            action=function_name,
+                            function_name=function_name,
                             **args
                         )
                         force_tool_name = None
@@ -170,10 +173,10 @@ class llmClient:
                     f_calls.append(f_call_identity)
                     f_call_time += elapsed
 
-                    logging.info(f"end execution {function_name} in {elapsed:.1f} secs.")
+                    logging.info(f"[{company.short_name}] Tool end execution: {function_name} in {elapsed:.1f} secs.")
 
                 if not function_calls:
-                    break           # no function call answer to send back to llm
+                    break           # no more function calls, the answer to send back to llm
 
                 # send results back to the LLM
                 tool_choice_value = "auto"
@@ -226,6 +229,7 @@ class llmClient:
                 'aditional_data': decoded_response.get('aditional_data', {}),
                 'response_id': response.id,
                 'query_id': query.id,
+                'model': model,
             }
         except SQLAlchemyError as db_error:
             # rollback

iatoolkit/locales/en.yaml

@@ -2,7 +2,7 @@
 ui:
   login_widget:
     title: "Sign In"
-    welcome_message: "Enter your credentials or register to access Sample Company’s AI platform."
+    welcome_message: "Enter your credentials or register to access this platform."
     email_placeholder: "Email address"
     password_placeholder: "Password"
     login_button: "Login"
@@ -42,8 +42,9 @@ ui:
 
   chat:
     welcome_message: "Hello! How can I help you today?"
-    input_placeholder: "Type your query here..."
+    input_placeholder: "Type here..."
     prompts_available: "Available prompts"
+    init_context: "Initializing the context ..."
 
   tooltips:
     history: "History of my queries"
@@ -130,6 +131,9 @@ errors:
     user_not_authorized: "user is not authorized for this company"
     account_not_verified: "Your account has not been verified. Please check your email."
     missing_response_id: "Can not found 'previous_response_id' for '{company_short_name}/{user_identifier}'. Reinit context"
+    context_rebuild_failed: "Company context rebuild failed."
+    cannot_read_excel: "Cannot read Excel file"
+    cannot_read_csv: "Cannot read CSV file"
 
 
 api_responses:
@@ -138,12 +142,14 @@ api_responses:
 services:
   mail_sent: "Email sent successfully."
   start_query: "Hello, what can I help you with today?"
+  mail_change_password: "mail sent for password change"
 
 
 flash_messages:
   password_changed_success: "Your password has been successfully reset. You can now log in."
   login_required: "Please log in to continue."
   signup_success: "Registration successful. Please check your email to verify your account."
+  signup_success_no_verification: "Registration successful."
   user_associated_success: "Existing user successfully associated with the new company."
   account_verified_success: "Your account has been successfully verified. Welcome!"
   forgot_password_success: "If your email is registered, you will receive a link to reset your password."
@@ -165,3 +171,5 @@ js_messages:
   loading: "Loading..."
   reload_init: "init reloading context in background..."
   no_history_found: "No query history found."
+  example: "Example:"
+