Share via


Frontend Tool Rendering with AG-UI

This tutorial shows you how to add frontend function tools to your AG-UI clients. Frontend tools are functions that execute on the client side, allowing the AI agent to interact with the user's local environment, access client-specific data, or perform UI operations. The server orchestrates when to call these tools, but the execution happens entirely on the client.

Prerequisites

Before you begin, ensure you have completed the Getting Started tutorial and have:

  • .NET 8.0 or later
  • Microsoft.Agents.AI.AGUI package installed
  • Microsoft.Agents.AI package installed
  • Basic understanding of AG-UI client setup

What are Frontend Tools?

Frontend tools are function tools that:

  • Are defined and registered on the client
  • Execute in the client's environment (not on the server)
  • Allow the AI agent to interact with client-specific resources
  • Provide results back to the server for the agent to incorporate into responses
  • Enable personalized, context-aware experiences

Common use cases:

  • Reading local sensor data (GPS, temperature, etc.)
  • Accessing client-side storage or preferences
  • Performing UI operations (changing themes, displaying notifications)
  • Interacting with device-specific features (camera, microphone)

Registering Frontend Tools on the Client

The key difference from the Getting Started tutorial is registering tools with the client agent. Here's what changes:

// Define a frontend function tool
[Description("Get the user's current location from GPS.")]
static string GetUserLocation()
{
    // Access client-side GPS
    return "Amsterdam, Netherlands (52.37°N, 4.90°E)";
}

// Create frontend tools
AITool[] frontendTools = [AIFunctionFactory.Create(GetUserLocation)];

// Pass tools when creating the agent
AIAgent agent = chatClient.CreateAIAgent(
    name: "agui-client",
    description: "AG-UI Client Agent",
    tools: frontendTools);

The rest of your client code remains the same as shown in the Getting Started tutorial.

How Tools Are Sent to the Server

When you register tools with CreateAIAgent(), the AGUIChatClient automatically:

  1. Captures the tool definitions (names, descriptions, parameter schemas)
  2. Sends the tools with each request to the server agent which maps them to ChatAgentRunOptions.ChatOptions.Tools

The server receives the client tool declarations and the AI model can decide when to call them.

Inspecting and Modifying Tools with Middleware

You can use agent middleware to inspect or modify the agent run, including accessing the tools:

// Create agent with middleware that inspects tools
AIAgent inspectableAgent = baseAgent
    .AsBuilder()
    .Use(runFunc: null, runStreamingFunc: InspectToolsMiddleware)
    .Build();

static async IAsyncEnumerable<AgentRunResponseUpdate> InspectToolsMiddleware(
    IEnumerable<ChatMessage> messages,
    AgentThread? thread,
    AgentRunOptions? options,
    AIAgent innerAgent,
    CancellationToken cancellationToken)
{
    // Access the tools from ChatClientAgentRunOptions
    if (options is ChatClientAgentRunOptions chatOptions)
    {
        IList<AITool>? tools = chatOptions.ChatOptions?.Tools;
        if (tools != null)
        {
            Console.WriteLine($"Tools available for this run: {tools.Count}");
            foreach (AITool tool in tools)
            {
                if (tool is AIFunction function)
                {
                    Console.WriteLine($"  - {function.Metadata.Name}: {function.Metadata.Description}");
                }
            }
        }
    }

    await foreach (AgentRunResponseUpdate update in innerAgent.RunStreamingAsync(messages, thread, options, cancellationToken))
    {
        yield return update;
    }
}

This middleware pattern allows you to:

  • Validate tool definitions before execution

Key Concepts

The following are new concepts for frontend tools:

  • Client-side registration: Tools are registered on the client using AIFunctionFactory.Create() and passed to CreateAIAgent()
  • Automatic capture: Tools are automatically captured and sent via ChatAgentRunOptions.ChatOptions.Tools

How Frontend Tools Work

Server-Side Flow

The server doesn't know the implementation details of frontend tools. It only knows:

  1. Tool names and descriptions (from client registration)
  2. Parameter schemas
  3. When to request tool execution

When the AI agent decides to call a frontend tool:

  1. Server sends a tool call request to the client via SSE
  2. Server waits for the client to execute the tool and return results
  3. Server incorporates the results into the agent's context
  4. Agent continues processing with the tool results

Client-Side Flow

The client handles frontend tool execution:

  1. Receives FunctionCallContent from server indicating a tool call request
  2. Matches the tool name to a locally registered function
  3. Deserializes parameters from the request
  4. Executes the function locally
  5. Serializes the result
  6. Sends FunctionResultContent back to the server
  7. Continues receiving agent responses

Expected Output with Frontend Tools

When the agent calls frontend tools, you'll see the tool call and result in the streaming output:

User (:q or quit to exit): Where am I located?

[Client Tool Call - Name: GetUserLocation]
[Client Tool Result: Amsterdam, Netherlands (52.37°N, 4.90°E)]

You are currently in Amsterdam, Netherlands, at coordinates 52.37°N, 4.90°E.

Server Setup for Frontend Tools

