ale979 commited on
Commit
117bf69
·
verified ·
1 Parent(s): 2411c1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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