Remove -pricePerUnit requirement for orchestrator with -AIWorker flag

livepeer · May 8, 2024 · 33f4b58 · 33f4b58
1 parent dfb4320
commit 33f4b58
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/cmd/livepeer/starter/starter.go b/cmd/livepeer/starter/starter.go
@@ -873,19 +873,16 @@ func StartLivepeer(ctx context.Context, cfg LivepeerConfig) {
 
 		if *cfg.Orchestrator {
 			// Set price per pixel base info
-			if *cfg.PixelsPerUnit <= 0 {
-				// Can't divide by 0
-				panic(fmt.Errorf("-pixelsPerUnit must be > 0, provided %d", *cfg.PixelsPerUnit))
-			}
-			if cfg.PricePerUnit == nil {
-				// Prevent orchestrators from unknowingly providing free transcoding
+			if cfg.PricePerUnit == nil && *cfg.AIWorker == false {
 				panic(fmt.Errorf("-pricePerUnit must be set"))
+			} else if cfg.PricePerUnit != nil {
+				if *cfg.PricePerUnit < 0 {
+					panic(fmt.Errorf("-pricePerUnit must be >= 0, provided %d", *cfg.PricePerUnit))
+				}
+
+				n.SetBasePrice("default", big.NewRat(int64(*cfg.PricePerUnit), int64(*cfg.PixelsPerUnit)))
+				glog.Infof("Price: %d wei for %d pixels\n ", *cfg.PricePerUnit, *cfg.PixelsPerUnit)
 			}
-			if *cfg.PricePerUnit < 0 {
-				panic(fmt.Errorf("-pricePerUnit must be >= 0, provided %d", *cfg.PricePerUnit))
-			}
-			n.SetBasePrice("default", big.NewRat(int64(*cfg.PricePerUnit), int64(*cfg.PixelsPerUnit)))
-			glog.Infof("Price: %d wei for %d pixels\n ", *cfg.PricePerUnit, *cfg.PixelsPerUnit)
 
 			if *cfg.PricePerBroadcaster != "" {
 				ppb := getBroadcasterPrices(*cfg.PricePerBroadcaster)