hyperforge-nucliadb 1.0.0.post85__tar.gz → 1.0.0.post91__tar.gz

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperforge_nucliadb
-Version: 1.0.0.post85
+Version: 1.0.0.post91
 Summary: NucliaDB Hyperforge agent
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hyperforge_nucliadb"
-version = "1.0.0.post85"
+version = "1.0.0.post91"
 license = "Apache-2.0"
 description = "NucliaDB Hyperforge agent"
 authors = [{ name = "Nuclia", email = "nucliadb@nuclia.com" }]

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/src/hyperforge_nucliadb/advanced_ask_agent.py

@@ -190,7 +190,7 @@ def build_ask_request(
     filter_expression = parse_filter_expression(agent, driver)
     ask_request = AskRequest(
         query=question,
-        generative_model=agent.generative_model,
+        generative_model=agent.generative_model.model_id,
         citations=True,
         filters=driver.filters,
     )

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/src/hyperforge_nucliadb/advanced_ask_config.py

@@ -1,6 +1,7 @@
 from typing import Any, List, Literal, Optional, Union
 
 from hyperforge.context.config import ContextAgentConfig
+from hyperforge.llm_config import LLMConfig, LLMField, llm_defaults
 from hyperforge.utils import WidgetType
 from nucliadb_models.filters import FilterExpression
 from nucliadb_models.search import (
@@ -30,11 +31,10 @@ class AdvancedAskAgentConfig(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},
     )
     features: Optional[list[ChatOptions]] = Field(
         default=None,

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/src/hyperforge_nucliadb/ask/config.py

@@ -1,6 +1,7 @@
 from typing import List, Literal, Optional
 
 from hyperforge.context.config import ContextAgentConfig
+from hyperforge.llm_config import LLMConfig, LLMField, llm_defaults
 from hyperforge.utils import WidgetType
 from nucliadb_models.search import KnowledgeGraphEntity
 from pydantic import Field
@@ -40,11 +41,10 @@ class AskAgentConfig(ContextAgentConfig):
             "show_in_node": True,
         },
     )
-    configuration_model: str = Field(
-        default="gemini-2.5-flash",
+    configuration_model: LLMField = Field(
+        default=LLMConfig(model_id=llm_defaults.fast),
         title="Generative model",
         description="Model used to generate the configuration",
-        json_schema_extra={"widget": WidgetType.MODEL_SELECT},
     )
     fast_answer: bool = True
     # Setting this to True so that this PR will not break the current behavior

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/src/hyperforge_nucliadb/basic_ask_agent.py

@@ -709,7 +709,7 @@ class BasicAskAgent(ContextAgent, Agent[BasicAskAgentConfig]):
         # Use full resource strategy by requesting title and summary fields
         ask_request = AskRequest(
             query=question,
-            generative_model=self.config.generative_model,
+            generative_model=self.config.generative_model.model_id,
             # TODO: Consider what to do when multiple resources match
             filter_expression=filter_expression,
             rag_strategies=[
@@ -1132,7 +1132,7 @@ class BasicAskAgent(ContextAgent, Agent[BasicAskAgentConfig]):
             query=question,
             show=[ResourceProperties.BASIC, ResourceProperties.ORIGIN],
             citations=CitationsType.LLM_FOOTNOTES,
-            generative_model=self.config.generative_model,
+            generative_model=self.config.generative_model.model_id,
             filter_expression=filter_expression,
             rag_strategies=rag_strategies,
         )
@@ -1244,7 +1244,7 @@ class BasicAskAgent(ContextAgent, Agent[BasicAskAgentConfig]):
         ask_request = AskRequest(
             query=question,
             citations=True,
-            generative_model=self.config.generative_model,
+            generative_model=self.config.generative_model.model_id,
             filters=nucliadb_driver.config.filters,
         )
         paragraphs = await nucliadb_driver.ask(

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/src/hyperforge_nucliadb/basic_ask_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
@@ -15,11 +16,10 @@ class BasicAskAgentConfig(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=(

{hyperforge_nucliadb-1.0.0.post85 → hyperforge_nucliadb-1.0.0.post91}/src/hyperforge_nucliadb.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperforge_nucliadb
-Version: 1.0.0.post85
+Version: 1.0.0.post91
 Summary: NucliaDB Hyperforge agent
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0