#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
LLM配置文件 - 千问模型配置
"""

from src.agent.llm_client import LLMConfig

# 千问模型配置
QWEN_CONFIG = LLMConfig(
    provider="openai",
    api_key="sk-your-qwen-api-key-here",  # 请替换为您的千问API密钥
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    model="qwen-turbo",  # 可选: qwen-turbo, qwen-plus, qwen-max
    temperature=0.7,
    max_tokens=2000
)

# 其他模型配置示例
OPENAI_CONFIG = LLMConfig(
    provider="openai",
    api_key="sk-your-openai-api-key-here",
    model="gpt-3.5-turbo",
    temperature=0.7,
    max_tokens=2000
)

ANTHROPIC_CONFIG = LLMConfig(
    provider="anthropic",
    api_key="sk-ant-your-anthropic-api-key-here",
    model="claude-3-sonnet-20240229",
    temperature=0.7,
    max_tokens=2000
)

# 默认使用千问模型
DEFAULT_CONFIG = QWEN_CONFIG