pikamomo commited on
Commit
8a70dca
·
1 Parent(s): c32cdfb

retrieve more file

Browse files
Files changed (1) hide show
  1. src/chatbot.py +2 -2
src/chatbot.py CHANGED
@@ -80,12 +80,12 @@ def create_rag_chain():
80
  # 2. Create retriever
81
  retriever = vectorstore.as_retriever(
82
  search_type="similarity",
83
- search_kwargs={"k": 5}
84
  )
85
 
86
  # 3. Create LLM
87
  llm = ChatOpenAI(
88
- model=os.getenv("OPEN_AI_CHAT_MODEL", "gpt-4o-mini"),
89
  temperature=0.3
90
  )
91
 
 
80
  # 2. Create retriever
81
  retriever = vectorstore.as_retriever(
82
  search_type="similarity",
83
+ search_kwargs={"k": 8}
84
  )
85
 
86
  # 3. Create LLM
87
  llm = ChatOpenAI(
88
+ model=os.getenv("OPEN_AI_CHAT_MODEL", "gpt-4o"),
89
  temperature=0.3
90
  )
91