%pip install git+https://github.com/sunishsheth2009/databricks-sdk-py@sunish-add-headers
import requests
import uuid
from databricks.sdk import WorkspaceClient
from databricks.sdk.service.serving import ExternalFunctionRequestHttpMethod

# Step 1: Initialize session
init_payload = {
    "jsonrpc": "2.0",
    "id": "init-1",
    "method": "initialize",
    "params": {}
}

init_response = WorkspaceClient().serving_endpoints.http_request(
  conn="github-u2m-managed-connection",
  method=ExternalFunctionRequestHttpMethod.POST,
  path="/mcp",
  json=init_payload,
)

session_id = init_response.headers["mcp-session-id"]
if not session_id:
    print("No session ID returned by server.")

print(f"✅ Got MCP Session ID: {session_id}")


Output:
✅ Got MCP Session ID: 53220122-015c-4e2b-9edb-47580cefffd0