From af477e8a262f201726c2b51a8b2f42bf848950f5 Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Sun, 9 Feb 2025 11:00:04 +0100 Subject: [PATCH] cleaned typing --- lightrag/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lightrag/base.py b/lightrag/base.py index 3fe58a77..a91595b2 100644 --- a/lightrag/base.py +++ b/lightrag/base.py @@ -66,10 +66,10 @@ class BaseVectorStorage(StorageNameSpace): embedding_func: EmbeddingFunc meta_fields: set = field(default_factory=set) - async def query(self, query: str, top_k: int) -> list[dict]: + async def query(self, query: str, top_k: int) -> list[dict[str, Any]]: raise NotImplementedError - async def upsert(self, data: dict[str, dict]): + async def upsert(self, data: dict[str, dict[str, Any]]) -> None: """Use 'content' field from value for embedding, use key as id. If embedding_func is None, use 'embedding' field from value """