[Partner Nodes] new Flux2ImageNode and GrokImageEditNodeV2 nodes

[bigcat88]

Nothing special, just rewritten nodes to use DynamicCombo and Autogrow

API Node PR Checklist

Scope

• Is API Node Change

Pricing & Billing

• Need pricing update
• No pricing update

If Need pricing update:

• Metronome rate cards updated
• Auto‑billing tests updated and passing

QA

• QA done
• QA not required

Comms

• Informed Kosinkadink

[coderabbitai]

📝 Walkthrough

Walkthrough

This PR introduces two upgraded node variants with dynamic model selection. Flux2ImageNode unifies Flux.2 [pro] and [max] generation with model-specific API endpoints, shared dimension inputs, and reference image support (up to 8 images, flattened from batches and encoded as base64). GrokImageEditNodeV2 replaces static input schema with model-driven dynamic inputs, validates image counts per model constraints (pro: 1 max, others: 3 max), and handles aspect-ratio rules. Both new nodes are registered in their respective extensions, and their legacy predecessors are marked deprecated.

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 0.00% which is insufficient. The required threshold is 80.00%.	Write docstrings for the functions missing them to satisfy the coverage threshold.

✅ Passed checks (4 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title clearly and specifically identifies the two new nodes being added (Flux2ImageNode and GrokImageEditNodeV2) and their purpose as partner nodes.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Description check	✅ Passed	The pull request description is related to the changeset, mentioning rewritten nodes using DynamicCombo and Autogrow, which aligns with the addition of Flux2ImageNode and GrokImageEditNodeV2.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

Tip

💬 Introducing Slack Agent: The best way for teams to turn conversations into code.

Slack Agent is built on CodeRabbit's deep understanding of your code, so your team can collaborate across the entire SDLC without losing context.

• Generate code and open pull requests
• Plan features and break down work
• Investigate incidents and troubleshoot customer tickets together
• Automate recurring tasks and respond to alerts with triggers
• Summarize progress and report instantly

Built for teams:

• Shared memory across your entire org—no repeating context
• Per-thread sandboxes to safely plan and execute work
• Governance built-in—scoped access, auditability, and budget controls

One agent for your entire SDLC. Right inside Slack.

👉 Get started

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[coderabbitai]

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents

Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@comfy_api_nodes/nodes_bfl.py`:
- Around line 735-743: Restore the removed "prompt_upsampling" input on the
unified Flux2 node and ensure it is propagated into the request object: add an
IO input named "prompt_upsampling" (matching the old behavior) back into
Flux2ImageNode's inputs and include that value when constructing the
Flux2ProGenerateRequest so the request field is set the same way as the
deprecated Flux2 nodes; update any related serialization or default handling
where Flux2ProGenerateRequest is built to pass the prompt_upsampling parameter
through.

🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

• Push a commit to this branch (recommended)
• Create a new PR with the fixes

---

ℹ️ Review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

Run ID: 6f99eabd-bd46-46cc-b955-ed01b464f133

📥 Commits

Reviewing files that changed from the base of the PR and between 52976f3 and 33ed6e6.

📒 Files selected for processing (2)

• comfy_api_nodes/nodes_bfl.py
• comfy_api_nodes/nodes_grok.py