fix: quiet subject agent model warnings
This commit is contained in:
10
api/main.py
10
api/main.py
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user