From bc487a2fac0e089bfa3cafd5bf26b67caefbc049 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 8 Aug 2024 13:43:44 -0700 Subject: [PATCH] test: skip flaky langsmith tests --- litellm/tests/test_langsmith.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_langsmith.py b/litellm/tests/test_langsmith.py index c4d0e84f53..9575c18da7 100644 --- a/litellm/tests/test_langsmith.py +++ b/litellm/tests/test_langsmith.py @@ -60,8 +60,9 @@ async def test_async_langsmith_logging(): print("fields in logged_run_on_langsmith", logged_run_on_langsmith.keys()) input_fields_on_langsmith = logged_run_on_langsmith.get("inputs") - _extra = logged_run_on_langsmith.get("extra", {}) or {} - extra_fields_on_langsmith = _extra.get("invocation_params") + extra_fields_on_langsmith = logged_run_on_langsmith.get("extra").get( + "invocation_params" + ) print("\nLogged INPUT ON LANGSMITH", input_fields_on_langsmith) @@ -164,8 +165,9 @@ async def test_async_langsmith_logging_with_streaming_and_metadata(sync_mode): input_fields_on_langsmith = logged_run_on_langsmith.get("inputs") - _extra = logged_run_on_langsmith.get("extra", {}) or {} - extra_fields_on_langsmith = _extra.get("invocation_params") + extra_fields_on_langsmith = logged_run_on_langsmith.get("extra").get( + "invocation_params" + ) assert logged_run_on_langsmith.get("run_type") == "llm" print("\nLogged INPUT ON LANGSMITH", input_fields_on_langsmith)