iatoolkit 2.25.1__tar.gz → 2.26.0__tar.gz

{iatoolkit-2.25.1/src/iatoolkit.egg-info → iatoolkit-2.26.0}/PKG-INFO

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

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/__init__.py

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

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/config/system_tools_pack.yaml

@@ -276,8 +276,8 @@ tools:
           description: "Opcional en la práctica. Usa null cuando no necesites filtrar por metadata. No inventes filtros ni envíes [] por defecto. Ejemplo: [{'key':'doc.type','value':'invoice'},{'key':'chunk.source_type','value':'table'}]."
       required:
         - query
-  - function_name: iat_knowledge_wiki_search
-    description: Busca páginas publicadas dentro de las wikis de conocimiento compartidas por la empresa. Úsala para políticas, playbooks, procesos, runbooks, conocimiento operativo, onboarding, guías por área o información curada del tenant. No es memoria personal del usuario ni búsqueda documental cruda.
+  - function_name: iat_wiki_search
+    description: Busca páginas publicadas dentro de las wikis de conocimiento compartidas por la empresa. Úsala como acelerador cuando necesites encontrar rápidamente una página relevante, pero no como punto de entrada por defecto. Si no conoces un `path` válido, primero llama `iat_wiki_get_page` con `path="/"` para leer el índice raíz de la wiki y seguir sus links exactos. `wiki_key` es el identificador estable de una wiki publicada, por ejemplo `sales`, `ops` o `hr`; úsalo cuando ya sepas en qué wiki buscar y omítelo cuando quieras buscar en todas las wikis publicadas disponibles en esta conexión MCP. Esta tool devuelve candidatos; luego usa `iat_wiki_get_page` para leer la página completa antes de responder.
     routing:
       force_include_capability: has_knowledge_wikis
     routing_profile:
@@ -292,15 +292,15 @@ tools:
         - onboarding
         - wiki
       intents:
-        - Search curated company wiki pages for policies, processes, playbooks, runbooks, or area knowledge.
-        - Retrieve published tenant knowledge before answering operational questions.
-        - Find the relevant company wiki page when the user asks about internal procedures.
+        - Search curated company wiki pages when direct lookup is faster than browsing the root index.
+        - Find the relevant company wiki page after the user asks about internal procedures, policies, or runbooks.
+        - Narrow down which exact page to open with iat_wiki_get_page.
       examples:
-        - iat_knowledge_wiki_search(query="politica de descuentos enterprise", wiki_key="sales", limit=5)
-        - iat_knowledge_wiki_search(query="como escalar incidentes sev1", wiki_key="ops", limit=3)
-        - iat_knowledge_wiki_search(query="onboarding de nuevos ejecutivos", limit=5)
+        - iat_wiki_search(query="politica de descuentos enterprise", wiki_key="sales", limit=5)
+        - iat_wiki_search(query="como escalar incidentes sev1", wiki_key="ops", limit=3)
+        - iat_wiki_search(query="onboarding de nuevos ejecutivos", limit=5)
       cost:
-        penalty: 0.07
+        penalty: 0.09
         latency_class: low
         expensive: false
     parameters:
@@ -308,22 +308,25 @@ tools:
       properties:
         query:
           type: string
-          description: Texto que describe el conocimiento publicado que deseas recuperar.
+          description: Consulta en lenguaje natural sobre políticas, playbooks, procesos, runbooks o conocimiento curado publicado en las wikis del tenant.
         wiki_key:
           anyOf:
             - type: string
             - type: "null"
-          description: Clave opcional de la wiki a consultar. Usa null para buscar en todas las wikis publicadas.
+          description: Clave estable de la wiki publicada, por ejemplo `sales`, `ops` o `hr`. Debe coincidir con el `wiki_key` expuesto por el catálogo/resource. Usa null para buscar en todas las wikis publicadas disponibles.
         limit:
           type: integer
