From 08e8a7ead19c76fa1cc82e5253eadd26e480ec99 Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 3 May 2025 00:46:28 +0800 Subject: [PATCH] Fix linting --- lightrag/kg/chroma_impl.py | 3 ++- lightrag/kg/faiss_impl.py | 14 ++++++++++++-- lightrag/kg/milvus_impl.py | 15 ++++++++------- lightrag/kg/mongo_impl.py | 15 ++++++++------- lightrag/kg/qdrant_impl.py | 19 ++++++++++--------- 5 files changed, 40 insertions(+), 26 deletions(-) diff --git a/lightrag/kg/chroma_impl.py b/lightrag/kg/chroma_impl.py index c7f8c47e..125f51d4 100644 --- a/lightrag/kg/chroma_impl.py +++ b/lightrag/kg/chroma_impl.py @@ -115,8 +115,9 @@ class ChromaVectorDBStorage(BaseVectorStorage): try: import time + current_time = int(time.time()) - + ids = list(data.keys()) documents = [v["content"] for v in data.values()] metadatas = [ diff --git a/lightrag/kg/faiss_impl.py b/lightrag/kg/faiss_impl.py index c924d2ad..6cf07f41 100644 --- a/lightrag/kg/faiss_impl.py +++ b/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 diff --git a/lightrag/kg/milvus_impl.py b/lightrag/kg/milvus_impl.py index 5e97a49a..3b98ed37 100644 --- a/lightrag/kg/milvus_impl.py +++ b/lightrag/kg/milvus_impl.py @@ -80,8 +80,9 @@ class MilvusVectorDBStorage(BaseVectorStorage): return import time + current_time = int(time.time()) - + list_data: list[dict[str, Any]] = [ { "id": k, @@ -124,10 +125,10 @@ class MilvusVectorDBStorage(BaseVectorStorage): print(results) return [ { - **dp["entity"], - "id": dp["id"], + **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] ] @@ -268,7 +269,7 @@ class MilvusVectorDBStorage(BaseVectorStorage): # Ensure the result contains created_at field if "created_at" not in result[0]: result[0]["created_at"] = None - + return result[0] except Exception as e: logger.error(f"Error retrieving vector data for ID {id}: {e}") @@ -297,12 +298,12 @@ class MilvusVectorDBStorage(BaseVectorStorage): filter=filter_expr, output_fields=list(self.meta_fields) + ["id", "created_at"], ) - + # Ensure each result contains created_at field for item in result: if "created_at" not in item: item["created_at"] = None - + return result or [] except Exception as e: logger.error(f"Error retrieving vector data for IDs {ids}: {e}") diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 2d4975b7..7e5cef9a 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -1001,8 +1001,9 @@ class MongoVectorDBStorage(BaseVectorStorage): # Add current time as Unix timestamp import time + current_time = int(time.time()) - + list_data = [ { "_id": k, @@ -1067,10 +1068,10 @@ class MongoVectorDBStorage(BaseVectorStorage): # Format and return the results with created_at field return [ { - **doc, - "id": doc["_id"], + **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 ] @@ -1165,10 +1166,10 @@ class MongoVectorDBStorage(BaseVectorStorage): # Format results, ensuring created_at is included results = [ { - **doc, + **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 ] diff --git a/lightrag/kg/qdrant_impl.py b/lightrag/kg/qdrant_impl.py index da264f92..35c3d308 100644 --- a/lightrag/kg/qdrant_impl.py +++ b/lightrag/kg/qdrant_impl.py @@ -88,10 +88,11 @@ class QdrantVectorDBStorage(BaseVectorStorage): logger.info(f"Inserting {len(data)} to {self.namespace}") if not data: return - + import time + current_time = int(time.time()) - + list_data = [ { "id": k, @@ -144,10 +145,10 @@ class QdrantVectorDBStorage(BaseVectorStorage): return [ { - **dp.payload, + **dp.payload, "distance": dp.score, - "created_at": dp.payload.get("created_at") - } + "created_at": dp.payload.get("created_at"), + } for dp in results ] @@ -310,12 +311,12 @@ class QdrantVectorDBStorage(BaseVectorStorage): if not result: return None - + # Ensure the result contains created_at field payload = result[0].payload if "created_at" not in payload: payload["created_at"] = None - + return payload except Exception as e: logger.error(f"Error retrieving vector data for ID {id}: {e}") @@ -343,7 +344,7 @@ class QdrantVectorDBStorage(BaseVectorStorage): ids=qdrant_ids, with_payload=True, ) - + # Ensure each result contains created_at field payloads = [] for point in results: @@ -351,7 +352,7 @@ class QdrantVectorDBStorage(BaseVectorStorage): if "created_at" not in payload: payload["created_at"] = None payloads.append(payload) - + return payloads except Exception as e: logger.error(f"Error retrieving vector data for IDs {ids}: {e}")