github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-23 11:30:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: use .get() for safe ls_params access (#33034)

Browse Source
This commit is contained in:
Mason Daugherty
2025-09-20 23:46:37 -04:00
committed by GitHub
parent 5b418d3f26
commit 043a7560a5
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
View File

@@ -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:

2
libs/partners/xai/tests/unit_tests/test_chat_models.py
View File

@@ -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: