ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter6/live-audio/backend/test_llm.js

@@ -0,0 +1,83 @@
+const axios = require('axios');
+const config = require('./config');
+
+async function testLLMEndpoint() {
+    console.log('Testing LLM endpoint...');
+    console.log(`URL: ${config.LLM_API_URL}`);
+    console.log(`Model: ${config.LLM_MODEL}`);
+    
+    const testPrompt = "Tell me a short joke.";
+    
+    try {
+        const response = await axios.post(
+            config.LLM_API_URL,
+            {
+                model: config.LLM_MODEL,
+                messages: [
+                    { role: "user", content: testPrompt }
+                ],
+                stream: true,
+                max_tokens: 100
+            },
+            {
+                headers: {
+                    'Authorization': `Bearer ${config.OPENAI_API_KEY}`,
+                    'Content-Type': 'application/json',
+                },
+                responseType: 'stream'
+            }
+        );
+
+        console.log('Connection established successfully');
+        
+        let fullResponse = '';
+        
+        response.data.on('data', chunk => {
+            const lines = chunk.toString().split('\n');
+            for (const line of lines) {
+                if (line.trim() === '') continue;
+                if (line.trim() === 'data: [DONE]') continue;
+                if (!line.startsWith('data: ')) continue;
+                
+                try {
+                    const jsonData = JSON.parse(line.replace('data: ', ''));
+                    const content = jsonData.choices[0]?.delta?.content || '';
+                    fullResponse += content;
+                    process.stdout.write(content); // Stream the response
+                } catch (e) {
+                    console.error('Error parsing chunk:', e);
+                    console.error('Raw chunk:', line);
+                }
+            }
+        });
+
+        response.data.on('end', () => {
+            console.log('\n\nFull response received:', fullResponse);
+            console.log('\nTest completed successfully');
+        });
+
+        response.data.on('error', (error) => {
+            console.error('Stream error:', error);
+        });
+
+    } catch (error) {
+        console.error('\nError testing LLM endpoint:', error.message);
+        if (error.response) {
+            console.error('Response status:', error.response.status);
+            console.error('Response headers:', JSON.stringify(error.response.headers, null, 2));
+        }
+        
+        // Log safe error properties
+        const safeError = {
+            message: error.message,
+            name: error.name,
+            stack: error.stack,
+            code: error.code,
+            status: error.status
+        };
+        console.error('\nError details:', JSON.stringify(safeError, null, 2));
+    }
+}
+
+// Run the test
+testLLMEndpoint();