接入真实模型调用与参考文件上传
This commit is contained in:
@@ -5,6 +5,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from database import get_db
|
||||
from models.ai_model import AiModel
|
||||
from schemas.schemas import AiModelCreate, AiModelUpdate, Response
|
||||
from services.ai_service import call_ai
|
||||
from services.security import decrypt_text, encrypt_text, mask_secret
|
||||
|
||||
router = APIRouter()
|
||||
@@ -88,10 +89,26 @@ async def test_model(model_id: int, db: AsyncSession = Depends(get_db)):
|
||||
if model is None:
|
||||
raise HTTPException(status_code=404, detail="模型不存在")
|
||||
|
||||
return Response(
|
||||
data={
|
||||
"id": model.id,
|
||||
"success": True,
|
||||
"message": f"模型 {model.name} 配置校验通过(当前为本地模拟测试)",
|
||||
}
|
||||
)
|
||||
class FakeParagraph:
|
||||
title = "连接测试"
|
||||
content = "请返回一段非常简短的测试文本。"
|
||||
need_prompt = False
|
||||
prompt_text = ""
|
||||
output_format = "text"
|
||||
|
||||
try:
|
||||
result = await call_ai(FakeParagraph(), model)
|
||||
return Response(
|
||||
data={
|
||||
"id": model.id,
|
||||
"success": True,
|
||||
"message": f"模型 {model.name} 连接测试成功",
|
||||
"preview": result.content,
|
||||
}
|
||||
)
|
||||
except Exception as error:
|
||||
return Response(
|
||||
code=-1,
|
||||
message=str(error),
|
||||
data={"id": model.id, "success": False},
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user