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

iatoolkit/__init__.py

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

iatoolkit/common/routes.py

@@ -32,6 +32,7 @@ def register_views(app):
     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'))
@@ -85,9 +86,21 @@ def register_views(app):
     # 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'])
+
+    # 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'))

iatoolkit/locales/en.yaml

@@ -52,8 +52,7 @@ ui:
     company_config: "Company Configuration (company.yaml)"
     prompts: "Prompts"
     prompts_description: "System Prompts"
-    knowledge: "Knowledge Base"
-    knowledge_rag: "RAG (Vector)"
+    knowledge_rag: "Knowledge Base (RAG)"
     knowledge_static: "Static Context"
     schemas: "Schemas"
     schemas_description: "Data Definitions (YAML)"
@@ -86,6 +85,10 @@ ui:
     logout: "Close session"
     error_loading: "Error loading file content"
     loading: "Loading..."
+    add: "Add"
+    create: "Create"
+    delete: "Delete"
+
 
   db_explorer:
     data_explorer: "Data Explorer"
@@ -108,6 +111,36 @@ ui:
     meta_synonyms: "Synonyms"
     pii_sesitive: "PII Sensitive"
 
+  prompts:
+    title: "Prompts Manager"
+    subtitle: "Manage and test your AI personas and templates"
+    library: "Prompt Library"
+    new_btn: "New"
+    filter_placeholder: "Filter prompts..."
+    select_prompt: "Select a prompt"
+    toggle_settings: "Toggle Variables & Config"
+    settings_btn: "Settings"
+    save_btn: "Save"
+    tab_editor: "Editor"
+    tab_playground: "Playground"
+    config_title: "Config"
+    desc_label: "Description"
+    desc_placeholder: "Describe this prompt..."
+    vars_label: "Input Variables"
+    no_vars: "No inputs defined"
+    playground_inputs: "Variables & Inputs"
+    no_vars_detected: "No variables detected in prompt template."
+    model_override: "Model Override (Optional)"
+    default_model: "Default (from config)"
+    run_btn: "Run"
+    output_placeholder: "Output will appear here..."
+    new_modal_title: "Create New Prompt"
+    name_label: "Name (Slug)"
+    name_help: "Use lowercase, numbers, and underscores only."
+    category_label: "Category"
+    delete_confirmation: "Delete Prompt?"
+
+
   config:
     editor_description: "IAToolkit configuration file"
     title: "Configuration Editor"
@@ -146,7 +179,7 @@ ui:
     delete_message: "This action cannot be undone. The file will be permanently removed."
     delete_button: "Delete"
     delete_cancel: "Cancel"
-    target_collection: "Target collection"
+    target_collection: "Collection"
     select_collection_placeholder: "Select a collection"
     collection_required: "Collection is required"
     collection: "Collection"

iatoolkit/locales/es.yaml

@@ -49,8 +49,7 @@ ui:
     workspace: "Recursos"
     configuration: "Configuración"
     company_config: "Configuración Empresa (company.yaml)"
-    knowledge: "Conocimiento"
-    knowledge_rag: "RAG (Vectorial)"
+    knowledge_rag: "Knowledge Base (RAG)"
     knowledge_static: "Contenido Estático"
     prompts: "Prompts"
     prompts_description: "Prompts de sistema"
@@ -83,6 +82,9 @@ ui:
     load_configuration: "Guardar configuración"
     goto_chat: "Ir al chat"
     logout: "Cerrar sesión"
+    add: "Agregar"
+    create: "Crear"
+    delete: "Borrar"
 
   db_explorer:
     data_explorer: "Explorador de datos"
@@ -105,6 +107,35 @@ ui:
     meta_synonyms: "Sinonimos"
     pii_sesitive: "IP Sensible"
 
