vrr/eigent
2
0
Fork 0
mirror of https://github.com/eigent-ai/eigent.git synced 2026-05-08 18:30:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

enhance: simplify loop

Browse source
This commit is contained in:
a7m-1st 2026-01-16 14:18:40 +03:00
parent 43e597298c
commit 01eecf3aa5
1 changed files with 26 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

45
backend/app/utils/agent.py
View file

@ -624,35 +624,42 @@ def agent_model(
"async_client",
"azure_deployment_name",
}
init_params = {
k: v
for k, v in extra_params.items()
if k in init_param_keys and v is not None and (not isinstance(v, str) or v.strip() != "")
}
init_params = {}
model_config: dict[str, Any] = {}
if options.is_cloud():
model_config["user"] = str(options.project_id)
model_config.update(
{
k: v
for k, v in extra_params.items()
if k not in init_param_keys and k not in ["model_platform", "model_type", "api_key", "url"]
}
)
excluded_keys = {"model_platform", "model_type", "api_key", "url"}
# Distribute extra_params between init_params and model_config
for k, v in extra_params.items():
if k in excluded_keys:
continue
# Skip empty values
if v is None or (isinstance(v, str) and not v.strip()):
continue
if k in init_param_keys:
init_params[k] = v
else:
model_config[k] = v
if agent_name == Agents.task_agent:
model_config["stream"] = True
if agent_name == Agents.search_agent:
try:
model_platform_enum = ModelPlatformType(options.model_platform.lower())
if model_platform_enum in {
ModelPlatformType.OPENAI,
ModelPlatformType.AZURE,
ModelPlatformType.AIHUBMIX,
}:
model_config["parallel_tool_calls"] = False
except (ValueError, AttributeError):
traceroot_logger.error(f"Invalid model platform for search agent: {options.model_platform}", exc_info=True)
model_platform_enum = None
if model_platform_enum in {
ModelPlatformType.OPENAI,
ModelPlatformType.AZURE,
ModelPlatformType.AIHUBMIX,
}:
model_config["parallel_tool_calls"] = False
return ListenChatAgent(
options.project_id,