cleaned code

2025-02-15 00:01:21 +01:00
parent 7e526d3436
commit 805da7b95b
4 changed files with 21 additions and 59 deletions
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -106,10 +106,16 @@ class BaseVectorStorage(StorageNameSpace):
        """
        raise NotImplementedError

+    async def delete_entity(self, entity_name: str) -> None:
+        raise NotImplementedError
+
+    async def delete_entity_relation(self, entity_name: str) -> None:
+        raise NotImplementedError
+

@dataclass
 class BaseKVStorage(StorageNameSpace):
-    embedding_func: EmbeddingFunc
+    embedding_func: EmbeddingFunc | None = None

    async def get_by_id(self, id: str) -> Union[dict[str, Any], None]:
        raise NotImplementedError