iatoolkit 0.8.1__py3-none-any.whl → 0.63.4__py3-none-any.whl

iatoolkit/__init__.py

@@ -4,37 +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 iatoolkit.services.query_service import QueryService
 from iatoolkit.services.sql_service import SqlService
-from iatoolkit.services.excel_service import ExcelService
-from iatoolkit.services.dispatcher_service import Dispatcher
 from iatoolkit.services.document_service import DocumentService
 from iatoolkit.services.search_service import SearchService
 from iatoolkit.services.load_documents_service import LoadDocumentsService
-from iatoolkit.repositories.profile_repo import ProfileRepo
-from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
-
-from iatoolkit.services.query_service import QueryService
-from iatoolkit.services.prompt_manager_service import PromptService
-from iatoolkit.repositories.database_manager import DatabaseManager
-
+from iatoolkit.services.excel_service import ExcelService
 from iatoolkit.infra.call_service import CallServiceClient
-from iatoolkit.common.util import Utility
-
-
-from iatoolkit.repositories.models import Base, Company, Function, TaskType, Prompt, PromptCategory
-
 
 __all__ = [
     'IAToolkit',
@@ -42,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

@@ -26,8 +26,18 @@ class BaseCompany(ABC):
         self.company = self.profile_repo.get_company_by_short_name(short_name)
         return self.company
 
-    def _create_company(self, name: str, short_name: str) -> Company:
-        company_obj = Company(name=name, short_name=short_name, allow_jwt=True)
+    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
 
@@ -67,6 +77,7 @@ class BaseCompany(ABC):
             **kwargs
         )
 
+
     @abstractmethod
     # initialize all the database tables  needed
     def register_company(self):
@@ -79,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

iatoolkit/common/routes.py

@@ -3,80 +3,93 @@
 #
 # IAToolkit is open source software.
 
-from flask import render_template, redirect, flash, url_for,send_from_directory, current_app, abort
-from iatoolkit.common.session_manager import SessionManager
+from flask import render_template, redirect, url_for,send_from_directory, current_app, abort
 from flask import jsonify
-from iatoolkit.views.history_view import HistoryView
-import os
-
-
-def logout(company_short_name: str):
-    SessionManager.clear()
-    flash("Has cerrado sesión correctamente", "info")
-    if company_short_name:
-        return redirect(url_for('login', company_short_name=company_short_name))
-    else:
-        return redirect(url_for('home'))
+from iatoolkit.views.history_api_view import HistoryApiView
 
 
 # this function register all the views
 def register_views(injector, app):
 
-    from iatoolkit.views.llmquery_view import LLMQueryView
-    from iatoolkit.views.tasks_view import TaskView
-    from iatoolkit.views.tasks_review_view import TaskReviewView
-    from iatoolkit.views.home_view import HomeView
-    from iatoolkit.views.chat_view import ChatView
-    from iatoolkit.views.login_view import LoginView
-    from iatoolkit.views.external_chat_login_view import ExternalChatLoginView
+    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_view import FileStoreView
-    from iatoolkit.views.user_feedback_view import UserFeedbackView
-    from iatoolkit.views.prompt_view import PromptView
-    from iatoolkit.views.chat_token_request_view import ChatTokenRequestView
-    from iatoolkit.views.external_login_view import ExternalLoginView
-    from iatoolkit.views.download_file_view import DownloadFileView
-
-    app.add_url_rule('/', view_func=HomeView.as_view('home'))
+    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.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
 
-    # main chat for iatoolkit front
-    app.add_url_rule('/<company_short_name>/chat', view_func=ChatView.as_view('chat'))
+    # iatoolkit home page
+    app.add_url_rule('/', view_func=IndexView.as_view('index'))
 
