公司git
cf9100fcc3
feat(api): 统一图像生成功能至generations接口 - 将gpt-image-2模型统一使用AI_BASE_URL下的generations接口， 不再区分有无参考图走不同接口 - 在payload中添加response_format字段并移除output_format字段 - 更新图像数据处理逻辑，将image_data转换为image数组格式 refactor(config): 更新图像生成API配置 - 将GPT_IMAGE_GENERATION_API从硬编码URL改为使用AI_BASE_URL动态构建 refactor(generation_service): 简化图像生成请求验证逻辑 - 移除has_reference_images判断逻辑，统一使用generations接口 - 积分模式下gpt-image-2改用PREMIUM_KEY而非专用编辑key - 移除不必要的image_data变量声明 refactor(task_service): 移除废弃的图像处理函数 - 删除_base64相关处理函数(_decode_base64_image等) - 移除_minio上传相关函数(_upload_image_bytes_to_minio等) - 移除gpt-image-2特定处理逻辑(_process_gpt_image_edit) - 简化图像生成任务处理流程 ```
93 lines
5.8 KiB
Python
93 lines
5.8 KiB
Python
import os
|
||
|
||
class Config:
|
||
# 基础配置
|
||
SECRET_KEY = os.getenv("SECRET_KEY", "vision-ai-secret-key")
|
||
|
||
# PostgreSQL 配置
|
||
SQLALCHEMY_DATABASE_URI = "postgresql://user_xREpkJ:password_DZz8DQ@331002.xyz:2022/ai_vision"
|
||
SQLALCHEMY_TRACK_MODIFICATIONS = False
|
||
SQLALCHEMY_ENGINE_OPTIONS = {
|
||
"pool_pre_ping": True,
|
||
"pool_recycle": 1800,
|
||
"pool_timeout": 30,
|
||
"connect_args": {
|
||
"connect_timeout": 5
|
||
}
|
||
}
|
||
|
||
# Redis 配置
|
||
REDIS_URL = "redis://:redis_WWjNyb@331002.xyz:2020/0"
|
||
|
||
# Session 配置
|
||
PERMANENT_SESSION_LIFETIME = 604800 # 7 天 (单位：秒)
|
||
SESSION_COOKIE_SECURE = False # 开发环境或非 HTTPS 生产环境设为 False
|
||
SESSION_COOKIE_HTTPONLY = True # 防止 XSS 获取 Cookie
|
||
|
||
# MinIO 配置
|
||
MINIO = {
|
||
"endpoint": "http://331002.xyz:9000",
|
||
"access_key": "l0VlsxrkASbXN2YSQrJk",
|
||
"secret_key": "ZK8nXHieorl3fpbssUMGGfr8zZmbpXB5gAbma3z1",
|
||
"bucket": "images",
|
||
"public_url": "/files/"
|
||
}
|
||
|
||
# AI API 配置
|
||
AI_API = "https://ai.t8star.cn/v1/images/generations"
|
||
AI_BASE_URL = "https://api.bltcy.ai"
|
||
AI_BACKUP_BASE_URL = ""
|
||
CHAT_API = f"{AI_BASE_URL}/v1/chat/completions"
|
||
VIDEO_GEN_API = f"{AI_BASE_URL}/v2/videos/generations"
|
||
VIDEO_POLL_API = f"{AI_BASE_URL}/v2/videos/generations/{{task_id}}"
|
||
GPT_IMAGE_GENERATION_API = f"{AI_BASE_URL}/v1/images/generations"
|
||
GPT_IMAGE_EDIT_API = "https://nas.4x4g.com:8317/v1/images/edits"
|
||
GPT_IMAGE_EDIT_KEY = "sk-vDmmYNzoe7UjWskKx"
|
||
|
||
# 试用模式配置
|
||
TRIAL_API = f"{AI_BASE_URL}/v1/images/generations"
|
||
TRIAL_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT"
|
||
TRIAL_BACKUP_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT"
|
||
PREMIUM_KEY = "sk-iMfeWAESLFIrHvxuBe2VbM4rC7ScZNvcaouqhVAd0J4KomYv"
|
||
PREMIUM_BACKUP_KEY = "sk-iMfeWAESLFIrHvxuBe2VbM4rC7ScZNvcaouqhVAd0J4KomYv"
|
||
VIDEO_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT"
|
||
VIDEO_BACKUP_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT"
|
||
GEMINI_FLASH_IMAGE_PREMIUM_KEY = "sk-0vTMf4vJ0RMvLK4J5wtkRDu7pNEDD69piFcDyf31NCdEYTC1"
|
||
GEMINI_FLASH_IMAGE_PREMIUM_BACKUP_KEY = "sk-0vTMf4vJ0RMvLK4J5wtkRDu7pNEDD69piFcDyf31NCdEYTC1"
|
||
|
||
DICT_URL = "https://nas.4x4g.com:10011/api/common/sys/dict"
|
||
PLATFORM = "lingmao"
|
||
|
||
# 阿里云短信配置
|
||
ALIBABA_CLOUD_ACCESS_KEY_ID = "LTAI5tAbHKxmPKVPYsABEdyq"
|
||
ALIBABA_CLOUD_ACCESS_KEY_SECRET = "v6URREddBqvGfwZrWH1DWoxs3w6RxZ"
|
||
SMS_SIGN_NAME = "速通互联验证码"
|
||
SMS_TEMPLATE_CODE = "100001"
|
||
SMS_NEED_PARAM = False # 该模板需要参数，如使用系统赠送模板请改为 False
|
||
|
||
# 支付宝配置 (待填)
|
||
ALIPAY_APP_ID = "2021006125623511" # 支付宝 AppID
|
||
ALIPAY_APP_PRIVATE_KEY = """-----BEGIN RSA PRIVATE KEY-----
|
||
MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQC+BMpGTJMzDoOnjyGh69rDLbV/8rlBRDAWdhyT3s0+iDpbk/j8kZvG3TG3rQdgQJLMMb9O+nIbkyl4BP/TJCnRummUGQlRW+S1mnfxmMZWCWRgqRecqQFnfWBwgn7+pNA/nKftTOplBzWZpAzRi4HWn0p2e72IcgJ2Pz6no0ldVYE2qJu5w7wmFFjRX+E4G94nZ0WPnZd+UnZwHzj1GXTSLhgF2OC6SVTgK4iOP2bTKQmnz5NjdQlIF0GVa1QqgS2VSE7hTbpM83YFIl9VwIiVpdvFuZApqFmECjdRSKSP4BB1XouQcTdtkE0mnrTMfeEswbsKgXh+dnuTs08OgninAgMBAAECggEAZt5C/vG/kpld7p0O8RC0+orJQj4NymZ2vgItEGRCqoKnTSJ7XYgaU/s1SiFS5+mtTvvzfHcaxRDTF1VlhVGDjjWjqHgxUik0VJgOLO2vGa6195oy3S2lQ/hc+Cu9Qm/JCVbulgnYrt7LRqTIB7zFioHrlt8ASVinOoqcibZdJjgNqvEnBwjqHxI7oyqEr5/Yz6hxNjoAeUr3b7Y37J3DKsD4L6koWRjvcoPi59FdkwBfCh63tYjJU1AXdU5pdHtl3+Ieu4G2oIRjwUlnH3FEVKhnFFECZ7i27QhK+IQrJdiJmSqCNzeJT0f0ZiUfMM/txYcpvI7i7FqyCJaYFhCtkQKBgQD+HM4Rhb+B7BWPKD/5h0zmplw+hyldOf7uo+jph861F73WV7DshD0inqyyXfXcX0fuMlNyXdxixCtlBdtGisR3IvY3wLPVzx7JcQ5z+NbXj4h3qPRfU3iu6gE2GCrkqVPjURRv/4Z6F6FxW03clSOp+SzzjyD8c2B1vrWBknBAbwKBgQC/bhxcNcNZ98EL51JI9H5aCSmOriZpA6Se4BQ6CLeGUzC+7/z2towNOswKNSJX3KRfsVUQuFuM4LejHXMvgkb1tAD0VFre/7P8XOn1RV96w3szv6kNXBzmvrl/K/Jk1UwoKkjqE6Ko6Oleru0WXw8C156gY+LFogBwNWagB3v3SQKBgHfXpeflqBfh+5j0ZpeME9yDgOWMYRmyjGSHsQTSI86MT9+So8In6500MEY+wgEYfpf77Bb35UQ/OCgAxxVRxK5bMrvAig/ACsj+JM7VAJ3Q949Xg0ih2TzE56VXaINGyGVgi772R7m1A1KuuN3bCGLVPTnwuqHU2B0vv/JH2eDJAoGAeoxgJXvZkqfz8s3ktIWoubTx4MCGIGZPpdHnwPtBmgF5zFJQjBRy9ik05Df9w2eh5wzcWzbajQZzKT46Nijdin/wuHmkLuxdaeqFcBo4NJweLBO/gqj8a4+fqtUUeQ6/kVr4kLl/QUjR1WaAqvix4GzuxNkNHN5ID41wJHSGHNkCgYEA4TVQiCPzyMIOizhLlNh0p5wIK+zUHs5fVamXm/pzgakhmzJxP9MfMhTuA3N11fR/JrJ2E+Nt9uDutWWGuRpty4/s0AKTCvXoVOZGwBvQ0vEk/S0dQuOxCC/ObQl9q3sZelEazVgqjVTgIUGuxRgKnA3mMoMoQvZ6c3JXUSym0rY=
|
||
-----END RSA PRIVATE KEY-----""" # 应用私钥
|
||
ALIPAY_PUBLIC_KEY = """-----BEGIN PUBLIC KEY-----
|
||
MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAlDx4KdOtOQE+tBq6jHKKFenRaRe2gbBnleBk++5gki9IQuxVyZUGTJixstf2gELFHWrGanpnwmGggXsqG+Rm5ZLJOlmFM1k0XeAIDvi6tP/rM+ZDFSu1bMBYtT5vzgVZC7mzIvOp9gsT/puqd3aNZmlviLD0R6OYN0zvFX+5qADZV7A9ziA+nXPFSHreBh7yY/q9ophVZNeHGPoYkDVI5++RrF1cALKOdit0giN5vxpe3ch9z3E6+FZg3LiP+1RW3tMiDQfp/SlVs6bNhLUtmlI5r7+mtFCKDUCEpnQ3S9e0II6rzyVXRyKCFs7qi5YzyhhmO3tJJoe9ilEFyNzfRQIDAQAB
|
||
-----END PUBLIC KEY-----""" # 支付宝公钥
|
||
ALIPAY_RETURN_URL = "https://860576.xyz/payment/return" # 支付成功跳转地址
|
||
ALIPAY_NOTIFY_URL = "https://860576.xyz/payment/notify" # 支付异步通知地址
|
||
ALIPAY_DEBUG = False # 是否使用沙箱环境
|
||
|
||
# 开发模式配置
|
||
DEV_MODE = False # True=开发模式（固定验证码），False=生产模式（真实短信）
|
||
DEV_SMS_CODE = "888888" # 开发模式下的固定验证码
|
||
|
||
# Proxy Config
|
||
USE_PROXY = True # 是否开启代理
|
||
PROXY_URL = "https://127007.xyz/"
|
||
|
||
# 代理超时配置 (单位：秒)
|
||
PROXY_TIMEOUT_DEFAULT = 60 # 默认超时
|
||
PROXY_TIMEOUT_SHORT = 30 # 短任务超时 (轮询、小图片下载)
|
||
PROXY_TIMEOUT_LONG = 300 # 长任务超时 (5分钟，视频下载、大文件)
|
||
PROXY_TIMEOUT_GENERATION = 600 # 生成任务超时 (10分钟，AI 生成请求)
|