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

fixed ids

Browse Source
This commit is contained in:
Yannick Stephan
2025-02-09 21:24:13 +01:00
parent f37f2c7555
commit 9f6621454e
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lightrag/kg/json_kv_impl.py
View File

@@ -39,7 +39,7 @@ class JsonKVStorage(BaseKVStorage):
] ]
async def filter_keys(self, data: set[str]) -> set[str]: async def filter_keys(self, data: set[str]) -> set[str]:
return set(self._data.keys()).difference(data) return set(data) - set(self._data.keys())
async def upsert(self, data: dict[str, dict[str, Any]]) -> None: async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
left_data = {k: v for k, v in data.items() if k not in self._data} left_data = {k: v for k, v in data.items() if k not in self._data}

4
lightrag/kg/jsondocstatus_impl.py
View File

@@ -48,8 +48,8 @@ Usage:
""" """
import os
from dataclasses import dataclass from dataclasses import dataclass
import os
from typing import Any, Union from typing import Any, Union
from lightrag.base import ( from lightrag.base import (
@@ -76,7 +76,7 @@ class JsonDocStatusStorage(DocStatusStorage):
async def filter_keys(self, data: set[str]) -> set[str]: async def filter_keys(self, data: set[str]) -> set[str]:
"""Return keys that should be processed (not in storage or not successfully processed)""" """Return keys that should be processed (not in storage or not successfully processed)"""
return set(self._data.keys()).difference(data) return set(data) - set(self._data.keys())
async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]: async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]:
result: list[dict[str, Any]] = [] result: list[dict[str, Any]] = []