up

SciPhi-AI · Oct 5, 2024 · df6bd31 · df6bd31
1 parent 0cc4671
commit df6bd31
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/py/core/providers/kg/postgres.py b/py/core/providers/kg/postgres.py
@@ -151,17 +151,15 @@ async def create_tables(self, project_name: str):
         await self.execute_query(query)
 
         # embeddings tables
-        # TODO: deprecating document ID
         query = f"""
             CREATE TABLE IF NOT EXISTS {self._get_table_name("entity_embedding")} (
             id SERIAL PRIMARY KEY,
             name TEXT NOT NULL,
             description TEXT NOT NULL,
             extraction_ids UUID[] NOT NULL,
             description_embedding vector({self.embedding_provider.config.base_dimension}) NOT NULL,
-            document_ids UUID[] NOT NULL, 
-            collection_id UUID NOT NULL, 
-            UNIQUE (name, collection_id)
+            document_id UUID NOT NULL, 
+            UNIQUE (name, document_id)
             );
         """