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

(test) azure gpt-4 vision test

Browse source
This commit is contained in:
ishaan-jaff 2023-12-26 13:18:32 +05:30
parent 08ed509b76
commit ae074814c9
1 changed files with 34 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

34
litellm/tests/test_completion.py
View file

@ -219,6 +219,40 @@ def test_completion_gpt4_vision():
# test_completion_gpt4_vision()
def test_completion_azure_gpt4_vision():
# azure gpt-4 vision takes 5s to respond
try:
litellm.set_verbose = True
response = completion(
model="azure/gpt-4-vision",
messages=[
{
"role": "user",
"content": [
{"type": "text", "text": "Whats in this image?"},
{
"type": "image_url",
"image_url": {
"url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
},
},
],
}
],
base_url="https://gpt-4-vision-resource.openai.azure.com/",
api_key=os.getenv("AZURE_VISION_API_KEY"),
)
print(response)
except openai.RateLimitError:
print("got a rate liimt error")
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_azure_gpt4_vision()
@pytest.mark.skip(reason="this test is flaky")
def test_completion_perplexity_api():
try: