phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix azure gpt-4o test

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-19 16:20:43 -07:00
parent b022247168
commit b54bbf510e
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion.py
View file

@ -935,6 +935,9 @@ def test_completion_base64(model):
print(f"\nResponse: {resp}") print(f"\nResponse: {resp}")
prompt_tokens = resp.usage.prompt_tokens prompt_tokens = resp.usage.prompt_tokens
except litellm.ServiceUnavailableError as e:
print("got service unavailable error: ", e)
pass
except Exception as e: except Exception as e:
if "500 Internal error encountered.'" in str(e): if "500 Internal error encountered.'" in str(e):
pass pass