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

test fix

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-16 20:13:31 -07:00
parent cf801f9642
commit c38146e180
1 changed files with 9 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_text_completion.py
View file

@ -4116,7 +4116,6 @@ async def test_async_text_completion_chat_model_stream():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_codestral_fim_api(model): async def test_completion_codestral_fim_api(model):
try: try:
if model == "vertex_ai/codestral@2405":
from test_amazing_vertex_completion import ( from test_amazing_vertex_completion import (
load_vertex_ai_credentials, load_vertex_ai_credentials,
) )
@ -4157,12 +4156,11 @@ async def test_completion_codestral_fim_api(model):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model", "model",
["vertex_ai/codestral@2405"], ["vertex_ai/mistral-nemo@2407"],
) )
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_codestral_fim_api_stream(model): async def test_completion_codestral_fim_api_stream(model):
try: try:
if model == "vertex_ai/codestral@2405":
from test_amazing_vertex_completion import ( from test_amazing_vertex_completion import (
load_vertex_ai_credentials, load_vertex_ai_credentials,
) )