grok-api / models.py
hf1024's picture
Upload 8 files
6c92e4f verified
Raw
History Blame Contribute Delete
675 Bytes
# !/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")