ranranrunforit commited on
Commit
30f3f8c
·
verified ·
1 Parent(s): ff39971

Update planner_agent.py

Browse files
Files changed (1) hide show
  1. planner_agent.py +2 -2
planner_agent.py CHANGED
@@ -5,10 +5,10 @@ import os
5
 
6
  gemini_client = AsyncOpenAI(base_url="https://generativelanguage.googleapis.com/v1beta/openai/", api_key=os.getenv('GOOGLE_API_KEY'))
7
 
8
- gemini_model = OpenAIChatCompletionsModel(model="gemini-2.0-flash", openai_client=gemini_client)
9
 
10
 
11
- HOW_MANY_SEARCHES = 5
12
 
13
  INSTRUCTIONS = f"You are a helpful research assistant. Given a query, come up with a set of web searches \
14
  to perform to best answer the query. Output {HOW_MANY_SEARCHES} terms to query for."
 
5
 
6
  gemini_client = AsyncOpenAI(base_url="https://generativelanguage.googleapis.com/v1beta/openai/", api_key=os.getenv('GOOGLE_API_KEY'))
7
 
8
+ gemini_model = OpenAIChatCompletionsModel(model="gemini-2.5-flash-preview-05-20", openai_client=gemini_client)
9
 
10
 
11
+ HOW_MANY_SEARCHES = 10
12
 
13
  INSTRUCTIONS = f"You are a helpful research assistant. Given a query, come up with a set of web searches \
14
  to perform to best answer the query. Output {HOW_MANY_SEARCHES} terms to query for."