Skip to content

Commit 477ec80

Browse files
committed
增加Evaluator
1 parent 0bd11e8 commit 477ec80

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

backend/modules/evaluation/domain/component/rpc/trace_agent.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,5 +25,7 @@ type CallTraceAgentParam struct {
2525

2626
EvalTargetType entity.EvalTargetType // now support prompt only
2727
EvalTargetID int64
28-
EvalTargetVersionID int64
28+
EvalTargetVersionID string
29+
30+
Evaluators []*entity.ExptEvaluatorRef
2931
}

backend/modules/evaluation/domain/service/insight_analysis_impl.go

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,11 +157,17 @@ func (e ExptInsightAnalysisServiceImpl) GenAnalysisReport(ctx context.Context, s
157157
if err != nil {
158158
return err
159159
}
160-
param.EvalTargetVersionID, err = strconv.ParseInt(target.EvalTargetVersion.SourceTargetVersion, 10, 64)
160+
param.EvalTargetVersionID = target.EvalTargetVersion.SourceTargetVersion
161161
if err != nil {
162162
return err
163163
}
164164

165+
evaluators, err := e.exptRepo.GetEvaluatorRefByExptIDs(ctx, []int64{exptID}, spaceID)
166+
if err != nil {
167+
return err
168+
}
169+
param.Evaluators = evaluators
170+
165171
// only allow prompt eval target, but not return error here. The task will fail in the CallTraceAgent.
166172
if param.EvalTargetType != entity.EvalTargetTypeLoopPrompt {
167173
logs.CtxWarn(ctx, "Illegal evaltarget type %d for expt %d", param.EvalTargetType, exptID)

0 commit comments

Comments
 (0)