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

refactor: fix imports

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-28 13:28:01 -07:00
parent 3560f0ef2c
commit 5ad01e59f6
8 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_streaming.py
View file

@ -1145,7 +1145,9 @@ def test_completion_claude_stream_bad_key():
@pytest.mark.parametrize("provider", ["vertex_ai_beta"]) # ""
def test_vertex_ai_stream(provider):
from litellm.tests.test_amazing_vertex_completion import load_vertex_ai_credentials
from tests.local_testing.test_amazing_vertex_completion import (
load_vertex_ai_credentials,
)
load_vertex_ai_credentials()
litellm.set_verbose = True
@ -3949,7 +3951,7 @@ def test_unit_test_perplexity_citations_chunk():
@pytest.mark.flaky(retries=3, delay=1)
def test_streaming_tool_calls_valid_json_str(model):
if "vertex_ai" in model:
from litellm.tests.test_amazing_vertex_completion import (
from tests.local_testing.test_amazing_vertex_completion import (
load_vertex_ai_credentials,
)