+  prompts:
+    title: "Gestor de Prompts"
+    subtitle: "Gestiona y prueba tus plantillas y personas de IA"
+    library: "Biblioteca de Prompts"
+    new_btn: "Nuevo"
+    filter_placeholder: "Filtrar prompts..."
+    select_prompt: "Selecciona un prompt"
+    toggle_settings: "Alternar Variables y Configuración"
+    settings_btn: "Ajustes"
+    save_btn: "Guardar"
+    tab_editor: "Editor"
+    tab_playground: "Playground"
+    config_title: "Configuración"
+    desc_label: "Descripción"
+    desc_placeholder: "Describe este prompt..."
+    vars_label: "Variables de Entrada"
+    no_vars: "Sin variables definidas"
+    playground_inputs: "Variables y Entradas"
+    no_vars_detected: "No se detectaron variables en la plantilla."
+    model_override: "Modelo Específico (Opcional)"
+    default_model: "Por defecto (según config)"
+    run_btn: "Ejecutar"
+    output_placeholder: "El resultado aparecerá aquí..."
+    new_modal_title: "Crear Nuevo Prompt"
+    name_label: "Nombre (Slug)"
+    name_help: "Solo minúsculas, números y guiones bajos."
+    category_label: "Categoría"
+    delete_confirmation: "Eliminar el prompt?"
+
   config:
     editor_description: "Editor de configuración"
     title: "Editor de configuraciones"
@@ -143,7 +174,7 @@ 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"
+    target_collection: "Categoría"
     select_collection_placeholder: "Selecciona una categoría"
     collection_required: "Debe seleccionar una categoría"
     all_collections: "Todas las categorías"

iatoolkit/repositories/llm_query_repo.py

@@ -3,10 +3,13 @@
 #
 # IAToolkit is open source software.
 
-from iatoolkit.repositories.models import LLMQuery, Tool, Company, Prompt, PromptCategory
+from iatoolkit.repositories.models import (LLMQuery, Tool,
+                    Company, Prompt, PromptCategory, PromptType)
 from injector import inject
 from iatoolkit.repositories.database_manager import DatabaseManager
 from sqlalchemy import or_
+from typing import List
+
 
 class LLMQueryRepo:
     @inject
@@ -54,7 +57,7 @@ class LLMQueryRepo:
             self.session.add(new_tool)
             tool = new_tool
 
-        self.session.flush()
+        self.session.commit()
         return tool
 
     def delete_tool(self, tool: Tool):
@@ -67,14 +70,14 @@ class LLMQueryRepo:
             prompt.category_id = new_prompt.category_id
             prompt.description = new_prompt.description
             prompt.order = new_prompt.order
-            prompt.is_system_prompt = new_prompt.is_system_prompt
+            prompt.prompt_type = new_prompt.prompt_type
             prompt.filename = new_prompt.filename
             prompt.custom_fields = new_prompt.custom_fields
         else:
             self.session.add(new_prompt)
             prompt = new_prompt
 
-        self.session.flush()
+        self.session.commit()
         return prompt
 
     def create_or_update_prompt_category(self, new_category: PromptCategory):
@@ -94,12 +97,31 @@ class LLMQueryRepo:
             LLMQuery.user_identifier == user_identifier,
         ).filter_by(company_id=company.id).order_by(LLMQuery.created_at.desc()).limit(100).all()
 
-    def get_prompts(self, company: Company) -> list[Prompt]:
-        return self.session.query(Prompt).filter_by(company_id=company.id, is_system_prompt=False).all()
+    def get_prompts(self, company: Company, include_all: bool = False) -> list[Prompt]:
+        if include_all:
+            # Include all  prompts: company, system, agent
+            return self.session.query(Prompt).filter(
+                Prompt.company_id == company.id,
+            ).all()
+        else:
+            # Only company prompts, excluding system (default behavior for end users)
+            return self.session.query(Prompt).filter(
+                Prompt.company_id == company.id,
+                Prompt.prompt_type == PromptType.COMPANY.value
+            ).all()
 
     def get_prompt_by_name(self, company: Company, prompt_name: str):
         return self.session.query(Prompt).filter_by(company_id=company.id, name=prompt_name).first()
 
+    def get_category_by_name(self, company_id: int, name: str) -> PromptCategory:
+        return self.session.query(PromptCategory).filter_by(company_id=company_id, name=name).first()
+
+    def get_all_categories(self, company_id: int) -> List[PromptCategory]:
+        return self.session.query(PromptCategory).filter_by(company_id=company_id).order_by(PromptCategory.order).all()
+
     def get_system_prompts(self) -> list[Prompt]:
