Update index.ts

Avaray · Nov 30, 2024 · 5ea17fc · 5ea17fc
1 parent d14b994
commit 5ea17fc
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/index.ts b/index.ts
@@ -5,7 +5,6 @@ import {
   getIds,
   getBranchName,
   getEnvironmentVariable,
-  normalizeFilename,
   runtime,
   saveFile,
   pvsUrl,
@@ -257,16 +256,20 @@ for (const ui of uis) {
     );
 
     if (template.id) {
-      console.log(`Template ${template.id} created successfully`);
-      templates[checkpoint.id!] = {
-        vastai: {
-          [ui.id]: template.id
-        }
+      console.log(`Template for ${ui.name}UI with ${checkpoint.name} created with id ${template.id}`);
+      // templates[checkpoint.id!].vastai[ui.id] = template.id;
+      if (!templates[checkpoint.id!]) {
+        templates[checkpoint.id!] = { vastai: {} };
       }
+      if (!templates[checkpoint.id!].vastai) {
+        templates[checkpoint.id!].vastai = {};
+      }
+      templates[checkpoint.id!].vastai[ui.id] = template.id;
     }
   }
 }
 
+
 await saveFile("data/templates.json", JSON.stringify(templates, null, 2));
 
 console.log("Done");