From 0dc712e0bc8fa369347562aa9e54b7b979e00be5 Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 2 May 2025 16:21:48 +0800 Subject: [PATCH] Fix created_at handling in Chroma vector db --- lightrag/kg/chroma_impl.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/lightrag/kg/chroma_impl.py b/lightrag/kg/chroma_impl.py index 627cf480..c7f8c47e 100644 --- a/lightrag/kg/chroma_impl.py +++ b/lightrag/kg/chroma_impl.py @@ -114,11 +114,17 @@ class ChromaVectorDBStorage(BaseVectorStorage): return try: + import time + current_time = int(time.time()) + ids = list(data.keys()) documents = [v["content"] for v in data.values()] metadatas = [ - {k: v for k, v in item.items() if k in self.meta_fields} - or {"_default": "true"} + { + **{k: v for k, v in item.items() if k in self.meta_fields}, + "created_at": current_time, + } + or {"_default": "true", "created_at": current_time} for item in data.values() ] @@ -183,6 +189,7 @@ class ChromaVectorDBStorage(BaseVectorStorage): "id": results["ids"][0][i], "distance": 1 - results["distances"][0][i], "content": results["documents"][0][i], + "created_at": results["metadatas"][0][i].get("created_at"), **results["metadatas"][0][i], } for i in range(len(results["ids"][0])) @@ -298,6 +305,7 @@ class ChromaVectorDBStorage(BaseVectorStorage): "id": result["ids"][0], "vector": result["embeddings"][0], "content": result["documents"][0], + "created_at": result["metadatas"][0].get("created_at"), **result["metadatas"][0], } except Exception as e: @@ -331,6 +339,7 @@ class ChromaVectorDBStorage(BaseVectorStorage): "id": result["ids"][i], "vector": result["embeddings"][i], "content": result["documents"][i], + "created_at": result["metadatas"][i].get("created_at"), **result["metadatas"][i], } for i in range(len(result["ids"]))