nucliadb-agentic-api 1.0.0.post207__tar.gz → 1.0.0.post211__tar.gz

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/PKG-INFO

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

nucliadb_agentic_api-1.0.0.post211/VERSION

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

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/pyproject.toml

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

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/pyproject.toml.orig

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

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/src/nucliadb_agentic_api/agentic/ask_result.py

@@ -15,6 +15,7 @@ from hyperforge_nucliadb_agentic.ask.model import (
     AskRetrievalMatch,
     AskTimings,
     AskTokens,
+    AugmentedContext,
     CitationsAskResponseItem,
     ConsumptionResponseItem,
     FootnoteCitationsAskResponseItem,
@@ -92,6 +93,14 @@ class AgenticAskResult(AskResult):
         self.generate_inner_answer = generate_inner_answer
         self.resource = resource
 
+        # These fields are populated by the native ask result, but the
+        # inherited synchronous serializer also expects them to exist.
+        self.prequeries_results = []
+        self.prompt_context = {}
+        self.prompt_context_order = {}
+        self.augmented_context = AugmentedContext()
+        self.debug_chat_model = None
+
         self._answer_text = ""
         self._reasoning_text: str | None = None
 
@@ -106,7 +115,6 @@ class AgenticAskResult(AskResult):
         self.best_matches: list[RetrievalMatch] = []
 
     async def loop(self):
-
         interaction = interaction_from_ask_request(self.ask_request)
         msg: AragAnswer
         async for msg in stream_response(
@@ -258,7 +266,6 @@ class AgenticAskResult(AskResult):
     async def websocket_to_ask(
         self,
     ) -> AsyncGenerator[TextGenerativeResponse | ReasoningGenerativeResponse, None]:
-
         output_nuclia_tokens = 0.0
         input_nuclia_tokens = 0.0
         timings = {}

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/src/nucliadb_agentic_api/db/agentic_configs.py

@@ -8,7 +8,6 @@ import sqlalchemy as sa
 from hyperforge.database import metadata
 from hyperforge.retrieval.config import RetrievalAgentConfig
 from hyperforge_google.config import GoogleDriverConfig, GoogleInnerConfig
-from hyperforge_nucliadb_agentic.ask.model import AskRequest
 from hyperforge_perplexity.config import PerplexityDriverConfig, PerplexityInnerConfig
 from lru import LRU
 from nucliadb_telemetry.utils import get_telemetry, init_telemetry
@@ -20,6 +19,9 @@ from nucliadb_agentic_api.db.sources import Sources
 from nucliadb_agentic_api.db.transform import transform_agentic_config
 from nucliadb_agentic_api.models import AgenticConfigSchema
 
+# Kept for the dormant AskRequest configuration path below.
+# from hyperforge_nucliadb_agentic.ask.model import AskRequest
+
 
 # Imported lazily in methods to avoid any load-order sensitivity between the two
 # table modules (both share the same `hyperforge.database.metadata`).
@@ -241,7 +243,9 @@ class AgenticConfigs:
         external_nucliadb_url: str | None = None,
         default_memory: bool = False,
         workflow_id: str = "default",
-        ask_request: AskRequest | None = None,
+        # The runtime AskRequest is consumed from QuestionMemory by NucliaDBAgent.
+        # Keep this dormant configuration parameter visible until its intended use is known.
+        # ask_request: AskRequest | None = None,
     ) -> RetrievalAgentConfig:
         # For now, we only support one config per KB, so we ignore agent_id and workflow_id, but in the future we can extend this method to support multiple configs per KB and select
         # the right one based on these parameters
@@ -257,7 +261,9 @@ class AgenticConfigs:
             internal_nucliadb_url=internal_nucliadb_url,
             external_nucliadb_url=external_nucliadb_url,
             external_nucliadb_key=external_nucliadb_key,
-            ask_request=ask_request,
+            # The runtime AskRequest is consumed from QuestionMemory by NucliaDBAgent.
+            # Keep this dormant config path visible until its intended use is known.
+            # ask_request=ask_request,
             kbid=kbid,
         )
 

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/src/nucliadb_agentic_api/db/transform.py

@@ -14,7 +14,6 @@ from hyperforge_nucliadb.driver_config import (
     NucliaDBConfig,
     NucliaDBConnection,
 )
