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

iatoolkit/templates/onboarding_shell.html

@@ -38,7 +38,6 @@
     <div class="ob-stack">
       <h1 id="ob-brand-header" class="ob-brand-header">
         <span class="brand-name text-brand-primary">{{ branding.name | default('IAToolkit') }}</span>
-        <span class="brand-rest"> IA</span>
       </h1>
 
       <div id="card-container" class="ob-card">
@@ -60,7 +59,7 @@
 
       <div id="loading-status" class="ob-loading-band">
         <div class="spinner"></div>
-        <p>Inicializando el contexto de {{ branding.name }} para la IA...</p>
+        <p>{{ t('ui.chat.init_context') }} </p>
       </div>
     </div>
   </div>

iatoolkit/templates/signup.html

@@ -59,6 +59,9 @@
                                 <input type="password" id="confirm_password" name="confirm_password" class="form-control" required>
                             </div>
 
+                            <!-- language: html -->
+                            <input type="hidden" name="lang" value="{{ lang or 'en' }}">
+
                             <!-- Botón actualizado con la clase de branding -->
                             <button type="submit" class="btn btn-branded-primary w-100 fw-bold py-2 mt-3">{{ t('ui.signup.signup_button') }}</button>
                         </form>

iatoolkit/views/base_login_view.py

@@ -12,7 +12,7 @@ from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.branding_service import BrandingService
 from iatoolkit.services.configuration_service import ConfigurationService
-from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.prompt_service import PromptService
 from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.common.util import Utility
 from iatoolkit.services.jwt_service import JWTService
@@ -74,6 +74,9 @@ class BaseLoginView(MethodView):
             )
         else:
             # --- FAST PATH: Render the chat page directly ---
+            # LLM configuration: default model and availables
+            default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
+
             prompts = self.prompt_service.get_user_prompts(company_short_name)
 
             # Get the entire 'js_messages' block in the correct language.
@@ -87,5 +90,7 @@ class BaseLoginView(MethodView):
                 branding=branding_data,
                 onboarding_cards=onboarding_cards,
                 js_translations=js_translations,
-                redeem_token=redeem_token
-            )
+                redeem_token=redeem_token,
+                llm_default_model=default_llm_model,
+                llm_available_models = available_llm_models,
+                )

iatoolkit/views/change_password_view.py

@@ -24,7 +24,7 @@ class ChangePasswordView(MethodView):
         self.branding_service = branding_service
         self.i18n_service = i18n_service
 
-        self.serializer = URLSafeTimedSerializer(os.getenv("PASS_RESET_KEY"))
+        self.serializer = URLSafeTimedSerializer(os.getenv("IATOOLKIT_SECRET_KEY"))
         self.bcrypt = Bcrypt()
 
     def get(self, company_short_name: str, token: str):

iatoolkit/views/chat_view.py

@@ -0,0 +1,76 @@
+from flask import render_template, redirect, url_for, request
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.configuration_service import ConfigurationService
+from iatoolkit.services.prompt_service import PromptService
+from iatoolkit.services.i18n_service import I18nService
+
+class ChatView(MethodView):
+    """
+    Handles direct access to the chat interface.
+    Validates if the user has an active session for the company.
+    """
+
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 config_service: ConfigurationService,
+                 prompt_service: PromptService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.config_service = config_service
+        self.prompt_service = prompt_service
+        self.i18n_service = i18n_service
+
+    def get(self, company_short_name: str):
+        # 1. Validate Company
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        # 2. Check Session
+        session_info = self.profile_service.get_current_session_info()
+        user_identifier = session_info.get('user_identifier')
+        session_company = session_info.get('company_short_name')
+
+        # If no user or session belongs to another company -> Redirect to Home
+        if not user_identifier or session_company != company_short_name:
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        # 3. Prepare Context for Chat
+        # (This logic mirrors the FAST PATH in BaseLoginView)
+        try:
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+            onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
+            default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
+            prompts = self.prompt_service.get_user_prompts(company_short_name)
+            js_translations = self.i18n_service.get_translation_block('js_messages')
+
+            return render_template(
+                "chat.html",
+                company_short_name=company_short_name,
+                user_identifier=user_identifier,
+                prompts=prompts,
+                branding=branding_data,
+                onboarding_cards=onboarding_cards,
+                js_translations=js_translations,
+                llm_default_model=default_llm_model,
+                llm_available_models=available_llm_models,
+                # redeem_token is None for direct access
+                redeem_token=None
+            )
+        except Exception as e:
+            # Fallback error handling
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500

