From 955d5fc9d0e9a76f49cc6d90e5b7fef96dc1f920 Mon Sep 17 00:00:00 2001
From: pantafive <56078241+pantafive@users.noreply.github.com>
Date: Thu, 21 Dec 2023 11:51:36 +0100
Subject: [PATCH] add templateId to create pods command

---
 cmd/pods/createPods.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cmd/pods/createPods.go b/cmd/pods/createPods.go
index f745780..ddd92de 100644
--- a/cmd/pods/createPods.go
+++ b/cmd/pods/createPods.go
@@ -22,6 +22,7 @@ var name string
 var podCount int
 var ports []string
 var secureCloud bool
+var templateId string
 var volumeInGb int
 var volumeMountPath string
 
@@ -42,6 +43,7 @@ var CreatePodsCmd = &cobra.Command{
 			MinMemoryInGb:     minMemoryInGb,
 			MinVcpuCount:      minVcpuCount,
 			Name:              name,
+			TemplateId:        templateId,
 			VolumeInGb:        volumeInGb,
 			VolumeMountPath:   volumeMountPath,
 		}
@@ -98,6 +100,7 @@ func init() {
 	CreatePodsCmd.Flags().StringVar(&gpuTypeId, "gpuType", "", "gpu type id, e.g. 'NVIDIA GeForce RTX 3090'")
 	CreatePodsCmd.Flags().StringVar(&imageName, "imageName", "", "container image name")
 	CreatePodsCmd.Flags().StringVar(&name, "name", "", "any pod name for easy reference")
+	CreatePodsCmd.Flags().StringVar(&templateId, "templateId", "", "templateId to use with the pods")
 	CreatePodsCmd.Flags().StringVar(&volumeMountPath, "volumePath", "/runpod", "container volume path")
 
 	CreatePodsCmd.MarkFlagRequired("gpuType")   //nolint