Function call_tool_with_reconnect

src/mcpo/utils/main.py:285–322

(
        request: Request, arguments: Dict[str, Any]
    )

    client_header_forwarding_config=None,
):
    async def call_tool_with_reconnect(
        request: Request, arguments: Dict[str, Any]
    ) -> CallToolResult:
        session_manager = getattr(request.app.state, "session_manager", None)

        async def _invoke(session):
            return await session.call_tool(endpoint_name, arguments=arguments)

        if session_manager:
            try:
                session, _ = await session_manager.ensure_initialized()
            except ClosedResourceError:
                logger.warning(
                    "Session closed while initializing '%s'; attempting reconnect",
                    endpoint_name,
                )
                session, _ = await session_manager.reconnect()
            request.app.state.session = session

            try:
                return await _invoke(session)
            except ClosedResourceError:
                logger.warning(
                    "Session closed during call to '%s'; attempting reconnect",
                    endpoint_name,
                )
                session, _ = await session_manager.reconnect()
                request.app.state.session = session
                return await _invoke(session)

        session = getattr(request.app.state, "session", None)
        if not session:
            raise HTTPException(
                status_code=500,
                detail={"message": "MCP session is not available"},
            )

        return await _invoke(session)

    if form_model_fields:
        FormModel = create_model(f"{endpoint_name}_form_model", **form_model_fields)