gzdaniel commited on
Commit
0df4f46
·
1 Parent(s): 79dbcba

Fix time handle bugs for nano vector db

Browse files
Files changed (1) hide show
  1. lightrag/kg/nano_vector_db_impl.py +15 -2
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
- return result[0]
 
 
 
 
 
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
- return client.get(ids)
 
 
 
 
 
 
 
 
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