Python: [Bug]: ResponsesHostServer drops oauth_consent_request and function_approval_request from toolbox MCP connectors #5535

import asyncio
import os

from agent_framework import Agent
from agent_framework.foundry import FoundryChatClient
from agent_framework_foundry_hosting import ResponsesHostServer
from azure.identity import DefaultAzureCredential
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()


def _patch_hosting_oauth_consent():
    """Patch _to_outputs to emit oauth_consent_request as an SSE event.

    PR #5070 (merged into agent-framework-foundry 1.2.0) fixed the streaming
    chunk parser so oauth_consent_request content is now surfaced from the
    Responses API. However, the ResponsesHostServer's _to_outputs function
    still drops it because there is no handler for this content type.
    This patch adds the missing branch.
    """
    import agent_framework_foundry_hosting._responses as hosting_mod

    _orig_to_outputs = hosting_mod._to_outputs

    async def _patched_to_outputs(stream, content):
        if content.type == "oauth_consent_request" and content.consent_link:
            label = (content.additional_properties or {}).get("server_label", "")
            text = (
                f"OAuth consent required for MCP server '{label}'. "
                f"Please visit:\n{content.consent_link}"
            )
            async for event in stream.aoutput_item_message(text):
                yield event
        else:
            async for event in _orig_to_outputs(stream, content):
                yield event

    hosting_mod._to_outputs = _patched_to_outputs


_patch_hosting_oauth_consent()


async def main():
    client = FoundryChatClient(
        project_endpoint=os.environ["FOUNDRY_PROJECT_ENDPOINT"],
        model=os.environ["AZURE_AI_MODEL_DEPLOYMENT_NAME"],
        credential=DefaultAzureCredential(),
    )

    # Load the named toolbox from the Foundry project. Omitting `version`
    # resolves the toolbox's current default version at runtime.
    toolbox = await client.get_toolbox(os.environ["TOOLBOX_NAME"])

    # Disable approval prompts for MCP tools — the hosting framework does
    # not yet support the function_approval_request content type.
    for tool in toolbox["tools"]:
        if tool.get("type") == "mcp":
            tool["require_approval"] = "never"

    agent = Agent(
        client=client,
        instructions="You are a friendly assistant. Keep your answers brief.",
        tools=toolbox,
        # History will be managed by the hosting infrastructure, thus there
        # is no need to store history by the service. Learn more at:
        # https://developers.openai.com/api/reference/resources/responses/methods/create
        default_options={"store": False},
    )

    server = ResponsesHostServer(agent)
    await server.run_async()


if __name__ == "__main__":
    asyncio.run(main())