test(ci): fix six CircleCI test regressions on main

tests/e2e/ui/fixtures/migratedPages.ts

@@ -99,7 +99,7 @@ export const MIGRATED_E2E_PAGES: Readonly<Record<string, MigratedPage>> = {
     group: "Settings",
     content: { role: "heading", name: "UI Theme Customization" },
   },
-  logs: { segment: "logs", linkName: "Logs", content: { role: "heading", name: "Request Logs" } },
+  logs: { segment: "logs", linkName: "Logs", content: { role: "tab", name: "Request Logs" } },
   "admin-panel": {
     segment: "admin-panel",
     linkName: "Admin Settings",

tests/integration/_support/bedrock_runtime_peer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 import re
 import threading

tests/integration/providers/test_bedrock_runtime_chat_completions_chaos.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import base64
 import binascii

tests/store_model_in_db_tests/test_mcp_servers.py

@@ -1,6 +1,6 @@
 import sys
 from datetime import datetime
-from typing import List, Optional
+from typing import Final, List, Optional
 import pytest
 from litellm._uuid import uuid
 import os
@@ -391,6 +391,7 @@ async def test_create_mcp_server_invalid_alias():
 @_SKIP_NO_MCP
 @pytest.mark.asyncio
 async def test_edit_mcp_server_redacts_credentials():
+    mock_get_server: Final = mock.AsyncMock()
     with (
         mock.patch(
             "litellm.proxy.management_endpoints.mcp_management_endpoints.MCP_AVAILABLE",
@@ -399,6 +400,10 @@ async def test_edit_mcp_server_redacts_credentials():
         mock.patch(
             "litellm.proxy.management_endpoints.mcp_management_endpoints.get_prisma_client_or_throw"
         ) as mock_get_prisma,
+        mock.patch(
+            "litellm.proxy.management_endpoints.mcp_management_endpoints.get_mcp_server",
+            new=mock_get_server,
+        ),
         mock.patch(
             "litellm.proxy.management_endpoints.mcp_management_endpoints.update_mcp_server",
             new_callable=mock.AsyncMock,
@@ -422,6 +427,18 @@ async def test_edit_mcp_server_redacts_credentials():
         mock_manager.reload_servers_from_database = mock.AsyncMock()
 
         server_id = str(uuid.uuid4())
+        stored_server: Final = LiteLLM_MCPServerTable(
+            server_id=server_id,
+            alias="Updated Server",
+            url="https://updated.example.com/mcp",
+            transport=MCPTransport.http,
+            created_at=datetime.now(),
+            updated_at=datetime.now(),
+            credentials={"auth_value": "secret"},
+            teams=[],
+        )
+        mock_get_server.return_value = stored_server
+
         updated_server = LiteLLM_MCPServerTable(
             server_id=server_id,
             alias="Updated Server",
@@ -458,6 +475,7 @@ async def test_edit_mcp_server_redacts_credentials():
         mock_update.assert_awaited_once()
         mock_manager.update_server.assert_called_once_with(updated_server)
         mock_manager.reload_servers_from_database.assert_awaited_once()
+        mock_get_server.assert_awaited_once_with(mock_prisma, server_id)
 
 
 def test_validate_mcp_server_name_direct():

tests/unit/proxy/test_custom_proxy.py

@@ -1,52 +1,45 @@
 import os
+from typing import Final
 
 import uvicorn
 from dotenv import load_dotenv
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
 
-load_dotenv()
 
-# Set the SERVER_ROOT_PATH environment variable to match the custom mount path
-os.environ["SERVER_ROOT_PATH"] = "/my-custom-path"
+def build_app() -> FastAPI:
+    load_dotenv()
+    os.environ["SERVER_ROOT_PATH"] = "/my-custom-path"
 
-from litellm.proxy.proxy_server import app as litellm_app
-from litellm.proxy.proxy_server import proxy_startup_event
+    from litellm.proxy.proxy_server import app as litellm_app
+    from litellm.proxy.proxy_server import proxy_startup_event
 
-# Create main FastAPI app
-app = FastAPI(title="Custom LiteLLM Server", lifespan=proxy_startup_event)
+    app: Final = FastAPI(title="Custom LiteLLM Server", lifespan=proxy_startup_event)
+    custom_path: Final = "/my-custom-path"
 
-# Add CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
 
-custom_path = "/my-custom-path"
+    app.mount(custom_path, litellm_app)
 
-# Mount LiteLLM app at /litellm
-app.mount(custom_path, litellm_app)
+    @app.get("/")
+    async def root() -> dict[str, str]:
+        return {
+            "message": "Welcome to the API Gateway",
+            "litellm_endpoint": custom_path,
+        }
 
+    @app.get("/health")
+    async def health_check() -> dict[str, str]:
+        return {"status": "healthy"}
 
-# Default route at /
-@app.get("/")
-async def root():
-    return {
-        "message": "Welcome to the API Gateway",
-        "litellm_endpoint": f"{custom_path}",
-    }
-
-
-# Health check endpoint
-@app.get("/health")
-async def health_check():
-    return {"status": "healthy"}
+    return app
 
 
 if __name__ == "__main__":
-    # Run the server on port 8000
-    uvicorn.run(app, host="0.0.0.0", port=4000, log_level="info")
+    uvicorn.run(build_app(), host="0.0.0.0", port=4000, log_level="info")