85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
|
import os
|
|||
|
|
from typing import Dict, Any
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
"""新闻稿生成器配置文件"""
|
|||
|
|
|
|||
|
|
# 模型配置
|
|||
|
|
MODELS = {
|
|||
|
|
'openai': {
|
|||
|
|
'name': 'OpenAI GPT-4',
|
|||
|
|
'api_key': os.getenv('OPENAI_API_KEY', ''),
|
|||
|
|
'model': 'gpt-4',
|
|||
|
|
'base_url': 'https://api.openai.com/v1',
|
|||
|
|
'max_tokens': 800,
|
|||
|
|
'temperature': 0.7
|
|||
|
|
},
|
|||
|
|
'volcengine': {
|
|||
|
|
'name': '火山引擎豆包',
|
|||
|
|
'api_key': os.getenv('VOLCENGINE_API_KEY', '6840c164-3b7e-4ddc-9d51-20624defc205'),
|
|||
|
|
'model': 'doubao-pro-4k',
|
|||
|
|
'base_url': 'maas-api.ml-platform-cn-beijing.volces.com',
|
|||
|
|
'max_tokens': 800,
|
|||
|
|
'temperature': 0.7
|
|||
|
|
},
|
|||
|
|
'dashscope': {
|
|||
|
|
'name': '阿里通义千问',
|
|||
|
|
'api_key': os.getenv('DASHSCOPE_API_KEY', 'sk-29a937c25bff4752af88514525c329b3'),
|
|||
|
|
'model': 'qwen-plus',
|
|||
|
|
'base_url': 'https://dashscope.aliyuncs.com',
|
|||
|
|
'max_tokens': 800,
|
|||
|
|
'temperature': 0.7
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 文案风格配置
|
|||
|
|
STYLES = ['正式', '简洁', '感性', '科技']
|
|||
|
|
|
|||
|
|
# Flask应用配置
|
|||
|
|
FLASK_CONFIG = {
|
|||
|
|
'debug': True,
|
|||
|
|
'host': '127.0.0.1',
|
|||
|
|
'port': 50001,
|
|||
|
|
'secret_key': os.getenv('FLASK_SECRET_KEY', 'your-secret-key-here')
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 文档配置
|
|||
|
|
DOCUMENT_CONFIG = {
|
|||
|
|
'default_filename': 'news.docx',
|
|||
|
|
'image_width': 4, # 图片宽度（英寸）
|
|||
|
|
'title': '新闻稿'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def get_model_config(cls, model_type: str) -> Dict[str, Any]:
|
|||
|
|
"""获取指定模型的配置"""
|
|||
|
|
if model_type not in cls.MODELS:
|
|||
|
|
raise ValueError(f"不支持的模型类型: {model_type}")
|
|||
|
|
return cls.MODELS[model_type]
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def validate_api_keys(cls) -> Dict[str, bool]:
|
|||
|
|
"""验证所有API密钥是否已设置"""
|
|||
|
|
validation_result = {}
|
|||
|
|
for model_type, config in cls.MODELS.items():
|
|||
|
|
validation_result[model_type] = bool(config['api_key'])
|
|||
|
|
return validation_result
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def get_available_models(cls) -> Dict[str, str]:
|
|||
|
|
"""获取可用的模型列表（已配置API密钥的模型）"""
|
|||
|
|
available = {}
|
|||
|
|
for model_type, config in cls.MODELS.items():
|
|||
|
|
if config['api_key']:
|
|||
|
|
available[model_type] = config['name']
|
|||
|
|
return available
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def update_api_key(cls, model_type: str, api_key: str) -> None:
|
|||
|
|
"""更新指定模型的API密钥"""
|
|||
|
|
if model_type not in cls.MODELS:
|
|||
|
|
raise ValueError(f"不支持的模型类型: {model_type}")
|
|||
|
|
cls.MODELS[model_type]['api_key'] = api_key
|
|||
|
|
|
|||
|
|
# 创建配置实例
|
|||
|
|
config = Config()
|