From fc62c72036126c72a5de1e9de003991bc85b96f4 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 24 Mar 2025 22:29:37 -0700 Subject: [PATCH] test: handle service unavailable errors --- .../test_fireworks_ai_translation.py | 23 +++++++++++-------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/tests/llm_translation/test_fireworks_ai_translation.py b/tests/llm_translation/test_fireworks_ai_translation.py index 9e78270c922..a51f460c855 100644 --- a/tests/llm_translation/test_fireworks_ai_translation.py +++ b/tests/llm_translation/test_fireworks_ai_translation.py @@ -186,16 +186,19 @@ def test_document_inlining_example(disable_add_transform_inline_image_block): disable_add_transform_inline_image_block=disable_add_transform_inline_image_block, ) else: - completion = litellm.completion( - model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct", - messages=[ - { - "role": "user", - "content": "this is a test request, write a short poem", - }, - ], - disable_add_transform_inline_image_block=disable_add_transform_inline_image_block, - ) + try: + completion = litellm.completion( + model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct", + messages=[ + { + "role": "user", + "content": "this is a test request, write a short poem", + }, + ], + disable_add_transform_inline_image_block=disable_add_transform_inline_image_block, + ) + except litellm.ServiceUnavailableError: + pytest.skip("ServiceUnavailableError") print(completion)