adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[modify] multi agent model parameter

Browse Source
This commit is contained in:
Mark
2026-01-05 11:49:01 +08:00
parent 3fe2ef6611
commit 393fbee551
3 changed files with 52 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
api/app/services/master_agent_router.py
View File

@@ -366,10 +366,26 @@ class MasterAgentRouter:
# if self.model_parameters:
# temperature = self.model_parameters["temperature"]
# max_tokens = self.model_parameters["max_tokens"]
extra_params = {"temperature": self.model_parameters.get("temperature", 0.3),
"max_tokens":self.model_parameters.get("max_tokens", 1000)
}
if self.model_parameters:
if hasattr(self.model_parameters, 'temperature'):
# Pydantic 模型
temperature = self.model_parameters.temperature
max_tokens = getattr(self.model_parameters, 'max_tokens', 1000)
elif isinstance(self.model_parameters, dict):
# 字典
temperature = self.model_parameters.get("temperature", 0.3)
max_tokens = self.model_parameters.get("max_tokens", 1000)
else:
temperature = 0.3
max_tokens = 1000
else:
temperature = 0.3
max_tokens = 1000
# extra_params = {"temperature": self.model_parameters.get("temperature", 0.3),
# "max_tokens":self.model_parameters.get("max_tokens", 1000)
# }
extra_params = {"temperature": temperature, "max_tokens": max_tokens}
# 创建 RedBearModelConfig
model_config = RedBearModelConfig(
model_name=api_key_config.model_name,