hyperforge-nucliadb 1.0.0.post29__tar.gz → 1.0.0.post34__tar.gz

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/PKG-INFO

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

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/pyproject.toml

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

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/src/hyperforge_nucliadb/ask/kb_analysis.py

@@ -9,7 +9,6 @@ from nuclia.lib.nua_responses import SemanticConfig
 class KnowledgeBoxInfo:
     content_types: Dict[str, int]
     languages: Dict[str, int]
-    facets: Dict[str, int]
     paragraph_facets: Dict[str, int]
     entity_model: str
     generative_model: str
@@ -22,8 +21,6 @@ def get_knowledge_base_analysis(source: Source):
     content_types = {}
     languages: Dict[str, int] = {}
 
-    facets: Dict[str, int] = {}
-
     for key, count in source.facets_native.facets.items():
         if key.startswith("/n/i"):
             content_type = key.replace("/n/i/", "")
@@ -32,14 +29,11 @@ def get_knowledge_base_analysis(source: Source):
         elif key.startswith("/s/p/"):
             language = key.replace("/s/p/", "")
             languages[language] = count
-        else:
-            facets[key] = count
 
     return KnowledgeBoxInfo(
         paragraph_facets=source.paragraph_facets,
         content_types=content_types,
         languages=languages,
-        facets=facets,
         semantic_models=source.learning_configuration.semantic_models,
         entity_model=source.learning_configuration.ner_model,
         generative_model=source.learning_configuration.generative_model,

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/src/hyperforge_nucliadb/ask/multi.py

@@ -122,7 +122,7 @@ async def choose_source(
 async def load_source_information(source: str, manager: Manager) -> Source:
     driver = manager.drivers.get(source)
     if not isinstance(driver, NucliaDBDriver) and not isinstance(driver, SyncDriver):
-        raise ValueError("Source is not a NucliaDB driver")
+        raise ValueError("Source is not a KnowledgeBox source")
     (
         description,
         labels,

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/src/hyperforge_nucliadb/driver.py

@@ -81,8 +81,8 @@ async def manager_connect(conn: NucliaDBConnection):
 
 @driver(
     id="nucliadb",
-    title="NucliaDB Driver",
-    description="Driver for interacting with the NucliaDB API.",
+    title="KnowledgeBox Source",
+    description="Source for interacting with the KnowledgeBox API.",
     config_schema=NucliaDBConfig,
 )
 class NucliaDBDriver(Driver):
@@ -264,13 +264,13 @@ class NucliaDBDriver(Driver):
 
         # Run requests in parallel as otherwise the facets endpoint can take too long on large knowledge boxes.
         tasks = []
-        for prefix, depth in [
-            ("/n/i", 2),  # content types
-            ("/s/p", 2),  # primary language
-            ("/l", 2),  # classification labels
+        for prefix in [
+            "/n/i",  # content types
+            "/s/p",  # primary language
+            # "/l",  # classification labels
         ]:
             request = CatalogFacetsRequest(
-                prefixes=[CatalogFacetsPrefix(prefix=prefix, depth=depth)]
+                prefixes=[CatalogFacetsPrefix(prefix=prefix)]
             )
             tasks.append(
                 self.driver.catalog_facets(kbid=self.config.kbid, content=request)

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/src/hyperforge_nucliadb/sync/agent.py

@@ -157,7 +157,7 @@ class SyncAskAgent(BasicAskAgent):  # type: ignore
                     or len(driver.sync_configs[sync_config_id]) == 0
                 ):
                     raise Exception(
-                        f"Connection ID {sync_config_id} not found in driver sync configs"
+                        f"Connection ID {sync_config_id} not found in source sync configs"
                     )
                 inner_connection_id = driver.sync_configs[sync_config_id][0]
                 creds_providers[sync_config_id] = driver.information[

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/src/hyperforge_nucliadb/sync/driver.py

@@ -52,8 +52,8 @@ class ExternalConnectionOutput(BaseModel):
 
 @driver(
     id="sync",
-    title="Sync Driver",
-    description="Driver for interacting with the Sync API.",
+    title="Sync Source",
+    description="Source for interacting with the Sync API.",
     config_schema=SyncDriverConfig,
 )
 class SyncDriver(NucliaDBDriver):

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/src/hyperforge_nucliadb.egg-info/PKG-INFO

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

{hyperforge_nucliadb-1.0.0.post29 → hyperforge_nucliadb-1.0.0.post34}/tests/test_sync.py

@@ -36,7 +36,7 @@ CONFIG = {
                 "kbid": "e103caf3-f8cb-4161-a57c-aad1192d0666",
                 "key": KB_E103CAF3_F8CB_4161_A57C_AAD1192D0666,
                 "filters": [],
-                "description": "Nuclia Sync driver for testing",
+                "description": "Nuclia Sync source for testing",
                 "connection_ids": ["019cade7-c177-77c5-99c2-c8771f85cf91"],
             },
         },