feat: adding icons to container images managed by the extension (#111)

* feat: adding icons to container images managed by the extension Signed-off-by: axel7083 <[email protected]> * fix: corrupt brain.woff2 Signed-off-by: axel7083 <[email protected]> * fix: prettier Signed-off-by: axel7083 <[email protected]> --------- Signed-off-by: axel7083 <[email protected]>
containers · Jan 23, 2024 · ef57545 · ef57545
1 parent 51d9a65
commit ef57545
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/packages/backend/brain.woff2 b/packages/backend/brain.woff2
diff --git 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
@@ -134,6 +134,9 @@ export class PlayGroundManager {
           ],
         },
       },
+      Labels: {
+        'ia-studio-model': modelId,
+      },
       Cmd: ['--models-path', '/models', '--context-size', '700', '--threads', '4'],
     });
-Original file line number
+Diff line change
@@ Expand Up / @@ -134,6 +134,9 @@ export class PlayGroundManager { @@
               ],
             },
           },
+          Labels: {
+            'ia-studio-model': modelId,
+          },
           Cmd: ['--models-path', '/models', '--context-size', '700', '--threads', '4'],
         });
@@ Expand Down @@