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

test: cleanup codestral tests - backend api unavailable

Browse source
This commit is contained in:
Krrish Dholakia 2024-10-23 22:19:57 -07:00
parent bcce21a5b0
commit ca09f4afec
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/local_testing/test_completion.py
View file

@ -1088,6 +1088,7 @@ def test_completion_mistral_api():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="backend api unavailable")
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_codestral_chat_api(): async def test_completion_codestral_chat_api():
try: try:

6
tests/local_testing/test_text_completion.py
View file

@ -4109,9 +4109,7 @@ async def test_async_text_completion_chat_model_stream():
# asyncio.run(test_async_text_completion_chat_model_stream()) # asyncio.run(test_async_text_completion_chat_model_stream())
@pytest.mark.parametrize( @pytest.mark.parametrize("model", ["vertex_ai/codestral@2405"]) #
"model", ["vertex_ai/codestral@2405", "text-completion-codestral/codestral-2405"] #
)
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_codestral_fim_api(model): async def test_completion_codestral_fim_api(model):
try: try:
@ -4154,7 +4152,7 @@ async def test_completion_codestral_fim_api(model):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model", "model",
["vertex_ai/codestral@2405", "text-completion-codestral/codestral-2405"], ["vertex_ai/codestral@2405"],
) )
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_codestral_fim_api_stream(model): async def test_completion_codestral_fim_api_stream(model):