freedak/OpenMontage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

comfyui: satisfy provider contract review items

Browse Source
This commit is contained in:
calesthio
2026-04-23 17:49:04 -07:00
committed by Alastair Beal
parent 4c62186c95
commit 7c4bb08890
9 changed files with 1152 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/tool_registry.py
+46
View File
@@ -282,6 +282,7 @@ class ToolRegistry:
"provider": tool.provider,
"runtime": tool.runtime.value,
"best_for": tool.best_for,
"dependencies": info.get("dependencies", []),
"install_instructions": tool.install_instructions,
"status": status.value,
}
@@ -291,6 +292,10 @@ class ToolRegistry:
"render_engines",
"remotion_note",
"provider_matrix",
"setup_offer",
"operation_statuses",
"resource_profiles",
"resource_profile_note",
):
if extra_key in info:
entry[extra_key] = info[extra_key]
@@ -398,6 +403,40 @@ class ToolRegistry:
setup_offers: list[dict[str, Any]] = []
for cap, bucket in menu.items():
for entry in bucket.get("unavailable", []):
offer = entry.get("setup_offer")
if offer:
setup_offers.append(
{
"capability": cap,
"tool": entry.get("name"),
"provider": entry.get("provider"),
"runtime": entry.get("runtime"),
"install_instructions": entry.get("install_instructions") or "",
**offer,
}
)
continue
env_vars = [
dep[4:]
for dep in entry.get("dependencies", [])
if isinstance(dep, str) and dep.startswith("env:")
]
if env_vars:
setup_offers.append(
{
"capability": cap,
"tool": entry.get("name"),
"provider": entry.get("provider"),
"runtime": entry.get("runtime"),
"kind": "env_var",
"fix_complexity": "1-minute env-var",
"env_vars": env_vars,
"install_instructions": entry.get("install_instructions") or "",
}
)
continue
hint = entry.get("install_instructions") or ""
# Heuristic: 1-minute fixes mention an env var or API key.
if any(k in hint.lower() for k in ["api key", "env", "_key=", "_api"]):
@@ -406,10 +445,17 @@ class ToolRegistry:
"capability": cap,
"tool": entry.get("name"),
"provider": entry.get("provider"),
"runtime": entry.get("runtime"),
"install_instructions": hint,
}
)
for entry in bucket.get("available", []) + bucket.get("unavailable", []):
if entry.get("resource_profile_note"):
runtime_warnings.append(
f"{entry.get('name')}: {entry.get('resource_profile_note')}"
)
result = {
"composition_runtimes": comp_runtimes,
"capabilities": capabilities,