-          description: Cantidad máxima de páginas a recuperar.
+          description: Cantidad máxima de páginas candidatas a recuperar antes de elegir una para leer completa.
           minimum: 1
           maximum: 25
       required:
         - query
         - limit
-  - function_name: iat_knowledge_wiki_get_page
-    description: Recupera una página completa de una wiki de conocimiento compartida por la empresa usando wiki_key y path/slug. Úsala después de iat_knowledge_wiki_search cuando necesites leer el contenido completo antes de responder.
+  - function_name: iat_wiki_get_page
+    description: |-
+      Wiki entrypoint: llama `iat_wiki_get_page` con `path="/"` para cargar el índice raíz de una wiki de conocimiento publicada del tenant.
+      Lee UNA página de wiki y devuelve su contenido en Markdown. La wiki documenta contexto de negocio, áreas, sistemas internos, arquitectura, procesos, políticas, runbooks, onboarding, operaciones y glosarios. Es una wiki navegable por links: la página raíz `path="/"` es el punto de entrada y lista las páginas disponibles con sus paths exactos; después abre solo las páginas relevantes siguiendo esos links o los paths devueltos por `iat_wiki_search`.
+      `wiki_key` identifica la wiki publicada, por ejemplo `sales`, `ops` o `hr`. Los `path` son sensibles a mayúsculas y deben coincidir EXACTAMENTE con el link o path publicado; no los inventes ni los reconstruyas de memoria. Si no conoces un path válido, vuelve a llamar con `path="/"`. Esta tool es de solo lectura y es segura de usar siempre que la documentación interna de la empresa ayude a responder. Prefiere abrir pocas páginas puntuales antes de recorrer el corpus completo. Cuando respondas usando material de la wiki, cita la página usada.
     routing:
       force_include_capability: has_knowledge_wikis
     routing_profile:
@@ -334,14 +337,21 @@ tools:
         - page
         - markdown
         - wiki
+        - entrypoint
+        - index
+        - root
+        - internal_docs
       intents:
-        - Open a published tenant wiki page by path or slug.
+        - Load the root index of a tenant knowledge wiki with path="/".
+        - Open a published tenant wiki page by exact path or slug.
         - Read the full markdown content of a company knowledge page before answering.
+        - Follow links from the root index or another wiki page to retrieve the exact page content.
       examples:
-        - iat_knowledge_wiki_get_page(wiki_key="sales", path="pricing.md")
-        - iat_knowledge_wiki_get_page(wiki_key="ops", path="incident-response.md")
+        - iat_wiki_get_page(wiki_key="sales", path="/")
+        - iat_wiki_get_page(wiki_key="sales", path="pricing.md")
+        - iat_wiki_get_page(wiki_key="ops", path="incident-response.md")
       cost:
-        penalty: 0.04
+        penalty: 0.01
         latency_class: low
         expensive: false
     parameters:
@@ -349,10 +359,10 @@ tools:
       properties:
         wiki_key:
           type: string
-          description: Clave de la wiki publicada.
+          description: Clave estable de la wiki publicada, por ejemplo `sales`, `ops` o `hr`.
         path:
           type: string
-          description: Path o slug de la página a recuperar.
+          description: Usa `"/"` para leer el índice raíz de la wiki. En cualquier otro caso, envía el path o slug exacto de la página dentro de esa wiki, idealmente uno listado en el índice raíz o devuelto por `iat_wiki_search`.
       required:
         - wiki_key
         - path

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/infra/connectors/file_connector.py

@@ -9,7 +9,7 @@ from typing import List, Optional
 
 class FileConnector(ABC):
     @abstractmethod
-    def list_files(self) -> List[str]:
+    def list_files(self, prefix: str | None = None) -> List[str]:
         pass
 
     @abstractmethod
@@ -25,4 +25,4 @@ class FileConnector(ABC):
         pass
 
     def generate_presigned_url(self, file_path: str, expiration: int = 3600) -> Optional[str]:
