lchakkei commited on
Commit
eddf540
1 Parent(s): 5d5df1a

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -59,7 +59,7 @@ class EndpointHandler():
59
  vectorstore = Chroma.from_documents(documents=all_splits, embedding=embedding_function)
60
  retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
61
 
62
- compressor = LLMChainExtractor.from_llm(self.llm)
63
  retriever = ContextualCompressionRetriever(base_compressor=compressor, base_retriever=retriever)
64
 
65
  _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question, in its original language.
 
59
  vectorstore = Chroma.from_documents(documents=all_splits, embedding=embedding_function)
60
  retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
61
 
62
+ compressor = LLMChainExtractor.from_llm(chat)
63
  retriever = ContextualCompressionRetriever(base_compressor=compressor, base_retriever=retriever)
64
 
65
  _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question, in its original language.