iatoolkit 0.58.4__tar.gz → 0.66.4__tar.gz

{iatoolkit-0.58.4 → iatoolkit-0.66.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.58.4
+Version: 0.66.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.58.4 → iatoolkit-0.66.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.58.4"
+version = "0.66.4"
 requires-python = ">=3.12"
 description = "IAToolkit"
 readme = "readme.md"
@@ -26,4 +26,4 @@ exclude = ["tests*", "*/tests*"]
 dependencies = { file = ["requirements.txt"] }
 
 [tool.setuptools.package-data]
-iatoolkit = ["system_prompts/*.prompt", "templates/**/*", "static/**/*"]
+iatoolkit = ["system_prompts/*.prompt", "templates/**/*", "static/**/*", "locales/*.yaml"]

{iatoolkit-0.58.4 → iatoolkit-0.66.4}/src/iatoolkit/base_company.py

@@ -29,11 +29,13 @@ class BaseCompany(ABC):
     def _create_company(self,
                         short_name: str,
                         name: str,
+                        parameters: dict | None = None,
                         branding: dict | None = None,
-                        onboarding_cards: 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)

{iatoolkit-0.58.4 → iatoolkit-0.66.4}/src/iatoolkit/common/routes.py

@@ -3,17 +3,8 @@
 #
 # 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_api_view import HistoryApiView
-import os
-
-
-def logout(company_short_name: str):
-    SessionManager.clear()
-    flash("Has cerrado sesión correctamente", "info")
-    return redirect(url_for('index', company_short_name=company_short_name))
 
 
 # this function register all the views
@@ -22,8 +13,8 @@ def register_views(injector, app):
     from iatoolkit.views.index_view import IndexView
     from iatoolkit.views.init_context_api_view import InitContextApiView
     from iatoolkit.views.llmquery_api_view import LLMQueryApiView
-    from iatoolkit.views.tasks_view import TaskView
-    from iatoolkit.views.tasks_review_view import TaskReviewView
+    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
@@ -32,12 +23,20 @@ def register_views(injector, app):
     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.chat_token_request_view import ChatTokenRequestView
+    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
 
     # iatoolkit home page
-    app.add_url_rule('/<company_short_name>', view_func=IndexView.as_view('index'))
+    app.add_url_rule('/', view_func=IndexView.as_view('index'))
+
+    # company home view
+    app.add_url_rule('/<company_short_name>/home', view_func=HomeView.as_view('home'))
 
     # login for the iatoolkit integrated frontend
     app.add_url_rule('/<company_short_name>/login', view_func=LoginView.as_view('login'))
@@ -57,22 +56,26 @@ def register_views(injector, app):
         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'))
 
-    # this endpoint is for requesting a chat token for external users
-    app.add_url_rule('/auth/chat_token',
-                     view_func=ChatTokenRequestView.as_view('chat-token'))
-
-    # init (reset) the company context (with api-key)
+    # init (reset) the company context
     app.add_url_rule('/<company_short_name>/api/init-context',
-                     view_func=InitContextApiView.as_view('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'))
@@ -84,13 +87,14 @@ def register_views(injector, app):
     # open the promt directory
     app.add_url_rule('/<company_short_name>/api/prompts', view_func=PromptApiView.as_view('prompt'))
 
-    # feedback and history
+    # 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=TaskView.as_view('tasks'))
-    app.add_url_rule('/tasks/review/<int:task_id>', view_func=TaskReviewView.as_view('tasks-review'))
+    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'))
@@ -132,6 +136,6 @@ def register_views(injector, app):
     # hacer que la raíz '/' vaya al home de iatoolkit
     @app.route('/')
     def root_redirect():
-        return redirect(url_for('index', company_short_name='sample_company'))
+        return redirect(url_for('index'))
 
 

{iatoolkit-0.58.4 → iatoolkit-0.66.4}/src/iatoolkit/common/session_manager.py

@@ -3,6 +3,8 @@
 #
 # IAToolkit is open source software.
 
+# This is the Flask connected session manager for IAToolkit
+
 from flask import session
 
 class SessionManager:

{iatoolkit-0.58.4 → iatoolkit-0.66.4}/src/iatoolkit/common/util.py

@@ -21,10 +21,8 @@ class Utility:
     def __init__(self):
         self.encryption_key = os.getenv('FERNET_KEY')
 
-
     def render_prompt_from_template(self,
                                     template_pathname: str,
-                                    query: str = None,
                                     client_data: dict = {},
                                     **kwargs) -> str:
 
@@ -37,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)
 
