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 8

Merge pull request #23377 from BerriAI/litellm_fix_flaky_mcp_and_agentcore_tests

Browse source 
[Fix] Flaky MCP Server and AgentCore Streaming Tests
This commit is contained in:
yuneng-jiang 2026-03-11 13:20:59 -07:00 • committed by GitHub
commit b5dfde892c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/mcp_tests/test_mcp_server.py
View file

@ -1456,6 +1456,7 @@ async def test_add_update_server_with_alias():
mock_mcp_server.authorization_url = None
mock_mcp_server.registration_url = None
mock_mcp_server.token_url = None
mock_mcp_server.oauth2_flow = None
# Additional fields used by build_mcp_server_from_table
mock_mcp_server.extra_headers = None
mock_mcp_server.allow_all_keys = False
@ -1511,6 +1512,7 @@ async def test_add_update_server_without_alias():
mock_mcp_server.authorization_url = None
mock_mcp_server.registration_url = None
mock_mcp_server.token_url = None
mock_mcp_server.oauth2_flow = None
# Additional fields used by build_mcp_server_from_table
mock_mcp_server.extra_headers = None
mock_mcp_server.allow_all_keys = False
@ -1566,6 +1568,7 @@ async def test_add_update_server_fallback_to_server_id():
mock_mcp_server.authorization_url = None
mock_mcp_server.registration_url = None
mock_mcp_server.token_url = None
mock_mcp_server.oauth2_flow = None
# Additional fields used by build_mcp_server_from_table - set explicitly
# to avoid MagicMock objects being passed to Pydantic MCPServer constructor
mock_mcp_server.extra_headers = None

4
tests/test_litellm/llms/bedrock/chat/agentcore/test_agentcore_transformation.py
View file

@ -219,6 +219,7 @@ class TestAgentCoreStreamingJsonFallback:
messages=[{"role": "user", "content": "test"}],
stream=True,
client=client,
api_key="test-jwt-token",
)
# Collect content across all chunks
@ -257,6 +258,7 @@ class TestAgentCoreStreamingJsonFallback:
messages=[{"role": "user", "content": "test"}],
stream=True,
client=client,
api_key="test-jwt-token",
)
# Collect content across all chunks
@ -289,6 +291,7 @@ class TestAgentCoreStreamingJsonFallback:
messages=[{"role": "user", "content": "test"}],
stream=True,
client=client,
api_key="test-jwt-token",
)
async def test_async_streaming_malformed_json_raises_error(self):
@ -316,4 +319,5 @@ class TestAgentCoreStreamingJsonFallback:
messages=[{"role": "user", "content": "test"}],
stream=True,
client=client,
api_key="test-jwt-token",
)