38 lines
896 B
Python
38 lines
896 B
Python
|
#!/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
|