From 40e88eec4b65509811b2c689ed08855d8173a6f6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 22 Nov 2023 14:23:52 -0800 Subject: [PATCH] (test)timeout errors --- litellm/tests/test_embedding.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 40fea7dd7..77ece41fb 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -2,6 +2,7 @@ import sys, os import traceback import pytest from dotenv import load_dotenv +import openai load_dotenv() @@ -56,6 +57,25 @@ def test_openai_azure_embedding_simple(): test_openai_azure_embedding_simple() + +def test_openai_azure_embedding_timeouts(): + try: + response = embedding( + model="azure/azure-embedding-model", + input=["good morning from litellm"], + timeout=0.00001 + ) + print(response) + response_keys = dict(response).keys() + assert set(["usage", "model", "object", "data"]) == set(response_keys) #assert litellm response has expected keys from OpenAI embedding response + except openai.APITimeoutError: + print("Good job got timeout error!") + pass + except Exception as e: + pytest.fail(f"Expected timeout error, did not get the correct error. Instead got {e}") + +# test_openai_azure_embedding_timeouts() + def test_openai_azure_embedding(): try: api_key = os.environ['AZURE_API_KEY']