qwen-opencode-provider 3.0.0 → 3.2.0

package/index.js

@@ -2,7 +2,7 @@
  * OpenCode Qwen API Plugin
  * 
  * Uses local proxy (like Pollinations) to handle auth automatically.
- * No manual config needed!
+ * Auto-fetches models, validates and refreshes tokens.
  */
 
 import http from 'http';
@@ -16,27 +16,6 @@ const LOG_FILE = path.join(LOG_DIR, 'debug.log');
 const PROVIDER_ID = 'qwen';
 const API_BASE_URL = 'https://qwen.aikit.club/v1';
 
-const QWEN_MODELS = {
-  'qwen3-max': { name: 'Qwen3 Max', limit: { context: 262144, output: 32768 } },
-  'qwen3-max-2026-01-23': { name: 'Qwen3 Max', limit: { context: 262144, output: 32768 } },
-  'qwen3-vl-plus': { name: 'Qwen3 VL Plus', limit: { context: 262144, output: 32768 } },
-  'qwen3-coder-plus': { name: 'Qwen3 Coder Plus', limit: { context: 1048576, output: 65536 } },
-  'qwen3-vl-32b': { name: 'Qwen3 VL 32B', limit: { context: 131072, output: 32768 } },
-  'qwen3-vl-30b-a3b': { name: 'Qwen3 VL 30B A3B', limit: { context: 131072, output: 32768 } },
-  'qwen3-omni-flash': { name: 'Qwen3 Omni Flash', limit: { context: 65536, output: 13684 } },
-  'qwen3-30b-a3b': { name: 'Qwen3 30B A3B', limit: { context: 131072, output: 32768 } },
-  'qwen3-coder-30b-a3b-instruct': { name: 'Qwen3 Coder Flash', limit: { context: 262144, output: 65536 } },
-  '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' },
-  'qwen-web-dev': { name: 'Qwen Web Dev' },
-  'qwen-full-stack': { name: 'Qwen Full Stack' },
-  'qwen-cogview': { name: 'Qwen CogView' }
-};
-
 function ensureLogDir() {
   try {
     if (!fs.existsSync(LOG_DIR)) {
@@ -67,11 +46,7 @@ function readApiKeyFromAuth() {
       
       const providerAuth = auth[PROVIDER_ID];
       if (providerAuth) {
-        // Read BOTH key and apiKey fields - OpenCode can use either
-        const apiKey = providerAuth.key || providerAuth.apiKey || providerAuth.token || null;
-        if (apiKey) {
-          return apiKey;
-        }
+        return providerAuth.key || providerAuth.apiKey || providerAuth.token || null;
       }
     }
   } catch (e) {
@@ -80,6 +55,255 @@ 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: 'POST',
+      headers: {
+        'Host': 'qwen.aikit.club',
+        'Content-Type': 'application/json'
+      }
+    };
+
+    if (apiKey) {
+      options.headers['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] Response: ${JSON.stringify(json)}`);
+          resolve(json);
+        } 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(JSON.stringify({ token: apiKey }));
+  });
+}
+
+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: 'POST',
+      headers: {
+        'Host': 'qwen.aikit.club',
+        'Content-Type': 'application/json'
+      }
+    };
+
+    if (currentKey) {
+      options.headers['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] Response: ${JSON.stringify(json)}`);
+          
+          if (json.token) {
+            saveApiKeyToAuth(json.token);
+            resolve(json.token);
+          } 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 options = {
+      hostname: 'qwen.aikit.club',
+      port: 443,
+      path: '/v1/models',
+      method: 'GET',
+      headers: {
+        'Host': 'qwen.aikit.club',
+        'Content-Type': 'application/json'
+      }
+    };
+
+    if (apiKey) {
+      options.headers['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);
+          if (json.data && Array.isArray(json.data)) {
+            const models = {};
+            json.data.forEach(model => {
+              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 from API`);
+            resolve(models);
+          } else {
+            log(`[Models] Using default models`);
+            resolve(getDefaultModels());
+          }
+        } catch (e) {
+          log(`[Models] Parse error: ${e.message}`);
+          resolve(getDefaultModels());
+        }
+      });
+    });
+
+    req.on('error', (e) => {
+      log(`[Models] Error: ${e.message}`);
+      resolve(getDefaultModels());
+    });
+
+    req.end();
+  });
+}
+
+function getDefaultModels() {
+  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;
+let proxyPort = 0;
+
 function startProxy() {
   return new Promise((resolve) => {
     const server = http.createServer(async (req, res) => {
@@ -94,18 +318,19 @@ 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', 
           provider: 'qwen',
-          hasKey: !!apiKey
+          hasKey: !!apiKey,
+          modelsCount: cachedModels ? Object.keys(cachedModels).length : 0
         }));
         return;
       }
 
       if (req.url.startsWith('/v1/')) {
-        const apiKey = readApiKeyFromAuth();
+        const apiKey = await checkAndRefreshToken();
 
         const options = {
           hostname: 'qwen.aikit.club',
@@ -142,9 +367,9 @@ function startProxy() {
     });
 
     server.listen(0, '127.0.0.1', () => {
-      const port = server.address().port;
-      log(`[Proxy] Started on port ${port}`);
-      resolve(port);
+      proxyPort = server.address().port;
+      log(`[Proxy] Started on port ${proxyPort}`);
+      resolve(proxyPort);
     });
 
     server.on('error', (e) => {
@@ -155,10 +380,18 @@ function startProxy() {
 }
 
 export const QwenPlugin = async (ctx) => {
-  log('[Plugin] Starting Qwen Plugin...');
+  log('[Plugin] Starting Qwen Plugin v3.2.0...');
+
+  await startProxy();
+  
+  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`);
 
-  const port = await startProxy();
-  const localBaseUrl = `http://127.0.0.1:${port}/v1`;
+  const localBaseUrl = `http://127.0.0.1:${proxyPort}/v1`;
 
   return {
     config: async (config) => {
@@ -170,10 +403,10 @@ export const QwenPlugin = async (ctx) => {
         id: PROVIDER_ID,
         name: 'Qwen AI',
         options: { baseURL: localBaseUrl },
-        models: QWEN_MODELS
+        models: cachedModels
       };
 
-      log(`[Hook] Registered provider with ${Object.keys(QWEN_MODELS).length} models`);
+      log(`[Hook] Registered provider with ${Object.keys(cachedModels).length} models`);
     }
   };
 };

package/package.json

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