Update server.js

server.js

@@ -15,11 +15,14 @@ const geminiApiUrl = 'https://generativelanguage.googleapis.com/v1/models/gemini
 
 const geminiTmpDir = path.join(require('os').tmpdir(), 'gemini_sessions');
 const deepseekTmpDir = path.join(require('os').tmpdir(), 'deepseek_sessions');
+const qwenTmpDir = path.join(require('os').tmpdir(), 'qwen_sessions');
+
 if (!fs.existsSync(geminiTmpDir)) fs.mkdirSync(geminiTmpDir);
 if (!fs.existsSync(deepseekTmpDir)) fs.mkdirSync(deepseekTmpDir);
+if (!fs.existsSync(qwenTmpDir)) fs.mkdirSync(qwenTmpDir);
 
 function getSessionFile(sessionId, model) {
-  const dir = model === 'gemini' ? geminiTmpDir : deepseekTmpDir;
+  const dir = model === 'gemini' ? geminiTmpDir : deepseekTmpDir && model === 'qwen' ? qwenTmpDir : '';
   return path.join(dir, `${sessionId}.json`);
 }
 
@@ -42,7 +45,7 @@ async function gemini(sessionId, prompt) {
   
   const reply = response.data.candidates?.[0]?.content?.parts?.[0]?.text || 'No response';  
   history.push({ role: 'assistant', content: { parts: [{ text: reply }] } });  
-  fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));  
+  fs.writeFileSync(sessionFile, history);  
   return reply;
 }
 
@@ -62,7 +65,27 @@ async function deepseek(sessionId, prompt) {
   if (!data.success) throw new Error(JSON.stringify(data, null, 2));  
   let response = data.data.response  
   history.push({ role: 'assistant', content: response.split("</think>")[1]?.trim() });  
-  fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));  
+  fs.writeFileSync(sessionFile, history);  
+  return response;
+}
+
+async function qwen(sessionId, prompt) {
+  const sessionFile = getSessionFile(sessionId, 'qwen');
+  let history = [];
+  if (fs.existsSync(sessionFile)) {
+    history = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
+  }
+  history.push({ role: 'user', content: prompt });
+
+  let { data } = await axios.post("https://ai.clauodflare.workers.dev/chat", {  
+    "model": "@cf/qwen/qwen1.5-0.5b-chat",  
+    "messages": history  
+  }).catch(e => e.response);  
+  
+  if (!data.success) throw new Error(JSON.stringify(data, null, 2));  
+  let response = data.data.response  
+  history.push({ role: 'assistant', content: response });  
+  fs.writeFileSync(sessionFile, history);  
   return response;
 }
 
@@ -82,7 +105,7 @@ app.all('/', (_, res) => {
 	status['memoryUsage'] =
 		`${utils.formatSize(totalmem - freemem)} / ${utils.formatSize(totalmem)}`
 	res.json({
-		router: ['/google/gemini-1.5-pro', '/deepseek-ai/deepseek-r1-distill-qwen-32b'],
+		router: ['/google/gemini-1.5-pro', '/deepseek-ai/deepseek-r1-distill-qwen-32b', '/qwen/qwen1.5-0.5b-chat'],
 		uptime: new Date(process.uptime() * 1000).toUTCString().split(' ')[4],
 		status
 	})
@@ -110,4 +133,15 @@ app.get('/deepseek-ai/deepseek-r1-distill-qwen-32b', async (req, res) => {
   }
 });
 
+app.get('/qwen/qwen1.5-0.5b-chat', async (req, res) => {
+  try {
+    const { question, sessionId } = req.query;
+    if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' });
+    const reply = await qwen(sessionId, question);
+    res.json({ reply });
+  } catch (error) {
+    res.status(500).json({ error: error.response?.data || error.message });
+  }
+});
+
 app.listen(7860);