89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
测试修复后的 ASR 端点
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
import os
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
|
|||
|
|
# 加载环境变量
|
|||
|
|
load_dotenv()
|
|||
|
|
|
|||
|
|
def test_asr_endpoint():
|
|||
|
|
"""测试 ASR 端点"""
|
|||
|
|
try:
|
|||
|
|
# 创建一个测试音频文件
|
|||
|
|
import wave
|
|||
|
|
import numpy as np
|
|||
|
|
|
|||
|
|
# 生成3秒的测试音频（16kHz, 单声道）
|
|||
|
|
sample_rate = 16000
|
|||
|
|
duration = 3
|
|||
|
|
frequency = 440 # A4音符
|
|||
|
|
|
|||
|
|
t = np.linspace(0, duration, sample_rate * duration, False)
|
|||
|
|
audio_data = np.sin(2 * np.pi * frequency * t) * 0.3
|
|||
|
|
audio_data = (audio_data * 32767).astype(np.int16)
|
|||
|
|
|
|||
|
|
# 保存为WAV文件
|
|||
|
|
with wave.open('test_audio.wav', 'wb') as wav_file:
|
|||
|
|
wav_file.setnchannels(1) # 单声道
|
|||
|
|
wav_file.setsampwidth(2) # 16-bit
|
|||
|
|
wav_file.setframerate(sample_rate) # 16kHz
|
|||
|
|
wav_file.writeframes(audio_data.tobytes())
|
|||
|
|
|
|||
|
|
print("✅ 创建测试音频文件")
|
|||
|
|
|
|||
|
|
# 测试ASR端点
|
|||
|
|
url = "http://192.168.1.141:30101/voice/call/asr"
|
|||
|
|
|
|||
|
|
# 获取token（如果需要）
|
|||
|
|
token = os.getenv('TEST_TOKEN', '')
|
|||
|
|
headers = {}
|
|||
|
|
if token:
|
|||
|
|
headers['Authorization'] = f'Bearer {token}'
|
|||
|
|
|
|||
|
|
print(f"🚀 测试 ASR 端点: {url}")
|
|||
|
|
|
|||
|
|
with open('test_audio.wav', 'rb') as f:
|
|||
|
|
files = {'audio': ('test_audio.wav', f, 'audio/wav')}
|
|||
|
|
|
|||
|
|
print("📤 发送请求...")
|
|||
|
|
response = requests.post(url, files=files, headers=headers, timeout=60)
|
|||
|
|
|
|||
|
|
print(f"📊 响应状态码: {response.status_code}")
|
|||
|
|
print(f"📋 响应内容: {response.text}")
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
data = response.json()
|
|||
|
|
if data.get('code') == 1 and data.get('data', {}).get('text'):
|
|||
|
|
print(f"✅ ASR 成功: {data['data']['text']}")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
print(f"⚠️ ASR 响应格式异常: {data}")
|
|||
|
|
return True # 仍然算成功，因为没有超时
|
|||
|
|
else:
|
|||
|
|
print(f"❌ ASR 请求失败: {response.status_code}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
except requests.exceptions.Timeout:
|
|||
|
|
print("❌ 请求超时")
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 测试失败: {e}")
|
|||
|
|
return False
|
|||
|
|
finally:
|
|||
|
|
# 清理测试文件
|
|||
|
|
try:
|
|||
|
|
os.remove('test_audio.wav')
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
print("🚀 开始测试修复后的 ASR 端点...")
|
|||
|
|
|
|||
|
|
if test_asr_endpoint():
|
|||
|
|
print("🎉 ASR 端点测试成功！")
|
|||
|
|
else:
|
|||
|
|
print("💥 ASR 端点测试失败！")
|