feat: [Coda] fix MockEvalTargetOutput method to handle SourceTargetID correctly

(LogID: 202509161632560100911101346893E51)

Co-Authored-By: Coda <[email protected]>

Author: tpfz

backend/modules/evaluation/application/eval_target_app.go

@@ -403,8 +403,8 @@ func (e EvalTargetApplicationImpl) MockEvalTargetOutput(ctx context.Context, req
 		return nil, errorx.NewByCode(errno.CommonInvalidParamCode, errorx.WithExtraMsg("request is nil"))
 	}
 
-	// 获取评测目标信息
-	evalTarget, err := e.evalTargetService.GetEvalTargetVersionByTarget(ctx, request.WorkspaceID, request.EvalTargetID, request.EvalTargetVersion, false)
+	// 获取评测目标信息，request.EvalTargetID实际上为SourceTargetID
+	evalTarget, err := e.evalTargetService.GetEvalTargetVersionBySourceTarget(ctx, request.WorkspaceID, strconv.FormatInt(request.EvalTargetID, 10), request.EvalTargetVersion, false)
 	if err != nil {
 		return nil, err
 	}

backend/modules/evaluation/domain/repo/target.go

@@ -15,6 +15,7 @@ type IEvalTargetRepo interface {
 	GetEvalTarget(ctx context.Context, targetID int64) (do *entity.EvalTarget, err error)
 	GetEvalTargetVersion(ctx context.Context, spaceID, versionID int64) (do *entity.EvalTarget, err error)
 	GetEvalTargetVersionByTarget(ctx context.Context, spaceID int64, targetID int64, sourceTargetVersion string) (do *entity.EvalTarget, err error)
+	GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string) (do *entity.EvalTarget, err error)
 	BatchGetEvalTargetBySource(ctx context.Context, param *BatchGetEvalTargetBySourceParam) (dos []*entity.EvalTarget, err error)
 	BatchGetEvalTargetVersion(ctx context.Context, spaceID int64, versionIDs []int64) (dos []*entity.EvalTarget, err error)
 

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

@@ -16,6 +16,7 @@ type IEvalTargetService interface {
 	GetEvalTargetVersion(ctx context.Context, spaceID int64, versionID int64, needSourceInfo bool) (do *entity.EvalTarget, err error)
 	GetEvalTargetVersionBySource(ctx context.Context, spaceID int64, targetID int64, sourceVersion string, needSourceInfo bool) (do *entity.EvalTarget, err error)
 	GetEvalTargetVersionByTarget(ctx context.Context, spaceID int64, targetID int64, sourceTargetVersion string, needSourceInfo bool) (do *entity.EvalTarget, err error)
+	GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string, needSourceInfo bool) (do *entity.EvalTarget, err error)
 	BatchGetEvalTargetBySource(ctx context.Context, param *entity.BatchGetEvalTargetBySourceParam) (dos []*entity.EvalTarget, err error)
 	BatchGetEvalTargetVersion(ctx context.Context, spaceID int64, versionIDs []int64, needSourceInfo bool) (dos []*entity.EvalTarget, err error)
 

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

@@ -87,6 +87,23 @@ func (e *EvalTargetServiceImpl) GetEvalTargetVersion(ctx context.Context, spaceI
 	return do, nil
 }
 
