mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
6fdf6111e6
* Rename GetNewSession to CreateSession * Address copilot feedback * Suppress warning * Suppress warning * Fix further warnings.
120 lines
3.8 KiB
C#
120 lines
3.8 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System.ComponentModel;
|
|
using Microsoft.Agents.AI;
|
|
using Microsoft.Agents.AI.AGUI;
|
|
using Microsoft.Extensions.AI;
|
|
|
|
string serverUrl = Environment.GetEnvironmentVariable("AGUI_SERVER_URL") ?? "http://localhost:8888";
|
|
|
|
Console.WriteLine($"Connecting to AG-UI server at: {serverUrl}\n");
|
|
|
|
// 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)];
|
|
|
|
// Create the AG-UI client agent with tools
|
|
using HttpClient httpClient = new()
|
|
{
|
|
Timeout = TimeSpan.FromSeconds(60)
|
|
};
|
|
|
|
AGUIChatClient chatClient = new(httpClient, serverUrl);
|
|
|
|
AIAgent agent = chatClient.AsAIAgent(
|
|
name: "agui-client",
|
|
description: "AG-UI Client Agent",
|
|
tools: frontendTools);
|
|
|
|
AgentSession session = await agent.CreateSessionAsync();
|
|
List<ChatMessage> messages =
|
|
[
|
|
new(ChatRole.System, "You are a helpful assistant.")
|
|
];
|
|
|
|
try
|
|
{
|
|
while (true)
|
|
{
|
|
// Get user input
|
|
Console.Write("\nUser (:q or quit to exit): ");
|
|
string? message = Console.ReadLine();
|
|
|
|
if (string.IsNullOrWhiteSpace(message))
|
|
{
|
|
Console.WriteLine("Request cannot be empty.");
|
|
continue;
|
|
}
|
|
|
|
if (message is ":q" or "quit")
|
|
{
|
|
break;
|
|
}
|
|
|
|
messages.Add(new ChatMessage(ChatRole.User, message));
|
|
|
|
// Stream the response
|
|
bool isFirstUpdate = true;
|
|
string? sessionId = null;
|
|
|
|
await foreach (AgentResponseUpdate update in agent.RunStreamingAsync(messages, session))
|
|
{
|
|
ChatResponseUpdate chatUpdate = update.AsChatResponseUpdate();
|
|
|
|
// First update indicates run started
|
|
if (isFirstUpdate)
|
|
{
|
|
sessionId = chatUpdate.ConversationId;
|
|
Console.ForegroundColor = ConsoleColor.Yellow;
|
|
Console.WriteLine($"\n[Run Started - Session: {chatUpdate.ConversationId}, Run: {chatUpdate.ResponseId}]");
|
|
Console.ResetColor();
|
|
isFirstUpdate = false;
|
|
}
|
|
|
|
// Display streaming content
|
|
foreach (AIContent content in update.Contents)
|
|
{
|
|
if (content is TextContent textContent)
|
|
{
|
|
Console.ForegroundColor = ConsoleColor.Cyan;
|
|
Console.Write(textContent.Text);
|
|
Console.ResetColor();
|
|
}
|
|
else if (content is FunctionCallContent functionCallContent)
|
|
{
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
Console.WriteLine($"\n[Client Tool Call - Name: {functionCallContent.Name}]");
|
|
Console.ResetColor();
|
|
}
|
|
else if (content is FunctionResultContent functionResultContent)
|
|
{
|
|
Console.ForegroundColor = ConsoleColor.Magenta;
|
|
Console.WriteLine($"[Client Tool Result: {functionResultContent.Result}]");
|
|
Console.ResetColor();
|
|
}
|
|
else if (content is ErrorContent errorContent)
|
|
{
|
|
Console.ForegroundColor = ConsoleColor.Red;
|
|
Console.WriteLine($"\n[Error: {errorContent.Message}]");
|
|
Console.ResetColor();
|
|
}
|
|
}
|
|
}
|
|
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
Console.WriteLine($"\n[Run Finished - Session: {sessionId}]");
|
|
Console.ResetColor();
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Console.WriteLine($"\nAn error occurred: {ex.Message}");
|
|
}
|