Ludovicollin commited on
Commit
bf7110c
·
1 Parent(s): ba4ef81

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -37,7 +37,7 @@ pinecone.init(
37
  )
38
 
39
  @cl.cache
40
- def retriever_to_cache(index_name, embeddings):
41
  vectorstore = Pinecone.from_existing_index(
42
  index_name=index_name, embedding=embeddings
43
  )
@@ -170,7 +170,7 @@ async def start():
170
  )
171
  #question_generator = LLMChain(llm=llm, prompt=CONDENSE_QUESTION_PROMPT)
172
  #doc_chain = load_qa_chain(streaming_llm, chain_type="stuff")
173
- relevant=retriever_to_cache(index_name, embeddings)
174
  qa = ConversationalRetrievalChain.from_llm(
175
  streaming_llm,
176
  chain_type="stuff",
 
37
  )
38
 
39
  @cl.cache
40
+ def retriever_to_cache():
41
  vectorstore = Pinecone.from_existing_index(
42
  index_name=index_name, embedding=embeddings
43
  )
 
170
  )
171
  #question_generator = LLMChain(llm=llm, prompt=CONDENSE_QUESTION_PROMPT)
172
  #doc_chain = load_qa_chain(streaming_llm, chain_type="stuff")
173
+ relevant=retriever_to_cache()
174
  qa = ConversationalRetrievalChain.from_llm(
175
  streaming_llm,
176
  chain_type="stuff",