Update handler.py
Browse files- 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(
|
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.
|