CoralLeiCN
commited on
Commit
·
d564629
1
Parent(s):
c1e5656
Refactor BasicAgent: add top_p and seed parameters for enhanced model configuration
Browse files- 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}")
|