mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-30 19:07:25 +08:00
Merge branch 'master' into alexis/utils
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Build package / Build Test (3.10) (push) Has been cancelled
Build package / Build Test (3.11) (push) Has been cancelled
Build package / Build Test (3.12) (push) Has been cancelled
Build package / Build Test (3.13) (push) Has been cancelled
Build package / Build Test (3.14) (push) Has been cancelled
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Build package / Build Test (3.10) (push) Has been cancelled
Build package / Build Test (3.11) (push) Has been cancelled
Build package / Build Test (3.12) (push) Has been cancelled
Build package / Build Test (3.13) (push) Has been cancelled
Build package / Build Test (3.14) (push) Has been cancelled
This commit is contained in:
commit
18c240e038
484
.github/workflows/backport_release.yaml
vendored
Normal file
484
.github/workflows/backport_release.yaml
vendored
Normal file
@ -0,0 +1,484 @@
|
|||||||
|
name: Backport Release
|
||||||
|
|
||||||
|
on:
|
||||||
|
workflow_dispatch:
|
||||||
|
inputs:
|
||||||
|
commit:
|
||||||
|
description: 'Full 40-char SHA of the tip commit of the backport source branch (the PR head commit that passed tests). The branch is resolved from this SHA and must be unique.'
|
||||||
|
required: true
|
||||||
|
type: string
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
contents: read
|
||||||
|
pull-requests: read
|
||||||
|
checks: read
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
backport-release:
|
||||||
|
name: Create backport release
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
environment: backport release
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Generate GitHub App token
|
||||||
|
id: app-token
|
||||||
|
uses: actions/create-github-app-token@bcd2ba49218906704ab6c1aa796996da409d3eb1
|
||||||
|
with:
|
||||||
|
app-id: ${{ secrets.FEN_RELEASE_APP_ID }}
|
||||||
|
private-key: ${{ secrets.FEN_RELEASE_PRIVATE_KEY }}
|
||||||
|
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd
|
||||||
|
with:
|
||||||
|
token: ${{ steps.app-token.outputs.token }}
|
||||||
|
fetch-depth: 0
|
||||||
|
fetch-tags: true
|
||||||
|
|
||||||
|
- name: Configure git
|
||||||
|
run: |
|
||||||
|
git config user.name "fen-release[bot]"
|
||||||
|
git config user.email "fen-release[bot]@users.noreply.github.com"
|
||||||
|
|
||||||
|
- name: Resolve source branch from commit SHA
|
||||||
|
id: resolve
|
||||||
|
env:
|
||||||
|
SOURCE_COMMIT: ${{ inputs.commit }}
|
||||||
|
DEFAULT_BRANCH: ${{ github.event.repository.default_branch }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Require a full 40-char lowercase-hex SHA. Short SHAs are ambiguous
|
||||||
|
# and we will be comparing this value against API responses (PR head
|
||||||
|
# SHA, ref tips) that always return the full form.
|
||||||
|
if [[ ! "${SOURCE_COMMIT}" =~ ^[0-9a-f]{40}$ ]]; then
|
||||||
|
echo "::error::Input commit '${SOURCE_COMMIT}' is not a full 40-char lowercase hex SHA."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Fetch all remote branches so we can search for which one(s) point
|
||||||
|
# at this SHA. `actions/checkout` with fetch-depth: 0 fetches full
|
||||||
|
# history of the checked-out ref but does not necessarily populate
|
||||||
|
# every refs/remotes/origin/*, so do it explicitly.
|
||||||
|
git fetch --prune origin '+refs/heads/*:refs/remotes/origin/*'
|
||||||
|
|
||||||
|
# Verify the commit actually exists in this repo's object DB.
|
||||||
|
if ! git cat-file -e "${SOURCE_COMMIT}^{commit}" 2>/dev/null; then
|
||||||
|
echo "::error::Commit ${SOURCE_COMMIT} was not found in the repository."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Find every remote branch whose tip == SOURCE_COMMIT. Exactly one
|
||||||
|
# branch must point at it. If zero, the commit isn't anyone's tip
|
||||||
|
# (likely stale, force-pushed past, or never the PR head). If more
|
||||||
|
# than one, the (branch -> SHA) mapping is ambiguous and we refuse
|
||||||
|
# to guess — the operator must give us a unique branch to release.
|
||||||
|
mapfile -t matching_branches < <(
|
||||||
|
git for-each-ref \
|
||||||
|
--format='%(refname:strip=3)' \
|
||||||
|
--points-at="${SOURCE_COMMIT}" \
|
||||||
|
refs/remotes/origin/ \
|
||||||
|
| grep -vx 'HEAD' || true
|
||||||
|
)
|
||||||
|
|
||||||
|
if [[ "${#matching_branches[@]}" -eq 0 ]]; then
|
||||||
|
echo "::error::No branch on origin has ${SOURCE_COMMIT} as its tip."
|
||||||
|
echo "::error::Either the branch was updated after you copied this SHA, or this commit was never the head of a branch."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "${#matching_branches[@]}" -gt 1 ]]; then
|
||||||
|
echo "::error::More than one branch on origin has ${SOURCE_COMMIT} as its tip; cannot pick one:"
|
||||||
|
for b in "${matching_branches[@]}"; do
|
||||||
|
echo "::error:: - ${b}"
|
||||||
|
done
|
||||||
|
echo "::error::Refusing to proceed with an ambiguous source branch."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
source_branch="${matching_branches[0]}"
|
||||||
|
|
||||||
|
if [[ "${source_branch}" == "${DEFAULT_BRANCH}" ]]; then
|
||||||
|
echo "::error::Source branch must not be the default branch ('${DEFAULT_BRANCH}')."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Resolved commit ${SOURCE_COMMIT} to branch '${source_branch}'."
|
||||||
|
echo "source_branch=${source_branch}" >> "$GITHUB_OUTPUT"
|
||||||
|
|
||||||
|
- name: Determine latest stable release
|
||||||
|
id: latest
|
||||||
|
env:
|
||||||
|
GH_TOKEN: ${{ steps.app-token.outputs.token }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# List all tags matching vMAJOR.MINOR.PATCH and pick the highest by numeric
|
||||||
|
# comparison of each component. We DO NOT use `sort -V` because it treats
|
||||||
|
# v0.19.99 as higher than v0.20.1.
|
||||||
|
latest_tag="$(
|
||||||
|
git tag --list 'v[0-9]*.[0-9]*.[0-9]*' \
|
||||||
|
| grep -E '^v[0-9]+\.[0-9]+\.[0-9]+$' \
|
||||||
|
| awk -F'[v.]' '{ printf "%010d %010d %010d %s\n", $2, $3, $4, $0 }' \
|
||||||
|
| sort -k1,1n -k2,2n -k3,3n \
|
||||||
|
| tail -n1 \
|
||||||
|
| awk '{print $4}'
|
||||||
|
)"
|
||||||
|
|
||||||
|
if [[ -z "${latest_tag}" ]]; then
|
||||||
|
echo "::error::No stable release tags (vMAJOR.MINOR.PATCH) were found."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Parse components
|
||||||
|
ver="${latest_tag#v}"
|
||||||
|
major="${ver%%.*}"
|
||||||
|
rest="${ver#*.}"
|
||||||
|
minor="${rest%%.*}"
|
||||||
|
patch="${rest#*.}"
|
||||||
|
|
||||||
|
new_patch=$((patch + 1))
|
||||||
|
new_version="v${major}.${minor}.${new_patch}"
|
||||||
|
release_branch="release/v${major}.${minor}"
|
||||||
|
|
||||||
|
latest_sha="$(git rev-list -n 1 "refs/tags/${latest_tag}")"
|
||||||
|
|
||||||
|
echo "latest_tag=${latest_tag}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "latest_sha=${latest_sha}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "major=${major}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "minor=${minor}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "patch=${patch}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "new_version=${new_version}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "new_version_no_v=${major}.${minor}.${new_patch}" >> "$GITHUB_OUTPUT"
|
||||||
|
echo "release_branch=${release_branch}" >> "$GITHUB_OUTPUT"
|
||||||
|
|
||||||
|
echo "Latest stable release: ${latest_tag} (${latest_sha})"
|
||||||
|
echo "New version will be: ${new_version}"
|
||||||
|
echo "Release branch: ${release_branch}"
|
||||||
|
|
||||||
|
- name: Validate source branch is cut directly from the latest stable release
|
||||||
|
env:
|
||||||
|
SOURCE_BRANCH: ${{ steps.resolve.outputs.source_branch }}
|
||||||
|
SOURCE_COMMIT: ${{ inputs.commit }}
|
||||||
|
LATEST_TAG_SHA: ${{ steps.latest.outputs.latest_sha }}
|
||||||
|
LATEST_TAG: ${{ steps.latest.outputs.latest_tag }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Use the user-provided SHA directly rather than re-resolving the branch
|
||||||
|
# tip — the resolve step already proved the branch tip equals SOURCE_COMMIT,
|
||||||
|
# and pinning to the SHA here makes the rest of the job TOCTOU-safe against
|
||||||
|
# someone pushing to the branch mid-run.
|
||||||
|
source_sha="${SOURCE_COMMIT}"
|
||||||
|
|
||||||
|
# Walking first-parent from the source tip must reach LATEST_TAG_SHA.
|
||||||
|
# We capture rev-list into a variable and grep against a here-string
|
||||||
|
# rather than piping `rev-list | grep -q`: under `set -o pipefail`,
|
||||||
|
# `grep -q` would exit on first match and SIGPIPE the still-streaming
|
||||||
|
# `rev-list`, propagating exit 141 as a spurious "not found".
|
||||||
|
first_parent_chain="$(git rev-list --first-parent "${source_sha}")"
|
||||||
|
if ! grep -Fxq "${LATEST_TAG_SHA}" <<< "${first_parent_chain}"; then
|
||||||
|
echo "::error::Source branch '${SOURCE_BRANCH}' is not cut from '${LATEST_TAG}'."
|
||||||
|
echo "::error::Its first-parent history does not include ${LATEST_TAG_SHA}."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Additionally, every commit added on top of the tag (the set we are
|
||||||
|
# about to publish) must itself be a descendant of the tag along
|
||||||
|
# first-parent — i.e. no sibling commits from master sneak in via a
|
||||||
|
# non-first-parent path. Enforce by requiring that the symmetric
|
||||||
|
# difference is empty in one direction: commits in source that are
|
||||||
|
# NOT first-parent-reachable from source starting at the tag.
|
||||||
|
# We do this by intersecting:
|
||||||
|
# A = commits reachable from source but not from tag (full DAG)
|
||||||
|
# B = commits on the first-parent chain from source down to tag
|
||||||
|
# and requiring A == B.
|
||||||
|
all_added="$(git rev-list "${LATEST_TAG_SHA}..${source_sha}" | sort)"
|
||||||
|
first_parent_added="$(
|
||||||
|
git rev-list --first-parent "${LATEST_TAG_SHA}..${source_sha}" | sort
|
||||||
|
)"
|
||||||
|
|
||||||
|
if [[ "${all_added}" != "${first_parent_added}" ]]; then
|
||||||
|
echo "::error::Source branch '${SOURCE_BRANCH}' contains commits not on its first-parent chain from '${LATEST_TAG}'."
|
||||||
|
echo "::error::This usually means the branch was cut from master (not from the tag) or contains a merge from master."
|
||||||
|
echo "Commits reachable but not on first-parent chain:"
|
||||||
|
comm -23 <(printf '%s\n' "${all_added}") <(printf '%s\n' "${first_parent_added}") \
|
||||||
|
| while read -r sha; do
|
||||||
|
echo " $(git log -1 --format='%h %s' "${sha}")"
|
||||||
|
done
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
added_count="$(printf '%s\n' "${all_added}" | grep -c . || true)"
|
||||||
|
echo "Source branch is cut directly from ${LATEST_TAG} with ${added_count} commit(s) on top."
|
||||||
|
|
||||||
|
- name: Validate PR exists, is open, named correctly, has latest commit, and checks pass
|
||||||
|
env:
|
||||||
|
GH_TOKEN: ${{ steps.app-token.outputs.token }}
|
||||||
|
SOURCE_BRANCH: ${{ steps.resolve.outputs.source_branch }}
|
||||||
|
SOURCE_COMMIT: ${{ inputs.commit }}
|
||||||
|
NEW_VERSION: ${{ steps.latest.outputs.new_version }}
|
||||||
|
REPO: ${{ github.repository }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
expected_title="ComfyUI backport release ${NEW_VERSION}"
|
||||||
|
|
||||||
|
# Find open PRs from this branch into master. The --state open filter
|
||||||
|
# is load-bearing: a closed/merged PR with passing checks must not be
|
||||||
|
# accepted as authorization for a new release.
|
||||||
|
pr_json="$(
|
||||||
|
gh pr list \
|
||||||
|
--repo "${REPO}" \
|
||||||
|
--state open \
|
||||||
|
--head "${SOURCE_BRANCH}" \
|
||||||
|
--base master \
|
||||||
|
--json number,title,headRefOid,state \
|
||||||
|
--limit 10
|
||||||
|
)"
|
||||||
|
|
||||||
|
pr_count="$(echo "${pr_json}" | jq 'length')"
|
||||||
|
if [[ "${pr_count}" -eq 0 ]]; then
|
||||||
|
echo "::error::No open PR found from '${SOURCE_BRANCH}' into 'master'. The PR must exist and be open."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Pick the PR matching the expected title
|
||||||
|
pr_number="$(echo "${pr_json}" | jq -r --arg t "${expected_title}" '
|
||||||
|
map(select(.title == $t)) | .[0].number // empty
|
||||||
|
')"
|
||||||
|
pr_head_sha="$(echo "${pr_json}" | jq -r --arg t "${expected_title}" '
|
||||||
|
map(select(.title == $t)) | .[0].headRefOid // empty
|
||||||
|
')"
|
||||||
|
|
||||||
|
if [[ -z "${pr_number}" ]]; then
|
||||||
|
echo "::error::No open PR from '${SOURCE_BRANCH}' into 'master' is titled '${expected_title}'."
|
||||||
|
echo "Found PRs:"
|
||||||
|
echo "${pr_json}" | jq -r '.[] | " #\(.number): \(.title)"'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# The PR's current head commit must equal the SHA the operator gave us.
|
||||||
|
# This is what closes the door on releasing stale code: if anyone has
|
||||||
|
# pushed to the branch since the operator validated tests passed, the
|
||||||
|
# PR head will have advanced past SOURCE_COMMIT and we abort. (The
|
||||||
|
# resolve step already proved the branch tip == SOURCE_COMMIT; this
|
||||||
|
# ties that same SHA to the PR that authorizes the release.)
|
||||||
|
if [[ "${pr_head_sha}" != "${SOURCE_COMMIT}" ]]; then
|
||||||
|
echo "::error::PR #${pr_number} head commit is ${pr_head_sha}, but the operator-provided commit is ${SOURCE_COMMIT}."
|
||||||
|
echo "::error::The PR has new commits since this release was authorized. Re-run with the new head SHA after verifying its checks."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Found open PR #${pr_number} titled '${expected_title}' at head ${pr_head_sha} (matches operator-provided commit)."
|
||||||
|
|
||||||
|
# Verify all check runs on the head commit have completed successfully.
|
||||||
|
# A check is considered passing if conclusion is success, neutral, or skipped.
|
||||||
|
checks_json="$(
|
||||||
|
gh api \
|
||||||
|
--paginate \
|
||||||
|
"repos/${REPO}/commits/${pr_head_sha}/check-runs" \
|
||||||
|
--jq '.check_runs[] | {name: .name, status: .status, conclusion: .conclusion}'
|
||||||
|
)"
|
||||||
|
|
||||||
|
if [[ -z "${checks_json}" ]]; then
|
||||||
|
echo "::error::No check runs found on PR head commit ${pr_head_sha}."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Check runs on ${pr_head_sha}:"
|
||||||
|
echo "${checks_json}" | jq -s '.'
|
||||||
|
|
||||||
|
failing="$(echo "${checks_json}" | jq -s '
|
||||||
|
map(select(
|
||||||
|
.status != "completed"
|
||||||
|
or (.conclusion as $c
|
||||||
|
| ["success","neutral","skipped"]
|
||||||
|
| index($c) | not)
|
||||||
|
))
|
||||||
|
')"
|
||||||
|
|
||||||
|
failing_count="$(echo "${failing}" | jq 'length')"
|
||||||
|
if [[ "${failing_count}" -gt 0 ]]; then
|
||||||
|
echo "::error::One or more checks have not passed on PR head commit ${pr_head_sha}:"
|
||||||
|
echo "${failing}" | jq -r '.[] | " - \(.name): status=\(.status) conclusion=\(.conclusion)"'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "All checks have passed on ${pr_head_sha}."
|
||||||
|
|
||||||
|
- name: Prepare release branch
|
||||||
|
id: prepare
|
||||||
|
env:
|
||||||
|
GH_TOKEN: ${{ steps.app-token.outputs.token }}
|
||||||
|
REPO: ${{ github.repository }}
|
||||||
|
RELEASE_BRANCH: ${{ steps.latest.outputs.release_branch }}
|
||||||
|
LATEST_TAG: ${{ steps.latest.outputs.latest_tag }}
|
||||||
|
LATEST_TAG_SHA: ${{ steps.latest.outputs.latest_sha }}
|
||||||
|
PATCH: ${{ steps.latest.outputs.patch }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Try to fetch the release branch. If patch == 0, it shouldn't exist yet
|
||||||
|
# and we'll create it from the latest stable tag. If patch > 0, it must
|
||||||
|
# already exist and its tip must equal the latest stable tag commit (i.e.
|
||||||
|
# the previous patch release).
|
||||||
|
if git ls-remote --exit-code --heads origin "${RELEASE_BRANCH}" >/dev/null 2>&1; then
|
||||||
|
echo "Release branch '${RELEASE_BRANCH}' already exists on origin."
|
||||||
|
git fetch origin "refs/heads/${RELEASE_BRANCH}:refs/remotes/origin/${RELEASE_BRANCH}"
|
||||||
|
git checkout -B "${RELEASE_BRANCH}" "refs/remotes/origin/${RELEASE_BRANCH}"
|
||||||
|
|
||||||
|
current_tip="$(git rev-parse HEAD)"
|
||||||
|
if [[ "${current_tip}" != "${LATEST_TAG_SHA}" ]]; then
|
||||||
|
echo "::error::Release branch '${RELEASE_BRANCH}' tip (${current_tip}) is not at the latest stable release '${LATEST_TAG}' (${LATEST_TAG_SHA})."
|
||||||
|
echo "::error::Refusing to release on top of a divergent branch."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "branch_existed=true" >> "$GITHUB_OUTPUT"
|
||||||
|
else
|
||||||
|
if [[ "${PATCH}" != "0" ]]; then
|
||||||
|
echo "::error::Release branch '${RELEASE_BRANCH}' does not exist on origin, but the latest stable release '${LATEST_TAG}' has patch=${PATCH} (>0). This is inconsistent."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "Release branch '${RELEASE_BRANCH}' does not exist. Creating from ${LATEST_TAG}."
|
||||||
|
git checkout -B "${RELEASE_BRANCH}" "refs/tags/${LATEST_TAG}"
|
||||||
|
echo "branch_existed=false" >> "$GITHUB_OUTPUT"
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: Fast-forward merge source branch into release branch
|
||||||
|
env:
|
||||||
|
SOURCE_BRANCH: ${{ steps.resolve.outputs.source_branch }}
|
||||||
|
SOURCE_COMMIT: ${{ inputs.commit }}
|
||||||
|
RELEASE_BRANCH: ${{ steps.latest.outputs.release_branch }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# --ff-only guarantees no merge commit is created. If a fast-forward is
|
||||||
|
# not possible (i.e. the release branch has commits the source branch
|
||||||
|
# doesn't), the merge will fail and we abort. Because we already validated
|
||||||
|
# that the source branch is rooted on the latest stable tag, and the
|
||||||
|
# release branch tip equals that same tag, this fast-forward should
|
||||||
|
# always succeed for a well-formed backport branch.
|
||||||
|
#
|
||||||
|
# We merge the operator-provided SHA, not the branch ref, so a push to
|
||||||
|
# the branch in the window between resolve and now cannot smuggle new
|
||||||
|
# commits into the release.
|
||||||
|
if ! git merge --ff-only "${SOURCE_COMMIT}"; then
|
||||||
|
echo "::error::Cannot fast-forward '${RELEASE_BRANCH}' to ${SOURCE_COMMIT} (tip of '${SOURCE_BRANCH}'). A merge commit would be required. Aborting."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Fast-forwarded '${RELEASE_BRANCH}' to ${SOURCE_COMMIT} (tip of '${SOURCE_BRANCH}')."
|
||||||
|
|
||||||
|
- name: Bump version files
|
||||||
|
env:
|
||||||
|
NEW_VERSION_NO_V: ${{ steps.latest.outputs.new_version_no_v }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
if [[ ! -f comfyui_version.py ]]; then
|
||||||
|
echo "::error::comfyui_version.py not found in repo root."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
if [[ ! -f pyproject.toml ]]; then
|
||||||
|
echo "::error::pyproject.toml not found in repo root."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Replace the version string in comfyui_version.py.
|
||||||
|
# Expected format: __version__ = "X.Y.Z"
|
||||||
|
python3 - "$NEW_VERSION_NO_V" <<'PY'
|
||||||
|
import re, sys, pathlib
|
||||||
|
new = sys.argv[1]
|
||||||
|
|
||||||
|
p = pathlib.Path("comfyui_version.py")
|
||||||
|
src = p.read_text()
|
||||||
|
new_src, n = re.subn(
|
||||||
|
r'(__version__\s*=\s*[\'"])[^\'"]+([\'"])',
|
||||||
|
lambda m: f'{m.group(1)}{new}{m.group(2)}',
|
||||||
|
src,
|
||||||
|
count=1,
|
||||||
|
)
|
||||||
|
if n != 1:
|
||||||
|
sys.exit("Could not find __version__ assignment in comfyui_version.py")
|
||||||
|
p.write_text(new_src)
|
||||||
|
|
||||||
|
p = pathlib.Path("pyproject.toml")
|
||||||
|
src = p.read_text()
|
||||||
|
# Replace the first `version = "..."` inside [project] or [tool.poetry].
|
||||||
|
new_src, n = re.subn(
|
||||||
|
r'(?m)^(version\s*=\s*")[^"]+(")',
|
||||||
|
lambda m: f'{m.group(1)}{new}{m.group(2)}',
|
||||||
|
src,
|
||||||
|
count=1,
|
||||||
|
)
|
||||||
|
if n != 1:
|
||||||
|
sys.exit("Could not find version assignment in pyproject.toml")
|
||||||
|
p.write_text(new_src)
|
||||||
|
PY
|
||||||
|
|
||||||
|
echo "Updated version to ${NEW_VERSION_NO_V} in comfyui_version.py and pyproject.toml."
|
||||||
|
git --no-pager diff -- comfyui_version.py pyproject.toml
|
||||||
|
|
||||||
|
- name: Commit version bump and tag release
|
||||||
|
env:
|
||||||
|
NEW_VERSION: ${{ steps.latest.outputs.new_version }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
git add comfyui_version.py pyproject.toml
|
||||||
|
git commit -m "ComfyUI ${NEW_VERSION}"
|
||||||
|
|
||||||
|
if git rev-parse -q --verify "refs/tags/${NEW_VERSION}" >/dev/null; then
|
||||||
|
echo "::error::Tag ${NEW_VERSION} already exists locally."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
git tag "${NEW_VERSION}"
|
||||||
|
|
||||||
|
- name: Verify tag does not already exist on origin
|
||||||
|
env:
|
||||||
|
NEW_VERSION: ${{ steps.latest.outputs.new_version }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
if git ls-remote --exit-code --tags origin "refs/tags/${NEW_VERSION}" >/dev/null 2>&1; then
|
||||||
|
echo "::error::Tag ${NEW_VERSION} already exists on origin. Aborting."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: Push release branch and tag
|
||||||
|
env:
|
||||||
|
RELEASE_BRANCH: ${{ steps.latest.outputs.release_branch }}
|
||||||
|
NEW_VERSION: ${{ steps.latest.outputs.new_version }}
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Push the branch first, then the tag. Atomic-ish: if the branch push
|
||||||
|
# fails we never publish the tag.
|
||||||
|
git push origin "refs/heads/${RELEASE_BRANCH}:refs/heads/${RELEASE_BRANCH}"
|
||||||
|
git push origin "refs/tags/${NEW_VERSION}"
|
||||||
|
|
||||||
|
echo "Released ${NEW_VERSION} on ${RELEASE_BRANCH}."
|
||||||
|
|
||||||
|
- name: Summary
|
||||||
|
if: always()
|
||||||
|
env:
|
||||||
|
NEW_VERSION: ${{ steps.latest.outputs.new_version }}
|
||||||
|
RELEASE_BRANCH: ${{ steps.latest.outputs.release_branch }}
|
||||||
|
LATEST_TAG: ${{ steps.latest.outputs.latest_tag }}
|
||||||
|
SOURCE_BRANCH: ${{ steps.resolve.outputs.source_branch }}
|
||||||
|
SOURCE_COMMIT: ${{ inputs.commit }}
|
||||||
|
run: |
|
||||||
|
# SOURCE_BRANCH is empty if the resolve step never produced an output
|
||||||
|
# (e.g. the workflow failed in or before that step). Show a placeholder
|
||||||
|
# in that case so the summary table still renders cleanly.
|
||||||
|
source_branch_display="${SOURCE_BRANCH:-(unresolved)}"
|
||||||
|
{
|
||||||
|
echo "## Backport release"
|
||||||
|
echo ""
|
||||||
|
echo "| Field | Value |"
|
||||||
|
echo "|---|---|"
|
||||||
|
echo "| Source commit | \`${SOURCE_COMMIT}\` |"
|
||||||
|
echo "| Source branch | \`${source_branch_display}\` |"
|
||||||
|
echo "| Previous stable | \`${LATEST_TAG}\` |"
|
||||||
|
echo "| New version | \`${NEW_VERSION}\` |"
|
||||||
|
echo "| Release branch | \`${RELEASE_BRANCH}\` |"
|
||||||
|
} >> "$GITHUB_STEP_SUMMARY"
|
||||||
@ -20,7 +20,7 @@
|
|||||||
[website-url]: https://www.comfy.org/
|
[website-url]: https://www.comfy.org/
|
||||||
<!-- Workaround to display total user from https://github.com/badges/shields/issues/4500#issuecomment-2060079995 -->
|
<!-- Workaround to display total user from https://github.com/badges/shields/issues/4500#issuecomment-2060079995 -->
|
||||||
[discord-shield]: https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fdiscord.com%2Fapi%2Finvites%2Fcomfyorg%3Fwith_counts%3Dtrue&query=%24.approximate_member_count&logo=discord&logoColor=white&label=Discord&color=green&suffix=%20total
|
[discord-shield]: https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fdiscord.com%2Fapi%2Finvites%2Fcomfyorg%3Fwith_counts%3Dtrue&query=%24.approximate_member_count&logo=discord&logoColor=white&label=Discord&color=green&suffix=%20total
|
||||||
[discord-url]: https://www.comfy.org/discord
|
[discord-url]: https://discord.com/invite/comfyorg
|
||||||
[twitter-shield]: https://img.shields.io/twitter/follow/ComfyUI
|
[twitter-shield]: https://img.shields.io/twitter/follow/ComfyUI
|
||||||
[twitter-url]: https://x.com/ComfyUI
|
[twitter-url]: https://x.com/ComfyUI
|
||||||
|
|
||||||
|
|||||||
@ -62,6 +62,8 @@ def get_comfy_package_versions():
|
|||||||
def check_comfy_packages_versions():
|
def check_comfy_packages_versions():
|
||||||
"""Warn for every comfy* package whose installed version is below requirements.txt."""
|
"""Warn for every comfy* package whose installed version is below requirements.txt."""
|
||||||
from packaging.version import InvalidVersion, parse as parse_pep440
|
from packaging.version import InvalidVersion, parse as parse_pep440
|
||||||
|
outdated_packages = []
|
||||||
|
|
||||||
for pkg in get_comfy_package_versions():
|
for pkg in get_comfy_package_versions():
|
||||||
installed_str = pkg["installed"]
|
installed_str = pkg["installed"]
|
||||||
required_str = pkg["required"]
|
required_str = pkg["required"]
|
||||||
@ -73,19 +75,26 @@ def check_comfy_packages_versions():
|
|||||||
logging.error(f"Failed to check {pkg['name']} version: {e}")
|
logging.error(f"Failed to check {pkg['name']} version: {e}")
|
||||||
continue
|
continue
|
||||||
if outdated:
|
if outdated:
|
||||||
app.logger.log_startup_warning(
|
outdated_packages.append((pkg["name"], installed_str, required_str))
|
||||||
f"""
|
else:
|
||||||
|
logging.info("{} version: {}".format(pkg["name"], installed_str))
|
||||||
|
|
||||||
|
if outdated_packages:
|
||||||
|
package_warnings = "\n".join(
|
||||||
|
f"Installed {name} version {installed} is lower than the recommended version {required}."
|
||||||
|
for name, installed, required in outdated_packages
|
||||||
|
)
|
||||||
|
app.logger.log_startup_warning(
|
||||||
|
f"""
|
||||||
________________________________________________________________________
|
________________________________________________________________________
|
||||||
WARNING WARNING WARNING WARNING WARNING
|
WARNING WARNING WARNING WARNING WARNING
|
||||||
|
|
||||||
Installed {pkg["name"]} version {installed_str} is lower than the recommended version {required_str}.
|
{package_warnings}
|
||||||
|
|
||||||
{get_missing_requirements_message()}
|
{get_missing_requirements_message()}
|
||||||
________________________________________________________________________
|
________________________________________________________________________
|
||||||
""".strip()
|
""".strip()
|
||||||
)
|
)
|
||||||
else:
|
|
||||||
logging.info("{} version: {}".format(pkg["name"], installed_str))
|
|
||||||
|
|
||||||
|
|
||||||
REQUEST_TIMEOUT = 10 # seconds
|
REQUEST_TIMEOUT = 10 # seconds
|
||||||
|
|||||||
@ -1613,6 +1613,16 @@ class ModelPatcherDynamic(ModelPatcher):
|
|||||||
#use all ModelPatcherDynamic this is ignored and its all done dynamically.
|
#use all ModelPatcherDynamic this is ignored and its all done dynamically.
|
||||||
return super().memory_required(input_shape=input_shape) * 1.3 + (1024 ** 3)
|
return super().memory_required(input_shape=input_shape) * 1.3 + (1024 ** 3)
|
||||||
|
|
||||||
|
def restore_loaded_backups(self):
|
||||||
|
restored = self.model.model_loaded_weight_memory
|
||||||
|
for key in list(self.backup.keys()):
|
||||||
|
bk = self.backup.pop(key)
|
||||||
|
comfy.utils.set_attr_param(self.model, key, bk.weight)
|
||||||
|
for key in list(self.backup_buffers.keys()):
|
||||||
|
comfy.utils.set_attr_buffer(self.model, key, self.backup_buffers.pop(key))
|
||||||
|
self.model.model_loaded_weight_memory = 0
|
||||||
|
return restored
|
||||||
|
|
||||||
|
|
||||||
def load(self, device_to=None, lowvram_model_memory=0, force_patch_weights=False, full_load=False, dirty=False):
|
def load(self, device_to=None, lowvram_model_memory=0, force_patch_weights=False, full_load=False, dirty=False):
|
||||||
|
|
||||||
@ -1629,7 +1639,7 @@ class ModelPatcherDynamic(ModelPatcher):
|
|||||||
|
|
||||||
num_patches = 0
|
num_patches = 0
|
||||||
allocated_size = 0
|
allocated_size = 0
|
||||||
self.model.model_loaded_weight_memory = 0
|
self.restore_loaded_backups()
|
||||||
|
|
||||||
with self.use_ejected():
|
with self.use_ejected():
|
||||||
self.unpatch_hooks()
|
self.unpatch_hooks()
|
||||||
@ -1716,6 +1726,9 @@ class ModelPatcherDynamic(ModelPatcher):
|
|||||||
force_load=True
|
force_load=True
|
||||||
|
|
||||||
if force_load:
|
if force_load:
|
||||||
|
if hasattr(m, "_v"):
|
||||||
|
comfy_aimdo.model_vbar.vbar_unpin(m._v)
|
||||||
|
delattr(m, "_v")
|
||||||
force_load_param(self, "weight", device_to)
|
force_load_param(self, "weight", device_to)
|
||||||
force_load_param(self, "bias", device_to)
|
force_load_param(self, "bias", device_to)
|
||||||
else:
|
else:
|
||||||
@ -1773,13 +1786,7 @@ class ModelPatcherDynamic(ModelPatcher):
|
|||||||
freed = 0 if vbar is None else vbar.free_memory(memory_to_free)
|
freed = 0 if vbar is None else vbar.free_memory(memory_to_free)
|
||||||
|
|
||||||
if freed < memory_to_free:
|
if freed < memory_to_free:
|
||||||
for key in list(self.backup.keys()):
|
freed += self.restore_loaded_backups()
|
||||||
bk = self.backup.pop(key)
|
|
||||||
comfy.utils.set_attr_param(self.model, key, bk.weight)
|
|
||||||
for key in list(self.backup_buffers.keys()):
|
|
||||||
comfy.utils.set_attr_buffer(self.model, key, self.backup_buffers.pop(key))
|
|
||||||
freed += self.model.model_loaded_weight_memory
|
|
||||||
self.model.model_loaded_weight_memory = 0
|
|
||||||
|
|
||||||
return freed
|
return freed
|
||||||
|
|
||||||
|
|||||||
@ -1019,10 +1019,11 @@ def bislerp(samples, width, height):
|
|||||||
|
|
||||||
def lanczos(samples, width, height):
|
def lanczos(samples, width, height):
|
||||||
#the below API is strict and expects grayscale to be squeezed
|
#the below API is strict and expects grayscale to be squeezed
|
||||||
samples = samples.squeeze(1) if samples.shape[1] == 1 else samples.movedim(1, -1)
|
if samples.ndim == 4:
|
||||||
|
samples = samples.squeeze(1) if samples.shape[1] == 1 else samples.movedim(1, -1)
|
||||||
images = [Image.fromarray(np.clip(255. * image.cpu().numpy(), 0, 255).astype(np.uint8)) for image in samples]
|
images = [Image.fromarray(np.clip(255. * image.cpu().numpy(), 0, 255).astype(np.uint8)) for image in samples]
|
||||||
images = [image.resize((width, height), resample=Image.Resampling.LANCZOS) for image in images]
|
images = [image.resize((width, height), resample=Image.Resampling.LANCZOS) for image in images]
|
||||||
images = [torch.from_numpy(np.array(image).astype(np.float32) / 255.0).movedim(-1, 0) for image in images]
|
images = [torch.from_numpy(t).movedim(-1, 0) if (t := np.array(image).astype(np.float32) / 255.0).ndim == 3 else torch.from_numpy(t) for image in images]
|
||||||
result = torch.stack(images)
|
result = torch.stack(images)
|
||||||
return result.to(samples.device, samples.dtype)
|
return result.to(samples.device, samples.dtype)
|
||||||
|
|
||||||
|
|||||||
@ -35,6 +35,19 @@ class AnthropicMessage(BaseModel):
|
|||||||
content: list[AnthropicTextContent | AnthropicImageContent] = Field(...)
|
content: list[AnthropicTextContent | AnthropicImageContent] = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class AnthropicThinkingConfig(BaseModel):
|
||||||
|
type: Literal["enabled", "disabled", "adaptive"] = Field(...)
|
||||||
|
budget_tokens: int | None = Field(
|
||||||
|
None, ge=1024,
|
||||||
|
description="Reasoning budget in tokens. Used when type is 'enabled'. Must be less than max_tokens.",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class AnthropicOutputConfig(BaseModel):
|
||||||
|
"""Used with `thinking.type='adaptive'` on models like Opus 4.7."""
|
||||||
|
effort: Literal["low", "medium", "high"] | None = Field(None)
|
||||||
|
|
||||||
|
|
||||||
class AnthropicMessagesRequest(BaseModel):
|
class AnthropicMessagesRequest(BaseModel):
|
||||||
model: str = Field(...)
|
model: str = Field(...)
|
||||||
messages: list[AnthropicMessage] = Field(...)
|
messages: list[AnthropicMessage] = Field(...)
|
||||||
@ -44,6 +57,8 @@ class AnthropicMessagesRequest(BaseModel):
|
|||||||
top_p: float | None = Field(None, ge=0.0, le=1.0)
|
top_p: float | None = Field(None, ge=0.0, le=1.0)
|
||||||
top_k: int | None = Field(None, ge=0)
|
top_k: int | None = Field(None, ge=0)
|
||||||
stop_sequences: list[str] | None = Field(None)
|
stop_sequences: list[str] | None = Field(None)
|
||||||
|
thinking: AnthropicThinkingConfig | None = Field(None)
|
||||||
|
output_config: AnthropicOutputConfig | None = Field(None)
|
||||||
|
|
||||||
|
|
||||||
class AnthropicResponseTextBlock(BaseModel):
|
class AnthropicResponseTextBlock(BaseModel):
|
||||||
@ -51,6 +66,14 @@ class AnthropicResponseTextBlock(BaseModel):
|
|||||||
text: str = Field(...)
|
text: str = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class AnthropicResponseThinkingBlock(BaseModel):
|
||||||
|
type: Literal["thinking"] = "thinking"
|
||||||
|
thinking: str = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
AnthropicResponseBlock = AnthropicResponseTextBlock | AnthropicResponseThinkingBlock
|
||||||
|
|
||||||
|
|
||||||
class AnthropicCacheCreationUsage(BaseModel):
|
class AnthropicCacheCreationUsage(BaseModel):
|
||||||
ephemeral_5m_input_tokens: int | None = Field(None)
|
ephemeral_5m_input_tokens: int | None = Field(None)
|
||||||
ephemeral_1h_input_tokens: int | None = Field(None)
|
ephemeral_1h_input_tokens: int | None = Field(None)
|
||||||
@ -69,7 +92,7 @@ class AnthropicMessagesResponse(BaseModel):
|
|||||||
type: str | None = Field(None)
|
type: str | None = Field(None)
|
||||||
role: str | None = Field(None)
|
role: str | None = Field(None)
|
||||||
model: str | None = Field(None)
|
model: str | None = Field(None)
|
||||||
content: list[AnthropicResponseTextBlock] | None = Field(None)
|
content: list[AnthropicResponseBlock] | None = Field(None)
|
||||||
stop_reason: str | None = Field(None)
|
stop_reason: str | None = Field(None)
|
||||||
stop_sequence: str | None = Field(None)
|
stop_sequence: str | None = Field(None)
|
||||||
usage: AnthropicMessagesUsage | None = Field(None)
|
usage: AnthropicMessagesUsage | None = Field(None)
|
||||||
|
|||||||
93
comfy_api_nodes/apis/openrouter.py
Normal file
93
comfy_api_nodes/apis/openrouter.py
Normal file
@ -0,0 +1,93 @@
|
|||||||
|
"""Pydantic models for the OpenRouter chat completions API.
|
||||||
|
|
||||||
|
See: https://openrouter.ai/docs/api/api-reference/chat/send-chat-completion-request
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Literal
|
||||||
|
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterTextContent(BaseModel):
|
||||||
|
type: Literal["text"] = "text"
|
||||||
|
text: str = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterImageUrl(BaseModel):
|
||||||
|
url: str = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterImageContent(BaseModel):
|
||||||
|
type: Literal["image_url"] = "image_url"
|
||||||
|
image_url: OpenRouterImageUrl = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterVideoUrl(BaseModel):
|
||||||
|
url: str = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterVideoContent(BaseModel):
|
||||||
|
type: Literal["video_url"] = "video_url"
|
||||||
|
video_url: OpenRouterVideoUrl = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
OpenRouterContentBlock = OpenRouterTextContent | OpenRouterImageContent | OpenRouterVideoContent
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterMessage(BaseModel):
|
||||||
|
role: Literal["system", "user", "assistant"] = Field(...)
|
||||||
|
content: str | list[OpenRouterContentBlock] = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterReasoningConfig(BaseModel):
|
||||||
|
effort: str | None = Field(None)
|
||||||
|
exclude: bool | None = Field(None, description="If true, model reasons but reasoning is excluded from response.")
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterWebSearchOptions(BaseModel):
|
||||||
|
search_context_size: str | None = Field(None)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterChatRequest(BaseModel):
|
||||||
|
model: str = Field(...)
|
||||||
|
messages: list[OpenRouterMessage] = Field(...)
|
||||||
|
seed: int | None = Field(None)
|
||||||
|
reasoning: OpenRouterReasoningConfig | None = Field(None)
|
||||||
|
web_search_options: OpenRouterWebSearchOptions | None = Field(None)
|
||||||
|
stream: bool = Field(False)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterUsage(BaseModel):
|
||||||
|
prompt_tokens: int | None = Field(None)
|
||||||
|
completion_tokens: int | None = Field(None)
|
||||||
|
total_tokens: int | None = Field(None)
|
||||||
|
cost: float | None = Field(None, description="Server-side authoritative USD cost of the call.")
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterResponseMessage(BaseModel):
|
||||||
|
role: str | None = Field(None)
|
||||||
|
content: str | None = Field(None)
|
||||||
|
reasoning: str | None = Field(None)
|
||||||
|
refusal: str | None = Field(None)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterChoice(BaseModel):
|
||||||
|
index: int | None = Field(None)
|
||||||
|
message: OpenRouterResponseMessage | None = Field(None)
|
||||||
|
finish_reason: str | None = Field(None)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterError(BaseModel):
|
||||||
|
code: int | str | None = Field(None)
|
||||||
|
message: str | None = Field(None)
|
||||||
|
metadata: dict | None = Field(None)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterChatResponse(BaseModel):
|
||||||
|
id: str | None = Field(None)
|
||||||
|
model: str | None = Field(None)
|
||||||
|
object: str | None = Field(None)
|
||||||
|
provider: str | None = Field(None)
|
||||||
|
choices: list[OpenRouterChoice] | None = Field(None)
|
||||||
|
usage: OpenRouterUsage | None = Field(None)
|
||||||
|
error: OpenRouterError | None = Field(None)
|
||||||
@ -9,8 +9,11 @@ from comfy_api_nodes.apis.anthropic import (
|
|||||||
AnthropicMessage,
|
AnthropicMessage,
|
||||||
AnthropicMessagesRequest,
|
AnthropicMessagesRequest,
|
||||||
AnthropicMessagesResponse,
|
AnthropicMessagesResponse,
|
||||||
|
AnthropicOutputConfig,
|
||||||
|
AnthropicResponseTextBlock,
|
||||||
AnthropicRole,
|
AnthropicRole,
|
||||||
AnthropicTextContent,
|
AnthropicTextContent,
|
||||||
|
AnthropicThinkingConfig,
|
||||||
)
|
)
|
||||||
from comfy_api_nodes.util import (
|
from comfy_api_nodes.util import (
|
||||||
ApiEndpoint,
|
ApiEndpoint,
|
||||||
@ -32,15 +35,29 @@ CLAUDE_MODELS: dict[str, str] = {
|
|||||||
"Haiku 4.5": "claude-haiku-4-5-20251001",
|
"Haiku 4.5": "claude-haiku-4-5-20251001",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
_THINKING_UNSUPPORTED = {"Haiku 4.5"}
|
||||||
|
# Models that use the newer "adaptive" thinking mode (Opus 4.7 requires it; older models keep the explicit budget API).
|
||||||
|
# Anthropic decides the actual budget when adaptive is used, based on the `output_config.effort` hint.
|
||||||
|
_ADAPTIVE_THINKING_MODELS = {"Opus 4.7", "Opus 4.6", "Sonnet 4.6"}
|
||||||
|
|
||||||
def _claude_model_inputs():
|
# Budget mode (Sonnet 4.5): effort -> reasoning budget in tokens. Must be < max_tokens.
|
||||||
return [
|
# Sized so even the "high" budget fits comfortably under the default max_tokens=32768.
|
||||||
|
_REASONING_BUDGET: dict[str, int] = {
|
||||||
|
"low": 2048,
|
||||||
|
"medium": 8192,
|
||||||
|
"high": 16384,
|
||||||
|
}
|
||||||
|
_REASONING_EFFORTS = ["off", "low", "medium", "high"]
|
||||||
|
|
||||||
|
|
||||||
|
def _claude_model_inputs(model_label: str):
|
||||||
|
inputs: list = [
|
||||||
IO.Int.Input(
|
IO.Int.Input(
|
||||||
"max_tokens",
|
"max_tokens",
|
||||||
default=16000,
|
default=32768,
|
||||||
min=32,
|
min=4096,
|
||||||
max=32000,
|
max=64000,
|
||||||
tooltip="Maximum number of tokens to generate before stopping.",
|
tooltip="Maximum number of tokens to generate (includes reasoning tokens when enabled).",
|
||||||
advanced=True,
|
advanced=True,
|
||||||
),
|
),
|
||||||
IO.Float.Input(
|
IO.Float.Input(
|
||||||
@ -49,10 +66,24 @@ def _claude_model_inputs():
|
|||||||
min=0.0,
|
min=0.0,
|
||||||
max=1.0,
|
max=1.0,
|
||||||
step=0.01,
|
step=0.01,
|
||||||
tooltip="Controls randomness. 0.0 is deterministic, 1.0 is most random. Ignored for Opus 4.7.",
|
tooltip=(
|
||||||
|
"Controls randomness. 0.0 is deterministic, 1.0 is most random. "
|
||||||
|
"Ignored for Opus 4.7 and any model when reasoning_effort is set."
|
||||||
|
),
|
||||||
advanced=True,
|
advanced=True,
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
if model_label not in _THINKING_UNSUPPORTED:
|
||||||
|
inputs.append(
|
||||||
|
IO.Combo.Input(
|
||||||
|
"reasoning_effort",
|
||||||
|
options=_REASONING_EFFORTS,
|
||||||
|
default="off",
|
||||||
|
tooltip="Extended thinking effort. 'off' disables reasoning.",
|
||||||
|
advanced=True,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
return inputs
|
||||||
|
|
||||||
|
|
||||||
def _model_price_per_million(model: str) -> tuple[float, float] | None:
|
def _model_price_per_million(model: str) -> tuple[float, float] | None:
|
||||||
@ -95,7 +126,11 @@ def calculate_tokens_price(response: AnthropicMessagesResponse) -> float | None:
|
|||||||
def _get_text_from_response(response: AnthropicMessagesResponse) -> str:
|
def _get_text_from_response(response: AnthropicMessagesResponse) -> str:
|
||||||
if not response.content:
|
if not response.content:
|
||||||
return ""
|
return ""
|
||||||
return "\n".join(block.text for block in response.content if block.text)
|
# Thinking blocks are silently dropped — we never want reasoning in the output.
|
||||||
|
return "\n".join(
|
||||||
|
block.text for block in response.content
|
||||||
|
if isinstance(block, AnthropicResponseTextBlock) and block.text
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
async def _build_image_content_blocks(
|
async def _build_image_content_blocks(
|
||||||
@ -133,7 +168,10 @@ class ClaudeNode(IO.ComfyNode):
|
|||||||
),
|
),
|
||||||
IO.DynamicCombo.Input(
|
IO.DynamicCombo.Input(
|
||||||
"model",
|
"model",
|
||||||
options=[IO.DynamicCombo.Option(label, _claude_model_inputs()) for label in CLAUDE_MODELS],
|
options=[
|
||||||
|
IO.DynamicCombo.Option(label, _claude_model_inputs(label))
|
||||||
|
for label in CLAUDE_MODELS
|
||||||
|
],
|
||||||
tooltip="The Claude model used to generate the response.",
|
tooltip="The Claude model used to generate the response.",
|
||||||
),
|
),
|
||||||
IO.Int.Input(
|
IO.Int.Input(
|
||||||
@ -207,8 +245,29 @@ class ClaudeNode(IO.ComfyNode):
|
|||||||
) -> IO.NodeOutput:
|
) -> IO.NodeOutput:
|
||||||
validate_string(prompt, strip_whitespace=True, min_length=1)
|
validate_string(prompt, strip_whitespace=True, min_length=1)
|
||||||
model_label = model["model"]
|
model_label = model["model"]
|
||||||
max_tokens = model["max_tokens"]
|
max_tokens = model.get("max_tokens", 32768)
|
||||||
temperature = None if model_label == "Opus 4.7" else model["temperature"]
|
reasoning_effort = model.get("reasoning_effort", "off")
|
||||||
|
thinking_enabled = reasoning_effort not in ("off", None) and model_label not in _THINKING_UNSUPPORTED
|
||||||
|
|
||||||
|
# Anthropic requires temperature to be unset (defaults to 1.0) when thinking is enabled.
|
||||||
|
# Opus 4.7 also rejects user-supplied temperature.
|
||||||
|
if thinking_enabled or model_label == "Opus 4.7":
|
||||||
|
temperature = None
|
||||||
|
else:
|
||||||
|
temperature = model.get("temperature", 1.0)
|
||||||
|
|
||||||
|
thinking_cfg: AnthropicThinkingConfig | None = None
|
||||||
|
output_cfg: AnthropicOutputConfig | None = None
|
||||||
|
if thinking_enabled:
|
||||||
|
if model_label in _ADAPTIVE_THINKING_MODELS:
|
||||||
|
# Adaptive mode - Anthropic chooses the budget based on effort hint
|
||||||
|
thinking_cfg = AnthropicThinkingConfig(type="adaptive")
|
||||||
|
output_cfg = AnthropicOutputConfig(effort=reasoning_effort)
|
||||||
|
else:
|
||||||
|
# Budget mode (Sonnet 4.5). Leave at least 1024 tokens for the actual response
|
||||||
|
budget = _REASONING_BUDGET[reasoning_effort]
|
||||||
|
budget = min(budget, max(1024, max_tokens - 1024))
|
||||||
|
thinking_cfg = AnthropicThinkingConfig(type="enabled", budget_tokens=budget)
|
||||||
|
|
||||||
image_tensors: list[Input.Image] = [t for t in (images or {}).values() if t is not None]
|
image_tensors: list[Input.Image] = [t for t in (images or {}).values() if t is not None]
|
||||||
if sum(get_number_of_images(t) for t in image_tensors) > CLAUDE_MAX_IMAGES:
|
if sum(get_number_of_images(t) for t in image_tensors) > CLAUDE_MAX_IMAGES:
|
||||||
@ -229,6 +288,8 @@ class ClaudeNode(IO.ComfyNode):
|
|||||||
messages=[AnthropicMessage(role=AnthropicRole.user, content=content)],
|
messages=[AnthropicMessage(role=AnthropicRole.user, content=content)],
|
||||||
system=system_prompt or None,
|
system=system_prompt or None,
|
||||||
temperature=temperature,
|
temperature=temperature,
|
||||||
|
thinking=thinking_cfg,
|
||||||
|
output_config=output_cfg,
|
||||||
),
|
),
|
||||||
price_extractor=calculate_tokens_price,
|
price_extractor=calculate_tokens_price,
|
||||||
)
|
)
|
||||||
|
|||||||
@ -43,15 +43,16 @@ from comfy_api_nodes.util import (
|
|||||||
ApiEndpoint,
|
ApiEndpoint,
|
||||||
download_url_to_image_tensor,
|
download_url_to_image_tensor,
|
||||||
download_url_to_video_output,
|
download_url_to_video_output,
|
||||||
|
downscale_video_to_max_pixels,
|
||||||
get_number_of_images,
|
get_number_of_images,
|
||||||
image_tensor_pair_to_batch,
|
image_tensor_pair_to_batch,
|
||||||
poll_op,
|
poll_op,
|
||||||
resize_video_to_pixel_budget,
|
|
||||||
sync_op,
|
sync_op,
|
||||||
upload_audio_to_comfyapi,
|
upload_audio_to_comfyapi,
|
||||||
upload_image_to_comfyapi,
|
upload_image_to_comfyapi,
|
||||||
upload_images_to_comfyapi,
|
upload_images_to_comfyapi,
|
||||||
upload_video_to_comfyapi,
|
upload_video_to_comfyapi,
|
||||||
|
upscale_video_to_min_pixels,
|
||||||
validate_image_aspect_ratio,
|
validate_image_aspect_ratio,
|
||||||
validate_image_dimensions,
|
validate_image_dimensions,
|
||||||
validate_string,
|
validate_string,
|
||||||
@ -110,12 +111,13 @@ def _validate_ref_video_pixels(video: Input.Video, model_id: str, resolution: st
|
|||||||
max_px = limits.get("max")
|
max_px = limits.get("max")
|
||||||
if min_px and pixels < min_px:
|
if min_px and pixels < min_px:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"Reference video {index} is too small: {w}x{h} = {pixels:,}px. " f"Minimum is {min_px:,}px for this model."
|
f"Reference video {index} is too small: {w}x{h} = {pixels:,} total pixels. "
|
||||||
|
f"Minimum for this model is {min_px:,} total pixels."
|
||||||
)
|
)
|
||||||
if max_px and pixels > max_px:
|
if max_px and pixels > max_px:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"Reference video {index} is too large: {w}x{h} = {pixels:,}px. "
|
f"Reference video {index} is too large: {w}x{h} = {pixels:,} total pixels. "
|
||||||
f"Maximum is {max_px:,}px for this model. Try downscaling the video."
|
f"Maximum for this model is {max_px:,} total pixels. Try downscaling the video."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ -1676,14 +1678,14 @@ class ByteDance2FirstLastFrameNode(IO.ComfyNode):
|
|||||||
"first_frame_asset_id",
|
"first_frame_asset_id",
|
||||||
default="",
|
default="",
|
||||||
tooltip="Seedance asset_id to use as the first frame. "
|
tooltip="Seedance asset_id to use as the first frame. "
|
||||||
"Mutually exclusive with the first_frame image input.",
|
"Mutually exclusive with the first_frame image input.",
|
||||||
optional=True,
|
optional=True,
|
||||||
),
|
),
|
||||||
IO.String.Input(
|
IO.String.Input(
|
||||||
"last_frame_asset_id",
|
"last_frame_asset_id",
|
||||||
default="",
|
default="",
|
||||||
tooltip="Seedance asset_id to use as the last frame. "
|
tooltip="Seedance asset_id to use as the last frame. "
|
||||||
"Mutually exclusive with the last_frame image input.",
|
"Mutually exclusive with the last_frame image input.",
|
||||||
optional=True,
|
optional=True,
|
||||||
),
|
),
|
||||||
IO.Int.Input(
|
IO.Int.Input(
|
||||||
@ -1865,11 +1867,20 @@ def _seedance2_reference_inputs(resolutions: list[str], default_ratio: str = "16
|
|||||||
IO.Boolean.Input(
|
IO.Boolean.Input(
|
||||||
"auto_downscale",
|
"auto_downscale",
|
||||||
default=False,
|
default=False,
|
||||||
advanced=True,
|
|
||||||
optional=True,
|
optional=True,
|
||||||
tooltip="Automatically downscale reference videos that exceed the model's pixel budget "
|
tooltip="Automatically downscale reference videos that exceed the model's pixel budget "
|
||||||
"for the selected resolution. Aspect ratio is preserved; videos already within limits are untouched.",
|
"for the selected resolution. Aspect ratio is preserved; videos already within limits are untouched.",
|
||||||
),
|
),
|
||||||
|
IO.Boolean.Input(
|
||||||
|
"auto_upscale",
|
||||||
|
default=False,
|
||||||
|
advanced=True,
|
||||||
|
optional=True,
|
||||||
|
tooltip="Automatically upscale reference videos that are below the model's minimum pixel count "
|
||||||
|
"for the selected resolution. Aspect ratio is preserved; videos already meeting the minimum are "
|
||||||
|
"untouched. Note: upscaling a low-resolution source does not add real detail and may produce "
|
||||||
|
"lower-quality generations.",
|
||||||
|
),
|
||||||
IO.Autogrow.Input(
|
IO.Autogrow.Input(
|
||||||
"reference_assets",
|
"reference_assets",
|
||||||
template=IO.Autogrow.TemplateNames(
|
template=IO.Autogrow.TemplateNames(
|
||||||
@ -2030,7 +2041,13 @@ class ByteDance2ReferenceNode(IO.ComfyNode):
|
|||||||
max_px = SEEDANCE2_REF_VIDEO_PIXEL_LIMITS.get(model_id, {}).get(model["resolution"], {}).get("max")
|
max_px = SEEDANCE2_REF_VIDEO_PIXEL_LIMITS.get(model_id, {}).get(model["resolution"], {}).get("max")
|
||||||
if max_px:
|
if max_px:
|
||||||
for key in reference_videos:
|
for key in reference_videos:
|
||||||
reference_videos[key] = resize_video_to_pixel_budget(reference_videos[key], max_px)
|
reference_videos[key] = downscale_video_to_max_pixels(reference_videos[key], max_px)
|
||||||
|
|
||||||
|
if model.get("auto_upscale") and reference_videos:
|
||||||
|
min_px = SEEDANCE2_REF_VIDEO_PIXEL_LIMITS.get(model_id, {}).get(model["resolution"], {}).get("min")
|
||||||
|
if min_px:
|
||||||
|
for key in reference_videos:
|
||||||
|
reference_videos[key] = upscale_video_to_min_pixels(reference_videos[key], min_px)
|
||||||
|
|
||||||
total_video_duration = 0.0
|
total_video_duration = 0.0
|
||||||
for i, key in enumerate(reference_videos, 1):
|
for i, key in enumerate(reference_videos, 1):
|
||||||
|
|||||||
374
comfy_api_nodes/nodes_openrouter.py
Normal file
374
comfy_api_nodes/nodes_openrouter.py
Normal file
@ -0,0 +1,374 @@
|
|||||||
|
"""API Nodes for OpenRouter LLM chat completions."""
|
||||||
|
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from typing import Literal
|
||||||
|
|
||||||
|
from typing_extensions import override
|
||||||
|
|
||||||
|
from comfy_api.latest import IO, ComfyExtension, Input
|
||||||
|
from comfy_api_nodes.apis.openrouter import (
|
||||||
|
OpenRouterChatRequest,
|
||||||
|
OpenRouterChatResponse,
|
||||||
|
OpenRouterContentBlock,
|
||||||
|
OpenRouterImageContent,
|
||||||
|
OpenRouterImageUrl,
|
||||||
|
OpenRouterMessage,
|
||||||
|
OpenRouterReasoningConfig,
|
||||||
|
OpenRouterTextContent,
|
||||||
|
OpenRouterVideoContent,
|
||||||
|
OpenRouterVideoUrl,
|
||||||
|
OpenRouterWebSearchOptions,
|
||||||
|
)
|
||||||
|
from comfy_api_nodes.util import (
|
||||||
|
ApiEndpoint,
|
||||||
|
get_number_of_images,
|
||||||
|
sync_op,
|
||||||
|
upload_images_to_comfyapi,
|
||||||
|
upload_video_to_comfyapi,
|
||||||
|
validate_string,
|
||||||
|
)
|
||||||
|
|
||||||
|
OPENROUTER_CHAT_ENDPOINT = "/proxy/openrouter/api/v1/chat/completions"
|
||||||
|
|
||||||
|
|
||||||
|
Profile = Literal["standard", "reasoning", "frontier_reasoning", "perplexity", "perplexity_reasoning"]
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class _ModelSpec:
|
||||||
|
slug: str # exact OpenRouter model id
|
||||||
|
profile: Profile
|
||||||
|
price_in: float # USD per token (prompt)
|
||||||
|
price_out: float # USD per token (completion)
|
||||||
|
max_images: int = 0 # 0 = no image input; otherwise max URL-passed images supported
|
||||||
|
max_videos: int = 0 # 0 = no video input; otherwise max URL-passed videos supported
|
||||||
|
|
||||||
|
|
||||||
|
MODELS: list[_ModelSpec] = [
|
||||||
|
_ModelSpec("anthropic/claude-opus-4.7", "frontier_reasoning", 0.000005, 0.000025, max_images=20),
|
||||||
|
_ModelSpec("openai/gpt-5.5-pro", "frontier_reasoning", 0.00003, 0.00018, max_images=20),
|
||||||
|
_ModelSpec("openai/gpt-5.5", "frontier_reasoning", 0.000005, 0.00003, max_images=20),
|
||||||
|
_ModelSpec("google/gemini-3.5-flash", "reasoning", 0.0000015, 0.000009, max_images=20, max_videos=4),
|
||||||
|
_ModelSpec("x-ai/grok-4.20", "reasoning", 0.00000125, 0.0000025, max_images=20),
|
||||||
|
_ModelSpec("x-ai/grok-4.3", "reasoning", 0.00000125, 0.0000025, max_images=20),
|
||||||
|
_ModelSpec("deepseek/deepseek-v4-pro", "reasoning", 0.000000435, 0.00000087),
|
||||||
|
_ModelSpec("deepseek/deepseek-v4-flash", "reasoning", 0.000000112, 0.000000224),
|
||||||
|
_ModelSpec("deepseek/deepseek-v3.2", "reasoning", 0.000000252, 0.000000378),
|
||||||
|
_ModelSpec("qwen/qwen3.6-max-preview", "reasoning", 0.00000104, 0.00000624),
|
||||||
|
_ModelSpec("qwen/qwen3.6-plus", "reasoning", 0.000000325, 0.00000195, max_images=10, max_videos=4),
|
||||||
|
_ModelSpec("qwen/qwen3.6-flash", "reasoning", 0.0000001875, 0.000001125, max_images=10, max_videos=4),
|
||||||
|
_ModelSpec("mistralai/mistral-large-2512", "standard", 0.0000005, 0.0000015, max_images=8),
|
||||||
|
_ModelSpec("mistralai/mistral-medium-3-5", "reasoning", 0.0000015, 0.0000075, max_images=8),
|
||||||
|
_ModelSpec("z-ai/glm-4.6", "reasoning", 0.00000043, 0.00000174),
|
||||||
|
_ModelSpec("z-ai/glm-5", "reasoning", 0.0000006, 0.00000192),
|
||||||
|
_ModelSpec("moonshotai/kimi-k2.6", "reasoning", 0.00000073, 0.00000349, max_images=10),
|
||||||
|
_ModelSpec("moonshotai/kimi-k2-thinking", "reasoning", 0.0000006, 0.0000025),
|
||||||
|
_ModelSpec("perplexity/sonar-pro", "perplexity", 0.000003, 0.000015),
|
||||||
|
_ModelSpec("perplexity/sonar-reasoning-pro", "perplexity_reasoning", 0.000002, 0.000008),
|
||||||
|
_ModelSpec("perplexity/sonar-deep-research", "perplexity_reasoning", 0.000002, 0.000008),
|
||||||
|
]
|
||||||
|
|
||||||
|
_MODELS_BY_SLUG: dict[str, _ModelSpec] = {m.slug: m for m in MODELS}
|
||||||
|
_REASONING_EFFORTS = ["off", "low", "medium", "high"]
|
||||||
|
_SEARCH_CONTEXT_SIZES = ["low", "medium", "high"]
|
||||||
|
|
||||||
|
|
||||||
|
def _reasoning_extra_inputs() -> list:
|
||||||
|
return [
|
||||||
|
IO.Combo.Input(
|
||||||
|
"reasoning_effort",
|
||||||
|
options=_REASONING_EFFORTS,
|
||||||
|
default="off",
|
||||||
|
tooltip="Reasoning effort. 'off' disables reasoning entirely.",
|
||||||
|
advanced=True,
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def _perplexity_extra_inputs() -> list:
|
||||||
|
return [
|
||||||
|
IO.Combo.Input(
|
||||||
|
"search_context_size",
|
||||||
|
options=_SEARCH_CONTEXT_SIZES,
|
||||||
|
default="medium",
|
||||||
|
tooltip="How much web search context to retrieve. Larger = more grounded but slower/pricier.",
|
||||||
|
advanced=True,
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def _profile_inputs(profile: Profile) -> list:
|
||||||
|
if profile == "standard":
|
||||||
|
return []
|
||||||
|
if profile in ("reasoning", "frontier_reasoning"):
|
||||||
|
return _reasoning_extra_inputs()
|
||||||
|
if profile == "perplexity":
|
||||||
|
return _perplexity_extra_inputs()
|
||||||
|
if profile == "perplexity_reasoning":
|
||||||
|
return _perplexity_extra_inputs() + _reasoning_extra_inputs()
|
||||||
|
raise ValueError(f"Unknown profile: {profile}")
|
||||||
|
|
||||||
|
|
||||||
|
def _media_inputs(spec: _ModelSpec) -> list:
|
||||||
|
extras: list = []
|
||||||
|
if spec.max_images > 0:
|
||||||
|
extras.append(
|
||||||
|
IO.Autogrow.Input(
|
||||||
|
"images",
|
||||||
|
template=IO.Autogrow.TemplateNames(
|
||||||
|
IO.Image.Input("image"),
|
||||||
|
names=[f"image_{i}" for i in range(1, spec.max_images + 1)],
|
||||||
|
min=0,
|
||||||
|
),
|
||||||
|
tooltip=f"Optional reference image(s) — up to {spec.max_images}. Sent as URLs.",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
if spec.max_videos > 0:
|
||||||
|
extras.append(
|
||||||
|
IO.Autogrow.Input(
|
||||||
|
"videos",
|
||||||
|
template=IO.Autogrow.TemplateNames(
|
||||||
|
IO.Video.Input("video"),
|
||||||
|
names=[f"video_{i}" for i in range(1, spec.max_videos + 1)],
|
||||||
|
min=0,
|
||||||
|
),
|
||||||
|
tooltip=f"Optional reference video(s) — up to {spec.max_videos}. Sent as URLs.",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
return extras
|
||||||
|
|
||||||
|
|
||||||
|
def _inputs_for_model(spec: _ModelSpec) -> list:
|
||||||
|
return _profile_inputs(spec.profile) + _media_inputs(spec)
|
||||||
|
|
||||||
|
|
||||||
|
def _build_model_options() -> list[IO.DynamicCombo.Option]:
|
||||||
|
return [IO.DynamicCombo.Option(spec.slug, _inputs_for_model(spec)) for spec in MODELS]
|
||||||
|
|
||||||
|
|
||||||
|
def _calculate_price(response: OpenRouterChatResponse) -> float | None:
|
||||||
|
if response.usage and response.usage.cost is not None:
|
||||||
|
return float(response.usage.cost)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def _price_badge_jsonata() -> str:
|
||||||
|
rates_pairs = []
|
||||||
|
for spec in MODELS:
|
||||||
|
prompt_per_1k = spec.price_in * 1000
|
||||||
|
completion_per_1k = spec.price_out * 1000
|
||||||
|
rates_pairs.append(f' "{spec.slug}": [{prompt_per_1k:.8g}, {completion_per_1k:.8g}]')
|
||||||
|
rates_block = ",\n".join(rates_pairs)
|
||||||
|
return (
|
||||||
|
"(\n"
|
||||||
|
" $rates := {\n"
|
||||||
|
f"{rates_block}\n"
|
||||||
|
" };\n"
|
||||||
|
" $r := $lookup($rates, widgets.model);\n"
|
||||||
|
" $r ? {\n"
|
||||||
|
' "type": "list_usd",\n'
|
||||||
|
' "usd": $r,\n'
|
||||||
|
' "format": { "approximate": true, "separator": "-", "suffix": " per 1K tokens" }\n'
|
||||||
|
' } : {"type": "text", "text": "Token-based"}\n'
|
||||||
|
")"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def _build_image_blocks(
|
||||||
|
cls: type[IO.ComfyNode], spec: _ModelSpec, images: list[Input.Image]
|
||||||
|
) -> list[OpenRouterImageContent]:
|
||||||
|
urls = await upload_images_to_comfyapi(
|
||||||
|
cls,
|
||||||
|
images,
|
||||||
|
max_images=spec.max_images,
|
||||||
|
total_pixels=2048 * 2048,
|
||||||
|
mime_type="image/png",
|
||||||
|
wait_label="Uploading reference images",
|
||||||
|
)
|
||||||
|
return [OpenRouterImageContent(image_url=OpenRouterImageUrl(url=url)) for url in urls]
|
||||||
|
|
||||||
|
|
||||||
|
async def _build_video_blocks(cls: type[IO.ComfyNode], videos: list[Input.Video]) -> list[OpenRouterVideoContent]:
|
||||||
|
blocks: list[OpenRouterVideoContent] = []
|
||||||
|
total = len(videos)
|
||||||
|
for idx, video in enumerate(videos):
|
||||||
|
label = "Uploading reference video"
|
||||||
|
if total > 1:
|
||||||
|
label = f"{label} ({idx + 1}/{total})"
|
||||||
|
url = await upload_video_to_comfyapi(cls, video, wait_label=label)
|
||||||
|
blocks.append(OpenRouterVideoContent(video_url=OpenRouterVideoUrl(url=url)))
|
||||||
|
return blocks
|
||||||
|
|
||||||
|
|
||||||
|
def _user_message(prompt: str, media_blocks: list[OpenRouterContentBlock]) -> OpenRouterMessage:
|
||||||
|
if not media_blocks:
|
||||||
|
return OpenRouterMessage(role="user", content=prompt)
|
||||||
|
blocks: list[OpenRouterContentBlock] = list(media_blocks)
|
||||||
|
blocks.append(OpenRouterTextContent(text=prompt))
|
||||||
|
return OpenRouterMessage(role="user", content=blocks)
|
||||||
|
|
||||||
|
|
||||||
|
def _build_messages(
|
||||||
|
system_prompt: str, prompt: str, media_blocks: list[OpenRouterContentBlock]
|
||||||
|
) -> list[OpenRouterMessage]:
|
||||||
|
messages: list[OpenRouterMessage] = []
|
||||||
|
if system_prompt:
|
||||||
|
messages.append(OpenRouterMessage(role="system", content=system_prompt))
|
||||||
|
messages.append(_user_message(prompt, media_blocks))
|
||||||
|
return messages
|
||||||
|
|
||||||
|
|
||||||
|
def _build_request(
|
||||||
|
slug: str,
|
||||||
|
system_prompt: str,
|
||||||
|
prompt: str,
|
||||||
|
media_blocks: list[OpenRouterContentBlock],
|
||||||
|
*,
|
||||||
|
seed: int,
|
||||||
|
reasoning_effort: str | None,
|
||||||
|
search_context_size: str | None,
|
||||||
|
) -> OpenRouterChatRequest:
|
||||||
|
reasoning_cfg: OpenRouterReasoningConfig | None = None
|
||||||
|
if reasoning_effort and reasoning_effort != "off":
|
||||||
|
# exclude=True asks providers to reason internally but not return the trace
|
||||||
|
reasoning_cfg = OpenRouterReasoningConfig(effort=reasoning_effort, exclude=True)
|
||||||
|
web_search_cfg: OpenRouterWebSearchOptions | None = None
|
||||||
|
if search_context_size:
|
||||||
|
web_search_cfg = OpenRouterWebSearchOptions(search_context_size=search_context_size)
|
||||||
|
return OpenRouterChatRequest(
|
||||||
|
model=slug,
|
||||||
|
messages=_build_messages(system_prompt, prompt, media_blocks),
|
||||||
|
seed=seed if seed > 0 else None,
|
||||||
|
reasoning=reasoning_cfg,
|
||||||
|
web_search_options=web_search_cfg,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def _extract_text(response: OpenRouterChatResponse) -> str:
|
||||||
|
if response.error:
|
||||||
|
code = response.error.code if response.error.code is not None else "unknown"
|
||||||
|
raise ValueError(f"OpenRouter error ({code}): {response.error.message or 'no message'}")
|
||||||
|
if not response.choices:
|
||||||
|
raise ValueError("Empty response from OpenRouter (no choices).")
|
||||||
|
message = response.choices[0].message
|
||||||
|
if not message:
|
||||||
|
raise ValueError("Empty response from OpenRouter (no message).")
|
||||||
|
if message.refusal:
|
||||||
|
raise ValueError(f"Model refused to respond: {message.refusal}")
|
||||||
|
return message.content or ""
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterLLMNode(IO.ComfyNode):
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def define_schema(cls):
|
||||||
|
return IO.Schema(
|
||||||
|
node_id="OpenRouterLLMNode",
|
||||||
|
display_name="OpenRouter LLM",
|
||||||
|
category="api node/text/OpenRouter",
|
||||||
|
essentials_category="Text Generation",
|
||||||
|
description=(
|
||||||
|
"Generate text responses through OpenRouter. Routes to a curated set of popular "
|
||||||
|
"models from xAI, DeepSeek, Qwen, Mistral, Z.AI (GLM), Moonshot (Kimi), and "
|
||||||
|
"Perplexity Sonar."
|
||||||
|
),
|
||||||
|
inputs=[
|
||||||
|
IO.String.Input(
|
||||||
|
"prompt",
|
||||||
|
multiline=True,
|
||||||
|
default="",
|
||||||
|
tooltip="Text input to the model.",
|
||||||
|
),
|
||||||
|
IO.DynamicCombo.Input(
|
||||||
|
"model",
|
||||||
|
options=_build_model_options(),
|
||||||
|
tooltip="The OpenRouter model used to generate the response.",
|
||||||
|
),
|
||||||
|
IO.Int.Input(
|
||||||
|
"seed",
|
||||||
|
default=0,
|
||||||
|
min=0,
|
||||||
|
max=2147483647,
|
||||||
|
control_after_generate=True,
|
||||||
|
tooltip="Seed for sampling. Set to 0 to omit. Most models treat this as a hint only.",
|
||||||
|
),
|
||||||
|
IO.String.Input(
|
||||||
|
"system_prompt",
|
||||||
|
multiline=True,
|
||||||
|
default="",
|
||||||
|
optional=True,
|
||||||
|
advanced=True,
|
||||||
|
tooltip="Foundational instructions that dictate the model's behavior.",
|
||||||
|
),
|
||||||
|
],
|
||||||
|
outputs=[IO.String.Output()],
|
||||||
|
hidden=[
|
||||||
|
IO.Hidden.auth_token_comfy_org,
|
||||||
|
IO.Hidden.api_key_comfy_org,
|
||||||
|
IO.Hidden.unique_id,
|
||||||
|
],
|
||||||
|
is_api_node=True,
|
||||||
|
price_badge=IO.PriceBadge(
|
||||||
|
depends_on=IO.PriceBadgeDepends(widgets=["model"]),
|
||||||
|
expr=_price_badge_jsonata(),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
async def execute(
|
||||||
|
cls,
|
||||||
|
prompt: str,
|
||||||
|
model: dict,
|
||||||
|
seed: int,
|
||||||
|
system_prompt: str = "",
|
||||||
|
) -> IO.NodeOutput:
|
||||||
|
validate_string(prompt, strip_whitespace=True, min_length=1)
|
||||||
|
slug: str = model["model"]
|
||||||
|
spec = _MODELS_BY_SLUG.get(slug)
|
||||||
|
if spec is None:
|
||||||
|
raise ValueError(f"Unknown OpenRouter model: {slug}")
|
||||||
|
|
||||||
|
reasoning_effort: str | None = model.get("reasoning_effort")
|
||||||
|
search_context_size: str | None = model.get("search_context_size")
|
||||||
|
|
||||||
|
image_tensors: list[Input.Image] = [t for t in (model.get("images") or {}).values() if t is not None]
|
||||||
|
if image_tensors and sum(get_number_of_images(t) for t in image_tensors) > spec.max_images:
|
||||||
|
raise ValueError(f"Up to {spec.max_images} images are supported for {slug}.")
|
||||||
|
video_inputs: list[Input.Video] = [v for v in (model.get("videos") or {}).values() if v is not None]
|
||||||
|
if video_inputs and len(video_inputs) > spec.max_videos:
|
||||||
|
raise ValueError(f"Up to {spec.max_videos} videos are supported for {slug}.")
|
||||||
|
|
||||||
|
media_blocks: list[OpenRouterContentBlock] = []
|
||||||
|
if image_tensors:
|
||||||
|
media_blocks.extend(await _build_image_blocks(cls, spec, image_tensors))
|
||||||
|
if video_inputs:
|
||||||
|
media_blocks.extend(await _build_video_blocks(cls, video_inputs))
|
||||||
|
|
||||||
|
request = _build_request(
|
||||||
|
slug,
|
||||||
|
system_prompt,
|
||||||
|
prompt,
|
||||||
|
media_blocks,
|
||||||
|
seed=seed,
|
||||||
|
reasoning_effort=reasoning_effort,
|
||||||
|
search_context_size=search_context_size,
|
||||||
|
)
|
||||||
|
|
||||||
|
response = await sync_op(
|
||||||
|
cls,
|
||||||
|
ApiEndpoint(path=OPENROUTER_CHAT_ENDPOINT, method="POST"),
|
||||||
|
response_model=OpenRouterChatResponse,
|
||||||
|
data=request,
|
||||||
|
price_extractor=_calculate_price,
|
||||||
|
)
|
||||||
|
return IO.NodeOutput(_extract_text(response))
|
||||||
|
|
||||||
|
|
||||||
|
class OpenRouterExtension(ComfyExtension):
|
||||||
|
@override
|
||||||
|
async def get_node_list(self) -> list[type[IO.ComfyNode]]:
|
||||||
|
return [OpenRouterLLMNode]
|
||||||
|
|
||||||
|
|
||||||
|
async def comfy_entrypoint() -> OpenRouterExtension:
|
||||||
|
return OpenRouterExtension()
|
||||||
@ -16,16 +16,17 @@ from .conversions import (
|
|||||||
convert_mask_to_image,
|
convert_mask_to_image,
|
||||||
downscale_image_tensor,
|
downscale_image_tensor,
|
||||||
downscale_image_tensor_by_max_side,
|
downscale_image_tensor_by_max_side,
|
||||||
|
downscale_video_to_max_pixels,
|
||||||
image_tensor_pair_to_batch,
|
image_tensor_pair_to_batch,
|
||||||
pil_to_bytesio,
|
pil_to_bytesio,
|
||||||
resize_mask_to_image,
|
resize_mask_to_image,
|
||||||
resize_video_to_pixel_budget,
|
|
||||||
tensor_to_base64_string,
|
tensor_to_base64_string,
|
||||||
tensor_to_bytesio,
|
tensor_to_bytesio,
|
||||||
tensor_to_pil,
|
tensor_to_pil,
|
||||||
text_filepath_to_base64_string,
|
text_filepath_to_base64_string,
|
||||||
text_filepath_to_data_uri,
|
text_filepath_to_data_uri,
|
||||||
trim_video,
|
trim_video,
|
||||||
|
upscale_video_to_min_pixels,
|
||||||
video_to_base64_string,
|
video_to_base64_string,
|
||||||
)
|
)
|
||||||
from .download_helpers import (
|
from .download_helpers import (
|
||||||
@ -88,16 +89,17 @@ __all__ = [
|
|||||||
"convert_mask_to_image",
|
"convert_mask_to_image",
|
||||||
"downscale_image_tensor",
|
"downscale_image_tensor",
|
||||||
"downscale_image_tensor_by_max_side",
|
"downscale_image_tensor_by_max_side",
|
||||||
|
"downscale_video_to_max_pixels",
|
||||||
"image_tensor_pair_to_batch",
|
"image_tensor_pair_to_batch",
|
||||||
"pil_to_bytesio",
|
"pil_to_bytesio",
|
||||||
"resize_mask_to_image",
|
"resize_mask_to_image",
|
||||||
"resize_video_to_pixel_budget",
|
|
||||||
"tensor_to_base64_string",
|
"tensor_to_base64_string",
|
||||||
"tensor_to_bytesio",
|
"tensor_to_bytesio",
|
||||||
"tensor_to_pil",
|
"tensor_to_pil",
|
||||||
"text_filepath_to_base64_string",
|
"text_filepath_to_base64_string",
|
||||||
"text_filepath_to_data_uri",
|
"text_filepath_to_data_uri",
|
||||||
"trim_video",
|
"trim_video",
|
||||||
|
"upscale_video_to_min_pixels",
|
||||||
"video_to_base64_string",
|
"video_to_base64_string",
|
||||||
# Validation utilities
|
# Validation utilities
|
||||||
"get_image_dimensions",
|
"get_image_dimensions",
|
||||||
|
|||||||
@ -415,14 +415,48 @@ def trim_video(video: Input.Video, duration_sec: float) -> Input.Video:
|
|||||||
raise RuntimeError(f"Failed to trim video: {str(e)}") from e
|
raise RuntimeError(f"Failed to trim video: {str(e)}") from e
|
||||||
|
|
||||||
|
|
||||||
def resize_video_to_pixel_budget(video: Input.Video, total_pixels: int) -> Input.Video:
|
def downscale_video_to_max_pixels(video: Input.Video, max_pixels: int) -> Input.Video:
|
||||||
"""Downscale a video to fit within ``total_pixels`` (w * h), preserving aspect ratio.
|
"""Downscale a video to fit within ``max_pixels`` (w * h), preserving aspect ratio.
|
||||||
|
|
||||||
Returns the original video object untouched when it already fits. Preserves frame rate, duration, and audio.
|
Returns the original video object untouched when it already fits. Preserves frame rate, duration, and audio.
|
||||||
Aspect ratio is preserved up to a fraction of a percent (even-dim rounding).
|
Aspect ratio is preserved up to a fraction of a percent (even-dim rounding).
|
||||||
"""
|
"""
|
||||||
src_w, src_h = video.get_dimensions()
|
src_w, src_h = video.get_dimensions()
|
||||||
scale_dims = _compute_downscale_dims(src_w, src_h, total_pixels)
|
scale_dims = _compute_downscale_dims(src_w, src_h, max_pixels)
|
||||||
|
if scale_dims is None:
|
||||||
|
return video
|
||||||
|
return _apply_video_scale(video, scale_dims)
|
||||||
|
|
||||||
|
|
||||||
|
def _compute_upscale_dims(src_w: int, src_h: int, total_pixels: int) -> tuple[int, int] | None:
|
||||||
|
"""Return upscaled (w, h) with even dims meeting at least ``total_pixels``, or None if already large enough.
|
||||||
|
|
||||||
|
Source aspect ratio is preserved; output may drift by a fraction of a percent because both dimensions
|
||||||
|
are rounded up to even values (many codecs require divisible-by-2). The result is guaranteed to be at
|
||||||
|
least ``total_pixels``.
|
||||||
|
"""
|
||||||
|
pixels = src_w * src_h
|
||||||
|
if pixels >= total_pixels:
|
||||||
|
return None
|
||||||
|
scale = math.sqrt(total_pixels / pixels)
|
||||||
|
new_w = math.ceil(src_w * scale)
|
||||||
|
new_h = math.ceil(src_h * scale)
|
||||||
|
if new_w % 2:
|
||||||
|
new_w += 1
|
||||||
|
if new_h % 2:
|
||||||
|
new_h += 1
|
||||||
|
return new_w, new_h
|
||||||
|
|
||||||
|
|
||||||
|
def upscale_video_to_min_pixels(video: Input.Video, min_pixels: int) -> Input.Video:
|
||||||
|
"""Upscale a video to meet at least ``min_pixels`` (w * h), preserving aspect ratio.
|
||||||
|
|
||||||
|
Returns the original video object untouched when it already meets the minimum. Preserves frame rate,
|
||||||
|
duration, and audio. Aspect ratio is preserved up to a fraction of a percent (even-dim rounding).
|
||||||
|
Note: upscaling a low-resolution source does not add real detail; downstream model quality may suffer.
|
||||||
|
"""
|
||||||
|
src_w, src_h = video.get_dimensions()
|
||||||
|
scale_dims = _compute_upscale_dims(src_w, src_h, min_pixels)
|
||||||
if scale_dims is None:
|
if scale_dims is None:
|
||||||
return video
|
return video
|
||||||
return _apply_video_scale(video, scale_dims)
|
return _apply_video_scale(video, scale_dims)
|
||||||
|
|||||||
@ -8,6 +8,82 @@ from comfy_api.latest import _io
|
|||||||
MISSING = object()
|
MISSING = object()
|
||||||
|
|
||||||
|
|
||||||
|
class NotNode(io.ComfyNode):
|
||||||
|
@classmethod
|
||||||
|
def define_schema(cls):
|
||||||
|
return io.Schema(
|
||||||
|
node_id="ComfyNotNode",
|
||||||
|
display_name="Not",
|
||||||
|
category="utils/logic",
|
||||||
|
description="Logical NOT operation. Returns true if the value is falsy. Uses Python's rules for truthiness.",
|
||||||
|
search_aliases=["invert", "toggle", "negate", "flip boolean"],
|
||||||
|
inputs=[
|
||||||
|
io.AnyType.Input("value"),
|
||||||
|
],
|
||||||
|
outputs=[
|
||||||
|
io.Boolean.Output(),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def execute(cls, value) -> io.NodeOutput:
|
||||||
|
return io.NodeOutput(not value)
|
||||||
|
|
||||||
|
|
||||||
|
class AndNode(io.ComfyNode):
|
||||||
|
@classmethod
|
||||||
|
def define_schema(cls):
|
||||||
|
template = io.Autogrow.TemplatePrefix(
|
||||||
|
input=io.AnyType.Input("value"),
|
||||||
|
prefix="value",
|
||||||
|
min=1,
|
||||||
|
)
|
||||||
|
return io.Schema(
|
||||||
|
node_id="ComfyAndNode",
|
||||||
|
display_name="And",
|
||||||
|
category="utils/logic",
|
||||||
|
description="Logical AND operation. Returns true if all of the values are truthy. Uses Python's rules for truthiness.",
|
||||||
|
search_aliases=["all", "every"],
|
||||||
|
inputs=[
|
||||||
|
io.Autogrow.Input("values", template=template),
|
||||||
|
],
|
||||||
|
outputs=[
|
||||||
|
io.Boolean.Output(),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def execute(cls, values: io.Autogrow.Type) -> io.NodeOutput:
|
||||||
|
return io.NodeOutput(all(values.values()))
|
||||||
|
|
||||||
|
|
||||||
|
class OrNode(io.ComfyNode):
|
||||||
|
@classmethod
|
||||||
|
def define_schema(cls):
|
||||||
|
template = io.Autogrow.TemplatePrefix(
|
||||||
|
input=io.AnyType.Input("value"),
|
||||||
|
prefix="value",
|
||||||
|
min=1,
|
||||||
|
)
|
||||||
|
return io.Schema(
|
||||||
|
node_id="ComfyOrNode",
|
||||||
|
display_name="Or",
|
||||||
|
category="utils/logic",
|
||||||
|
description="Logical OR operation. Returns true if any of the values are truthy. Uses Python's rules for truthiness.",
|
||||||
|
search_aliases=["any", "some"],
|
||||||
|
inputs=[
|
||||||
|
io.Autogrow.Input("values", template=template),
|
||||||
|
],
|
||||||
|
outputs=[
|
||||||
|
io.Boolean.Output(),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def execute(cls, values: io.Autogrow.Type) -> io.NodeOutput:
|
||||||
|
return io.NodeOutput(any(values.values()))
|
||||||
|
|
||||||
|
|
||||||
class SwitchNode(io.ComfyNode):
|
class SwitchNode(io.ComfyNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def define_schema(cls):
|
def define_schema(cls):
|
||||||
@ -261,6 +337,9 @@ class LogicExtension(ComfyExtension):
|
|||||||
return [
|
return [
|
||||||
SwitchNode,
|
SwitchNode,
|
||||||
CustomComboNode,
|
CustomComboNode,
|
||||||
|
NotNode,
|
||||||
|
AndNode,
|
||||||
|
OrNode,
|
||||||
# SoftSwitchNode,
|
# SoftSwitchNode,
|
||||||
# ConvertStringToComboNode,
|
# ConvertStringToComboNode,
|
||||||
# DCTestNode,
|
# DCTestNode,
|
||||||
|
|||||||
@ -1,5 +1,5 @@
|
|||||||
comfyui-frontend-package==1.43.18
|
comfyui-frontend-package==1.43.18
|
||||||
comfyui-workflow-templates==0.9.79
|
comfyui-workflow-templates==0.9.82
|
||||||
comfyui-embedded-docs==0.5.0
|
comfyui-embedded-docs==0.5.0
|
||||||
torch
|
torch
|
||||||
torchsde
|
torchsde
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user