37 lines
887 B
Python
37 lines
887 B
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
LLM配置文件 - 千问模型配置
|
|
"""
|
|
|
|
from src.agent.llm_client import LLMConfig
|
|
|
|
# 千问模型配置
|
|
QWEN_CONFIG = LLMConfig(
|
|
provider="qwen",
|
|
api_key="sk-c0dbefa1718d46eaa897199135066f00", # 请替换为您的千问API密钥
|
|
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
model="qwen-plus-latest", # 可选: 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
|