61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""测试 Dify 原生 API 和 dify2openai 代理"""
|
||
|
|
import httpx
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
async def test_dify():
|
||
|
|
async with httpx.AsyncClient(timeout=30) as client:
|
||
|
|
# 1. 测试原生 API 路径
|
||
|
|
print("=== Test 1: Dify Native API (v1/chat-messages) ===")
|
||
|
|
url1 = "http://yw-dify.dc.servyou-it.com/v1/chat-messages"
|
||
|
|
headers1 = {
|
||
|
|
"Authorization": "Bearer app-7jkRkAzvX4QM9v9SM3P8mMEO",
|
||
|
|
"Content-Type": "application/json"
|
||
|
|
}
|
||
|
|
payload1 = {
|
||
|
|
"inputs": {"user_query": "hello"},
|
||
|
|
"query": "hello",
|
||
|
|
"response_mode": "blocking",
|
||
|
|
"user": "test-user-001"
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
r1 = await client.post(url1, json=payload1, headers=headers1)
|
||
|
|
print(f"Status: {r1.status_code}")
|
||
|
|
print(f"Response: {r1.text[:300]}")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error: {e}")
|
||
|
|
|
||
|
|
print()
|
||
|
|
|
||
|
|
# 2. 测试 dify2openai 代理路径
|
||
|
|
print("=== Test 2: dify2openai Proxy ===")
|
||
|
|
url2 = "http://yw-dify.dc.servyou-it.com/dify2openai/v1/chat/completions"
|
||
|
|
headers2 = {
|
||
|
|
"Authorization": "Bearer http://yw-dify.dc.servyou-it.com/v1|app-7jkRkAzvX4QM9v9SM3P8mMEO|Chat",
|
||
|
|
"Content-Type": "application/json"
|
||
|
|
}
|
||
|
|
payload2 = {
|
||
|
|
"model": "Chat",
|
||
|
|
"messages": [{"role": "user", "content": "hello"}],
|
||
|
|
"stream": False
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
r2 = await client.post(url2, json=payload2, headers=headers2)
|
||
|
|
print(f"Status: {r2.status_code}")
|
||
|
|
print(f"Response: {r2.text[:300]}")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error: {e}")
|
||
|
|
|
||
|
|
print()
|
||
|
|
|
||
|
|
# 3. 测试 Dify 健康检查
|
||
|
|
print("=== Test 3: Dify Health Check ===")
|
||
|
|
try:
|
||
|
|
r3 = await client.get("http://yw-dify.dc.servyou-it.com/")
|
||
|
|
print(f"Status: {r3.status_code}")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error: {e}")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(test_dify())
|