feat: fix

stepfun_api.py

@@ -159,7 +159,16 @@ class StepFunAPIClient:
         Raises:
             Exception: 如果转录失败
         """
-        url = f"{API_BASE_URL}/audio/asr/sse"
+        if streaming:
+            return self._transcribe_audio_sse_streaming(audio_path, progress_callback)
+        else:
+            return self._transcribe_audio_sse_sync(audio_path, progress_callback)
+
+    def _transcribe_audio_sse_sync(self, audio_path: str, progress_callback=None) -> str:
+        """
+        同步ASR转录，返回最终文本
+        """
+        url = f"{API_BASE_URL}/audio/asr"
 
         # 读取音频文件并转换为base64
         try:
@@ -167,13 +176,9 @@ class StepFunAPIClient:
                 audio_data = base64.b64encode(audio_file.read()).decode('utf-8')
         except Exception as e:
             logger.error(f"❌ Failed to read audio file: {e}")
-            if streaming:
-                yield f"[读取文件失败: {e}]"
-                return
-            else:
-                raise Exception(f"Failed to read audio file: {e}")
+            raise Exception(f"Failed to read audio file: {e}")
 
-        # 构建请求payload - 修正格式根据文档
+        # 构建请求payload
         payload = {
             "audio": {
                 "data": audio_data,
@@ -229,66 +234,150 @@ class StepFunAPIClient:
                                 accumulated_text += delta
                                 logger.debug(f"📝 ASR delta: {delta}")
 
-                                # 处理增量更新
-                                if streaming:
-                                    yield accumulated_text
-                                elif progress_callback:
+                                # 处理增量更新回调
+                                if progress_callback:
                                     progress_callback(accumulated_text)
 
                             elif event_type == "transcript.text.done":
                                 final_text = event_data.get("text", accumulated_text)
                                 logger.info(f"✅ ASR transcription complete: {final_text}")
-
-                                if streaming:
-                                    yield final_text
-                                    return
-                                else:
-                                    break
+                                break
 
                             elif event_type == "error":
                                 error_msg = event_data.get("message", "Unknown ASR error")
                                 logger.error(f"❌ ASR error: {error_msg}")
-
-                                if streaming:
-                                    yield f"[ASR错误: {error_msg}]"
-                                    return
-                                else:
-                                    raise Exception(f"ASR API error: {error_msg}")
+                                raise Exception(f"ASR API error: {error_msg}")
 
                         except json.JSONDecodeError as e:
                             logger.warning(f"⚠️ Failed to parse SSE line: {line}, error: {e}")
                             continue
                         except Exception as e:
                             logger.error(f"❌ Error processing SSE event: {e}")
-                            if streaming:
-                                yield f"[处理错误: {str(e)}]"
-                                return
-                            else:
-                                raise
+                            raise
 
             # 如果没有获得final_text，使用accumulated_text
             if not final_text:
                 final_text = accumulated_text
 
             if not final_text:
-                error_msg = "No transcription result received"
-                if streaming:
-                    yield f"[转录无结果]"
-                    return
-                else:
-                    raise Exception(error_msg)
+                raise Exception("No transcription result received")
 
             logger.info(f"🎯 Final transcription: {final_text}")
+            return final_text
+
+        except Exception as e:
+            logger.error(f"❌ ASR transcription failed: {e}")
+            raise Exception(f"ASR transcription failed: {e}")
+
+    def _transcribe_audio_sse_streaming(self, audio_path: str, progress_callback=None):
+        """
+        流式ASR转录，返回生成器
+        """
+        url = f"{API_BASE_URL}/audio/asr"
+
+        # 读取音频文件并转换为base64
+        try:
+            with open(audio_path, 'rb') as audio_file:
+                audio_data = base64.b64encode(audio_file.read()).decode('utf-8')
+        except Exception as e:
+            logger.error(f"❌ Failed to read audio file: {e}")
+            yield f"[读取文件失败: {e}]"
+            return
+
+        # 构建请求payload
+        payload = {
+            "audio": {
+                "data": audio_data,
+                "input": {
+                    "transcription": {
+                        "language": "zh",
+                        "prompt": "请记录下你所听到的语音内容。",
+                        "model": "step-asr",
+                        "full_rerun_on_commit": True,
+                        "enable_itn": True
+                    },
+                    "format": {
+                        "type": "pcm",
+                        "codec": "pcm_s16le",
+                        "rate": 16000,
+                        "bits": 16,
+                        "channel": 1
+                    }
+                }
+            }
+        }
+
+        headers = {
+            **self.base_headers,
+            "Content-Type": "application/json",
+            "Accept": "text/event-stream"
+        }
+
+        logger.info("🎙️ Starting ASR transcription...")
+
+        try:
+            response = requests.post(url, headers=headers, data=json.dumps(payload), stream=True, timeout=120)
+            response.raise_for_status()
+
+            final_text = ""
+            accumulated_text = ""
 
-            if not streaming:
-                return final_text
+            for line in response.iter_lines(decode_unicode=True):
+                if line:
+                    line = line.strip()
+                    if line.startswith("data: "):
+                        try:
+                            # 解析SSE数据
+                            data_str = line[6:]  # 去掉 "data: " 前缀
+                            if data_str == "[DONE]":
+                                break
+
+                            event_data = json.loads(data_str)
+                            event_type = event_data.get("type")
+
+                            if event_type == "transcript.text.delta":
+                                delta = event_data.get("delta", "")
+                                accumulated_text += delta
+                                logger.debug(f"📝 ASR delta: {delta}")
+
+                                # 流式更新
+                                yield accumulated_text
+                                if progress_callback:
+                                    progress_callback(accumulated_text)
+
+                            elif event_type == "transcript.text.done":
+                                final_text = event_data.get("text", accumulated_text)
+                                logger.info(f"✅ ASR transcription complete: {final_text}")
+                                yield final_text
+                                return
+
+                            elif event_type == "error":
+                                error_msg = event_data.get("message", "Unknown ASR error")
+                                logger.error(f"❌ ASR error: {error_msg}")
+                                yield f"[ASR错误: {error_msg}]"
+                                return
+
+                        except json.JSONDecodeError as e:
+                            logger.warning(f"⚠️ Failed to parse SSE line: {line}, error: {e}")
+                            continue
+                        except Exception as e:
+                            logger.error(f"❌ Error processing SSE event: {e}")
+                            yield f"[处理错误: {str(e)}]"
+                            return
+
+            # 如果没有获得final_text，使用accumulated_text
+            if not final_text:
+                final_text = accumulated_text
+
+            if not final_text:
+                yield f"[转录无结果]"
+                return
+
+            logger.info(f"🎯 Final transcription: {final_text}")
 
         except Exception as e:
             logger.error(f"❌ ASR transcription failed: {e}")
-            if streaming:
-                yield f"[转录失败: {str(e)}]"
-            else:
-                raise Exception(f"ASR transcription failed: {e}")
+            yield f"[转录失败: {str(e)}]"
 
 
 # Global API client instance (singleton)