iatoolkit 0.95.4__tar.gz → 1.10.0__tar.gz

{iatoolkit-0.95.4/src/iatoolkit.egg-info → iatoolkit-1.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.95.4
+Version: 1.10.0
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
@@ -14,7 +14,6 @@ Requires-Dist: botocore==1.36.22
 Requires-Dist: build==1.2.2.post1
 Requires-Dist: click==8.1.8
 Requires-Dist: cryptography==44.0.3
-Requires-Dist: deepseek==1.0.0
 Requires-Dist: Flask==3.1.0
 Requires-Dist: Flask-Bcrypt==1.0.1
 Requires-Dist: flask-cors==6.0.0

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/requirements.txt

@@ -4,7 +4,6 @@ botocore==1.36.22
 build==1.2.2.post1
 click==8.1.8
 cryptography==44.0.3
-deepseek==1.0.0
 Flask==3.1.0
 Flask-Bcrypt==1.0.1
 flask-cors==6.0.0

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "0.95.4"
+__version__ = "1.10.0"
 
 # Expose main classes and functions at the top level of the package
 
@@ -17,13 +17,14 @@ from .base_company import BaseCompany
 # --- Services ---
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.document_service import DocumentService
-from iatoolkit.services.search_service import SearchService
+from iatoolkit.services.knowledge_base_service import KnowledgeBaseService
 from iatoolkit.services.sql_service import SqlService
 from iatoolkit.services.load_documents_service import LoadDocumentsService
 from iatoolkit.infra.call_service import CallServiceClient
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.mail_service import MailService
 from iatoolkit.repositories.models import Base as OrmModel
+from iatoolkit.base_company import BaseCompany
 
 __all__ = [
     'IAToolkit',
@@ -35,10 +36,11 @@ __all__ = [
     'QueryService',
     'SqlService',
     'DocumentService',
-    'SearchService',
+    'KnowledgeBaseService',
     'LoadDocumentsService',
     'CallServiceClient',
     'ProfileService',
     'MailService',
-    'OrmModel'
+    'OrmModel',
+    'BaseCompany',
 ]

iatoolkit-1.10.0/src/iatoolkit/base_company.py

@@ -0,0 +1,21 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+# companies/base_company.py
+from abc import ABC, abstractmethod
+
+class BaseCompany(ABC):
+
+    @abstractmethod
+    # execute the specific action configured in the intent table
+    def handle_request(self, tag: str, params: dict) -> dict:
+        raise NotImplementedError("La subclase debe implementar el método handle_request()")
+
+    @abstractmethod
+    def register_cli_commands(self, app):
+        pass
+
+    def unsupported_operation(self, tag):
+        raise NotImplementedError(f"La operación '{tag}' no está soportada por esta empresa.")

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/cli_commands.py

@@ -14,12 +14,13 @@ def register_core_commands(app):
 
     @app.cli.command("api-key")
     @click.argument("company_short_name")
-    def api_key(company_short_name: str):
+    @click.argument("key_name")
+    def api_key(company_short_name: str, key_name: str):
         """⚙️ Genera una nueva API key para una compañía ya registrada."""
         try:
             profile_service = IAToolkit.get_instance().get_injector().get(ProfileService)
             click.echo(f"🔑 Generating API-KEY for company: '{company_short_name}'...")
-            result = profile_service.new_api_key(company_short_name)
+            result = profile_service.new_api_key(company_short_name, key_name)
 
             if 'error' in result:
                 click.echo(f"❌ Error: {result['error']}")

iatoolkit-1.10.0/src/iatoolkit/common/interfaces/asset_storage.py

@@ -0,0 +1,34 @@
+from enum import Enum
+from typing import List
+import abc
+
+
+class AssetType(Enum):
+    CONFIG = "config"
+    PROMPT = "prompts"
+    SCHEMA = "schema"
+    CONTEXT = "context"
+
+
+class AssetRepository(abc.ABC):
+    @abc.abstractmethod
+    def exists(self, company_short_name: str, asset_type: AssetType, filename: str) -> bool:
+        pass
+
+    @abc.abstractmethod
+    def read_text(self, company_short_name: str, asset_type: AssetType, filename: str) -> str:
+        pass
+
+    @abc.abstractmethod
+    def list_files(self, company_short_name: str, asset_type: AssetType, extension: str = None) -> List[str]:
+        pass
+
+    @abc.abstractmethod
+    def write_text(self, company_short_name: str, asset_type: AssetType, filename: str, content: str) -> None:
+        """Creates or updates a text asset."""
+        pass
+
+    @abc.abstractmethod
+    def delete(self, company_short_name: str, asset_type: AssetType, filename: str) -> None:
+        """Deletes an asset if it exists."""
+        pass

iatoolkit-1.10.0/src/iatoolkit/common/interfaces/database_provider.py

@@ -0,0 +1,43 @@
+import abc
+from typing import Any, List, Dict, Union
+
+class DatabaseProvider(abc.ABC):
+    """
+    Abstract interface for interacting with a database source.
+    Handles both metadata introspection and query execution.
+    """
+
+    # --- Schema Methods ---
+    @abc.abstractmethod
+    def get_database_structure(self) -> dict:
+        """
+        Returns the structure of the database (tables, columns, types)
+        Format:
+        {
+            "table_name": {
+                "columns": [
+                    {"name": "col1", "type": "VARCHAR", "nullable": True, "pk": True},
+                    ...
+                ]
+            }
+        }
+        """
+        pass
+
+    # --- Execution Methods ---
+    @abc.abstractmethod
+    def execute_query(self, query: str, commit: bool = False) -> Union[List[Dict[str, Any]], Dict[str, int]]:
+        """
+        Executes a query and returns:
+         - A list of dicts for SELECT (rows).
+         - A dict {'rowcount': N} for INSERT/UPDATE/DELETE.
+        """
+        pass
+
+    @abc.abstractmethod
+    def commit(self) -> None:
+        pass
+
+    @abc.abstractmethod
+    def rollback(self) -> None:
+        pass

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/common/model_registry.py

@@ -42,7 +42,7 @@ class ModelRegistry:
         # Hardcoded rules for now; can be extended or loaded from config later.
         # The order of patterns matters: first match wins.
         self._provider_patterns: dict[ProviderType, tuple[str, ...]] = {
-            "openai": ("gpt", "gpt-5"),
+            "openai": ("gpt", "gpt-5", "gpt-5-mini", "gpt-5.1"),
             "gemini": ("gemini", "gemini-3"),
             "deepseek": ("deepseek",),
             "xai": ("grok", "grok-1", "grok-beta"),
@@ -72,6 +72,57 @@ class ModelRegistry:
 
         return "unknown"
 
+    def get_request_defaults(self, model: str) -> dict:
+        """
+        Return per-model request defaults to keep model-specific policy centralized.
+
+        Notes:
+        - This should only include keys that are supported by the target provider.
+        - Callers should merge these defaults with user-provided params (do not mutate inputs).
+        """
+        model_lower = (model or "").lower()
+        provider = self.get_provider(model_lower)
+
+        # Conservative defaults: do not send provider-specific knobs unless we know they are supported.
+        defaults = {"text": {}, "reasoning": {}}
+
+        # OpenAI/xAI (OpenAI-compatible) support 'text.verbosity' and 'reasoning.effort' in our current integration.
+        if provider in ("openai", "xai"):
+            defaults["text"] = {"verbosity": "low"}
+
+            # Fine-grained per-model tuning.
+            if model_lower in ("gpt-5", "gpt-5-mini"):
+                defaults["reasoning"] = {"effort": "minimal"}
+            elif model_lower == "gpt-5.1":
+                defaults["reasoning"] = {"effort": "low", "summary": "auto"}
+
+        # Gemini/DeepSeek/unknown: keep defaults empty to avoid sending unsupported parameters.
+        return defaults
+
+    def resolve_request_params(self, model: str, text: dict | None = None, reasoning: dict | None = None) -> dict:
+        """
+        Resolve provider/model defaults and merge them with caller-provided overrides.
+
+        Rules:
+        - Defaults come from get_request_defaults(model).
+        - Caller overrides win over defaults.
+        - Input dictionaries are never mutated.
+        """
+        defaults = self.get_request_defaults(model)
+
+        merged_text: dict = {}
+        merged_text.update(defaults.get("text") or {})
+        merged_text.update(text or {})
+
+        merged_reasoning: dict = {}
+        merged_reasoning.update(defaults.get("reasoning") or {})
+        merged_reasoning.update(reasoning or {})
+
+        return {
+            "text": merged_text,
+            "reasoning": merged_reasoning,
+        }
+
     def get_history_type(self, model: str) -> HistoryType:
         """
         Returns the history strategy for a given model.

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/common/routes.py

@@ -24,10 +24,15 @@ def register_views(app):
     from iatoolkit.views.profile_api_view import UserLanguageApiView
     from iatoolkit.views.embedding_api_view import EmbeddingApiView
     from iatoolkit.views.login_view import LoginView, FinalizeContextView
+    from iatoolkit.views.configuration_api_view import ConfigurationApiView, ValidateConfigurationApiView
     from iatoolkit.views.logout_api_view import LogoutApiView
     from iatoolkit.views.home_view import HomeView
+    from iatoolkit.views.chat_view import ChatView
     from iatoolkit.views.static_page_view import StaticPageView
     from iatoolkit.views.root_redirect_view import RootRedirectView
+    from iatoolkit.views.users_api_view import UsersApiView
+    from iatoolkit.views.rag_api_view import RagApiView
+    from iatoolkit.views.categories_api_view import CategoriesApiView
 
     # assign root '/' to our new redirect logic
     app.add_url_rule('/home', view_func=RootRedirectView.as_view('root_redirect'))
@@ -38,6 +43,10 @@ def register_views(app):
     # login for the iatoolkit integrated frontend
     app.add_url_rule('/<company_short_name>/login', view_func=LoginView.as_view('login'))
 
+    # Chat Route (Direct Access)
+    app.add_url_rule('/<company_short_name>/chat',
+                     view_func=ChatView.as_view('chat'))
+
     # this endpoint is called when onboarding_shell finish the context load
     app.add_url_rule(
         '/<company_short_name>/finalize',
@@ -68,19 +77,62 @@ def register_views(app):
     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'))
+    app.add_url_rule(
+        '/<string:company_short_name>/api/company-users',
+        view_func=UsersApiView.as_view('company-users')
+    )
 
     # 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'))
 
-    # open the promt directory
-    app.add_url_rule('/<company_short_name>/api/prompts', view_func=PromptApiView.as_view('prompt'))
-
+    # Categories Endpoint
+    app.add_url_rule('/<company_short_name>/api/categories',
+                     view_func=CategoriesApiView.as_view('categories_api'),
+                     methods=['GET', 'POST'])
+
+    # open the promt directory and specific prompt management
+    prompt_view = PromptApiView.as_view('prompt')
+    app.add_url_rule('/<company_short_name>/api/prompts',
+                     view_func=prompt_view,
+                     methods=['GET', 'POST'],
+                     defaults={'prompt_name': None})
+
+    app.add_url_rule('/<company_short_name>/api/prompts/<prompt_name>',
+                     view_func=prompt_view,
+                     methods=['GET', 'POST','PUT', 'DELETE'])
     # 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'))
 
+    # --- RAG API Routes ---
+    rag_view = RagApiView.as_view('rag_api')
+
+    # 1. List Files (POST for filters)
+    app.add_url_rule('/api/rag/<company_short_name>/files',
+                     view_func=rag_view,
+                     methods=['POST'],
+                     defaults={'action': 'list_files'})
+
+    # 2. Delete File
+    app.add_url_rule('/api/rag/<company_short_name>/files/<int:document_id>',
+                     view_func=rag_view,
+                     methods=['DELETE'],
+                     defaults={'action': 'delete_file'})
+
+    # 3. Search Lab
+    app.add_url_rule('/api/rag/<company_short_name>/search',
+                     view_func=rag_view,
+                     methods=['POST'],
+                     defaults={'action': 'search'})
+
+    # 4. Get File Content (View/Download)
+    app.add_url_rule('/api/rag/<company_short_name>/files/<int:document_id>/content',
+                     view_func=rag_view,
+                     methods=['GET'],
+                     defaults={'action': 'get_file_content'})
+
     # this endpoint is for upload documents into the vector store (api-key)
     app.add_url_rule('/api/load-document', view_func=LoadDocumentApiView.as_view('load-document'), methods=['POST'])
 
@@ -88,6 +140,15 @@ def register_views(app):
     app.add_url_rule('/<company_short_name>/api/embedding',
                      view_func=EmbeddingApiView.as_view('embedding_api'))
 
+    # company configuration
+    app.add_url_rule('/<company_short_name>/api/configuration',
+                     view_func=ConfigurationApiView.as_view('configuration'),
+                     methods=['GET', 'POST', 'PATCH'],)
+
+    app.add_url_rule('/<company_short_name>/api/configuration/validate',
+                     view_func=ValidateConfigurationApiView.as_view('configuration-validate'),
+                     methods=['GET'])
+
     # static pages
     # url: /pages/foundation o /pages/implementation_plan
     static_view = StaticPageView.as_view('static_pages')

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/common/util.py

@@ -157,6 +157,31 @@ class Utility:
             schema = yaml.safe_load(f)
         return schema
 
+    def load_yaml_from_string(self, yaml_content: str) -> dict:
+        """
+        Parses a YAML string into a dictionary securely.
+        """
+        try:
+            yaml_content = yaml_content.replace('\t', '  ')
+            return yaml.safe_load(yaml_content) or {}
+        except yaml.YAMLError as e:
+            logging.error(f"Error parsing YAML string: {e}")
+            return {}
+
+    def dump_yaml_to_string(self, config: dict) -> str:
+        """
+        Dumps a dictionary into a YAML formatted string.
+        It uses default flow style False to ensure block format (readable YAML).
+        """
+        try:
+            # default_flow_style=False ensures lists and dicts are expanded (not inline like JSON)
+            # allow_unicode=True ensures characters like accents are preserved
+            return yaml.safe_dump(config, default_flow_style=False, allow_unicode=True, sort_keys=False)
+        except yaml.YAMLError as e:
+            logging.error(f"Error dumping YAML to string: {e}")
+            raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
+                                     f"Failed to generate YAML: {e}")
+
     def generate_context_for_schema(self, entity_name: str, schema_file: str = None, schema: dict = {}) -> str:
         if not schema_file and not schema:
             raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
@@ -186,9 +211,13 @@ class Utility:
             root_name = list(schema.keys())[0]
             root_details = schema[root_name]
 
+            # support this format
+            if root_details.get('columns'):
+                root_details = root_details['columns']
+
             if isinstance(root_details, dict):
                 # Las claves de metadatos describen el objeto en sí, no sus propiedades hijas.
-                METADATA_KEYS = ['description', 'type', 'format', 'items', 'properties']
+                METADATA_KEYS = ['description', 'type', 'format', 'items', 'properties', 'pk']
 
                 # Las propiedades son las claves restantes en el diccionario.
                 properties = {
@@ -231,6 +260,8 @@ class Utility:
             description = details.get('description', '')
             data_type = details.get('type', 'any')
             output.append(f"{indent_str}- **`{name.lower()}`** ({data_type}): {description}")
+            # if 'pk' in details and details['pk']:
+            #    output.append(f"{indent_str}- **Primary Key**: {details['pk']}")
 
             child_indent_str = '  ' * (indent_level + 1)
 

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/company_registry.py

@@ -6,6 +6,7 @@
 from typing import Dict, Type, Any, Optional
 from .base_company import BaseCompany
 import logging
+from injector import inject
 
 
 class CompanyRegistry:
@@ -14,6 +15,7 @@ class CompanyRegistry:
     Allow the client to register companies and instantiate them with dependency injection.
     """
 
+    @inject
     def __init__(self):
         self._company_classes: Dict[str, Type[BaseCompany]] = {}
         self._company_instances: Dict[str, BaseCompany] = {}
@@ -85,6 +87,9 @@ def get_company_registry() -> CompanyRegistry:
     """Get the global company registry instance."""
     return _company_registry
 
+def get_registered_companies() -> Dict[str, Type[BaseCompany]]:
+    return _company_registry.get_registered_companies()
+
 
 def set_company_registry(registry: CompanyRegistry) -> None:
     """

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/core.py

@@ -9,20 +9,26 @@ from flask_injector import FlaskInjector
 from flask_bcrypt import Bcrypt
 from flask_cors import CORS
 from iatoolkit.common.exceptions import IAToolkitException
-from typing import Optional, Dict, Any
 from iatoolkit.repositories.database_manager import DatabaseManager
+from iatoolkit.common.interfaces.asset_storage import AssetRepository
+from iatoolkit.company_registry import get_registered_companies
 from werkzeug.middleware.proxy_fix import ProxyFix
 from injector import Binder, Injector, singleton
+from typing import Optional, Dict, Any
 from urllib.parse import urlparse
 import redis
 import logging
 import os
 
 from iatoolkit import __version__ as IATOOLKIT_VERSION
+from iatoolkit.services.configuration_service import ConfigurationService
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
 
+def is_bound(injector: Injector, cls) -> bool:
+    return cls in injector.binder._bindings
+
 class IAToolkit:
     """
     IAToolkit main class
@@ -49,8 +55,8 @@ class IAToolkit:
         self.config = config or {}
         self.app = None
         self.db_manager = None
-        self._injector = None
-        self.version = IATOOLKIT_VERSION    # default version
+        self._injector = Injector()         # init empty injector
+        self.version = IATOOLKIT_VERSION
         self.license = "Community Edition"
 
     @classmethod
@@ -61,7 +67,7 @@ class IAToolkit:
             _iatoolkit_instance = cls()
         return _iatoolkit_instance
 
-    def create_iatoolkit(self):
+    def create_iatoolkit(self, start: bool = True):
         """
             Creates, configures, and returns the Flask application instance.
             this is the main entry point for the application factory.
@@ -77,8 +83,8 @@ class IAToolkit:
         # Step 2: Set up the core components that DI depends on
         self._setup_database()
 
-        # Step 3: Create the Injector and configure all dependencies in one place
-        self._injector = Injector(self._configure_core_dependencies)
+        # Step 3: Configure dependencies using the existing injector
+        self._configure_core_dependencies(self._injector)
 
         # Step 4: Register routes using the fully configured injector
         self._register_routes()
@@ -98,6 +104,7 @@ class IAToolkit:
 
         # Step 8: Finalize setup within the application context
         self._setup_redis_sessions()
+
         self._setup_cors()
         self._setup_additional_services()
         self._setup_cli_commands()
@@ -107,11 +114,21 @@ class IAToolkit:
         # Step 9: define the download_dir
         self._setup_download_dir()
 
+        # register data source
+        if start:
+            self.register_data_sources()
+
         logging.info(f"🎉 IAToolkit {self.license} version {self.version} correctly initialized.")
         self._initialized = True
 
         return self.app
 
+    def register_data_sources(self):
+        # load the company configurations
+        configuration_service = self._injector.get(ConfigurationService)
+        for company in get_registered_companies():
+            configuration_service.register_data_sources(company)
+
     def _get_config_value(self, key: str, default=None):
         # get a value from the config dict or the environment variable
         return self.config.get(key, os.getenv(key, default))
@@ -181,11 +198,11 @@ class IAToolkit:
             os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
     def _setup_database(self):
-        database_uri = self._get_config_value('DATABASE_URI')
+        database_uri = self._get_config_value('DATABASE_URI') or self._get_config_value('DATABASE_URL')
         if not database_uri:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,
-                "DATABASE_URI is requires (config dict or env. variable)"
+                "DATABASE_URI is required (config dict or env. variable)"
             )
 
         self.db_manager = DatabaseManager(database_url=database_uri, schema='iatoolkit')
@@ -240,8 +257,9 @@ class IAToolkit:
         extra_origins = []
         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
+            if company_instance.company:
+                cors_origin = company_instance.company.parameters.get('cors_origin', [])
+                extra_origins += cors_origin
 
         all_origins = default_origins + extra_origins
 
@@ -256,8 +274,11 @@ class IAToolkit:
 
         logging.info(f"✅ CORS configured for: {all_origins}")
 
-    def _configure_core_dependencies(self, binder: Binder):
+    def _configure_core_dependencies(self, injector: Injector):
         """⚙️ Configures all system dependencies."""
+
+        # get the binder from injector
+        binder = injector.binder
         try:
             # Core dependencies
             binder.bind(Flask, to=self.app)
@@ -282,12 +303,17 @@ class IAToolkit:
         from iatoolkit.repositories.profile_repo import ProfileRepo
         from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
         from iatoolkit.repositories.vs_repo import VSRepo
+        from iatoolkit.repositories.filesystem_asset_repository import FileSystemAssetRepository
 
         binder.bind(DocumentRepo, to=DocumentRepo)
         binder.bind(ProfileRepo, to=ProfileRepo)
         binder.bind(LLMQueryRepo, to=LLMQueryRepo)
         binder.bind(VSRepo, to=VSRepo)
 
+        # this class can be setup befor by iatoolkit enterprise
+        if not is_bound(self._injector, AssetRepository):
+            binder.bind(AssetRepository, to=FileSystemAssetRepository)
+
     def _bind_services(self, binder: Binder):
         from iatoolkit.services.query_service import QueryService
         from iatoolkit.services.benchmark_service import BenchmarkService
@@ -308,7 +334,8 @@ class IAToolkit:
         from iatoolkit.services.tool_service import ToolService
         from iatoolkit.services.llm_client_service import llmClient
         from iatoolkit.services.auth_service import AuthService
-
+        from iatoolkit.services.sql_service import SqlService
+        from iatoolkit.services.knowledge_base_service import KnowledgeBaseService
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(BenchmarkService, to=BenchmarkService)
@@ -329,6 +356,8 @@ class IAToolkit:
         binder.bind(ToolService, to=ToolService)
         binder.bind(llmClient, to=llmClient)
         binder.bind(AuthService, to=AuthService)
+        binder.bind(SqlService, to=SqlService)
+        binder.bind(KnowledgeBaseService, to=KnowledgeBaseService)
 
     def _bind_infrastructure(self, binder: Binder):
         from iatoolkit.infra.llm_proxy import LLMProxy
@@ -405,11 +434,13 @@ class IAToolkit:
                 'app_name': 'IAToolkit',
                 'user_identifier': SessionManager.get('user_identifier'),
                 'company_short_name': SessionManager.get('company_short_name'),
+                'user_role': user_profile.get('user_role'),
                 'user_is_local': user_profile.get('user_is_local'),
                 'user_email': user_profile.get('user_email'),
                 'iatoolkit_base_url': request.url_root,
                 'flashed_messages': get_flashed_messages(with_categories=True),
-                't': translate_for_template
+                't': translate_for_template,
+                'google_analytics_id': self._get_config_value('GOOGLE_ANALYTICS_ID', ''),
             }
 
     def _get_default_static_folder(self) -> str:
@@ -473,6 +504,7 @@ class IAToolkit:
         logging.info(f"✅ download dir created in: {download_dir}")
 
 
+
 def current_iatoolkit() -> IAToolkit:
     return IAToolkit.get_instance()
 

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/infra/connectors/file_connector_factory.py

@@ -40,6 +40,7 @@ class FileConnectorFactory:
             return S3Connector(
                 bucket=config['bucket'],
                 prefix=config.get('prefix', ''),
+                folder=config.get('folder', ''),
                 auth=auth
             )
 

{iatoolkit-0.95.4 → iatoolkit-1.10.0}/src/iatoolkit/infra/connectors/s3_connector.py

@@ -9,14 +9,16 @@ from typing import List
 
 
 class S3Connector(FileConnector):
-    def __init__(self, bucket: str, prefix: str, auth: dict):
+    def __init__(self, bucket: str, prefix: str, folder: str, auth: dict):
         self.bucket = bucket
         self.prefix = prefix
+        self.folder = folder
         self.s3 = boto3.client('s3', **auth)
 
     def list_files(self) -> List[dict]:
         # list all the files as dictionaries, with keys:  'path', 'name' y 'metadata'.
-        response = self.s3.list_objects_v2(Bucket=self.bucket, Prefix=self.prefix)
+        prefix = f'{self.prefix}/{self.folder}/'
+        response = self.s3.list_objects_v2(Bucket=self.bucket, Prefix=prefix)
         files = response.get('Contents', [])
 
         return [