fixed bugs

2025-02-09 19:21:49 +01:00
parent 832a9bb6fb
commit 7d63898015
7 changed files with 67 additions and 61 deletions
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -421,7 +421,7 @@ class PGDocStatusStorage(DocStatusStorage):
    def __post_init__(self):
        pass

-    async def filter_keys(self, data: list[str]) -> set[str]:
+    async def filter_keys(self, data: set[str]) -> set[str]:
        """Return keys that don't exist in storage"""
        keys = ",".join([f"'{_id}'" for _id in data])
        sql = (