nucliadb-agentic-api 1.0.0.post294__tar.gz → 1.0.0.post301__tar.gz

{nucliadb_agentic_api-1.0.0.post294 → nucliadb_agentic_api-1.0.0.post301}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_agentic_api
-Version: 1.0.0.post294
+Version: 1.0.0.post301
 Summary: NucliaDB Agentic API for Hyperforge
 Author: AI Data Team
 Author-email: AI Data Team <learning@nuclia.com>

nucliadb_agentic_api-1.0.0.post301/VERSION

@@ -0,0 +1 @@
+1.0.0.post301

nucliadb_agentic_api-1.0.0.post301/alembic/versions/b3781ec6a8ad_migration_of_aws_claude_4_1_opus.py

@@ -0,0 +1,26 @@
+"""migration of aws claude 4 1 opus
+
+Revision ID: b3781ec6a8ad
+Revises: 0fe6c7b5f33d
+Create Date: 2026-10-01 13:04:16.465758
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+from nucliadb_agentic_api.db.llm_migration_utils import migrate_llm_models
+
+# revision identifiers, used by Alembic.
+revision: str = "b3781ec6a8ad"
+down_revision: Union[str, Sequence[str], None] = "0fe6c7b5f33d"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    migrate_llm_models(op.get_bind(), {"aws-claude-4-1-opus": "aws-claude-4-8-opus"})
+
+
+def downgrade() -> None:
+    pass

{nucliadb_agentic_api-1.0.0.post294 → nucliadb_agentic_api-1.0.0.post301}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "nucliadb_agentic_api"
-version = "1.0.0.post294"
+version = "1.0.0.post301"
 description = "NucliaDB Agentic API for Hyperforge"
 readme = "README.md"
 requires-python = ">=3.12"

{nucliadb_agentic_api-1.0.0.post294 → nucliadb_agentic_api-1.0.0.post301}/pyproject.toml.orig

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "nucliadb_agentic_api"
-version = "1.0.0.post294"
+version = "1.0.0.post301"
 authors = [{ name = "AI Data Team", email = "learning@nuclia.com" }]
 description = "NucliaDB Agentic API for Hyperforge"
 readme = "README.md"

{nucliadb_agentic_api-1.0.0.post294 → nucliadb_agentic_api-1.0.0.post301}/src/nucliadb_agentic_api/v1/mcp_nucliadb.py

@@ -45,6 +45,7 @@ from hyperforge_nucliadb_agentic.ask.exceptions import (
 )
 from hyperforge_nucliadb_agentic.ask.model import (
     AskRequest,
+    ChatOptions,
     CitationsType,
     FieldExtensionStrategy,
     Filter,
@@ -52,6 +53,7 @@ from hyperforge_nucliadb_agentic.ask.model import (
     MetadataExtensionType,
     NeighbouringParagraphsStrategy,
 )
+from hyperforge_nucliadb_agentic.ask.search import rpc
 from hyperforge_nucliadb_agentic.ask.search.ask import (
     AskResult,
     ask,
@@ -100,11 +102,11 @@ class SearchDocumentsArgs(BaseModel):
     )
     search_configuration: str | None = Field(
         None,
-        description="Optional. The name of the Search Configuration to use. If not provided, default settings are used.",
+        description="Optional. The name of a saved Search Configuration (ask or find) whose settings (filters, retrieval and RAG strategies, prompts, model...) are applied to the search. If not provided, default settings are used.",
     )
     filters: list[Filter] | None = Field(
         None,
-        description="Optional. A list of filter objects to narrow results. Each filter can include 'all' (AND), 'any' (OR), and/or 'not_all' (NOT AND) label paths like '/l/labelset/label'.",
+        description="Optional. A list of filter objects to narrow results. Each filter can include 'all' (AND), 'any' (OR), and/or 'not_all' (NOT AND) label paths like '/l/labelset/label'. When a Search Configuration is used, these filters are combined (AND) with its saved filters.",
     )
 
 
@@ -217,35 +219,155 @@ def _append_document_ids_to_answer(
     return f"{answer_text}\n\nDocument IDs:\n{document_ids_text}"
 
 
-async def _tool_search_documents(
-    context: MCPContext,
-    args: SearchDocumentsArgs,
-) -> Sequence[TextContent | ImageContent | EmbeddedResource]:
-    ask_request = AskRequest(query=args.query)
-    ask_request.audit_metadata = {"source": "mcp_tool"}
-
-    ask_request.rag_strategies = [
-        FieldExtensionStrategy(fields=["a/title", "a/summary"]),
-        NeighbouringParagraphsStrategy(before=5, after=5),
-        MetadataExtensionStrategy(
-            types=[
-                MetadataExtensionType.CLASSIFICATION_LABELS,
-                MetadataExtensionType.ORIGIN,
-            ]
-        ),
+def _mcp_default_ask_parameters() -> dict[str, Any]:
+    """Parameters used by the MCP tool when the search configuration does not set them."""
+    return {
+        "rag_strategies": [
+            FieldExtensionStrategy(fields=["a/title", "a/summary"]),
+            NeighbouringParagraphsStrategy(before=5, after=5),
+            MetadataExtensionStrategy(
+                types=[
+                    MetadataExtensionType.CLASSIFICATION_LABELS,
+                    MetadataExtensionType.ORIGIN,
+                ]
+            ),
+        ],
+        "debug": True,
+    }
+
+
+# Parameters the MCP tool response format depends on (answer text + footnote
+# citations mapped to documents), so they always win over the search configuration.
+_MCP_FORCED_ASK_PARAMETERS: dict[str, Any] = {
+    "generate_answer": True,
+    "citations": CitationsType.LLM_FOOTNOTES,
+}
+
+
+async def _load_search_configuration(context: MCPContext, name: str) -> dict[str, Any]:
+    """Fetch a saved Search Configuration and return the ask parameters it sets.
+
+    Both `ask` and `find` configurations are supported: `search_documents` runs an
+    ask, so `find` configurations contribute their retrieval parameters (filters,
+    top_k, features, reranker, ...) and parameters with no ask equivalent are ignored.
+    """
+    try:
+        search_config = await rpc.get_search_configuration(
+            context.ndb_reader, context.kbid, name=name
+        )
+    except Exception as e:
+        logger.exception(f"Error fetching search configuration '{name}'")
+        raise ResourceError(
+            f"Failed to load search configuration '{name}'"
+            + (f": {e}" if str(e) else "")
+        ) from e
+    if search_config is None:
+        raise ResourceError(f"Search configuration not found: '{name}'")
+
+    # AskConfig/FindConfig are built with `create_model`, so mypy can't see them as models
+    config_model: BaseModel = search_config.config
+    params = config_model.model_dump(mode="json", exclude_unset=True)
+    params.pop("query", None)  # the tool query always wins
+    ignored = sorted(set(params) - set(AskRequest.model_fields))
+    if "features" in params:
+        # Find supports features ask does not (e.g. `graph`)
+        ask_features = {o.value for o in ChatOptions}
+        features = [f for f in params["features"] if f in ask_features]
+        if len(features) != len(params["features"]):
+            ignored.append(f"features={sorted(set(params['features']) - ask_features)}")
+        if features:
+            params["features"] = features
+        else:
+            params.pop("features")  # nothing left, use the defaults
+    if ignored:
+        logger.info(
+            f"Search configuration '{name}' ({search_config.kind}) parameters not "
+            f"applicable to search_documents are ignored: {ignored}"
+        )
+    return {k: v for k, v in params.items() if k in AskRequest.model_fields}
+
+
+def _combine_filters(
+    config_params: dict[str, Any], tool_filters: list[Filter], name: str
+) -> list[dict[str, Any]]:
+    """AND the tool filters with the search configuration filters.
+
+    The tool filters can only narrow results further: they never replace the
+    filters saved in the search configuration.
+    """
+    if config_params.get("filter_expression") is not None:
+        raise ResourceError(
+            f"Search configuration '{name}' defines a filter expression, which "
+            "cannot be combined with the 'filters' argument. Retry without 'filters'."
+        )
+    combined: list[dict[str, Any]] = [
+        {"all": [f]} if isinstance(f, str) else f
+        for f in config_params.get("filters") or []
     ]
+    combined.extend(f.model_dump(mode="json", exclude_unset=True) for f in tool_filters)
+    return combined
 
-    if args.filters is not None:
-        ask_request.filters = args.filters
 
+async def _build_ask_request(
+    context: MCPContext, args: SearchDocumentsArgs
+) -> AskRequest:
+    """Build the ask request for `search_documents`.
+
+    Precedence: tool query/filters and MCP output requirements > saved search
+    configuration > MCP defaults.
+    """
+    params: dict[str, Any] = {}
     if args.search_configuration is not None:
-        ask_request.search_configuration = args.search_configuration
+        params = await _load_search_configuration(context, args.search_configuration)
+        if params.get("answer_json_schema") is not None:
+            # The MCP response only renders text answers with footnote citations
+            raise ResourceError(
+                f"Search configuration '{args.search_configuration}' uses a structured "
+                "answer (answer_json_schema), which search_documents does not support"
+            )
+        params["search_configuration"] = args.search_configuration
+
+    for key, value in _mcp_default_ask_parameters().items():
+        params.setdefault(key, value)
+
+    if args.filters is not None:
+        if args.search_configuration is not None:
+            params["filters"] = _combine_filters(
+                params, args.filters, args.search_configuration
+            )
+        else:
+            params["filters"] = args.filters
+
+    params.update(_MCP_FORCED_ASK_PARAMETERS)
+    params["audit_metadata"] = {
+        **(params.get("audit_metadata") or {}),
+        "source": "mcp_tool",
+    }
+    params["query"] = args.query
+
+    try:
+        return AskRequest.model_validate(params)
+    except ValidationError as e:
+        error_details = "; ".join(
+            f"'{'.'.join(str(p) for p in err['loc'])}': {err['msg']}"
+            for err in e.errors()
+        )
+        raise ResourceError(
+            "Invalid search parameters"
+            + (
+                f" from search configuration '{args.search_configuration}'"
+                if args.search_configuration
+                else ""
+            )
+            + f": {error_details}"
+        ) from e
 
-    # citations does not work with rag strategies yet
 
-    ask_request.generate_answer = True
-    ask_request.citations = CitationsType.LLM_FOOTNOTES
-    ask_request.debug = True
+async def _tool_search_documents(
+    context: MCPContext,
+    args: SearchDocumentsArgs,
+) -> Sequence[TextContent | ImageContent | EmbeddedResource]:
+    ask_request = await _build_ask_request(context, args)
     try:
         ask_result: AskResult = await ask(
             search_sdk=context.ndb_search,

nucliadb_agentic_api-1.0.0.post294/VERSION

@@ -1 +0,0 @@
-1.0.0.post294