mirror of
https://github.com/langgenius/dify.git
synced 2026-07-15 01:17:04 +08:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
"""
|
|
Helper module for Creators Platform integration.
|
|
|
|
Provides functionality to upload DSL files to the Creators Platform
|
|
and generate redirect URLs with OAuth authorization codes.
|
|
"""
|
|
|
|
import logging
|
|
from urllib.parse import urlencode
|
|
|
|
import httpx
|
|
from yarl import URL
|
|
|
|
from configs import dify_config
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
creators_platform_api_url = URL(str(dify_config.CREATORS_PLATFORM_API_URL))
|
|
|
|
|
|
def upload_dsl(dsl_file_bytes: bytes, filename: str = "template.yaml") -> str:
|
|
url = str(creators_platform_api_url / "api/v1/templates/anonymous-upload")
|
|
response = httpx.post(url, files={"file": (filename, dsl_file_bytes)}, timeout=30)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
claim_code = data.get("data", {}).get("claim_code")
|
|
if not isinstance(claim_code, str) or not claim_code:
|
|
raise ValueError("Creators Platform did not return a valid claim_code")
|
|
return claim_code
|
|
|
|
|
|
def get_redirect_url(user_account_id: str, claim_code: str) -> str:
|
|
base_url = str(dify_config.CREATORS_PLATFORM_API_URL).rstrip("/")
|
|
params: dict[str, str] = {"dsl_claim_code": claim_code}
|
|
client_id = str(dify_config.CREATORS_PLATFORM_OAUTH_CLIENT_ID or "")
|
|
if client_id:
|
|
from services.oauth_server import OAuthServerService
|
|
|
|
oauth_code = OAuthServerService.sign_oauth_authorization_code(client_id, user_account_id)
|
|
params["oauth_code"] = oauth_code
|
|
return f"{base_url}?{urlencode(params)}"
|