fix: labels used

Signed-off-by: axel7083 <[email protected]>
containers · May 22, 2024 · c6d5aac · c6d5aac
1 parent ebbe98e
commit c6d5aac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/packages/backend/src/managers/applicationManager.spec.ts b/packages/backend/src/managers/applicationManager.spec.ts
@@ -1212,6 +1212,10 @@ describe('pod detection', async () => {
         'ai-lab-model-id': 'model-id-1',
       },
     });
+    expect(podManager.findPodByLabelsValues).toHaveBeenCalledWith({
+      'ai-lab-recipe-id': 'recipe-id-1',
+      'ai-lab-model-id': 'model-id-1',
+    });
   });
 
   test('deleteApplication calls stopPod and removePod', async () => {

diff --git a/packages/backend/src/managers/applicationManager.ts b/packages/backend/src/managers/applicationManager.ts
@@ -746,8 +746,8 @@ export class ApplicationManager extends Publisher<ApplicationState[]> implements
 
   private async findPod(recipeId: string, modelId: string): Promise<PodInfo | undefined> {
     return this.podManager.findPodByLabelsValues({
-      LABEL_RECIPE_ID: recipeId,
-      LABEL_MODEL_ID: modelId,
+      [LABEL_RECIPE_ID]: recipeId,
+      [LABEL_MODEL_ID]: modelId,
     });
   }