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

@@ -163,7 +163,7 @@ class ElasticsearchDB(BaseVectorDB):
def query(
self,
input_query: list[str],
input_query: str,
n_results: int,
where: dict[str, any],
citations: bool = False,
@@ -172,8 +172,8 @@ class ElasticsearchDB(BaseVectorDB):
"""
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
@@ -185,7 +185,7 @@ class ElasticsearchDB(BaseVectorDB):
along with url of the source and doc_id (if citations flag is true)
:rtype: list[str], if citations=False, otherwise list[tuple[str, str, str]]
"""
input_query_vector = self.embedder.embedding_fn(input_query)
input_query_vector = self.embedder.embedding_fn([input_query])
query_vector = input_query_vector[0]
# `https://www.elastic.co/guide/en/elasticsearch/reference/7.17/query-dsl-script-score-query.html`