diff --git a/api/internal/storeLink/octopus.go b/api/internal/storeLink/octopus.go index 49968007..75f4ebef 100644 --- a/api/internal/storeLink/octopus.go +++ b/api/internal/storeLink/octopus.go @@ -479,8 +479,8 @@ func (o *OctopusLink) GetTrainingTask(ctx context.Context, taskId string) (*coll if err != nil { return nil, err } - jobresp := (resp).(*octopus.GetTrainJobResp) - if !jobresp.Success { + jobresp, ok := (resp).(*octopus.GetTrainJobResp) + if !jobresp.Success || !ok { return nil, errors.New("get training task failed") } var task collector.Task