mistralai[patch]: Use public attribute for eventsource.response (#20580)

Minor change, use the public attribute instead of the protected one.
pull/20609/head^2
Eugene Yurtsev 3 months ago committed by GitHub
parent 66fb0b1f35
commit 8c29b7bf35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -148,9 +148,7 @@ async def _aiter_sse(
) -> AsyncIterator[Dict]:
"""Iterate over the server-sent events."""
async with event_source_mgr as event_source:
# TODO(Team): Remove after this is fixed in httpx dependency
# https://github.com/florimondmanca/httpx-sse/pull/25/files
await _araise_on_error(event_source._response)
await _araise_on_error(event_source.response)
async for event in event_source.aiter_sse():
if event.data == "[DONE]":
return
@ -370,9 +368,7 @@ class ChatMistralAI(BaseChatModel):
with connect_sse(
self.client, "POST", "/chat/completions", json=kwargs
) as event_source:
# TODO(Team): Remove after this is fixed in httpx dependency
# https://github.com/florimondmanca/httpx-sse/pull/25/files
_raise_on_error(event_source._response)
_raise_on_error(event_source.response)
for event in event_source.iter_sse():
if event.data == "[DONE]":
return

Loading…
Cancel
Save