cleaned base and added missing status

This commit is contained in:
Yannick Stephan
2025-02-08 22:57:37 +01:00
parent 7f407ce949
commit ff0bd1512e

View File

@@ -56,7 +56,7 @@ class QueryParam:
@dataclass @dataclass
class StorageNameSpace: class StorageNameSpace:
namespace: str namespace: str
global_config: dict global_config: dict[str, Any]
async def index_done_callback(self): async def index_done_callback(self):
"""commit the storage operations after indexing""" """commit the storage operations after indexing"""
@@ -92,21 +92,23 @@ class BaseKVStorage(Generic[T], StorageNameSpace):
async def get_by_id(self, id: str) -> Union[T, None]: async def get_by_id(self, id: str) -> Union[T, None]:
raise NotImplementedError raise NotImplementedError
async def get_by_ids( async def get_by_ids(self, ids: list[str]) -> list[Union[T, None]]:
self, ids: list[str], fields: Union[set[str], None] = None
) -> list[Union[T, None]]:
raise NotImplementedError raise NotImplementedError
async def filter_keys(self, data: list[str]) -> set[str]: async def filter_keys(self, data: list[str]) -> set[str]:
"""return un-exist keys""" """return un-exist keys"""
raise NotImplementedError raise NotImplementedError
async def upsert(self, data: dict[str, T]): async def upsert(self, data: dict[str, T]) -> None:
raise NotImplementedError raise NotImplementedError
async def drop(self): async def drop(self) -> None:
raise NotImplementedError raise NotImplementedError
async def get_by_status_and_ids(
self, status: str, ids: list[str]
) -> list[dict[str, Any]]:
raise NotImplementedError
@dataclass @dataclass
class BaseGraphStorage(StorageNameSpace): class BaseGraphStorage(StorageNameSpace):