test: fix test

This commit is contained in:
Krrish Dholakia 2024-08-27 10:46:57 -07:00
parent cd7dd2a511
commit 18b67a455e
3 changed files with 21 additions and 11 deletions

View file

@ -637,7 +637,10 @@ def client(original_function):
if is_coroutine is True:
pass
else:
if isinstance(original_response, ModelResponse):
if (
isinstance(original_response, ModelResponse)
and len(original_response.choices) > 0
):
model_response: Optional[str] = original_response.choices[
0
].message.content # type: ignore