fix: bug #370, support temperature configuration

This commit is contained in:
lemon 2026-02-12 19:12:38 +08:00
parent a05e58cf79
commit a3599b97b9
2 changed files with 8 additions and 2 deletions

View File

@ -41,6 +41,7 @@ class AgentLoop:
workspace: Path,
model: str | None = None,
max_iterations: int = 20,
temperature: float = 0.7,
brave_api_key: str | None = None,
exec_config: "ExecToolConfig | None" = None,
cron_service: "CronService | None" = None,
@ -54,6 +55,7 @@ class AgentLoop:
self.workspace = workspace
self.model = model or provider.get_default_model()
self.max_iterations = max_iterations
self.temperature = temperature
self.brave_api_key = brave_api_key
self.exec_config = exec_config or ExecToolConfig()
self.cron_service = cron_service
@ -195,7 +197,8 @@ class AgentLoop:
response = await self.provider.chat(
messages=messages,
tools=self.tools.get_definitions(),
model=self.model
model=self.model,
temperature=self.temperature
)
# Handle tool calls
@ -305,7 +308,8 @@ class AgentLoop:
response = await self.provider.chat(
messages=messages,
tools=self.tools.get_definitions(),
model=self.model
model=self.model,
temperature=self.temperature
)
if response.has_tool_calls:

View File

@ -323,6 +323,7 @@ def gateway(
provider=provider,
workspace=config.workspace_path,
model=config.agents.defaults.model,
temperature=config.agents.defaults.temperature,
max_iterations=config.agents.defaults.max_tool_iterations,
brave_api_key=config.tools.web.search.api_key or None,
exec_config=config.tools.exec,
@ -428,6 +429,7 @@ def agent(
bus=bus,
provider=provider,
workspace=config.workspace_path,
temperature=config.agents.defaults.temperature,
brave_api_key=config.tools.web.search.api_key or None,
exec_config=config.tools.exec,
restrict_to_workspace=config.tools.restrict_to_workspace,