ai chat init

This commit is contained in:
xie7654
2025-07-12 16:54:49 +08:00
parent 62bba7d23f
commit f5314efc81
10 changed files with 390 additions and 18 deletions

11
backend/ai/utils.py Normal file
View File

@@ -0,0 +1,11 @@
from ai.models import AIModel
from utils.models import CommonStatus
from asgiref.sync import sync_to_async
@sync_to_async
def get_first_available_ai_config():
# 这里只取第一个可用的,可以根据实际业务加筛选条件
ai = AIModel.objects.filter(status=CommonStatus.ENABLED).prefetch_related('key').first()
if not ai:
raise Exception('没有可用的AI配置')
return ai.model, ai.key.api_key, ai.key.url