Fixed: Error when user does not choose migration location (#268)

* Hack: Return empty * Fix test.
Comfy-Org · Nov 16, 2024 · 98db404 · 98db404
1 parent b4f54b1
commit 98db404
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/config/comfyServerConfig.ts b/src/config/comfyServerConfig.ts
@@ -208,9 +208,9 @@ export class ComfyServerConfig {
   public static async getMigrationConfig(
     migrationSource?: string,
     migrationItemIds: Set<string> = new Set()
-  ): Promise<Record<string, ModelPaths>> {
+  ): Promise<{ comfyui: ModelPaths } & Record<string, ModelPaths>> {
     if (!migrationSource || !migrationItemIds.has('models')) {
-      return {};
+      return { comfyui: {} };
     }
     // The yaml file exited in migration source repo.
     const migrationServerConfig = await ComfyServerConfig.getConfigFromRepoPath(migrationSource);

diff --git a/tests/unit/comfyServerConfig.test.ts b/tests/unit/comfyServerConfig.test.ts
@@ -141,7 +141,9 @@ comfyui:
   describe('getMigrationConfig', () => {
     it('should return empty object when no migration source is provided', async () => {
       const result = await ComfyServerConfig.getMigrationConfig(undefined);
-      expect(result).toEqual({});
+      expect(result).toEqual({
+        comfyui: {},
+      });
     });
 
     it('should merge configs and remove custom_nodes when migration source is provided', async () => {