Change list[str] -> str for vectordbs (#1388)

This commit is contained in:
Anu
2024-06-07 10:15:40 -06:00
committed by GitHub
parent fd07513004
commit e0d0f6e94c
11 changed files with 27 additions and 27 deletions

View File

@@ -219,12 +219,12 @@ class WeaviateDB(BaseVectorDB):
)
def query(
self, input_query: list[str], n_results: int, where: dict[str, any], citations: bool = False
self, input_query: str, n_results: int, where: dict[str, any], citations: bool = False
) -> Union[list[tuple[str, dict]], list[str]]:
"""
query contents from vector database based on vector similarity
:param input_query: list of query string
:type input_query: list[str]
:param input_query: query string
:type input_query: str
:param n_results: no of similar documents to fetch from database
:type n_results: int
:param where: Optional. to filter data