feat: fetch todos from cached AgentRun (#564)

* fetch todos implemented, using cached AgentRun to serve Todos list from

Author: kinjalh

internal/mcpserver/agent/tool/fetchtodos.go

@@ -0,0 +1,62 @@
+package tool
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+)
+
+func (in *GetAgentRunTodosTool) Install(server *server.MCPServer) {
+	server.AddTool(
+		mcp.NewTool(
+			in.name,
+			mcp.WithDescription(in.description),
+		),
+		in.handler,
+	)
+}
+
+func (in *GetAgentRunTodosTool) handler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	runId, err := request.RequireString("runId")
+	if err != nil {
+		return mcp.NewToolResultError(fmt.Sprintf("failed to get runId: %v", err)), nil
+	}
+
+	// if the agentRun is not in the cache, fetch it from the API and cache it
+	if cachedAgentRun == nil {
+		fragment, err := in.client.GetAgentRun(ctx, runId)
+		if err != nil {
+			return mcp.NewToolResultError(fmt.Sprintf("failed to get agent run: %v", err)), nil
+		}
+		cachedAgentRun = fragment
+	}
+
+	// we need to convert the todos in order to return the correct format for the MCP server
+	todos := make([]map[string]interface{}, 0)
+	if cachedAgentRun != nil && cachedAgentRun.Todos != nil {
+		for _, todo := range cachedAgentRun.Todos {
+			todoMap := map[string]interface{}{
+				"title":       todo.Title,
+				"done":        todo.Done,
+				"description": todo.Description,
+			}
+			todos = append(todos, todoMap)
+		}
+	}
+
+	return mcp.NewToolResultJSON(struct {
+		Success bool                     `json:"success"`
+		Message string                   `json:"message"`
+		RunId   string                   `json:"runId"`
+		Todos   []map[string]interface{} `json:"todos"`
+		Source  string                   `json:"source"`
+	}{
+		Success: true,
+		Message: fmt.Sprintf("Successfully fetched todos for agent run %s", runId),
+		RunId:   runId,
+		Todos:   todos,
+		Source:  "cache",
+	})
+}

internal/mcpserver/agent/tool/tool.go

@@ -56,3 +56,11 @@ type CreateBranch struct {
 	CommitMessage string
 	BranchName    string
 }
+
+// GetAgentRunTodos is an MCP tool that gets the todos for a given agent run
+type GetAgentRunTodosTool struct {
+	ConsoleTool
+}
+
+// cachedAgentRun holds the current agent run fragment shared across tools
+var cachedAgentRun *client.AgentRunFragment

internal/mcpserver/agent/tool/updateanalysis.go

@@ -44,6 +44,9 @@ func (in *UpdateAnalysis) handler(ctx context.Context, request mcp.CallToolReque
 	if err != nil {
 		return mcp.NewToolResultError(fmt.Sprintf("failed to update analysis: %v", err)), nil
 	}
+	cachedAgentRun = &client.AgentRunFragment{
+		Todos: agentRun.GetTodos(),
+	}
 
 	return mcp.NewToolResultJSON(struct {
 		Success bool   `json:"success"`

internal/mcpserver/agent/tool/updatetodos.go

@@ -34,6 +34,7 @@ func (in *UpdateTodos) handler(ctx context.Context, request mcp.CallToolRequest)
 	if err != nil {
 		return mcp.NewToolResultError(fmt.Sprintf("failed to update todos: %v", err)), nil
 	}
+	cachedAgentRun = nil
 
 	return mcp.NewToolResultJSON(struct {
 		Success bool                        `json:"success"`