Skip to content

Commit 6dadfa2

Browse files
committed
convert BFL API nodes to use new API client; remove deprecated BFL nodes
1 parent fab58dd commit 6dadfa2

File tree

2 files changed

+152
-505
lines changed

2 files changed

+152
-505
lines changed

comfy_api_nodes/apis/bfl_api.py

Lines changed: 3 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -50,44 +50,6 @@ class BFLFluxFillImageRequest(BaseModel):
5050
mask: str = Field(None, description='A Base64-encoded string representing the mask of the areas you with to modify.')
5151

5252

53-
class BFLFluxCannyImageRequest(BaseModel):
54-
prompt: str = Field(..., description='Text prompt for image generation')
55-
prompt_upsampling: Optional[bool] = Field(
56-
None, description='Whether to perform upsampling on the prompt. If active, automatically modifies the prompt for more creative generation.'
57-
)
58-
canny_low_threshold: Optional[int] = Field(None, description='Low threshold for Canny edge detection')
59-
canny_high_threshold: Optional[int] = Field(None, description='High threshold for Canny edge detection')
60-
seed: Optional[int] = Field(None, description='The seed value for reproducibility.')
61-
steps: conint(ge=15, le=50) = Field(..., description='Number of steps for the image generation process')
62-
guidance: confloat(ge=1, le=100) = Field(..., description='Guidance strength for the image generation process')
63-
safety_tolerance: Optional[conint(ge=0, le=6)] = Field(
64-
6, description='Tolerance level for input and output moderation. Between 0 and 6, 0 being most strict, 6 being least strict. Defaults to 2.'
65-
)
66-
output_format: Optional[BFLOutputFormat] = Field(
67-
BFLOutputFormat.png, description="Output format for the generated image. Can be 'jpeg' or 'png'.", examples=['png']
68-
)
69-
control_image: Optional[str] = Field(None, description='Base64 encoded image to use as control input if no preprocessed image is provided')
70-
preprocessed_image: Optional[str] = Field(None, description='Optional pre-processed image that will bypass the control preprocessing step')
71-
72-
73-
class BFLFluxDepthImageRequest(BaseModel):
74-
prompt: str = Field(..., description='Text prompt for image generation')
75-
prompt_upsampling: Optional[bool] = Field(
76-
None, description='Whether to perform upsampling on the prompt. If active, automatically modifies the prompt for more creative generation.'
77-
)
78-
seed: Optional[int] = Field(None, description='The seed value for reproducibility.')
79-
steps: conint(ge=15, le=50) = Field(..., description='Number of steps for the image generation process')
80-
guidance: confloat(ge=1, le=100) = Field(..., description='Guidance strength for the image generation process')
81-
safety_tolerance: Optional[conint(ge=0, le=6)] = Field(
82-
6, description='Tolerance level for input and output moderation. Between 0 and 6, 0 being most strict, 6 being least strict. Defaults to 2.'
83-
)
84-
output_format: Optional[BFLOutputFormat] = Field(
85-
BFLOutputFormat.png, description="Output format for the generated image. Can be 'jpeg' or 'png'.", examples=['png']
86-
)
87-
control_image: Optional[str] = Field(None, description='Base64 encoded image to use as control input if no preprocessed image is provided')
88-
preprocessed_image: Optional[str] = Field(None, description='Optional pre-processed image that will bypass the control preprocessing step')
89-
90-
9153
class BFLFluxProGenerateRequest(BaseModel):
9254
prompt: str = Field(..., description='The text prompt for image generation.')
9355
prompt_upsampling: Optional[bool] = Field(
@@ -160,15 +122,8 @@ class BFLStatus(str, Enum):
160122
error = "Error"
161123

162124

163-
class BFLFluxProStatusResponse(BaseModel):
125+
class BFLFluxStatusResponse(BaseModel):
164126
id: str = Field(..., description="The unique identifier for the generation task.")
165127
status: BFLStatus = Field(..., description="The status of the task.")
166-
result: Optional[Dict[str, Any]] = Field(
167-
None, description="The result of the task (null if not completed)."
168-
)
169-
progress: confloat(ge=0.0, le=1.0) = Field(
170-
..., description="The progress of the task (0.0 to 1.0)."
171-
)
172-
details: Optional[Dict[str, Any]] = Field(
173-
None, description="Additional details about the task (null if not available)."
174-
)
128+
result: Optional[Dict[str, Any]] = Field(None, description="The result of the task (null if not completed).")
129+
progress: Optional[float] = Field(None, description="The progress of the task (0.0 to 1.0).", ge=0.0, le=1.0)

0 commit comments

Comments
 (0)