From dd2c0abd339176fef2e6087ba5dc6a7d9864dc92 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 28 Sep 2024 13:33:50 -0700 Subject: [PATCH] refactor(test_stream_chunk_builder.py): fix import --- tests/local_testing/test_stream_chunk_builder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_stream_chunk_builder.py b/tests/local_testing/test_stream_chunk_builder.py index 477f28f2e..f45175f39 100644 --- a/tests/local_testing/test_stream_chunk_builder.py +++ b/tests/local_testing/test_stream_chunk_builder.py @@ -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."