He Tao
14 lines
456 B
Python
14 lines
456 B
Python
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
|
|
}
|