refactor: Move preflight config to preflight package

Co-authored-by: logan <[email protected]>
Lilypad-Tech · Feb 3, 2025 · f01160d · f01160d
1 parent 4582da5
commit f01160d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 13 deletions.
diff --git a/pkg/options/resource-provider.go b/pkg/options/resource-provider.go
@@ -6,7 +6,6 @@ import (
 
 	"github.com/lilypad-tech/lilypad/pkg/data"
 	"github.com/lilypad-tech/lilypad/pkg/resourceprovider"
-	"github.com/lilypad-tech/lilypad/pkg/resourceprovider/preflight"
 	"github.com/lilypad-tech/lilypad/pkg/system"
 	"github.com/spf13/cobra"
 )
@@ -19,13 +18,6 @@ func NewResourceProviderOptions() resourceprovider.ResourceProviderOptions {
 		Pow:       GetDefaultResourceProviderPowOptions(),
 		IPFS:      GetDefaultIPFSOptions(),
 		Telemetry: GetDefaultTelemetryOptions(),
-		Preflight: preflight.PreflightConfig{
-			GPU: struct {
-				MinMemoryGB int64
-			}{
-				MinMemoryGB: preflight.RequiredGPUMemoryGB,
-			},
-		},
 	}
 	options.Web3.Service = system.ResourceProviderService
 	return options

diff --git a/pkg/resourceprovider/preflight/preflight.go b/pkg/resourceprovider/preflight/preflight.go
@@ -22,7 +22,7 @@ type CheckResult struct {
 	Error   error
 }
 
-type PreflightConfig struct {
+type preflightConfig struct {
 	GPU struct {
 		MinMemoryGB int64
 	}
@@ -35,9 +35,16 @@ type preflightChecker struct {
 	gpuInfo []GPUInfo
 }
 
-func RunPreflightChecks(ctx context.Context, config PreflightConfig) error {
+func RunPreflightChecks(ctx context.Context) error {
 	log.Info().Msg("Starting preflight checks...")
 	checker := &preflightChecker{}
+	config := preflightConfig{
+		GPU: struct {
+			MinMemoryGB int64
+		}{
+			MinMemoryGB: RequiredGPUMemoryGB,
+		},
+	}
 
 	// Logging GPU requirements
 	gpuInfo, err := checker.GetGPUInfo(ctx)
@@ -58,7 +65,7 @@ func RunPreflightChecks(ctx context.Context, config PreflightConfig) error {
 	return nil
 }
 
-func (p *preflightChecker) RunAllChecks(ctx context.Context, config PreflightConfig) error {
+func (p *preflightChecker) RunAllChecks(ctx context.Context, config preflightConfig) error {
 
 	gpuResult := p.CheckGPU(ctx, &GPUCheckConfig{
 		MinMemory: config.GPU.MinMemoryGB * 1024 * 1024 * 1024,

diff --git a/pkg/resourceprovider/resourceprovider.go b/pkg/resourceprovider/resourceprovider.go
@@ -74,7 +74,6 @@ type ResourceProviderOptions struct {
 	Pow       ResourceProviderPowOptions
 	IPFS      ipfs.IPFSOptions
 	Telemetry system.TelemetryOptions
-	Preflight preflight.PreflightConfig
 }
 
 type ResourceProvider struct {
@@ -90,7 +89,7 @@ func NewResourceProvider(
 	executor executor.Executor,
 	tracer trace.Tracer,
 ) (*ResourceProvider, error) {
-	if err := preflight.RunPreflightChecks(context.Background(), options.Preflight); err != nil {
+	if err := preflight.RunPreflightChecks(context.Background()); err != nil {
 		return nil, fmt.Errorf("preflight checks failed: %w", err)
 	}