Method WrapToolCall

pkg/middleware/hitl.go:170–247

WrapToolCall 拦截工具调用,请求人工审核

(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler)

// WrapToolCall 拦截工具调用,请求人工审核
func (m *HumanInTheLoopMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error) {
	// 检查是否需要审核
	interruptCfg, needsApproval := m.interruptConfigs[req.ToolName]
	if !needsApproval {
		// 不需要审核,直接执行
		return handler(ctx, req)
	}

	hitlLog.Info(ctx, "tool requires approval", map[string]any{"tool": req.ToolName})

	// 构建审核请求
	reviewRequest := &ReviewRequest{
		ActionRequests: []ActionRequest{
			{
				ToolCallID: req.ToolCallID,
				ToolName:   req.ToolName,
				Input:      req.ToolInput,
				Message:    interruptCfg.Message,
			},
		},
		ReviewConfigs: []InterruptConfig{*interruptCfg},
	}

	// 获取人工决策
	decisions, err := m.getApproval(ctx, reviewRequest)
	if err != nil {
		return &ToolCallResponse{
			Result: map[string]any{
				"ok":    false,
				"error": fmt.Sprintf("approval request failed: %v", err),
			},
		}, nil
	}

	if len(decisions) == 0 {
		return &ToolCallResponse{
			Result: map[string]any{
				"ok":    false,
				"error": "no decision received",
			},
		}, nil
	}

	decision := decisions[0]

	// 处理决策
	switch decision.Type {
	case DecisionApprove:
		hitlLog.Info(ctx, "tool approved", map[string]any{"tool": req.ToolName})
		return handler(ctx, req)

	case DecisionEdit:
		hitlLog.Info(ctx, "tool approved with edited input", map[string]any{"tool": req.ToolName})
		// 使用编辑后的参数
		editedReq := *req
		editedReq.ToolInput = decision.EditedInput
		return handler(ctx, &editedReq)