From 535b1136c5ffcfac50a4b3b642275cf41894d394 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Fri, 22 Aug 2025 09:22:42 +0800
Subject: [PATCH] 'commit'

---
 .../Config/__pycache__/Config.cpython-310.pyc | Bin 2087 -> 2174 bytes
 dsLightRag/Test/auc_websocket_demo.py         | 116 ----------------
 dsLightRag/Util/ASRClient.py                  | 128 ++++++++++++++++++
 3 files changed, 128 insertions(+), 116 deletions(-)
 delete mode 100644 dsLightRag/Test/auc_websocket_demo.py
 create mode 100644 dsLightRag/Util/ASRClient.py

diff --git a/dsLightRag/Config/__pycache__/Config.cpython-310.pyc b/dsLightRag/Config/__pycache__/Config.cpython-310.pyc
index 68346443072d3d46785061b5462cfadc8505a8d4..b834a481475eaa95eee9bdfe413ea6199cc4edec 100644
GIT binary patch
delta 258
zcmZ23@K1m*pO=@50SI0lUY^0lJ&{j_@zg}^!#q*iDLPR)DZ0T7ntBsIMKIpl?9KR|
zQ80?5AiuaIy(qOfin%B;y-IKLL?+qI8O#$I4WqaWO%07rjg1T~&8ieiGL8Jgb+ePQ
zJd?7klFgjFO-&P%OY%J9%R_=JDzZKNG<7%sVX<L!jN%RO4-Rn;at)6605Xuc0scYA
z+|XdxpeRAO;sD3s;BfySm#E2=Z2#15@p=TuI|c{EI|c;Ad%E1>fpSCqy<PoQG8C}^
ZeOSacxq;o8F=g@*_Cz)oMiv$hMgYK6PI>?U

delta 161
zcmew-uv~yIpO=@50SJBzF3IrWn#d=^cxa;bVU}P9O`VCqA{e7K2Q$8B<i5pGkY8Mq
zUX)sVYqKx&7Di@2P3_G_tTv2}w|E2mgG1bdT!Z61fD9yVfPWA&H#FEa=$0T{ae!lR
kaJYYv%dN@l+5T~^WGDi;qKIqqWp-!AgvsI@iA)@f08WH5+yDRo

diff --git a/dsLightRag/Test/auc_websocket_demo.py b/dsLightRag/Test/auc_websocket_demo.py
deleted file mode 100644
index 064199b3..00000000
--- a/dsLightRag/Test/auc_websocket_demo.py
+++ /dev/null
@@ -1,116 +0,0 @@
-import json
-import time
-import uuid
-import requests
-
-from Config.Config import HS_ASR_APP_ID, HS_ASR_TOKEN
-
-
-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
-
-appid = HS_ASR_APP_ID
-token = HS_ASR_TOKEN
-
-if __name__ == '__main__':
-    main()
diff --git a/dsLightRag/Util/ASRClient.py b/dsLightRag/Util/ASRClient.py
new file mode 100644
index 00000000..a75ea933
--- /dev/null
+++ b/dsLightRag/Util/ASRClient.py
@@ -0,0 +1,128 @@
+import json
+import time
+import uuid
+import requests
+
+from Config.Config import HS_ASR_APP_ID, HS_ASR_TOKEN
+
+
+class ASRClient:
+    def __init__(self, appid=None, token=None, file_url=None):
+        self.appid = appid or HS_ASR_APP_ID
+        self.token = token or HS_ASR_TOKEN
+        self.file_url = file_url or "https://ttc-advisory-oss.oss-cn-hangzhou.aliyuncs.com/lark_audio/int/T_APLA_1941058348698869760.mp3"
+        self.submit_url = "https://openspeech-direct.zijieapi.com/api/v3/auc/bigmodel/submit"
+        self.query_url = "https://openspeech-direct.zijieapi.com/api/v3/auc/bigmodel/query"
+        self.task_id = None
+        self.x_tt_logid = None
+
+    def _prepare_headers(self, task_id=None):
+        """准备请求头"""
+        headers = {
+            "X-Api-App-Key": self.appid,
+            "X-Api-Access-Key": self.token,
+            "X-Api-Resource-Id": "volc.bigasr.auc",
+            "X-Api-Request-Id": task_id or str(uuid.uuid4()),
+        }
+        if task_id and self.x_tt_logid:
+            headers["X-Tt-Logid"] = self.x_tt_logid
+        else:
+            headers["X-Api-Sequence"] = "-1"
+        return headers
+
+    def _prepare_request_body(self):
+        """准备请求体"""
+        return {
+            "user": {
+                "uid": "fake_uid"
+            },
+            "audio": {
+                "url": self.file_url
+            },
+            "request": {
+                "model_name": "bigmodel",
+                "enable_channel_split": True,
+                "enable_ddc": True,
+                "enable_speaker_info": True,
+                "enable_punc": True,
+                "enable_itn": True,
+                "corpus": {
+                    "correct_table_name": "",
+                    "context": ""
+                }
+            }
+        }
+
+    def submit_task(self):
+        """提交ASR任务"""
+        headers = self._prepare_headers()
+        self.task_id = headers["X-Api-Request-Id"]
+        request_body = self._prepare_request_body()
+
+        print(f'Submit task id: {self.task_id}')
+        response = requests.post(self.submit_url, data=json.dumps(request_body), 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"]}')
+            self.x_tt_logid = response.headers.get("X-Tt-Logid", "")
+            print(f'Submit task response header X-Tt-Logid: {self.x_tt_logid}\n')
+            return self.task_id, self.x_tt_logid
+        else:
+            print(f'Submit task failed and the response headers are: {response.headers}')
+            exit(1)
+
+    def query_task(self):
+        """查询ASR任务状态"""
+        if not self.task_id or not self.x_tt_logid:
+            print("Task not submitted yet. Please call submit_task() first.")
+            exit(1)
+
+        headers = self._prepare_headers(self.task_id)
+        response = requests.post(self.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 extract_text_from_result(self, result):
+        """从识别结果中提取文本内容"""
+        if isinstance(result, dict) and 'result' in result and 'text' in result['result']:
+            return result['result']['text']
+        return ""
+
+    def process_task(self):
+        """处理ASR任务：提交并轮询结果"""
+        self.submit_task()
+        while True:
+            query_response = self.query_task()
+            code = query_response.headers.get('X-Api-Status-Code', "")
+            if code == '20000000':  # task finished
+                result_json = query_response.json()
+                # 提取并返回文本内容
+                text_content = self.extract_text_from_result(result_json)
+                print(f"识别文本: {text_content}")
+                return text_content
+            elif code != '20000001' and code != '20000002':  # task failed
+                print("FAILED!")
+                exit(1)
+            time.sleep(1)
+
+
+def main(audio_url=None):
+    # 创建ASR客户端实例，传入指定的音频文件URL
+    asr_client = ASRClient(file_url=audio_url)
+    # 处理ASR任务并获取文本结果
+    text_result = asr_client.process_task()
+    return text_result
+
+
+if __name__ == '__main__':
+    # 使用指定的音频文件URL进行测试
+    audio_url = "https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Temp/temp_audio_20250822085243.wav"
+    main(audio_url)