CoralLeiCN commited on
Commit
04e6f8d
·
1 Parent(s): 8127077

Refactor BasicAgent to allow model-determined thinking budget

Browse files
Files changed (1) hide show
  1. 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
- thinking_config = ThinkingConfig(
99
- thinking_budget=1024, # Use `0` to turn off thinking
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