diff --git a/tests/unit/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py b/tests/unit/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py index 682eab957be..ed6bc06183a 100644 --- a/tests/unit/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py +++ b/tests/unit/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py @@ -4,6 +4,7 @@ import unittest from typing import TYPE_CHECKING, Final, List, Literal, Optional, Tuple, get_args from unittest.mock import ANY, MagicMock, Mock, patch +import httpx import pytest import litellm