Files
ai-agent-server/app/config/ai_configs.py
2026-04-04 20:49:51 +08:00

53 lines
1.8 KiB
Python

from app.config.env import env
ai_configs = {
# /*---------------------------------------local-------------------------------------------*/
"local_glm": {
"model": "glm-4-9b-0414",
"url": "http://127.0.0.1:1234/v1/chat/completions",
"key": env.llm_key_local
},
"local_embedding": {
"model": "text-embedding-text2vec-large-chinese",
"url": "http://127.0.0.1:1234/v1/embeddings",
"key": env.llm_key_local
},
# /*---------------------------------------huoshan-------------------------------------------*/
'huoshan-doubao': {
'model': 'doubao-1-5-lite-32k-250115',
'url': 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
'key': env.llm_key_huoshan,
},
'huoshan-doubao-seed': {
'model': 'doubao-seed-1-6-250615',
'url': 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
'key': env.llm_key_huoshan,
},
'huoshan-doubao-vision': {
'model': 'doubao-1-5-vision-pro-32k-250115',
'url': 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
'key': env.llm_key_huoshan,
},
# /*---------------------------------------bailian-------------------------------------------*/
'bailian-qwen-turbo': {
'model': 'qwen-turbo',
'url': 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
'key': env.llm_key_bailian,
},
'bailian-qwen3-max': {
'model': 'qwen3-max',
'url': 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
'key': env.llm_key_bailian,
},
'bailian-qwen3.6-plus': {
'model': 'qwen3.6-plus',
'url': 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
'key': env.llm_key_bailian,
},
'bailian-embedding': {
'model': 'text-embedding-v4',
'url': 'https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings',
'key': env.llm_key_bailian,
},
}