Fix is not found error for snapshots (#227)

e2b-dev · Jan 8, 2025 · 72de688 · 72de688
2 parents de3decb + 72af861
commit 72de688
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/shared/pkg/db/snapshot.go b/packages/shared/pkg/db/snapshot.go
@@ -130,9 +130,13 @@ func (db *DB) GetLastSnapshot(ctx context.Context, sandboxID string, teamID uuid
 
 	notFound := models.IsNotFound(err)
 
-	if err != nil && !notFound {
+	if err != nil {
 		return nil, nil, fmt.Errorf("failed to get snapshot build for '%s': %w", sandboxID, err)
 	}
 
+	if notFound {
+		return nil, nil, fmt.Errorf("no snapshot build found for '%s'", sandboxID)
+	}
+
 	return e.Edges.Snapshots[0], e.Edges.Builds[0], nil
 }