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,

9
api/app/services/multi_agent_service.py
View File

@@ -293,12 +293,13 @@ class MultiAgentService:
# 处理 model_parameters（可能是 None、字典或 Pydantic 模型）
if data.model_parameters is None:
model_parameters_data = None
elif isinstance(data.model_parameters, dict):
# 过滤掉值为 None 的字段
model_parameters_data = {k: v for k, v in data.model_parameters.items() if v is not None}
# elif isinstance(data.model_parameters, dict):
# # 过滤掉值为 None 的字段
# model_parameters_data = {k: v for k, v in data.model_parameters.items() if v is not None}
else:
# 过滤掉值为 None 的字段
model_parameters_data = {k: v for k, v in data.model_parameters.model_dump().items() if v is not None}
# model_parameters_data = {k: v for k, v in data.model_parameters.model_dump().items() if v is not None}
model_parameters_data = data.model_parameters
config = MultiAgentConfig(
app_id=app_id,