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

iatoolkit/base_company.py

@@ -26,10 +26,18 @@ class BaseCompany(ABC):
         self.company = self.profile_repo.get_company_by_short_name(short_name)
         return self.company
 
-    def _create_company(self, short_name: str, name: str, branding: dict | None = None) -> 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,
-                              branding=branding)
+                              parameters=parameters,
+                              branding=branding,
+                              onboarding_cards=onboarding_cards)
         self.company = self.profile_repo.create_company(company_obj)
         return self.company
 
@@ -82,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,102 @@
 #
 # 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'))
 
 
 # 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.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
 
-    # 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')
+    )
+
+    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>/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']}))
+    # 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.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 +121,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)
@@ -108,6 +81,23 @@ class Utility:
                                f'No se pudo renderizar el template desde el string, error: {str(e)}') from e
 
 
+    def get_company_template(self, company_short_name: str, template_name: str) -> str:
+        # 1. get the path to the company specific template
+        template_path = os.path.join(os.getcwd(), f'companies/{company_short_name}/templates/{template_name}')
+        if not os.path.exists(template_path):
+            return None
+
+        # 2. read the file
+        try:
+            with open(template_path, 'r') as f:
+                template_string = f.read()
+
+            return template_string
+        except Exception as e:
+            logging.exception(e)
+            return None
+
+
     def serialize(self, obj):
         if isinstance(obj, datetime) or isinstance(obj, date):
             return obj.isoformat()

iatoolkit/iatoolkit.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from flask import Flask, url_for
+from flask import Flask, url_for, get_flashed_messages
 from flask_session import Session
 from flask_injector import FlaskInjector
 from flask_bcrypt import Bcrypt
@@ -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.66.2"
 
 # 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
@@ -95,16 +96,12 @@ class IAToolkit:
         self._setup_cors()
         self._setup_additional_services()
         self._setup_cli_commands()
+        self._setup_request_globals()
         self._setup_context_processors()
 
         # 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
@@ -113,9 +110,26 @@ class IAToolkit:
         # get a value from the config dict or the environment variable
         return self.config.get(key, os.getenv(key, default))
 
+    def _setup_request_globals(self):
+        """
+        Configures functions to run before each request to set up
+        request-global variables, such as language.
+        """
+        injector = self._injector
+
+        @self.app.before_request
+        def set_request_language():
+            """
+            Determines and caches the language for the current request in g.lang.
+            """
+            from iatoolkit.services.language_service import LanguageService
+            language_service = injector.get(LanguageService)
+            language_service.get_current_language()
+
     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 +158,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 +180,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 +202,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 +241,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 +268,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 +279,6 @@ class IAToolkit:
             self._bind_repositories(binder)
             self._bind_services(binder)
             self._bind_infrastructure(binder)
-            self._bind_views(binder)
 
             logging.info("✅ Dependencias configuradas correctamente")
 
@@ -279,6 +316,8 @@ class IAToolkit:
         from iatoolkit.services.jwt_service import JWTService
         from iatoolkit.services.dispatcher_service import Dispatcher
         from iatoolkit.services.branding_service import BrandingService
+        from iatoolkit.services.i18n_service import I18nService
+        from iatoolkit.services.language_service import LanguageService
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(TaskService, to=TaskService)
@@ -292,39 +331,24 @@ class IAToolkit:
         binder.bind(JWTService, to=JWTService)
         binder.bind(Dispatcher, to=Dispatcher)
         binder.bind(BrandingService, to=BrandingService)
-
+        binder.bind(I18nService, to=I18nService)
+        binder.bind(LanguageService, to=LanguageService)
 
     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)
 
@@ -362,12 +386,32 @@ class IAToolkit:
         @self.app.context_processor
         def inject_globals():
             from iatoolkit.common.session_manager import SessionManager
+            from iatoolkit.services.profile_service import ProfileService
+            from iatoolkit.services.i18n_service import I18nService
+
+            # Get services from the injector
+            profile_service = self._injector.get(ProfileService)
+            i18n_service = self._injector.get(I18nService)
+
+            # The 't' function wrapper no longer needs to determine the language itself.
+            # It will be automatically handled by the refactored I18nService.
+            def translate_for_template(key: str, **kwargs):
+                return i18n_service.t(key, **kwargs)
+
+            # Get user profile if a session exists
+            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', ''),
+                'flashed_messages': get_flashed_messages(with_categories=True),
+                't': translate_for_template
             }
 
     def _get_default_static_folder(self) -> str:

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)