8_multi_model.py

# 通过OpenAI Python SDK 支持其他大模型
# 或者通过 https://github.com/songquanpeng/one-api 支持其他大模型
from GeneralAgent import Agent
from dotenv import load_dotenv

load_dotenv()

models = [
    ('deepseek-chat', 32000, 'sk-xxx', 'https://api.deepseek.com/v1'),  # DeepSeek官方支持
    ('moonshot-v1-128k', 128000, '$MOONSHOT_API_KEY', 'https://api.moonshot.cn/v1'),  # Moonshot官方支持
    ('SparkDesk-v3.5', 4000, None, None),
    ('glm-4v', 128000, None, None),
    ('ERNIE-4.0-8K', 8000, None, None),
    ('qwen-turbo', 6000, None, None),
    ('hunyuan', 8000, None, None),
]

for model, token_limit, api_key, base_url in models:
    agent = Agent('You are a helpful agent.', model=model, token_limit=token_limit, api_key=api_key, base_url=base_url)
    agent.user_input('介绍一下成都')