Skip to content

Commit a4f8ec1

Browse files
committed
fix prompt too long
1 parent 3b7855a commit a4f8ec1

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

src/utils/query.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -574,10 +574,9 @@ def handle_known_apistatus_errors(
574574
Returns:
575575
AbstractErrorResponse: The error response model.
576576
"""
577-
if error.status_code == 400:
578-
error_message = getattr(error, "message", str(error))
579-
if is_context_length_error(error_message):
580-
return PromptTooLongResponse(model=model_id)
581-
elif error.status_code == 429:
577+
error_message = getattr(error, "message", str(error))
578+
if is_context_length_error(error_message):
579+
return PromptTooLongResponse(model=model_id)
580+
if error.status_code == 429:
582581
return QuotaExceededResponse.model(model_id)
583582
return InternalServerErrorResponse.generic()

0 commit comments

Comments
 (0)