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.
|