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()