未来
2bd8b61e91
新功能: - 大纲/章节生成改为服务端后台任务，支持断线续传 - 后台任务队列排队执行，按用户排队（同用户串行不同用户并发） - 章节管理页面添加后台任务列表弹窗和进度面板 - 章节状态添加 pending（待处理）选项 - 集成json5容错解析器 + 上下文感知JSON修复 - SSE流式生成添加心跳保活，防止连接超时 - SSEPostClient添加credentials:include修复network error - 每章最大伏笔数从2调整为5 - 添加大纲读区伏笔的功能 Bug修复: - 修复AI生成JSON中未转义引号/中文标点/多对象属性值未合并 - 修复JSON非法转义字符清洗和中文引号处理 - 修复MCP插件TimeoutError/连接失败上下文清理 - MCP插件后台注册添加重试机制 - 续写模式添加缺失的mcp_references参数 - 修复Alembic迁移链分叉 - 使用torch CPU版本加速Docker构建
1336 lines
48 KiB
Python
1336 lines
48 KiB
Python
"""
|
||
设置管理 API
|
||
"""
|
||
from fastapi import APIRouter, HTTPException, Request, Depends
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy import select
|
||
from typing import Dict, Any, List, Optional
|
||
from pathlib import Path
|
||
from pydantic import BaseModel
|
||
from datetime import datetime
|
||
import httpx
|
||
import json
|
||
import time
|
||
|
||
from app.database import get_db
|
||
from app.models.settings import Settings
|
||
from app.services.cover_generation_service import cover_generation_service
|
||
from app.schemas.settings import (
|
||
SettingsCreate, SettingsUpdate, SettingsResponse,
|
||
APIKeyPreset, APIKeyPresetConfig, PresetCreateRequest,
|
||
PresetUpdateRequest, PresetResponse, PresetListResponse,
|
||
SystemSMTPSettingsResponse, SystemSMTPSettingsUpdate, SMTPTestRequest
|
||
)
|
||
from app.user_manager import User
|
||
from app.logger import get_logger
|
||
from app.config import settings as app_settings, PROJECT_ROOT
|
||
from app.services.ai_service import AIService, create_user_ai_service, create_user_ai_service_with_mcp, normalize_provider
|
||
from app.services.email_service import email_service
|
||
from app.security import validate_public_http_url
|
||
|
||
logger = get_logger(__name__)
|
||
|
||
router = APIRouter(prefix="/settings", tags=["设置管理"])
|
||
|
||
|
||
class CoverSettingsTestRequest(BaseModel):
|
||
cover_api_provider: str
|
||
cover_api_key: str
|
||
cover_api_base_url: Optional[str] = None
|
||
cover_image_model: str
|
||
|
||
|
||
def read_env_defaults() -> Dict[str, Any]:
|
||
"""从.env文件读取默认配置（仅读取，不修改）"""
|
||
return {
|
||
"api_provider": app_settings.default_ai_provider,
|
||
"api_key": app_settings.openai_api_key or app_settings.anthropic_api_key or "",
|
||
"api_base_url": app_settings.openai_base_url or app_settings.anthropic_base_url or "",
|
||
"llm_model": app_settings.default_model,
|
||
"temperature": app_settings.default_temperature,
|
||
"max_tokens": app_settings.default_max_tokens,
|
||
}
|
||
|
||
|
||
def require_login(request: Request):
|
||
"""依赖：要求用户已登录"""
|
||
if not hasattr(request.state, "user") or not request.state.user:
|
||
raise HTTPException(status_code=401, detail="需要登录")
|
||
return request.state.user
|
||
|
||
|
||
def require_admin(user: User = Depends(require_login)):
|
||
"""依赖：要求管理员权限"""
|
||
if not user.is_admin:
|
||
raise HTTPException(status_code=403, detail="仅管理员可访问系统设置")
|
||
return user
|
||
|
||
|
||
async def get_or_create_admin_settings(db: AsyncSession, user: User) -> Settings:
|
||
"""获取或创建管理员设置，系统级 SMTP 配置挂在管理员设置记录上"""
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user.user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
env_defaults = read_env_defaults()
|
||
settings = Settings(
|
||
user_id=user.user_id,
|
||
smtp_provider=app_settings.SMTP_PROVIDER,
|
||
smtp_host=app_settings.SMTP_HOST,
|
||
smtp_port=app_settings.SMTP_PORT,
|
||
smtp_username=app_settings.SMTP_USERNAME,
|
||
smtp_password=app_settings.SMTP_PASSWORD,
|
||
smtp_use_tls=app_settings.SMTP_USE_TLS,
|
||
smtp_use_ssl=app_settings.SMTP_USE_SSL,
|
||
smtp_from_email=app_settings.SMTP_FROM_EMAIL,
|
||
smtp_from_name=app_settings.SMTP_FROM_NAME,
|
||
email_auth_enabled=app_settings.EMAIL_AUTH_ENABLED,
|
||
email_register_enabled=app_settings.EMAIL_REGISTER_ENABLED,
|
||
verification_code_ttl_minutes=app_settings.EMAIL_VERIFICATION_CODE_TTL_MINUTES,
|
||
verification_resend_interval_seconds=app_settings.EMAIL_VERIFICATION_RESEND_INTERVAL_SECONDS,
|
||
**env_defaults
|
||
)
|
||
db.add(settings)
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
|
||
return settings
|
||
|
||
|
||
async def get_user_ai_service(
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
) -> AIService:
|
||
"""
|
||
依赖：获取当前用户的AI服务实例（支持MCP工具自动加载）
|
||
|
||
从数据库读取用户设置并创建对应的AI服务。
|
||
自动传递 user_id 和 db_session，使得 AIService 能够加载用户配置的MCP工具。
|
||
根据用户的所有MCP插件状态决定是否启用MCP：如果有启用的插件则启用，否则禁用。
|
||
"""
|
||
from app.models.mcp_plugin import MCPPlugin
|
||
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user.user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
# 如果用户没有设置，从.env读取并保存
|
||
env_defaults = read_env_defaults()
|
||
settings = Settings(
|
||
user_id=user.user_id,
|
||
**env_defaults
|
||
)
|
||
db.add(settings)
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"用户 {user.user_id} 首次使用AI服务，已从.env同步设置到数据库")
|
||
|
||
# 查询用户的所有MCP插件状态
|
||
mcp_result = await db.execute(
|
||
select(MCPPlugin).where(MCPPlugin.user_id == user.user_id)
|
||
)
|
||
mcp_plugins = mcp_result.scalars().all()
|
||
|
||
# 检查是否有启用的MCP插件
|
||
enable_mcp = any(plugin.enabled for plugin in mcp_plugins) if mcp_plugins else False
|
||
|
||
if mcp_plugins:
|
||
enabled_count = sum(1 for p in mcp_plugins if p.enabled)
|
||
logger.info(f"用户 {user.user_id} 有 {len(mcp_plugins)} 个MCP插件，{enabled_count} 个启用，{enable_mcp} 决定使用MCP")
|
||
else:
|
||
logger.debug(f"用户 {user.user_id} 没有配置MCP插件，禁用MCP")
|
||
|
||
# ✅ 使用支持MCP的工厂函数创建AI服务实例
|
||
# 传递 user_id 和 db_session，使得 AIService 能够自动加载用户配置的MCP工具
|
||
return create_user_ai_service_with_mcp(
|
||
api_provider=settings.api_provider,
|
||
api_key=settings.api_key,
|
||
api_base_url=settings.api_base_url or "",
|
||
model_name=settings.llm_model,
|
||
temperature=settings.temperature,
|
||
max_tokens=settings.max_tokens,
|
||
user_id=user.user_id, # ✅ 传递 user_id
|
||
db_session=db, # ✅ 传递 db_session
|
||
system_prompt=settings.system_prompt,
|
||
enable_mcp=enable_mcp, # 根据MCP插件状态动态决定
|
||
)
|
||
|
||
|
||
async def get_user_ai_service_from_db(user_id: str, db: AsyncSession) -> AIService:
|
||
"""
|
||
从数据库直接创建用户AI服务实例（用于后台任务，不依赖FastAPI的Depends）
|
||
"""
|
||
from app.models.mcp_plugin import MCPPlugin
|
||
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
env_defaults = read_env_defaults()
|
||
settings = Settings(user_id=user_id, **env_defaults)
|
||
db.add(settings)
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
|
||
mcp_result = await db.execute(
|
||
select(MCPPlugin).where(MCPPlugin.user_id == user_id)
|
||
)
|
||
mcp_plugins = mcp_result.scalars().all()
|
||
enable_mcp = any(plugin.enabled for plugin in mcp_plugins) if mcp_plugins else False
|
||
|
||
return create_user_ai_service_with_mcp(
|
||
api_provider=settings.api_provider,
|
||
api_key=settings.api_key,
|
||
api_base_url=settings.api_base_url or "",
|
||
model_name=settings.llm_model,
|
||
temperature=settings.temperature,
|
||
max_tokens=settings.max_tokens,
|
||
user_id=user_id,
|
||
db_session=db,
|
||
system_prompt=settings.system_prompt,
|
||
enable_mcp=enable_mcp,
|
||
)
|
||
|
||
|
||
@router.get("", response_model=SettingsResponse)
|
||
async def get_settings(
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
获取当前用户的设置
|
||
如果用户没有保存过设置，自动从.env创建并保存到数据库
|
||
"""
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user.user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
# 如果用户没有保存过设置，从.env读取默认配置并保存到数据库
|
||
env_defaults = read_env_defaults()
|
||
logger.info(f"用户 {user.user_id} 首次获取设置，自动从.env同步到数据库")
|
||
|
||
# 创建新设置并保存到数据库
|
||
settings = Settings(
|
||
user_id=user.user_id,
|
||
**env_defaults
|
||
)
|
||
db.add(settings)
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"用户 {user.user_id} 的设置已从.env同步到数据库")
|
||
|
||
logger.info(f"用户 {user.user_id} 获取已保存的设置")
|
||
return settings
|
||
|
||
|
||
@router.post("/cover/test")
|
||
async def test_cover_settings(
|
||
data: CoverSettingsTestRequest,
|
||
user: User = Depends(require_login),
|
||
):
|
||
result = await cover_generation_service.test_cover_settings(
|
||
provider=data.cover_api_provider,
|
||
api_key=data.cover_api_key,
|
||
api_base_url=data.cover_api_base_url,
|
||
model=data.cover_image_model,
|
||
)
|
||
return {
|
||
"success": result.success,
|
||
"message": result.message,
|
||
"provider": result.provider,
|
||
"model": result.model,
|
||
}
|
||
|
||
|
||
@router.get("/system/smtp", response_model=SystemSMTPSettingsResponse)
|
||
async def get_system_smtp_settings(
|
||
user: User = Depends(require_admin),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取系统 SMTP 设置（仅管理员）"""
|
||
settings = await get_or_create_admin_settings(db, user)
|
||
return settings
|
||
|
||
|
||
@router.put("/system/smtp", response_model=SystemSMTPSettingsResponse)
|
||
async def update_system_smtp_settings(
|
||
data: SystemSMTPSettingsUpdate,
|
||
user: User = Depends(require_admin),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""更新系统 SMTP 设置（仅管理员）"""
|
||
settings = await get_or_create_admin_settings(db, user)
|
||
update_data = data.model_dump(exclude_unset=True)
|
||
|
||
if update_data.get("smtp_provider") == "qq":
|
||
update_data.setdefault("smtp_host", "smtp.qq.com")
|
||
update_data.setdefault("smtp_port", 465)
|
||
update_data.setdefault("smtp_use_ssl", True)
|
||
update_data.setdefault("smtp_use_tls", False)
|
||
|
||
if update_data.get("smtp_use_ssl") and update_data.get("smtp_use_tls"):
|
||
raise HTTPException(status_code=400, detail="SSL 和 TLS 不能同时启用")
|
||
|
||
for key, value in update_data.items():
|
||
setattr(settings, key, value)
|
||
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"管理员 {user.user_id} 更新系统 SMTP 设置")
|
||
return settings
|
||
|
||
|
||
@router.post("/system/smtp/test")
|
||
async def test_system_smtp_settings(
|
||
data: SMTPTestRequest,
|
||
user: User = Depends(require_admin),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""测试系统 SMTP 设置（真实发送测试邮件）"""
|
||
settings = await get_or_create_admin_settings(db, user)
|
||
|
||
if not settings.smtp_host or not settings.smtp_username or not settings.smtp_password:
|
||
raise HTTPException(status_code=400, detail="请先完善 SMTP 主机、用户名和授权码")
|
||
|
||
if settings.smtp_provider == "qq" and settings.smtp_host != "smtp.qq.com":
|
||
raise HTTPException(status_code=400, detail="QQ 邮箱 SMTP 主机必须为 smtp.qq.com")
|
||
|
||
if "@" not in data.to_email or "." not in data.to_email.split("@")[-1]:
|
||
raise HTTPException(status_code=400, detail="测试收件邮箱格式不正确")
|
||
|
||
from_email = settings.smtp_from_email or settings.smtp_username
|
||
if not from_email:
|
||
raise HTTPException(status_code=400, detail="请先配置发件人邮箱或 SMTP 用户名")
|
||
|
||
subject = "MuMuAINovel SMTP 测试邮件"
|
||
text_body = (
|
||
"这是一封来自 MuMuAINovel 系统设置页面的 SMTP 测试邮件。\n\n"
|
||
f"发送时间：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n"
|
||
f"SMTP 服务商：{settings.smtp_provider}\n"
|
||
f"SMTP 主机：{settings.smtp_host}:{settings.smtp_port}\n"
|
||
"如果你收到这封邮件，说明当前 SMTP 配置可正常发送邮件。"
|
||
)
|
||
html_body = f"""
|
||
<div style=\"font-family: Arial, sans-serif; line-height: 1.7; color: #1f1f1f;\">
|
||
<h2 style=\"margin-bottom: 12px;\">MuMuAINovel SMTP 测试邮件</h2>
|
||
<p>这是一封来自系统设置页面的 SMTP 测试邮件。</p>
|
||
<ul>
|
||
<li><strong>发送时间：</strong>{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}</li>
|
||
<li><strong>SMTP 服务商：</strong>{settings.smtp_provider}</li>
|
||
<li><strong>SMTP 主机：</strong>{settings.smtp_host}:{settings.smtp_port}</li>
|
||
</ul>
|
||
<p>如果你收到这封邮件，说明当前 SMTP 配置可正常发送邮件。</p>
|
||
</div>
|
||
"""
|
||
|
||
try:
|
||
await email_service.send_mail(
|
||
host=settings.smtp_host,
|
||
port=settings.smtp_port,
|
||
username=settings.smtp_username,
|
||
password=settings.smtp_password,
|
||
use_tls=settings.smtp_use_tls,
|
||
use_ssl=settings.smtp_use_ssl,
|
||
from_email=from_email,
|
||
from_name=settings.smtp_from_name,
|
||
to_email=data.to_email,
|
||
subject=subject,
|
||
text_body=text_body,
|
||
html_body=html_body,
|
||
)
|
||
except Exception as exc:
|
||
logger.exception(f"SMTP 测试邮件发送失败: {exc}")
|
||
raise HTTPException(status_code=400, detail=f"SMTP 测试邮件发送失败: {str(exc)}") from exc
|
||
|
||
return {
|
||
"success": True,
|
||
"message": f"测试邮件已发送至 {data.to_email}，请检查收件箱和垃圾箱",
|
||
"provider": settings.smtp_provider,
|
||
"host": settings.smtp_host,
|
||
"port": settings.smtp_port,
|
||
}
|
||
|
||
|
||
@router.post("", response_model=SettingsResponse)
|
||
async def save_settings(
|
||
data: SettingsCreate,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
创建或更新当前用户的设置（Upsert）
|
||
如果设置已存在则更新，否则创建新设置
|
||
仅保存到数据库
|
||
|
||
注意：手动保存配置后会自动取消之前激活的预设状态，
|
||
因为手动修改的配置可能与预设不一致
|
||
"""
|
||
# 查找现有设置
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user.user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
# 准备数据
|
||
settings_dict = data.model_dump(exclude_unset=True)
|
||
|
||
if settings:
|
||
# 更新现有设置
|
||
for key, value in settings_dict.items():
|
||
setattr(settings, key, value)
|
||
|
||
# 检查并取消预设激活状态
|
||
# 因为用户手动修改了配置，可能与之前激活的预设不一致
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 找到激活的预设并检查是否与当前保存的配置一致
|
||
active_preset = next((p for p in presets if p.get('is_active')), None)
|
||
if active_preset:
|
||
preset_config = active_preset.get('config', {})
|
||
# 检查配置是否发生变化
|
||
config_changed = (
|
||
preset_config.get('api_provider') != settings_dict.get('api_provider', settings.api_provider) or
|
||
preset_config.get('api_key') != settings_dict.get('api_key', settings.api_key) or
|
||
preset_config.get('api_base_url') != settings_dict.get('api_base_url', settings.api_base_url) or
|
||
preset_config.get('llm_model') != settings_dict.get('llm_model', settings.llm_model) or
|
||
preset_config.get('temperature') != settings_dict.get('temperature', settings.temperature) or
|
||
preset_config.get('max_tokens') != settings_dict.get('max_tokens', settings.max_tokens)
|
||
)
|
||
|
||
if config_changed:
|
||
# 取消激活状态
|
||
active_preset['is_active'] = False
|
||
prefs['api_presets'] = api_presets
|
||
settings.preferences = json.dumps(prefs, ensure_ascii=False)
|
||
logger.info(f"用户 {user.user_id} 手动修改配置，已取消预设 {active_preset.get('name')} 的激活状态")
|
||
except (json.JSONDecodeError, TypeError) as e:
|
||
logger.warning(f"解析用户 {user.user_id} 的preferences失败: {e}")
|
||
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"用户 {user.user_id} 更新设置")
|
||
else:
|
||
# 创建新设置
|
||
settings = Settings(
|
||
user_id=user.user_id,
|
||
**settings_dict
|
||
)
|
||
db.add(settings)
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"用户 {user.user_id} 创建设置")
|
||
|
||
return settings
|
||
|
||
|
||
@router.put("", response_model=SettingsResponse)
|
||
async def update_settings(
|
||
data: SettingsUpdate,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
更新当前用户的设置
|
||
仅保存到数据库
|
||
"""
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user.user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
raise HTTPException(status_code=404, detail="设置不存在，请先创建设置")
|
||
|
||
# 更新设置
|
||
update_data = data.model_dump(exclude_unset=True)
|
||
for key, value in update_data.items():
|
||
setattr(settings, key, value)
|
||
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"用户 {user.user_id} 更新设置")
|
||
|
||
return settings
|
||
|
||
|
||
@router.delete("")
|
||
async def delete_settings(
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
删除当前用户的设置
|
||
"""
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user.user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
raise HTTPException(status_code=404, detail="设置不存在")
|
||
|
||
await db.delete(settings)
|
||
await db.commit()
|
||
logger.info(f"用户 {user.user_id} 删除设置")
|
||
|
||
return {"message": "设置已删除", "user_id": user.user_id}
|
||
|
||
|
||
@router.get("/models")
|
||
async def get_available_models(
|
||
api_key: str,
|
||
api_base_url: str,
|
||
provider: str = "openai",
|
||
user: User = Depends(require_login)
|
||
):
|
||
"""
|
||
从配置的 API 获取可用的模型列表
|
||
|
||
Args:
|
||
api_key: API 密钥
|
||
api_base_url: API 基础 URL
|
||
provider: API 提供商 (openai, anthropic, azure, custom)
|
||
|
||
Returns:
|
||
模型列表
|
||
"""
|
||
try:
|
||
provider = normalize_provider(provider)
|
||
api_base_url = validate_public_http_url(api_base_url)
|
||
async with httpx.AsyncClient(timeout=10.0) as client:
|
||
if provider == "openai" or provider == "azure" or provider == "custom":
|
||
# OpenAI 兼容接口获取模型列表
|
||
url = f"{api_base_url.rstrip('/')}/models"
|
||
headers = {
|
||
"Authorization": f"Bearer {api_key}",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
logger.info(f"正在从 {url} 获取模型列表")
|
||
response = await client.get(url, headers=headers)
|
||
response.raise_for_status()
|
||
|
||
data = response.json()
|
||
models = []
|
||
|
||
if "data" in data and isinstance(data["data"], list):
|
||
for model in data["data"]:
|
||
model_id = model.get("id", "")
|
||
# 返回所有模型，不进行过滤
|
||
if model_id:
|
||
models.append({
|
||
"value": model_id,
|
||
"label": model_id,
|
||
"description": model.get("description", "") or f"Created: {model.get('created', 'N/A')}"
|
||
})
|
||
|
||
if not models:
|
||
raise HTTPException(
|
||
status_code=404,
|
||
detail="未能从 API 获取到可用的模型列表"
|
||
)
|
||
|
||
logger.info(f"成功获取 {len(models)} 个模型")
|
||
return {
|
||
"provider": provider,
|
||
"models": models,
|
||
"count": len(models)
|
||
}
|
||
|
||
elif provider == "anthropic":
|
||
# Anthropic models API
|
||
url = f"{api_base_url.rstrip('/')}/v1/models"
|
||
headers = {"x-api-key": api_key, "anthropic-version": "2023-06-01"}
|
||
response = await client.get(url, headers=headers)
|
||
response.raise_for_status()
|
||
data = response.json()
|
||
models = [{"value": m["id"], "label": m["id"], "description": m.get("display_name", "")} for m in data.get("data", [])]
|
||
return {"provider": provider, "models": models, "count": len(models)}
|
||
|
||
elif provider == "gemini":
|
||
# Gemini models API
|
||
url = f"{api_base_url.rstrip('/')}/models?key={api_key}"
|
||
response = await client.get(url)
|
||
response.raise_for_status()
|
||
data = response.json()
|
||
models = []
|
||
for m in data.get("models", []):
|
||
if "generateContent" in m.get("supportedGenerationMethods", []):
|
||
mid = m.get("name", "").replace("models/", "")
|
||
models.append({"value": mid, "label": m.get("displayName", mid), "description": ""})
|
||
return {"provider": provider, "models": models, "count": len(models)}
|
||
|
||
else:
|
||
raise HTTPException(status_code=400, detail=f"不支持的提供商: {provider}")
|
||
|
||
except httpx.HTTPStatusError as e:
|
||
logger.error(f"获取模型列表失败 (HTTP {e.response.status_code}): {e.response.text}")
|
||
if e.response.status_code == 404:
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"该 API 提供商不支持模型列表查询接口 (/models 返回 404)，请手动输入模型名称。当前请求地址: {api_base_url.rstrip('/')}/models"
|
||
)
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"无法从 API 获取模型列表 (HTTP {e.response.status_code})"
|
||
)
|
||
except httpx.RequestError as e:
|
||
logger.error(f"请求模型列表失败: {str(e)}")
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"无法连接到 API: {str(e)}"
|
||
)
|
||
except HTTPException:
|
||
raise
|
||
except Exception as e:
|
||
logger.error(f"获取模型列表时发生错误: {str(e)}")
|
||
raise HTTPException(
|
||
status_code=500,
|
||
detail=f"获取模型列表失败: {str(e)}"
|
||
)
|
||
|
||
|
||
class ApiTestRequest(BaseModel):
|
||
"""API 测试请求模型"""
|
||
api_key: str
|
||
api_base_url: str
|
||
provider: str
|
||
llm_model: str
|
||
temperature: Optional[float] = None
|
||
max_tokens: Optional[int] = None
|
||
|
||
|
||
@router.post("/check-function-calling")
|
||
async def check_function_calling_support(data: ApiTestRequest):
|
||
"""
|
||
检查模型是否支持 Function Calling（工具调用）
|
||
|
||
基于业界最佳实践的测试方法：
|
||
1. 发送包含工具定义的请求
|
||
2. 检查响应的 finish_reason 是否为 "tool_calls"
|
||
3. 验证响应中是否包含有效的 tool_calls 数据
|
||
|
||
Args:
|
||
data: 包含 API 配置的请求数据
|
||
|
||
Returns:
|
||
检测结果包含支持状态、详细信息和建议
|
||
"""
|
||
api_key = data.api_key
|
||
api_base_url = data.api_base_url
|
||
provider = normalize_provider(data.provider)
|
||
llm_model = data.llm_model
|
||
|
||
try:
|
||
start_time = time.time()
|
||
|
||
# 定义一个简单的测试工具（天气查询）
|
||
test_tools = [{
|
||
"type": "function",
|
||
"function": {
|
||
"name": "get_weather",
|
||
"description": "获取指定城市的当前天气信息",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"city": {
|
||
"type": "string",
|
||
"description": "城市名称，例如：北京、上海、深圳"
|
||
},
|
||
"unit": {
|
||
"type": "string",
|
||
"enum": ["celsius", "fahrenheit"],
|
||
"description": "温度单位"
|
||
}
|
||
},
|
||
"required": ["city"]
|
||
}
|
||
}
|
||
}]
|
||
|
||
# 测试提示：故意设计一个需要调用工具的问题
|
||
test_prompt = "请告诉我北京现在的天气情况如何？"
|
||
|
||
logger.info(f"🧪 开始检测 Function Calling 支持")
|
||
logger.info(f" - 提供商: {provider}")
|
||
logger.info(f" - 模型: {llm_model}")
|
||
logger.info(f" - 测试工具: get_weather")
|
||
|
||
# 创建临时 AI 服务实例进行测试
|
||
test_service = AIService(
|
||
api_provider=provider,
|
||
api_key=api_key,
|
||
api_base_url=api_base_url,
|
||
default_model=llm_model,
|
||
default_temperature=0.3, # 使用较低温度以获得更确定的行为
|
||
default_max_tokens=200
|
||
)
|
||
|
||
# 发送带工具的测试请求
|
||
response = await test_service.generate_text(
|
||
prompt=test_prompt,
|
||
provider=provider,
|
||
model=llm_model,
|
||
temperature=0.3,
|
||
max_tokens=200,
|
||
tools=test_tools,
|
||
tool_choice="auto", # 让模型自动决定是否使用工具
|
||
auto_mcp=False # 禁用 MCP 自动加载
|
||
)
|
||
|
||
end_time = time.time()
|
||
response_time = round((end_time - start_time) * 1000, 2)
|
||
|
||
# 分析响应以确定是否支持 Function Calling
|
||
supported = False
|
||
finish_reason = None
|
||
tool_calls = None
|
||
response_content = None
|
||
|
||
if isinstance(response, dict):
|
||
# 检查 finish_reason（OpenAI 标准）
|
||
finish_reason = response.get("finish_reason")
|
||
|
||
# 检查是否有 tool_calls
|
||
if "tool_calls" in response and response["tool_calls"]:
|
||
supported = True
|
||
tool_calls = response["tool_calls"]
|
||
logger.info(f"✅ 检测到工具调用: {len(tool_calls)} 个")
|
||
|
||
# 记录返回的内容（如果有）
|
||
if "content" in response:
|
||
response_content = response["content"]
|
||
elif isinstance(response, str):
|
||
# 如果只返回字符串，说明不支持工具调用
|
||
response_content = response
|
||
|
||
logger.info(f" - 响应时间: {response_time}ms")
|
||
logger.info(f" - finish_reason: {finish_reason}")
|
||
logger.info(f" - 支持状态: {'✅ 支持' if supported else '❌ 不支持'}")
|
||
|
||
# 构建详细的返回信息
|
||
result = {
|
||
"success": True,
|
||
"supported": supported,
|
||
"message": "✅ 模型支持 Function Calling" if supported else "❌ 模型不支持 Function Calling",
|
||
"response_time_ms": response_time,
|
||
"provider": provider,
|
||
"model": llm_model,
|
||
"details": {
|
||
"finish_reason": finish_reason,
|
||
"has_tool_calls": bool(tool_calls),
|
||
"tool_call_count": len(tool_calls) if tool_calls else 0,
|
||
"test_tool": "get_weather",
|
||
"test_prompt": test_prompt,
|
||
"response_type": "tool_calls" if supported else "text"
|
||
}
|
||
}
|
||
|
||
# 添加工具调用详情
|
||
if tool_calls:
|
||
result["tool_calls"] = tool_calls
|
||
result["suggestions"] = [
|
||
"✅ 该模型支持 Function Calling，可以正常使用 MCP 插件",
|
||
"建议：启用需要的 MCP 插件以扩展 AI 能力",
|
||
"提示：测试成功检测到工具调用，模型能够正确解析和使用外部工具"
|
||
]
|
||
else:
|
||
result["response_preview"] = response_content[:200] if response_content else None
|
||
result["suggestions"] = [
|
||
"❌ 该模型不支持 Function Calling，无法使用 MCP 插件功能",
|
||
"建议：更换支持工具调用的模型",
|
||
"推荐模型：GPT-4 系列、GPT-4-turbo、Claude 3 Opus/Sonnet、Gemini 1.5 Pro 等",
|
||
"说明：模型返回了文本回复而非工具调用，表明不支持该功能"
|
||
]
|
||
|
||
return result
|
||
|
||
except ValueError as e:
|
||
error_msg = str(e)
|
||
logger.error(f"❌ Function Calling 检测配置错误: {error_msg}")
|
||
return {
|
||
"success": False,
|
||
"supported": False,
|
||
"message": "配置错误",
|
||
"error": error_msg,
|
||
"error_type": "ConfigurationError",
|
||
"suggestions": [
|
||
"请检查 API Key 是否正确",
|
||
"请确认 API Base URL 格式是否正确",
|
||
"请验证所选提供商与配置是否匹配"
|
||
]
|
||
}
|
||
|
||
except TimeoutError as e:
|
||
error_msg = str(e)
|
||
logger.error(f"❌ Function Calling 检测超时: {error_msg}")
|
||
return {
|
||
"success": False,
|
||
"supported": None,
|
||
"message": "检测超时",
|
||
"error": error_msg,
|
||
"error_type": "TimeoutError",
|
||
"suggestions": [
|
||
"请检查网络连接是否正常",
|
||
"请确认 API 服务是否可访问",
|
||
"建议：稍后重试或使用其他网络环境"
|
||
]
|
||
}
|
||
|
||
except Exception as e:
|
||
error_msg = str(e)
|
||
error_type = type(e).__name__
|
||
|
||
logger.error(f"❌ Function Calling 检测失败: {error_msg}")
|
||
logger.error(f" - 错误类型: {error_type}")
|
||
|
||
# 智能分析错误原因
|
||
suggestions = []
|
||
if "tool" in error_msg.lower() or "function" in error_msg.lower():
|
||
suggestions = [
|
||
"该模型可能不支持 Function Calling 功能",
|
||
"API 返回了与工具调用相关的错误",
|
||
"建议：更换支持工具调用的模型或联系 API 提供商"
|
||
]
|
||
elif "unauthorized" in error_msg.lower() or "401" in error_msg:
|
||
suggestions = [
|
||
"API Key 认证失败",
|
||
"请检查 API Key 是否正确且有效",
|
||
"请确认 API Key 是否有足够的权限"
|
||
]
|
||
elif "not found" in error_msg.lower() or "404" in error_msg:
|
||
suggestions = [
|
||
"模型不存在或不可用",
|
||
"请检查模型名称是否正确",
|
||
"请确认该模型在当前 API 中是否可用"
|
||
]
|
||
else:
|
||
suggestions = [
|
||
"检测过程中遇到未知错误",
|
||
"建议：检查所有配置参数是否正确",
|
||
"提示：查看详细错误信息以获取更多线索"
|
||
]
|
||
|
||
return {
|
||
"success": False,
|
||
"supported": False,
|
||
"message": "Function Calling 检测失败",
|
||
"error": error_msg,
|
||
"error_type": error_type,
|
||
"suggestions": suggestions
|
||
}
|
||
|
||
|
||
@router.post("/test")
|
||
async def test_api_connection(data: ApiTestRequest):
|
||
"""
|
||
测试 API 连接和配置是否正确
|
||
|
||
Args:
|
||
data: 包含 API 配置的请求数据（包括 temperature 和 max_tokens）
|
||
|
||
Returns:
|
||
测试结果包含状态、响应时间和详细信息
|
||
"""
|
||
api_key = data.api_key
|
||
api_base_url = data.api_base_url
|
||
provider = normalize_provider(data.provider)
|
||
llm_model = data.llm_model
|
||
# 使用前端传递的参数，如果未传递则使用默认值
|
||
temperature = data.temperature if data.temperature is not None else 0.7
|
||
max_tokens = data.max_tokens if data.max_tokens is not None else 2000
|
||
import time
|
||
|
||
try:
|
||
start_time = time.time()
|
||
|
||
# 创建临时 AI 服务实例，使用前端传递的参数
|
||
test_service = AIService(
|
||
api_provider=provider,
|
||
api_key=api_key,
|
||
api_base_url=api_base_url,
|
||
default_model=llm_model,
|
||
default_temperature=temperature,
|
||
default_max_tokens=max_tokens
|
||
)
|
||
|
||
# 发送简单的测试请求
|
||
test_prompt = "请用一句话回复：测试成功"
|
||
|
||
logger.info(f"🧪 开始测试 API 连接")
|
||
logger.info(f" - 提供商: {provider}")
|
||
logger.info(f" - 模型: {llm_model}")
|
||
logger.info(f" - Base URL: {api_base_url}")
|
||
logger.info(f" - Temperature: {temperature}")
|
||
logger.info(f" - Max Tokens: {max_tokens}")
|
||
|
||
response = await test_service.generate_text(
|
||
prompt=test_prompt,
|
||
provider=provider,
|
||
model=llm_model,
|
||
temperature=temperature,
|
||
max_tokens=max_tokens,
|
||
auto_mcp=False # 测试时不加载MCP工具
|
||
)
|
||
|
||
end_time = time.time()
|
||
response_time = round((end_time - start_time) * 1000, 2) # 转换为毫秒
|
||
|
||
logger.info(f"✅ API 测试成功")
|
||
logger.info(f" - 响应时间: {response_time}ms")
|
||
|
||
# 安全地处理响应内容（确保是字符串）
|
||
response_str = str(response) if response else 'N/A'
|
||
logger.info(f" - 响应内容: {response_str[:100]}")
|
||
|
||
return {
|
||
"success": True,
|
||
"message": "API 连接测试成功",
|
||
"response_time_ms": response_time,
|
||
"provider": provider,
|
||
"model": llm_model,
|
||
"response_preview": response_str[:100] if len(response_str) > 100 else response_str,
|
||
"details": {
|
||
"api_available": True,
|
||
"model_accessible": True,
|
||
"response_valid": bool(response),
|
||
"temperature": temperature,
|
||
"max_tokens": max_tokens
|
||
}
|
||
}
|
||
|
||
except ValueError as e:
|
||
# 配置错误
|
||
error_msg = str(e)
|
||
logger.error(f"❌ API 配置错误: {error_msg}")
|
||
return {
|
||
"success": False,
|
||
"message": "API 配置错误",
|
||
"error": error_msg,
|
||
"error_type": "ConfigurationError",
|
||
"suggestions": [
|
||
"请检查 API Key 是否正确",
|
||
"请确认 API Base URL 格式正确",
|
||
"请验证所选提供商是否匹配"
|
||
]
|
||
}
|
||
|
||
except TimeoutError as e:
|
||
# 超时错误
|
||
error_msg = str(e)
|
||
logger.error(f"❌ API 请求超时: {error_msg}")
|
||
return {
|
||
"success": False,
|
||
"message": "API 请求超时",
|
||
"error": error_msg,
|
||
"error_type": "TimeoutError",
|
||
"suggestions": [
|
||
"请检查网络连接",
|
||
"请确认 API Base URL 是否可访问",
|
||
"如果使用代理，请检查代理设置"
|
||
]
|
||
}
|
||
|
||
except Exception as e:
|
||
# 其他错误
|
||
error_msg = str(e)
|
||
error_type = type(e).__name__
|
||
|
||
logger.error(f"❌ API 测试失败: {error_msg}")
|
||
logger.error(f" - 错误类型: {error_type}")
|
||
|
||
# 分析错误原因并提供建议
|
||
suggestions = []
|
||
if "blocked" in error_msg.lower():
|
||
suggestions = [
|
||
"请求被 API 提供商阻止",
|
||
"可能原因：API Key 被限制或地区限制",
|
||
"建议：检查 API Key 状态和账户余额",
|
||
"建议：尝试更换 API Base URL 或使用代理"
|
||
]
|
||
elif "unauthorized" in error_msg.lower() or "401" in error_msg:
|
||
suggestions = [
|
||
"API Key 认证失败",
|
||
"建议：检查 API Key 是否正确",
|
||
"建议：确认 API Key 是否过期"
|
||
]
|
||
elif "not found" in error_msg.lower() or "404" in error_msg:
|
||
suggestions = [
|
||
"API 端点不存在或模型不可用",
|
||
"建议：检查 API Base URL 是否正确",
|
||
"建议：确认模型名称是否正确"
|
||
]
|
||
elif "rate limit" in error_msg.lower() or "429" in error_msg:
|
||
suggestions = [
|
||
"API 请求频率超限",
|
||
"建议：稍后重试",
|
||
"建议：升级 API 套餐"
|
||
]
|
||
elif "insufficient" in error_msg.lower() or "quota" in error_msg.lower():
|
||
suggestions = [
|
||
"API 配额不足",
|
||
"建议：检查账户余额",
|
||
"建议：充值或升级套餐"
|
||
]
|
||
else:
|
||
suggestions = [
|
||
"请检查所有配置参数是否正确",
|
||
"请确认网络连接正常",
|
||
"请查看详细错误信息"
|
||
]
|
||
|
||
return {
|
||
"success": False,
|
||
"message": "API 测试失败",
|
||
"error": error_msg,
|
||
"error_type": error_type,
|
||
"suggestions": suggestions
|
||
}
|
||
|
||
|
||
# ========== API配置预设管理（零数据库改动方案）==========
|
||
|
||
async def get_user_settings(user_id: str, db: AsyncSession) -> Settings:
|
||
"""获取用户settings，如果不存在则创建"""
|
||
result = await db.execute(
|
||
select(Settings).where(Settings.user_id == user_id)
|
||
)
|
||
settings = result.scalar_one_or_none()
|
||
|
||
if not settings:
|
||
# 创建默认设置
|
||
env_defaults = read_env_defaults()
|
||
settings = Settings(
|
||
user_id=user_id,
|
||
**env_defaults,
|
||
preferences='{}' # 初始化为空JSON
|
||
)
|
||
db.add(settings)
|
||
await db.commit()
|
||
await db.refresh(settings)
|
||
logger.info(f"用户 {user_id} 首次访问，已创建默认设置")
|
||
|
||
return settings
|
||
|
||
|
||
@router.get("/presets", response_model=PresetListResponse)
|
||
async def get_presets(
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
获取所有API配置预设
|
||
|
||
从preferences字段读取预设列表
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 解析preferences
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
except json.JSONDecodeError:
|
||
logger.warning(f"用户 {user.user_id} 的preferences字段JSON格式错误，重置为空")
|
||
prefs = {}
|
||
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 找到激活的预设
|
||
active_preset_id = next(
|
||
(p['id'] for p in presets if p.get('is_active')),
|
||
None
|
||
)
|
||
|
||
logger.info(f"用户 {user.user_id} 获取预设列表，共 {len(presets)} 个")
|
||
|
||
return {
|
||
"presets": presets,
|
||
"total": len(presets),
|
||
"active_preset_id": active_preset_id
|
||
}
|
||
|
||
|
||
@router.post("/presets", response_model=PresetResponse)
|
||
async def create_preset(
|
||
data: PresetCreateRequest,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
创建新预设
|
||
|
||
将预设添加到preferences字段的JSON中
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 解析preferences
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
except json.JSONDecodeError:
|
||
prefs = {}
|
||
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 创建新预设
|
||
new_preset = {
|
||
"id": f"preset_{int(datetime.now().timestamp() * 1000)}",
|
||
"name": data.name,
|
||
"description": data.description,
|
||
"is_active": False,
|
||
"created_at": datetime.now().isoformat(),
|
||
"config": {
|
||
**data.config.model_dump(),
|
||
"api_provider": normalize_provider(data.config.api_provider)
|
||
}
|
||
}
|
||
|
||
presets.append(new_preset)
|
||
|
||
# 保存回preferences
|
||
api_presets['presets'] = presets
|
||
prefs['api_presets'] = api_presets
|
||
settings.preferences = json.dumps(prefs, ensure_ascii=False)
|
||
|
||
await db.commit()
|
||
|
||
logger.info(f"用户 {user.user_id} 创建预设: {data.name}")
|
||
return new_preset
|
||
|
||
|
||
@router.put("/presets/{preset_id}", response_model=PresetResponse)
|
||
async def update_preset(
|
||
preset_id: str,
|
||
data: PresetUpdateRequest,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
更新预设
|
||
|
||
在preferences字段的JSON中更新指定预设
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 解析preferences
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
except json.JSONDecodeError:
|
||
raise HTTPException(status_code=500, detail="配置数据格式错误")
|
||
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 找到并更新预设
|
||
target_preset = next((p for p in presets if p['id'] == preset_id), None)
|
||
if not target_preset:
|
||
raise HTTPException(status_code=404, detail="预设不存在")
|
||
|
||
# 更新字段
|
||
if data.name is not None:
|
||
target_preset['name'] = data.name
|
||
if data.description is not None:
|
||
target_preset['description'] = data.description
|
||
if data.config is not None:
|
||
target_preset['config'] = {
|
||
**data.config.model_dump(),
|
||
'api_provider': normalize_provider(data.config.api_provider)
|
||
}
|
||
|
||
# 保存回preferences
|
||
prefs['api_presets'] = api_presets
|
||
settings.preferences = json.dumps(prefs, ensure_ascii=False)
|
||
|
||
await db.commit()
|
||
|
||
logger.info(f"用户 {user.user_id} 更新预设: {preset_id}")
|
||
return target_preset
|
||
|
||
|
||
@router.delete("/presets/{preset_id}")
|
||
async def delete_preset(
|
||
preset_id: str,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
删除预设
|
||
|
||
从preferences字段的JSON中删除指定预设
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 解析preferences
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
except json.JSONDecodeError:
|
||
raise HTTPException(status_code=500, detail="配置数据格式错误")
|
||
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 找到预设
|
||
target_preset = next((p for p in presets if p['id'] == preset_id), None)
|
||
if not target_preset:
|
||
raise HTTPException(status_code=404, detail="预设不存在")
|
||
|
||
# 检查是否是激活的预设
|
||
if target_preset.get('is_active'):
|
||
raise HTTPException(status_code=400, detail="无法删除激活中的预设，请先激活其他预设")
|
||
|
||
# 删除预设
|
||
presets = [p for p in presets if p['id'] != preset_id]
|
||
|
||
# 保存回preferences
|
||
api_presets['presets'] = presets
|
||
prefs['api_presets'] = api_presets
|
||
settings.preferences = json.dumps(prefs, ensure_ascii=False)
|
||
|
||
await db.commit()
|
||
|
||
logger.info(f"用户 {user.user_id} 删除预设: {preset_id}")
|
||
return {"message": "预设已删除", "preset_id": preset_id}
|
||
|
||
|
||
@router.post("/presets/{preset_id}/activate")
|
||
async def activate_preset(
|
||
preset_id: str,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
激活预设
|
||
|
||
将预设的配置应用到Settings主字段
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 解析preferences
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
except json.JSONDecodeError:
|
||
raise HTTPException(status_code=500, detail="配置数据格式错误")
|
||
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 找到目标预设
|
||
target_preset = next((p for p in presets if p['id'] == preset_id), None)
|
||
if not target_preset:
|
||
raise HTTPException(status_code=404, detail="预设不存在")
|
||
|
||
# 应用配置到Settings主字段
|
||
config = target_preset['config']
|
||
settings.api_provider = normalize_provider(config['api_provider'])
|
||
settings.api_key = config['api_key']
|
||
settings.api_base_url = config.get('api_base_url')
|
||
settings.llm_model = config['llm_model']
|
||
settings.temperature = config['temperature']
|
||
settings.max_tokens = config['max_tokens']
|
||
settings.system_prompt = config.get('system_prompt')
|
||
|
||
# 更新所有预设的is_active状态
|
||
for preset in presets:
|
||
preset['is_active'] = (preset['id'] == preset_id)
|
||
|
||
# 保存回preferences
|
||
prefs['api_presets'] = api_presets
|
||
settings.preferences = json.dumps(prefs, ensure_ascii=False)
|
||
|
||
await db.commit()
|
||
|
||
logger.info(f"用户 {user.user_id} 激活预设: {target_preset['name']}")
|
||
return {
|
||
"message": "预设已激活",
|
||
"preset_id": preset_id,
|
||
"preset_name": target_preset['name']
|
||
}
|
||
|
||
|
||
@router.post("/presets/{preset_id}/test")
|
||
async def test_preset(
|
||
preset_id: str,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
测试预设的API连接
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 解析preferences
|
||
try:
|
||
prefs = json.loads(settings.preferences or '{}')
|
||
except json.JSONDecodeError:
|
||
raise HTTPException(status_code=500, detail="配置数据格式错误")
|
||
|
||
api_presets = prefs.get('api_presets', {'presets': [], 'version': '1.0'})
|
||
presets = api_presets.get('presets', [])
|
||
|
||
# 找到预设
|
||
target_preset = next((p for p in presets if p['id'] == preset_id), None)
|
||
if not target_preset:
|
||
raise HTTPException(status_code=404, detail="预设不存在")
|
||
|
||
# 使用现有的test_api_connection逻辑
|
||
# 确保传递完整参数，与当前配置测试保持一致
|
||
config = target_preset['config']
|
||
test_request = ApiTestRequest(
|
||
api_key=config['api_key'],
|
||
api_base_url=config.get('api_base_url', ''),
|
||
provider=config['api_provider'],
|
||
llm_model=config['llm_model'],
|
||
temperature=config.get('temperature'), # 使用预设中的温度参数
|
||
max_tokens=config.get('max_tokens') # 使用预设中的最大tokens参数
|
||
)
|
||
|
||
logger.info(f"用户 {user.user_id} 测试预设: {target_preset['name']}")
|
||
return await test_api_connection(test_request)
|
||
|
||
|
||
@router.post("/presets/from-current", response_model=PresetResponse)
|
||
async def create_preset_from_current(
|
||
name: str,
|
||
description: Optional[str] = None,
|
||
user: User = Depends(require_login),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""
|
||
从当前配置创建新预设
|
||
|
||
快捷方式：将当前激活的配置保存为新预设
|
||
"""
|
||
settings = await get_user_settings(user.user_id, db)
|
||
|
||
# 从当前Settings主字段读取配置
|
||
current_config = APIKeyPresetConfig(
|
||
api_provider=normalize_provider(settings.api_provider),
|
||
api_key=settings.api_key,
|
||
api_base_url=settings.api_base_url,
|
||
llm_model=settings.llm_model,
|
||
temperature=settings.temperature,
|
||
max_tokens=settings.max_tokens,
|
||
system_prompt=settings.system_prompt
|
||
)
|
||
|
||
# 创建预设
|
||
create_request = PresetCreateRequest(
|
||
name=name,
|
||
description=description,
|
||
config=current_config
|
||
)
|
||
|
||
logger.info(f"用户 {user.user_id} 从当前配置创建预设: {name}")
|
||
return await create_preset(create_request, user, db)
|