@pedrofariasx/qwenproxy 1.6.0 → 1.6.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pedrofariasx/qwenproxy",
-  "version": "1.6.0",
+  "version": "1.6.2",
   "description": "Local OpenAI-compatible proxy API that routes requests to Qwen (chat.qwen.ai) via Playwright browser automation.",
   "main": "index.js",
   "scripts": {

package/src/core/accounts.ts

@@ -74,9 +74,8 @@ export function listAccounts(): QwenAccount[] {
 }
 
 export function getAccountCredentials(id: string): QwenAccount | undefined {
-  const db = getDatabase()
-  const row = db.prepare('SELECT id, email, password, cooldown_until, cooldown_reason FROM accounts WHERE id = ?').get(id)
-  return row as QwenAccount | undefined
+  const cached = getCachedAccounts()
+  return cached.find(a => a.id === id)
 }
 
 export function updateAccountCooldown(id: string, cooldownUntil: number, reason: string | null): void {

package/src/routes/chat.ts

@@ -566,12 +566,12 @@ export async function chatCompletions(c: Context) {
         const createdTimestamp = Math.floor(Date.now() / 1000);
 
         const fastWriteContent = (content: string) => {
-          const escaped = content.replace(/\\/g, '\\\\').replace(/"/g, '\\"').replace(/\n/g, '\\n').replace(/\r/g, '\\r').replace(/\t/g, '\\t');
+          const escaped = JSON.stringify(content).slice(1, -1);
           streamWriter.write(`data: {"id":"${completionId}","object":"chat.completion.chunk","created":${createdTimestamp},"model":"${body.model}","choices":[{"index":0,"delta":{"content":"${escaped}"},"logprobs":null,"finish_reason":null}]}\n\n`);
         };
 
         const fastWriteReasoning = (content: string) => {
-          const escaped = content.replace(/\\/g, '\\\\').replace(/"/g, '\\"').replace(/\n/g, '\\n').replace(/\r/g, '\\r').replace(/\t/g, '\\t');
+          const escaped = JSON.stringify(content).slice(1, -1);
           streamWriter.write(`data: {"id":"${completionId}","object":"chat.completion.chunk","created":${createdTimestamp},"model":"${body.model}","choices":[{"index":0,"delta":{"reasoning_content":"${escaped}"},"logprobs":null,"finish_reason":null}]}\n\n`);
         };
 

package/src/services/playwright.ts

@@ -562,21 +562,23 @@ export async function getGuestHeaders(): Promise<Record<string, string>> {
         await guestPage!.focus(inputSelector);
         await guestPage!.fill(inputSelector, '');
         await guestPage!.type(inputSelector, 'a', { delay: 50 });
-        await sleep(1000);
+        await sleep(1500);
         
+        // Try pressing Enter first as it is highly reliable
+        await guestPage!.focus(inputSelector);
+        await guestPage!.keyboard.press('Enter');
+
         const selectors = ['.message-input-right-button-send .send-button', '.chat-prompt-send-button', 'button.send-button'];
-        let clicked = false;
         for (const selector of selectors) {
-          const btn = await guestPage!.$(selector);
-          if (btn && await btn.isVisible()) {
-            await btn.click({ force: true, delay: 50 }).catch(() => {});
-            clicked = true;
-            break;
+          try {
+            const btn = await guestPage!.$(selector);
+            if (btn && await btn.isVisible()) {
+              await btn.click({ force: true, delay: 50 }).catch(() => {});
+            }
+          } catch (e) {
+            // ignore click errors
           }
         }
-        if (!clicked) {
-          await guestPage!.keyboard.press('Enter');
-        }
       } catch (e) {
         clearTimeout(timeout);
         reject(e);
@@ -825,47 +827,45 @@ async function _getQwenHeadersInternal(forceNew = false, accountId?: string): Pr
       console.log(`[Playwright] Triggering request for ${cacheKey}...`);
       const inputSelector = 'textarea:visible, [contenteditable="true"]:visible';
 
-      await page.focus(inputSelector);
-      await page.fill(inputSelector, '');
-      await page.type(inputSelector, 'a', { delay: 100 });
-      console.log(`[Playwright] Typed char for ${cacheKey}, waiting for UI to update...`);
-      await sleep(2000);
-
-      const selectors = [
-        '.message-input-right-button-send .send-button',
-        '.chat-prompt-send-button',
-        'button.send-button'
-      ];
+      try {
+        await page.focus(inputSelector);
+        await page.fill(inputSelector, '');
+        await page.type(inputSelector, 'a', { delay: 50 });
+        console.log(`[Playwright] Typed char for ${cacheKey}, waiting for UI to update...`);
+        await sleep(1500);
 
-      let clicked = false;
-      for (const selector of selectors) {
-        try {
-          const btn = await page.$(selector);
-          if (btn && await btn.isVisible()) {
-            console.log(`[Playwright] Attempting click on: ${selector}`);
-
-            await page.evaluate((sel) => {
-              const element = document.querySelector(sel) as HTMLElement;
-              if (element) {
-                element.focus();
-                element.click();
-              }
-            }, selector);
+        // Try pressing Enter first on the input field as it is highly reliable
+        console.log(`[Playwright] Pressing Enter for ${cacheKey}...`);
+        await page.focus(inputSelector);
+        await page.keyboard.press('Enter');
 
-            await btn.click({ force: true, delay: 50 }).catch(() => {});
+        // Also attempt to click the send button in case Enter didn't submit
+        const selectors = [
+          '.message-input-right-button-send .send-button',
+          '.chat-prompt-send-button',
+          'button.send-button'
+        ];
 
-            clicked = true;
-            break;
+        for (const selector of selectors) {
+          try {
+            const btn = await page.$(selector);
+            if (btn && await btn.isVisible()) {
+              console.log(`[Playwright] Also attempting click on: ${selector}`);
+              await page.evaluate((sel) => {
+                const element = document.querySelector(sel) as HTMLElement;
+                if (element) {
+                  element.focus();
+                  element.click();
+                }
+              }, selector);
+              await btn.click({ force: true, delay: 50 }).catch(() => {});
+            }
+          } catch (e) {
+            // ignore click errors
           }
-        } catch (e) {
-          console.error(`[Playwright] Error clicking ${selector} for ${cacheKey}:`, e);
         }
-      }
-
-      if (!clicked) {
-        console.log(`[Playwright] No send button found/clicked for ${cacheKey}, fallback to Enter...`);
-        await page.focus(inputSelector);
-        await page.keyboard.press('Enter');
+      } catch (triggerErr: any) {
+        console.error(`[Playwright] Failed to trigger request for ${cacheKey}:`, triggerErr.message);
       }
     });
   });

package/src/services/qwen.ts

@@ -86,6 +86,7 @@ const refillPromises: Map<string, Promise<void>> = new Map();
 
 const WARM_POOL_SIZE = 10;
 const WARM_POOL_TTL_MS = 10 * 60 * 1000;
+const WARM_POOL_LOW_WATER = 3;
 
 function cleanupStalePool(accountId: string) {
   const pool = warmPool.get(accountId);
@@ -243,6 +244,11 @@ export async function getWarmedChat(accountId?: string) {
   let pool = warmPool.get(key);
   if (!pool) { pool = []; warmPool.set(key, pool); }
   cleanupStalePool(key);
+
+  if (pool.length < WARM_POOL_LOW_WATER && !refillPromises.has(key)) {
+    refillPromises.set(key, refillPoolForAccount(key).finally(() => refillPromises.delete(key)));
+  }
+
   if (pool.length === 0) {
     if (!refillPromises.has(key)) {
       refillPromises.set(key, refillPoolForAccount(key).finally(() => refillPromises.delete(key)));