hyperforge-nucliadb 1.0.0.post72__tar.gz → 1.0.0.post87__tar.gz

{hyperforge_nucliadb-1.0.0.post72 → hyperforge_nucliadb-1.0.0.post87}/PKG-INFO

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

{hyperforge_nucliadb-1.0.0.post72 → hyperforge_nucliadb-1.0.0.post87}/pyproject.toml

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

{hyperforge_nucliadb-1.0.0.post72 → hyperforge_nucliadb-1.0.0.post87}/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.post72 → hyperforge_nucliadb-1.0.0.post87}/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.post72 → hyperforge_nucliadb-1.0.0.post87}/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.post72 → hyperforge_nucliadb-1.0.0.post87}/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.post72 → hyperforge_nucliadb-1.0.0.post87}/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.post72 → hyperforge_nucliadb-1.0.0.post87}/src/hyperforge_nucliadb.egg-info/PKG-INFO

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

{hyperforge_nucliadb-1.0.0.post72 → hyperforge_nucliadb-1.0.0.post87}/tests/test_nucliadb.py

@@ -28,7 +28,7 @@ from hyperforge_nucliadb.driver_config import (
 
 NUA_KEY = os.environ.get(
     "NUA_KEY",
-) or cassette_nua_key("https://europe-1.nuclia.cloud/")
+) or cassette_nua_key("https://europe-1.dp.progress.cloud/")
 KB_DE48CFAA_3209_4041_BB64_8604AFF061FB = os.environ.get(
     "KB_DE48CFAA_3209_4041_BB64_8604AFF061FB"
 ) or cassette_nua_key("https://europe-1.nuclia.cloud/")
@@ -250,7 +250,7 @@ async def test_nucliadb_agent():
 
 
 @pytest.mark.asyncio
-@pytest.mark.vcr(ignore_localhost=True, ignore_hosts=["europe-1.dp.progress.cloud"])
+@pytest.mark.vcr(ignore_localhost=True)
 async def test_nucliadb_agent_simple():
     answers = []
 
@@ -276,7 +276,7 @@ async def test_nucliadb_agent_simple():
 
 
 @pytest.mark.asyncio
-@pytest.mark.vcr(ignore_localhost=True, ignore_hosts=["europe-1.dp.progress.cloud"])
+@pytest.mark.vcr(ignore_localhost=True)
 async def test_nucliadb_agent_simple_disable_ai_parameter_search():
     question_memory = await arag_main(
         agent_id="default",
@@ -294,7 +294,7 @@ async def test_nucliadb_agent_simple_disable_ai_parameter_search():
 
 
 @pytest.mark.asyncio
-@pytest.mark.vcr(ignore_localhost=True, ignore_hosts=["europe-1.dp.progress.cloud"])
+@pytest.mark.vcr(ignore_localhost=True)
 async def test_nucliadb_agent_basic_ask():
     config = deepcopy(CONFIG_SIMPLE)
 
@@ -475,13 +475,17 @@ def test_build_ask_request():
 
 @pytest.mark.asyncio
 async def test_nucliadb_not_localhost():
+    config = deepcopy(CONFIG_LOCAL)
+    config["preprocess"] = []
+    config["generation"] = []
+
     with pytest.raises(PrivateUrlError):
         await arag_main(
             agent_id="default",
             internal_nua=False,
-            external_nua_api_key=NUA_KEY,
+            external_nua_api_key=None,
             question="Como usar max_tokens.answer. En español y dame link a la doucmentación",
-            config=CONFIG_LOCAL,
+            config=config,
             loaded_modules=[
                 "hyperforge_nucliadb",
                 "hyperforge_summarize",

{hyperforge_nucliadb-1.0.0.post72 → hyperforge_nucliadb-1.0.0.post87}/tests/test_sync.py

@@ -15,7 +15,7 @@ from hyperforge.pubsub import UserToAgentInteraction
 
 NUA_KEY = os.environ.get(
     "NUA_KEY",
-) or cassette_nua_key("https://europe-1.nuclia.cloud/")
+) or cassette_nua_key("https://europe-1.dp.progress.cloud/")
 
 
 KB_E103CAF3_F8CB_4161_A57C_AAD1192D0666 = os.environ.get(
@@ -23,7 +23,7 @@ KB_E103CAF3_F8CB_4161_A57C_AAD1192D0666 = os.environ.get(
 ) or cassette_nua_key("https://europe-1.nuclia.cloud/")
 
 pytestmark = [
-    pytest.mark.vcr(ignore_localhost=True, ignore_hosts=["europe-1.dp.progress.cloud"]),
+    pytest.mark.vcr(ignore_localhost=True),
     pytest.mark.asyncio,
 ]
 
@@ -87,7 +87,7 @@ def _mock_sync_response(
     )
 
 
-async def test_sync_agent(arag_api: AsyncClient):
+async def test_sync_agent():
     sync_driver_config = cast(dict[str, Any], CONFIG["drivers"][0]["config"])  # type: ignore
     sync_base_url = f"{sync_driver_config['url']}/v1/kb/{sync_driver_config['kbid']}"