Skip to content

Commit fc62c72

Browse files
test: handle service unavailable errors
1 parent 6a9cf29 commit fc62c72

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

tests/llm_translation/test_fireworks_ai_translation.py

+13-10
Original file line numberDiff line numberDiff line change
@@ -186,16 +186,19 @@ def test_document_inlining_example(disable_add_transform_inline_image_block):
186186
disable_add_transform_inline_image_block=disable_add_transform_inline_image_block,
187187
)
188188
else:
189-
completion = litellm.completion(
190-
model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct",
191-
messages=[
192-
{
193-
"role": "user",
194-
"content": "this is a test request, write a short poem",
195-
},
196-
],
197-
disable_add_transform_inline_image_block=disable_add_transform_inline_image_block,
198-
)
189+
try:
190+
completion = litellm.completion(
191+
model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct",
192+
messages=[
193+
{
194+
"role": "user",
195+
"content": "this is a test request, write a short poem",
196+
},
197+
],
198+
disable_add_transform_inline_image_block=disable_add_transform_inline_image_block,
199+
)
200+
except litellm.ServiceUnavailableError:
201+
pytest.skip("ServiceUnavailableError")
199202
print(completion)
200203

201204

0 commit comments

Comments
 (0)