feat: expose models directory in the api

containers · Jan 30, 2024 · a579f93 · a579f93
1 parent 6527e08
commit a579f93
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/packages/backend/src/managers/modelsManager.ts b/packages/backend/src/managers/modelsManager.ts
@@ -58,6 +58,10 @@ export class ModelsManager {
     });
   }
 
+  getModelsDirectory(): string {
+    return this.#modelsDir;
+  }
+
   getLocalModelsFromDisk(): void {
     if (!fs.existsSync(this.#modelsDir)) {
       return;

diff --git a/packages/backend/src/studio-api-impl.ts b/packages/backend/src/studio-api-impl.ts
@@ -118,4 +118,8 @@ export class StudioApiImpl implements StudioAPI {
   async deleteLocalModel(modelId: string): Promise<void> {
     await this.modelsManager.deleteLocalModel(modelId);
   }
+
+  async getModelsDirectory(): Promise<string> {
+    return this.modelsManager.getModelsDirectory();
+  }
 }
diff --git a/packages/shared/src/StudioAPI.ts b/packages/shared/src/StudioAPI.ts
@@ -25,4 +25,6 @@ export abstract class StudioAPI {
   abstract askPlayground(modelId: string, prompt: string): Promise<number>;
   abstract getPlaygroundQueriesState(): Promise<QueryState[]>;
   abstract getPlaygroundsState(): Promise<PlaygroundState[]>;
+
+  abstract getModelsDirectory(): Promise<string>;
 }