-from hyperforge_nucliadb_agentic.ask.model import AskRequest
 from hyperforge_nucliadb_agentic.internal_driver import (
     InternalNucliaDBConfig,
     InternalNucliaDBConnection,
@@ -23,6 +22,9 @@ from hyperforge_nucliadb_agentic.internal_driver import (
 from nucliadb_agentic_api.db.sources import Sources
 from nucliadb_agentic_api.models import AgenticConfigSchema
 
+# Kept for the dormant AskRequest configuration path below.
+# from hyperforge_nucliadb_agentic.ask.model import AskRequest
+
 
 async def transform_agentic_config(
     agentic_config: AgenticConfigSchema,
@@ -32,7 +34,9 @@ async def transform_agentic_config(
     internal_nucliadb_url: str | None = None,
     external_nucliadb_url: str | None = None,
     external_nucliadb_key: str | None = None,
-    ask_request: AskRequest | None = None,
+    # The runtime AskRequest is consumed from QuestionMemory by NucliaDBAgent.
+    # Keep this dormant configuration parameter visible until its intended use is known.
+    # ask_request: AskRequest | None = None,
     kbid: str = "",
 ) -> Tuple[RetrievalAgentConfig, Dict[str, DriverConfig], list[str]]:
     drivers: Dict[str, DriverConfig] = {}
@@ -89,6 +93,7 @@ async def transform_agentic_config(
                 # Same KB different
                 uid = uuid4().hex
 
+                source_config["id"] = f"nucliadb-agent-{source}"
                 source_config["sources"] = [uid]
                 source_config["module"] = "nucliadb_agent"
 
@@ -135,6 +140,9 @@ async def transform_agentic_config(
                 registered_agents.append(source_config)
             elif source_obj.type == "mcp":
                 uid = uuid4().hex
+                # See the nucliadb branch above for why this must be
+                # deterministic rather than random.
+                source_config["id"] = f"mcp-agent-{source}"
                 source_config["sources"] = [uid]
                 source_config["transport"] = Transport.HTTP
                 source_config["module"] = "mcp"
@@ -151,11 +159,13 @@ async def transform_agentic_config(
                 registered_agents.append(source_config)
 
             elif source_obj.type == "google":
+                source_config["id"] = f"google-agent-{source}"
                 source_config["module"] = "google"
                 global_driver.append("google")
                 registered_agents.append(source_config)
 
             elif source_obj.type == "perplexity":
+                source_config["id"] = f"perplexity-agent-{source}"
                 source_config["module"] = "perplexity"
                 global_driver.append("perplexity")
                 registered_agents.append(source_config)

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/src/nucliadb_agentic_api/server/session.py

@@ -24,6 +24,7 @@ from nucliadb_telemetry import errors
 from nucliadb_telemetry.utils import get_telemetry
 from nucliadb_utils.settings import AuditSettings
 from opentelemetry import trace
+from pydantic import ValidationError
 
 from nucliadb_agentic_api import logger
 from nucliadb_agentic_api.db.agentic_configs import AgenticConfigs
@@ -90,11 +91,6 @@ class NucliaDBAgenticSessionManager(SessionManager):
     async def activate(self, message: StartInteraction):
         topic = None
 
-        ask_request_json = message.arguments.get("ask_request")
-        ask_request = None
-        if ask_request_json:
-            ask_request = AskRequest.model_validate_json(ask_request_json)
-
         logger.info("Activation message received: %s", message)
         observation = activation_observer()
         observation.start()
@@ -115,6 +111,24 @@ class NucliaDBAgenticSessionManager(SessionManager):
                 message.workflow_id,
             )
 
+            ask_request_json = message.arguments.get("ask_request")
+            if ask_request_json is not None:
+                try:
+                    AskRequest.model_validate_json(ask_request_json)
+                except ValidationError as exc:
+                    logger.warning("Invalid ask_request argument: %s", exc)
+                    observation.set_status("error")
+                    await self.callback(
+                        topic,
+                        AragAnswer(
+                            exception=ARAGException(detail=exc.json()),
+                            operation=AnswerOperation.ERROR,
+                        ),
+                    )
+                    await self.send_message(topic, AgentDone())
+                    observation.end()
+                    return
+
             # Get or load session
             config = await self.agent_manager.get_agent_config(
                 account=message.account,
@@ -124,7 +138,9 @@ class NucliaDBAgenticSessionManager(SessionManager):
                 external_nucliadb_url=self.settings.external_nucliadb_url,
                 external_nucliadb_key=self.settings.external_nucliadb_key,
                 workflow_id=message.workflow_id,
-                ask_request=ask_request,
+                # The AskRequest is consumed from QuestionMemory by NucliaDBAgent.
+                # Keep this dormant config path visible until its intended use is known.
+                # ask_request=ask_request,
             )
 
             state = await get_state(

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/src/nucliadb_agentic_api/v1/ask.py

@@ -39,7 +39,7 @@ from nucliadb_agentic_api.v1.router import router
     status_code=200,
     summary="Ask Knowledge Box",
     description="Ask questions on a Knowledge Box",
-    tags=["Search"],
+    tags=["Ask"],
     response_model=SyncAskResponse,
 )
 @requires(NucliaDBRoles.READER)
@@ -123,7 +123,7 @@ async def ask_knowledgebox_endpoint(
     status_code=200,
     summary="Ask a resource (by id)",
     description="Ask questions to a resource",
-    tags=["Search"],
+    tags=["Ask"],
     response_model=SyncAskResponse,
 )
 @requires(NucliaDBRoles.READER)
@@ -185,7 +185,7 @@ async def resource_ask_endpoint_by_uuid(
     status_code=200,
     summary="Ask a resource (by slug)",
     description="Ask questions to a resource",
-    tags=["Search"],
+    tags=["Ask"],
     response_model=SyncAskResponse,
 )
 @requires(NucliaDBRoles.READER)

{nucliadb_agentic_api-1.0.0.post207 → nucliadb_agentic_api-1.0.0.post211}/src/nucliadb_agentic_api/v1/mcp_nucliadb.py

@@ -621,7 +621,7 @@ async def read_resource(
         )
 
 
-@router.get("/.well-known/oauth-protected-resource/api/v1/kb/{kbid}/mcp")
+@router.get("/.well-known/oauth-protected-resource/api/v1/kb/{kbid}/mcp", tags=["MCP"])
 async def mcp_protected_resource_metadata(
     request: Request,
     kbid: str,
@@ -642,8 +642,8 @@ async def mcp_protected_resource_metadata(
     }
 
 
-@router.get("/api/v1/kb/{kbid}/mcp")
-@router.post("/api/v1/kb/{kbid}/mcp")
+@router.get("/api/v1/kb/{kbid}/mcp", tags=["MCP"])
+@router.post("/api/v1/kb/{kbid}/mcp", tags=["MCP"])
 @requires_one([NucliaDBRoles.READER])
 async def mcp_handler(
     request: Request,

nucliadb_agentic_api-1.0.0.post207/VERSION

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