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

test(test_openai_endpoints.py): change key

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-06 11:19:34 -07:00
parent 6a06aba443
commit b014a72f7a
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
View file

@ -1621,6 +1621,7 @@ def test_completion_ollama_function_call_stream(model):
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="local test")
@pytest.mark.parametrize( @pytest.mark.parametrize(
("model"), ("model"),
[ [
@ -1669,6 +1670,7 @@ async def test_acompletion_ollama_function_call(model):
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="local test")
@pytest.mark.parametrize( @pytest.mark.parametrize(
("model"), ("model"),
[ [

5
tests/test_openai_endpoints.py
View file

@ -328,11 +328,10 @@ async def test_chat_completion_old_key():
""" """
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
try: try:
key = "sk-ecMXHujzUtKCvHcwacdaTw" key = "sk--W0Ph0uDZLVD7V7LQVrslg"
await chat_completion(session=session, key=key) await chat_completion(session=session, key=key)
except Exception as e: except Exception as e:
key = "sk-ecMXHujzUtKCvHcwacdaTw" # try diff db key (in case db url is for the other db) pytest.fail("Invalid api key")
await chat_completion(session=session, key=key)
@pytest.mark.asyncio @pytest.mark.asyncio