Method createHandler

pkg/tools/a2a/a2a.go:215–249

()

}

func (t *Toolset) createHandler() tools.ToolHandler {
	return func(ctx context.Context, toolCall tools.ToolCall) (*tools.ToolCallResult, error) {
		var args struct {
			Message string `json:"message"`
		}
		if err := json.Unmarshal([]byte(toolCall.Function.Arguments), &args); err != nil {
			return nil, fmt.Errorf("failed to parse arguments: %w", err)
		}

		t.mu.RLock()
		client := t.client
		t.mu.RUnlock()

		if client == nil {
			return nil, errors.New("A2A client not initialized")
		}

		params := &a2a.MessageSendParams{
			Message: a2a.NewMessage(a2a.MessageRoleUser, &a2a.TextPart{Text: args.Message}),
		}

		var response strings.Builder
		for event, err := range client.SendStreamingMessage(ctx, params) {
			if err != nil {
				return nil, fmt.Errorf("A2A call failed: %w", err)
			}
			response.WriteString(extractText(event))
		}

		result := cmp.Or(response.String(), "No response from agent")

		// TODO(dga): could this be a tool call error?
		return tools.ResultSuccess(result), nil
	}
}

func extractText(event a2a.Event) string {
	var parts a2a.ContentParts