From fe6412e1d4cb7b6ba1553ee5cb66793441e1dd9b Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Thu, 21 Nov 2024 13:13:29 +0700
Subject: [PATCH] feat: configure HuggingfaceToken via cortex.cpp

---
 extensions/model-extension/src/index.ts | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/extensions/model-extension/src/index.ts b/extensions/model-extension/src/index.ts
index c63510c37..e29084bc2 100644
--- a/extensions/model-extension/src/index.ts
+++ b/extensions/model-extension/src/index.ts
@@ -20,6 +20,10 @@ import { deleteModelFiles } from './legacy/delete'
 
 declare const SETTINGS: Array<any>
 
+export enum Settings {
+  huggingfaceToken = 'hugging-face-access-token',
+}
+
 /**
  * A extension for models
  */
@@ -33,10 +37,29 @@ export default class JanModelExtension extends ModelExtension {
   async onLoad() {
     this.registerSettings(SETTINGS)
 
+    // Configure huggingface token if available
+    const huggingfaceToken = await this.getSetting<string>(
+      Settings.huggingfaceToken,
+      undefined
+    )
+    if (huggingfaceToken)
+      this.cortexAPI.configs({ huggingface_token: huggingfaceToken })
+
     // Listen to app download events
     this.handleDesktopEvents()
   }
 
+  /**
+   * Subscribe to settings update and make change accordingly
+   * @param key
+   * @param value
+   */
+  onSettingUpdate<T>(key: string, value: T): void {
+    if (key === Settings.huggingfaceToken) {
+      this.cortexAPI.configs({ huggingface_token: value })
+    }
+  }
+
   /**
    * Called when the extension is unloaded.
    * @override