wassname/stampy-chat
1
0
Fork 0
mirror of https://github.com/wassname/stampy-chat.git synced 2026-09-11 12:50:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

pr fix

Browse Source
This commit is contained in:
Daniel O'Connell
2023-11-01 12:02:13 +01:00
parent b05d743f65
commit 9930a83587
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/src/stampy_chat/citations.py
+2 -2
View File
@@ -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]: