Function toolWithMaxBudget

pkg/workflow/unified_prompt_step.go:513–518

toolWithMaxBudget formats a tool name with a per-call budget annotation when the configured maximum is greater than 1. This helps agents understand that multiple calls to the same tool are allowed for the current workflow. Returns "toolname" when max is nil or "1" (default single-call behavior). Re

(name string, max *string)

// Returns "toolname" when max is nil or "1" (default single-call behavior).
// Returns "toolname(max:N)" when max > 1 so agents know the real action budget.
func toolWithMaxBudget(name string, max *string) string {
	if max == nil || *max == "1" {
		return name
	}
	return fmt.Sprintf("%s(max:%s)", name, *max)
}

// buildSafeOutputsSections returns the PromptSections that form the <safe-output-tools> block.
// The block contains: