ethanfelandClaude Opus 4.8
fbd9087758
push.py and console_relay.py take --ip (default 127.0.0.1) and --port (9188), building http://<ip>:<port>/mcp; --url still overrides. console_relay.sh takes the IP as its first arg (./console_relay.sh 192.168.1.12) or prompts for it. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
57 lines
2.1 KiB
Python
Executable File
57 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Manual test client: push a message to a bridge channel.
|
|
|
|
Usage:
|
|
python examples/push.py <channel> <text> [--image PATH] [--seed N] [--url URL]
|
|
|
|
Examples:
|
|
python examples/push.py sxcp_eval_out "a red fox in autumn leaves"
|
|
python examples/push.py sxcp_eval_out "result" --image /path/out.png --seed 123
|
|
COMFY_BRIDGE_URL=http://192.168.1.12:9188/mcp python examples/push.py main "hi"
|
|
|
|
Needs: pip install "mcp>=1.2.0"
|
|
Default URL: $COMFY_BRIDGE_URL or http://127.0.0.1:9188/mcp
|
|
(This is only a manual tester — a real agent calls comfy_push as a native tool.)
|
|
"""
|
|
import argparse
|
|
import asyncio
|
|
import os
|
|
|
|
from mcp.client.streamable_http import streamablehttp_client
|
|
from mcp import ClientSession
|
|
|
|
|
|
async def push(url, channel, text, image, seed):
|
|
args = {"channel": channel}
|
|
if text is not None:
|
|
args["text"] = text
|
|
if image is not None:
|
|
args["image_path"] = image
|
|
if seed is not None:
|
|
args["seed"] = seed
|
|
async with streamablehttp_client(url) as (r, w, _):
|
|
async with ClientSession(r, w) as s:
|
|
await s.initialize()
|
|
res = await s.call_tool("comfy_push", args)
|
|
print(res.content[0].text if res.content else "(no content)")
|
|
|
|
|
|
def main():
|
|
p = argparse.ArgumentParser(description="Push a message to a ComfyUI bridge channel.")
|
|
p.add_argument("channel")
|
|
p.add_argument("text", nargs="?", default=None)
|
|
p.add_argument("--image", default=None, help="image path (must be openable by ComfyUI)")
|
|
p.add_argument("--seed", type=int, default=None)
|
|
p.add_argument("--ip", default=os.environ.get("COMFY_BRIDGE_IP", "127.0.0.1"),
|
|
help="bridge host/IP (default 127.0.0.1)")
|
|
p.add_argument("--port", default=os.environ.get("COMFY_BRIDGE_PORT", "9188"))
|
|
p.add_argument("--url", default=os.environ.get("COMFY_BRIDGE_URL"),
|
|
help="full MCP URL (overrides --ip/--port)")
|
|
a = p.parse_args()
|
|
url = a.url or f"http://{a.ip}:{a.port}/mcp"
|
|
asyncio.run(push(url, a.channel, a.text, a.image, a.seed))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|