fix(worker): fix mishandled workflow not found

instill-ai · Jul 2, 2024 · 0189dd8 · 0189dd8
1 parent 26c76b2
commit 0189dd8
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pkg/service/worker.go b/pkg/service/worker.go
@@ -65,15 +65,21 @@ func (s *service) GetNamespaceLatestModelOperation(ctx context.Context, ns resou
 		return nil, err
 	}
 
-	outputWorkflowID := s.redisClient.Get(ctx, fmt.Sprintf("model_trigger_output_key:%s:%s", userUID, dbModel.UID.String())).Val()
+	outputWorkflowID, err := s.redisClient.Get(ctx, fmt.Sprintf("model_trigger_output_key:%s:%s", userUID, dbModel.UID.String())).Result()
+	if err != nil {
+		if errors.Is(err, redis.Nil) {
+			return nil, nil
+		}
+		return nil, err
+	}
+
 	operationID, err := resource.GetOperationID(outputWorkflowID)
 	if err != nil {
 		return nil, err
 	}
 
 	workflowExecutionRes, err := s.temporalClient.DescribeWorkflowExecution(ctx, operationID, "")
 	if err != nil {
-		fmt.Println(err)
 		return nil, err
 	}