From cfa0a8c72b00deb3bb49baee0e3962ea86e66b34 Mon Sep 17 00:00:00 2001 From: Dan Volz Date: Mon, 30 Mar 2026 21:21:41 -0700 Subject: [PATCH] Remove no-op patch of track_llm_api_timing in test The decorator is applied at class definition time, so patching the module-level name at call time has no effect. The test works without it. --- .../oci/chat/test_oci_chat_transformation.py | 23 +++++++++---------- 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/tests/test_litellm/llms/oci/chat/test_oci_chat_transformation.py b/tests/test_litellm/llms/oci/chat/test_oci_chat_transformation.py index 4248d65316f..dc542abfaf8 100644 --- a/tests/test_litellm/llms/oci/chat/test_oci_chat_transformation.py +++ b/tests/test_litellm/llms/oci/chat/test_oci_chat_transformation.py @@ -777,7 +777,7 @@ class TestSyncStreamSplitChunks: def test_sync_stream_wrapper_splits_batched_chunks(self): """Verify get_sync_custom_stream_wrapper splits multi-event HTTP chunks.""" - from unittest.mock import MagicMock, patch + from unittest.mock import MagicMock config = OCIChatConfig() @@ -792,17 +792,16 @@ class TestSyncStreamSplitChunks: mock_client = MagicMock() mock_client.post.return_value = mock_response - with patch("litellm.llms.oci.chat.transformation.track_llm_api_timing", lambda: lambda f: f): - result = config.get_sync_custom_stream_wrapper( - model=TEST_MODEL_NAME, - custom_llm_provider="oci", - logging_obj=MagicMock(), - api_base="https://example.com/chat", - headers={}, - data={"test": "data"}, - messages=[], - client=mock_client, - ) + result = config.get_sync_custom_stream_wrapper( + model=TEST_MODEL_NAME, + custom_llm_provider="oci", + logging_obj=MagicMock(), + api_base="https://example.com/chat", + headers={}, + data={"test": "data"}, + messages=[], + client=mock_client, + ) # Consume the stream wrapper's completion_stream to verify splitting chunks = list(result.completion_stream)