03e3094dcc* Propagate CancellationToken request cancellation to remote endpoint * Refactor existing tests to have a shared base class * Add cancellation tests --------- Co-authored-by: Stephen Toub <stoub@microsoft.com>
38 lines
1.2 KiB
C#
38 lines
1.2 KiB
C#
using ModelContextProtocol;
|
|
using ModelContextProtocol.Protocol.Types;
|
|
using ModelContextProtocol.Server;
|
|
using System.ComponentModel;
|
|
|
|
namespace EverythingServer.Tools;
|
|
|
|
[McpServerToolType]
|
|
public class LongRunningTool
|
|
{
|
|
[McpServerTool(Name = "longRunningOperation"), Description("Demonstrates a long running operation with progress updates")]
|
|
public static async Task<string> LongRunningOperation(
|
|
IMcpServer server,
|
|
RequestContext<CallToolRequestParams> context,
|
|
int duration = 10,
|
|
int steps = 5)
|
|
{
|
|
var progressToken = context.Params?.Meta?.ProgressToken;
|
|
var stepDuration = duration / steps;
|
|
|
|
for (int i = 1; i <= steps + 1; i++)
|
|
{
|
|
await Task.Delay(stepDuration * 1000);
|
|
|
|
if (progressToken is not null)
|
|
{
|
|
await server.SendNotificationAsync("notifications/progress", new
|
|
{
|
|
Progress = i,
|
|
Total = steps,
|
|
progressToken
|
|
});
|
|
}
|
|
}
|
|
|
|
return $"Long running operation completed. Duration: {duration} seconds. Steps: {steps}.";
|
|
}
|
|
}
|