civerson916 commited on
Commit
89326bd
·
verified ·
1 Parent(s): bbccf68

Using o4-mini for notation conversion

Browse files
Files changed (1) hide show
  1. agent.py +1 -8
agent.py CHANGED
@@ -49,9 +49,6 @@ class ResearchAgent:
49
  verbosity_level=1,
50
  model=LiteLLMModel(
51
  model_id=OpenRouterModelID.GPT_O4_MINI,
52
- # model_id=OpenRouterModelID.GPT_4_1_MINI,
53
- # model_id=OpenRouterModelID.GROK_3_BETA,
54
- # model_id=OpenRouterModelID.GROK_3_MINI_BETA,
55
  api_key = settings.openrouter_api_key.get_secret_value(),
56
  temperature=0.0, timeout=180
57
  )
@@ -65,7 +62,7 @@ class ChessAgent:
65
  add_base_tools=False,
66
  tools=[ChessBoardFENTool(),
67
  BestChessMoveTool(settings),
68
- ConvertChessMoveTool(settings, OpenRouterModelID.QWEN_3_14B_FREE),
69
  ],
70
  additional_authorized_imports=[
71
  "unicodedata",
@@ -99,10 +96,6 @@ class ManagerAgent:
99
  tools=[GetTaskFileTool(settings), FinalAnswerTool()],
100
  model=LiteLLMModel(
101
  model_id=OpenRouterModelID.GPT_O4_MINI,
102
- # model_id=OpenRouterModelID.GPT_4_1_MINI,
103
- # model_id=OpenRouterModelID.GROK_3_BETA,
104
- # model_id=OpenRouterModelID.GROK_3_MINI_BETA,
105
- # model_id=OpenRouterModelID.QWEN_3_14B_FREE,
106
  api_key = settings.openrouter_api_key.get_secret_value(),
107
  temperature=0.0, timeout=180
108
  ),
 
49
  verbosity_level=1,
50
  model=LiteLLMModel(
51
  model_id=OpenRouterModelID.GPT_O4_MINI,
 
 
 
52
  api_key = settings.openrouter_api_key.get_secret_value(),
53
  temperature=0.0, timeout=180
54
  )
 
62
  add_base_tools=False,
63
  tools=[ChessBoardFENTool(),
64
  BestChessMoveTool(settings),
65
+ ConvertChessMoveTool(settings, OpenRouterModelID.GPT_O4_MINI),
66
  ],
67
  additional_authorized_imports=[
68
  "unicodedata",
 
96
  tools=[GetTaskFileTool(settings), FinalAnswerTool()],
97
  model=LiteLLMModel(
98
  model_id=OpenRouterModelID.GPT_O4_MINI,
 
 
 
 
99
  api_key = settings.openrouter_api_key.get_secret_value(),
100
  temperature=0.0, timeout=180
101
  ),