scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: tighten SSE done parser + rubrik /v1/messages match

Browse source 
Co-authored-by: Yassin Kortam <yassin@berri.ai>
This commit is contained in:
Cursor Agent 2026-05-21 00:31:35 +00:00
parent 7ce378ec49
commit 577d1adab5
No known key found for this signature in database
2 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/integrations/rubrik.py
View file

@ -34,7 +34,7 @@ if TYPE_CHECKING:
Logging as LiteLLMLoggingObj,
)
_ENDPOINT_ANTHROPIC_MESSAGES = "/messages"
_ENDPOINT_ANTHROPIC_MESSAGES = "/v1/messages"
_WEBHOOK_PATH_TOOL_BLOCKING = "/v1/after_completion/openai/v1"
_WEBHOOK_PATH_LOGGING_BATCH = "/v1/litellm/batch"

8
litellm/llms/chatgpt/responses/transformation.py
View file

@ -167,8 +167,6 @@ class ChatGPTResponsesAPIConfig(OpenAIResponsesAPIConfig):
parsed_chunk = self._parse_sse_json_chunk(chunk)
if parsed_chunk is None:
continue
if parsed_chunk == STREAM_SSE_DONE_STRING:
break
event_type = parsed_chunk.get("type")
if event_type == ResponsesAPIStreamEvents.OUTPUT_ITEM_DONE:
@ -209,7 +207,7 @@ class ChatGPTResponsesAPIConfig(OpenAIResponsesAPIConfig):
return completed_response, error_message
def _parse_sse_json_chunk(self, chunk: str) -> Optional[Any]:
def _parse_sse_json_chunk(self, chunk: str) -> Optional[Dict[str, Any]]:
# Strip outer whitespace before removing the SSE `data:` prefix.
# `_strip_sse_data_from_chunk` only matches the prefix at position 0,
# so chunks with leading whitespace (e.g. ` data: {...}`) would
@ -218,10 +216,8 @@ class ChatGPTResponsesAPIConfig(OpenAIResponsesAPIConfig):
if not stripped_chunk:
return None
stripped_chunk = stripped_chunk.strip()
if not stripped_chunk:
if not stripped_chunk or stripped_chunk == STREAM_SSE_DONE_STRING:
return None
if stripped_chunk == STREAM_SSE_DONE_STRING:
return STREAM_SSE_DONE_STRING
try:
parsed_chunk = json.loads(stripped_chunk)
except json.JSONDecodeError: