22 lines
551 B
Python
22 lines
551 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", "code"]
|
|
|
|
# Define agent-LLM mapping
|
|
AGENT_LLM_MAP: dict[str, LLMType] = {
|
|
"coordinator": "basic",
|
|
"planner": "basic",
|
|
"researcher": "basic",
|
|
"analyst": "basic",
|
|
"coder": "basic",
|
|
"reporter": "basic",
|
|
"podcast_script_writer": "basic",
|
|
"ppt_composer": "basic",
|
|
"prose_writer": "basic",
|
|
"prompt_enhancer": "basic",
|
|
}
|