init

custom_components/tencentcloud_asr/tencentcloud_api.py

@@ -0,0 +1,54 @@
+import logging
+from homeassistant.components.stt import AudioFormats
+from tencentcloud.common import credential
+from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
+from tencentcloud.asr.v20190614.asr_client import AsrClient
+from tencentcloud.asr.v20190614.models import SentenceRecognitionRequest, SentenceRecognitionResponse
+
+_LOGGER = logging.getLogger(__name__)
+
+
+SentenceRecognitionSourceTypePost = 1
+SentenceRecognitionSourceTypeURL = 0
+Modules = {
+    "16k_zh": "中文",
+    "16k_zh-PY": "中英粤",
+    "16k_en": "英语",
+    "16k_yue": "粤语",
+    "16k_zh_dialect": "多方言",
+}
+ModuleSupportLanguage = {
+    "16k_zh": ["zh"],
+    "16k_zh-PY": ["zh", "en", "yue"],
+    "16k_en": ["en"],
+    "16k_yue": ["yue"],
+    "16k_zh_dialect": ["zh"],
+}
+
+
+class TencentCloudAsrAPi(object):
+    def __init__(self, secretId, secretKey):
+        try:
+            cred = credential.Credential(secretId, secretKey)
+            region = ""
+            self.asrClient = AsrClient(cred, region)
+        except TencentCloudSDKException as err:
+            _LOGGER.error("tencent cloud sdk err:%s", err.message)
+            raise err
+
+    def SentenceRecognition(self, engSerViceType, data: str) -> (tuple[bool, str | None]):
+        req = SentenceRecognitionRequest()
+        req.EngSerViceType = engSerViceType
+        req.SourceType = SentenceRecognitionSourceTypePost
+        req.VoiceFormat = AudioFormats.WAV
+        req.ProjectId = 0
+        req.SubServiceType = 2
+        req.UsrAudioKey = ""
+        req.Data = data
+        req.DataLen = len(data)
+        try:
+            res: SentenceRecognitionResponse = self.asrClient.SentenceRecognition(req)
+            return True, res.Result
+        except TencentCloudSDKException as err:
+            _LOGGER.error("recognition failed:%s", err.message)
+            return False, str(err)