From 5c1fa3e70e2e0ae1d73f80a191ee0ee1a6214a5a Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sun, 18 Jun 2023 18:33:15 -0700 Subject: [PATCH] Harrison/typesense fix (#6391) Co-authored-by: Gaurav Chauhan <2796gaurav@gmail.com> Co-authored-by: gaurav --- langchain/vectorstores/typesense.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/langchain/vectorstores/typesense.py b/langchain/vectorstores/typesense.py index 63ddab74..1b89a4dc 100644 --- a/langchain/vectorstores/typesense.py +++ b/langchain/vectorstores/typesense.py @@ -42,10 +42,10 @@ class Typesense(VectorStore): embedding = OpenAIEmbeddings() vectorstore = Typesense( - typesense_client, - typesense_collection_name, - embedding.embed_query, - "text", + typesense_client=typesense_client, + embedding=embedding, + typesense_collection_name=typesense_collection_name, + text_key="text", ) """ @@ -138,14 +138,15 @@ class Typesense(VectorStore): def similarity_search_with_score( self, query: str, - k: int = 4, + k: int = 10, filter: Optional[str] = "", ) -> List[Tuple[Document, float]]: """Return typesense documents most similar to query, along with scores. Args: query: Text to look up documents similar to. - k: Number of Documents to return. Defaults to 4. + k: Number of Documents to return. Defaults to 10. + Minimum 10 results would be returned. filter: typesense filter_by expression to filter documents on Returns: @@ -173,7 +174,7 @@ class Typesense(VectorStore): def similarity_search( self, query: str, - k: int = 4, + k: int = 10, filter: Optional[str] = "", **kwargs: Any, ) -> List[Document]: @@ -181,7 +182,8 @@ class Typesense(VectorStore): Args: query: Text to look up documents similar to. - k: Number of Documents to return. Defaults to 4. + k: Number of Documents to return. Defaults to 10. + Minimum 10 results would be returned. filter: typesense filter_by expression to filter documents on Returns: