fix: importing model fresh install (#1133)

Signed-off-by: axel7083 <[email protected]>
containers · May 30, 2024 · 4441e14 · 4441e14
1 parent 26454aa
commit 4441e14
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/packages/backend/src/managers/catalogManager.spec.ts b/packages/backend/src/managers/catalogManager.spec.ts
@@ -42,6 +42,7 @@ vi.mock('node:fs', () => {
       readFile: vi.fn(),
       stat: vi.fn(),
       writeFile: vi.fn(),
+      mkdir: vi.fn(),
     },
   };
 });
@@ -183,6 +184,7 @@ test('expect to call writeFile in addLocalModelsToCatalog with catalog updated',
     },
   ]);
 
+  expect(promises.mkdir).toHaveBeenCalled();
   expect(writeFileMock).toBeCalledWith('path', JSON.stringify(updatedCatalog, undefined, 2), 'utf-8');
 });
 

diff --git a/packages/backend/src/managers/catalogManager.ts b/packages/backend/src/managers/catalogManager.ts
@@ -225,6 +225,9 @@ export class CatalogManager extends Publisher<ApplicationCatalog> implements Dis
     // Add all our models infos to the user's models catalog
     content.models.push(...models);
 
+    // ensure parent directory exists
+    await promises.mkdir(path.dirname(userCatalogPath), { recursive: true });
+
     // overwrite the existing catalog
     return promises.writeFile(userCatalogPath, JSON.stringify(content, undefined, 2), 'utf-8');
   }