typing fixes

microsoft · Oct 24, 2024 · 4b886b0 · 4b886b0
1 parent 347fb60
commit 4b886b0
Show file tree

Hide file tree

Showing 5 changed files with 188 additions and 172 deletions.
diff --git a/python/packages/autogen-ext/src/autogen_ext/storage/_base.py b/python/packages/autogen-ext/src/autogen_ext/storage/_base.py
@@ -156,7 +156,7 @@ async def delete_docs(self, ids: Sequence[ItemID], collection_name: Optional[str
 
     async def retrieve_docs(
         self,
-        queries: Sequence[str],
+        queries: List[str],
         collection_name: Optional[str] = None,
         n_results: int = 10,
         distance_threshold: float = -1,
@@ -183,7 +183,7 @@ async def get_docs_by_ids(
         self,
         ids: Optional[Sequence[ItemID]] = None,
         collection_name: Optional[str] = None,
-        include: Optional[Sequence[str]] = None,
+        include: Optional[List[str]] = None,
         **kwargs: Any,
     ) -> List[Document]:
         """
@@ -324,7 +324,7 @@ def delete_docs(self, ids: Sequence[ItemID], collection_name: Optional[str] = No
 
     def retrieve_docs(
         self,
-        queries: Sequence[str],
+        queries: List[str],
         collection_name: Optional[str] = None,
         n_results: int = 10,
         distance_threshold: float = -1,