Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -86,7 +86,7 @@ def answer_query(message, chat_history):
|
|
| 86 |
base_compressor = LLMChainExtractor.from_llm(chat)
|
| 87 |
|
| 88 |
db = Chroma(persist_directory = "./slide_chromaDB4", embedding_function=embedding_function, collection_name="slideCollection")
|
| 89 |
-
base_retriever = db.as_retriever()
|
| 90 |
mq_retriever = MultiQueryRetriever.from_llm(retriever = base_retriever, llm=chat)
|
| 91 |
compression_retriever = ContextualCompressionRetriever(base_compressor=base_compressor, base_retriever=mq_retriever)
|
| 92 |
|
|
|
|
| 86 |
base_compressor = LLMChainExtractor.from_llm(chat)
|
| 87 |
|
| 88 |
db = Chroma(persist_directory = "./slide_chromaDB4", embedding_function=embedding_function, collection_name="slideCollection")
|
| 89 |
+
base_retriever = db.as_retriever(search_type="similarity_score_threshold",search_kwargs={'score_threshold': 0.2})
|
| 90 |
mq_retriever = MultiQueryRetriever.from_llm(retriever = base_retriever, llm=chat)
|
| 91 |
compression_retriever = ContextualCompressionRetriever(base_compressor=base_compressor, base_retriever=mq_retriever)
|
| 92 |
|