The server doesn't need special configuration to support frontend tools. Use the standard AG-UI server from the Getting Started tutorial - it automatically:

  • Receives frontend tool declarations during client connection
  • Requests tool execution when the AI agent needs them
  • Waits for results from the client
  • Incorporates results into the agent's decision-making

Next Steps

Now that you understand frontend tools, you can:

Additional Resources

This tutorial shows you how to add frontend function tools to your AG-UI clients. Frontend tools are functions that execute on the client side, allowing the AI agent to interact with the user's local environment, access client-specific data, or perform UI operations.

Prerequisites

Before you begin, ensure you have completed the Getting Started tutorial and have:

  • Python 3.10 or later
  • httpx installed for HTTP client functionality
  • Basic understanding of AG-UI client setup
  • Azure OpenAI service configured

What are Frontend Tools?

Frontend tools are function tools that:

  • Are defined and registered on the client
  • Execute in the client's environment (not on the server)
  • Allow the AI agent to interact with client-specific resources
  • Provide results back to the server for the agent to incorporate into responses

Common use cases:

  • Reading local sensor data
  • Accessing client-side storage or preferences
  • Performing UI operations
  • Interacting with device-specific features

Creating Frontend Tools

Frontend tools in Python are defined similarly to backend tools but are registered with the client:

from typing import Annotated
from pydantic import BaseModel, Field


class SensorReading(BaseModel):
    """Sensor reading from client device."""
    temperature: float
    humidity: float
    air_quality_index: int


def read_climate_sensors(
    include_temperature: Annotated[bool, Field(description="Include temperature reading")] = True,
    include_humidity: Annotated[bool, Field(description="Include humidity reading")] = True,
) -> SensorReading:
    """Read climate sensor data from the client device."""
    # Simulate reading from local sensors
    return SensorReading(
        temperature=22.5 if include_temperature else 0.0,
        humidity=45.0 if include_humidity else 0.0,
        air_quality_index=75,
    )


def change_background_color(color: Annotated[str, Field(description="Color name")] = "blue") -> str:
    """Change the console background color."""
    # Simulate UI change
    print(f"\n🎨 Background color changed to {color}")
    return f"Background changed to {color}"

Creating an AG-UI Client with Frontend Tools

Here's a complete client implementation with frontend tools:

"""AG-UI client with frontend tools."""

import asyncio
import json
import os
from typing import Annotated, AsyncIterator

import httpx
from pydantic import BaseModel, Field


class SensorReading(BaseModel):
    """Sensor reading from client device."""
    temperature: float
    humidity: float
    air_quality_index: int


# Define frontend tools
def read_climate_sensors(
    include_temperature: Annotated[bool, Field(description="Include temperature")] = True,
    include_humidity: Annotated[bool, Field(description="Include humidity")] = True,
) -> SensorReading:
    """Read climate sensor data from the client device."""
    return SensorReading(
        temperature=22.5 if include_temperature else 0.0,
        humidity=45.0 if include_humidity else 0.0,
        air_quality_index=75,
    )


def get_user_location() -> dict:
    """Get the user's current GPS location."""
    # Simulate GPS reading
    return {
        "latitude": 52.3676,
        "longitude": 4.9041,
        "accuracy": 10.0,
        "city": "Amsterdam",
    }


# Tool registry maps tool names to functions
FRONTEND_TOOLS = {
    "read_climate_sensors": read_climate_sensors,
    "get_user_location": get_user_location,
}


class AGUIClientWithTools:
    """AG-UI client with frontend tool support."""

    def __init__(self, server_url: str, tools: dict):
        self.server_url = server_url
        self.tools = tools
        self.thread_id: str | None = None

    async def send_message(self, message: str) -> AsyncIterator[dict]:
        """Send a message and handle streaming response with tool execution."""
        # Prepare tool declarations for the server
        tool_declarations = []
        for name, func in self.tools.items():
            tool_declarations.append({
                "name": name,
                "description": func.__doc__ or "",
                # Add parameter schema from function signature
            })

        request_data = {
            "messages": [
                {"role": "system", "content": "You are a helpful assistant with access to client tools."},
                {"role": "user", "content": message},
            ],
            "tools": tool_declarations,  # Send tool declarations to server
        }

        if self.thread_id:
            request_data["thread_id"] = self.thread_id

        async with httpx.AsyncClient(timeout=60.0) as client:
            async with client.stream(
                "POST",
                self.server_url,
                json=request_data,
                headers={"Accept": "text/event-stream"},
            ) as response:
                response.raise_for_status()

                async for line in response.aiter_lines():
                    if line.startswith("data: "):
                        data = line[6:]
                        try:
                            event = json.loads(data)

                            # Handle tool call requests from server
                            if event.get("type") == "TOOL_CALL_REQUEST":
                                await self._handle_tool_call(event, client)
                            else:
                                yield event

                            # Capture thread_id
                            if event.get("type") == "RUN_STARTED" and not self.thread_id:
                                self.thread_id = event.get("threadId")

                        except json.JSONDecodeError:
                            continue

    async def _handle_tool_call(self, event: dict, client: httpx.AsyncClient):
        """Execute frontend tool and send result back to server."""
        tool_name = event.get("toolName")
        tool_call_id = event.get("toolCallId")
        arguments = event.get("arguments", {})

        print(f"\n\033[95m[Client Tool Call: {tool_name}]\033[0m")
        print(f"  Arguments: {arguments}")

        try:
            # Execute the tool
            tool_func = self.tools.get(tool_name)
            if not tool_func:
                raise ValueError(f"Unknown tool: {tool_name}")

            result = tool_func(**arguments)

            # Convert Pydantic models to dict
            if hasattr(result, "model_dump"):
                result = result.model_dump()

            print(f"\033[94m[Client Tool Result: {result}]\033[0m")

            # Send result back to server
            await client.post(
                f"{self.server_url}/tool_result",
                json={
                    "tool_call_id": tool_call_id,
                    "result": result,
                },
            )

        except Exception as e:
            print(f"\033[91m[Tool Error: {e}]\033[0m")
            # Send error back to server
            await client.post(
                f"{self.server_url}/tool_result",
                json={
                    "tool_call_id": tool_call_id,
                    "error": str(e),
                },
            )


