fix issues with merge

Author: djeebus

packages/orchestrator/benchmark_test.go

@@ -37,6 +37,7 @@ import (
 	sbxlogger "github.com/e2b-dev/infra/packages/shared/pkg/logger/sandbox"
 	"github.com/e2b-dev/infra/packages/shared/pkg/smap"
 	"github.com/e2b-dev/infra/packages/shared/pkg/storage"
+	"github.com/e2b-dev/infra/packages/shared/pkg/storage/providers"
 	"github.com/e2b-dev/infra/packages/shared/pkg/telemetry"
 	"github.com/e2b-dev/infra/packages/shared/pkg/utils"
 )
@@ -153,7 +154,7 @@ func BenchmarkBaseImageLaunch(b *testing.B) {
 	limiter, err := limit.New(b.Context(), featureFlags)
 	require.NoError(b, err)
 
-	persistence, err := storage.GetTemplateStorageProvider(b.Context(), limiter)
+	persistence, err := providers.GetTemplateStorageProvider(b.Context(), limiter)
 	require.NoError(b, err)
 
 	blockMetrics, err := blockmetrics.NewMetrics(&noop.MeterProvider{})
@@ -196,10 +197,10 @@ func BenchmarkBaseImageLaunch(b *testing.B) {
 	artifactRegistry, err := artifactsregistry.GetArtifactsRegistryProvider(b.Context())
 	require.NoError(b, err)
 
-	persistenceTemplate, err := storage.GetTemplateStorageProvider(b.Context(), nil)
+	persistenceTemplate, err := providers.GetTemplateStorageProvider(b.Context(), nil)
 	require.NoError(b, err)
 
-	persistenceBuild, err := storage.GetBuildCacheStorageProvider(b.Context(), nil)
+	persistenceBuild, err := providers.GetBuildCacheStorageProvider(b.Context(), nil)
 	require.NoError(b, err)
 
 	var proxyPort uint16 = 5007
@@ -231,6 +232,7 @@ func BenchmarkBaseImageLaunch(b *testing.B) {
 		sandboxes,
 		templateCache,
 		buildMetrics,
+		featureFlags,
 	)
 
 	buildPath := filepath.Join(os.Getenv("LOCAL_TEMPLATE_STORAGE_BASE_PATH"), buildID, "rootfs.ext4")