63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""REST test for Qwen3-TTS /v1/audio/speech endpoint."""
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
import requests
|
||
|
|
|
||
|
|
BASE_URL = "http://localhost:8091"
|
||
|
|
OUTPUT_FILE = "test_output.wav"
|
||
|
|
|
||
|
|
|
||
|
|
def check_health():
|
||
|
|
try:
|
||
|
|
r = requests.get(f"{BASE_URL}/health", timeout=5)
|
||
|
|
r.raise_for_status()
|
||
|
|
body = r.json() if r.content else "(leer)"
|
||
|
|
print(f"[OK] /health → HTTP {r.status_code} {body}")
|
||
|
|
return True
|
||
|
|
except Exception as e:
|
||
|
|
print(f"[FAIL] /health: {e}")
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def test_tts(text: str, voice: str = "Ryan", language: str = "German"):
|
||
|
|
payload = {
|
||
|
|
"model": "Qwen/Qwen3-TTS-12Hz-1.7B-CustomVoice",
|
||
|
|
"input": text,
|
||
|
|
"voice": voice,
|
||
|
|
"language": language,
|
||
|
|
"response_format": "wav",
|
||
|
|
}
|
||
|
|
print(f"\n[→] POST /v1/audio/speech")
|
||
|
|
print(f" text: {text!r}")
|
||
|
|
print(f" voice: {voice}, language: {language}")
|
||
|
|
|
||
|
|
t0 = time.perf_counter()
|
||
|
|
r = requests.post(f"{BASE_URL}/v1/audio/speech", json=payload, timeout=60)
|
||
|
|
elapsed = time.perf_counter() - t0
|
||
|
|
|
||
|
|
if r.status_code != 200:
|
||
|
|
print(f"[FAIL] HTTP {r.status_code}: {r.text[:300]}")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
with open(OUTPUT_FILE, "wb") as f:
|
||
|
|
f.write(r.content)
|
||
|
|
|
||
|
|
kb = len(r.content) / 1024
|
||
|
|
print(f"[OK] {elapsed:.2f}s → {kb:.1f} KB WAV saved to {OUTPUT_FILE}")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
print(f"Testing Qwen3-TTS at {BASE_URL}\n")
|
||
|
|
|
||
|
|
if not check_health():
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
test_tts(
|
||
|
|
text="Hallo! Ich bin ein KI-Sprachassistent. Wie kann ich Ihnen helfen?",
|
||
|
|
voice="Ryan",
|
||
|
|
language="German",
|
||
|
|
)
|
||
|
|
|
||
|
|
print("\nAll tests passed.")
|