xicocdi commited on
Commit
a49221e
·
1 Parent(s): d33b898

remove multiquery_retriever

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -12,7 +12,6 @@ from langchain.chains import ConversationalRetrievalChain
12
  from langchain_community.vectorstores import Qdrant
13
  from langchain.memory import ConversationBufferMemory
14
  from langchain_huggingface import HuggingFaceEmbeddings
15
- from langchain.retrievers.multi_query import MultiQueryRetriever
16
 
17
  import chainlit as cl
18
 
@@ -80,11 +79,6 @@ llm = ChatOpenAI(
80
  streaming=True,
81
  )
82
 
83
- retriever_llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
84
- multiquery_retriever = MultiQueryRetriever.from_llm(
85
- retriever=retriever, llm=retriever_llm
86
- )
87
-
88
 
89
  @cl.on_chat_start
90
  async def start_chat():
@@ -94,7 +88,7 @@ async def start_chat():
94
 
95
  qa = ConversationalRetrievalChain.from_llm(
96
  llm,
97
- retriever=multiquery_retriever,
98
  memory=memory,
99
  combine_docs_chain_kwargs={"prompt": PROMPT},
100
  return_source_documents=True,
 
12
  from langchain_community.vectorstores import Qdrant
13
  from langchain.memory import ConversationBufferMemory
14
  from langchain_huggingface import HuggingFaceEmbeddings
 
15
 
16
  import chainlit as cl
17
 
 
79
  streaming=True,
80
  )
81
 
 
 
 
 
 
82
 
83
  @cl.on_chat_start
84
  async def start_chat():
 
88
 
89
  qa = ConversationalRetrievalChain.from_llm(
90
  llm,
91
+ retriever=retriever,
92
  memory=memory,
93
  combine_docs_chain_kwargs={"prompt": PROMPT},
94
  return_source_documents=True,