df18925e8d
Preserve existing workflows while adding capability-based catalogs, dedicated image requests, resumable native video, validated audio and explicit chat tiers. Cover compatibility, billing-sensitive retries, credentials and cancellation. Co-authored-by: Arthur Reboul Salze <22814805+ArthurReboulSalze@users.noreply.github.com> Co-authored-by: Dookie <me@djdookie.de>
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from .node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
|
|
|
|
def register_routes():
|
|
"""Expose public metadata and server-side credits without returning a key."""
|
|
try:
|
|
from server import PromptServer
|
|
from aiohttp import web
|
|
except ImportError:
|
|
return # Standalone tests and metadata tools do not run a Comfy server.
|
|
if not hasattr(PromptServer, "instance"):
|
|
return
|
|
import asyncio
|
|
from . import openrouter_catalog
|
|
from .node import OpenRouterNode
|
|
|
|
@PromptServer.instance.routes.get("/openrouter/model_catalog")
|
|
async def model_catalog(request):
|
|
# This only schedules background work and copies the last snapshot.
|
|
snapshot = openrouter_catalog.get_catalog(refresh=request.query.get("refresh") == "1")
|
|
snapshot["video"] = openrouter_catalog.video_generation_models(snapshot)
|
|
return web.json_response(snapshot)
|
|
|
|
@PromptServer.instance.routes.get("/openrouter/credits")
|
|
async def credits(request):
|
|
def fetch():
|
|
node = OpenRouterNode.__new__(OpenRouterNode)
|
|
return node.fetch_credits("", timeout=15)
|
|
return web.json_response({"credits": await asyncio.to_thread(fetch)})
|
|
|
|
|
|
register_routes()
|
|
|