From e41f46fd10659e76bf562cf7cadd81e331c7231b Mon Sep 17 00:00:00 2001
From: velaraptor-runpod <chris.vela@runpod.io>
Date: Fri, 6 Mar 2026 13:51:42 -0600
Subject: [PATCH] fix: add computeType for create Pod; add category for create
 template

---
 src/index.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/index.ts b/src/index.ts
index 590c0b3..b69eee6 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -181,6 +181,7 @@ server.tool(
   {
     name: z.string().optional().describe('Name for the pod'),
     imageName: z.string().describe('Docker image to use'),
+    computeType: z.string().optional().describe('Compute type (NVIDIA, CPU) for the pod'),
     cloudType: z
       .enum(['SECURE', 'COMMUNITY'])
       .optional()
@@ -530,6 +531,7 @@ server.tool(
   {
     name: z.string().describe('Name for the template'),
     imageName: z.string().describe('Docker image to use'),
+    category: z.string().optional().describe('Compute category (NVIDIA, AMD, CPU) for the template'),
     isServerless: z
       .boolean()
       .optional()