Method primeWorkspaceMCPCache

coderd/x/chatd/chatd.go:628–666

primeWorkspaceMCPCache populates workspaceMCPToolsCache after the create_workspace or start_workspace tool finishes waiting for the workspace agent to become reachable. By the time it runs the agent is already Ready, so a single ListMCPTools call usually succeeds. When the agent's MCP server is stil

(
	ctx context.Context,
	logger slog.Logger,
	chatID uuid.UUID,
	workspaceCtx *turnWorkspaceContext,
)

// the budget. The next user turn re-runs top-of-turn discovery from
// scratch.
func (p *Server) primeWorkspaceMCPCache(
	ctx context.Context,
	logger slog.Logger,
	chatID uuid.UUID,
	workspaceCtx *turnWorkspaceContext,
) {
	deadline := p.clock.Now().Add(workspaceMCPPrimeMaxWait)
	attempt := 0
	for {
		attempt++
		tools := p.discoverWorkspaceMCPTools(ctx, logger, chatID, workspaceCtx)
		if len(tools) > 0 {
			logger.Debug(ctx, "primed workspace MCP cache",
				slog.F("chat_id", chatID),
				slog.F("tool_count", len(tools)),
				slog.F("attempts", attempt),
			)
			return
		}
		if ctx.Err() != nil {
			return
		}
		if !p.clock.Now().Before(deadline) {
			logger.Debug(ctx,
				"workspace MCP cache primer gave up waiting for tools",
				slog.F("chat_id", chatID),
				slog.F("attempts", attempt),
			)
			return
		}
		timer := p.clock.NewTimer(workspaceMCPPrimeRetryInterval, "chatd", "workspace-mcp-prime")
		select {
		case <-timer.C:
		case <-ctx.Done():
			timer.Stop()
			return
		}
	}
}

type turnWorkspaceContext struct {
	server           *Server