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

chore: add max_retries=5 for 4o-mini Aggregate tool

Browse files
Files changed (1) hide show
  1. tools.py +1 -1
tools.py CHANGED
@@ -145,7 +145,7 @@ def aggregate_information(query: str, results: list[str]) -> str:
145
  # Convert to LangChain Document objects
146
  docs = [Document(page_content=chunk) for chunk in results]
147
 
148
- llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
149
 
150
  # 🔍 Map Prompt — Summarize each document in light of the query
151
  map_prompt = PromptTemplate.from_template(
 
145
  # Convert to LangChain Document objects
146
  docs = [Document(page_content=chunk) for chunk in results]
147
 
148
+ llm = ChatOpenAI(model="gpt-4o-mini", temperature=0, max_retries=5)
149
 
150
  # 🔍 Map Prompt — Summarize each document in light of the query
151
  map_prompt = PromptTemplate.from_template(