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

refactor(test_stream_chunk_builder.py): fix import

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-28 13:33:50 -07:00
parent 5ad01e59f6
commit dd2c0abd33
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_stream_chunk_builder.py
View file

@ -16,7 +16,7 @@ import pytest
from openai import OpenAI
import litellm
import litellm.tests.stream_chunk_testdata
from tests.local_testing import stream_chunk_testdata
from litellm import completion, stream_chunk_builder
dotenv.load_dotenv()
@ -200,7 +200,7 @@ def test_stream_chunk_builder_litellm_usage_chunks():
def test_stream_chunk_builder_litellm_mixed_calls():
response = stream_chunk_builder(litellm.tests.stream_chunk_testdata.chunks)
response = stream_chunk_builder(stream_chunk_testdata.chunks)
assert (
response.choices[0].message.content
== "To answer your question about how many rows are in the 'users' table, I'll need to run a SQL query. Let me do that for you."