29 lines
478 B
Python
29 lines
478 B
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class BBox(BaseModel):
|
||
|
|
x: int
|
||
|
|
y: int
|
||
|
|
w: int
|
||
|
|
h: int
|
||
|
|
|
||
|
|
|
||
|
|
class QuadrupleItem(BaseModel):
|
||
|
|
subject: str
|
||
|
|
predicate: str
|
||
|
|
object: str
|
||
|
|
qualifier: str | None = None
|
||
|
|
bbox: BBox
|
||
|
|
cropped_image_path: str
|
||
|
|
|
||
|
|
|
||
|
|
class ImageExtractRequest(BaseModel):
|
||
|
|
file_path: str
|
||
|
|
task_id: int
|
||
|
|
model: str | None = None
|
||
|
|
prompt_template: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class ImageExtractResponse(BaseModel):
|
||
|
|
items: list[QuadrupleItem]
|