[diffusion] CI: support returning request id from endpoint (#15844)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2025-12-26 23:38:03 +08:00
parent 51dbdb2202
commit b70914969b
8 changed files with 50 additions and 30 deletions
@@ -12,6 +12,7 @@ class ImageResponseData(BaseModel):


 class ImageResponse(BaseModel):
+    id: str
    created: int = Field(default_factory=lambda: int(time.time()))
    data: List[ImageResponseData]
    peak_memory_mb: Optional[float] = None