From 011659b8bc23f282184f963002f995867f1be22b Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 1 May 2025 23:33:54 +0800 Subject: [PATCH] Fix created_at handling in FAISS vector db --- lightrag/kg/faiss_impl.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lightrag/kg/faiss_impl.py b/lightrag/kg/faiss_impl.py index d4e6ad01..c924d2ad 100644 --- a/lightrag/kg/faiss_impl.py +++ b/lightrag/kg/faiss_impl.py @@ -107,7 +107,7 @@ class FaissVectorDBStorage(BaseVectorStorage): if not data: return - current_time = time.time() + current_time = int(time.time()) # Prepare data for embedding list_data = [] @@ -425,7 +425,7 @@ class FaissVectorDBStorage(BaseVectorStorage): if not metadata: return None - return {**metadata, "id": metadata.get("__id__")} + 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 +445,7 @@ class FaissVectorDBStorage(BaseVectorStorage): if fid is not None: metadata = self._id_to_meta.get(fid, {}) if metadata: - results.append({**metadata, "id": metadata.get("__id__")}) + results.append({**metadata, "id": metadata.get("__id__"), "created_at": metadata.get("__created_at__")}) return results