Upload 3 files

gss-sdk.js

@@ -43,6 +43,8 @@ class GSSClient {
     this.model       = opts.model || 'llama-3.1-8b-instant';
     this.storageKey  = opts.storageKey || 'gss_jwt';
     this._token      = null;
+    // Start keep-alive sync (only in browser context)
+    if (typeof window !== 'undefined') this._startKeepAlive();
   }
 
   // ── JWT management ──────────────────────────────────────────────────────────
@@ -52,9 +54,11 @@ class GSSClient {
     try {
       const raw = localStorage.getItem(this.storageKey);
       if (!raw) return null;
-      const { token, exp } = JSON.parse(raw);
-      // Expire 5 minutes early to avoid edge cases
-      if (Date.now() / 1000 < exp - 300) return token;
+      const { token, exp, hf_space } = JSON.parse(raw);
+      if (Date.now() / 1000 < exp - 300) {
+        if (hf_space) this.hfEngineUrl = hf_space.replace(/\/$/, '');
+        return token;
+      }
       localStorage.removeItem(this.storageKey);
     } catch { /* ignore */ }
     return null;
@@ -83,10 +87,12 @@ class GSSClient {
       throw new Error(`[GSSClient] Lease failed: ${res.status}`);
     }
 
-    const { token } = await res.json();
+    const { token, hf_space } = await res.json();
     const exp = this._decodeExp(token);
-    localStorage.setItem(this.storageKey, JSON.stringify({ token, exp }));
+    localStorage.setItem(this.storageKey, JSON.stringify({ token, exp, hf_space }));
     this._token = token;
+    // Switch to the assigned HF Space for load balancing
+    if (hf_space) this.hfEngineUrl = hf_space.replace(/\/$/, '');
     return token;
   }
 
@@ -167,6 +173,25 @@ class GSSClient {
     this._token = null;
     return this._lease();
   }
+
+  // ── Keep-alive sync ────────────────────────────────────────────────────────
+  /** Signal Cloudflare to push this subscriber's keys to HF Space.
+   *  Called automatically every 30 minutes to keep the key pool warm. */
+  async _syncKeys() {
+    try {
+      const token = await this._getToken();
+      await fetch(`${this.cfWorkerUrl}/auth/keep-alive`, {
+        method: 'POST',
+        headers: { 'Authorization': `Bearer ${token}`, 'Content-Type': 'application/json' },
+      });
+    } catch (_) { /* silent — non-critical */ }
+  }
+
+  _startKeepAlive() {
+    // Fire once 5s after page load, then every 30 minutes
+    setTimeout(() => this._syncKeys(), 5000);
+    setInterval(() => this._syncKeys(), 30 * 60 * 1000);
+  }
 }
 
 // Support both browser global and CommonJS/ESM

sdk/demo.html

@@ -155,7 +155,7 @@
   <div class="config">
     <div>
       <label>GSS API Key</label>
-      <input id="apiKey" type="text" placeholder="gss_..." value="gss_EraxhzKQHoeXahnkzCIMgg0PaP1lidCl" />
+      <input id="apiKey" type="text" placeholder="gss_..." value="gss_ZsXcS4qQigXT3DUgOZyKfs0zp1AB5i9T" />
     </div>
     <div class="row2">
       <div>

sdk/gss-sdk.js

@@ -54,9 +54,11 @@ class GSSClient {
     try {
       const raw = localStorage.getItem(this.storageKey);
       if (!raw) return null;
-      const { token, exp } = JSON.parse(raw);
-      // Expire 5 minutes early to avoid edge cases
-      if (Date.now() / 1000 < exp - 300) return token;
+      const { token, exp, hf_space } = JSON.parse(raw);
+      if (Date.now() / 1000 < exp - 300) {
+        if (hf_space) this.hfEngineUrl = hf_space.replace(/\/$/, '');
+        return token;
+      }
       localStorage.removeItem(this.storageKey);
     } catch { /* ignore */ }
     return null;
@@ -85,10 +87,12 @@ class GSSClient {
       throw new Error(`[GSSClient] Lease failed: ${res.status}`);
     }
 
-    const { token } = await res.json();
+    const { token, hf_space } = await res.json();
     const exp = this._decodeExp(token);
-    localStorage.setItem(this.storageKey, JSON.stringify({ token, exp }));
+    localStorage.setItem(this.storageKey, JSON.stringify({ token, exp, hf_space }));
     this._token = token;
+    // Switch to the assigned HF Space for load balancing
+    if (hf_space) this.hfEngineUrl = hf_space.replace(/\/$/, '');
     return token;
   }