raghuv-aditya commited on
Commit
09474c3
·
verified ·
1 Parent(s): ce728b2

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -1
chatbot.py CHANGED
@@ -16,7 +16,7 @@ def create_chatbot(vector_store):
16
  Returns:
17
  RetrievalQA: A retrieval-based QA system.
18
  """
19
- llm = OpenAI(temperature=0.5)
20
  retriever = vector_store.as_retriever(search_type="mmr", k=3)
21
 
22
  qa = RetrievalQA.from_chain_type(
 
16
  Returns:
17
  RetrievalQA: A retrieval-based QA system.
18
  """
19
+ llm = OpenAI(model="4o mini", temperature=0.5)
20
  retriever = vector_store.as_retriever(search_type="mmr", k=3)
21
 
22
  qa = RetrievalQA.from_chain_type(