Fix time handle bugs for nano vector db
Browse files
lightrag/kg/nano_vector_db_impl.py
CHANGED
@@ -291,7 +291,12 @@ class NanoVectorDBStorage(BaseVectorStorage):
|
|
291 |
client = await self._get_client()
|
292 |
result = client.get([id])
|
293 |
if result:
|
294 |
-
|
|
|
|
|
|
|
|
|
|
|
295 |
return None
|
296 |
|
297 |
async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]:
|
@@ -307,7 +312,15 @@ class NanoVectorDBStorage(BaseVectorStorage):
|
|
307 |
return []
|
308 |
|
309 |
client = await self._get_client()
|
310 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
311 |
|
312 |
async def drop(self) -> dict[str, str]:
|
313 |
"""Drop all vector data from storage and clean up resources
|
|
|
291 |
client = await self._get_client()
|
292 |
result = client.get([id])
|
293 |
if result:
|
294 |
+
dp = result[0]
|
295 |
+
return {
|
296 |
+
**dp,
|
297 |
+
"id": dp.get("__id__"),
|
298 |
+
"created_at": dp.get("__created_at__"),
|
299 |
+
}
|
300 |
return None
|
301 |
|
302 |
async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]:
|
|
|
312 |
return []
|
313 |
|
314 |
client = await self._get_client()
|
315 |
+
results = client.get(ids)
|
316 |
+
return [
|
317 |
+
{
|
318 |
+
**dp,
|
319 |
+
"id": dp.get("__id__"),
|
320 |
+
"created_at": dp.get("__created_at__"),
|
321 |
+
}
|
322 |
+
for dp in results
|
323 |
+
]
|
324 |
|
325 |
async def drop(self) -> dict[str, str]:
|
326 |
"""Drop all vector data from storage and clean up resources
|