Skip to content

fix: map SSE error types to correct HTTP status codes in streaming #1263

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Jah-yee wants to merge 1 commit into
base: main
Choose a base branch
from
Open

fix: map SSE error types to correct HTTP status codes in streaming #1263

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 46 additions & 2 deletions src/anthropic/_streaming.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,28 @@


_T = TypeVar("_T")
# Mapping from SSE error type to HTTP status code
# See: https://docs.anthropic.com/en/api/error-codes
_SSE_ERROR_TYPE_TO_STATUS = {
"overloaded_error": 529,
"rate_limit_error": 429,
"api_error": 500,
"internal_server_error": 500,
"authentication_error": 401,
"invalid_request_error": 400,
"not_found_error": 404,
}


def _get_status_code_from_sse_error(body: dict, default_status_code: int) -> int:
"""Extract the appropriate HTTP status code from an SSE error body."""
if isinstance(body, dict) and "error" in body:
error_type = body["error"].get("type")
if error_type in _SSE_ERROR_TYPE_TO_STATUS:
return _SSE_ERROR_TYPE_TO_STATUS[error_type]
return default_status_code




class _SyncStreamMeta(abc.ABCMeta):
Expand Down Expand Up @@ -111,10 +133,21 @@ def __stream__(self) -> Iterator[_T]:
except Exception:
err_msg = sse.data or f"Error code: {response.status_code}"

# Create a mock response with the correct status code based on SSE error type
status_code = _get_status_code_from_sse_error(
body if isinstance(body, dict) else {},
response.status_code
)
error_response = httpx.Response(
status_code=status_code,
content=response.content,
headers=response.headers,
request=response.request,
)
raise self._client._make_status_error(
err_msg,
body=body,
response=self.response,
response=error_response,
)
finally:
# Ensure the response is closed even if the consumer doesn't read all data
Expand Down Expand Up @@ -231,10 +264,21 @@ async def __stream__(self) -> AsyncIterator[_T]:
except Exception:
err_msg = sse.data or f"Error code: {response.status_code}"

# Create a mock response with the correct status code based on SSE error type
status_code = _get_status_code_from_sse_error(
body if isinstance(body, dict) else {},
response.status_code
)
error_response = httpx.Response(
status_code=status_code,
content=response.content,
headers=response.headers,
request=response.request,
)
raise self._client._make_status_error(
err_msg,
body=body,
response=self.response,
response=error_response,
)
finally:
# Ensure the response is closed even if the consumer doesn't read all data
Expand Down