From 2d8bee0d6d3984fe2b234ec65068e377112c28bc Mon Sep 17 00:00:00 2001 From: Makonike Date: Wed, 9 Jul 2025 15:04:18 +0800 Subject: [PATCH] feat: add validation for empty streaming responses in LiteLLM handler --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index 59a00045..2b914c9f 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -436,4 +436,8 @@ 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") + return full_response, finish_reason