From 74df3f8bd58d2c52c287f36b5717e5067868aabc Mon Sep 17 00:00:00 2001 From: Makonike Date: Thu, 10 Jul 2025 15:14:25 +0800 Subject: [PATCH] fix(ai_handler): improve empty streaming response validation logic --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index 32d1420c..596ff7f9 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -459,8 +459,10 @@ class LiteLLMAIHandler(BaseAiHandler): get_logger().error(f"Error handling streaming response: {e}") raise - if not full_response: - get_logger().warning("Streaming response resulted in empty content") - raise openai.APIError("Empty streaming response received") + if not full_response and finish_reason is None: + get_logger().warning("Streaming response resulted in empty content with no finish reason") + raise openai.APIError("Empty streaming response received without proper completion") + elif not full_response and finish_reason: + get_logger().debug(f"Streaming response resulted in empty content but completed with finish_reason: {finish_reason}") return full_response, finish_reason