gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-05-03 00:46:28 +08:00
parent debe8f329a
commit 08e8a7ead1
5 changed files with 40 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/kg/chroma_impl.py
View File

@@ -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 = [

14
lightrag/kg/faiss_impl.py
View File

@@ -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

15
lightrag/kg/milvus_impl.py
View File

@@ -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}")

15
lightrag/kg/mongo_impl.py
View File

@@ -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
]

19
lightrag/kg/qdrant_impl.py
View File

@@ -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}")