chore: add support to pass definition for stt

langgenius · Dec 11, 2024 · 15783b9 · 15783b9
1 parent ed1f849
commit 15783b9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/api/core/tools/provider/builtin/azurespeech/tools/asr.py b/api/core/tools/provider/builtin/azurespeech/tools/asr.py
@@ -34,4 +34,7 @@ def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) -> list[ToolInv
         if len(combinedPhrases) == 0:
             raise Exception("No text detected, error: {}".format(json.dumps(data)))
 
-        return [self.create_text_message(data.get("combinedPhrases", [])[0].get("text", ""))]
+        return [
+            self.create_text_message(data.get("combinedPhrases", [])[0].get("text", "")),
+            self.create_json_message(data),
+        ]
diff --git a/api/core/tools/provider/builtin/azurespeech/tools/asr.yaml b/api/core/tools/provider/builtin/azurespeech/tools/asr.yaml
@@ -20,3 +20,14 @@ parameters:
       zh_Hans: 要转换的音频文件。
     llm_description: The audio file to be converted.
     form: llm
+  - name: definition
+    type: string
+    required: false
+    label:
+      en_US: Definition
+      zh_Hans: asr 详细配置
+    human_description:
+      en_US: Definition
+      zh_Hans: asr 详细配置
+    llm_description: Definition
+    form: form