From 043a7560a5434e8a1d165ef219f53d04a6976cf3 Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Sat, 20 Sep 2025 23:46:37 -0400 Subject: [PATCH] test: use `.get()` for safe `ls_params` access (#33034) --- .../openai/tests/unit_tests/chat_models/test_azure.py | 4 ++-- libs/partners/xai/tests/unit_tests/test_chat_models.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py index a42cc69008e..05e60f85c7b 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py +++ b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py @@ -40,8 +40,8 @@ def test_initialize_more() -> None: assert llm.temperature == 0 ls_params = llm._get_ls_params() - assert ls_params["ls_provider"] == "azure" - assert ls_params["ls_model_name"] == "gpt-35-turbo-0125" + assert ls_params.get("ls_provider") == "azure" + assert ls_params.get("ls_model_name") == "gpt-35-turbo-0125" def test_initialize_azure_openai_with_openai_api_base_set() -> None: diff --git a/libs/partners/xai/tests/unit_tests/test_chat_models.py b/libs/partners/xai/tests/unit_tests/test_chat_models.py index b5af091004a..9763601917d 100644 --- a/libs/partners/xai/tests/unit_tests/test_chat_models.py +++ b/libs/partners/xai/tests/unit_tests/test_chat_models.py @@ -34,7 +34,7 @@ def test_xai_model_param() -> None: llm = ChatXAI(model_name="foo") # type: ignore[call-arg] assert llm.model_name == "foo" ls_params = llm._get_ls_params() - assert ls_params["ls_provider"] == "xai" + assert ls_params.get("ls_provider") == "xai" def test_chat_xai_invalid_streaming_params() -> None: