firasbouzazi commented on PR #74198: URL: https://github.com/apache/airflow/pull/74198#issuecomment-6103760493
Live test against a Databricks workspace with a PAT, through Unity Gateway MCP Services in front of public MCP servers: - Listing tools works. - A missing service fails with HTTP 403 → `DatabricksUnityMCPAccessDeniedError`; an invalid token with 401 → the same error. - The server's errors for bad arguments reach the model as `ModelRetry`. It also found an issue: Unity Gateway buffers the whole response until the upstream server closes its stream (11 s for GitMCP, vs 0.7 s direct). That's longer than pydantic-ai's 5 s MCP handshake timeout, so the connection fails with a bare `RuntimeError`. Proposal: add an optional `init_timeout` to the toolset (default 30 s) and report a handshake timeout as `DatabricksUnityMCPTransportError`. Happy to add it to this PR if you agree. common.ai's `MCPToolset` has the same 5 s default and could hit this behind the gateway too. Not tested: a successful tool call through the gateway (blocked by the timeout above), service principal OAuth and Azure AD. @zozo123 @moomindani -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