iatoolkit/views/forgot_password_view.py

@@ -21,7 +21,7 @@ class ForgotPasswordView(MethodView):
         self.branding_service = branding_service
         self.i18n_service = i18n_service
 
-        self.serializer = URLSafeTimedSerializer(os.getenv("PASS_RESET_KEY"))
+        self.serializer = URLSafeTimedSerializer(os.getenv("IATOOLKIT_SECRET_KEY"))
 
     def get(self, company_short_name: str):
         # get company info
@@ -31,9 +31,11 @@ class ForgotPasswordView(MethodView):
                                    message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
         branding_data = self.branding_service.get_company_branding(company_short_name)
+        current_lang = request.args.get("lang", "en")
         return render_template('forgot_password.html',
                                company_short_name=company_short_name,
-                               branding=branding_data
+                               branding=branding_data,
+                               lang=current_lang
                                )
 
     def post(self, company_short_name: str):
@@ -53,7 +55,9 @@ class ForgotPasswordView(MethodView):
                                 company_short_name=company_short_name,
                                 token=token, _external=True)
 
-            response = self.profile_service.forgot_password(email=email, reset_url=reset_url)
+            response = self.profile_service.forgot_password(
+                        company_short_name=company_short_name,
+                        email=email, reset_url=reset_url)
             if "error" in response:
                 flash(response["error"], 'error')
                 return render_template(
@@ -63,7 +67,8 @@ class ForgotPasswordView(MethodView):
                     form_data={"email": email}), 400
 
             flash(self.i18n_service.t('flash_messages.forgot_password_success'), 'success')
-            return redirect(url_for('home', company_short_name=company_short_name))
+            lang = request.args.get("lang", "en")
+            return redirect(url_for('home', company_short_name=company_short_name, lang=lang))
 
         except Exception as e:
             flash(self.i18n_service.t('errors.general.unexpected_error'), 'error')

iatoolkit/views/history_api_view.py

@@ -5,7 +5,7 @@
 
 from flask import request, jsonify
 from flask.views import MethodView
-from iatoolkit.services.history_service import HistoryService
+from iatoolkit.services.history_manager_service import HistoryManagerService
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.i18n_service import I18nService
 from injector import inject
@@ -21,7 +21,7 @@ class HistoryApiView(MethodView):
     @inject
     def __init__(self,
                  auth_service: AuthService,
-                 history_service: HistoryService,
+                 history_service: HistoryManagerService,
                  i18n_service: I18nService):
         self.auth_service = auth_service
         self.history_service = history_service
@@ -39,7 +39,7 @@ class HistoryApiView(MethodView):
 
             # 2. Call the history service with the unified identifier.
             # The service's signature should now only expect user_identifier.
