Spaces:
Sleeping
Sleeping
chore: add max_retries=5 for 4o-mini Final Answer
Browse files- 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. "
|