test: update test

krrishdholakia · krrishdholakia · commit 9199078f360c · 2025-05-17T07:56:05.000-07:00
diff --git a/tests/litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py b/tests/litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
@@ -27,7 +27,7 @@ def test_gemini_realtime_transformation_session_created():
 
     session_created_message_str = json.dumps(session_created_message)
     logging_obj = MagicMock()
-    logging_obj.litellm_trace_id.return_value = "123"
+    logging_obj.litellm_trace_id = "123"
 
     transformed_message = config.transform_realtime_response(
         session_created_message_str,
