Agent-Native/runtime/mcp/ask.go

package mcp

import "context"

type askKey struct{}

// WithAsk 把当前回合的 AskUser 放进上下文。MCP 返回 input_required 时使用它。
func WithAsk(ctx context.Context, fn func(context.Context, string) (string, error)) context.Context {
	return context.WithValue(ctx, askKey{}, fn)
}

func askFrom(ctx context.Context) func(context.Context, string) (string, error) {
	if ctx == nil {
		return nil
	}
	fn, _ := ctx.Value(askKey{}).(func(context.Context, string) (string, error))
	return fn
}

func inputQuestion(res callToolResult) (string, bool) {
	for _, blk := range res.Content {
		if blk.Type == "input_required" {
			return blk.Text, true
		}
	}
	return "", false
}