-            response = self.history_service.get_history(
+            response = self.history_service.get_full_history(
                 company_short_name=company_short_name,
                 user_identifier=user_identifier
             )

iatoolkit/views/home_view.py

@@ -1,5 +1,5 @@
 # iatoolkit/views/home_view.py
-from flask import render_template,  render_template_string
+from flask import render_template,  render_template_string, request
 from flask.views import MethodView
 from injector import inject
 from iatoolkit.services.profile_service import ProfileService
@@ -33,7 +33,9 @@ class HomeView(MethodView):
                                        message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
             branding_data = self.branding_service.get_company_branding(company_short_name)
-            home_template = self.util.get_company_template(company_short_name, "home.html")
+
+            template_name = self.util.get_template_by_language("home")
+            home_template = self.util.get_company_template(company_short_name, template_name)
 
             # 2. Verificamos si el archivo de plantilla personalizado no existe.
             if not home_template:

iatoolkit/views/init_context_api_view.py

@@ -54,7 +54,7 @@ class InitContextApiView(MethodView):
             response_message = {'status': 'OK', 'message': success_message}
 
             # if received a response ID with the context, return it
-            if response.get('response_id'):
+            if response and response.get('response_id'):
                 response_message['response_id'] = response['response_id']
 
             return jsonify(response_message), 200

iatoolkit/views/llmquery_api_view.py

@@ -41,18 +41,19 @@ class LLMQueryApiView(MethodView):
             result = self.query_service.llm_query(
                 company_short_name=company_short_name,
                 user_identifier=user_identifier,
+                model=data.get('model', ''),
                 question=data.get('question', ''),
                 prompt_name=data.get('prompt_name'),
                 client_data=data.get('client_data', {}),
-                response_id = data.get('response_id'),
+                ignore_history=data.get('ignore_history', False),
                 files=data.get('files', [])
             )
             if 'error' in result:
-                return jsonify(result), 407
+                return jsonify(result), 409
 
             return jsonify(result), 200
 
         except Exception as e:
             logging.exception(
                 f"Unexpected error: {e}")
-            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error', error=str(e))}), 500
+            return jsonify({"error": True, "error_message": self.i18n_service.t('errors.general.unexpected_error')}), 500

iatoolkit/views/load_company_configuration_api_view.py

@@ -0,0 +1,49 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import redirect, url_for, jsonify, request, g
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.configuration_service import ConfigurationService
+import logging
+
+class LoadCompanyConfigurationApiView(MethodView):
+    @inject
+    def __init__(self,
+                 configuration_service: ConfigurationService,
+                 profile_service: ProfileService,
+                 auth_service: AuthService):
+        self.configuration_service = configuration_service
+        self.profile_service = profile_service
+        self.auth_service = auth_service
+
+    def get(self, company_short_name: str = None):
+        try:
+            # 1. Get the authenticated user's
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return jsonify({"error": "company not found."}), 404
+
+            config, errors = self.configuration_service.load_configuration(company_short_name)
+            if config:
+                self.configuration_service.register_data_sources(company_short_name)
+
+            # this is fo avoid serialization issues
+            if 'company' in config:
+                config.pop('company')
+
+            status_code = 200 if not errors else 400
+            return {'config': config, 'errors': [errors]}, status_code
+        except Exception as e:
+            logging.exception(f"Unexpected error: {e}")
+            return {'status': 'error'}, 500
+
+

iatoolkit/views/{file_store_api_view.py → load_document_api_view.py}

@@ -5,21 +5,22 @@
 
 from flask.views import MethodView
 from flask import request, jsonify
-from iatoolkit.services.load_documents_service import LoadDocumentsService
-from iatoolkit.services.auth_service import AuthService
-from iatoolkit.repositories.profile_repo import ProfileRepo
 from injector import inject
 import base64
 
+from iatoolkit.services.knowledge_base_service import KnowledgeBaseService
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.repositories.profile_repo import ProfileRepo
 
-class FileStoreApiView(MethodView):
+
+class LoadDocumentApiView(MethodView):
     @inject
     def __init__(self,
                  auth_service: AuthService,
-                 doc_service: LoadDocumentsService,
-                 profile_repo: ProfileRepo,):
+                 knowledge_base_service: KnowledgeBaseService,
+                 profile_repo: ProfileRepo):
         self.auth_service = auth_service
-        self.doc_service = doc_service
+        self.knowledge_base_service = knowledge_base_service
         self.profile_repo = profile_repo
 
     def post(self):
@@ -48,18 +49,21 @@ class FileStoreApiView(MethodView):
             # get the file content from base64
             content = base64.b64decode(base64_content)
 
-            new_document = self.doc_service._file_processing_callback(
+            # Use KnowledgeBaseService for ingestion
+            new_document = self.knowledge_base_service.ingest_document_sync(
+                company=company,
                 filename=filename,
                 content=content,
-                company=company,
-                context={'metadata': metadata})
+                metadata=metadata
+            )
 
             return jsonify({
                 "document_id": new_document.id,
+                "status": "active" # ingest_document_sync returns ACTIVE on success
             }), 200
 
         except Exception as e:
             response = jsonify({"error": str(e)})
             response.status_code = 500
 
-            return response
+            return response

iatoolkit/views/login_view.py

@@ -5,12 +5,12 @@
 
 from flask.views import MethodView
 from flask import (request, redirect, render_template, url_for,
-                   render_template_string, flash)
+                   render_template_string, flash, make_response)
 from injector import inject
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.jwt_service import JWTService
 from iatoolkit.services.query_service import QueryService
-from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.prompt_service import PromptService
 from iatoolkit.services.branding_service import BrandingService
 from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.i18n_service import I18nService
@@ -32,6 +32,7 @@ class LoginView(BaseLoginView):
         branding_data = self.branding_service.get_company_branding(company_short_name)
         email = request.form.get('email')
         password = request.form.get('password')
