search working

SciPhi-AI · Oct 20, 2024 · a220ace · a220ace
1 parent 40cbc09
commit a220ace
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/py/core/configs/full.toml b/py/core/configs/full.toml
@@ -11,6 +11,6 @@ overlap = 256
 
 [orchestration]
 provider = "hatchet"
-kg_creation_concurrency_limit = 32
+kg_creation_concurrency_lipmit = 32
 ingestion_concurrency_limit = 128
 kg_enrichment_concurrency_limit = 8
diff --git a/py/core/pipes/kg/deduplication_summary.py b/py/core/pipes/kg/deduplication_summary.py
@@ -174,8 +174,6 @@ async def _run_logic(
         entity_descriptions = (
             await self.kg_provider.get_entities(
                 collection_id,                
-                offset,
-                -1,
                 entity_names=entity_names,
                 entity_table_name="entity_embedding",
             )

diff --git a/py/core/providers/kg/postgres.py b/py/core/providers/kg/postgres.py
@@ -457,7 +457,7 @@ async def vector_query(self, query: str, **kwargs: Any) -> Any:
 
         table_name = ""
         if search_type == "__Entity__":
-            table_name = "entity_embedding"
+            table_name = "entity_deduplicated"
         elif search_type == "__Relationship__":
             table_name = "triple_raw"
         elif search_type == "__Community__":

diff --git a/py/shared/abstractions/vector.py b/py/shared/abstractions/vector.py
@@ -109,7 +109,7 @@ class VectorTableName(str, Enum):
 
     CHUNKS = "chunks"
     ENTITIES = "entity_embedding"
-    ENTITY_DEDUPLICATED = "entity_deduplicated"
+    ENTITIES_DEDUPLICATED = "entity_deduplicated"
     # TODO: Add support for triples
     # TRIPLES = "triple_raw"
     COMMUNITIES = "community_report"