Spaces:
Sleeping
Sleeping
Update planner_agent.py
Browse files- 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.
|
| 9 |
|
| 10 |
|
| 11 |
-
HOW_MANY_SEARCHES =
|
| 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."
|