tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-05 08:23:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
tests/local_testing/test_text_completion.py
+9 -11
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