ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 19:05:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add validation for empty streaming responses in LiteLLM handler

Browse source
This commit is contained in:
Makonike 2025-07-09 15:04:18 +08:00
parent 5e82d0a316
commit 2d8bee0d6d
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View file

@ -436,4 +436,8 @@ class LiteLLMAIHandler(BaseAiHandler):
get_logger().error(f"Error handling streaming response: {e}") get_logger().error(f"Error handling streaming response: {e}")
raise 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 return full_response, finish_reason