240241002/ai_v
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

```

Browse Source 
feat(config): 更新AI备份API配置和密钥

- 添加AI_BACKUP_BASE_URL配置项，设置为"https://ai.comfly.chat"
- 更新多个API的备用密钥，包括TRIAL、PREMIUM、VIDEO和GEMINI_FLASH_IMAGE等服务
- 确保系统拥有有效的备份API端点和认证密钥

fix(utils): 改进备份API密钥获取逻辑

- 修改get_backup_api_key函数，当未找到对应的备份密钥时返回None而不是原密钥
- 更新get_api_candidates函数，添加backup_key存在性检查以避免使用空密钥
```
This commit is contained in:
24024 2026-05-14 22:41:07 +08:00
parent cf9100fcc3
commit 2bcb53e2eb
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
config.py
View File

@ -36,7 +36,7 @@ class Config:
# AI API 配置 # AI API 配置
AI_API = "https://ai.t8star.cn/v1/images/generations" AI_API = "https://ai.t8star.cn/v1/images/generations"
AI_BASE_URL = "https://api.bltcy.ai" AI_BASE_URL = "https://api.bltcy.ai"
AI_BACKUP_BASE_URL = "" AI_BACKUP_BASE_URL = "https://ai.comfly.chat"
CHAT_API = f"{AI_BASE_URL}/v1/chat/completions" CHAT_API = f"{AI_BASE_URL}/v1/chat/completions"
VIDEO_GEN_API = f"{AI_BASE_URL}/v2/videos/generations" VIDEO_GEN_API = f"{AI_BASE_URL}/v2/videos/generations"
VIDEO_POLL_API = f"{AI_BASE_URL}/v2/videos/generations/{{task_id}}" VIDEO_POLL_API = f"{AI_BASE_URL}/v2/videos/generations/{{task_id}}"
@ -47,13 +47,13 @@ class Config:
# 试用模式配置 # 试用模式配置
TRIAL_API = f"{AI_BASE_URL}/v1/images/generations" TRIAL_API = f"{AI_BASE_URL}/v1/images/generations"
TRIAL_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT" TRIAL_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT"
TRIAL_BACKUP_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT" TRIAL_BACKUP_KEY = "sk-Rr8L5noW8Qga7K4jmey3yYZYL1a4SlhlNlo5iZrwqJRK1Pa1"
PREMIUM_KEY = "sk-iMfeWAESLFIrHvxuBe2VbM4rC7ScZNvcaouqhVAd0J4KomYv" PREMIUM_KEY = "sk-iMfeWAESLFIrHvxuBe2VbM4rC7ScZNvcaouqhVAd0J4KomYv"
PREMIUM_BACKUP_KEY = "sk-iMfeWAESLFIrHvxuBe2VbM4rC7ScZNvcaouqhVAd0J4KomYv" PREMIUM_BACKUP_KEY = "sk-168trRxnemem6nTpQn1rbmJ4SFKLwTMsZ0G6uk5OipP7FKAY"
VIDEO_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT" VIDEO_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT"
VIDEO_BACKUP_KEY = "sk-pxowRoSbyavisIbaVLdedS7g5UePzhIxsTfjlJFOuqTRYQzT" VIDEO_BACKUP_KEY = "sk-Rr8L5noW8Qga7K4jmey3yYZYL1a4SlhlNlo5iZrwqJRK1Pa1"
GEMINI_FLASH_IMAGE_PREMIUM_KEY = "sk-0vTMf4vJ0RMvLK4J5wtkRDu7pNEDD69piFcDyf31NCdEYTC1" GEMINI_FLASH_IMAGE_PREMIUM_KEY = "sk-0vTMf4vJ0RMvLK4J5wtkRDu7pNEDD69piFcDyf31NCdEYTC1"
GEMINI_FLASH_IMAGE_PREMIUM_BACKUP_KEY = "sk-0vTMf4vJ0RMvLK4J5wtkRDu7pNEDD69piFcDyf31NCdEYTC1" GEMINI_FLASH_IMAGE_PREMIUM_BACKUP_KEY = "sk-OEbEnJORrKx4YEnLPEbwQL3eS5sp0eeSbtUepUrsIqjaLc1X"
DICT_URL = "https://nas.4x4g.com:10011/api/common/sys/dict" DICT_URL = "https://nas.4x4g.com:10011/api/common/sys/dict"
PLATFORM = "lingmao" PLATFORM = "lingmao"

4
utils.py
View File

@ -41,7 +41,7 @@ def get_backup_api_key(api_key):
Config.VIDEO_KEY: Config.VIDEO_BACKUP_KEY, Config.VIDEO_KEY: Config.VIDEO_BACKUP_KEY,
Config.GEMINI_FLASH_IMAGE_PREMIUM_KEY: Config.GEMINI_FLASH_IMAGE_PREMIUM_BACKUP_KEY, Config.GEMINI_FLASH_IMAGE_PREMIUM_KEY: Config.GEMINI_FLASH_IMAGE_PREMIUM_BACKUP_KEY,
} }
return fallback_map.get(api_key, api_key) return fallback_map.get(api_key)
def get_api_candidates(url, api_key): def get_api_candidates(url, api_key):
""" """
@ -51,7 +51,7 @@ def get_api_candidates(url, api_key):
backup_url = get_backup_api_url(url) backup_url = get_backup_api_url(url)
backup_key = get_backup_api_key(api_key) backup_key = get_backup_api_key(api_key)
if backup_url and backup_url != url: if backup_url and backup_url != url and backup_key:
candidates.append({"url": backup_url, "api_key": backup_key, "label": "backup"}) candidates.append({"url": backup_url, "api_key": backup_key, "label": "backup"})
return candidates return candidates