Fix linting

lightrag/kg/chroma_impl.py

@@ -115,6 +115,7 @@ class ChromaVectorDBStorage(BaseVectorStorage):
 
         try:
             import time
+
             current_time = int(time.time())
 
             ids = list(data.keys())

lightrag/kg/faiss_impl.py

@@ -425,7 +425,11 @@ class FaissVectorDBStorage(BaseVectorStorage):
         if not metadata:
             return None
 
-        return {**metadata, "id": metadata.get("__id__"), "created_at": metadata.get("__created_at__")}
+        return {
+            **metadata,
+            "id": metadata.get("__id__"),
+            "created_at": metadata.get("__created_at__"),
+        }
 
     async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]:
         """Get multiple vector data by their IDs
@@ -445,7 +449,13 @@ class FaissVectorDBStorage(BaseVectorStorage):
             if fid is not None:
                 metadata = self._id_to_meta.get(fid, {})
                 if metadata:
-                    results.append({**metadata, "id": metadata.get("__id__"), "created_at": metadata.get("__created_at__")})
+                    results.append(
+                        {
+                            **metadata,
+                            "id": metadata.get("__id__"),
+                            "created_at": metadata.get("__created_at__"),
+                        }
+                    )
 
         return results
 

lightrag/kg/milvus_impl.py

@@ -80,6 +80,7 @@ class MilvusVectorDBStorage(BaseVectorStorage):
             return
 
         import time
+
         current_time = int(time.time())
 
         list_data: list[dict[str, Any]] = [
@@ -127,7 +128,7 @@ class MilvusVectorDBStorage(BaseVectorStorage):
                 **dp["entity"],
                 "id": dp["id"],
                 "distance": dp["distance"],
-                "created_at": dp["entity"].get("created_at")
+                "created_at": dp["entity"].get("created_at"),
             }
             for dp in results[0]
         ]

lightrag/kg/mongo_impl.py

@@ -1001,6 +1001,7 @@ class MongoVectorDBStorage(BaseVectorStorage):
 
         # Add current time as Unix timestamp
         import time
+
         current_time = int(time.time())
 
         list_data = [
@@ -1070,7 +1071,7 @@ class MongoVectorDBStorage(BaseVectorStorage):
                 **doc,
                 "id": doc["_id"],
                 "distance": doc.get("score", None),
-                "created_at": doc.get("created_at")  # Include created_at field
+                "created_at": doc.get("created_at"),  # Include created_at field
             }
             for doc in results
         ]
@@ -1167,7 +1168,7 @@ class MongoVectorDBStorage(BaseVectorStorage):
                 {
                     **doc,
                     "id": doc["_id"],
-                    "created_at": doc.get("created_at")  # Include created_at field
+                    "created_at": doc.get("created_at"),  # Include created_at field
                 }
                 for doc in matching_records
             ]

lightrag/kg/qdrant_impl.py

@@ -90,6 +90,7 @@ class QdrantVectorDBStorage(BaseVectorStorage):
             return
 
         import time
+
         current_time = int(time.time())
 
         list_data = [
@@ -146,7 +147,7 @@ class QdrantVectorDBStorage(BaseVectorStorage):
             {
                 **dp.payload,
                 "distance": dp.score,
-                "created_at": dp.payload.get("created_at")
+                "created_at": dp.payload.get("created_at"),
             }
             for dp in results
         ]