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

iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "1.4.2"
+__version__ = "1.7.0"
 
 # Expose main classes and functions at the top level of the package
 

iatoolkit/common/interfaces/database_provider.py

@@ -9,14 +9,19 @@ class DatabaseProvider(abc.ABC):
 
     # --- Schema Methods ---
     @abc.abstractmethod
-    def get_all_table_names(self) -> List[str]:
-        pass
-
-    @abc.abstractmethod
-    def get_table_description(self,
-                              table_name: str,
-                              schema_object_name: str | None = None,
-                              exclude_columns: List[str] | None = None) -> str:
+    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 ---

iatoolkit/common/routes.py

@@ -24,7 +24,7 @@ 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.load_company_configuration_api_view import LoadCompanyConfigurationApiView
+    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
@@ -128,8 +128,13 @@ def register_views(app):
                      view_func=EmbeddingApiView.as_view('embedding_api'))
 
     # company configuration
-    app.add_url_rule('/<company_short_name>/api/load_configuration',
-                     view_func=LoadCompanyConfigurationApiView.as_view('load-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

iatoolkit/common/util.py

@@ -168,6 +168,20 @@ class Utility:
             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,
@@ -197,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 = {
@@ -242,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/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/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 [

iatoolkit/locales/en.yaml

@@ -78,7 +78,6 @@ ui:
     saved_ok: "File saved successfully"
     save_file: "Save"
     credentials: "Email and password required for login."
-    saved_ok: "Saved successfully"
     deleted_ok: "Deleted successfully"
     user_manager: "User management"
     admin_page_title: "Admin access"
@@ -88,6 +87,38 @@ ui:
     error_loading: "Error loading file content"
     loading: "Loading..."
 
+  db_explorer:
+    data_explorer: "Data Explorer"
+    database_connection: "Database Connection"
+    tables: "Tables"
+    data_explorer_description: "Explore your database tables ."
+    select_database: "Please select a database above to view its tables."
+    not_table_selected: "No table selected"
+    view_yaml: "View YAML"
+    save_schema: "Save schema"
+    select_table: "Select a table from the left panel"
+    table_semantics: "Table Semantics"
+    table_description: "Description (AI Context)"
+    table_placeholder: "Describe what data this table contains (e.g., 'Active and inactive customer records including billing addresses')..."
+    column_metadata: "Column Metadata"
+    auto_detect: "Auto-detected from DB"
+    meta_column: "Column"
+    meta_type: "Type"
+    meta_description: "Description"
+    meta_synonyms: "Synonyms"
+    pii_sesitive: "PII Sensitive"
+
+  config:
+    editor_description: "IAToolkit configuration file"
+    title: "Configuration Editor"
+    sections: "Sections"
+    refresh: "Refresh"
+    validate: "Validate"
+    select_section: "Select a section from the left panel"
+    no_section_selected: "No section selected"
+    view_yaml: "View YAML"
+
+
   rag:
     ingestion: "Ingestion"
     ingestion_description: "Ingest documents into the knowledge base."
@@ -115,8 +146,11 @@ ui:
     delete_message: "This action cannot be undone. The file will be permanently removed."
     delete_button: "Delete"
     delete_cancel: "Cancel"
-
-
+    target_collection: "Target collection"
+    select_collection_placeholder: "Select a collection"
+    collection_required: "Collection is required"
+    collection: "Collection"
+    all_collections: "All collections"
 
   tooltips:
     history: "History of my queries"

iatoolkit/locales/es.yaml

@@ -62,7 +62,6 @@ ui:
     monitoring: "Monitoreo"
     teams: "Usuarios"
     billing: "Facturación"
-    company: "Empresa"
     files: "Archivos"
     select_file: "Seleccione un archivo a editar"
     select_category: "Seleccione una categoría"
@@ -85,6 +84,38 @@ ui:
     goto_chat: "Ir al chat"
     logout: "Cerrar sesión"
 
+  db_explorer:
+    data_explorer: "Explorador de datos"
+    database_connection: "Bases de datos"
+    tables: "Tablas"
+    data_explorer_description: "Explora tus tablas de base de datos."
+    select_database: "Seleccione una base de datos para ver sus tablas."
+    not_table_selected: "No hay tabla seleccionada."
+    view_yaml: "Ver YAML"
+    save_schema: "Guardar esquema"
+    select_table: "Seleccionar una tabla del panel izquierdo"
+    table_semantics: "Significado de la tabla"
+    table_description: "Descripción de la tabla (Contexto IA)"
+    table_placeholder: "Describe el significado de la tabla, ejemplo: tabla de usuarios."
+    column_metadata: "Metadatos de columna"
+    auto_detect: "Auto-detectar desde la BD"
+    meta_column: "Columna"
+    meta_type: "Tipo"
+    meta_description: "Descripción"
+    meta_synonyms: "Sinonimos"
+    pii_sesitive: "IP Sensible"
+
+  config:
+    editor_description: "Editor de configuración"
+    title: "Editor de configuraciones"
+    sections: "Secciones"
+    refresh: "Refrescar"
+    validate: "Validar"
+    select_section: "Seleccione una sección del panel izquierdo"
+    no_section_selected: "No hay sección seleccionada."
+    view_yaml: "Ver YAML"
+
+
   rag:
     ingestion: "Ingesta"
     ingestion_description: "Ingestar documentos en la base de conocimiento."
@@ -101,7 +132,7 @@ ui:
     filename_placeholder: "Contiene..."
     user: "Usuario"
     status: "Estado"
-    all_status: "Todos los estados"
+    all_status: "Estados"
     status_active: "Activo"
     status_pending: "Pendiente"
     status_processing: "Procesando"
@@ -112,6 +143,11 @@ ui:
     delete_message: "Esta acción no se puede deshacer. El archivo se eliminará permanentemente."
     delete_button: "Eliminar"
     delete_cancel: "Cancelar"
+    target_collection: "Categoría seleccionada"
+    select_collection_placeholder: "Selecciona una categoría"
+    collection_required: "Debe seleccionar una categoría"
+    all_collections: "Todas las categorías"
+    collection: "Categoría"
 
 
   tooltips:

iatoolkit/repositories/database_manager.py

@@ -11,6 +11,7 @@ from iatoolkit.repositories.models import Base
 from injector import inject
 from pgvector.psycopg2 import register_vector
 from iatoolkit.common.interfaces.database_provider import DatabaseProvider
+import logging
 
 
 class DatabaseManager(DatabaseProvider):
@@ -135,51 +136,30 @@ class DatabaseManager(DatabaseProvider):
         self.get_session().rollback()
 
     # -- schema methods ----
-    def get_all_table_names(self) -> list[str]:
-        # Returns a list of all table names in the database
+    def get_database_structure(self) -> dict:
         inspector = inspect(self._engine)
-        return inspector.get_table_names(schema=self.schema)
-
-    def get_table_description(self,
-                              table_name: str,
-                              schema_object_name: str | None = None,
-                              exclude_columns: list[str] | None = None) -> str:
-        inspector = inspect(self._engine)
-
-        # search the table in the specified schema
-        if table_name not in inspector.get_table_names(schema=self.schema):
-            raise RuntimeError(f"Table '{table_name}' does not exist in database schema '{self.schema}'.")
-
-        if exclude_columns is None:
-            exclude_columns = []
-
-        # get all the table columns
-        columns = inspector.get_columns(table_name, schema=self.schema)
-
-        # construct a json dictionary with the table definition
-        json_dict = {
-            "table": table_name,
-            "schema": self.schema,
-            "description": f"The table belongs to the **`{self.schema}`** schema.",
-            "fields": []
-        }
-
-        if schema_object_name:
-            json_dict["description"] += (
-                f"The meaning of each field in this table is detailed in the **`{schema_object_name}`** object."
-            )
-
-        # now add every column to the json dictionary
-        for col in columns:
-            name = col["name"]
-
-            # omit the excluded columns.
-            if name in exclude_columns:
-                continue
-
-            json_dict["fields"].append({
-                "name": name,
-                "type": str(col["type"]),
-            })
-
-        return "\n\n" + str(json_dict)
+        structure = {}
+        for table in inspector.get_table_names(schema=self.schema):
+            columns_data = []
+
+            # get columns
+            try:
+                columns = inspector.get_columns(table, schema=self.schema)
+                # Obtener PKs para marcarlas
+                pks = inspector.get_pk_constraint(table, schema=self.schema).get('constrained_columns', [])
+
+                for col in columns:
+                    columns_data.append({
+                        "name": col['name'],
+                        "type": str(col['type']),
+                        "nullable": col.get('nullable', True),
+                        "pk": col['name'] in pks
+                    })
+            except Exception as e:
+                logging.warning(f"Could not inspect columns for table {table}: {e}")
+
+            structure[table] = {
+                "columns": columns_data
+            }
+
+        return structure

iatoolkit/services/company_context_service.py

@@ -104,16 +104,17 @@ class CompanyContextService:
                 continue
 
             # get database schema definition, for this source.
-            database_schema_name = source.get('schema')
+            database_schema_name = source.get('schema', 'public')
 
             try:
-                db_provider = self.sql_service.get_database_provider(company_short_name, db_name)
+                # 1. Get the full database structure at once using the SQL service
+                db_structure = self.sql_service.get_database_structure(company_short_name, db_name)
             except IAToolkitException as e:
-                logging.warning(f"Could not get DB provider for '{db_name}': {e}")
+                logging.warning(f"Could not get DB structure for '{db_name}': {e}")
                 continue
 
             db_description = source.get('description', '')
-            sql_context = f"***Database (`database_key`)***: {db_name}\n"
+            sql_context += f"***Database (`database_key`)***: {db_name}\n"
 
             if db_description:
                 sql_context += (
@@ -131,28 +132,32 @@ class CompanyContextService:
                 f"Use exactly: `database_key='{db_name}'`.\n"
             )
 
-            # 1. get the list of tables to process.
+            # 2. get the list of tables to process based on structure and config
             tables_to_process = []
             if source.get('include_all_tables', False):
-                all_tables = db_provider.get_all_table_names()
+                # Use keys from the fetched structure
+                all_tables = list(db_structure.keys())
                 tables_to_exclude = set(source.get('exclude_tables', []))
                 tables_to_process = [t for t in all_tables if t not in tables_to_exclude]
             elif 'tables' in source:
-                # if not include_all_tables, use the list of tables explicitly specified in the map.
-                tables_to_process = list(source['tables'].keys())
+                # Use keys from the config map, but check if they exist in DB structure
+                config_tables = list(source['tables'].keys())
+                tables_to_process = [t for t in config_tables if t in db_structure]
 
-            # 2. get the global settings and overrides.
+            # 3. get the global settings and overrides.
             global_exclude_columns = source.get('exclude_columns', [])
             table_prefix = source.get('table_prefix')
             table_overrides = source.get('tables', {})
 
-            # 3. iterate over the tables.
+            # 4. iterate over the tables.
             for table_name in tables_to_process:
                 try:
-                    # 4. get the table specific configuration.
+                    table_data = db_structure[table_name]
+
+                    # 5. get the table specific configuration.
                     table_config = table_overrides.get(table_name, {})
 
-                    # 5. define the schema object name, using the override if it exists.
+                    # 6. define the schema object name, using the override if it exists.
                     # Priority 1: Explicit override from the 'tables' map.
                     schema_object_name = table_config.get('schema_name')
 
@@ -164,17 +169,36 @@ class CompanyContextService:
                             # Priority 4: Default to the table name itself.
                             schema_object_name = table_name
 
-                    # 6. define the list of columns to exclude, (local vs. global).
+                    # 7. define the list of columns to exclude, (local vs. global).
                     local_exclude_columns = table_config.get('exclude_columns')
                     final_exclude_columns = local_exclude_columns if local_exclude_columns is not None else global_exclude_columns
 
-                    # 7. get the table schema definition.
-                    table_definition = db_provider.get_table_description(
-                        table_name=table_name,
-                        schema_object_name=schema_object_name,
-                        exclude_columns=final_exclude_columns
-                    )
-                    sql_context += table_definition
+                    # 8. Build the table definition dictionary manually using the structure data
+                    json_dict = {
+                        "table": table_name,
+                        "schema": database_schema_name,
+                        "description": f"The table belongs to the **`{database_schema_name}`** schema.",
+                        "fields": []
+                    }
+
+                    if schema_object_name:
+                        json_dict["description"] += (
+                            f"The meaning of each field in this table is detailed in the **`{schema_object_name}`** object."
+                        )
+
+                    for col in table_data.get('columns', []):
+                        name = col["name"]
+                        if name in final_exclude_columns:
+                            continue
+
+                        json_dict["fields"].append({
+                            "name": name,
+                            "type": col["type"]
+                        })
+
+                    # Append as string representation of dict (consistent with previous behavior)
+                    sql_context += "\n\n" + str(json_dict)
+
                 except (KeyError, RuntimeError) as e:
                     logging.warning(f"Could not generate schema for table '{table_name}': {e}")
 

iatoolkit/services/configuration_service.py

@@ -97,6 +97,141 @@ class ConfigurationService:
         logging.info(f"✅ Company '{company_short_name}' configured successfully.")
         return config, errors
 
+    def update_configuration_key(self, company_short_name: str, key: str, value) -> tuple[dict, list[str]]:
+        """
+        Updates a specific key in the company's configuration file, validates the result,
+        and saves it to the asset repository if valid.
+
+        Args:
+            company_short_name: The company identifier.
+            key: The configuration key to update (supports dot notation, e.g., 'llm.model').
+            value: The new value for the key.
+
+        Returns:
+            A tuple containing the updated configuration dict and a list of error strings (if any).
+        """
+        # 1. Load raw config from file (to avoid working with merged supplementary files if possible,
+        # but for simplicity we load the main yaml structure)
+        main_config_filename = "company.yaml"
+
+        if not self.asset_repo.exists(company_short_name, AssetType.CONFIG, main_config_filename):
+            raise FileNotFoundError(f"Configuration file not found for {company_short_name}")
+
+        yaml_content = self.asset_repo.read_text(company_short_name, AssetType.CONFIG, main_config_filename)
+        config = self.utility.load_yaml_from_string(yaml_content) or {}
+
+        # 2. Update the key in the dictionary
+        self._set_nested_value(config, key, value)
+
+        # 3. Validate the new configuration structure
+        errors = self._validate_configuration(company_short_name, config)
+
+        if errors:
+            logging.warning(f"Configuration update failed validation: {errors}")
+            return config, errors
+
+        # 4. Save back to repository
+        # Assuming Utility has a method to dump YAML. If not, standard yaml library would be needed.
+        # For this example, we assume self.utility.dump_yaml_to_string exists.
+        new_yaml_content = self.utility.dump_yaml_to_string(config)
+        self.asset_repo.write_text(company_short_name, AssetType.CONFIG, main_config_filename, new_yaml_content)
+
+        # 5. Invalidate cache so next reads get the new version
+        if company_short_name in self._loaded_configs:
+            del self._loaded_configs[company_short_name]
+
+        return config, []
+
+    def add_configuration_key(self, company_short_name: str, parent_key: str, key: str, value) -> tuple[dict, list[str]]:
+        """
+        Adds a new key-value pair under a specific parent key in the configuration.
+
+        Args:
+            company_short_name: The company identifier.
+            parent_key: The parent configuration key under which to add the new key (e.g., 'llm').
+            key: The new key name to add.
+            value: The value for the new key.
+
+        Returns:
+            A tuple containing the updated configuration dict and a list of error strings (if any).
+        """
+        # 1. Load raw config from file
+        main_config_filename = "company.yaml"
+
+        if not self.asset_repo.exists(company_short_name, AssetType.CONFIG, main_config_filename):
+            raise FileNotFoundError(f"Configuration file not found for {company_short_name}")
+
+        yaml_content = self.asset_repo.read_text(company_short_name, AssetType.CONFIG, main_config_filename)
+        config = self.utility.load_yaml_from_string(yaml_content) or {}
+
+        # 2. Construct full path and set the value
+        # If parent_key is provided, we append the new key to it (e.g., 'llm.new_setting')
+        full_path = f"{parent_key}.{key}" if parent_key else key
+        self._set_nested_value(config, full_path, value)
+
+        # 3. Validate the new configuration structure
+        errors = self._validate_configuration(company_short_name, config)
+
+        if errors:
+            logging.warning(f"Configuration add failed validation: {errors}")
+            return config, errors
+
+        # 4. Save back to repository
+        new_yaml_content = self.utility.dump_yaml_to_string(config)
+        self.asset_repo.write_text(company_short_name, AssetType.CONFIG, main_config_filename, new_yaml_content)
+
+        # 5. Invalidate cache
+        if company_short_name in self._loaded_configs:
+            del self._loaded_configs[company_short_name]
+
+        return config, []
+
+
+    def validate_configuration(self, company_short_name: str) -> list[str]:
+        """
+        Public method to trigger validation of the current configuration.
+        """
+        config = self._load_and_merge_configs(company_short_name)
+        return self._validate_configuration(company_short_name, config)
+
+    def _set_nested_value(self, data: dict, key: str, value):
+        """
+        Helper to set a value in a nested dictionary or list using dot notation (e.g. 'llm.model', 'tools.0.name').
+        Handles traversal through both dictionaries and lists.
+        """
+        keys = key.split('.')
+        current = data
+
+        # Traverse up to the parent of the target key
+        for i, k in enumerate(keys[:-1]):
+            if isinstance(current, dict):
+                # If it's a dict, we can traverse or create the path
+                current = current.setdefault(k, {})
+            elif isinstance(current, list):
+                # If it's a list, we MUST use an integer index
+                try:
+                    idx = int(k)
+                    current = current[idx]
+                except (ValueError, IndexError) as e:
+                    raise ValueError(
+                        f"Invalid path: cannot access index '{k}' in list at '{'.'.join(keys[:i + 1])}'") from e
+            else:
+                raise ValueError(
+                    f"Invalid path: '{k}' is not a container (got {type(current)}) at '{'.'.join(keys[:i + 1])}'")
+
+        # Set the final value
+        last_key = keys[-1]
+        if isinstance(current, dict):
+            current[last_key] = value
+        elif isinstance(current, list):
+            try:
+                idx = int(last_key)
+                current[idx] = value
+            except (ValueError, IndexError) as e:
+                raise ValueError(f"Invalid path: cannot assign to index '{last_key}' in list") from e
+        else:
+            raise ValueError(f"Cannot assign value to non-container type {type(current)} at '{key}'")
+
     def _load_and_merge_configs(self, company_short_name: str) -> dict:
         """
         Loads the main company.yaml and merges data from supplementary files
@@ -239,13 +374,12 @@ class ConfigurationService:
         from iatoolkit.services.prompt_service import PromptService
         prompt_service = current_iatoolkit().get_injector().get(PromptService)
 
-        prompts_config = config.get('prompts', [])
-        categories_config = config.get('prompt_categories', [])
+        prompts_config = config.get('prompts', {})
 
         prompt_service.sync_company_prompts(
             company_short_name=company_short_name,
-            prompts_config=prompts_config,
-            categories_config=categories_config
+            prompts_config=prompts_config.get('prompt_list', []),
+            categories_config=prompts_config.get('prompt_categories', []),
         )
 
     def _register_knowledge_base(self, company_short_name: str, config: dict):
@@ -328,8 +462,11 @@ class ConfigurationService:
                 add_error(f"tools[{i}]", "'params' key must be a dictionary.")
 
         # 6. Prompts
-        category_set = set(config.get("prompt_categories", []))
-        for i, prompt in enumerate(config.get("prompts", [])):
+        prompt_list = config.get("prompts", {}).get("prompt_list", [])
+        prompt_categories = config.get("prompts", {}).get("prompt_categories", [])
+
+        category_set = set(prompt_categories)
+        for i, prompt in enumerate(prompt_list):
             prompt_name = prompt.get("name")
             if not prompt_name:
                 add_error(f"prompts[{i}]", "Missing required key: 'name'")

iatoolkit/services/knowledge_base_service.py

@@ -308,7 +308,7 @@ class KnowledgeBaseService:
 
         # filter by collection
         if collection:
-            query = query.join(CollectionType).filter(CollectionType.name == collection)
+            query = query.join(Document.collection_type).filter(CollectionType.name == collection)
 
         # Filter by user identifier
         if user_identifier:
@@ -403,6 +403,19 @@ class KnowledgeBaseService:
 
         session.commit()
 
+    def get_collection_names(self, company_short_name: str) -> List[str]:
+        """
+        Retrieves the names of all collections defined for a specific company.
+        """
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            logging.warning(f"Company {company_short_name} not found when listing collections.")
+            return []
+
+        session = self.document_repo.session
+        collections = session.query(CollectionType).filter_by(company_id=company.id).all()
+        return [c.name for c in collections]
+
     def _get_collection_type_id(self, company_id: int, collection_name: str) -> Optional[int]:
         """Helper to get ID by name"""
         if not collection_name:

iatoolkit/services/load_documents_service.py

@@ -65,17 +65,24 @@ class LoadDocumentsService:
                 logging.warning(f"Source '{source_name}' not found in configuration for company '{company.short_name}'. Skipping.")
                 continue
 
+            collection = source_config.get('collection')
+            if not collection:
+                logging.warning(
+                    f"Document Source '{source_name}' missing collection definition en company.yaml, Skipping.")
+                continue
+
             try:
-                logging.info(f"Processing source '{source_name}' for company '{company.short_name}'...")
+                logging.info(f"company {company.short_name}: loading source '{source_name}' into collection '{collection}'...")
 
                 # Combine the base connector configuration with the specific path from the source.
                 full_connector_config = base_connector_config.copy()
                 full_connector_config['path'] = source_config.get('path')
+                full_connector_config['folder'] = source_config.get('folder')
 
                 # Prepare the context for the callback function.
                 context = {
                     'company': company,
-                    'collection': source_config.get('metadata', {}).get('collection'),
+                    'collection': collection,
                     'metadata': source_config.get('metadata', {})
                 }
 
@@ -132,7 +139,7 @@ class LoadDocumentsService:
                 company=company,
                 filename=filename,
                 content=content,
-                collection=predefined_metadata.get('collection'),
+                collection=context.get('collection'),
                 metadata=predefined_metadata
             )
 

iatoolkit/services/sql_service.py

@@ -171,4 +171,21 @@ class SqlService:
             logging.error(f"Error while committing sql: '{str(e)}'")
             raise IAToolkitException(
                 IAToolkitException.ErrorType.DATABASE_ERROR, str(e)
+            )
+
+    def get_database_structure(self, company_short_name: str, db_name: str) -> dict:
+        """
+        Introspects the specified database and returns its structure (Tables & Columns).
+        Used for the Schema Editor 2.0
+        """
+        try:
+            provider = self.get_database_provider(company_short_name, db_name)
+            return provider.get_database_structure()
+        except IAToolkitException:
+            raise
+        except Exception as e:
+            logging.error(f"Error introspecting database '{db_name}': {e}")
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.DATABASE_ERROR,
+                f"Failed to introspect database: {str(e)}"
             )

iatoolkit/templates/chat.html

@@ -8,6 +8,7 @@
     {# Movemos los estilos y los links aquí para que se rendericen en el <head> #}
     <style>
         {{ branding.css_variables | safe }}
+        {{ branding.css_variables | safe }}
     </style>
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css">
     <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.11.3/font/bootstrap-icons.min.css">
@@ -104,7 +105,7 @@
                        style="color: {{ branding.header_text_color }};">
                         <i class="bi bi-question-circle-fill"></i>
                 </a>
-                {% if user_role == "admin" and license == 'enterprise' %}
+                {% if user_role != "user" and license == 'enterprise' %}
                     <a href="/{{ company_short_name }}/admin/dashboard"
                        target="_blank"
                            id="preferences-button"

iatoolkit/views/configuration_api_view.py

@@ -0,0 +1,163 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import jsonify, request
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.configuration_service import ConfigurationService
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.auth_service import AuthService
+import logging
+
+
+class ConfigurationApiView(MethodView):
+    """
+    API View to manage company configuration.
+    Supports loading, updating specific keys, and validating the configuration.
+    """
+    @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):
+        """
+        Loads the current configuration for the company.
+        """
+        try:
+            # 1. Verify authentication
+            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)
+
+            # Register data sources to ensure services are up to date with loaded config
+            if config:
+                self.configuration_service.register_data_sources(company_short_name)
+
+            # Remove non-serializable objects
+            if 'company' in config:
+                config.pop('company')
+
+            status_code = 200 if not errors else 400
+            return jsonify({'config': config, 'errors': errors}), status_code
+        except Exception as e:
+            logging.exception(f"Unexpected error loading config: {e}")
+            return jsonify({'status': 'error', 'message': str(e)}), 500
+
+    def patch(self, company_short_name: str):
+        """
+        Updates a specific configuration key.
+        Body: { "key": "llm.model", "value": "gpt-4" }
+        """
+        try:
+            auth_result = self.auth_service.verify(anonymous=False) # Require valid user for updates
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            payload = request.get_json()
+            key = payload.get('key')
+            value = payload.get('value')
+
+            if not key:
+                return jsonify({'error': 'Missing "key" in payload'}), 400
+
+            logging.info(f"Updating config key '{key}' for company '{company_short_name}'")
+
+            updated_config, errors = self.configuration_service.update_configuration_key(
+                company_short_name, key, value
+            )
+
+            # Remove non-serializable objects
+            if 'company' in updated_config:
+                updated_config.pop('company')
+
+            if errors:
+                return jsonify({'status': 'invalid', 'errors': errors, 'config': updated_config}), 400
+
+            return jsonify({'status': 'success', 'config': updated_config}), 200
+
+        except FileNotFoundError:
+            return jsonify({'error': 'Configuration file not found'}), 404
+        except Exception as e:
+            logging.exception(f"Error updating config: {e}")
+            return jsonify({'status': 'error', 'message': str(e)}), 500
+
+    def post(self, company_short_name: str):
+        """
+        Adds a new configuration key.
+        Body: { "parent_key": "llm", "key": "max_tokens", "value": 2048 }
+        """
+        try:
+            auth_result = self.auth_service.verify(anonymous=False)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            payload = request.get_json()
+            parent_key = payload.get('parent_key', '')  # Optional, defaults to root
+            key = payload.get('key')
+            value = payload.get('value')
+
+            if not key:
+                return jsonify({'error': 'Missing "key" in payload'}), 400
+
+            logging.info(f"Adding config key '{key}' under '{parent_key}' for company '{company_short_name}'")
+
+            updated_config, errors = self.configuration_service.add_configuration_key(
+                company_short_name, parent_key, key, value
+            )
+
+            # Remove non-serializable objects
+            if 'company' in updated_config:
+                updated_config.pop('company')
+
+            if errors:
+                return jsonify({'status': 'invalid', 'errors': errors, 'config': updated_config}), 400
+
+            return jsonify({'status': 'success', 'config': updated_config}), 200
+
+        except FileNotFoundError:
+            return jsonify({'error': 'Configuration file not found'}), 404
+        except Exception as e:
+            logging.exception(f"Error adding config key: {e}")
+            return jsonify({'status': 'error', 'message': str(e)}), 500
+
+class ValidateConfigurationApiView(MethodView):
+    """
+    API View to trigger an explicit validation of the current configuration.
+    Useful for UI to check status without modifying data.
+    """
+    @inject
+    def __init__(self,
+                 configuration_service: ConfigurationService,
+                 auth_service: AuthService):
+        self.configuration_service = configuration_service
+        self.auth_service = auth_service
+
+    def get(self, company_short_name: str):
+        try:
+            auth_result = self.auth_service.verify(anonymous=False)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            errors = self.configuration_service.validate_configuration(company_short_name)
+
+            if errors:
+                return jsonify({'status': 'invalid', 'errors': errors}), 200  # 200 OK because check succeeded
+
+            return jsonify({'status': 'valid', 'errors': []}), 200
+
+        except Exception as e:
+            logging.exception(f"Error validating config: {e}")
+            return jsonify({'status': 'error', 'message': str(e)}), 500

{iatoolkit-1.4.2.dist-info → iatoolkit-1.7.0.dist-info}/METADATA

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

{iatoolkit-1.4.2.dist-info → iatoolkit-1.7.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-iatoolkit/__init__.py,sha256=4kPCwhdZff5rt-FhnHfafMCPvD5hDFGEdRwINPm8ATc,1425
+iatoolkit/__init__.py,sha256=lBC4IhNqHSyQzU5dQEHh49nkOgc97JfQq2E27-VVPa0,1425
 iatoolkit/base_company.py,sha256=TzGvdseDpJMi9fl5nbR2pMARgU6pN6ISatFoMJKtNxw,648
 iatoolkit/cli_commands.py,sha256=EDky1QpvHnatzXsd1evYW0OQt-k3JIbPGMc13vgDvy4,1814
 iatoolkit/company_registry.py,sha256=XopFzqIpUxBtQJzdZwu2-c0v98MLT7SP_xLPxn_iENM,4176
@@ -6,12 +6,12 @@ iatoolkit/core.py,sha256=QKqu-QDrNWunDU9uX96a86kFNv3a_aAxonkpxJptzoE,20885
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/exceptions.py,sha256=xmNi5Et13pCvCM1so79z36I-7Kdie8nVUAqW9acGs9k,1202
 iatoolkit/common/model_registry.py,sha256=HiFr1FtRKGjpp3YurkiF4l6sGuirigDVL-b_3_R-vlM,5561
-iatoolkit/common/routes.py,sha256=cJCFLgSqrTXPOC9CraXLQtoOhVCphWXHyFadYiTXQGw,7292
+iatoolkit/common/routes.py,sha256=iVZawjKfGt6iK9tZ8t3y6dofP4KJpIQex48OVVvlRYo,7540
 iatoolkit/common/session_manager.py,sha256=OUYMzT8hN1U-NCUidR5tUAXB1drd8nYTOpo60rUNYeY,532
-iatoolkit/common/util.py,sha256=e000sUjUXeY4BJrJ3HtnsVMMhn8guclRZkl0f2Dmfy0,15270
+iatoolkit/common/util.py,sha256=NfSuDbzrGVS59fIgqJETYUlD5otMLwAjyp7OVhaMoo8,16310
 iatoolkit/common/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/interfaces/asset_storage.py,sha256=IYXyc73gmwFT7T8RQhC4MymQDW0vGDSEzuKk43w7Ct4,989
-iatoolkit/common/interfaces/database_provider.py,sha256=Cf-33SZ7HPxTv2b_ahew1TcqDmFH442OPwIv6YEiubk,1097
+iatoolkit/common/interfaces/database_provider.py,sha256=7ayYuEYRunkm4udoOGumoX9PfdgohBcrj0JDPfyfM00,1156
 iatoolkit/infra/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/infra/brevo_mail_app.py,sha256=xvy3KxEEgjFpHlIAogP6SU5KXmg3w7lC16nnNmOYU8Y,5323
 iatoolkit/infra/call_service.py,sha256=iRk9VxbXaAwlLIl8fUzGDWIAdzwfsbs1MtP84YeENxg,4929
@@ -21,19 +21,19 @@ iatoolkit/infra/llm_response.py,sha256=6hSmI909m0Pdgs7UnJHrv_nokbCZLzHpdwBUED6VN
 iatoolkit/infra/redis_session_manager.py,sha256=EPr3E_g7LHxn6U4SV5lT_L8WQsAwg8VzA_WIEZ3TwOw,3667
 iatoolkit/infra/connectors/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/infra/connectors/file_connector.py,sha256=HOjRTFd-WfDOcFyvHncAhnGNZuFgChIwC-P6osPo9ZM,352
-iatoolkit/infra/connectors/file_connector_factory.py,sha256=3qvyfH4ZHKuiMxJFkawOxhW2-TGKKtsBYHgoPpZMuKU,2118
+iatoolkit/infra/connectors/file_connector_factory.py,sha256=iCOhklUaYG3YxtJGBYOrLN8spF5mjKu6vyA6nj71FHI,2167
 iatoolkit/infra/connectors/google_cloud_storage_connector.py,sha256=IXpL3HTo7Ft4EQsYiQq5wXRRQK854jzOEB7ZdWjLa4U,2050
 iatoolkit/infra/connectors/google_drive_connector.py,sha256=WR1AlO5-Bl3W89opdja0kKgHTJzVOjTsy3H4SlIvwVg,2537
 iatoolkit/infra/connectors/local_file_connector.py,sha256=hrzIgpMJOTuwTqzlQeTIU_50ZbZ6yl8lcWPv6hMnoqI,1739
-iatoolkit/infra/connectors/s3_connector.py,sha256=Nj4_YaLobjfcnbZewJf21_K2EXohgcc3mJll1Pzn4zg,1123
+iatoolkit/infra/connectors/s3_connector.py,sha256=r8bgye8ydMzayAAg5RS22oakpKd-vNw-cSrOCvlwG9g,1209
 iatoolkit/infra/llm_providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/infra/llm_providers/deepseek_adapter.py,sha256=ui4V05xplUvLWHu7y_dUMKpP4Af-BkZi5tvdgZY16dU,10719
 iatoolkit/infra/llm_providers/gemini_adapter.py,sha256=xjYAMgS4bh7qfpCEv6n92tzpKdjnNBVFfr8GkSrcatM,14555
 iatoolkit/infra/llm_providers/openai_adapter.py,sha256=9SuHYtPJcadWSV_GHfx8DDj9BqkUS3RvLSyDd_ZVB2M,4848
-iatoolkit/locales/en.yaml,sha256=cKPJQB2sV_AuFd80eOK4r-Y08cKlmG9eG7b34iuBqNo,12210
-iatoolkit/locales/es.yaml,sha256=1LT81MPQAW0mhHOVxWtgiOTxYs6B9Jyc6wgQGz04xGo,13027
+iatoolkit/locales/en.yaml,sha256=NX1LMzpBNtgRl0q5RqD9wwTbF9IKzdVheTOaGzI1P7E,13571
+iatoolkit/locales/es.yaml,sha256=ote9JSzWq8lGwbfX7wKKIp7ccdZ76-sbXIDnTp_szQQ,14443
 iatoolkit/repositories/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
-iatoolkit/repositories/database_manager.py,sha256=JAi-A_siEZlyQeV7xk1NIYFZjfBCCo12RVSV_pLYYhI,6631
+iatoolkit/repositories/database_manager.py,sha256=F6oYrmNp29SrYHJeoe7UIzQuXdZ6fVyRMgdWhGo6oCs,5905
 iatoolkit/repositories/document_repo.py,sha256=5oCrq8pjcyOCBrOc4XYeUvza9nUd0VJ4eNL0g1MQb68,1441
 iatoolkit/repositories/filesystem_asset_repository.py,sha256=ZFr_FpHQD5nszZnJSki6nc-W3PKvmhTxKmVZMd2DUbM,1825
 iatoolkit/repositories/llm_query_repo.py,sha256=8rGSVx7WZlGbUfeVw_uGygTUuJXRK5vGfFHj1znc1ss,3981
@@ -44,8 +44,8 @@ iatoolkit/services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X
 iatoolkit/services/auth_service.py,sha256=B0gTUHDf56WiTnA5viBEvbmf2A6y_6bkAoFlMvu0IWY,7756
 iatoolkit/services/benchmark_service.py,sha256=CdbFYyS3FHFhNzWQEa9ZNjUlmON10DT1nKNbZQ1EUi8,5880
 iatoolkit/services/branding_service.py,sha256=Fsjcos7hGLx2qhoZ9hTaFCgb7x5eYHMDGvXg4mXx8D0,8120
-iatoolkit/services/company_context_service.py,sha256=MV07jvef_9u_55n92yHCDwThBykn5rQT1tKg51Ebxoo,9339
-iatoolkit/services/configuration_service.py,sha256=bYjhR5cZcJunFr1OtX2ReOJaHbYrxBlxizEdoo52zaw,17933
+iatoolkit/services/company_context_service.py,sha256=Oqldfdo8MSGv_KB8-4EztUVcqycoabrVLSbfky2n5s4,10435
+iatoolkit/services/configuration_service.py,sha256=TsrTzou2Z8rC50favIlPgAloHBcy2p1jlZjRNtg3cq8,24184
 iatoolkit/services/dispatcher_service.py,sha256=f3aPbKU9UkNS61dNVZjT-J9hg3x-H2Nb_n-cZRnZ9aI,5247
 iatoolkit/services/document_service.py,sha256=XrAwbNPyhmP94bGdDwHy76Mg_PCg6O2QyZwii1T6kZ8,6469
 iatoolkit/services/embedding_service.py,sha256=ngRnFXEpko-rqpicqbJbVdr91IjSe1jXSYVWprOVlE0,6296
@@ -54,16 +54,16 @@ iatoolkit/services/file_processor_service.py,sha256=5xS7zrdL_kpCFXQ5seqNl9tiseI7
 iatoolkit/services/history_manager_service.py,sha256=nq1hk39WDbyHG6s8irpIdCS98EBDagF0djH6JHq9WuY,8734
 iatoolkit/services/i18n_service.py,sha256=mR4pS0z56NgZLeSnEvDXiMvVBeOCl5CkUWdYBTVEhyM,3941
 iatoolkit/services/jwt_service.py,sha256=pVoH1rzRwWixpvT3AhFdiE1BDmdbo4KQMyOF0P7tz-Y,2939
-iatoolkit/services/knowledge_base_service.py,sha256=kjVI04WjOf5q_fYEDNkg-lRN1hqbQrS0URzUGy3dkuo,16500
+iatoolkit/services/knowledge_base_service.py,sha256=UuFWdjobFqwsBuLwAqi4FRNaVuYabSIjSj4vPKxZzLk,17098
 iatoolkit/services/language_service.py,sha256=ktNzj0hrCPm_VMIpWTxFEpEHKQ1lOHOfuhzCbZud2Dc,3428
 iatoolkit/services/license_service.py,sha256=pdl9szSEPbIu2ISl6g-WIR1p4HQu2gSGdAjViKxY6Hg,2751
 iatoolkit/services/llm_client_service.py,sha256=YSvoNKeBQRGL3znMDIyLH2PGZt4ImmYKD_aBlBeS91k,19024
-iatoolkit/services/load_documents_service.py,sha256=zK1mXsj9aLDDKoV9WKgNuFlg02ROnlf9umcTm_Y4OIs,6761
+iatoolkit/services/load_documents_service.py,sha256=WEaPMRxginbsxOXopofPj6FeH-iIrSTlipFFRV20t4I,7069
 iatoolkit/services/mail_service.py,sha256=6Kx1CIbXzAr_ucoqwqTlhhwE6y2Jw64eNDjXysaZP-c,8297
 iatoolkit/services/profile_service.py,sha256=mUXW5dwFF4SBH04adQuoSj-E0FmuLDt_VcPo9oxdr1M,23680
 iatoolkit/services/prompt_service.py,sha256=mr3szmOThNi6MtqCbh4rwJJOlk-Be35bkIJXqh7_hvM,12898
 iatoolkit/services/query_service.py,sha256=aS2-NtEMdu-QViw58d-g9iZI1j3KZIJb0xrN8N_c__U,20406
-iatoolkit/services/sql_service.py,sha256=kgLEE7iiw9WjnyvOBfm4VJ60-F6JAg26cAKF6LlF-zU,7087
+iatoolkit/services/sql_service.py,sha256=nafxpDdv1AbOYMeHzoiD6G8V64TPMniUlLhrJ0J_KOU,7802
 iatoolkit/services/tool_service.py,sha256=FQuplPsgs1AdyOk2pxUBkPIvAvUqJU1ElOpOWz5wkr8,10002
 iatoolkit/services/user_feedback_service.py,sha256=KbjMjl0CDF7WpPHioJ3sqQVwocjfq_cisrpeqYUqtas,5361
 iatoolkit/services/user_session_context_service.py,sha256=RSv_NxOSTVPA6nFLiZKSIuUWa59cmZWBwiN_TWSljBo,9074
@@ -95,7 +95,7 @@ iatoolkit/templates/_company_header.html,sha256=L1QtrMpqYXALqtinCa3WhG4CrgHKLObl
 iatoolkit/templates/_login_widget.html,sha256=qziV70-n7SwC11rL4cU1LenJM83jYvuyw2xSz6OORJA,2063
 iatoolkit/templates/base.html,sha256=sRHUnHaKIycLRKZ_PQOTb88kSBC6kigzxlm60vpXj1I,3434
 iatoolkit/templates/change_password.html,sha256=p0GWZ9gldluYQM8OPSz2Q4CYhU8UJmb-72iz_Sl_6Ho,3485
-iatoolkit/templates/chat.html,sha256=HtmTBbbBPvVHscEymSWqAFjCnxE4hlMckzRgdi-1MXs,15642
+iatoolkit/templates/chat.html,sha256=8tLQlpCS7Nd1-07kefIZzv4S5Hcd1UIhQmC5P_Z760s,15685
 iatoolkit/templates/chat_modals.html,sha256=x0Do7I7X9cfgZFHSAwWreIN55nYPCpZK7PTBxGXjrYY,8501
 iatoolkit/templates/error.html,sha256=4aT6cIYQUILciekAceHEiUuZxcAqkmNPUMnB9CD4BNA,1778
 iatoolkit/templates/forgot_password.html,sha256=XLVMmLJK5SAXmPqIH_OyTlSbu5QTAdekO6ggzfcNup0,2376
@@ -105,6 +105,7 @@ iatoolkit/views/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,1
 iatoolkit/views/base_login_view.py,sha256=l45msQUrLGxxxZrSb5l7tnTLLQyIoPTu4TgbkHBvVlc,3981
 iatoolkit/views/change_password_view.py,sha256=TXFvvtNrR37yrNAHYzcMSSIkV4-MIzmIpwbPffaJJyE,4957
 iatoolkit/views/chat_view.py,sha256=QNlC0n1hxjWbI3wyLhV6D-U2kJZkIqClNg94iQP4PrA,3534
+iatoolkit/views/configuration_api_view.py,sha256=K8aclK6oKkoTZoYwck-mozLchEOFCkrXFtraHh7tI00,6370
 iatoolkit/views/embedding_api_view.py,sha256=9n5utihq8zHiG0V2sVCOYQ72wOKa4denqE951gqaJWo,2371
 iatoolkit/views/forgot_password_view.py,sha256=C4eJe9GmGrl6M3bTt_anwuPQib9ouXZtKYp6kB2xBeU,3380
 iatoolkit/views/help_content_api_view.py,sha256=8EUw2iSE6X-Lm2dGlYZxSp-29HzaAGwD3CpL4Otqp5k,2011
@@ -112,7 +113,6 @@ iatoolkit/views/history_api_view.py,sha256=oNYfo-fZbz4sY6VpE_goaCFQyH2k9IX85VhPo
 iatoolkit/views/home_view.py,sha256=GaU2-qyTnE8jE1ejdNjHW3azo72X_Awhp1y1Zx9BXRg,2662
 iatoolkit/views/init_context_api_view.py,sha256=axfNno8E1v4dCNuXNkdV1XpDZtypICj48hW3cRyl_eA,2961
 iatoolkit/views/llmquery_api_view.py,sha256=ohlYmFcvhvZL2BTMeCzd7ZPeTBzIPcIZbC9CRWqQM70,2290
-iatoolkit/views/load_company_configuration_api_view.py,sha256=ufX_jrSO7NTUsgNJNrAXxFKmyp2JyNYoplMUm4biULs,1902
 iatoolkit/views/load_document_api_view.py,sha256=RaO4KRz19uSOv3LL7Fn8FJSLzHtRkPAIYRPU-yT1nxQ,2485
 iatoolkit/views/login_view.py,sha256=NKPjR9cCOsw7FhOfXCM32m6veKxRdLhsc0H7DLc5_ug,7517
 iatoolkit/views/logout_api_view.py,sha256=7c0rL5sLTuoRPqQs73wlH_7eb3--S6GLcr7Yu4hNOYU,2035
@@ -125,9 +125,9 @@ iatoolkit/views/static_page_view.py,sha256=V0oTybQ5lFo3hp5xOKau-yMO8vK8NCM6zWiwV
 iatoolkit/views/user_feedback_api_view.py,sha256=QOTBtpNqYAmewXDgVAoaIprnVjvf1xM0ExWxSFp3ICI,2098
 iatoolkit/views/users_api_view.py,sha256=hVR6rxLNyhGiXhCIHvttUeSinH071Pyk-2_Fcm-v2TM,1139
 iatoolkit/views/verify_user_view.py,sha256=Uc9P2wiRjZ7pMRTjWUn8Y-p0DFqw_AwNYQsdbAwhbg8,2659
-iatoolkit-1.4.2.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
-iatoolkit-1.4.2.dist-info/licenses/LICENSE_COMMUNITY.md,sha256=9lVNcggPNUnleMF3_h3zu9PbNTeaRqB1tHAbiBLQJZU,566
-iatoolkit-1.4.2.dist-info/METADATA,sha256=j6HullcLZAkwHSYnpcXew6VVsL6nzFkyoXwpuSBDy1E,8395
-iatoolkit-1.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-1.4.2.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-1.4.2.dist-info/RECORD,,
+iatoolkit-1.7.0.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
+iatoolkit-1.7.0.dist-info/licenses/LICENSE_COMMUNITY.md,sha256=9lVNcggPNUnleMF3_h3zu9PbNTeaRqB1tHAbiBLQJZU,566
+iatoolkit-1.7.0.dist-info/METADATA,sha256=V0ZDSh08B6jbggfwSY10wfIt7oDiFVWHuPTg-Dbkg30,8395
+iatoolkit-1.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-1.7.0.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-1.7.0.dist-info/RECORD,,

iatoolkit/views/load_company_configuration_api_view.py

@@ -1,49 +0,0 @@
-# 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
-
-