#!/usr/bin/env python3
"""Test multiple Dify API keys to find a working one"""
import httpx
import json
import time

base_url = "http://yw-dify.dc.servyou-it.com"

# Test multiple API keys
api_keys = [
    ("app-7jkRkAzvX4QM9v9SM3P8mMEO", "审批意图副本"),
    ("app-J3s8sHarZQ2SCaNF3xCppliL", "自建应用"),
    ("app-z3S9AEUUAVPbtR2rioxpiIvp", "分诊应用"),
]

for api_key, name in api_keys:
    url = f"{base_url}/v1/chat-messages"
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }
    payload = {
        "inputs": {},
        "query": "密码忘记了怎么办",
        "response_mode": "blocking",
        "user": "test_verify"
    }

    print(f"=== Testing: {name} ({api_key[:20]}...) ===")
    start = time.time()
    try:
        with httpx.Client(timeout=httpx.Timeout(30.0)) as client:
            resp = client.post(url, headers=headers, json=payload)
        elapsed = (time.time() - start) * 1000
        print(f"Status: {resp.status_code}, Time: {elapsed:.0f}ms")

        if resp.status_code == 200:
            data = resp.json()
            answer = data.get('answer', '')
            print(f"conversation_id: {data.get('conversation_id', 'N/A')}")
            print(f"answer (first 300 chars): {answer[:300]}")

            # Try JSON parse
            try:
                parsed = json.loads(answer)
                print(f"JSON Parse: SUCCESS - keys: {list(parsed.keys())}")
            except:
                print(f"JSON Parse: FAILED (plain text)")
        else:
            print(f"Error: {resp.text[:300]}")
    except Exception as e:
        print(f"Exception: {type(e).__name__}: {e}")

    print()