diff --git a/packages/backend/brain.woff2 b/packages/backend/brain.woff2 new file mode 100644 index 000000000..ae843b25c Binary files /dev/null and b/packages/backend/brain.woff2 differ diff --git a/packages/backend/package.json b/packages/backend/package.json index f9580ddf8..ec55e336b 100644 --- a/packages/backend/package.json +++ b/packages/backend/package.json @@ -11,8 +11,23 @@ }, "main": "./dist/extension.js", "contributes": { - "configuration": {}, - "menus": {} + "icons": { + "brain-icon": { + "description": "Brain icon", + "default": { + "fontPath": "brain.woff2", + "fontCharacter": "\\E001" + } + } + }, + "views": { + "icons/containersList": [ + { + "when": "ia-studio-model in containerLabelKeys", + "icon": "${brain-icon}" + } + ] + } }, "scripts": { "build": "vite build", diff --git a/packages/backend/src/managers/playground.ts b/packages/backend/src/managers/playground.ts index af1016929..04820f4c8 100644 --- a/packages/backend/src/managers/playground.ts +++ b/packages/backend/src/managers/playground.ts @@ -134,6 +134,9 @@ export class PlayGroundManager { ], }, }, + Labels: { + 'ia-studio-model': modelId, + }, Cmd: ['--models-path', '/models', '--context-size', '700', '--threads', '4'], });