diff --git a/dsLightRag/Routes/XueBan.py b/dsLightRag/Routes/XueBan.py index 9a7401dd..e46ec4bd 100644 --- a/dsLightRag/Routes/XueBan.py +++ b/dsLightRag/Routes/XueBan.py @@ -46,8 +46,8 @@ async def upload_audio(file: UploadFile = File(...)): } # 删除临时文件 - #os.remove(temp_file_path) - #logger.info(f"临时文件已删除: {temp_file_path}") + os.remove(temp_file_path) + logger.info(f"临时文件已删除: {temp_file_path}") # 返回识别结果 return JSONResponse(content={ diff --git a/dsLightRag/Test/auc_websocket_demo.py b/dsLightRag/Test/auc_websocket_demo.py new file mode 100644 index 00000000..2cba28c5 --- /dev/null +++ b/dsLightRag/Test/auc_websocket_demo.py @@ -0,0 +1,115 @@ +import json +import time +import uuid +import requests + + +def submit_task(): + submit_url = "https://openspeech-direct.zijieapi.com/api/v3/auc/bigmodel/submit" + + task_id = str(uuid.uuid4()) + + headers = { + "X-Api-App-Key": appid, + "X-Api-Access-Key": token, + "X-Api-Resource-Id": "volc.bigasr.auc", + "X-Api-Request-Id": task_id, + "X-Api-Sequence": "-1" + } + + request = { + "user": { + "uid": "fake_uid" + }, + "audio": { + "url": file_url, + # "format": "mp3", + # "codec": "map3", + # "rate": 48000, + # "bits": 16, + #"channel": 2 + }, + "request": { + "model_name": "bigmodel", + # "model_name": "bigmodel", + "enable_channel_split": True, + "enable_ddc": True, + "enable_speaker_info": True, + "enable_punc": True, + "enable_itn": True, + # "enable_itn": True, + # "enable_punc": True, + # "enable_ddc": True, + # "show_utterances": True, + # "enable_channel_split": True, + # "vad_segment": True, + # "enable_speaker_info": True, + "corpus": { + # "boosting_table_name": "test", + "correct_table_name": "", + "context": "" + } + } + } + print(f'Submit task id: {task_id}') + response = requests.post(submit_url, data=json.dumps(request), headers=headers) + if 'X-Api-Status-Code' in response.headers and response.headers["X-Api-Status-Code"] == "20000000": + print(f'Submit task response header X-Api-Status-Code: {response.headers["X-Api-Status-Code"]}') + print(f'Submit task response header X-Api-Message: {response.headers["X-Api-Message"]}') + x_tt_logid = response.headers.get("X-Tt-Logid", "") + print(f'Submit task response header X-Tt-Logid: {response.headers["X-Tt-Logid"]}\n') + return task_id, x_tt_logid + else: + print(f'Submit task failed and the response headers are: {response.headers}') + exit(1) + return task_id + + +def query_task(task_id, x_tt_logid): + query_url = "https://openspeech-direct.zijieapi.com/api/v3/auc/bigmodel/query" + + headers = { + "X-Api-App-Key": appid, + "X-Api-Access-Key": token, + "X-Api-Resource-Id": "volc.bigasr.auc", + "X-Api-Request-Id": task_id, + "X-Tt-Logid": x_tt_logid # 固定传递 x-tt-logid + } + + response = requests.post(query_url, json.dumps({}), headers=headers) + + if 'X-Api-Status-Code' in response.headers: + print(f'Query task response header X-Api-Status-Code: {response.headers["X-Api-Status-Code"]}') + print(f'Query task response header X-Api-Message: {response.headers["X-Api-Message"]}') + print(f'Query task response header X-Tt-Logid: {response.headers["X-Tt-Logid"]}\n') + else: + print(f'Query task failed and the response headers are: {response.headers}') + exit(1) + return response + + +def main(): + task_id, x_tt_logid = submit_task() + while True: + query_response = query_task(task_id, x_tt_logid) + code = query_response.headers.get('X-Api-Status-Code', "") + if code == '20000000': # task finished + print(query_response.json()) + print("SUCCESS!") + exit(0) + elif code != '20000001' and code != '20000002': # task failed + print("FAILED!") + exit(1) + time.sleep(1) + +# 需要使用在线url,推荐使用TOS +file_url = "https://ttc-advisory-oss.oss-cn-hangzhou.aliyuncs.com/lark_audio/int/T_APLA_1941058348698869760.mp3" + +# 填入控制台获取的app id和access token + +# https://console.volcengine.com/speech/service/10012?AppID=1513532196 +appid = "1513532196" +token = "ti2NW-kbjIbkzc6BK55actnI_wTR8xkH" + +if __name__ == '__main__': + main()