# !/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")