gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cleaned base and optional unuseful

Browse source
This commit is contained in:
Yannick Stephan 2025-02-09 10:32:59 +01:00
parent 50c7f26262
commit d267f06602
1 changed files with 3 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
lightrag/base.py
View file

@ -5,14 +5,9 @@ from typing import (
Union,
Literal,
TypeVar,
Optional,
Dict,
Any,
List,
)
from enum import Enum
import numpy as np
from .utils import EmbeddingFunc
@ -88,10 +83,10 @@ class BaseKVStorage(StorageNameSpace):
async def all_keys(self) -> list[str]:
raise NotImplementedError
async def get_by_id(self, id: str) -> Union[dict[str, Any], None]:
async def get_by_id(self, id: str) -> dict[str, Any]:
raise NotImplementedError
async def get_by_ids(self, ids: list[str]) -> list[Union[dict[str, Any], None]]:
async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]:
raise NotImplementedError
async def filter_keys(self, data: list[str]) -> set[str]:
@ -104,7 +99,7 @@ class BaseKVStorage(StorageNameSpace):
async def drop(self) -> None:
raise NotImplementedError
async def get_by_status_and_ids(
async def get_by_status(
self, status: str
) -> Union[list[dict[str, Any]], None]:
raise NotImplementedError