Pijush2023 commited on
Commit
28ed422
·
verified ·
1 Parent(s): 4f78c42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -40,6 +40,10 @@ gpt_vectorstore = PineconeVectorStore(index_name="radardata11122024", embedding=
40
  gpt_retriever = gpt_vectorstore.as_retriever(search_kwargs={'k': 2})
41
 
42
 
 
 
 
 
43
 
44
  # Pinecone setup
45
  from pinecone import Pinecone
 
40
  gpt_retriever = gpt_vectorstore.as_retriever(search_kwargs={'k': 2})
41
 
42
 
43
+ async def async_get_retriever_result(query, retriever, k=2):
44
+ search_kwargs = {'k': k}
45
+ results = await retriever.as_retriever(search_kwargs=search_kwargs)(query)
46
+ return results
47
 
48
  # Pinecone setup
49
  from pinecone import Pinecone