dify/scripts/stress-test/setup/publish_workflow.py

#!/usr/bin/env python3

import sys
from pathlib import Path

sys.path.append(str(Path(__file__).parent.parent))

import httpx
import json
from common import config_helper
from common import Logger


def publish_workflow() -> None:
    """Publish the imported workflow app."""

    log = Logger("PublishWorkflow")
    log.header("Publishing Workflow")

    # Read token from config
    access_token = config_helper.get_token()
    if not access_token:
        log.error("No access token found in config")
        return

    # Read app_id from config
    app_id = config_helper.get_app_id()
    if not app_id:
        log.error("No app_id found in config")
        return

    log.step(f"Publishing workflow for app: {app_id}")

    # API endpoint for publishing workflow
    base_url = "http://localhost:5001"
    publish_endpoint = f"{base_url}/console/api/apps/{app_id}/workflows/publish"

    # Publish payload
    publish_payload = {"marked_name": "", "marked_comment": ""}

    headers = {
        "Accept": "*/*",
        "Accept-Language": "en-US,en;q=0.9",
        "Cache-Control": "no-cache",
        "Connection": "keep-alive",
        "DNT": "1",
        "Origin": "http://localhost:3000",
        "Pragma": "no-cache",
        "Referer": "http://localhost:3000/",
        "Sec-Fetch-Dest": "empty",
        "Sec-Fetch-Mode": "cors",
        "Sec-Fetch-Site": "same-site",
        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/139.0.0.0 Safari/537.36",
        "authorization": f"Bearer {access_token}",
        "content-type": "application/json",
        "sec-ch-ua": '"Not;A=Brand";v="99", "Google Chrome";v="139", "Chromium";v="139"',
        "sec-ch-ua-mobile": "?0",
        "sec-ch-ua-platform": '"macOS"',
    }

    cookies = {"locale": "en-US"}

    try:
        # Make the publish request
        with httpx.Client() as client:
            response = client.post(
                publish_endpoint,
                json=publish_payload,
                headers=headers,
                cookies=cookies,
            )

            if response.status_code == 200 or response.status_code == 201:
                log.success("Workflow published successfully!")
                log.key_value("App ID", app_id)

                # Try to parse response if it has JSON content
                if response.text:
                    try:
                        response_data = response.json()
                        if response_data:
                            log.debug(
                                f"Response: {json.dumps(response_data, indent=2)}"
                            )
                    except json.JSONDecodeError:
                        # Response might be empty or non-JSON
                        pass

            elif response.status_code == 401:
                log.error("Workflow publish failed: Unauthorized")
                log.info("Token may have expired. Please run login_admin.py again")
            elif response.status_code == 404:
                log.error("Workflow publish failed: App not found")
                log.info("Make sure the app was imported successfully")
            else:
                log.error(
                    f"Workflow publish failed with status code: {response.status_code}"
                )
                log.debug(f"Response: {response.text}")

    except httpx.ConnectError:
        log.error("Could not connect to Dify API at http://localhost:5001")
        log.info("Make sure the API server is running with: ./dev/start-api")
    except Exception as e:
        log.error(f"An error occurred: {e}")


if __name__ == "__main__":
    publish_workflow()