async def main():
    """Main client loop with frontend tools."""
    server_url = os.environ.get("AGUI_SERVER_URL", "http://127.0.0.1:8888/")
    print(f"Connecting to AG-UI server at: {server_url}\n")

    client = AGUIClientWithTools(server_url, FRONTEND_TOOLS)

    try:
        while True:
            message = input("\nUser (:q or quit to exit): ")
            if not message.strip():
                continue

            if message.lower() in (":q", "quit"):
                break

            print()
            async for event in client.send_message(message):
                event_type = event.get("type", "")

                if event_type == "RUN_STARTED":
                    print(f"\033[93m[Run Started]\033[0m")

                elif event_type == "TEXT_MESSAGE_CONTENT":
                    print(f"\033[96m{event.get('delta', '')}\033[0m", end="", flush=True)

                elif event_type == "RUN_FINISHED":
                    print(f"\n\033[92m[Run Finished]\033[0m")

                elif event_type == "RUN_ERROR":
                    error_msg = event.get("message", "Unknown error")
                    print(f"\n\033[91m[Error: {error_msg}]\033[0m")

            print()

    except KeyboardInterrupt:
        print("\n\nExiting...")
    except Exception as e:
        print(f"\n\033[91mError: {e}\033[0m")


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

How Frontend Tools Work

Protocol Flow

  1. Client Registration: Client sends tool declarations (names, descriptions, parameters) to server
  2. Server Orchestration: AI agent decides when to call frontend tools based on user request
  3. Tool Call Request: Server sends TOOL_CALL_REQUEST event to client via SSE
  4. Client Execution: Client executes the tool locally
  5. Result Submission: Client sends result back to server via POST request
  6. Agent Processing: Server incorporates result and continues response

Key Events

  • TOOL_CALL_REQUEST: Server requests frontend tool execution
  • TOOL_CALL_RESULT: Client submits execution result (via HTTP POST)

Expected Output

User (:q or quit to exit): What's the temperature reading from my sensors?

[Run Started]

[Client Tool Call: read_climate_sensors]
  Arguments: {'include_temperature': True, 'include_humidity': True}
[Client Tool Result: {'temperature': 22.5, 'humidity': 45.0, 'air_quality_index': 75}]

Based on your sensor readings, the current temperature is 22.5°C and the 
humidity is at 45%. These are comfortable conditions!
[Run Finished]

Server Setup

The standard AG-UI server from the Getting Started tutorial automatically supports frontend tools. No changes needed on the server side - it handles tool orchestration automatically.

Best Practices

Security

def access_sensitive_data() -> str:
    """Access user's sensitive data."""
    # Always check permissions first
    if not has_permission():
        return "Error: Permission denied"

    try:
        # Access data
        return "Data retrieved"
    except Exception as e:
        # Don't expose internal errors
        return "Unable to access data"

Error Handling

def read_file(path: str) -> str:
    """Read a local file."""
    try:
        with open(path, "r") as f:
            return f.read()
    except FileNotFoundError:
        return f"Error: File not found: {path}"
    except PermissionError:
        return f"Error: Permission denied: {path}"
    except Exception as e:
        return f"Error reading file: {str(e)}"

Async Operations

async def capture_photo() -> str:
    """Capture a photo from device camera."""
    # Simulate camera access
    await asyncio.sleep(1)
    return "photo_12345.jpg"

Troubleshooting

Tools Not Being Called

  1. Ensure tool declarations are sent to server
  2. Verify tool descriptions clearly indicate purpose
  3. Check server logs for tool registration

Execution Errors

  1. Add comprehensive error handling
  2. Validate parameters before processing
  3. Return user-friendly error messages
  4. Log errors for debugging

Type Issues

  1. Use Pydantic models for complex types
  2. Convert models to dicts before serialization
  3. Handle type conversions explicitly

Next Steps

Additional Resources