File size: 675 Bytes
6c92e4f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | # !/usr/bin/python3
# -*- coding: utf-8 -*-
"""
@Author : Ailove
------------------------------------
@File : models.py
@CreateTime : 2025/5/22 20:36
------------------------------------
"""
from pydantic import BaseModel, Field
class ImageRequest(BaseModel):
url: str = None
class ImageCompareRequest(BaseModel):
img_1: str = None
url_1: str = None
img_2: str = None
url_2: str = None
class QuestionRequest(BaseModel):
text: str = Field(..., description="用户提出的问题")
end_flag: bool = True
streaming: bool = False
model: str = Field(..., description="要使用的模型ID")
|