mathdatech/awesome-fastapi-projects
1
0
Fork 0
mirror of https://github.com/Kludex/awesome-fastapi-projects.git synced 2025-05-16 14:16:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ReadTimeout for errors to retry for in SourceGraph client

Browse Source
This commit is contained in:
Vladyslav Fedoriuk 2023-08-06 00:52:54 +02:00
parent ffc62be92a
commit 1ec3c0d57f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/source_graph/client.py
View File

@ -90,7 +90,9 @@ class AsyncSourceGraphSSEClient:
self: Self, **kwargs: MutableMapping[str, Any] self: Self, **kwargs: MutableMapping[str, Any]
) -> AsyncGenerator[ServerSentEvent, None]: ) -> AsyncGenerator[ServerSentEvent, None]:
"""Iterate over the SourceGraph SSE API with retries.""" """Iterate over the SourceGraph SSE API with retries."""
async for attempt in stamina.retry_context(on=httpx.ReadError): async for attempt in stamina.retry_context(
on=(httpx.ReadError, httpx.ReadTimeout)
):
with attempt: with attempt:
await asyncio.sleep(self._reconnection_delay) await asyncio.sleep(self._reconnection_delay)
async for event in self._aiter_sse(**kwargs): async for event in self._aiter_sse(**kwargs):