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

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

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

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

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

{hyperforge_nucliadb-1.0.0.post62 → hyperforge_nucliadb-1.0.0.post72}/src/hyperforge_nucliadb/driver.py

@@ -111,14 +111,10 @@ class NucliaDBDriver(Driver):
 
     async def labels(self) -> Dict[str, List[str]]:
         labelsets = await self.driver.get_labelsets(kbid=self.config.kbid)
-        result = {}
-        for labelset in labelsets.labelsets:
-            labels = await self.driver.get_labelset(
-                kbid=self.config.kbid, labelset=labelset
-            )
-            result[labelset] = [x.title for x in labels.labels]
-
-        return result
+        return {
+            labelset_id: [label.title for label in labelset.labels]
+            for labelset_id, labelset in labelsets.labelsets.items()
+        }
 
     async def field_facets(self) -> Dict[str, int]:
         field_labels = {}

{hyperforge_nucliadb-1.0.0.post62 → hyperforge_nucliadb-1.0.0.post72}/src/hyperforge_nucliadb.egg-info/PKG-INFO

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

{hyperforge_nucliadb-1.0.0.post62 → hyperforge_nucliadb-1.0.0.post72}/src/hyperforge_nucliadb.egg-info/SOURCES.txt

@@ -34,5 +34,6 @@ src/hyperforge_nucliadb/sync/config.py
 src/hyperforge_nucliadb/sync/config_driver.py
 src/hyperforge_nucliadb/sync/driver.py
 tests/test_ask_utils.py
+tests/test_driver.py
 tests/test_nucliadb.py
 tests/test_sync.py

hyperforge_nucliadb-1.0.0.post72/tests/test_driver.py

@@ -0,0 +1,34 @@
+from types import SimpleNamespace
+from unittest.mock import AsyncMock
+
+import pytest
+
+from hyperforge_nucliadb.driver import NucliaDBDriver
+
+
+def make_driver(client):
+    return NucliaDBDriver.model_construct(
+        provider="nucliadb",
+        name="test",
+        config=SimpleNamespace(kbid="test-kb"),
+        driver=client,
+        manager=SimpleNamespace(),
+        _synonyms=None,
+    )
+
+
+@pytest.mark.asyncio
+async def test_labels_uses_complete_labelsets_response():
+    client = AsyncMock()
+    client.get_labelsets.return_value = SimpleNamespace(
+        labelsets={
+            "type": SimpleNamespace(labels=[SimpleNamespace(title="report")]),
+            "topic": SimpleNamespace(labels=[SimpleNamespace(title="science")]),
+        }
+    )
+    driver = make_driver(client)
+
+    assert await driver.labels() == {"type": ["report"], "topic": ["science"]}
+
+    client.get_labelsets.assert_awaited_once_with(kbid="test-kb")
+    client.get_labelset.assert_not_awaited()