-    # front if the company internal portal
-    app.add_url_rule('/<company_short_name>/chat_login', view_func=ExternalChatLoginView.as_view('external_chat_login'))
-    app.add_url_rule('/<company_short_name>/external_login/<external_user_id>', view_func=ExternalLoginView.as_view('external_login'))
-    app.add_url_rule('/auth/chat_token', view_func=ChatTokenRequestView.as_view('chat-token'))
+    # company home view
+    app.add_url_rule('/<company_short_name>/home', view_func=HomeView.as_view('home'))
 
-    # main pages for the iatoolkit frontend
+    # 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')
+    )
+
+    # 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>/logout', 'logout', logout)
-    app.add_url_rule('/logout', 'logout', logout)
     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'))
 
-    # this are backend endpoints mainly
-    app.add_url_rule('/<company_short_name>/llm_query', view_func=LLMQueryView.as_view('llm_query'))
-    app.add_url_rule('/<company_short_name>/feedback', view_func=UserFeedbackView.as_view('feedback'))
-    app.add_url_rule('/<company_short_name>/prompts', view_func=PromptView.as_view('prompt'))
-    app.add_url_rule('/<company_short_name>/history', view_func=HistoryView.as_view('history'))
-    app.add_url_rule('/tasks', view_func=TaskView.as_view('tasks'))
-    app.add_url_rule('/tasks/review/<int:task_id>', view_func=TaskReviewView.as_view('tasks-review'))
-    app.add_url_rule('/load', view_func=FileStoreView.as_view('load'))
+    # 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'))
 
-    app.add_url_rule(
-        '/about',  # URL de la ruta
-        view_func=lambda: render_template('about.html'))
+    # open the promt directory
+    app.add_url_rule('/<company_short_name>/api/prompts', view_func=PromptApiView.as_view('prompt'))
 
-    app.add_url_rule('/version', 'version',
-                     lambda: jsonify({"version": app.config['VERSION']}))
+    # feedback and history
+    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'))
+
+    # 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.add_url_rule('/<company_short_name>/<external_user_id>/download-file/<path:filename>',
-                     view_func=DownloadFileView.as_view('download-file'))
 
     @app.route('/download/<path:filename>')
     def download_file(filename):
@@ -99,3 +112,21 @@ 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/session_manager.py

@@ -5,7 +5,6 @@
 
 from flask import session
 
-
 class SessionManager:
     @staticmethod
     def set(key, value):

iatoolkit/common/util.py

@@ -9,7 +9,6 @@ from iatoolkit.common.exceptions import IAToolkitException
 from injector import inject
 import os
 from jinja2 import Environment, FileSystemLoader
-from iatoolkit.common.session_manager import SessionManager
 from datetime import datetime, date
 from decimal import Decimal
 import yaml
@@ -22,30 +21,8 @@ class Utility:
     def __init__(self):
         self.encryption_key = os.getenv('FERNET_KEY')
 
-    @staticmethod
-    def resolve_user_identifier(external_user_id: str = None, local_user_id: int = 0) -> tuple[str, bool]:
-        """
-        Resuelve un identificador único de usuario desde external_user_id o local_user_id.
-
-        Lógica:
-        - Si external_user_id existe y no está vacío: usar external_user_id
-        - Si no, y local_user_id > 0: obtener email de la sesión actual y retornarlo como ID
-        - Si ninguno: retornar string vacío
-
-        """
-        if external_user_id and external_user_id.strip():
-            return external_user_id.strip(), False
-        elif local_user_id and local_user_id > 0:
-            # get the user information from the session
-            user_data = SessionManager.get('user')
-            if user_data:
-                return user_data.get('email', ''), True
-
-        return "", False
-
     def render_prompt_from_template(self,
                                     template_pathname: str,
-                                    query: str = None,
                                     client_data: dict = {},
                                     **kwargs) -> str:
 
@@ -58,8 +35,6 @@ class Utility:
             env = Environment(loader=FileSystemLoader(template_dir))
             template = env.get_template(template_file)
 
-            kwargs["query"] = query
-
             # add all the keys in client_data to kwargs
             kwargs.update(client_data)
 
