tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-12 23:05:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21275 from BerriAI/fix/watsonx-gpt-oss-async-mock

Browse Source 
fix(test): use async side_effect for client.post mock in watsonx test
This commit is contained in:
jquinter
2026-02-15 19:44:33 -03:00
committed by GitHub
parent 03215c6495 be63bac1c1
commit b0f2b4bb2d
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/llms/watsonx/test_watsonx.py
+7 -6
View File
@@ -289,15 +289,19 @@ async def test_watsonx_gpt_oss_prompt_transformation(monkeypatch):
# template is always used regardless of test execution order.
hf_model = "openai/gpt-oss-120b"
litellm.known_tokenizer_config[hf_model] = mock_tokenizer_config
# Also create sync mock functions in case the fallback sync path is used
def mock_get_tokenizer_config(hf_model_name: str):
return mock_tokenizer_config
def mock_get_chat_template_file(hf_model_name: str):
return {"status": "failure"}
with patch.object(client, "post") as mock_post, patch.object(
# Async mock function for client.post to properly handle async method mocking
async def mock_post_func(*args, **kwargs):
return mock_completion_response
with patch.object(client, "post", side_effect=mock_post_func) as mock_post, patch.object(
litellm.module_level_client, "post", return_value=mock_token_response
), patch(
"litellm.litellm_core_utils.prompt_templates.huggingface_template_handler._aget_tokenizer_config",
@@ -312,9 +316,6 @@ async def test_watsonx_gpt_oss_prompt_transformation(monkeypatch):
"litellm.litellm_core_utils.prompt_templates.huggingface_template_handler._get_chat_template_file",
side_effect=mock_get_chat_template_file,
):
# Set the mock to return the completion response
mock_post.return_value = mock_completion_response
try:
# Call acompletion with messages
await litellm.acompletion(