From 8056fc58506832d85bd5e73d4af138977aa6bcfd Mon Sep 17 00:00:00 2001
From: Nikhil Vasan <nikhil@skip.money>
Date: Fri, 15 Mar 2024 08:44:25 -0700
Subject: [PATCH] digital ocean img id

---
 core/provider/digitalocean/droplet.go | 2 +-
 core/provider/digitalocean/types.go   | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/core/provider/digitalocean/droplet.go b/core/provider/digitalocean/droplet.go
index 5e35b27..1fa94bf 100644
--- a/core/provider/digitalocean/droplet.go
+++ b/core/provider/digitalocean/droplet.go
@@ -32,7 +32,7 @@ func (p *Provider) CreateDroplet(ctx context.Context, definition provider.TaskDe
 		Region: doConfig.Region,
 		Size:   doConfig.Size,
 		Image: godo.DropletCreateImage{
-			ID: 148794370,
+			ID: doConfig.ImageID,
 		},
 		SSHKeys: []godo.DropletCreateSSHKey{
 			{
diff --git a/core/provider/digitalocean/types.go b/core/provider/digitalocean/types.go
index 42d299c..8806f8b 100644
--- a/core/provider/digitalocean/types.go
+++ b/core/provider/digitalocean/types.go
@@ -3,4 +3,5 @@ package digitalocean
 type DigitalOceanTaskConfig struct {
 	Region string
 	Size   string
+	ImageID int
 }