-        return self.session.query(Prompt).filter_by(is_system_prompt=True, active=True).order_by(Prompt.order).all()
+        return self.session.query(Prompt).filter_by(prompt_type=PromptType.SYSTEM.value, active=True).order_by(Prompt.order).all()
 
+    def delete_prompt(self, prompt: Prompt):
+        self.session.delete(prompt)
+        self.session.commit()

iatoolkit/repositories/models.py

@@ -17,6 +17,19 @@ import enum
 class Base(DeclarativeBase):
     pass
 
+
+class DocumentStatus(str, enum.Enum):
+    PENDING = "pending"
+    PROCESSING = "processing"
+    ACTIVE = "active"
+    FAILED = "failed"
+
+class PromptType(str, enum.Enum):
+    SYSTEM = "system"
+    COMPANY = "company"
+    AGENT = "agent"
+
+
 # relation table for many-to-many relationship between companies and users
 user_company = Table('iat_user_company',
                      Base.metadata,
@@ -149,11 +162,6 @@ class Tool(Base):
         return {column.key: getattr(self, column.key) for column in class_mapper(self.__class__).columns}
 
 
-class DocumentStatus(str, enum.Enum):
-    PENDING = "pending"
-    PROCESSING = "processing"
-    ACTIVE = "active"
-    FAILED = "failed"
 
 
 class CollectionType(Base):
@@ -290,12 +298,13 @@ class Prompt(Base):
     description = Column(String, nullable=False)
     filename = Column(String, nullable=False)
     active = Column(Boolean, default=True)
-    is_system_prompt = Column(Boolean, default=False)
+    prompt_type = Column(String, default=PromptType.COMPANY.value, nullable=False)
     order = Column(Integer, nullable=True, default=0)
     category_id = Column(Integer, ForeignKey('iat_prompt_categories.id'), nullable=True)
     custom_fields = Column(JSON, nullable=False, default=[])
-
     created_at = Column(DateTime, default=datetime.now)
+    def to_dict(self):
+        return {column.key: getattr(self, column.key) for column in class_mapper(self.__class__).columns}
 
     company = relationship("Company", back_populates="prompts")
     category = relationship("PromptCategory", back_populates="prompts")

iatoolkit/services/configuration_service.py

@@ -38,41 +38,11 @@ class ConfigurationService:
         if company_short_name not in self._loaded_configs:
             self._loaded_configs[company_short_name] = self._load_and_merge_configs(company_short_name)
 
-    def get_configuration(self, company_short_name: str, content_key: str):
-        """
-        Public method to provide a specific section of a company's configuration.
-        It uses a cache to avoid reading files from disk on every call.
-        """
-        self._ensure_config_loaded(company_short_name)
-        return self._loaded_configs[company_short_name].get(content_key)
-
-    def get_llm_configuration(self, company_short_name: str):
-        """
-        Convenience helper to obtain the 'llm' configuration block for a company.
-        Kept separate from get_configuration() to avoid coupling tests that
-        assert the number of calls to get_configuration().
-        """
-        default_llm_model = None
-        available_llm_models = []
-        self._ensure_config_loaded(company_short_name)
-        llm_config = self._loaded_configs[company_short_name].get("llm")
-        if llm_config:
-            default_llm_model = llm_config.get("model")
-            available_llm_models = llm_config.get('available_models') or []
-
-        # fallback: if no explicit list of models is provided, use the default model
-        if not available_llm_models and default_llm_model:
-            available_llm_models = [{
-                "id": default_llm_model,
-                "label": default_llm_model,
-                "description": "Modelo por defecto configurado para esta compañía."
-            }]
-        return default_llm_model, available_llm_models
-
     def load_configuration(self, company_short_name: str):
         """
         Main entry point for configuring a company instance.
         This method is invoked by the dispatcher for each registered company.
+        And for the configurator, for editing the configuration of a company.
         """
         logging.info(f"⚙️  Starting configuration for company '{company_short_name}'...")
 
@@ -97,6 +67,14 @@ class ConfigurationService:
         logging.info(f"✅ Company '{company_short_name}' configured successfully.")
         return config, errors
 
+    def get_configuration(self, company_short_name: str, content_key: str):
+        """
+        Public method to provide a specific section of a company's configuration.
+        It uses a cache to avoid reading files from disk on every call.
+        """
+        self._ensure_config_loaded(company_short_name)
+        return self._loaded_configs[company_short_name].get(content_key)
+
     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,
@@ -186,7 +164,6 @@ class ConfigurationService:
 
         return config, []
 
-
     def validate_configuration(self, company_short_name: str) -> list[str]:
         """
         Public method to trigger validation of the current configuration.
@@ -194,83 +171,6 @@ class ConfigurationService:
         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
-        specified in the 'content_files' section using AssetRepository.
-        """
-        main_config_filename = "company.yaml"
-
-        # verify existence of the main configuration file
-        if not self.asset_repo.exists(company_short_name, AssetType.CONFIG, main_config_filename):
-            # raise FileNotFoundError(f"Main configuration file not found: {main_config_filename}")
-            logging.exception(f"Main configuration file not found: {main_config_filename}")
-
-            # return the minimal configuration needed for starting the IAToolkit
-            # this is a for solving a chicken/egg problem when trying to migrate the configuration
-            # from filesystem to database in enterprise installation
-            # see create_assets cli command in enterprise-iatoolkit)
-            return {
-                'id': company_short_name,
-                'name': company_short_name,
-                'llm': {'model': 'gpt-5', 'provider_api_keys': {'openai':''} },
-                }
-
-        # read text and parse
-        yaml_content = self.asset_repo.read_text(company_short_name, AssetType.CONFIG, main_config_filename)
-        config = self.utility.load_yaml_from_string(yaml_content)
-        if not config:
-            return {}
-
-        # Load and merge supplementary content files (e.g., onboarding_cards)
-        for key, filename in config.get('help_files', {}).items():
-            if self.asset_repo.exists(company_short_name, AssetType.CONFIG, filename):
-                supp_content = self.asset_repo.read_text(company_short_name, AssetType.CONFIG, filename)
-                config[key] = self.utility.load_yaml_from_string(supp_content)
-            else:
-                logging.warning(f"⚠️  Warning: Content file not found: {filename}")
-                config[key] = None
-
-        return config
-
     def _register_company_database(self, config: dict) -> Company:
         # register the company in the database: create_or_update logic
         if not config:
@@ -374,12 +274,11 @@ class ConfigurationService:
         from iatoolkit.services.prompt_service import PromptService
         prompt_service = current_iatoolkit().get_injector().get(PromptService)
 
-        prompts_config = config.get('prompts', {})
-
+        prompt_list, categories_config = self._get_prompt_config(config)
         prompt_service.sync_company_prompts(
             company_short_name=company_short_name,
-            prompts_config=prompts_config.get('prompt_list', []),
-            categories_config=prompts_config.get('prompt_categories', []),
+            prompt_list=prompt_list,
+            categories_config=categories_config,
         )
 
     def _register_knowledge_base(self, company_short_name: str, config: dict):
@@ -394,7 +293,6 @@ class ConfigurationService:
         # sync collection types in database
         knowledge_base.sync_collection_types(company_short_name, categories_config)
 
-
     def _validate_configuration(self, company_short_name: str, config: dict):
         """
         Validates the structure and consistency of the company.yaml configuration.
@@ -462,10 +360,9 @@ class ConfigurationService:
                 add_error(f"tools[{i}]", "'params' key must be a dictionary.")
 
         # 6. Prompts
-        prompt_list = config.get("prompts", {}).get("prompt_list", [])
-        prompt_categories = config.get("prompts", {}).get("prompt_categories", [])
+        prompt_list, categories_config = self._get_prompt_config(config)
 
-        category_set = set(prompt_categories)
+        category_set = set(categories_config)
         for i, prompt in enumerate(prompt_list):
             prompt_name = prompt.get("name")
             if not prompt_name:
@@ -480,10 +377,12 @@ class ConfigurationService:
                     add_error(f"prompts[{i}]", "Missing required key: 'description'")
 
             prompt_cat = prompt.get("category")
-            if not prompt_cat:
-                add_error(f"prompts[{i}]", "Missing required key: 'category'")
-            elif prompt_cat not in category_set:
-                add_error(f"prompts[{i}]", f"Category '{prompt_cat}' is not defined in 'prompt_categories'.")
+            prompt_type = prompt.get("prompt_type", 'company').lower()
+            if prompt_type == 'company':
+                if not prompt_cat:
+                    add_error(f"prompts[{i}]", "Missing required key: 'category'")
+                elif prompt_cat not in category_set:
+                    add_error(f"prompts[{i}]", f"Category '{prompt_cat}' is not defined in 'prompt_categories'.")
 
         # 7. User Feedback
         feedback_config = config.get("parameters", {}).get("user_feedback", {})
@@ -530,3 +429,123 @@ class ConfigurationService:
 
         return errors
 
+
+    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)
+                    # Allow accessing existing index
+                    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)
+                # If index equals length, it means append
+                if idx == len(current):
+                    current.append(value)
+                elif 0 <= idx < len(current):
+                    current[idx] = value
+                else:
+                    raise IndexError(f"Index {idx} out of range for list of size {len(current)}")
+            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 get_llm_configuration(self, company_short_name: str):
+        """
+        Convenience helper to obtain the 'llm' configuration block for a company.
+        Kept separate from get_configuration() to avoid coupling tests that
+        assert the number of calls to get_configuration().
+        """
+        default_llm_model = None
+        available_llm_models = []
+        self._ensure_config_loaded(company_short_name)
+        llm_config = self._loaded_configs[company_short_name].get("llm")
+        if llm_config:
+            default_llm_model = llm_config.get("model")
+            available_llm_models = llm_config.get('available_models') or []
+
+        # fallback: if no explicit list of models is provided, use the default model
+        if not available_llm_models and default_llm_model:
+            available_llm_models = [{
+                "id": default_llm_model,
+                "label": default_llm_model,
+                "description": "Modelo por defecto configurado para esta compañía."
+            }]
+        return default_llm_model, available_llm_models
+
+
+    def _load_and_merge_configs(self, company_short_name: str) -> dict:
+        """
+        Loads the main company.yaml and merges data from supplementary files
+        specified in the 'content_files' section using AssetRepository.
+        """
+        main_config_filename = "company.yaml"
+
+        # verify existence of the main configuration file
+        if not self.asset_repo.exists(company_short_name, AssetType.CONFIG, main_config_filename):
+            # raise FileNotFoundError(f"Main configuration file not found: {main_config_filename}")
+            logging.exception(f"Main configuration file not found: {main_config_filename}")
+
+            # return the minimal configuration needed for starting the IAToolkit
+            # this is a for solving a chicken/egg problem when trying to migrate the configuration
+            # from filesystem to database in enterprise installation
+            # see create_assets cli command in enterprise-iatoolkit)
+            return {
+                'id': company_short_name,
+                'name': company_short_name,
+                'llm': {'model': 'gpt-5', 'provider_api_keys': {'openai':''} },
+                }
+
+        # read text and parse
+        yaml_content = self.asset_repo.read_text(company_short_name, AssetType.CONFIG, main_config_filename)
+        config = self.utility.load_yaml_from_string(yaml_content)
+        if not config:
+            return {}
+
+        # Load and merge supplementary content files (e.g., onboarding_cards)
+        for key, filename in config.get('help_files', {}).items():
+            if self.asset_repo.exists(company_short_name, AssetType.CONFIG, filename):
+                supp_content = self.asset_repo.read_text(company_short_name, AssetType.CONFIG, filename)
+                config[key] = self.utility.load_yaml_from_string(supp_content)
+            else:
+                logging.warning(f"⚠️  Warning: Content file not found: {filename}")
+                config[key] = None
+
+        return config
+
+    def _get_prompt_config(self, config):
+        prompts_config = config.get('prompts', {})
+        if isinstance(prompts_config, dict):
+            prompt_list = prompts_config.get('prompt_list', [])
+            categories_config = prompts_config.get('prompt_categories', [])
+        else:
+            prompt_list = config.get('prompts', [])
+            categories_config = config.get('prompt_categories', [])
+
+        return prompt_list, categories_config
+

iatoolkit/services/dispatcher_service.py

@@ -87,9 +87,6 @@ class Dispatcher:
             # system tools registration
             self.tool_service.register_system_tools()
 
-            # system prompts registration
-            self.prompt_service.register_system_prompts()
-
         except Exception as e:
             self.llmquery_repo.rollback()
             raise IAToolkitException(IAToolkitException.ErrorType.DATABASE_ERROR, str(e))