feat: start a recipe with llama-stack backend

packages/backend/src/managers/application/applicationManager.spec.ts

@@ -31,6 +31,7 @@ import { VMType } from '@shared/models/IPodman';
 import { POD_LABEL_MODEL_ID, POD_LABEL_RECIPE_ID } from '../../utils/RecipeConstants';
 import type { InferenceServer } from '@shared/models/IInference';
 import type { RpcExtension } from '@shared/messages/MessageProxy';
+import type { LlamaStackManager } from '../llama-stack/llamaStackManager';
 
 const taskRegistryMock = {
   createTask: vi.fn(),
@@ -75,6 +76,10 @@ const recipeManager = {
   buildRecipe: vi.fn(),
 } as unknown as RecipeManager;
 
+const llamaStackManager = {
+  getLlamaStackContainer: vi.fn(),
+} as unknown as LlamaStackManager;
+
 vi.mock('@podman-desktop/api', () => ({
   window: {
     withProgress: vi.fn(),
@@ -139,6 +144,11 @@ beforeEach(() => {
     id: 'fake-task',
   }));
   vi.mocked(modelsManagerMock.uploadModelToPodmanMachine).mockResolvedValue('downloaded-model-path');
+  vi.mocked(llamaStackManager.getLlamaStackContainer).mockResolvedValue({
+    containerId: 'container1',
+    port: 10001,
+    playgroundPort: 10002,
+  });
 });
 
 function getInitializedApplicationManager(): ApplicationManager {
@@ -151,6 +161,7 @@ function getInitializedApplicationManager(): ApplicationManager {
     telemetryMock,
     podManager,
     recipeManager,
+    llamaStackManager,
   );
 
   manager.init();
@@ -160,11 +171,11 @@ function getInitializedApplicationManager(): ApplicationManager {
 describe('requestPullApplication', () => {
   test('task should be set to error if pull application raise an error', async () => {
     vi.mocked(window.withProgress).mockRejectedValue(new Error('pull application error'));
-    const trackingId = await getInitializedApplicationManager().requestPullApplication(
-      connectionMock,
-      recipeMock,
-      remoteModelMock,
-    );
+    const trackingId = await getInitializedApplicationManager().requestPullApplication({
+      connection: connectionMock,
+      recipe: recipeMock,
+      model: remoteModelMock,
+    });
 
     // ensure the task is created
     await vi.waitFor(() => {
@@ -292,9 +303,16 @@ describe('startApplication', () => {
 
 describe('pullApplication', () => {
   test('labels should be propagated', async () => {
-    await getInitializedApplicationManager().pullApplication(connectionMock, recipeMock, remoteModelMock, {
-      'test-label': 'test-value',
-    });
+    await getInitializedApplicationManager().pullApplication(
+      {
+        connection: connectionMock,
+        recipe: recipeMock,
+        model: remoteModelMock,
+      },
+      {
+        'test-label': 'test-value',
+      },
+    );
 
     // clone the recipe
     expect(recipeManager.cloneRecipe).toHaveBeenCalledWith(recipeMock, {
@@ -314,11 +332,18 @@ describe('pullApplication', () => {
       'model-id': remoteModelMock.id,
     });
     // build the recipe
-    expect(recipeManager.buildRecipe).toHaveBeenCalledWith(connectionMock, recipeMock, remoteModelMock, {
-      'test-label': 'test-value',
-      'recipe-id': recipeMock.id,
-      'model-id': remoteModelMock.id,
-    });
+    expect(recipeManager.buildRecipe).toHaveBeenCalledWith(
+      {
+        connection: connectionMock,
+        recipe: recipeMock,
+        model: remoteModelMock,
+      },
+      {
+        'test-label': 'test-value',
+        'recipe-id': recipeMock.id,
+        'model-id': remoteModelMock.id,
+      },
+    );
     // create AI App task must be created
     expect(taskRegistryMock.createTask).toHaveBeenCalledWith('Creating AI App', 'loading', {
       'test-label': 'test-value',
@@ -361,9 +386,17 @@ describe('pullApplication', () => {
         },
       } as InferenceServer,
     });
-    await getInitializedApplicationManager().pullApplication(connectionMock, recipeMock, remoteModelMock, {
-      'test-label': 'test-value',
-    });
+    vi.mocked(modelsManagerMock.requestDownloadModel).mockResolvedValue('/path/to/model');
+    await getInitializedApplicationManager().pullApplication(
+      {
+        connection: connectionMock,
+        recipe: recipeMock,
+        model: remoteModelMock,
+      },
+      {
+        'test-label': 'test-value',
+      },
+    );
 
     // clone the recipe
     expect(recipeManager.cloneRecipe).toHaveBeenCalledWith(recipeMock, {
@@ -379,11 +412,18 @@ describe('pullApplication', () => {
     // upload model to podman machine
     expect(modelsManagerMock.uploadModelToPodmanMachine).not.toHaveBeenCalled();
     // build the recipe
-    expect(recipeManager.buildRecipe).toHaveBeenCalledWith(connectionMock, recipeMock, remoteModelMock, {
-      'test-label': 'test-value',
-      'recipe-id': recipeMock.id,
-      'model-id': remoteModelMock.id,
-    });
+    expect(recipeManager.buildRecipe).toHaveBeenCalledWith(
+      {
+        connection: connectionMock,
+        recipe: recipeMock,
+        model: remoteModelMock,
+      },
+      {
+        'test-label': 'test-value',
+        'recipe-id': recipeMock.id,
+        'model-id': remoteModelMock.id,
+      },
+    );
     // create AI App task must be created
     expect(taskRegistryMock.createTask).toHaveBeenCalledWith('Creating AI App', 'loading', {
       'test-label': 'test-value',
@@ -427,7 +467,11 @@ describe('pullApplication', () => {
       },
     } as unknown as PodInfo);
 
-    await getInitializedApplicationManager().pullApplication(connectionMock, recipeMock, remoteModelMock);
+    await getInitializedApplicationManager().pullApplication({
+      connection: connectionMock,
+      recipe: recipeMock,
+      model: remoteModelMock,
+    });
 
     // removing existing application should create a task to notify the user
     expect(taskRegistryMock.createTask).toHaveBeenCalledWith('Removing AI App', 'loading', {
@@ -456,7 +500,11 @@ describe('pullApplication', () => {
       ],
     });
 
-    await getInitializedApplicationManager().pullApplication(connectionMock, recipeMock, remoteModelMock);
+    await getInitializedApplicationManager().pullApplication({
+      connection: connectionMock,
+      recipe: recipeMock,
+      model: remoteModelMock,
+    });
 
     // the remove pod should have been called
     expect(containerEngine.createContainer).toHaveBeenCalledWith(