-        return None
+        return None

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/infra/connectors/google_cloud_storage_connector.py

@@ -36,11 +36,12 @@ class GoogleCloudStorageConnector(FileConnector):
             client = storage.Client.from_service_account_json(self.service_account_path)
         return client
 
-    def list_files(self) -> List[dict]:
+    def list_files(self, prefix: str | None = None) -> List[dict]:
         """
         Lista todos los archivos en el bucket de GCS como diccionarios con claves 'path', 'name' y 'metadata'.
         """
-        blobs = self.bucket.list_blobs()
+        normalized_prefix = str(prefix or "").strip().strip("/") or None
+        blobs = self.bucket.list_blobs(prefix=normalized_prefix)
 
         return [
             {

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/infra/connectors/google_drive_connector.py

@@ -52,7 +52,7 @@ class GoogleDriveConnector(FileConnector):
         )
         return service
 
-    def list_files(self) -> List[dict]:
+    def list_files(self, prefix: str | None = None) -> List[dict]:
         """
         Estándar: Lista todos los archivos como diccionarios con claves 'path', 'name' y 'metadata'.
         """

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/infra/connectors/local_file_connector.py

@@ -14,7 +14,7 @@ class LocalFileConnector(FileConnector):
         local_root = os.getenv("ROOT_DIR_LOCAL_FILES", '')
         self.directory = os.path.join(local_root, directory)
 
-    def list_files(self) -> List[dict]:
+    def list_files(self, prefix: str | None = None) -> List[dict]:
         """
         Estándar: Lista todos los archivos como diccionarios con claves 'path', 'name' y 'metadata'.
         """

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/infra/connectors/s3_connector.py

@@ -15,21 +15,32 @@ class S3Connector(FileConnector):
         self.folder = folder
         self.s3 = boto3.client('s3', **auth)
 
-    def list_files(self) -> List[dict]:
+    def list_files(self, prefix: str | None = None) -> List[dict]:
         # List only real S3 objects representing files, excluding folder placeholders.
-        # Construimos el prefijo evitando dobles barras si folder está vacío
-        parts = [p.strip('/') for p in [self.prefix, self.folder] if p]
-        prefix = "/".join(parts)
-
-        # Agregamos slash final solo si hay prefijo, para listar "contenido de carpeta"
-        if prefix:
-            prefix += "/"
-
-        response = self.s3.list_objects_v2(Bucket=self.bucket, Prefix=prefix)
-        files = [
-            obj for obj in response.get('Contents', [])
-            if self._is_file_key(obj.get('Key'))
-        ]
+        list_prefix = self._resolve_list_prefix(prefix)
+        files = []
+        continuation_token = None
+
+        while True:
+            params = {
+                "Bucket": self.bucket,
+                "Prefix": list_prefix,
+            }
+            if continuation_token:
+                params["ContinuationToken"] = continuation_token
+
+            response = self.s3.list_objects_v2(**params)
+            files.extend(
+                obj for obj in response.get('Contents', [])
+                if self._is_file_key(obj.get('Key'))
+            )
+
+            if not response.get("IsTruncated"):
+                break
+
+            continuation_token = response.get("NextContinuationToken")
+            if not continuation_token:
+                break
 
         return [
             {
@@ -44,6 +55,27 @@ class S3Connector(FileConnector):
             for obj in files
         ]
 
+    def _resolve_list_prefix(self, requested_prefix: str | None) -> str:
+        base_prefix = "/".join(p.strip("/") for p in [self.prefix, self.folder] if p)
+        normalized_requested = str(requested_prefix or "").strip().strip("/")
+
+        if normalized_requested:
+            if base_prefix and (
+                normalized_requested == base_prefix
+                or normalized_requested.startswith(f"{base_prefix}/")
+            ):
+                effective_prefix = normalized_requested
+            elif base_prefix:
+                effective_prefix = "/".join([base_prefix, normalized_requested])
+            else:
+                effective_prefix = normalized_requested
+        else:
+            effective_prefix = base_prefix
+
+        if effective_prefix:
+            return f"{effective_prefix.rstrip('/')}/"
+        return ""
+
     def _is_file_key(self, key: str | None) -> bool:
         return isinstance(key, str) and key != "" and not key.endswith("/")
 

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/locales/en.yaml

@@ -626,10 +626,15 @@ ui:
     title: "Knowledge wikis"
     wikis: "Wikis"
     new_wiki: "New wiki"
+    create_wiki: "Create wiki"
+    select_wiki: "Select a wiki"
+    workspace_placeholder: "Choose a wiki from the list or create a new one."
     configure: "Configuration"
     key: "Key"
     name: "Name"
     root: "Storage root"
+    edit_root: "Edit storage root"
+    lock_root: "Lock storage root editing"
     status: "Status"
     status_published: "Published"
     status_draft: "Draft"
@@ -638,25 +643,64 @@ ui:
     pages: "Pages"
     index: "Index"
     mcp: "MCP"
-    lint: "Review"
+    mcp_status: "MCP publication"
+    mcp_status_published: "Published"
+    mcp_status_unpublished: "Not published"
+    mcp_status_published_help: "This wiki index is already exposed as a resource in the tenant MCP catalog."
+    mcp_status_unpublished_help: "This wiki does not yet expose its index as an MCP resource."
+    mcp_wiki_key: "MCP wiki_key"
+    lint: "Validate"
+    validate: "Validate"
+    validation_results: "Validation results"
     sync: "Sync"
     issues: "issues"
-    index_resource: "Index resource"
+    index_resource: "Resource URI"
+    mcp_publish_help: "This action publishes only this wiki's markdown index resource into MCP."
+    mcp_tools_help: "The tools iat_wiki_search and iat_wiki_get_page are enabled separately from the MCP manager."
+    mcp_step_publish_title: "1. Publish resource"
+    mcp_step_publish_text: "The Publish to MCP button adds or updates `iat://knowledge-wikis/{wiki_key}/index` in the tenant MCP resources."
+    mcp_step_tools_title: "2. Enable tools"
+    mcp_step_tools_text: "If you want tool-based search or page reads, enable `iat_wiki_search` and `iat_wiki_get_page` from the MCP manager."
+    mcp_step_usage_title: "3. Use wiki_key and path"
+    mcp_step_usage_text: "MCP clients can read this index first to discover pages, then use `wiki_key` and `path` with the tools."
     copy_uri: "Copy URI"
     publish_mcp: "Publish to MCP"
-    open_mcp: "Open MCP"
+    republish_mcp: "Update publication"
+    open_mcp: "Open MCP manager"
     filter_pages: "Filter pages"
     loading: "Loading..."
     empty: "No wikis configured."
     no_pages: "No pages."
-    no_issues: "No issues."
+    no_issues: "No issues found."
+    lint_summary_clean: "Checked {count} indexed pages. No issues found."
+    lint_summary_issues: "Checked {count} indexed pages. Found {issues} issues."
     missing_key: "Wiki key is required."
     save_ok: "Wiki saved."
     save_error: "Could not save wiki."
-    sync_ok: "Sync queued."
+    delete_confirmation: "Delete wiki"
+    delete_message: "This removes the wiki configuration, generated index, and indexed pages. Source markdown files in storage are not deleted."
+    delete_ok: "Wiki deleted."
+    delete_error: "Could not delete wiki."
+    sync_ok: "Sync completed."
+    sync_partial_ok: "Sync completed with errors."
     sync_error: "Could not sync wiki."
-    lint_ok: "Review completed."
-    lint_error: "Could not review wiki."
+    sync_running: "Syncing..."
+    sync_results: "Sync result"
+    sync_status_idle: "Not run"
+    sync_status_success: "Completed"
+    sync_status_partial: "Completed with errors"
+    sync_status_error: "Failed"
+    sync_seen: "Files seen"
+    sync_indexed: "Pages indexed"
+    sync_failed: "Failed"
+    sync_finished_at: "Finished"
+    sync_first_error: "First error"
+    sync_summary_success: "Reviewed {seen} markdown files and indexed {indexed} pages."
+    sync_summary_partial: "Reviewed {seen} markdown files, indexed {indexed} pages, and found {failed} failures."
+    sync_summary_error: "Sync failed after reviewing {seen} markdown files. {failed} files could not be indexed."
+    sync_summary_empty: "No markdown files were found under this storage root."
+    lint_ok: "Validation completed."
+    lint_error: "Could not validate wiki."
     mcp_ok: "Wiki published to MCP."
     mcp_error: "Could not publish to MCP."
     copy_ok: "URI copied."

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/locales/es.yaml

@@ -626,10 +626,15 @@ ui:
     title: "Wikis de conocimiento"
     wikis: "Wikis"
     new_wiki: "Nueva wiki"
+    create_wiki: "Crear wiki"
+    select_wiki: "Selecciona una wiki"
+    workspace_placeholder: "Elige una wiki del listado o crea una nueva."
     configure: "Configuración"
     key: "Clave"
     name: "Nombre"
     root: "Raíz storage"
+    edit_root: "Editar raíz storage"
+    lock_root: "Bloquear edición de raíz storage"
     status: "Estado"
     status_published: "Publicada"
     status_draft: "Borrador"
@@ -638,25 +643,64 @@ ui:
     pages: "Páginas"
     index: "Índice"
     mcp: "MCP"
-    lint: "Revisar"
+    mcp_status: "Publicación MCP"
+    mcp_status_published: "Publicado"
+    mcp_status_unpublished: "No publicado"
+    mcp_status_published_help: "El índice de esta wiki ya está expuesto como resource en el catálogo MCP del tenant."
+    mcp_status_unpublished_help: "Esta wiki todavía no expone su índice como resource MCP."
+    mcp_wiki_key: "wiki_key MCP"
+    lint: "Validar"
+    validate: "Validar"
+    validation_results: "Resultados de validación"
     sync: "Sincronizar"
     issues: "hallazgos"
-    index_resource: "Resource de índice"
+    index_resource: "URI del resource"
+    mcp_publish_help: "Esta acción publica en MCP solo el resource del índice markdown de esta wiki."
+    mcp_tools_help: "Las tools iat_wiki_search e iat_wiki_get_page se habilitan por separado desde el gestor MCP."
+    mcp_step_publish_title: "1. Publicar resource"
+    mcp_step_publish_text: "El botón Publicar en MCP agrega o actualiza `iat://knowledge-wikis/{wiki_key}/index` dentro de los resources MCP del tenant."
+    mcp_step_tools_title: "2. Habilitar tools"
+    mcp_step_tools_text: "Si quieres búsqueda o lectura por tool, habilita `iat_wiki_search` e `iat_wiki_get_page` desde el gestor MCP."
+    mcp_step_usage_title: "3. Usar wiki_key y path"
+    mcp_step_usage_text: "Los clientes MCP pueden leer primero este índice para descubrir páginas y luego usar `wiki_key` y `path` en las tools."
     copy_uri: "Copiar URI"
     publish_mcp: "Publicar en MCP"
-    open_mcp: "Abrir MCP"
+    republish_mcp: "Actualizar publicación"
+    open_mcp: "Abrir gestor MCP"
     filter_pages: "Filtrar páginas"
     loading: "Cargando..."
     empty: "No hay wikis configuradas."
     no_pages: "No hay páginas."
-    no_issues: "Sin issues."
+    no_issues: "Sin hallazgos."
+    lint_summary_clean: "Se revisaron {count} páginas indexadas. No se encontraron hallazgos."
+    lint_summary_issues: "Se revisaron {count} páginas indexadas. Se encontraron {issues} hallazgos."
     missing_key: "Debes ingresar una key de wiki."
     save_ok: "Wiki guardada."
     save_error: "No se pudo guardar la wiki."
-    sync_ok: "Sincronización encolada."
+    delete_confirmation: "Eliminar wiki"
+    delete_message: "Se borrará la configuración de la wiki, el índice generado y las páginas indexadas. Los archivos markdown fuente del storage no se borran."
+    delete_ok: "Wiki eliminada."
+    delete_error: "No se pudo eliminar la wiki."
+    sync_ok: "Sincronización completada."
+    sync_partial_ok: "Sincronización completada con errores."
     sync_error: "No se pudo sincronizar la wiki."
-    lint_ok: "Revisión completada."
-    lint_error: "No se pudo revisar la wiki."
+    sync_running: "Sincronizando..."
+    sync_results: "Resultado de sincronización"
+    sync_status_idle: "Sin ejecutar"
+    sync_status_success: "Completada"
+    sync_status_partial: "Con errores"
+    sync_status_error: "Falló"
+    sync_seen: "Archivos vistos"
+    sync_indexed: "Páginas indexadas"
+    sync_failed: "Fallidas"
+    sync_finished_at: "Finalizó"
+    sync_first_error: "Primer error"
+    sync_summary_success: "Se revisaron {seen} archivos markdown y se indexaron {indexed} páginas."
+    sync_summary_partial: "Se revisaron {seen} archivos markdown, se indexaron {indexed} páginas y {failed} fallaron."
+    sync_summary_error: "La sincronización falló después de revisar {seen} archivos markdown. {failed} archivos no pudieron indexarse."
+    sync_summary_empty: "No se encontraron archivos markdown bajo esta raíz de storage."
+    lint_ok: "Validación completada."
+    lint_error: "No se pudo validar la wiki."
     mcp_ok: "Wiki publicada en MCP."
     mcp_error: "No se pudo publicar en MCP."
     copy_ok: "URI copiada."

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/repositories/knowledge_wiki_repo.py

@@ -89,6 +89,14 @@ class KnowledgeWikiRepo:
         self.session.commit()
         return wiki
 
+    def delete_wiki(self, company_id: int, wiki_key: str) -> KnowledgeWiki | None:
+        wiki = self.get_wiki_by_key(company_id, wiki_key)
+        if wiki is None:
+            return None
+        self.session.delete(wiki)
+        self.session.commit()
+        return wiki
+
     def get_page_by_path(self, wiki_id: int, path: str) -> KnowledgeWikiPage | None:
         if not wiki_id or not path:
             return None

{iatoolkit-2.25.1 → iatoolkit-2.26.0}/src/iatoolkit/services/embedding_service.py

@@ -181,16 +181,73 @@ class EmbeddingClientFactory:
         )
         self._clients = {}  # Cache for storing initialized client wrappers
 
+    @staticmethod
+    def _freeze_value(value):
+        if isinstance(value, dict):
+            return tuple(
+                (str(key), EmbeddingClientFactory._freeze_value(item))
+                for key, item in sorted(value.items(), key=lambda pair: str(pair[0]))
+            )
+        if isinstance(value, list):
+            return tuple(EmbeddingClientFactory._freeze_value(item) for item in value)
+        return value
+
+    def _build_cache_key(
+        self,
+        company_short_name: str,
+        model_type: str,
+        config_section: str,
+        provider: str,
+        model: str | None,
+        dimensions: Optional[int],
+        embedding_config: dict,
+        openai_client_config: dict | None = None,
+    ) -> tuple:
+        cache_payload = {"embedding_config": dict(embedding_config or {})}
+        if provider == "openai" and isinstance(openai_client_config, dict):
+            cache_payload["openai_client_config"] = {
+                "api_key": openai_client_config.get("api_key") or "",
+                "base_url": openai_client_config.get("base_url") or "",
+                "default_headers": dict(openai_client_config.get("default_headers") or {}),
+            }
+
+        return (
+            company_short_name,
+            model_type,
+            config_section,
+            provider,
+            model or "",
+            dimensions,
+            self._freeze_value(cache_payload),
+        )
+
+    def _drop_stale_cache_entries(self, company_short_name: str, model_type: str, active_cache_key: tuple) -> None:
+        stale_keys = [
+            key
+            for key in self._clients
+            if key[:2] == (company_short_name, model_type) and key != active_cache_key
+        ]
+        for stale_key in stale_keys:
+            self._clients.pop(stale_key, None)
+
+    @staticmethod
+    def _describe_transport(provider: str, openai_client_config: dict | None = None) -> str:
+        if provider != "openai":
+            return provider
+
+        base_url = str((openai_client_config or {}).get("base_url") or "").strip().lower()
+        if "gateway.ai.cloudflare.com" in base_url:
+            return "cloudflare/provider_native"
+        if base_url:
+            return "custom"
+        return "direct"
+
     def get_client(self, company_short_name: str, model_type: str = 'text') -> EmbeddingClientWrapper:
         """
         Retrieves a configured embedding client wrapper for a specific company.
         If the client is not in the cache, it creates and stores it.
         model_type: 'text' or 'image'
         """
-        cache_key = (company_short_name, model_type)
-        if cache_key in self._clients:
-            return self._clients[cache_key]
-
         # Determine config section based on model type
         config_section = 'visual_embedding_provider' if model_type in ['image', 'image_query'] else 'embedding_provider'
 
@@ -213,6 +270,32 @@ class EmbeddingClientFactory:
         # Extract class path if provider is custom
         class_path = embedding_config.get('class_path')
 
+        openai_client_config = None
+        if provider == 'openai':
+            openai_client_config = self._get_openai_client_config(company_short_name, embedding_config)
+
+        cache_key = self._build_cache_key(
+            company_short_name=company_short_name,
+            model_type=model_type,
+            config_section=config_section,
+            provider=provider,
+            model=model,
+            dimensions=dimensions,
+            embedding_config=embedding_config,
+            openai_client_config=openai_client_config,
+        )
+        if cache_key in self._clients:
+            logging.debug(
+                "Reusing embedding client (%s) for '%s' from %s with model: %s provider=%s transport=%s",
+                model_type,
+                company_short_name,
+                config_section,
+                model,
+                provider,
+                self._describe_transport(provider, openai_client_config),
+            )
+            return self._clients[cache_key]
+
         # Logic to handle multiple providers
         wrapper = None
         if provider == 'custom_class':
@@ -276,11 +359,10 @@ class EmbeddingClientFactory:
             )
 
         elif provider == 'openai':
-            client_config = self._get_openai_client_config(company_short_name, embedding_config)
             client = OpenAI(
-                api_key=client_config["api_key"],
-                base_url=client_config["base_url"] or None,
-                default_headers=client_config["default_headers"] or None,
+                api_key=openai_client_config["api_key"],
+                base_url=openai_client_config["base_url"] or None,
+                default_headers=openai_client_config["default_headers"] or None,
             )
             if not model:
                 model='text-embedding-ada-002'
@@ -288,7 +370,16 @@ class EmbeddingClientFactory:
         else:
             raise NotImplementedError(f"Embedding provider '{provider}' is not implemented.")
 
-        logging.debug(f"Embedding client ({model_type}) for '{company_short_name}' created with model: {model}")
+        self._drop_stale_cache_entries(company_short_name, model_type, cache_key)
+        logging.debug(
+            "Embedding client (%s) for '%s' created from %s with model: %s provider=%s transport=%s",
+            model_type,
+            company_short_name,
+            config_section,
+            model,
+            provider,
+            self._describe_transport(provider, openai_client_config),
+        )
         self._clients[cache_key] = wrapper
         return wrapper