fix: quiet subject agent model warnings

This commit is contained in:
2026-05-20 13:07:58 +08:00
parent 83b151df04
commit e10b1a6e1a
2 changed files with 22 additions and 14 deletions

View File

@@ -23,7 +23,7 @@ from dotenv import load_dotenv
from fastapi import BackgroundTasks, FastAPI, File, Form, HTTPException, Request, Response, UploadFile
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse
from pydantic import BaseModel, Field
from pydantic import BaseModel, ConfigDict, Field
load_dotenv()
@@ -738,6 +738,8 @@ class AudioScript(BaseModel):
class SubjectAgentAnalysis(BaseModel):
model_config = ConfigDict(protected_namespaces=())
model_bundle: SubjectModelBundle = "gpt"
model: str = ""
source_frame_indices: list[int] = Field(default_factory=list)
@@ -758,6 +760,8 @@ class SubjectAgentMessage(BaseModel):
class SubjectAgentState(BaseModel):
model_config = ConfigDict(protected_namespaces=())
model_bundle: SubjectModelBundle = "gpt"
source_frame_indices: list[int] = Field(default_factory=list)
analysis: SubjectAgentAnalysis | None = None
@@ -4164,11 +4168,15 @@ class CreateJobReq(BaseModel):
class SubjectAgentAnalyzeReq(BaseModel):
model_config = ConfigDict(protected_namespaces=())
model_bundle: SubjectModelBundle = "gpt"
source_frame_indices: list[int] = Field(default_factory=list)
class SubjectAgentMessageReq(BaseModel):
model_config = ConfigDict(protected_namespaces=())
model_bundle: SubjectModelBundle = "gpt"
source_frame_indices: list[int] = Field(default_factory=list)
selected_mode: Literal["realistic", "cartoon", "elements", "custom"] = "custom"