tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 02:22:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_dynamic_azure_params

Browse Source
This commit is contained in:
Ishaan Jaff
2025-03-18 17:26:23 -07:00
parent 55ea2370ba
commit c010cdef59
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_completion.py
+5 -5
View File
@@ -4364,14 +4364,14 @@ async def test_dynamic_azure_params(stream, sync_mode):
## recreate mock client
if sync_mode:
mock_client = MagicMock(return_value="Hello world!")
new_mock_client = MagicMock(return_value="Hello world!")
else:
mock_client = AsyncMock(return_value="Hello world!")
new_mock_client = AsyncMock(return_value="Hello world!")
## CHECK IF NEW CLIENT IS USED (PARAM CHANGE)
with patch.object(
client.chat.completions.with_raw_response, "create", new=mock_client
) as mock_client:
client.chat.completions.with_raw_response, "create", new=new_mock_client
) as new_mock_client:
try:
if sync_mode:
_ = completion(
@@ -4393,7 +4393,7 @@ async def test_dynamic_azure_params(stream, sync_mode):
pass
try:
mock_client.assert_not_called()
new_mock_client.assert_called()
except Exception as e:
raise e