SuperAPI AI 工具集

update-llm-model-list

智能体与模型

查看项目

更新于 10-15 16:42

技能介绍

用途:审核并更新 Agenta 中受支持的 LLM 模型列表,使其与 litellm 的注册表(models.litellm.ai)保持一致。规范列表位于 sdks/python/agenta/sdk/utils/assets.py 的 supported_llm_models,它驱动 playground 的模型下拉框、成本元数据以及 model_to_provider_mapping;外部权威来源为 litellm.model_cost(2600+ 条目)。

主要能力与规则:每个模型必须存在于 litellm.model_cost(直接键名,或去掉 provider 前缀后匹配);anthropic/claude-* 在 litellm 中存为 claude-,cohere/command- 存为 command-*,其他 provider 保留完整前缀(如 gemini/、groq/、together_ai/);provider 键须与 api/oss/src/core/secrets/enums.py 中的 Secrets API 枚举(StandardProviderKind)一致;同一 provider 列表内不得重复。pytest 守卫位于 sdks/python/oss/tests/pytest/unit/test_supported_llm_models.py。

使用方式:第一步检查过时或错误的模型,可运行 uvx --with litellm python /tmp/check_agenta_models.py,或直接执行 pytest sdks/python/oss/tests/pytest/unit/test_supported_llm_models.py -v。第二步用 /tmp/find_missing.py 找出 Agenta 缺失的模型(过滤音频、视频、嵌入、codex、快照等噪声)。第三步编辑 assets.py,按 provider 正确列表、新模型在前添加;Gemini 1.5 加到 gemini;OpenAI o 系列 pro 层加在基础模型之后;Groq 需核对 litellm.groq_models;DeepInfra/Together AI 核对 litellm.deepinfra_models / litellm.together_ai_models。第四步运行 ruff 格式化与检查:uvx --from ruff==0.14.0 ruff format sdks/python/agenta/sdk/utils/assets.py 与 uvx --from ruff==0.14.0 ruff check --fix sdks/python/agenta/sdk/utils/assets.py,再重跑校验脚本,全部通过后方可提交。

相关文件:assets.py(规范列表与成本元数据构建器)、test_supported_llm_models.py(pytest 守卫)、enums.py(provider 键需同步)、api/oss/src/resources/evaluators/evaluators.py(评估器下拉框的独立较短列表)。限制:脚本需 Python >=3.11 与 litellm,部分脚本需先手动从 assets.py 填入 AGENTA_* 集合;评测器下拉框为独立列表,本技能未说明其同步方式。