OpenAI: Chat

# Python 3.10+: pip install comfy-sdk==0.4.0
import os

from comfy_sdk import Comfy

api_key = os.environ.get("COMFY_API_KEY", "").strip()
if not api_key: raise RuntimeError("Set COMFY_API_KEY")
os.environ.setdefault("COMFY_BASE_URL", "https://cloud.comfy.org")
client = Comfy(api_key=api_key)
workflow = client.workflows.from_json({
    "2": {
        "class_type": "PreviewAny",
        "inputs": {
            "source": [
                "3",
                0
            ]
        },
        "_meta": {
            "title": "Preview as Text"
        }
    },
    "3": {
        "class_type": "OpenAIChatNode",
        "inputs": {
            "advanced_options": [
                "4",
                0
            ],
            "images": [
                "5",
                0
            ],
            "model": "o1-pro",
            "persist_context": False,
            "prompt": "Describe this image"
        },
        "_meta": {
            "title": "OpenAI ChatGPT"
        }
    },
    "4": {
        "class_type": "OpenAIChatConfig",
        "inputs": {
            "instructions": "You are an AI specialized in image-to-text prompt generation. Analyze the given image carefully and describe it in a detailed, structured, and prompt-friendly format. Include elements such as subject, clothing, style, environment, lighting, composition, color palette, mood, camera angle, and any other relevant visual details. Your output should resemble a high-quality text-to-image prompt that can help recreate the image. Use concise, descriptive language. Do not include explanations, just output the prompt.",
            "max_output_tokens": 4096,
            "reasoning_effort": "default",
            "truncation": "auto"
        },
        "_meta": {
            "title": "OpenAI ChatGPT Advanced Options"
        }
    },
    "5": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "UPLOADED_image_FILENAME"
        },
        "_meta": {
            "title": "Load Image"
        }
    }
})
# Replace each example.com URL with a public URL of your file.
workflow.set_input("5", "image", client.assets.from_url("https://example.com/replace-with-a-url/image"))

# run() also passes api_key to the partner nodes in this workflow.
job = client.run(workflow, api_key=api_key)