yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-10-31 09:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed base and generic

Browse Source
This commit is contained in:
Yannick Stephan 2025-02-08 23:16:04 +01:00
parent fe3050adce
commit eb552afcdc
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lightrag/base.py
View File

@ -4,7 +4,6 @@ from typing import (
TypedDict,
Union,
Literal,
Generic,
TypeVar,
Optional,
Dict,
@ -83,30 +82,30 @@ class BaseVectorStorage(StorageNameSpace):
@dataclass
class BaseKVStorage(Generic[T], StorageNameSpace):
class BaseKVStorage(StorageNameSpace):
embedding_func: EmbeddingFunc
async def all_keys(self) -> list[str]:
raise NotImplementedError
async def get_by_id(self, id: str) -> Union[T, None]:
async def get_by_id(self, id: str) -> Union[dict[str, Any], None]:
raise NotImplementedError
async def get_by_ids(self, ids: list[str]) -> list[Union[T, None]]:
async def get_by_ids(self, ids: list[str]) -> list[Union[dict[str, Any], None]]:
raise NotImplementedError
async def filter_keys(self, data: list[str]) -> set[str]:
"""return un-exist keys"""
raise NotImplementedError
async def upsert(self, data: dict[str, T]) -> None:
async def upsert(self, data: dict[str, Any]) -> None:
raise NotImplementedError
async def drop(self) -> None:
raise NotImplementedError
async def get_by_status_and_ids(
self, status: str, ids: list[str]
self, status: str
) -> list[dict[str, Any]]:
raise NotImplementedError