Cline で作った Devin 風ワークフローを Microsoft Agent Framework で再構築する③ - Interrupt / Resume 実装編

dotnet new console

dotnet add package Microsoft.Agents.AI.Workflows --prerelease

// ----------------------------------------
// Request / Response のデータ型
// ----------------------------------------

public sealed record KanbanRequest(string TaskId);

public sealed record KanbanResult(
    string TaskId,
    bool Success,
    string Message
);

// ----------------------------------------
// Workflow を作成
// ----------------------------------------

// KanbanRequest を送り、KanbanResult を受け取る RequestPort
var kanbanPort =
    RequestPort.Create<KanbanRequest, KanbanResult>("Kanban");

// WorkflowOutputEvent イベント発生用 Executor
var outputExecutor =
    new FunctionExecutor<KanbanResult, KanbanResult>(
        "KanbanResultOutput",
        (result, _, _) => result
    );

// 実際には Planner や Executor から
// この RequestPort に Request を送る Workflow を構築する想定
Workflow workflow = new WorkflowBuilder(kanbanPort)
    .AddEdge(kanbanPort, outputExecutor)
    .WithOutputFrom(outputExecutor)
    .Build();

// ----------------------------------------
// Workflow 開始
// ----------------------------------------

await using StreamingRun run =
    await InProcessExecution.RunStreamingAsync(
        workflow,
        new KanbanRequest("T0001")
    );

// ----------------------------------------
// Workflow のイベントを監視
// ----------------------------------------

await foreach (WorkflowEvent evt in run.WatchStreamAsync())
{
    switch (evt)
    {
        // Workflow が外部応答待ちになった
        case RequestInfoEvent requestEvent:
        {
            Console.WriteLine("Kanban の完了待ちに入りました。");

            // --------------------------------
            // KanbanRequest データを取得
            // --------------------------------
            if (!requestEvent.Request.TryGetDataAs<KanbanRequest>(
                    out KanbanRequest? kanbanRequest) ||
                kanbanRequest is null)
            {
                throw new InvalidOperationException(
                    "KanbanRequest データを取得できませんでした。"
                );
            }

            // --------------------------------
            // ここで Kanban を監視
            // --------------------------------
            KanbanResult result =
                await WaitForKanbanAsync(kanbanRequest.TaskId);

            Console.WriteLine("Kanban が完了しました。");

            // --------------------------------
            // Workflow に Response を返す
            // --------------------------------
            var response =
                requestEvent.Request.CreateResponse(result);

            await run.SendResponseAsync(response);

            break;
        }

        // Workflow 全体が完了
        case WorkflowOutputEvent outputEvent:
        {
            Console.WriteLine(
                $"Workflow 完了: {outputEvent.Data}"
            );

            return;
        }
    }
}

// ----------------------------------------
// Kanban 監視処理
// ----------------------------------------

static async Task<KanbanResult> WaitForKanbanAsync(string taskId)
{
    while (true)
    {
        Console.WriteLine($"{taskId} を監視中...");

        // 本番では
        // var status = ReadKanbanStatus(taskId);
        // のような処理になる。

        await Task.Delay(TimeSpan.FromSeconds(3));

        // サンプルなので Done になったことにする
        bool done = true;

        if (done)
        {
            return new KanbanResult(
                taskId,
                true,
                "Kanban task completed"
            );
        }
    }
}