Skip to content

Commit fe0324f

Browse files
authored
fix: add reference_image to FILE_FIELDS for proper file handling (#44)
fix: add reference_image to FILE_FIELDS for consistent file handling
1 parent 37dbaa2 commit fe0324f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

decart/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ async def process(self, options: dict[str, Any]) -> bytes:
175175
inputs = {k: v for k, v in options.items() if k not in ("model", "cancel_token")}
176176

177177
# File fields that need special handling (not validated by Pydantic)
178-
FILE_FIELDS = {"data", "start", "end"}
178+
FILE_FIELDS = {"data", "start", "end", "reference_image"}
179179

180180
# Separate file inputs from regular inputs
181181
file_inputs = {k: v for k, v in inputs.items() if k in FILE_FIELDS}

decart/queue/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ async def submit(self, options: dict[str, Any]) -> JobSubmitResponse:
9595
inputs = {k: v for k, v in options.items() if k not in ("model", "cancel_token")}
9696

9797
# File fields that need special handling
98-
FILE_FIELDS = {"data", "start", "end"}
98+
FILE_FIELDS = {"data", "start", "end", "reference_image"}
9999

100100
# Separate file inputs from regular inputs
101101
file_inputs = {k: v for k, v in inputs.items() if k in FILE_FIELDS}

0 commit comments

Comments
 (0)