ai-agent/internal/services/agent_evaluation_service.go

package services

import (
	"context"
	"fmt"

	"code.tczkiot.com/wlw/ai-agent/internal/pkg/dto/request"
	"code.tczkiot.com/wlw/ai-agent/internal/pkg/dto/response"
	"code.tczkiot.com/wlw/ai-agent/internal/pkg/errorsx"
)

var AgentEvaluationService = newAgentEvaluationService()
var AgentEvaluationRunHook func(context.Context, request.RunAgentEvaluationRequest) (*response.AgentEvaluationReportResponse, error)

type agentEvaluationService struct{}

func newAgentEvaluationService() *agentEvaluationService { return &agentEvaluationService{} }

func (s *agentEvaluationService) Run(ctx context.Context, req request.RunAgentEvaluationRequest) (*response.AgentEvaluationReportResponse, error) {
	if req.AIAgentID <= 0 {
		return nil, errorsx.InvalidParam("ai agent id is required")
	}
	if len(req.Cases) == 0 {
		return nil, errorsx.InvalidParam("evaluation cases are required")
	}
	if len(req.Cases) > 100 {
		return nil, errorsx.InvalidParam("evaluation case limit exceeded")
	}
	if AgentEvaluationRunHook == nil {
		return nil, fmt.Errorf("agent evaluation runner is not initialized")
	}
	return AgentEvaluationRunHook(ctx, req)
}