@@ -74,7 +49,6 @@ class Utility:
     def render_prompt_from_string(self,
                                   template_string: str,
                                   searchpath: str | list[str] = None,
-                                  query: str = None,
                                   client_data: dict = {},
                                   **kwargs) -> str:
         """
@@ -97,7 +71,6 @@ class Utility:
             env = Environment(loader=loader)
             template = env.from_string(template_string)
 
-            kwargs["query"] = query
             kwargs.update(client_data)
 
             prompt = template.render(**kwargs)

iatoolkit/iatoolkit.py

@@ -15,10 +15,11 @@ import logging
 import os
 from typing import Optional, Dict, Any
 from iatoolkit.repositories.database_manager import DatabaseManager
-
-from injector import Binder, singleton, Injector
+from werkzeug.middleware.proxy_fix import ProxyFix
+from injector import Binder, Injector, singleton
 from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
+IATOOLKIT_VERSION = "0.63.4"
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
@@ -51,7 +52,7 @@ class IAToolkit:
         self.app = None
         self.db_manager = None
         self._injector = None
-        self.version = "0.0.0+dev"
+        self.version = IATOOLKIT_VERSION    # default version
 
     @classmethod
     def get_instance(cls) -> 'IAToolkit':
@@ -87,7 +88,7 @@ class IAToolkit:
         # and other integrations, as views are handled manually.
         FlaskInjector(app=self.app, injector=self._injector)
 
-        # Step 6: initialize dispatcher and registered compaies
+        # Step 6: initialize dispatcher and registered companies
         self._init_dispatcher_and_company_instances()
 
         # Step 7: Finalize setup within the application context
@@ -100,11 +101,6 @@ class IAToolkit:
         # Step 8: define the download_dir for excel's
         self._setup_download_dir()
 
-        try:
-            self.version = _pkg_version("iatoolkit")
-        except PackageNotFoundError:
-            pass
-
         logging.info(f"🎉 IAToolkit v{self.version} inicializado correctamente")
         self._initialized = True
         return self.app
@@ -114,8 +110,9 @@ class IAToolkit:
         return self.config.get(key, os.getenv(key, default))
 
     def _setup_logging(self):
-        log_level_str = self._get_config_value('FLASK_ENV', 'production')
-        log_level = logging.INFO if log_level_str in ('dev', 'development') else logging.WARNING
+        # Lee el nivel de log desde una variable de entorno, con 'INFO' como valor por defecto.
+        log_level_name = os.getenv('LOG_LEVEL', 'INFO').upper()
+        log_level = getattr(logging, log_level_name, logging.INFO)
 
         logging.basicConfig(
             level=log_level,
@@ -144,11 +141,21 @@ class IAToolkit:
         is_https = self._get_config_value('USE_HTTPS', 'false').lower() == 'true'
         is_dev = self._get_config_value('FLASK_ENV') == 'development'
 
+        # get the iatoolkit domain
+        parsed_url = urlparse(os.getenv('IATOOLKIT_BASE_URL'))
+        domain = parsed_url.netloc
+
+        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'),
-            'SESSION_COOKIE_SAMESITE': "None" if is_https else "Lax",
-            'SESSION_COOKIE_SECURE': is_https,
+            'SESSION_COOKIE_SAMESITE': "None",
+            'SESSION_COOKIE_SECURE': True,
             'SESSION_PERMANENT': False,
             'SESSION_USE_SIGNER': True,
             'JWT_SECRET_KEY': self._get_config_value('JWT_SECRET_KEY', 'iatoolkit-jwt-secret'),
@@ -156,6 +163,12 @@ class IAToolkit:
             'JWT_EXPIRATION_SECONDS_CHAT': int(self._get_config_value('JWT_EXPIRATION_SECONDS_CHAT', 3600))
         })
 
+        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)
+
         # Configuración para tokenizers en desarrollo
         if is_dev:
             os.environ["TOKENIZERS_PARALLELISM"] = "false"
@@ -172,6 +185,15 @@ class IAToolkit:
         self.db_manager.create_all()
         logging.info("✅ Base de datos configurada correctamente")
 
+        @self.app.teardown_appcontext
+        def remove_session(exception=None):
+            """
+            Flask calls this after each request.
+            It ensures the SQLAlchemy session is properly closed
+            and the DB connection is returned to the pool.
+            """
+            self.db_manager.scoped_session.remove()
+
     def _setup_redis_sessions(self):
         redis_url = self._get_config_value('REDIS_URL')
         if not redis_url:
@@ -202,19 +224,19 @@ class IAToolkit:
 
     def _setup_cors(self):
         """🌐 Configura CORS"""
-        # Origins por defecto para desarrollo
+        from iatoolkit.company_registry import get_company_registry
+
+        # default CORS origin
         default_origins = [
-            "http://localhost:5001",
-            "http://127.0.0.1:5001",
             os.getenv('IATOOLKIT_BASE_URL')
         ]
 
-        # Obtener origins adicionales desde configuración/env
+        # Iterate through the registered company names
         extra_origins = []
-        for i in range(1, 11):  # Soporte para CORS_ORIGIN_1 a CORS_ORIGIN_10
-            origin = self._get_config_value(f'CORS_ORIGIN_{i}')
-            if origin:
-                extra_origins.append(origin)
+        all_company_instances = get_company_registry().get_all_company_instances()
+        for company_name, company_instance in all_company_instances.items():
+            cors_origin = company_instance.company.parameters.get('cors_origin', [])
+            extra_origins += cors_origin
 
         all_origins = default_origins + extra_origins
 
@@ -229,7 +251,6 @@ class IAToolkit:
 
         logging.info(f"✅ CORS configurado para: {all_origins}")
 
-
     def _configure_core_dependencies(self, binder: Binder):
         """⚙️ Configures all system dependencies."""
         try:
@@ -241,7 +262,6 @@ class IAToolkit:
             self._bind_repositories(binder)
             self._bind_services(binder)
             self._bind_infrastructure(binder)
-            self._bind_views(binder)
 
             logging.info("✅ Dependencias configuradas correctamente")
 
@@ -278,6 +298,7 @@ class IAToolkit:
         from iatoolkit.services.profile_service import ProfileService
         from iatoolkit.services.jwt_service import JWTService
         from iatoolkit.services.dispatcher_service import Dispatcher
+        from iatoolkit.services.branding_service import BrandingService
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(TaskService, to=TaskService)
@@ -290,38 +311,23 @@ class IAToolkit:
         binder.bind(ProfileService, to=ProfileService)
         binder.bind(JWTService, to=JWTService)
         binder.bind(Dispatcher, to=Dispatcher)
+        binder.bind(BrandingService, to=BrandingService)
 
     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.common.auth import IAuthentication
+        from iatoolkit.services.auth_service import AuthService
         from iatoolkit.common.util import Utility
 
-
-
-        binder.bind(LLMProxy, to=LLMProxy, scope=singleton)
-        binder.bind(llmClient, to=llmClient, scope=singleton)
+        binder.bind(LLMProxy, to=LLMProxy)
+        binder.bind(llmClient, to=llmClient)
         binder.bind(GoogleChatApp, to=GoogleChatApp)
         binder.bind(MailApp, to=MailApp)
-        binder.bind(IAuthentication, to=IAuthentication)
+        binder.bind(AuthService, to=AuthService)
         binder.bind(Utility, to=Utility)
 
-    def _bind_views(self, binder: Binder):
-        """Vincula las vistas después de que el injector ha sido creado"""
-        from iatoolkit.views.llmquery_view import LLMQueryView
-        from iatoolkit.views.home_view import HomeView
-        from iatoolkit.views.chat_view import ChatView
-        from iatoolkit.views.change_password_view import ChangePasswordView
-
-        binder.bind(HomeView, to=HomeView)
-        binder.bind(ChatView, to=ChatView)
-        binder.bind(ChangePasswordView, to=ChangePasswordView)
-        binder.bind(LLMQueryView, to=LLMQueryView)
-
-        logging.info("✅ Views configuradas correctamente")
-
     def _setup_additional_services(self):
         Bcrypt(self.app)
 
@@ -359,14 +365,23 @@ class IAToolkit:
         @self.app.context_processor
         def inject_globals():
             from iatoolkit.common.session_manager import SessionManager
+            from iatoolkit.services.profile_service import ProfileService
+
+            profile_service = self._injector.get(ProfileService)
+            user_profile = profile_service.get_current_session_info().get('profile', {})
+
             return {
                 'url_for': url_for,
                 'iatoolkit_version': self.version,
                 'app_name': 'IAToolkit',
-                'user': SessionManager.get('user'),
-                'user_company': SessionManager.get('company_short_name'),
+                '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', ''),
             }
 
+
     def _get_default_static_folder(self) -> str:
         try:
             current_dir = os.path.dirname(os.path.abspath(__file__))  # .../src/iatoolkit
@@ -409,7 +424,7 @@ class IAToolkit:
 
     def _setup_download_dir(self):
         # 1. set the default download directory
-        default_download_dir = os.path.join(os.getcwd(), 'downloads')
+        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)

iatoolkit/infra/llm_client.py

@@ -21,6 +21,7 @@ import tiktoken
 from typing import Dict, Optional, List
 from iatoolkit.services.dispatcher_service import Dispatcher
 
+CONTEXT_ERROR_MESSAGE = 'Tu consulta supera el límite de contexto, utiliza el boton de recarga de contexto.'
 
 class llmClient:
     _llm_clients_cache = {}      # class attribute, for the clients cache
@@ -116,7 +117,7 @@ class llmClient:
 
                 # in case of context error
                 if "context_length_exceeded" in str(e):
-                    error_message = 'Tu consulta supera el limite de contexto, sale e ingresa de nuevo a IAToolkit'
+                    error_message = CONTEXT_ERROR_MESSAGE
 
                 raise IAToolkitException(IAToolkitException.ErrorType.LLM_ERROR, error_message)
 
@@ -256,25 +257,23 @@ class llmClient:
 
             # in case of context error
             if "context_length_exceeded" in str(e):
-                error_message = 'Tu consulta supera el limite de contexto, sale e ingresa de nuevo a IAToolkit'
+                error_message = CONTEXT_ERROR_MESSAGE
             elif "string_above_max_length" in str(e):
-                error_message = 'La respuesta es muy larga, trata de filtrar/restringuir tu consulta'
+                error_message = 'La respuesta es muy extensa, trata de filtrar/restringuir tu consulta'
 
             raise IAToolkitException(IAToolkitException.ErrorType.LLM_ERROR, error_message)
 
     def set_company_context(self,
             company: Company,
             company_base_context: str,
-            model: str = None) -> str:
+            model) -> str:
 
-        if model:
-            self.model = model
-        logging.info(f"initializing model '{self.model}' with company context: {self.count_tokens(company_base_context)} tokens...")
+        logging.info(f"initializing model '{model}' with company context: {self.count_tokens(company_base_context)} tokens...")
 
         llm_proxy = self.llm_proxy_factory.create_for_company(company)
         try:
             response = llm_proxy.create_response(
-                model=self.model,
+                model=model,
                 input=[{
                     "role": "system",
                     "content": company_base_context

iatoolkit/infra/openai_adapter.py

@@ -55,7 +55,7 @@ class OpenAIAdapter:
 
             # En caso de error de contexto
             if "context_length_exceeded" in str(e):
-                error_message = 'Tu consulta supera el limite de contexto, sale e ingresa de nuevo a IAToolkit'
+                error_message = 'Tu consulta supera el limite de contexto. Reinicia el contexto con el boton de la barra superior.'
 
             raise IAToolkitException(IAToolkitException.ErrorType.LLM_ERROR, error_message)