phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: cleanup testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-15 17:57:06 -08:00
parent a48445c11c
commit a3d280baa3
3 changed files with 79 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_completion.py
View file

@ -9,9 +9,8 @@ sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
import pytest
from openai import Timeout
import litellm
from litellm import embedding, completion, completion_cost
from litellm import embedding, completion, completion_cost, Timeout
from litellm import RateLimitError
litellm.num_retries = 3
litellm.cache = None
@ -419,7 +418,7 @@ def test_completion_openai():
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai()
# test_completion_openai()
def test_completion_text_openai():
try:
@ -442,11 +441,13 @@ def test_completion_openai_with_optional_params():
)
# Add any assertions here to check the response
print(response)
except Timeout as e:
except litellm.Timeout as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai_with_optional_params()
def test_completion_openai_litellm_key():
try:
litellm.api_key = os.environ['OPENAI_API_KEY']
@ -606,7 +607,7 @@ def test_completion_openai_azure_with_functions():
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai_azure_with_functions()
# test_completion_openai_azure_with_functions()
def test_completion_azure():