laverdes commited on
Commit
d738df2
·
verified ·
1 Parent(s): 3a1bf60

chore: add max_retries=5 for 4o-mini Final Answer

Browse files
Files changed (1) hide show
  1. basic_agent.py +1 -1
basic_agent.py CHANGED
@@ -80,7 +80,7 @@ def print_conversation(messages):
80
  def generate_final_answer(qa: dict[str, str]) -> str:
81
  """Invokes gpt-4o-mini to extract generate a final answer based on the content query, response, and metadata"""
82
 
83
- final_answer_llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
84
 
85
  system_prompt = (
86
  "You will be given a JSON object containing a user's query, a response from an AI assistant, and optional metadata. "
 
80
  def generate_final_answer(qa: dict[str, str]) -> str:
81
  """Invokes gpt-4o-mini to extract generate a final answer based on the content query, response, and metadata"""
82
 
83
+ final_answer_llm = ChatOpenAI(model="gpt-4o-mini", temperature=0, max_retries=5)
84
 
85
  system_prompt = (
86
  "You will be given a JSON object containing a user's query, a response from an AI assistant, and optional metadata. "