CoralLeiCN
commited on
Commit
·
04e6f8d
1
Parent(s):
8127077
Refactor BasicAgent to allow model-determined thinking budget
Browse files- agent/agents.py +2 -3
agent/agents.py
CHANGED
@@ -95,9 +95,8 @@ class BasicAgent:
|
|
95 |
if model.startswith("gemini-2.0-flash"):
|
96 |
thinking_config = None
|
97 |
else:
|
98 |
-
|
99 |
-
|
100 |
-
)
|
101 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
102 |
# Run the agent to find the best catering service
|
103 |
|
|
|
95 |
if model.startswith("gemini-2.0-flash"):
|
96 |
thinking_config = None
|
97 |
else:
|
98 |
+
# let model decide the budget
|
99 |
+
thinking_config = ThinkingConfig()
|
|
|
100 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
101 |
# Run the agent to find the best catering service
|
102 |
|