@@ -53,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:
         """
@@ -76,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)
@@ -87,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-0.58.4 → iatoolkit-0.66.4}/src/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
@@ -16,10 +16,10 @@ 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, singleton, Injector
+from injector import Binder, Injector, singleton
 from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
-IATOOLKIT_VERSION = "0.58.4"
+IATOOLKIT_VERSION = "0.66.4"
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
@@ -52,7 +52,7 @@ class IAToolkit:
         self.app = None
         self.db_manager = None
         self._injector = None
-        self.version = IATOOLKIT_VERSION
+        self.version = IATOOLKIT_VERSION    # default version
 
     @classmethod
     def get_instance(cls) -> 'IAToolkit':
@@ -88,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
@@ -96,13 +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()
 
-
-
         logging.info(f"🎉 IAToolkit v{self.version} inicializado correctamente")
         self._initialized = True
         return self.app
@@ -111,6 +110,22 @@ 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):
         # 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()
@@ -226,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
 
@@ -253,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:
@@ -302,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)
@@ -315,6 +331,8 @@ 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
@@ -324,8 +342,8 @@ class IAToolkit:
         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(AuthService, to=AuthService)
@@ -369,8 +387,18 @@ class IAToolkit:
         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 {
@@ -382,6 +410,8 @@ class IAToolkit:
                 '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-0.66.4/src/iatoolkit/locales/en.yaml

@@ -0,0 +1,167 @@
+# Language: English
+ui:
+  login_widget:
+    title: "Sign In"
+    welcome_message: "Enter your credentials or register to access Sample Company’s AI platform."
+    email_placeholder: "Email address"
+    password_placeholder: "Password"
+    login_button: "Login"
+    forgot_password_link: "Forgot your password?"
+    no_account_prompt: "Don't have an account?"
+    signup_link: "Sign Up"
+
+  signup:
+    title: "Create an Account"
+    subtitle: "Start your journey with us today."
+    first_name_label: "First Name"
+    last_name_label: "Last Name"
+    email_label: "Email Address"
+    password_label: "Password"
+    confirm_password_label: "Confirm Password"
+    signup_button: "Create Account"
+    already_have_account: "Already have an account?"
+    login_link: "Log In"
+    disclaimer: "🔒We value your privacy. Your data will be used exclusively for the operation of the platform."
+
+  forgot_password:
+    title: "Recover Password"
+    subtitle: "Enter your email address and we will send you a link to reset your password."
+    submit_button: "Send Recovery Link"
+    back_to_login: "Back to Login"
+
+  change_password:
+    title: "Create New Password"
+    subtitle: "You are changing the password for <strong>{email}</strong>."
+    temp_code_label: "Temporary Code"
+    temp_code_placeholder: "Check your email"
+    new_password_label: "New Password"
+    password_instructions: "Must contain at least 8 characters, an uppercase letter, a lowercase letter, a number, and a special character."
+    confirm_password_label: "Confirm New Password"
+    save_button: "Save Password"
+    back_to_home: "Back to home"
+
+  chat:
+    welcome_message: "Hello! How can I help you today?"
+    input_placeholder: "Type your query here..."
+    prompts_available: "Available prompts"
+
+  tooltips:
+    history: "History of my queries"
+    reload_context: "Force Context Reload"
+    feedback: "Your feedback is very important"
+    usage_guide: "Usage Guide"
+    onboarding: "How to ask better questions"
+    logout: "Log out"
+    use_prompt_assistant: "Use Prompt Assistant"
+    attach_files: "Attach files"
+    view_attached_files: "View attached files"
+    send: "Send"
+    stop: "Stop"
+
+  modals:
+    files_title: "Uploaded Files"
+    feedback_title: "Your Opinion is Important"
+    feedback_prompt: "How useful was the assistant's response?"
+    feedback_comment_label: "Your feedback helps us improve:"
+    feedback_comment_placeholder: "Write your opinion, suggestions, or comments here..."
+    history_title: "Query History"
+    history_table_date: "Date"
+    history_table_query: "Query"
+    loading_history: "Loading history..."
+    no_history_found: "No query history found."
+    help_title: "AI Assistant User Guide"
+
+  buttons:
+    cancel: "Close"
+    send: "Send"
+    stop: "Stop"
+
+errors:
+  company_not_found: "The company {company_short_name} does not exist."
+  timeout: "timeout expired."
+  auth:
+    invalid_password: "The provided password is incorrect."
+    user_not_found: "A user with that email address was not found."
+    invalid_or_expired_token: "Invalid or expired token."
+    session_creation_failed: "Could not create user session."
+    authentication_required: "Authentication required. No session cookie or API Key provided."
+    invalid_api_key: "Invalid or inactive API Key."
+    no_user_identifier_api: "No user_identifier provided for API call."
+  templates:
+    company_not_found: "Company not found."
+    home_template_not_found: "The home page template for the company '{company_name}' is not configured."
+    template_not_found: "Template not found: '{template_name}'."
+
+    processing_error: "An error occurred while processing the custom home page template: {error}"
+  general:
+    unexpected_error: "An unexpected error has occurred. Please contact support."
+    unsupported_language: "The selected language is not supported."
+  signup:
+    company_not_found: "The company {company_name} does not exist."
+    incorrect_password_for_existing_user: "The password for the user {email} is incorrect."
+    user_already_registered: "The user with email '{email}' already exists in this company."
+    password_mismatch: "The passwords do not match. Please try again."
+  change_password:
+    token_expired: "The password change link has expired. Please request a new one."
+    password_mismatch: "The passwords do not match. Please try again."
+    invalid_temp_code: "The temporary code is not valid. Please check it or request a new one."
+  forgot_password:
+    user_not_registered: "The user with email {email} is not registered."
+  verification:
+    token_expired: "The verification link has expired. Please contact support if you need a new one."
+    user_not_found: "The user you are trying to verify does not exist."
+  validation:
+    password_too_short: "Password must be at least 8 characters long."
+    password_no_uppercase: "Password must contain at least one uppercase letter."
+    password_no_lowercase: "Password must contain at least one lowercase letter."
+    password_no_digit: "Password must contain at least one number."
+    password_no_special_char: "Password must contain at least one special character."
+
+  services:
+    no_text_file: "The file is not text or the encoding is not UTF-8"
+    no_output_file: "Missing output file name"
+    no_data_for_excel: "Missing data or it is not a list of dictionaries"
+    no_download_directory: "Temporary directory for saving Excel files is not configured"
+    cannot_create_excel: "Could not create the Excel file"
+    invalid_filename: "Invalid filename"
+    file_not_exist: "File not found"
+    path_is_not_a_file: "The path does not correspond to a file"
+    file_validation_error: "Error validating file"
+    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"
+
+
+api_responses:
+  context_reloaded_success: "The context has been successfully reloaded."
+
+services:
+  mail_sent: "Email sent successfully."
+  start_query: "Hello, what can I help you with today?"
+
+
+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."
+  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."
+
+# Keys specifically for JavaScript
+js_messages:
+  feedback_sent_success_title: "Feedback Sent"
+  feedback_sent_success_body: "Thank you for your feedback!"
+  feedback_sent_error: "Could not send feedback, please try again."
+  feedback_rating_error: "Please rate the assistant using the stars."
+  feedback_comment_error: "Please write your comment before sending."
+  context_reloaded: "Context has been reloaded."
+  error_loading_history: "An error occurred while loading the history."
+  request_aborted: "The response generation has been stopped."
+  processing_error: "An error occurred while processing the request."
+  server_comm_error: "Server communication error ({status}). Please try again later."
+  network_error: "A network error occurred. Please try again in a few moments."
+  unknown_server_error: "Unknown server error."
+  loading: "Loading..."
+  reload_init: "init reloading context in background..."
+  no_history_found: "No query history found."