qwen-opencode-provider 3.1.0 → 3.2.1

package/index.js

@@ -2,7 +2,7 @@
  * OpenCode Qwen API Plugin
  * 
  * Uses local proxy (like Pollinations) to handle auth automatically.
- * Auto-fetches all models from Qwen API.
+ * Auto-fetches models, validates and refreshes tokens.
  */
 
 import http from 'http';
@@ -46,8 +46,7 @@ function readApiKeyFromAuth() {
       
       const providerAuth = auth[PROVIDER_ID];
       if (providerAuth) {
-        const apiKey = providerAuth.key || providerAuth.apiKey || providerAuth.token || null;
-        return apiKey;
+        return providerAuth.key || providerAuth.apiKey || providerAuth.token || null;
       }
     }
   } catch (e) {
@@ -56,10 +55,188 @@ function readApiKeyFromAuth() {
   return null;
 }
 
+function saveApiKeyToAuth(newKey) {
+  try {
+    const authPath = getAuthFilePath();
+    let auth = {};
+    
+    if (fs.existsSync(authPath)) {
+      const content = fs.readFileSync(authPath, 'utf-8');
+      auth = JSON.parse(content);
+    }
+    
+    auth[PROVIDER_ID] = {
+      type: 'api',
+      key: newKey
+    };
+    
+    fs.writeFileSync(authPath, JSON.stringify(auth, null, 2));
+    log(`[Auth] Saved new token to auth file`);
+    return true;
+  } catch (e) {
+    log(`[Auth] Error saving: ${e.message}`);
+    return false;
+  }
+}
+
+async function validateToken(apiKey) {
+  return new Promise((resolve) => {
+    const options = {
+      hostname: 'qwen.aikit.club',
+      port: 443,
+      path: '/v1/validate',
+      method: 'GET',
+      headers: {
+        'Host': 'qwen.aikit.club',
+        'Authorization': `Bearer ${apiKey}`
+      }
+    };
+
+    const req = https.request(options, (res) => {
+      let data = '';
+      res.on('data', chunk => data += chunk);
+      res.on('end', () => {
+        try {
+          const json = JSON.parse(data);
+          log(`[Validate] Status: ${res.statusCode}, Response: ${JSON.stringify(json).substring(0, 100)}`);
+          
+          // Valid token returns user info, invalid returns error
+          if (res.statusCode === 200 && json.id) {
+            resolve({ valid: true, data: json });
+          } else {
+            resolve({ valid: false, error: json.error || 'Unknown error' });
+          }
+        } catch (e) {
+          log(`[Validate] Parse error: ${e.message}`);
+          resolve({ valid: false, error: e.message });
+        }
+      });
+    });
+
+    req.on('error', (e) => {
+      log(`[Validate] Error: ${e.message}`);
+      resolve({ valid: false, error: e.message });
+    });
+
+    req.end();
+  });
+}
+
+async function refreshToken() {
+  const currentKey = readApiKeyFromAuth();
+  if (!currentKey) {
+    log(`[Refresh] No current token to refresh`);
+    return null;
+  }
+
+  return new Promise((resolve) => {
+    const options = {
+      hostname: 'qwen.aikit.club',
+      port: 443,
+      path: '/v1/refresh',
+      method: 'GET',
+      headers: {
+        'Host': 'qwen.aikit.club',
+        'Authorization': `Bearer ${currentKey}`
+      }
+    };
+
+    const req = https.request(options, (res) => {
+      let data = '';
+      res.on('data', chunk => data += chunk);
+      res.on('end', () => {
+        try {
+          const json = JSON.parse(data);
+          log(`[Refresh] Status: ${res.statusCode}, Response: ${JSON.stringify(json).substring(0, 200)}`);
+          
+          if (res.statusCode === 200 && json.access_token) {
+            saveApiKeyToAuth(json.access_token);
+            resolve(json.access_token);
+          } else if (res.statusCode === 200 && json.id) {
+            // Token is still valid, no refresh needed
+            resolve(currentKey);
+          } else {
+            resolve(null);
+          }
+        } catch (e) {
+          log(`[Refresh] Parse error: ${e.message}`);
+          resolve(null);
+        }
+      });
+    });
+
+    req.on('error', (e) => {
+      log(`[Refresh] Error: ${e.message}`);
+      resolve(null);
+    });
+
+    req.end();
+  });
+}
+
+async function checkAndRefreshToken() {
+  const apiKey = readApiKeyFromAuth();
+  if (!apiKey) {
+    log(`[Token] No token found`);
+    return null;
+  }
+
+  log(`[Token] Validating current token...`);
+  const result = await validateToken(apiKey);
+  
+  if (result.valid) {
+    log(`[Token] Token is valid`);
+    return apiKey;
+  }
+
+  log(`[Token] Token invalid/expired, attempting refresh...`);
+  const newToken = await refreshToken();
+  
+  if (newToken) {
+    log(`[Token] Token refreshed successfully`);
+    return newToken;
+  }
+
+  log(`[Token] Could not refresh token`);
+  return null;
+}
+
+const MODEL_CAPABILITIES = {
+  'qvq-max': { name: 'QVQ Max', vision: true, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen-deep-research': { name: 'Qwen Deep Research', vision: false, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen2.5-max': { name: 'Qwen2.5 Max', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen3-next-80b-a3b': { name: 'Qwen3 Next 80B A3B', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen2.5-plus': { name: 'Qwen2.5 Plus', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen2.5-turbo': { name: 'Qwen2.5 Turbo', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen2.5-14b-instruct-1m': { name: 'Qwen2.5 14B Instruct 1M', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen2.5-72b-instruct': { name: 'Qwen2.5 72B Instruct', vision: true, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen2.5-coder-32b-instruct': { name: 'Qwen2.5 Coder 32B', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen2.5-omni-7b': { name: 'Qwen2.5 Omni 7B', vision: true, reasoning: false, webSearch: true, toolCalling: false },
+  'qwen2.5-vl-32b-instruct': { name: 'Qwen2.5 VL 32B', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen3-235b-a22b-2507': { name: 'Qwen3 235B A22B', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen3-30b-a3b-2507': { name: 'Qwen3 30B A3B', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen3-coder': { name: 'Qwen3 Coder', vision: true, reasoning: false, webSearch: true, toolCalling: true },
+  'qwen3-coder-flash': { name: 'Qwen3 Coder Flash', vision: true, reasoning: false, webSearch: true, toolCalling: false },
+  'qwen3-max': { name: 'Qwen3 Max', vision: true, reasoning: false, webSearch: true, toolCalling: false },
+  'qwen3-omni-flash': { name: 'Qwen3 Omni Flash', vision: true, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen3-vl-235b-a22b': { name: 'Qwen3 VL 235B', vision: true, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen3-vl-32b': { name: 'Qwen3 VL 32B', vision: true, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen3-vl-30b-a3b': { name: 'Qwen3 VL 30B A3B', vision: true, reasoning: true, webSearch: false, toolCalling: false },
+  'qwen3-max-2026-01-23': { name: 'Qwen3 Max 2026-01-23', vision: true, reasoning: false, webSearch: true, toolCalling: false },
+  'qwen3-vl-plus': { name: 'Qwen3 VL Plus', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen3-coder-plus': { name: 'Qwen3 Coder Plus', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwq-32b': { name: 'QWQ 32B', vision: false, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen-web-dev': { name: 'Qwen Web Dev', vision: true, reasoning: false, webSearch: false, toolCalling: false },
+  'qwen-full-stack': { name: 'Qwen Full Stack', vision: true, reasoning: false, webSearch: false, toolCalling: false },
+  'qwen-cogview': { name: 'Qwen CogView', vision: false, reasoning: false, webSearch: false, toolCalling: false },
+  'qwen-max': { name: 'Qwen Max', vision: true, reasoning: true, webSearch: true, toolCalling: false },
+  'qwen-max-latest': { name: 'Qwen Max Latest', vision: true, reasoning: true, webSearch: true, toolCalling: false }
+};
+
 async function fetchModels() {
+  const apiKey = await checkAndRefreshToken();
+  
   return new Promise((resolve) => {
-    const apiKey = readApiKeyFromAuth();
-    
     const options = {
       hostname: 'qwen.aikit.club',
       port: 443,
@@ -84,12 +261,17 @@ async function fetchModels() {
           if (json.data && Array.isArray(json.data)) {
             const models = {};
             json.data.forEach(model => {
-              models[model.id] = { name: model.id };
+              const id = model.id;
+              const caps = MODEL_CAPABILITIES[id] || {};
+              models[id] = { 
+                name: caps.name || id,
+                limit: caps.vision ? { context: 262144, output: 32768 } : undefined
+              };
             });
-            log(`[Models] Fetched ${Object.keys(models).length} models`);
+            log(`[Models] Fetched ${Object.keys(models).length} models from API`);
             resolve(models);
           } else {
-            log(`[Models] Unexpected response: ${data.substring(0, 200)}`);
+            log(`[Models] Using default models`);
             resolve(getDefaultModels());
           }
         } catch (e) {
@@ -109,14 +291,15 @@ async function fetchModels() {
 }
 
 function getDefaultModels() {
-  return {
-    'qwen-max': { name: 'Qwen Max' },
-    'qwen2.5-max': { name: 'Qwen2.5 Max' },
-    'qwen2.5-plus': { name: 'Qwen2.5 Plus' },
-    'qwen2.5-turbo': { name: 'Qwen2.5 Turbo' },
-    'qwq-32b': { name: 'QWQ 32B' },
-    'qwen-deep-research': { name: 'Qwen Deep Research' }
-  };
+  const models = {};
+  Object.keys(MODEL_CAPABILITIES).forEach(id => {
+    const caps = MODEL_CAPABILITIES[id];
+    models[id] = { 
+      name: caps.name,
+      limit: caps.vision ? { context: 262144, output: 32768 } : undefined
+    };
+  });
+  return models;
 }
 
 let cachedModels = null;
@@ -136,7 +319,7 @@ function startProxy() {
       }
 
       if (req.method === 'GET' && req.url === '/health') {
-        const apiKey = readApiKeyFromAuth();
+        const apiKey = await checkAndRefreshToken();
         res.writeHead(200, { 'Content-Type': 'application/json' });
         res.end(JSON.stringify({ 
           status: 'ok', 
@@ -148,7 +331,7 @@ function startProxy() {
       }
 
       if (req.url.startsWith('/v1/')) {
-        const apiKey = readApiKeyFromAuth();
+        const apiKey = await checkAndRefreshToken();
 
         const options = {
           hostname: 'qwen.aikit.club',
@@ -198,11 +381,14 @@ function startProxy() {
 }
 
 export const QwenPlugin = async (ctx) => {
-  log('[Plugin] Starting Qwen Plugin...');
+  log('[Plugin] Starting Qwen Plugin v3.2.0...');
 
   await startProxy();
   
-  // Fetch models from API
+  log('[Plugin] Checking and refreshing token...');
+  await checkAndRefreshToken();
+  
+  log('[Plugin] Fetching models from API...');
   cachedModels = await fetchModels();
   log(`[Plugin] Loaded ${Object.keys(cachedModels).length} models`);
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "qwen-opencode-provider",
-  "version": "3.1.0",
+  "version": "3.2.1",
   "description": "OpenCode plugin for Qwen API - auto adds provider with 28+ models",
   "main": "index.js",
   "type": "module",