cleaned import

2025-02-09 11:24:08 +01:00
parent 61fd3e6127
commit 4cce14e65e
8 changed files with 62 additions and 70 deletions
--- a/lightrag/kg/mongo_impl.py
+++ b/lightrag/kg/mongo_impl.py
@@ -77,9 +77,7 @@ class MongoKVStorage(BaseKVStorage):
        """Drop the collection"""
        await self._data.drop()

-    async def get_by_status(
-        self, status: str
-    ) -> Union[list[dict[str, Any]], None]:
+    async def get_by_status(self, status: str) -> Union[list[dict[str, Any]], None]:
        """Get documents by status and ids"""
        return self._data.find({"status": status})