+        current_lang = request.form.get('lang') or request.args.get('lang') or 'en'
 
         # 1. Authenticate internal user
         auth_response = self.auth_service.login_local_user(
@@ -42,7 +43,14 @@ class LoginView(BaseLoginView):
 
         if not auth_response['success']:
             flash(auth_response["message"], 'error')
-            home_template = self.utility.get_company_template(company_short_name, "home.html")
+
+            # Resolve the correct template name based on language (e.g., home_en.html or home_es.html)
+            template_name = self.utility.get_template_by_language("home")
+            home_template = self.utility.get_company_template(company_short_name, template_name)
+
+            if not home_template:
+                return render_template('error.html',
+                                       message=f'Home template ({template_name}) not found.'), 500
 
             return render_template_string(
                 home_template,
@@ -57,7 +65,8 @@ class LoginView(BaseLoginView):
         # 3. define URL to call when slow path is finished
         target_url = url_for('finalize_no_token',
                              company_short_name=company_short_name,
-                             _external=True)
+                             _external=True,
+                             lang=current_lang)
 
         # 2. Delegate the path decision to the centralized logic.
         try:
@@ -97,6 +106,9 @@ class FinalizeContextView(MethodView):
 
     def get(self, company_short_name: str, token: str = None):
         try:
+            # get the languaje from the query string if it exists
+            current_lang = request.args.get('lang') or 'en'
+
             session_info = self.profile_service.get_current_session_info()
             if session_info:
                 # session exists, internal user
@@ -107,12 +119,12 @@ class FinalizeContextView(MethodView):
                 payload = self.jwt_service.validate_chat_jwt(token)
                 if not payload:
                     logging.warning("Fallo crítico: No se pudo leer el auth token.")
-                    return redirect(url_for('home', company_short_name=company_short_name))
+                    return redirect(url_for('home', company_short_name=company_short_name, lang=current_lang))
 
                 user_identifier = payload.get('user_identifier')
             else:
                 logging.error("missing session information or auth token")
-                return redirect(url_for('home', company_short_name=company_short_name))
+                return redirect(url_for('home', company_short_name=company_short_name, lang=current_lang))
 
             company = self.profile_service.get_company_by_short_name(company_short_name)
             if not company:
@@ -121,6 +133,8 @@ class FinalizeContextView(MethodView):
                             message="Empresa no encontrada"), 404
             branding_data = self.branding_service.get_company_branding(company_short_name)
 
+            default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
+
             # 2. Finalize the context rebuild (the heavy task).
             self.query_service.set_context_for_llm(
                 company_short_name=company_short_name,
@@ -134,6 +148,8 @@ class FinalizeContextView(MethodView):
             # Get the entire 'js_messages' block in the correct language.
             js_translations = self.i18n_service.get_translation_block('js_messages')
 
+            # Importante: no envolver con make_response; dejar que Flask gestione
+            # tanto strings como tuplas (string, status) que pueda devolver render_template
             return render_template(
                 "chat.html",
                 company_short_name=company_short_name,
@@ -142,7 +158,9 @@ class FinalizeContextView(MethodView):
                 prompts=prompts,
                 onboarding_cards=onboarding_cards,
                 js_translations=js_translations,
-                redeem_token=token
+                redeem_token=token,
+                llm_default_model=default_llm_model,
+                llm_available_models=available_llm_models,
             )
 
         except Exception as e:
@@ -150,4 +168,3 @@ class FinalizeContextView(MethodView):
                                    company_short_name=company_short_name,
                                    branding=branding_data,
                                    message=f"An unexpected error occurred during context loading: {str(e)}"), 500
-

iatoolkit/views/logout_api_view.py

@@ -4,7 +4,7 @@
 # IAToolkit is open source software.
 
 from flask.views import MethodView
-from flask import redirect, url_for, jsonify
+from flask import redirect, url_for, jsonify, request, g
 from injector import inject
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.profile_service import ProfileService
@@ -33,7 +33,15 @@ class LogoutApiView(MethodView):
             # get URL for redirection
             url_for_redirect = company.parameters.get('external_urls', {}).get('logout_url')
             if not url_for_redirect:
-                url_for_redirect = url_for('home', company_short_name=company_short_name)
+                current_lang = (
+                        request.args.get('lang')
+                        or getattr(g, 'lang', None)
+                        or 'en'
+                )
+
+                url_for_redirect = url_for('home',
+                                           company_short_name=company_short_name,
+                                           lang=current_lang)
 
             # clear de session cookie
             SessionManager.clear()

iatoolkit/views/prompt_api_view.py

@@ -5,7 +5,7 @@
 
 from flask import jsonify
 from flask.views import MethodView
-from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.prompt_service import PromptService
 from iatoolkit.services.auth_service import AuthService
 from injector import inject
 import logging