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 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
async def test_completion_codestral_fim_api(model):
try:
if model == "vertex_ai/codestral@2405":
from test_amazing_vertex_completion import (
load_vertex_ai_credentials,
)
from test_amazing_vertex_completion import (
load_vertex_ai_credentials,
)
load_vertex_ai_credentials()
load_vertex_ai_credentials()
litellm.set_verbose = True
import logging
@ -4157,17 +4156,16 @@ async def test_completion_codestral_fim_api(model):
@pytest.mark.parametrize(
"model",
["vertex_ai/codestral@2405"],
["vertex_ai/mistral-nemo@2407"],
)
@pytest.mark.asyncio
async def test_completion_codestral_fim_api_stream(model):
try:
if model == "vertex_ai/codestral@2405":
from test_amazing_vertex_completion import (
load_vertex_ai_credentials,
)
from test_amazing_vertex_completion import (
load_vertex_ai_credentials,
)
load_vertex_ai_credentials()
load_vertex_ai_credentials()
import logging
from litellm._logging import verbose_logger