fix: fix test

This commit is contained in:
Krrish Dholakia 2025-03-10 21:46:05 -07:00
parent bfbe26b91d
commit 01aa536716
4 changed files with 22 additions and 10 deletions

View file

@ -3716,7 +3716,8 @@ async def chat_completion( # noqa: PLR0915
message=getattr(e, "message", error_msg),
type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"),
code=getattr(e, "code", getattr(e, "status_code", 500)),
openai_code=getattr(e, "code", None),
code=getattr(e, "status_code", 500),
headers=headers,
)
@ -3929,7 +3930,8 @@ async def completion( # noqa: PLR0915
message=getattr(e, "message", error_msg),
type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"),
code=getattr(e, "code", getattr(e, "status_code", 500)),
openai_code=getattr(e, "code", None),
code=getattr(e, "status_code", 500),
)
@ -4138,7 +4140,8 @@ async def embeddings( # noqa: PLR0915
message=getattr(e, "message", error_msg),
type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"),
code=getattr(e, "code", getattr(e, "status_code", 500)),
openai_code=getattr(e, "code", None),
code=getattr(e, "status_code", 500),
)
@ -4257,7 +4260,8 @@ async def image_generation(
message=getattr(e, "message", error_msg),
type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"),
code=getattr(e, "code", getattr(e, "status_code", 500)),
openai_code=getattr(e, "code", None),
code=getattr(e, "status_code", 500),
)
@ -4518,7 +4522,8 @@ async def audio_transcriptions(
message=getattr(e, "message", error_msg),
type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"),
code=getattr(e, "code", getattr(e, "status_code", 500)),
openai_code=getattr(e, "code", None),
code=getattr(e, "status_code", 500),
)
@ -4667,7 +4672,8 @@ async def get_assistants(
message=getattr(e, "message", error_msg),
type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"),
code=getattr(e, "code", getattr(e, "status_code", 500)),
openai_code=getattr(e, "code", None),
code=getattr(e, "status_code", 500),
)