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

#!/usr/bin/env python3

import sys
from pathlib import Path

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

import httpx
from common import config_helper
from common import Logger


def configure_openai_plugin() -> None:
    """Configure OpenAI plugin with mock server credentials."""

    log = Logger("ConfigPlugin")
    log.header("Configuring OpenAI Plugin")

    # Read token from config
    access_token = config_helper.get_token()
    if not access_token:
        log.error("No access token found in config")
        log.info("Please run login_admin.py first to get access token")
        return

    log.step("Configuring OpenAI plugin with mock server...")

    # API endpoint for plugin configuration
    base_url = "http://localhost:5001"
    config_endpoint = f"{base_url}/console/api/workspaces/current/model-providers/langgenius/openai/openai/credentials"

    # Configuration payload with mock server
    config_payload = {
        "credentials": {
            "openai_api_key": "apikey",
            "openai_organization": None,
            "openai_api_base": "http://host.docker.internal:5004",
        }
    }

    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 configuration request
        with httpx.Client() as client:
            response = client.post(
                config_endpoint,
                json=config_payload,
                headers=headers,
                cookies=cookies,
            )

            if response.status_code == 200:
                log.success("OpenAI plugin configured successfully!")
                log.key_value(
                    "API Base", config_payload["credentials"]["openai_api_base"]
                )
                log.key_value(
                    "API Key", config_payload["credentials"]["openai_api_key"]
                )

            elif response.status_code == 201:
                log.success("OpenAI plugin credentials created successfully!")
                log.key_value(
                    "API Base", config_payload["credentials"]["openai_api_base"]
                )
                log.key_value(
                    "API Key", config_payload["credentials"]["openai_api_key"]
                )

            elif response.status_code == 401:
                log.error("Configuration failed: Unauthorized")
                log.info("Token may have expired. Please run login_admin.py again")
            else:
                log.error(
                    f"Configuration 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__":
    configure_openai_plugin()