set correct pool name and machine id

beam-cloud · Dec 26, 2024 · ea353b9 · ea353b9
1 parent 396b967
commit ea353b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/repository/provider_redis.go b/pkg/repository/provider_redis.go
@@ -311,8 +311,8 @@ func (r *ProviderRedisRepository) RegisterMachine(providerName, poolName, machin
 		machineInfo.Gpu = poolConfig.GPUType
 		machineInfo.Created = fmt.Sprintf("%d", time.Now().UTC().Unix())
 		machineInfo.LastKeepalive = fmt.Sprintf("%d", time.Now().UTC().Unix())
-		machineInfo.PoolName = newMachineInfo.PoolName
-		machineInfo.MachineId = newMachineInfo.MachineId
+		machineInfo.PoolName = poolName
+		machineInfo.MachineId = machineId
 
 		// Add machine to index
 		machineIndexKey := common.RedisKeys.ProviderMachineIndex(providerName, poolName)