From 8b1bf16e5f01d11a808405323ac9fa372cf8a980 Mon Sep 17 00:00:00 2001 From: Adir Amsalem Date: Sun, 5 Apr 2026 10:59:58 +0300 Subject: [PATCH] fix: add reference_image to FILE_FIELDS for consistent file handling --- decart/client.py | 2 +- decart/queue/client.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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}