Function forward_to_nlweb

AskAgent/python/chatbot_interface.py:23–59

Forward a request to the NLWeb MCP endpoint

(function_name: str, arguments: dict[str, Any], server_url: str, endpoint: str)

DEFAULT_ENDPOINT = "/mcp"

async def forward_to_nlweb(function_name: str, arguments: dict[str, Any], server_url: str, endpoint: str) -> dict[str, Any]:
    """Forward a request to the NLWeb MCP endpoint"""
    nlweb_mcp_url = f"{server_url}{endpoint}"
    try:
        # Format the request in MCP format
        payload = {
            "function_call": {
                "name": function_name,
                "arguments": json.dumps(arguments)
            }
        }

        # Print some debug info to stderr (won't interfere with stdio protocol)
        print(f"Forwarding to {nlweb_mcp_url}: {function_name}", file=sys.stderr)

        async with httpx.AsyncClient() as client:
            response = await client.post(
                nlweb_mcp_url,
                json=payload,
                headers={"Content-Type": "application/json"},
                timeout=30
            )

            if response.status_code != 200:
                print(f"Error from server: {response.status_code} - {response.text}", file=sys.stderr)
                return {
                    "error": f"Server error: {response.status_code} - {response.text}"
                }

            result = response.json()
            return result

    except Exception as e:
        print(f"Request failed: {e!s}", file=sys.stderr)
        return {
            "error": f"Request failed: {e!s}"
        }

async def serve(server_url: str = DEFAULT_SERVER_URL, endpoint: str = DEFAULT_ENDPOINT) -> None:
    """