diff --git a/decart/client.py b/decart/client.py index 10e0fad..9f02b95 100644 --- a/decart/client.py +++ b/decart/client.py @@ -175,7 +175,7 @@ async def process(self, options: dict[str, Any]) -> bytes: inputs = {k: v for k, v in options.items() if k not in ("model", "cancel_token")} # File fields that need special handling (not validated by Pydantic) - FILE_FIELDS = {"data", "start", "end"} + FILE_FIELDS = {"data", "start", "end", "reference_image"} # Separate file inputs from regular inputs file_inputs = {k: v for k, v in inputs.items() if k in FILE_FIELDS} diff --git a/decart/queue/client.py b/decart/queue/client.py index f85686b..43fb1c5 100644 --- a/decart/queue/client.py +++ b/decart/queue/client.py @@ -95,7 +95,7 @@ async def submit(self, options: dict[str, Any]) -> JobSubmitResponse: inputs = {k: v for k, v in options.items() if k not in ("model", "cancel_token")} # File fields that need special handling - FILE_FIELDS = {"data", "start", "end"} + FILE_FIELDS = {"data", "start", "end", "reference_image"} # Separate file inputs from regular inputs file_inputs = {k: v for k, v in inputs.items() if k in FILE_FIELDS}