修复流式传输重试问题

Author: DoiiarX

InsightEngine/llms/base.py

@@ -83,7 +83,6 @@ def invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
             return self.validate_response(response.choices[0].message.content)
         return ""
 
-    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Generator[str, None, None]:
         """
         流式调用LLM，逐步返回响应内容
@@ -131,6 +130,7 @@ def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Gener
             logger.error(f"流式请求失败: {str(e)}")
             raise e
 
+    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke_to_string(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
         """
         流式调用LLM并安全地拼接为完整字符串（避免UTF-8多字节字符截断）

MediaEngine/llms/base.py

@@ -86,7 +86,6 @@ def invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
             return self.validate_response(response.choices[0].message.content)
         return ""
 
-    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Generator[str, None, None]:
         """
         流式调用LLM，逐步返回响应内容
@@ -134,6 +133,7 @@ def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Gener
             logger.error(f"流式请求失败: {str(e)}")
             raise e
 
+    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke_to_string(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
         """
         流式调用LLM并安全地拼接为完整字符串（避免UTF-8多字节字符截断）

QueryEngine/llms/base.py

@@ -83,7 +83,6 @@ def invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
             return self.validate_response(response.choices[0].message.content)
         return ""
 
-    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Generator[str, None, None]:
         """
         流式调用LLM，逐步返回响应内容
@@ -131,6 +130,7 @@ def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Gener
             logger.error(f"流式请求失败: {str(e)}")
             raise e
 
+    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke_to_string(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
         """
         流式调用LLM并安全地拼接为完整字符串（避免UTF-8多字节字符截断）

ReportEngine/llms/base.py

@@ -76,7 +76,6 @@ def invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
             return self.validate_response(response.choices[0].message.content)
         return ""
 
-    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Generator[str, None, None]:
         """
         流式调用LLM，逐步返回响应内容
@@ -118,6 +117,7 @@ def stream_invoke(self, system_prompt: str, user_prompt: str, **kwargs) -> Gener
             logger.error(f"流式请求失败: {str(e)}")
             raise e
 
+    @with_retry(LLM_RETRY_CONFIG)
     def stream_invoke_to_string(self, system_prompt: str, user_prompt: str, **kwargs) -> str:
         """
         流式调用LLM并安全地拼接为完整字符串（避免UTF-8多字节字符截断）