Update worker.js

worker.js

@@ -161,9 +161,9 @@ function testBasicConfiguration() {
     return { passed, failed };
 }
 
-// 新增：测试函数 - 网络连接性
+// 修改：测试函数 - 网络连接性（使用单个浏览器实例）
 async function testNetworkConnectivity() {
-    log('info', '🌐 开始网络连接性测试...');
+    log('info', '🌐 开始网络连接性测试（使用单个浏览器实例）...');
     
     const testUrls = [
         { name: '主站', url: 'https://toolbaz.com' },
@@ -173,40 +173,74 @@ async function testNetworkConnectivity() {
     ];
     
     const results = [];
+    let testBrowser = null;
+    let testPage = null;
     
-    for (const test of testUrls) {
-        try {
-            const startTime = Date.now();
-            const tempBrowser = await puppeteer.launch({ headless: true });
-            const tempPage = await tempBrowser.newPage();
-            
-            await tempPage.goto(test.url, { 
-                waitUntil: 'networkidle2', 
-                timeout: 15000 
-            });
-            
-            const responseTime = Date.now() - startTime;
-            const title = await tempPage.title();
-            
-            results.push({
-                name: test.name,
-                url: test.url,
-                status: 'success',
-                responseTime,
-                title: title.substring(0, 50)
-            });
-            
-            await tempBrowser.close();
-            log('verbose', `✅ ${test.name}: ${responseTime}ms - ${title.substring(0, 30)}...`);
-            
-        } catch (error) {
-            results.push({
-                name: test.name,
-                url: test.url,
-                status: 'failed',
-                error: error.message
-            });
-            log('error', `❌ ${test.name}: ${error.message}`);
+    try {
+        // 启动单个浏览器实例
+        testBrowser = await puppeteer.launch({ 
+            headless: true,
+            args: [
+                '--no-sandbox', 
+                '--disable-setuid-sandbox',
+                '--disable-dev-shm-usage',
+                '--disable-gpu'
+            ]
+        });
+        testPage = await testBrowser.newPage();
+        
+        // 设置更长的超时时间以应对网络问题
+        testPage.setDefaultTimeout(30000);
+        
+        for (const test of testUrls) {
+            try {
+                const startTime = Date.now();
+                
+                // 重用同一个页面进行测试
+                await testPage.goto(test.url, { 
+                    waitUntil: 'networkidle2', 
+                    timeout: 25000 // 增加到25秒
+                });
+                
+                const responseTime = Date.now() - startTime;
+                const title = await testPage.title();
+                
+                results.push({
+                    name: test.name,
+                    url: test.url,
+                    status: 'success',
+                    responseTime,
+                    title: title.substring(0, 50)
+                });
+                
+                log('verbose', `✅ ${test.name}: ${responseTime}ms - ${title.substring(0, 30)}...`);
+                
+                // 在测试之间添加短暂延迟，避免过于频繁的请求
+                await new Promise(resolve => setTimeout(resolve, 1000));
+                
+            } catch (error) {
+                results.push({
+                    name: test.name,
+                    url: test.url,
+                    status: 'failed',
+                    error: error.message
+                });
+                log('error', `❌ ${test.name}: ${error.message}`);
+                
+                // 错误后也短暂延迟
+                await new Promise(resolve => setTimeout(resolve, 500));
+            }
+        }
+        
+    } catch (error) {
+        log('error', `❌ 网络测试初始化失败: ${error.message}`);
+    } finally {
+        // 清理资源
+        if (testPage) {
+            await testPage.close().catch(e => log('warn', `页面关闭失败: ${e.message}`));
+        }
+        if (testBrowser) {
+            await testBrowser.close().catch(e => log('warn', `浏览器关闭失败: ${e.message}`));
         }
     }
     
@@ -261,19 +295,22 @@ async function testPromptPrefixFiltering() {
     return { results, passed: passCount, failed: results.length - passCount };
 }
 
-// 新增：测试函数 - 浏览器环境
+// 修改：测试函数 - 浏览器环境（使用全局浏览器实例）
 async function testBrowserEnvironment() {
     log('info', '🌍 开始浏览器环境测试...');
     
+    let testPage = null;
+    
     try {
-        const testBrowser = await puppeteer.launch({ 
-            headless: true,
-            args: ['--no-sandbox', '--disable-setuid-sandbox']
-        });
-        const testPage = await testBrowser.newPage();
+        // 尝试使用现有的浏览器实例，如果没有则创建
+        const testBrowser = browser || await initBrowser();
+        testPage = await testBrowser.newPage();
+        
+        // 设置更长的超时时间
+        testPage.setDefaultTimeout(20000);
         
         // 测试基本功能
-        await testPage.goto('https://example.com', { waitUntil: 'networkidle2' });
+        await testPage.goto('https://example.com', { waitUntil: 'networkidle2', timeout: 15000 });
         
         const browserInfo = await testPage.evaluate(() => {
             return {
@@ -301,8 +338,6 @@ async function testBrowserEnvironment() {
             }
         });
         
-        await testBrowser.close();
-        
         log('verbose', '✅ 浏览器环境测试通过');
         log('verbose', '📋 浏览器信息:', browserInfo);
         log('verbose', '📋 JavaScript功能:', jsTest);
@@ -312,117 +347,132 @@ async function testBrowserEnvironment() {
     } catch (error) {
         log('error', `❌ 浏览器环境测试失败: ${error.message}`);
         return { success: false, error: error.message };
+    } finally {
+        // 清理页面资源
+        if (testPage) {
+            await testPage.close().catch(e => log('warn', `页面关闭失败: ${e.message}`));
+        }
     }
 }
 
-// 新增：测试函数 - 深度会话检测
+// 修改：测试函数 - 深度会话检测（使用全局浏览器实例）
 async function testSessionDetection() {
     log('info', '🔍 开始深度会话检测测试...');
     
-    const tempBrowser = await puppeteer.launch({ headless: true });
-    const tempPage = await tempBrowser.newPage();
+    let testPage = null;
     
     try {
-        await tempPage.goto('https://toolbaz.com/writer/ai-writer', { 
+        // 使用现有的浏览器实例或创建新的
+        const testBrowser = browser || await initBrowser();
+        testPage = await testBrowser.newPage();
+        
+        // 设置更长的超时时间
+        testPage.setDefaultTimeout(30000);
+        
+        await testPage.goto('https://toolbaz.com/writer/ai-writer', { 
             waitUntil: 'networkidle2', 
             timeout: 30000 
         });
         
-        // 增强的会话检测
-        const sessionAnalysis = await tempPage.evaluate(() => {
+        // 简化的会话检测（减少资源消耗）
+        const sessionAnalysis = await testPage.evaluate(() => {
             const analysis = {
                 timestamp: Date.now(),
                 url: window.location.href,
                 title: document.title,
                 elementsFound: [],
-                forms: [],
                 hiddenInputs: [],
-                scripts: [],
                 cookies: {},
                 sessionStorage: {},
-                localStorage: {},
-                networkRequests: [],
-                potentialSessions: []
+                localStorage: {}
             };
             
-            // 分析所有元素
-            const allElements = document.querySelectorAll('*');
+            // 专注于关键元素，避免过度遍历
             const sessionKeywords = ['session', 'sess', 'sid', 'token', 'csrf', 'auth'];
             
-            allElements.forEach(elem => {
-                const info = {
-                    tag: elem.tagName,
-                    id: elem.id || '',
-                    name: elem.name || '',
-                    className: elem.className || '',
-                    type: elem.type || '',
-                    value: elem.value || ''
-                };
-                
-                const searchText = [info.id, info.name, info.className, info.value].join(' ').toLowerCase();
-                
-                if (sessionKeywords.some(keyword => searchText.includes(keyword))) {
-                    analysis.elementsFound.push(info);
+            // 只检查可能包含会话信息的元素
+            const relevantSelectors = [
+                'input[type="hidden"]',
+                'input[name*="session"]',
+                'input[name*="token"]',
+                'input[id*="session"]',
+                'input[id*="token"]',
+                '[data-session]',
+                '[data-token]'
+            ];
+            
+            relevantSelectors.forEach(selector => {
+                try {
+                    const elements = document.querySelectorAll(selector);
+                    elements.forEach(elem => {
+                        const info = {
+                            tag: elem.tagName,
+                            id: elem.id || '',
+                            name: elem.name || '',
+                            value: elem.value || ''
+                        };
+                        
+                        const searchText = [info.id, info.name, info.value].join(' ').toLowerCase();
+                        
+                        if (sessionKeywords.some(keyword => searchText.includes(keyword))) {
+                            analysis.elementsFound.push(info);
+                        }
+                    });
+                } catch (e) {
+                    // 忽略选择器错误
                 }
             });
             
-            // 分析表单
-            const forms = document.querySelectorAll('form');
-            forms.forEach(form => {
-                analysis.forms.push({
-                    action: form.action,
-                    method: form.method,
-                    id: form.id,
-                    className: form.className
-                });
-            });
-            
             // 分析隐藏输入
             const hiddenInputs = document.querySelectorAll('input[type="hidden"]');
             hiddenInputs.forEach(input => {
-                analysis.hiddenInputs.push({
-                    name: input.name,
-                    id: input.id,
-                    value: input.value.substring(0, 100) // 限制长度
-                });
-            });
-            
-            // 分析脚本
-            const scripts = document.querySelectorAll('script');
-            scripts.forEach(script => {
-                if (script.src) {
-                    analysis.scripts.push(script.src);
+                if (input.name || input.value) {
+                    analysis.hiddenInputs.push({
+                        name: input.name,
+                        id: input.id,
+                        value: input.value.substring(0, 100)
+                    });
                 }
             });
             
-            // 分析存储
-            for (let i = 0; i < sessionStorage.length; i++) {
-                const key = sessionStorage.key(i);
-                analysis.sessionStorage[key] = sessionStorage.getItem(key);
+            // 简化存储分析
+            try {
+                for (let i = 0; i < Math.min(sessionStorage.length, 10); i++) {
+                    const key = sessionStorage.key(i);
+                    if (key && sessionKeywords.some(k => key.toLowerCase().includes(k))) {
+                        analysis.sessionStorage[key] = sessionStorage.getItem(key);
+                    }
+                }
+                
+                for (let i = 0; i < Math.min(localStorage.length, 10); i++) {
+                    const key = localStorage.key(i);
+                    if (key && sessionKeywords.some(k => key.toLowerCase().includes(k))) {
+                        analysis.localStorage[key] = localStorage.getItem(key);
+                    }
+                }
+            } catch (e) {
+                // 忽略存储访问错误
             }
             
-            for (let i = 0; i < localStorage.length; i++) {
-                const key = localStorage.key(i);
-                analysis.localStorage[key] = localStorage.getItem(key);
+            // 简化cookie分析
+            try {
+                document.cookie.split(';').forEach(cookie => {
+                    const [name, value] = cookie.trim().split('=');
+                    if (name && value && sessionKeywords.some(k => name.toLowerCase().includes(k))) {
+                        analysis.cookies[name] = value;
+                    }
+                });
+            } catch (e) {
+                // 忽略cookie访问错误
             }
             
-            // 分析cookies
-            document.cookie.split(';').forEach(cookie => {
-                const [name, value] = cookie.trim().split('=');
-                if (name && value) {
-                    analysis.cookies[name] = value;
-                }
-            });
-            
             return analysis;
         });
         
         log('verbose', `📊 会话分析结果:`);
         log('verbose', `  - 页面: ${sessionAnalysis.title}`);
         log('verbose', `  - 相关元素: ${sessionAnalysis.elementsFound.length}`);
-        log('verbose', `  - 表单数量: ${sessionAnalysis.forms.length}`);
         log('verbose', `  - 隐藏输入: ${sessionAnalysis.hiddenInputs.length}`);
-        log('verbose', `  - 脚本数量: ${sessionAnalysis.scripts.length}`);
         log('verbose', `  - Cookie数量: ${Object.keys(sessionAnalysis.cookies).length}`);
         log('verbose', `  - SessionStorage: ${Object.keys(sessionAnalysis.sessionStorage).length}`);
         log('verbose', `  - LocalStorage: ${Object.keys(sessionAnalysis.localStorage).length}`);
@@ -430,48 +480,63 @@ async function testSessionDetection() {
         if (sessionAnalysis.elementsFound.length > 0) {
             log('info', '🎯 找到潜在的会话相关元素:');
             sessionAnalysis.elementsFound.forEach(elem => {
-                log('verbose', `  - ${elem.tag}#${elem.id}.${elem.className} (${elem.type}): ${elem.value.substring(0, 50)}`);
+                log('verbose', `  - ${elem.tag}#${elem.id} (${elem.name}): ${elem.value.substring(0, 50)}`);
             });
         }
         
-        await tempBrowser.close();
         return { success: true, analysis: sessionAnalysis };
         
     } catch (error) {
         log('error', `❌ 深度会话检测失败: ${error.message}`);
-        await tempBrowser.close();
         return { success: false, error: error.message };
+    } finally {
+        // 清理页面资源
+        if (testPage) {
+            await testPage.close().catch(e => log('warn', `页面关闭失败: ${e.message}`));
+        }
     }
 }
 
-// 新增：测试函数 - AJAX请求模拟
+// 修改：测试函数 - AJAX请求模拟（使用全局浏览器实例）
 async function testAjaxRequest() {
     log('info', '📡 开始AJAX请求模拟测试...');
     
-    const tempBrowser = await puppeteer.launch({ headless: true });
-    const tempPage = await tempBrowser.newPage();
+    let testPage = null;
     
     try {
-        await tempPage.goto('https://toolbaz.com/writer/ai-writer', { 
+        // 使用现有的浏览器实例或创建新的
+        const testBrowser = browser || await initBrowser();
+        testPage = await testBrowser.newPage();
+        
+        // 设置更长的超时时间
+        testPage.setDefaultTimeout(30000);
+        
+        await testPage.goto('https://toolbaz.com/writer/ai-writer', { 
             waitUntil: 'networkidle2', 
             timeout: 30000 
         });
         
-        // 设置请求监听
+        // 简化的请求监听
         const networkRequests = [];
-        await tempPage.setRequestInterception(true);
-        tempPage.on('request', request => {
-            networkRequests.push({
-                url: request.url(),
-                method: request.method(),
-                headers: request.headers(),
-                postData: request.postData()
+        try {
+            await testPage.setRequestInterception(true);
+            testPage.on('request', request => {
+                if (request.url().includes('writing.php')) {
+                    networkRequests.push({
+                        url: request.url(),
+                        method: request.method(),
+                        headers: request.headers(),
+                        postData: request.postData()
+                    });
+                }
+                request.continue();
             });
-            request.continue();
-        });
+        } catch (e) {
+            log('warn', `请求拦截设置失败，继续测试: ${e.message}`);
+        }
         
         // 测试AJAX请求
-        const ajaxTest = await tempPage.evaluate(async () => {
+        const ajaxTest = await testPage.evaluate(async () => {
             return new Promise((resolve) => {
                 const testData = {
                     text: 'Hello, this is a test message',
@@ -485,7 +550,9 @@ async function testAjaxRequest() {
                     method: 'POST',
                     headers: {
                         'Content-Type': 'application/x-www-form-urlencoded',
-                        'Accept': '*/*'
+                        'Accept': '*/*',
+                        'Origin': 'https://toolbaz.com',
+                        'Referer': 'https://toolbaz.com/writer/ai-writer'
                     },
                     body: new URLSearchParams(testData).toString()
                 })
@@ -520,13 +587,16 @@ async function testAjaxRequest() {
         
         log('verbose', `  - 网络请求数量: ${networkRequests.length}`);
         
-        await tempBrowser.close();
         return { success: ajaxTest.success, data: ajaxTest, networkRequests };
         
     } catch (error) {
         log('error', `❌ AJAX测试失败: ${error.message}`);
-        await tempBrowser.close();
         return { success: false, error: error.message };
+    } finally {
+        // 清理页面资源
+        if (testPage) {
+            await testPage.close().catch(e => log('warn', `页面关闭失败: ${e.message}`));
+        }
     }
 }
 
@@ -575,9 +645,9 @@ function showPerformanceStats() {
     log('info', `  - 平均响应时间: ${PERFORMANCE_STATS.averageResponseTime}ms`);
 }
 
-// 新增：综合测试函数
+// 修改：综合测试函数（优化浏览器资源管理）
 async function runComprehensiveTests() {
-    log('info', '🧪 开始综合测试套件...');
+    log('info', '🧪 开始综合测试套件（优化版）...');
     
     const results = {
         configuration: null,
@@ -588,8 +658,10 @@ async function runComprehensiveTests() {
         overall: { passed: 0, failed: 0 }
     };
     
+    let sharedBrowser = null;
+    
     try {
-        // 1. 基本配置测试
+        // 1. 基本配置测试（不需要浏览器）
         results.configuration = testBasicConfiguration();
         if (results.configuration.failed === 0) {
             results.overall.passed++;
@@ -597,7 +669,7 @@ async function runComprehensiveTests() {
             results.overall.failed++;
         }
         
-        // 2. 网络连接性测试
+        // 2. 网络连接性测试（已优化为使用单个浏览器）
         results.network = await testNetworkConnectivity();
         if (results.network.filter(r => r.status === 'success').length >= 3) {
             results.overall.passed++;
@@ -605,7 +677,7 @@ async function runComprehensiveTests() {
             results.overall.failed++;
         }
         
-        // 3. 浏览器环境测试
+        // 3. 浏览器环境测试（使用全局浏览器）
         results.browser = await testBrowserEnvironment();
         if (results.browser.success) {
             results.overall.passed++;
@@ -613,7 +685,7 @@ async function runComprehensiveTests() {
             results.overall.failed++;
         }
         
-        // 4. 会话检测测试
+        // 4. 会话检测测试（使用全局浏览器）
         results.session = await testSessionDetection();
         if (results.session.success) {
             results.overall.passed++;
@@ -621,7 +693,7 @@ async function runComprehensiveTests() {
             results.overall.failed++;
         }
         
-        // 5. AJAX请求测试
+        // 5. AJAX请求测试（使用全局浏览器）
         results.ajax = await testAjaxRequest();
         if (results.ajax.success) {
             results.overall.passed++;
@@ -629,7 +701,7 @@ async function runComprehensiveTests() {
             results.overall.failed++;
         }
         
-        // 6. 前缀过滤测试
+        // 6. 前缀过滤测试（不需要浏览器）
         results.prefixFiltering = await testPromptPrefixFiltering();
         if (results.prefixFiltering.failed === 0) {
             results.overall.passed++;
@@ -643,10 +715,23 @@ async function runComprehensiveTests() {
             log('warn', '⚠️ 部分测试失败，请检查相关配置');
         }
         
+        // 添加资源使用统计
+        const browserInstance = browser || sharedBrowser;
+        if (browserInstance) {
+            const pages = await browserInstance.pages().catch(() => []);
+            log('info', `📊 浏览器资源统计: ${pages.length} 个页面`);
+        }
+        
         return results;
         
     } catch (error) {
         log('error', `❌ 综合测试失败: ${error.message}`);
+        
+        // 确保清理共享浏览器资源
+        if (sharedBrowser) {
+            await sharedBrowser.close().catch(e => log('warn', `共享浏览器关闭失败: ${e.message}`));
+        }
+        
         return { ...results, error };
     }
 }
@@ -1875,69 +1960,413 @@ async function handleImageGenerations(req, res) {
     });
 }
 
+// 模型缓存（避免频繁重新提取）
+let modelCache = {
+    data: null,
+    timestamp: null,
+    expiresAt: null,
+    isValid: function() {
+        return this.data && this.expiresAt && Date.now() < this.expiresAt;
+    },
+    set: function(models, ttlMinutes = 60) {
+        this.data = models;
+        this.timestamp = Date.now();
+        this.expiresAt = Date.now() + (ttlMinutes * 60 * 1000);
+        log('info', `🔐 模型列表已缓存，有效期 ${ttlMinutes} 分钟`);
+    },
+    clear: function() {
+        this.data = null;
+        this.timestamp = null;
+        this.expiresAt = null;
+        log('info', '🔐 模型缓存已清空');
+    }
+};
+
 async function handleModels(req, res) {
     try {
         log('info', '🔍 开始提取网站模型列表...');
-        const browser = await initBrowser();
-        const page = await browser.newPage();
         
-        await page.goto('https://toolbaz.com/writer/ai-writer', { waitUntil: 'networkidle2', timeout: 30000 });
+        // 检查缓存
+        if (modelCache.isValid()) {
+            log('info', `✅ 使用缓存模型列表 (${modelCache.data.length} 个模型)`);
+            res.writeHead(200, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
+            res.end(JSON.stringify({
+                object: 'list',
+                data: modelCache.data
+            }));
+            return;
+        }
         
-        // 等待并点击模型选择器
-        await page.waitForSelector('#selected-model', { timeout: 10000 });
-        await page.click('#selected-model');
+        // 尝试多种方法获取模型列表
+        let models = null;
+        let extractionMethod = '';
         
-        // 等待下拉列表出现
-        await page.waitForSelector('.model-dropdown .model-option', { timeout: 5000 });
+        // 方法1: 尝试访问网站获取最新模型
+        try {
+            log('info', '🌐 方法1: 从网站提取模型列表...');
+            models = await extractModelsFromWebsite();
+            extractionMethod = 'website';
+        } catch (websiteError) {
+            log('warn', `⚠️ 网站提取失败: ${websiteError.message}`);
+            
+            // 方法2: 尝试AJAX方式获取模型
+            try {
+                log('info', '📡 方法2: 尝试AJAX方式获取模型...');
+                models = await extractModelsViaAjax();
+                extractionMethod = 'ajax';
+            } catch (ajaxError) {
+                log('warn', `⚠️ AJAX提取失败: ${ajaxError.message}`);
+                
+                // 方法3: 使用增强的静态配置
+                try {
+                    log('info', '🔧 方法3: 使用增强静态配置...');
+                    models = await getEnhancedStaticModels();
+                    extractionMethod = 'enhanced-static';
+                } catch (staticError) {
+                    log('error', `❌ 所有方法都失败了，使用基础静态配置: ${staticError.message}`);
+                    models = getBasicStaticModels();
+                    extractionMethod = 'basic-static';
+                }
+            }
+        }
+        
+        // 更新缓存
+        if (models && models.length > 0) {
+            modelCache.set(models, 60); // 缓存60分钟
+            
+            log('info', `✅ 成功提取 ${models.length} 个模型 (方法: ${extractionMethod})`);
+            
+            // 记录提取到的模型（调试用）
+            if (CONFIG.DEBUG_LEVEL === 'verbose') {
+                models.forEach((model, index) => {
+                    log('verbose', `  ${index + 1}. ${model.name} (${model.id})`);
+                });
+            }
+            
+            res.writeHead(200, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
+            res.end(JSON.stringify({
+                object: 'list',
+                data: models,
+                metadata: {
+                    extraction_method: extractionMethod,
+                    extracted_at: new Date().toISOString(),
+                    total_models: models.length
+                }
+            }));
+        } else {
+            throw new Error('没有获取到任何模型数据');
+        }
         
-        // 提取所有模型选项
+    } catch (error) {
+        log('error', `❌ 模型提取完全失败: ${error.message}`);
+        
+        // 最后的兜底策略
+        const fallbackModels = getBasicStaticModels();
+        
+        res.writeHead(200, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
+        res.end(JSON.stringify({
+            object: 'list',
+            data: fallbackModels,
+            metadata: {
+                extraction_method: 'fallback',
+                error: error.message,
+                extracted_at: new Date().toISOString()
+            }
+        }));
+    }
+}
+
+// 新增：从网站提取模型的专用函数
+async function extractModelsFromWebsite() {
+    const browser = await initBrowser();
+    const page = await browser.newPage();
+    
+    try {
+        // 设置更长的超时时间
+        page.setDefaultTimeout(30000);
+        
+        log('info', '🌐 访问写作页面...');
+        await page.goto('https://toolbaz.com/writer/ai-writer', { 
+            waitUntil: 'networkidle2', 
+            timeout: 30000 
+        });
+        
+        // 等待页面加载
+        await page.waitForTimeout(2000); // 给页面一些加载时间
+        
+        log('info', '🔍 查找模型选择器...');
+        
+        // 尝试多种选择器
+        const modelSelectors = [
+            '#selected-model',
+            '.model-selector',
+            '[data-model]',
+            '.model-dropdown-trigger'
+        ];
+        
+        let selectorFound = null;
+        for (const selector of modelSelectors) {
+            try {
+                await page.waitForSelector(selector, { timeout: 5000 });
+                selectorFound = selector;
+                log('info', `✅ 找到模型选择器: ${selector}`);
+                break;
+            } catch (e) {
+                continue;
+            }
+        }
+        
+        if (!selectorFound) {
+            throw new Error('未找到模型选择器');
+        }
+        
+        // 尝试点击选择器
+        try {
+            await page.click(selectorFound);
+            log('info', '✅ 成功点击模型选择器');
+        } catch (e) {
+            log('warn', `⚠️ 点击选择器失败，尝试直接查找模型: ${e.message}`);
+        }
+        
+        // 等待下拉列表或模型列表出现
+        await page.waitForTimeout(1000);
+        
+        // 查找模型选项
         const models = await page.evaluate(() => {
-            const modelOptions = document.querySelectorAll('.model-dropdown .model-option');
             const modelList = [];
             
-            modelOptions.forEach(option => {
-                const modelNameElement = option.querySelector('.model-name');
-                const modelIconElement = option.querySelector('.model-icon');
-                
-                if (modelNameElement) {
-                    const name = modelNameElement.innerText.trim();
-                    const icon = modelIconElement ? modelIconElement.src : '';
-                    modelList.push({
-                        id: name.toLowerCase().replace(/[^a-z0-9]/g, '-'),
-                        name: name,
-                        icon: icon,
-                        object: 'model',
-                        created: Math.floor(Date.now() / 1000),
-                        owned_by: 'toolbaz'
-                    });
+            // 尝试多种模型选项选择器
+            const modelSelectors = [
+                '.model-dropdown .model-option',
+                '.model-list .model-item',
+                '[data-model-option]',
+                '.model-selector option',
+                'select[name="model"] option',
+                '.ai-models .model',
+                '.available-models .model-item'
+            ];
+            
+            for (const selector of modelSelectors) {
+                try {
+                    const elements = document.querySelectorAll(selector);
+                    if (elements.length > 0) {
+                        log(`✅ 使用选择器 ${selector} 找到 ${elements.length} 个模型选项`);
+                        
+                        elements.forEach((element, index) => {
+                            let name = '';
+                            let id = '';
+                            
+                            // 尝试获取模型名称
+                            const nameElement = element.querySelector('.model-name') || 
+                                             element.querySelector('.name') || 
+                                             element.querySelector('span') ||
+                                             element;
+                            
+                            name = nameElement.innerText || nameElement.textContent || 
+                                   nameElement.value || nameElement.getAttribute('data-model') || '';
+                            
+                            name = name.trim();
+                            
+                            if (name && name.length > 0) {
+                                id = name.toLowerCase()
+                                     .replace(/[^a-z0-9\s]/g, '')
+                                     .replace(/\s+/g, '-')
+                                     .substring(0, 50);
+                                
+                                modelList.push({
+                                    id: id || `model-${index}`,
+                                    name: name,
+                                    icon: element.querySelector('.model-icon, .icon')?.src || '',
+                                    object: 'model',
+                                    created: Math.floor(Date.now() / 1000),
+                                    owned_by: 'toolbaz'
+                                });
+                            }
+                        });
+                        
+                        // 如果找到了模型，就不再尝试其他选择器
+                        if (modelList.length > 0) {
+                            break;
+                        }
+                    }
+                } catch (e) {
+                    continue;
                 }
-            });
+            }
             
             return modelList;
         });
         
-        await page.close();
+        if (models.length === 0) {
+            throw new Error('页面中没有找到模型选项');
+        }
         
-        log('info', `✅ 成功提取 ${models.length} 个模型`);
+        return models;
         
-        res.writeHead(200, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
-        res.end(JSON.stringify({
-            object: 'list',
-            data: models
-        }));
+    } finally {
+        await page.close().catch(e => log('warn', `页面关闭失败: ${e.message}`));
+    }
+}
+
+// 新增：通过AJAX方式获取模型
+async function extractModelsViaAjax() {
+    const browser = await initBrowser();
+    const page = await browser.newPage();
+    
+    try {
+        await page.goto('https://toolbaz.com/writer/ai-writer', { 
+            waitUntil: 'networkidle2', 
+            timeout: 20000 
+        });
         
-    } catch (error) {
-        log('error', '❌ 模型提取失败:', error.message);
-        // 降级到静态配置
-        const allModels = [...CONFIG.CHAT_MODELS, ...CONFIG.IMAGE_MODELS];
-        res.writeHead(200, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
-        res.end(JSON.stringify({
-            object: 'list',
-            data: allModels.map(id => ({ id, object: 'model', created: Math.floor(Date.now() / 1000), owned_by: 'toolbaz-2api' }))
-        }));
+        const models = await page.evaluate(async () => {
+            return new Promise((resolve) => {
+                // 尝试通过AJAX获取模型列表
+                const modelEndpoints = [
+                    'https://data.toolbaz.com/models.php',
+                    'https://data.toolbaz.com/api/models',
+                    '/api/models',
+                    '/models'
+                ];
+                
+                let requestCount = 0;
+                
+                const tryFetch = (url) => {
+                    return fetch(url, {
+                        method: 'GET',
+                        headers: {
+                            'Accept': 'application/json',
+                            'X-Requested-With': 'XMLHttpRequest',
+                            'Referer': window.location.href
+                        }
+                    })
+                    .then(response => {
+                        if (response.ok) {
+                            return response.json();
+                        }
+                        throw new Error(`HTTP ${response.status}`);
+                    })
+                    .then(data => {
+                        if (Array.isArray(data) && data.length > 0) {
+                            return data.map((model, index) => ({
+                                id: model.id || model.name?.toLowerCase().replace(/[^a-z0-9]/g, '-') || `model-${index}`,
+                                name: model.name || model.displayName || `Model ${index}`,
+                                icon: model.icon || '',
+                                object: 'model',
+                                created: Math.floor(Date.now() / 1000),
+                                owned_by: 'toolbaz'
+                            }));
+                        }
+                        throw new Error('无效的模型数据');
+                    });
+                };
+                
+                // 尝试所有端点
+                const promises = modelEndpoints.map(url => 
+                    tryFetch(url).catch(err => {
+                        requestCount++;
+                        if (requestCount === modelEndpoints.length) {
+                            throw new Error('所有AJAX端点都失败了');
+                        }
+                        return null;
+                    })
+                );
+                
+                Promise.all(promises)
+                    .then(results => {
+                        const validResults = results.filter(r => r !== null);
+                        if (validResults.length > 0) {
+                            resolve(validResults[0]); // 使用第一个成功的结果
+                        } else {
+                            throw new Error('没有成功的AJAX请求');
+                        }
+                    })
+                    .catch(() => resolve([])); // 返回空数组表示失败
+            });
+        });
+        
+        if (models.length === 0) {
+            throw new Error('AJAX方式未获取到模型');
+        }
+        
+        return models;
+        
+    } finally {
+        await page.close().catch(e => log('warn', `页面关闭失败: ${e.message}`));
     }
 }
 
+// 新增：增强的静态模型配置
+async function getEnhancedStaticModels() {
+    // 结合配置中的模型和常见的AI模型
+    const enhancedModels = [
+        // 聊天模型
+        ...CONFIG.CHAT_MODELS.map(name => ({
+            id: name.toLowerCase().replace(/[^a-z0-9]/g, '-'),
+            name: name,
+            object: 'model',
+            created: Math.floor(Date.now() / 1000),
+            owned_by: 'toolbaz-2api'
+        })),
+        
+        // 图片模型
+        ...CONFIG.IMAGE_MODELS.map(name => ({
+            id: name.toLowerCase().replace(/[^a-z0-9]/g, '-'),
+            name: name,
+            object: 'model',
+            created: Math.floor(Date.now() / 1000),
+            owned_by: 'toolbaz-2api'
+        })),
+        
+        // 额外的常见模型
+        {
+            id: 'gpt-4',
+            name: 'GPT-4',
+            object: 'model',
+            created: Math.floor(Date.now() / 1000),
+            owned_by: 'toolbaz-2api'
+        },
+        {
+            id: 'gpt-3.5-turbo',
+            name: 'GPT-3.5 Turbo',
+            object: 'model',
+            created: Math.floor(Date.now() / 1000),
+            owned_by: 'toolbaz-2api'
+        },
+        {
+            id: 'claude-3-sonnet',
+            name: 'Claude 3 Sonnet',
+            object: 'model',
+            created: Math.floor(Date.now() / 1000),
+            owned_by: 'toolbaz-2api'
+        }
+    ];
+    
+    // 去重
+    const uniqueModels = [];
+    const seenIds = new Set();
+    
+    for (const model of enhancedModels) {
+        if (!seenIds.has(model.id)) {
+            seenIds.add(model.id);
+            uniqueModels.push(model);
+        }
+    }
+    
+    return uniqueModels;
+}
+
+// 新增：基础静态模型配置
+function getBasicStaticModels() {
+    return [...CONFIG.CHAT_MODELS, ...CONFIG.IMAGE_MODELS].map(id => ({
+        id: id.toLowerCase().replace(/[^a-z0-9]/g, '-'),
+        name: id,
+        object: 'model',
+        created: Math.floor(Date.now() / 1000),
+        owned_by: 'toolbaz-2api'
+    }));
+}
+
 // 新增：测试API端点
 async function handleTests(req, res) {
     if (!verifyAuth(req)) return sendError(res, 401, 'Unauthorized');
@@ -2003,7 +2432,11 @@ async function handleStatus(req, res) {
                 max_retries: CONFIG.MAX_RETRIES,
                 performance_monitoring: CONFIG.PERFORMANCE_MONITORING
             },
-            browser: browser ? 'running' : 'stopped'
+            browser: browser ? 'running' : 'stopped',
+            cache: {
+                models: modelCache.isValid() ? 'valid' : 'expired',
+                credentials: credentialCache.isValid() ? 'valid' : 'expired'
+            }
         };
         
         res.writeHead(200, { 'Content-Type': 'application/json', 'Access-Control-Allow-Origin': '*' });
@@ -2015,6 +2448,27 @@ async function handleStatus(req, res) {
     }
 }
 
+// 新增：模型刷新处理函数
+async function handleModelsRefresh(req, res) {
+    if (!verifyAuth(req)) return sendError(res, 401, 'Unauthorized');
+    
+    try {
+        log('info', '🔄 收到模型刷新请求...');
+        
+        // 清除模型缓存
+        modelCache.clear();
+        
+        // 强制重新提取模型
+        const response = await handleModels(req, res);
+        
+        log('info', '✅ 模型刷新完成');
+        
+    } catch (error) {
+        log('error', `❌ 模型刷新失败: ${error.message}`);
+        sendError(res, 500, error.message);
+    }
+}
+
 function handleUI(req, res) {
     const html = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width,initial-scale=1.0"><title>${CONFIG.PROJECT_NAME}</title><style>:root{--bg:#121212;--panel:#1E1E1E;--border:#333;--text:#E0E0E0;--primary:#00FF9D;--secondary:#00BFA5;--error:#F44336;--success:#4CAF50;--warning:#FF9800}body{font-family:Consolas,monospace;background:var(--bg);color:var(--text);margin:0;height:100vh;display:flex;overflow:hidden;font-size:13px}.sidebar{width:320px;background:var(--panel);border-right:1px solid var(--border);display:flex;flex-direction:column;padding:10px;gap:10px}.main{flex:1;display:flex;flex-direction:column}.log-panel{height:40%;border-top:1px solid var(--border);background:#000;overflow-y:auto;padding:10px}.box{background:#252525;padding:15px;border-radius:6px;border:1px solid var(--border)}.label{font-size:11px;color:#888;display:block;margin-bottom:5px;text-transform:uppercase}input,select,textarea{width:100%;background:#333;border:1px solid #444;color:#fff;padding:8px;border-radius:4px;margin-bottom:10px;box-sizing:border-box;font-family:inherit}button{width:100%;padding:10px;background:var(--primary);border:none;border-radius:4px;font-weight:bold;cursor:pointer;color:#000;margin-bottom:5px;transition:all 0.2s}button:disabled{background:#555;cursor:not-allowed}button.secondary{background:#333;color:#fff;border:1px solid #555}button.danger{background:var(--error);color:#fff}button.warning{background:var(--warning);color:#000}.chat-window{flex:1;padding:20px;overflow-y:auto;display:flex;flex-direction:column;gap:15px}.msg{max-width:85%;padding:10px 15px;border-radius:6px;line-height:1.5;word-wrap:break-word}.msg.user{align-self:flex-end;background:#333;color:#fff}.msg.ai{align-self:flex-start;background:#1a1a1a;border:1px solid #333;width:100%}.msg.img img{max-width:100%;border-radius:4px}.log-entry{margin-bottom:8px;border-bottom:1px solid #222;padding-bottom:8px}.log-time{color:#666;margin-right:10px}.log-data{color:#ccc;white-space:pre-wrap;word-break:break-all;margin-top:4px;display:block}.tabs{display:flex;border-bottom:1px solid var(--border);margin-bottom:10px}.tab{flex:1;text-align:center;padding:8px;cursor:pointer;color:#888;border-bottom:2px solid transparent;transition:all 0.2s}.tab.active{color:var(--primary);border-bottom-color:var(--primary)}.status-item{display:flex;justify-content:space-between;margin-bottom:5px;font-size:12px}.status-value{color:var(--primary)}.test-result{margin:5px 0;padding:5px;border-radius:3px;font-size:11px}.test-pass{background:var(--success);color:#000}.test-fail{background:var(--error);color:#fff}.performance-grid{display:grid;grid-template-columns:1fr 1fr;gap:5px;margin-top:10px}.perf-item{text-align:center;padding:5px;background:#333;border-radius:3px}.perf-value{font-size:16px;color:var(--primary)}.perf-label{font-size:10px;color:#888}</style></head><body><div class="sidebar"><div class="box"><h2 style="margin:0;color:var(--primary)">${CONFIG.PROJECT_NAME}</h2><span style="font-size:11px;color:#666">v${CONFIG.PROJECT_VERSION} | Enhanced Testing</span></div><div class="box"><div class="tabs"><div class="tab active" onclick="switchMode('chat')">聊天</div><div class="tab" onclick="switchMode('image')">绘图</div><div class="tab" onclick="switchMode('test')">测试</div></div><div id="chat-controls"><span class="label">Model</span><select id="chat-model">${CONFIG.CHAT_MODELS.map(m => `<option value="${m}">${m}</option>`).join('')}</select><span class="label">Prompt</span><textarea id="chat-prompt" rows="5" placeholder="输入内容...">你好，请介绍一下你自己</textarea></div><div id="image-controls" style="display:none"><span class="label">Model</span><select id="image-model">${CONFIG.IMAGE_MODELS.map(m => `<option value="${m}">${m}</option>`).join('')}</select><span class="label">Prompt</span><textarea id="image-prompt" rows="5" placeholder="输入图片描述...">A futuristic city at sunset</textarea></div><div id="test-controls" style="display:none"><span class="label">测试类型</span><select id=\"test-type\"><option value=\"comprehensive\">综合测试</option><option value=\"config\">配置测试</option><option value=\"network\">网络测试</option><option value=\"browser\">浏览器测试</option><option value=\"session\">会话测试</option><option value=\"ajax\">AJAX测试</option><option value=\"prefix\">前缀过滤测试</option></select><button class="warning" onclick="runTest()">🧪 运行测试</button></div><button id="btn-send" onclick="handleSend()">发送请求</button></div><div class="box"><div class="status-item"><span>服务状态:</span><span class="status-value" id="service-status">检查中...</span></div><div class="status-item"><span>浏览器:</span><span class="status-value" id="browser-status">未知</span></div><div class="status-item"><span>总请求:</span><span class="status-value" id="total-requests">0</span></div><div class="status-item"><span>成功率:</span><span class="status-value" id="success-rate">0%</span></div><div class="performance-grid"><div class="perf-item"><div class="perf-value" id="ajax-success">0</div><div class="perf-label">AJAX成功</div></div><div class="perf-item"><div class="perf-value" id="ui-fallback">0</div><div class="perf-label">UI回退</div></div><div class="perf-item"><div class="perf-value" id="avg-response">0ms</div><div class="perf-label">平均响应</div></div><div class="perf-item"><div class="perf-value" id="uptime">0m</div><div class="perf-label">运行时间</div></div></div></div><div class="box" style="margin-top:auto"><button class="secondary" onclick="copyLogs()">📋 复制日志</button><button class="secondary" onclick="clearLogs()">🗑️ 清空日志</button><button class="secondary" onclick="refreshStatus()">🔄 刷新状态</button></div></div><div class="main"><div class="chat-window" id="chat"><div class="msg ai">🎉 ${CONFIG.PROJECT_NAME} v${CONFIG.PROJECT_VERSION} 已就绪<br><br>✨ 新功能:<br>• 🧪 增强的测试套件<br>• 📊 性能监控和统计<br>• 🔍 深度调试信息<br>• 🔄 智能重试机制<br><br>请选择聊天或测试模式开始使用！</div></div><div class="log-panel" id="log-container"></div></div><script>const API_KEY="${CONFIG.API_MASTER_KEY}";const ORIGIN=window.location.origin;let currentMode='chat';let statusInterval;function switchMode(mode){currentMode=mode;document.querySelectorAll('.tab').forEach(t=>t.classList.remove('active'));event.target.classList.add('active');document.getElementById('chat-controls').style.display=mode==='chat'?'block':'none';document.getElementById('image-controls').style.display=mode==='image'?'block':'none';document.getElementById('test-controls').style.display=mode==='test'?'block':'none';const sendBtn=document.getElementById('btn-send');sendBtn.style.display=mode==='test'?'none':'block';}function appendLog(msg,type='info'){const div=document.createElement('div');div.className='log-entry';let color='#ccc';switch(type){case 'error':color='#F44336';break;case 'warn':color='#FF9800';break;case 'success':color='#4CAF50';break;case 'info':color='#2196F3';break;}div.innerHTML=\`<span class="log-time">\${new Date().toLocaleTimeString()}</span><span class="log-data" style="color:\${color}">\${msg}</span>\`;document.getElementById('log-container').appendChild(div);div.scrollIntoView()}function clearLogs(){document.getElementById('log-container').innerHTML=''}function copyLogs(){const logs=Array.from(document.querySelectorAll('.log-entry')).map(e=>e.innerText).join('\\n');navigator.clipboard.writeText(logs).then(()=>appendLog('日志已复制到剪贴板','success'))}function appendMsg(role,content){const div=document.createElement('div');div.className=\`msg \${role}\`;div.innerHTML=content;document.getElementById('chat').appendChild(div);div.scrollIntoView({behavior:"smooth"});return div}async function handleSend(){const btn=document.getElementById('btn-send');btn.disabled=true;clearLogs();appendLog('🚀 开始请求...');if(currentMode==='chat'){const prompt=document.getElementById('chat-prompt').value;const model=document.getElementById('chat-model').value;if(!prompt){btn.disabled=false;return}appendMsg('user',prompt);const aiMsg=appendMsg('ai','...');let fullText='';try{appendLog('📡 发送到浏览器...');const res=await fetch(ORIGIN+'/v1/chat/completions',{method:'POST',headers:{'Authorization':'Bearer '+API_KEY,'Content-Type':'application/json'},body:JSON.stringify({model,messages:[{role:'user',content:prompt}],stream:true})});appendLog('✅ 接收响应...');const reader=res.body.getReader();const decoder=new TextDecoder();while(true){const{done,value}=await reader.read();if(done)break;const chunk=decoder.decode(value);const lines=chunk.split('\\n');for(const line of lines){if(line.startsWith('data: ')){const dataStr=line.slice(6);if(dataStr==='[DONE]')break;try{const json=JSON.parse(dataStr);if(json.choices&&json.choices[0].delta.content){fullText+=json.choices[0].delta.content;aiMsg.innerText=fullText}}catch(e){}}}}appendLog('🎉 完成！','success');refreshStatus();}catch(e){aiMsg.innerText+='\\n[错误]: '+e.message;appendLog('❌ 错误: '+e.message,'error')}}else{const prompt=document.getElementById('image-prompt').value;const model=document.getElementById('image-model').value;if(!prompt){btn.disabled=false;return}appendMsg('user',prompt);const aiMsg=appendMsg('ai','生成中...');try{appendLog('🎨 发送图片请求...');const res=await fetch(ORIGIN+'/v1/images/generations',{method:'POST',headers:{'Authorization':'Bearer '+API_KEY,'Content-Type':'application/json'},body:JSON.stringify({model,prompt})});const data=await res.json();if(data.error)throw new Error(data.error.message);const imgUrl=data.data[0].url;aiMsg.innerHTML=\`<img src="\${imgUrl}" onclick="window.open(this.src)">\`;aiMsg.className='msg ai img';appendLog('✅ 图片生成成功！','success');refreshStatus();}catch(e){aiMsg.innerText='生成失败: '+e.message;appendLog('❌ 错误: '+e.message,'error')}}btn.disabled=false}async function runTest(){const testType=document.getElementById('test-type').value;appendLog(\`🧪 开始运行\${testType}测试...\`,'info');const testMsg=appendMsg('ai','<div id="test-results">正在运行测试...</div>');try{const res=await fetch(ORIGIN+'/v1/tests?type='+testType,{headers:{'Authorization':'Bearer '+API_KEY}});const data=await res.json();let html='<h3>测试结果</h3>';if(data.overall){html+=\`<p><strong>综合测试:</strong> \${data.overall.passed}/\${data.overall.passed+data.overall.failed} 通过</p>\`;if(data.configuration){html+=\`<div class="test-result test-pass">配置测试: \${data.configuration.passed} 通过, \${data.configuration.failed} 失败</div>\`;}}if(data.success===false){html+=\`<div class="test-result test-fail">测试失败: \${data.error}</div>\`;}else if(data.success){html+=\`<div class="test-result test-pass">测试通过</div>\`;}if(data.analysis){html+=\`<p><strong>会话分析:</strong> 找到 \${data.analysis.elementsFound.length} 个相关元素</p>\`;}if(data.length>0){html+='<h4>详细结果:</h4><ul>';data.forEach(result=>{const status=result.status==='success'?'pass':'fail';html+=\`<li class="test-result test-\${status}">\${result.name}: \${result.status}</li>\`;});html+='</ul>';}testMsg.querySelector('#test-results').innerHTML=html;appendLog('测试完成','success');refreshStatus();}catch(e){testMsg.innerHTML='<div class="test-result test-fail">测试执行失败: '+e.message+'</div>';appendLog('测试执行失败: '+e.message,'error');}}async function refreshStatus(){try{const res=await fetch(ORIGIN+'/v1/status',{headers:{'Authorization':'Bearer '+API_KEY}});const data=await res.json();document.getElementById('service-status').textContent='运行中';document.getElementById('browser-status').textContent=data.browser||'未知';document.getElementById('total-requests').textContent=data.performance.totalRequests||0;const successRate=data.performance.totalRequests>0?Math.round(data.performance.successfulRequests/data.performance.totalRequests*100):0;document.getElementById('success-rate').textContent=successRate+'%';document.getElementById('ajax-success').textContent=data.performance.ajaxSuccess||0;document.getElementById('ui-fallback').textContent=data.performance.uiFallback||0;document.getElementById('avg-response').textContent=(data.performance.averageResponseTime||0)+'ms';document.getElementById('uptime').textContent=Math.floor(data.uptime/60)+'m';}catch(e){appendLog('状态刷新失败: '+e.message,'error');}}function init(){refreshStatus();statusInterval=setInterval(refreshStatus,30000);}init();</script></body></html>`;
     res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
@@ -2048,6 +2502,7 @@ const server = http.createServer(async (req, res) => {
     // 新增的测试和状态端点
     if (url.pathname === '/v1/tests') return await handleTests(req, res);
     if (url.pathname === '/v1/status') return await handleStatus(req, res);
+    if (url.pathname === '/v1/models/refresh') return await handleModelsRefresh(req, res);
     
     sendError(res, 404, 'Not Found');
 });
@@ -2103,8 +2558,9 @@ server.listen(CONFIG.PORT, async () => {
     log('info', '  - GET  /          : Web界面');
     log('info', '  - POST /v1/chat/completions  : 聊天API');
     log('info', '  - POST /v1/images/generations : 图片生成API');
-    log('info', '  - GET  /v1/models : 模型列表API');
-    log('info', '  - GET  /v1/status : 服务状态API');
+    log('info', '  - GET  /v1/models : 模型列表API（带60分钟缓存）');
+    log('info', '  - GET  /v1/models/refresh : 强制刷新模型列表API');
+    log('info', '  - GET  /v1/status : 服务状态API（包含缓存状态）');
     log('info', '  - GET  /v1/tests : 测试API (?type=config|network|browser|session|ajax|comprehensive)');
     
     // 启动性能统计定时显示