Merge pull request #24 from mrslimslim/mrslimslim/stt-prompt

✨ feat: CreateOpenaiAudioTranscriptionsOptions add <prompt> parmas to get better transcript result
lobehub · Jan 5, 2024 · 921fa55 · 921fa55
2 parents cbc7bd4 + 91ba99b
commit 921fa55
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/core/OpenAISTT/index.ts b/src/core/OpenAISTT/index.ts
@@ -13,6 +13,10 @@ export interface OpenAISTTPayload {
      * @title 语音识别的模型名称
      */
     model: string;
+    /**
+     * @title 语音识别的prmopt 以更好的获得whisper的解析效果
+     */
+    prompt?: string;
   };
   /**
    * @title 语音识别的文件

diff --git a/src/server/createOpenaiAudioTranscriptions.ts b/src/server/createOpenaiAudioTranscriptions.ts
@@ -21,6 +21,7 @@ export const createOpenaiAudioTranscriptions = async ({
     {
       file,
       model: options.model,
+      prompt: options.prompt || '',
     },
     { headers: { Accept: '*/*' } },
   );