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

Loading…
Cancel
Save