+func (e *EvalTargetServiceImpl) GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string, needSourceInfo bool) (do *entity.EvalTarget, err error) {
+	do, err = e.evalTargetRepo.GetEvalTargetVersionBySourceTarget(ctx, spaceID, sourceTargetID, sourceTargetVersion)
+	if err != nil {
+		return nil, err
+	}
+	// 包装source info信息
+	if needSourceInfo {
+		for _, op := range e.typedOperators {
+			err = op.PackSourceVersionInfo(ctx, spaceID, []*entity.EvalTarget{do})
+			if err != nil {
+				return nil, err
+			}
+		}
+	}
+	return do, nil
+}
+
 func (e *EvalTargetServiceImpl) GetEvalTargetVersionBySource(ctx context.Context, spaceID int64, targetID int64, sourceVersion string, needSourceInfo bool) (do *entity.EvalTarget, err error) {
 	// 根据spaceID、targetID和sourceVersion查询版本
 	versions, err := e.evalTargetRepo.BatchGetEvalTargetBySource(ctx, &repo.BatchGetEvalTargetBySourceParam{
@@ -133,6 +150,7 @@ func (e *EvalTargetServiceImpl) GetEvalTargetVersionByTarget(ctx context.Context
 	return do, nil
 }
 
+
 func (e *EvalTargetServiceImpl) BatchGetEvalTargetBySource(ctx context.Context, param *entity.BatchGetEvalTargetBySourceParam) (dos []*entity.EvalTarget, err error) {
 	return e.evalTargetRepo.BatchGetEvalTargetBySource(ctx, &repo.BatchGetEvalTargetBySourceParam{
 		SpaceID:        param.SpaceID,

backend/modules/evaluation/infra/repo/target/eval_target_repo_impl.go

@@ -164,6 +164,41 @@ func (e *EvalTargetRepoImpl) GetEvalTargetVersion(ctx context.Context, spaceID,
 	return targetDO, nil
 }
 
+func (e *EvalTargetRepoImpl) GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string) (targetDO *entity.EvalTarget, err error) {
+	var opts []db.Option
+	
+	// 第一步：根据sourceTargetID查找target，获取targetID
+	targetPO, err := e.evalTargetDao.GetEvalTargetBySourceID(ctx, spaceID, sourceTargetID, int32(entity.EvalTargetTypeLoopPrompt), opts...)
+	if err != nil {
+		return nil, err
+	}
+	if targetPO == nil {
+		return nil, nil // 没有找到对应的target
+	}
+	
+	// 第二步：根据targetID和sourceTargetVersion查找版本信息
+	var versionOpts []db.Option
+	if e.lwt.CheckWriteFlagByID(ctx, platestwrite.ResourceTypeTargetVersion, targetPO.ID) {
+		versionOpts = append(versionOpts, db.WithMaster())
+		logs.CtxInfo(ctx, "GetEvalTargetVersionBySourceTarget CheckWriteFlagByID true")
+	}
+	
+	versionPO, err := e.evalTargetVersionDao.GetEvalTargetVersionByTarget(ctx, spaceID, targetPO.ID, sourceTargetVersion, versionOpts...)
+	if err != nil {
+		return nil, err
+	}
+	if versionPO == nil {
+		return nil, nil // 没有找到对应的版本
+	}
+	
+	// 转换为DO对象
+	targetDO = convertor.EvalTargetPO2DO(targetPO)
+	versionDO := convertor.EvalTargetVersionPO2DO(versionPO, targetDO.EvalTargetType)
+	targetDO.EvalTargetVersion = versionDO
+	
+	return targetDO, nil
+}
+
 func (e *EvalTargetRepoImpl) BatchGetEvalTargetBySource(ctx context.Context, param *repo.BatchGetEvalTargetBySourceParam) (dos []*entity.EvalTarget, err error) {
 	targets, err := e.evalTargetDao.BatchGetEvalTargetBySource(ctx, param.SpaceID, param.SourceTargetID, int32(param.TargetType))
 	if err != nil {
@@ -208,6 +243,7 @@ func (e *EvalTargetRepoImpl) GetEvalTargetVersionByTarget(ctx context.Context, s
 	return targetDO, nil
 }
 
+
 func (e *EvalTargetRepoImpl) BatchGetEvalTargetVersion(ctx context.Context, spaceID int64, versionIDs []int64) (dos []*entity.EvalTarget, err error) {
 	versions, err := e.evalTargetVersionDao.BatchGetEvalTargetVersion(ctx, spaceID, versionIDs)
 	if err != nil {