CoralLeiCN commited on
Commit
d564629
·
1 Parent(s): c1e5656

Refactor BasicAgent: add top_p and seed parameters for enhanced model configuration

Browse files
Files changed (1) hide show
  1. agent/agents.py +3 -1
agent/agents.py CHANGED
@@ -8,6 +8,7 @@ from smolagents import CodeAgent, VisitWebpageTool, WebSearchTool
8
 
9
  from agent.prompts import formatter_system_prompt, system_prompt
10
  from agent.tools import (
 
11
  CodeExecutionTool,
12
  DownloadFile,
13
  ReadExcelFileBytes,
@@ -15,7 +16,6 @@ from agent.tools import (
15
  TranscribeYoutubeVideo,
16
  UnderstandImageBytes,
17
  WikipediaSearchTool,
18
- ChessBestMove,
19
  )
20
  from agent.utils import gemini_client, gemini_model_liteLLM
21
 
@@ -115,6 +115,8 @@ class BasicAgent:
115
  temperature=0.0,
116
  response_schema=final_answer,
117
  thinking_config=thinking_config,
 
 
118
  ),
119
  )
120
  print(f"Final answer after formatter by model: {response.parsed.answer}")
 
8
 
9
  from agent.prompts import formatter_system_prompt, system_prompt
10
  from agent.tools import (
11
+ ChessBestMove,
12
  CodeExecutionTool,
13
  DownloadFile,
14
  ReadExcelFileBytes,
 
16
  TranscribeYoutubeVideo,
17
  UnderstandImageBytes,
18
  WikipediaSearchTool,
 
19
  )
20
  from agent.utils import gemini_client, gemini_model_liteLLM
21
 
 
115
  temperature=0.0,
116
  response_schema=final_answer,
117
  thinking_config=thinking_config,
118
+ top_p=0.95,
119
+ seed=42,
120
  ),
121
  )
122
  print(f"Final answer after formatter by model: {response.parsed.answer}")