phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +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

20
tests/local_testing/test_text_completion.py
View file

@ -4116,12 +4116,11 @@ 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, )
)
load_vertex_ai_credentials() load_vertex_ai_credentials()
litellm.set_verbose = True litellm.set_verbose = True
import logging import logging
@ -4157,17 +4156,16 @@ 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, )
)
load_vertex_ai_credentials() load_vertex_ai_credentials()
import logging import logging
from litellm._logging import verbose_logger from litellm._logging import verbose_logger