mirror of
https://github.com/agentuniverse-ai/agentUniverse.git
synced 2026-02-09 01:59:19 +08:00
36 lines
2.6 KiB
Python
36 lines
2.6 KiB
Python
# !/usr/bin/env python3
|
|
# -*- coding:utf-8 -*-
|
|
|
|
# @Time : 2024/7/26 10:10
|
|
# @Author : wangchongshi
|
|
# @Email : wangchongshi.wcs@antgroup.com
|
|
# @FileName: llm_constant.py
|
|
# llm model name list
|
|
LLM_MODEL_NAME = {
|
|
'demo_llm': ['gpt-4o', 'gpt-4o-mini', 'o1', 'o3-mini', 'qwen-max', 'qwen-long', 'qwen-plus',
|
|
'qwen-turbo', 'qwen2.5-72b-instruct', 'qwen2.5-7b-instruct'],
|
|
'openai_llm': ['gpt-3.5-turbo', 'gpt-4-turbo', 'gpt-4o', 'gpt-4o-mini', 'o1', 'o1-mini', 'o3-mini'],
|
|
'default_openai_llm': ['gpt-3.5-turbo', 'gpt-4-turbo', 'gpt-4o', 'gpt-4o-mini', 'o1', 'o1-mini', 'o3-mini'],
|
|
'qwen_llm': ['qwen-max', 'qwen-long', 'qwen-plus', 'qwen-turbo', 'qwen-coder-plus',
|
|
'qwen2.5-72b-instruct', 'qwen2.5-32b-instruct', 'qwen2.5-14b-instruct', 'qwen2.5-7b-instruct'],
|
|
'default_qwen_llm': ['qwen-max', 'qwen-long', 'qwen-plus', 'qwen-turbo', 'qwen-coder-plus',
|
|
'qwen2.5-72b-instruct', 'qwen2.5-32b-instruct', 'qwen2.5-14b-instruct', 'qwen2.5-7b-instruct'],
|
|
'wenxin_llm': ['ERNIE-Speed-AppBuilder-8K-0516', 'ERNIE-Lite-8K-0725', 'ERNIE-Speed-128K', 'ERNIE-3.5-128K',
|
|
'ERNIE-3.5-8K-0701', 'ERNIE-4.0-8K-0613', 'ERNIE-4.0-8K-Preview', 'ERNIE-3.5-8K-Preview',
|
|
'ERNIE-Tiny-8K', 'ERNIE-4.0-8K-Latest', 'ERNIE-4.0-Turbo-8K','ERNIE-4.5-8K-Preview',
|
|
'ERNIE-4.5-Turbo-32K', 'ERNIE-4.5-Turbo-128K'],
|
|
'default_wenxin_llm': ['ERNIE-Speed-AppBuilder-8K-0516', 'ERNIE-Lite-8K-0725', 'ERNIE-Speed-128K', 'ERNIE-3.5-128K',
|
|
'ERNIE-3.5-8K-0701', 'ERNIE-4.0-8K-0613', 'ERNIE-4.0-8K-Preview', 'ERNIE-3.5-8K-Preview',
|
|
'ERNIE-Tiny-8K', 'ERNIE-4.0-8K-Latest', 'ERNIE-4.0-Turbo-8K','ERNIE-4.5-8K-Preview',
|
|
'ERNIE-4.5-Turbo-32K', 'ERNIE-4.5-Turbo-128K'],
|
|
'kimi_llm': ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
|
|
'default_kimi_llm': ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
|
|
'deepseek_llm': ['deepseek-chat', 'deepseek-coder', 'deepseek-reasoner', 'deepseek-v3', 'deepseek-r1',
|
|
'DeepSeek-R1', 'DeepSeek-V3'],
|
|
'default_deepseek_llm': ['deepseek-chat', 'deepseek-coder', 'deepseek-reasoner', 'deepseek-v3', 'deepseek-r1',
|
|
'DeepSeek-R1', 'DeepSeek-V3'],
|
|
'baichuan_llm': ['Baichuan4', 'Baichuan3-Turbo', 'Baichuan3-Turbo-128k', 'Baichuan2-Turbo', 'Baichuan2-Turbo-192k'],
|
|
'default_baichuan_llm': ['Baichuan4', 'Baichuan3-Turbo', 'Baichuan3-Turbo-128k', 'Baichuan2-Turbo',
|
|
'Baichuan2-Turbo-192k'],
|
|
}
|