Method SelectTools

pkg/tools/constraints.go:205–243

SelectTools 选择工具

(ctx context.Context, allTools []Tool, constraints ToolConstraints)

// SelectTools 选择工具
func (s *DefaultToolSelector) SelectTools(ctx context.Context, allTools []Tool, constraints ToolConstraints) ([]Tool, error) {
	if constraints == nil || constraints.GetConstraintType() == ConstraintTypeNone {
		return allTools, nil
	}

	// 根据约束类型过滤
	switch constraints.GetConstraintType() {
	case ConstraintTypeWhitelist, ConstraintTypeRequired:
		allowedNames := constraints.GetAllowedTools(ctx)
		if allowedNames == nil {
			return allTools, nil
		}

		allowed := make(map[string]bool, len(allowedNames))
		for _, name := range allowedNames {
			allowed[name] = true
		}

		filtered := make([]Tool, 0, len(allTools))
		for _, tool := range allTools {
			if allowed[tool.Name()] {
				filtered = append(filtered, tool)
			}
		}
		return filtered, nil

	case ConstraintTypeBlacklist:
		filtered := make([]Tool, 0, len(allTools))
		for _, tool := range allTools {
			if constraints.IsAllowed(ctx, tool.Name()) {
				filtered = append(filtered, tool)
			}
		}
		return filtered, nil

	default:
		return allTools, nil
	}
}

// ShouldUseToolChoice 判断是否应该使用 tool_choice
func (s *DefaultToolSelector) ShouldUseToolChoice(ctx context.Context, constraints ToolConstraints) (*ToolChoice, bool) {