File tree 1 file changed +13
-10
lines changed
1 file changed +13
-10
lines changed Original file line number Diff line number Diff line change @@ -186,16 +186,19 @@ def test_document_inlining_example(disable_add_transform_inline_image_block):
186
186
disable_add_transform_inline_image_block = disable_add_transform_inline_image_block ,
187
187
)
188
188
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" )
199
202
print (completion )
200
203
201
204
You can’t perform that action at this time.
0 commit comments