nucliadb-agentic-api 1.0.0.post244__tar.gz → 1.0.0.post262__tar.gz

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/PKG-INFO

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

nucliadb_agentic_api-1.0.0.post262/VERSION

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

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "nucliadb_agentic_api"
-version = "1.0.0.post244"
+version = "1.0.0.post262"
 description = "NucliaDB Agentic API for Hyperforge"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -72,15 +72,15 @@ server = [
     "nucliadb-sdk>=6.14.0.post6532",
     "nucliadb-telemetry[otel]>=6.14.0.post6534",
     "nucliadb-utils[storages,fastapi]>=6.15.0.post6671",
-    "hyperforge>=1.0.0.post237",
+    "hyperforge>=1.0.0.post281",
     "hyperforge_conditional>=1.0.0.post27",
-    "hyperforge_mcp>=1.0.0.post41",
+    "hyperforge_mcp>=1.0.0.post95",
     "hyperforge_nucliadb>=1.0.0.post46",
-    "hyperforge_rephrase>=1.0.0.post40",
-    "hyperforge_smart>=1.0.0.post30",
-    "hyperforge_summarize>=1.0.0.post29",
-    "hyperforge_google>=1.0.0.post38",
-    "hyperforge_perplexity>=1.0.0.post30",
+    "hyperforge_rephrase>=1.0.0.post74",
+    "hyperforge_smart>=1.0.0.post117",
+    "hyperforge_summarize>=1.0.0.post85",
+    "hyperforge_google>=1.0.0.post71",
+    "hyperforge_perplexity>=1.0.0.post57",
     "pydantic_settings",
     "mcp>=1.26.0",
     "prometheus_client",

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/pyproject.toml.orig

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "nucliadb_agentic_api"
-version = "1.0.0.post244"
+version = "1.0.0.post262"
 authors = [{ name = "AI Data Team", email = "learning@nuclia.com" }]
 description = "NucliaDB Agentic API for Hyperforge"
 readme = "README.md"
@@ -67,15 +67,15 @@ server = [
     "nucliadb-sdk>=6.14.0.post6532",
     "nucliadb-telemetry[otel]>=6.14.0.post6534",
     "nucliadb-utils[storages,fastapi]>=6.15.0.post6671",
-    "hyperforge>=1.0.0.post237",
+    "hyperforge>=1.0.0.post281",
     "hyperforge_conditional>=1.0.0.post27",
-    "hyperforge_mcp>=1.0.0.post41",
+    "hyperforge_mcp>=1.0.0.post95",
     "hyperforge_nucliadb>=1.0.0.post46",
-    "hyperforge_rephrase>=1.0.0.post40",
-    "hyperforge_smart>=1.0.0.post30",
-    "hyperforge_summarize>=1.0.0.post29",
-    "hyperforge_google>=1.0.0.post38",
-    "hyperforge_perplexity>=1.0.0.post30",
+    "hyperforge_rephrase>=1.0.0.post74",
+    "hyperforge_smart>=1.0.0.post117",
+    "hyperforge_summarize>=1.0.0.post85",
+    "hyperforge_google>=1.0.0.post71",
+    "hyperforge_perplexity>=1.0.0.post57",
     "pydantic_settings",
     "mcp>=1.26.0",
     "prometheus_client",

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/src/nucliadb_agentic_api/agentic/ask_transform_to_interaction.py

@@ -6,5 +6,7 @@ def interaction_from_ask_request(ask_request: AskRequest) -> InteractionRequest:
     # Transform the AskRequest into an ARAG interaction
     # This is a placeholder implementation and should be adapted to the actual ARAG interaction format
     interaction = InteractionRequest(question=ask_request.query, streaming=True)
-    interaction.arguments["ask_request"] = ask_request.model_dump_json()
+    interaction.arguments["ask_request"] = ask_request.model_dump_json(
+        exclude_unset=True
+    )
     return interaction

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/src/nucliadb_agentic_api/db/transform.py

@@ -63,7 +63,7 @@ async def transform_agentic_config(
             kb=rephrase.ask_to,
             history=rephrase.history if rephrase.history is not None else False,
         )
-        if rephrase.model:
+        if rephrase.model is not None:
             rephrase_config.model = rephrase.model
         preprocess.append(rephrase_config)
 
@@ -139,16 +139,16 @@ async def transform_agentic_config(
                     ),
                 )  # TODO: pass real config if needed
                 drivers[uid] = mcp_driver_config
-                registered_agents.append(
-                    MCPAgentConfig(
-                        id=f"mcp-agent-{source}",
-                        title=source_title,
-                        source=uid,
-                        transport=Transport.HTTP,
-                        tool_choice_model=source_obj.tool_choice_model or "chatgpt-4.1",
-                        valid_headers=source_obj.valid_headers or [],
-                    )
+                mcp_agent_config = MCPAgentConfig(
+                    id=f"mcp-agent-{source}",
+                    title=source_title,
+                    source=uid,
+                    transport=Transport.HTTP,
+                    valid_headers=source_obj.valid_headers or [],
                 )
+                if source_obj.tool_choice_model is not None:
+                    mcp_agent_config.tool_choice_model = source_obj.tool_choice_model
+                registered_agents.append(mcp_agent_config)
 
             elif source_obj.type == "google":
                 global_driver.append("google")
@@ -167,7 +167,6 @@ async def transform_agentic_config(
                     )
                 )
 
-        models = smart_agent.models
         smart_config = SmartAgentConfig(
             title=f"{title} - Smart Agent",
             planning_mode=smart_agent.mode.value,
@@ -176,14 +175,13 @@ async def transform_agentic_config(
             # SmartAgentConfig's before-validator accepts serialized agent configs.
             registered_agents=[agent.model_dump() for agent in registered_agents],  # type: ignore[misc]
         )
-        # Only set the models if they are provided; otherwise, leave them as None to use hyperforge's default models.
-        # TODO: Provide nucliadb agentic api level defaults for models
-        if models:
-            if models.context_validation:
+        models = smart_agent.models
+        if models is not None:
+            if models.context_validation is not None:
                 smart_config.context_validation_model = models.context_validation
-            if models.planner:
+            if models.planner is not None:
                 smart_config.planner_model = models.planner
-            if models.executor:
+            if models.executor is not None:
                 smart_config.executor_model = models.executor
         context.append(smart_config)
 
@@ -198,7 +196,7 @@ async def transform_agentic_config(
             force_chunk_level_citations=False,
             history=summarize.history if summarize.history is not None else False,
         )
-        if summarize.model:
+        if summarize.model is not None:
             summarize_config.model = summarize.model
         generation.append(summarize_config)
 

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/src/nucliadb_agentic_api/models.py

@@ -2,6 +2,7 @@ from enum import Enum
 from typing import Any, Dict, List, Literal, Optional, Union
 
 from hyperforge.driver import DriverConfig
+from hyperforge.llm_config import LLMConfig, LLMField, llm_defaults
 from hyperforge.models import Rules
 from nucliadb_models import FilterExpression, TextFormat
 from pydantic import BaseModel, Field
@@ -132,7 +133,7 @@ class InteractionRequest(BaseModel):
 class AgenticRephraseConfiguration(BaseModel):
     ask_to: Optional[str] = None
     prompt: Optional[str] = None
-    model: Optional[str] = None
+    model: LLMField | None = Field(default=LLMConfig(model_id=llm_defaults.default))
     history: Optional[bool] = True
 
 
@@ -142,9 +143,11 @@ class AgenticSmartAgentMode(str, Enum):
 
 
 class AgenticSmartAgentModels(BaseModel):
-    context_validation: Optional[str] = None
-    planner: Optional[str] = None
-    executor: Optional[str] = None
+    context_validation: LLMField | None = Field(
+        default=LLMConfig(model_id=llm_defaults.default)
+    )
+    planner: LLMField | None = Field(default=LLMConfig(model_id=llm_defaults.smart))
+    executor: LLMField | None = Field(default=LLMConfig(model_id=llm_defaults.smart))
 
 
 class NucliaDBAgenticSource(BaseModel):
@@ -188,7 +191,9 @@ class MCPAgenticSource(BaseModel):
     description: Optional[str] = None
     uri: str
     headers: Optional[Dict[str, Any]] = None
-    tool_choice_model: Optional[str] = None
+    tool_choice_model: LLMField | None = Field(
+        default=LLMConfig(model_id=llm_defaults.smart)
+    )
     valid_headers: Optional[List[str]] = None
 
 
@@ -216,7 +221,9 @@ AgenticSource = Annotated[
 class AgenticSmartAgentConfiguration(BaseModel):
     mode: AgenticSmartAgentMode = AgenticSmartAgentMode.REACTIVE
     extra_prompt: Optional[str] = None
-    models: Optional[AgenticSmartAgentModels] = None
+    models: AgenticSmartAgentModels | None = Field(
+        default_factory=AgenticSmartAgentModels
+    )
     sources: List[str] = Field(default_factory=list)
     history: Optional[bool] = True
 
@@ -225,7 +232,7 @@ class AgenticSummarizeConfiguration(BaseModel):
     user_prompt: Optional[str] = None
     system_prompt: Optional[str] = None
     conversational: bool = True
-    model: Optional[str] = None
+    model: LLMField | None = Field(default=LLMConfig(model_id=llm_defaults.default))
     history: Optional[bool] = True
 
 
@@ -292,8 +299,8 @@ class MCPSourceSchema(BaseModel):
         default=None,
         description="HTTP headers forwarded with every MCP request",
     )
-    tool_choice_model: Optional[str] = Field(
-        default=None,
+    tool_choice_model: LLMField | None = Field(
+        default=LLMConfig(model_id=llm_defaults.smart),
         description="Model used for MCP tool selection",
     )
     valid_headers: Optional[List[str]] = Field(

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/src/nucliadb_agentic_api/v1/agentic_configs.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, cast
+from typing import TYPE_CHECKING, Any, cast
 
 from fastapi import Header, HTTPException, Request, Response
 from nucliadb_models.resource import NucliaDBRoles
@@ -12,6 +12,20 @@ if TYPE_CHECKING:
     from nucliadb_agentic_api.app import HTTPApplication
 
 
+@router.get(
+    "/api/v1/kb/{kbid}/agentic_configs/schema",
+    status_code=200,
+    summary="Get agentic configuration schema",
+    tags=["Agentic configs"],
+)
+@requires(NucliaDBRoles.READER)
+async def get_agentic_config_schema_endpoint(
+    request: Request,
+    kbid: str,
+) -> dict[str, Any]:
+    return AgenticConfigSchema.model_json_schema()
+
+
 @router.post(
     "/api/v1/kb/{kbid}/agentic_configs/{agentic_id}",
     status_code=201,

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/src/nucliadb_agentic_api/v1/ask.py

@@ -72,9 +72,7 @@ async def ask_knowledgebox_endpoint(
             rpc.get_sdk("reader"), kbid, item
         )
     except rpc.SearchConfigurationNotFound:
-        return HTTPClientError(
-            status_code=400, detail="Search configuration not found"
-        )
+        return HTTPClientError(status_code=400, detail="Search configuration not found")
     except rpc.InvalidAskSearchConfiguration:
         return HTTPClientError(
             status_code=400,

{nucliadb_agentic_api-1.0.0.post244 → nucliadb_agentic_api-1.0.0.post262}/src/nucliadb_agentic_api/v1/ask_websocket.py

@@ -75,9 +75,7 @@ async def websocket_endpoint(
             rpc.get_sdk("reader"), kbid, item
         )
     except rpc.SearchConfigurationNotFound:
-        return HTTPClientError(
-            status_code=400, detail="Search configuration not found"
-        )
+        return HTTPClientError(status_code=400, detail="Search configuration not found")
     except rpc.InvalidAskSearchConfiguration:
         return HTTPClientError(
             status_code=400,
@@ -110,7 +108,7 @@ async def websocket_endpoint(
             interaction.headers[header] = header_value
 
         item.query = interaction.question
-        interaction.arguments["ask_request"] = item.model_dump_json()
+        interaction.arguments["ask_request"] = item.model_dump_json(exclude_unset=True)
 
         async for msg in stream_response(
             websocket.app,

nucliadb_agentic_api-1.0.0.post244/VERSION

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