From 9930a8358762ec854945500bbacae7108eea3e95 Mon Sep 17 00:00:00 2001 From: Daniel O'Connell Date: Wed, 1 Nov 2023 12:02:13 +0100 Subject: [PATCH] pr fix --- api/src/stampy_chat/citations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/src/stampy_chat/citations.py b/api/src/stampy_chat/citations.py index 34740e4..8c9be85 100644 --- a/api/src/stampy_chat/citations.py +++ b/api/src/stampy_chat/citations.py @@ -76,10 +76,10 @@ class ReferencesSelector(SemanticSimilarityExampleSelector): return examples -def make_example_selector(k: int, **params) -> ReferencesSelector: +def make_example_selector(**params) -> ReferencesSelector: embeddings = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY) vectorstore = Pinecone(PINECONE_INDEX, embeddings.embed_query, "hash_id", namespace=PINECONE_NAMESPACE) - return ReferencesSelector(vectorstore=vectorstore, k=k, **params) + return ReferencesSelector(vectorstore=vectorstore, **params) def format_block(block) -> Dict[str, Any]: