He Tao
17 lines
546 B
Python
17 lines
546 B
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from typing import Literal
|
|
|
|
# Define available LLM types
|
|
LLMType = Literal["basic", "reasoning", "vision"]
|
|
|
|
# Define agent-LLM mapping
|
|
AGENT_LLM_MAP: dict[str, LLMType] = {
|
|
"coordinator": "basic", # 协调默认使用basic llm
|
|
"planner": "basic", # 计划默认使用basic llm
|
|
"researcher": "basic", # 简单搜索任务使用basic llm
|
|
"coder": "basic", # 编程任务使用basic llm
|
|
"reporter": "basic", # 报告使用basic llm
|
|
}
|