hyperforge-nucliadb-agentic 1.0.0.post108__tar.gz → 1.0.0.post126__tar.gz

{hyperforge_nucliadb_agentic-1.0.0.post108 → hyperforge_nucliadb_agentic-1.0.0.post126}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperforge_nucliadb_agentic
-Version: 1.0.0.post108
+Version: 1.0.0.post126
 Summary: NucliaDB Hyperforge agent
 Author: Nuclia
 Author-email: Nuclia <nucliadb@nuclia.com>
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Dist: hyperforge>=1.0.0.post184
+Requires-Dist: hyperforge>=1.0.0.post281
 Requires-Dist: lingua-language-detector
 Requires-Dist: nucliadb-sdk
 Requires-Python: >=3.12, <4

{hyperforge_nucliadb_agentic-1.0.0.post108 → hyperforge_nucliadb_agentic-1.0.0.post126}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "hyperforge_nucliadb_agentic"
-version = "1.0.0.post108"
+version = "1.0.0.post126"
 license = "Apache-2.0"
 description = "NucliaDB Hyperforge agent"
 readme = "README.md"
@@ -18,7 +18,7 @@ classifiers = [
 ]
 requires-python = ">=3.12, <4"
 dependencies = [
-    "hyperforge>=1.0.0.post184",
+    "hyperforge>=1.0.0.post281",
     "lingua-language-detector",
     "nucliadb_sdk",
 ]

{hyperforge_nucliadb_agentic-1.0.0.post108 → hyperforge_nucliadb_agentic-1.0.0.post126}/pyproject.toml.orig

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "hyperforge_nucliadb_agentic"
-version = "1.0.0.post108"
+version = "1.0.0.post126"
 license = "Apache-2.0"
 description = "NucliaDB Hyperforge agent"
 authors = [{ name = "Nuclia", email = "nucliadb@nuclia.com" }]
@@ -19,7 +19,7 @@ classifiers = [
 ]
 requires-python = ">=3.12, <4"
 dependencies = [
-    "hyperforge>=1.0.0.post184",
+    "hyperforge>=1.0.0.post281",
     "lingua-language-detector",
     "nucliadb_sdk",
 ]

{hyperforge_nucliadb_agentic-1.0.0.post108 → hyperforge_nucliadb_agentic-1.0.0.post126}/src/hyperforge_nucliadb_agentic/agent.py

@@ -9,7 +9,7 @@ from hyperforge.agent import Agent
 from hyperforge.configure import agent
 from hyperforge.context.agent import ContextAgent
 from hyperforge.definition import FunctionDefinition
-from hyperforge.manager import Manager
+from hyperforge.manager import Manager, build_reasoning
 from hyperforge.memory import Chunk, Context, QuestionMemory, Source
 from hyperforge.models import JSONObject
 from hyperforge_nucliadb.ask.multi import choose_source
@@ -1010,9 +1010,10 @@ class NucliaDBAgent(ContextAgent, Agent[NucliaDBAgentConfig]):
         fallback_values = {
             "show": [ResourceProperties.BASIC, ResourceProperties.ORIGIN],
             "citations": CitationsType.LLM_FOOTNOTES,
-            "generative_model": self.config.generative_model,
+            "generative_model": self.config.generative_model.model_id,
             "rag_strategies": rag_strategies,
             "generate_answer": self.config.generate_inner_answer,
+            "reasoning": build_reasoning(self.config.generative_model),
         }
         return ask_request.model_copy(
             update={

{hyperforge_nucliadb_agentic-1.0.0.post108 → hyperforge_nucliadb_agentic-1.0.0.post126}/src/hyperforge_nucliadb_agentic/ask/audit.py

@@ -9,7 +9,7 @@ import mmh3
 import nats
 from fastapi import Request
 from hyperforge.feature_flag import Features, has_feature
-from hyperforge.models import ExternalUsage, Step
+from hyperforge.models import ExternalUsage, ExternalUsageOperation, Step
 from nucliadb_models.retrieval import RawQuery, RetrievalRequest
 from nucliadb_models.search import (
     NucliaDBClientType,
@@ -48,20 +48,28 @@ from hyperforge_nucliadb_agentic.ask.model import (
 from hyperforge_nucliadb_agentic.ask.predict import AnswerStatusCode
 from hyperforge_nucliadb_agentic.ask.utils.proto import client_type
 
+EXTERNAL_USAGE_PREDICT_TYPES = {
+    ExternalUsageOperation.INTERNET_SEARCH: PredictType.INTERNET_SEARCH,
+}
+
 
 def external_usage_to_predict(
     event: ExternalUsage, ndb_client_type: NucliaDBClientType
-) -> Predict:
-    return Predict(
-        client=KbUsageClientType.Value(ndb_client_type.name),
-        type=PredictType.QUESTION_ANSWER,
-        model=f"{event.provider}/{event.model}",
-        input=event.input_tokens,
-        output=event.output_tokens,
-        num_predicts=event.requests,
-        image=event.image,
-        customer_key=False,
-    )
+) -> list[Predict]:
+    client = KbUsageClientType.Value(ndb_client_type.name)
+    return [
+        Predict(
+            client=client,
+            type=EXTERNAL_USAGE_PREDICT_TYPES[event.operation],
+            # Accounting uses model to identify the external service used.
+            model=event.provider,
+            input=event.input_tokens,
+            output=event.output_tokens,
+            image=event.image,
+            external_requests=event.requests,
+            customer_key=False,
+        )
+    ]
 
 
 class RequestContext:
@@ -209,8 +217,9 @@ class StreamAuditStorage:
         trace_id: str | None = None,
     ) -> None:
         predicts = [
-            external_usage_to_predict(event, client_type)
+            predict
             for event in step.external_usage or []
+            for predict in external_usage_to_predict(event, client_type)
         ]
         if not predicts or self.kb_usage_utility is None:
             logger.warning(

{hyperforge_nucliadb_agentic-1.0.0.post108 → hyperforge_nucliadb_agentic-1.0.0.post126}/src/hyperforge_nucliadb_agentic/config.py

@@ -1,6 +1,7 @@
 from typing import List, Literal, Optional, Tuple
 
 from hyperforge.context.agent import ContextAgentConfig
+from hyperforge.llm_config import LLMConfig, LLMField, llm_defaults
 from hyperforge.utils import WidgetType
 from pydantic import Field
 from pydantic.config import ConfigDict
@@ -19,11 +20,10 @@ class NucliaDBAgentConfig(ContextAgentConfig):
             "show_in_node": True,
         },
     )
-    generative_model: str = Field(
-        default="chatgpt-azure-4o-mini",
+    generative_model: LLMField = Field(
+        default=LLMConfig(model_id=llm_defaults.default),
         title="Generative model",
         description="Model used to generate answers",
-        json_schema_extra={"widget": WidgetType.MODEL_SELECT},
     )
     published_functions: Optional[Tuple[str, ...]] = Field(
         default=(