From 8cbac096e57ebe896c92ca5e44407542ca9a8f3a Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Fri, 6 Dec 2024 10:13:45 +0100
Subject: [PATCH] fix: fixes database pruning issue in pgvector

---
 cognee/shared/CodeGraphEntities.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cognee/shared/CodeGraphEntities.py b/cognee/shared/CodeGraphEntities.py
index 87c37bcc9..8859fd0d6 100644
--- a/cognee/shared/CodeGraphEntities.py
+++ b/cognee/shared/CodeGraphEntities.py
@@ -7,7 +7,7 @@ class Repository(DataPoint):
     type: Optional[str] = "Repository"
 
 class CodeFile(DataPoint):
-    __tablename__ = "CodeFile"
+    __tablename__ = "codefile"
     extracted_id: str  # actually file path
     type: Optional[str] = "CodeFile"
     source_code: Optional[str] = None
@@ -21,7 +21,7 @@ class CodeFile(DataPoint):
     }
 
 class CodePart(DataPoint):
-    __tablename__ = "CodePart"
+    __tablename__ = "codepart"
     # part_of: Optional[CodeFile]
     source_code: str
     type: Optional[str] = "CodePart"