Skip to content

Commit ac45f2b

Browse files
dsf86HearyShen
authored andcommitted
[feat] [eval] fix video (#432)
fix video
1 parent dfc65bf commit ac45f2b

2 files changed

Lines changed: 20 additions & 0 deletions

File tree

backend/modules/evaluation/application/convertor/evaluation_set/openapi.go

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -578,6 +578,8 @@ func OpenAPIContentDO2DTO(content *entity.Content) *common.Content {
578578
ContentType: convertDOContentTypeToOpenAPI(gptr.Indirect(content.ContentType)),
579579
Text: content.Text,
580580
Image: ConvertImageDO2DTO(content.Image),
581+
Audio: ConvertAudioDO2DTO(content.Audio),
582+
Video: ConvertVideoDO2DTO(content.Video),
581583
MultiPart: multiPart,
582584
ContentOmitted: content.ContentOmitted,
583585
FullContent: ConvertObjectStorageDO2DTO(content.FullContent),
@@ -617,6 +619,18 @@ func ConvertAudioDO2DTO(audio *entity.Audio) *common.Audio {
617619
}
618620
}
619621

622+
func ConvertVideoDO2DTO(video *entity.Video) *common.Video {
623+
if video == nil {
624+
return nil
625+
}
626+
return &common.Video{
627+
Name: video.Name,
628+
URL: video.URL,
629+
ThumbURL: video.ThumbURL,
630+
URI: video.URI,
631+
}
632+
}
633+
620634
func OpenAPIItemErrorGroupDO2DTOs(dos []*entity.ItemErrorGroup) []*openapi_eval_set.ItemErrorGroup {
621635
if dos == nil {
622636
return nil

backend/modules/evaluation/application/convertor/evaluation_set/openapi_test.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -533,6 +533,10 @@ func TestOpenAPIItemDOToDTOConversions(t *testing.T) {
533533
Format: &audioFormat,
534534
URL: &audioURL,
535535
},
536+
Video: &entity.Video{
537+
Name: &imageName,
538+
URL: &imageURL,
539+
},
536540
}
537541

538542
do := &entity.EvaluationSetItem{
@@ -570,6 +574,8 @@ func TestOpenAPIItemDOToDTOConversions(t *testing.T) {
570574
URL: &imageURL,
571575
ThumbURL: &thumbURL,
572576
},
577+
Audio: &common.Audio{Format: &audioFormat, URL: &audioURL},
578+
Video: &common.Video{Name: &imageName, URL: &imageURL},
573579
MultiPart: []*common.Content{
574580
{
575581
ContentType: ptr[common.ContentType](common.ContentTypeText),

0 commit comments

Comments
 (0)