File tree 1 file changed +11
-0
lines changed
packages/backend/src/managers/application 1 file changed +11
-0
lines changed Original file line number Diff line number Diff line change @@ -31,6 +31,7 @@ import { VMType } from '@shared/models/IPodman';
31
31
import { POD_LABEL_MODEL_ID , POD_LABEL_RECIPE_ID } from '../../utils/RecipeConstants' ;
32
32
import type { InferenceServer } from '@shared/models/IInference' ;
33
33
import type { RpcExtension } from '@shared/messages/MessageProxy' ;
34
+ import type { LlamaStackManager } from '../llama-stack/llamaStackManager' ;
34
35
35
36
const taskRegistryMock = {
36
37
createTask : vi . fn ( ) ,
@@ -75,6 +76,10 @@ const recipeManager = {
75
76
buildRecipe : vi . fn ( ) ,
76
77
} as unknown as RecipeManager ;
77
78
79
+ const llamaStackManager = {
80
+ getLlamaStackContainer : vi . fn ( ) ,
81
+ } as unknown as LlamaStackManager ;
82
+
78
83
vi . mock ( '@podman-desktop/api' , ( ) => ( {
79
84
window : {
80
85
withProgress : vi . fn ( ) ,
@@ -139,6 +144,11 @@ beforeEach(() => {
139
144
id : 'fake-task' ,
140
145
} ) ) ;
141
146
vi . mocked ( modelsManagerMock . uploadModelToPodmanMachine ) . mockResolvedValue ( 'downloaded-model-path' ) ;
147
+ vi . mocked ( llamaStackManager . getLlamaStackContainer ) . mockResolvedValue ( {
148
+ containerId : 'container1' ,
149
+ port : 10001 ,
150
+ playgroundPort : 10002 ,
151
+ } ) ;
142
152
} ) ;
143
153
144
154
function getInitializedApplicationManager ( ) : ApplicationManager {
@@ -151,6 +161,7 @@ function getInitializedApplicationManager(): ApplicationManager {
151
161
telemetryMock ,
152
162
podManager ,
153
163
recipeManager ,
164
+ llamaStackManager ,
154
165
) ;
155
166
156
167
manager . init ( ) ;
You can’t perform that action at this time.
0 commit comments