feat: [Coda] add TargetType parameter to MockEvalTargetOutput query chain

(LogID: 202509161632560100911101346893E51)

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

Author: tpfz

backend/modules/evaluation/application/eval_target_app.go

@@ -404,7 +404,7 @@ func (e EvalTargetApplicationImpl) MockEvalTargetOutput(ctx context.Context, req
 	}
 
 	// 获取评测目标信息，request.EvalTargetID实际上为SourceTargetID
-	evalTarget, err := e.evalTargetService.GetEvalTargetVersionBySourceTarget(ctx, request.WorkspaceID, strconv.FormatInt(request.SourceTargetID, 10), request.EvalTargetVersion, false)
+	evalTarget, err := e.evalTargetService.GetEvalTargetVersionBySourceTarget(ctx, request.WorkspaceID, strconv.FormatInt(request.SourceTargetID, 10), request.EvalTargetVersion, entity.EvalTargetType(request.TargetType), false)
 	if err != nil {
 		return nil, err
 	}
@@ -438,4 +438,4 @@ func (e EvalTargetApplicationImpl) MockEvalTargetOutput(ctx context.Context, req
 		EvalTarget: target.EvalTargetDO2DTO(evalTarget),
 		MockOutput: mockOutput,
 	}, nil
-}
+}

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

@@ -15,7 +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)
+	GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string, targetType entity.EvalTargetType) (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,7 +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)
+	GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string, targetType entity.EvalTargetType, 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,8 +87,8 @@ 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)
+func (e *EvalTargetServiceImpl) GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string, targetType entity.EvalTargetType, needSourceInfo bool) (do *entity.EvalTarget, err error) {
+	do, err = e.evalTargetRepo.GetEvalTargetVersionBySourceTarget(ctx, spaceID, sourceTargetID, sourceTargetVersion, targetType)
 	if err != nil {
 		return nil, err
 	}

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

@@ -164,11 +164,11 @@ 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) {
+func (e *EvalTargetRepoImpl) GetEvalTargetVersionBySourceTarget(ctx context.Context, spaceID int64, sourceTargetID string, sourceTargetVersion string, targetType entity.EvalTargetType) (targetDO *entity.EvalTarget, err error) {
 	var opts []db.Option
 
-	// 第一步：根据sourceTargetID查找target，获取targetID
-	targetPO, err := e.evalTargetDao.GetEvalTargetBySourceID(ctx, spaceID, sourceTargetID, int32(entity.EvalTargetTypeLoopPrompt), opts...)
+	// 第一步：根据sourceTargetID查找target，获取targetID，使用传入的targetType
+	targetPO, err := e.evalTargetDao.GetEvalTargetBySourceID(ctx, spaceID, sourceTargetID, int32(targetType), opts...)
 	if err != nil {
 		return nil, err
 	}