From e7d49eb3873efa484ad5b79e3910c38ee5ac5bad Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 10 Apr 2025 21:45:23 -0700 Subject: [PATCH] test: update test --- tests/llm_translation/test_optional_params.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llm_translation/test_optional_params.py b/tests/llm_translation/test_optional_params.py index 078a0beef0..1eaa8673d5 100644 --- a/tests/llm_translation/test_optional_params.py +++ b/tests/llm_translation/test_optional_params.py @@ -1410,7 +1410,7 @@ def test_litellm_proxy_thinking_param(): custom_llm_provider="litellm_proxy", thinking={"type": "enabled", "budget_tokens": 1024}, ) - assert optional_params["thinking"] == {"type": "enabled", "budget_tokens": 1024} + assert optional_params["extra_body"]["thinking"] == {"type": "enabled", "budget_tokens": 1024} def test_gemini_modalities_param(): optional_params = get_optional_params(