diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py index 0164d2d217a..7186bc5da7e 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py @@ -995,6 +995,12 @@ class TestSigV4BuildFromTable: table_record.instructions = None table_record.source_url = None table_record.submitted_by = None + table_record.token_exchange_endpoint = None + table_record.audience = None + table_record.subject_token_type = None + table_record.token_exchange_profile = None + table_record.timeout = None + table_record.max_concurrent_requests = None manager = MCPServerManager() @@ -1058,6 +1064,12 @@ class TestSigV4BuildFromTable: table_record.instructions = None table_record.source_url = None table_record.submitted_by = "user-submitter" + table_record.token_exchange_endpoint = None + table_record.audience = None + table_record.subject_token_type = None + table_record.token_exchange_profile = None + table_record.timeout = None + table_record.max_concurrent_requests = None manager = MCPServerManager() diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py index e92647a2feb..024d7a4302e 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py @@ -2570,7 +2570,7 @@ class TestPreviewOpenAPITools: registered_summary_to_name: dict = {} - def fake_create_tool_function(path, method, operation, base_url): # noqa: ANN001 + def fake_create_tool_function(path, method, operation, base_url): def _f(): return None @@ -2583,7 +2583,7 @@ class TestPreviewOpenAPITools: ) class _StubRegistry: - def register_tool(self, name, description, input_schema, handler): # noqa: ANN001 + def register_tool(self, name, description, input_schema, handler): registered_summary_to_name[description] = name monkeypatch.setattr(