scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

test: Let MCP tool-execution mock accept new kwargs for streaming tests

Browse source
This commit is contained in:
Yuta Saito 2026-01-19 07:00:14 +09:00
parent 1ff5b3ed28
commit 1a7dc22002
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/mcp_tests/test_aresponses_api_with_mcp.py
View file

@ -665,7 +665,7 @@ async def test_streaming_mcp_events_validation():
"model",
[
pytest.param("gpt-4o-mini", id="openai"),
pytest.param("anthropic/claude-4-5-haiku", id="anthropic"),
pytest.param("claude-haiku-4-5", id="anthropic"),
],
)
async def test_streaming_responses_api_with_mcp_tools(
@ -717,7 +717,9 @@ async def test_streaming_responses_api_with_mcp_tools(
mock_get_tools.return_value = (mock_mcp_tools, ["litellm_proxy"])
# Create a dynamic mock that will match the actual tool call ID from the LLM response
def mock_execute_tool_calls_side_effect(tool_calls, user_api_key_auth):
def mock_execute_tool_calls_side_effect(
tool_calls, user_api_key_auth, **kwargs
):
"""Mock function that returns results matching the actual tool call IDs from the LLM"""
results = []
for tool_call in tool_calls: