diff --git a/schema.prisma b/schema.prisma index f7c07112417..7d90e1fd0ca 100644 --- a/schema.prisma +++ b/schema.prisma @@ -70,6 +70,8 @@ model LiteLLM_AgentsTable { created_by String updated_at DateTime @default(now()) @updatedAt @map("updated_at") updated_by String + static_headers Json? @default("{}") + extra_headers String[] @default([]) } model LiteLLM_OrganizationTable { diff --git a/tests/mcp_tests/test_mcp_server.py b/tests/mcp_tests/test_mcp_server.py index d3ec7863504..c71e55b086a 100644 --- a/tests/mcp_tests/test_mcp_server.py +++ b/tests/mcp_tests/test_mcp_server.py @@ -1453,6 +1453,9 @@ 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.tool_name_to_display_name = None + mock_mcp_server.tool_name_to_description = None + mock_mcp_server.byok_api_key_help_url = None # Add server to manager await test_manager.add_server(mock_mcp_server) @@ -1535,6 +1538,9 @@ 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.tool_name_to_display_name = None + mock_mcp_server.tool_name_to_description = None + mock_mcp_server.byok_api_key_help_url = None # Add server to manager await test_manager.add_server(mock_mcp_server) diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 799d9167065..002e31c7ad2 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -754,6 +754,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): "/v1/audio/transcriptions", "/v1/audio/speech", "/v1/ocr", + "/vertex_ai/live", ], }, },