Update index.js

index.js

@@ -1,13 +1,11 @@
-import express from 'express';
-import fetch from 'node-fetch';
-import { v4 as uuidv4 } from 'uuid';
-import dotenv from 'dotenv';
-import { Buffer } from 'buffer';
-
-dotenv.config();
-
+const express = require('express');
+const axios = require('axios');
+const { v4: uuidv4 } = require('uuid');
 const app = express();
 
+// 中间件
+app.use(express.json());
+app.use(express.text());
 
 const modelMapping = {
   "gpt-4o-mini": "GPT-4o mini",
@@ -15,34 +13,42 @@ const modelMapping = {
   "llama-3": "Llama 3",
   "gemini-1.5": "Gemini 1.5",
   "gemini-flash": "Gemini Flash",
-  "command-r": "Command R"
+  "command-r": "Command R",
 };
 
-app.use(express.json());
+// 认证中间件
+const authMiddleware = (req, res, next) => {
+  const authToken = process.env.AUTH_TOKEN;
+  if (authToken) {
+    const requestToken = req.headers.authorization || '';
+    const token = requestToken.replace('Bearer ', '');
+    if (token !== authToken) {
+      return res.status(401).send('Access Denied');
+    }
+  }
+  next();
+};
 
-const getCurrentTimestamp = () => Math.floor(Date.now() / 1000);
+// 获取临时用户ID
+const getTempUserID = async () => {
+  const response = await axios.get('https://playground.julius.ai/api/temp_user_id');
+  return response.data.temp_user_id;
+};
 
+// 分割文本为块
 const splitIntoChunks = (text, chunkSize) => {
+  const chars = Array.from(text);
   const chunks = [];
-  for (let i = 0; i < text.length; i += chunkSize) {
-    chunks.push(text.slice(i, i + chunkSize));
+  for (let i = 0; i < chars.length; i += chunkSize) {
+    chunks.push(chars.slice(i, i + chunkSize).join(''));
   }
   return chunks;
 };
 
-const getTempUserID = async () => {
-  try {
-    const response = await fetch('https://playground.julius.ai/api/temp_user_id');
-    const data = await response.json();
-    return data.temp_user_id;
-  } catch (error) {
-    throw new Error('Failed to get temp user ID');
-  }
-};
-
+// 发送消息到 Julius
 const sendToJulius = async (tempUserID, message, model) => {
   const conversationID = uuidv4();
-  const juliusReq = {
+  const data = {
     message: {
       content: message,
       role: "user"
@@ -51,148 +57,120 @@ const sendToJulius = async (tempUserID, message, model) => {
     chat_mode: "auto",
     client_version: "20240130",
     theme: "dark",
+    new_images: null,
+    new_attachments: null,
+    dataframe_format: "json",
     selectedModels: [model]
   };
 
-  try {
-    const response = await fetch('https://playground.julius.ai/api/chat/message', {
-      method: 'POST',
-      headers: {
-        'is-demo': tempUserID,
-        'Content-Type': 'application/json',
-        'Platform': 'web',
-        'conversation-id': conversationID,
-        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36'
-      },
-      body: JSON.stringify(juliusReq)
-    });
-
-    const reader = response.body.pipeThrough(new TextDecoderStream()).getReader();
-    let fullResponse = '';
-
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) break;
-      
-      const lines = value.split('\n');
-      for (const line of lines) {
-        try {
-          const jsonResp = JSON.parse(line);
-          if (jsonResp.content) {
-            fullResponse += jsonResp.content;
-          }
-        } catch (e) {
-          // 忽略解析错误
-        }
-      }
-    }
-
-    return fullResponse;
-  } catch (error) {
-    throw new Error('Failed to send to Julius');
-  }
+  const response = await axios({
+    method: 'post',
+    url: 'https://playground.julius.ai/api/chat/message',
+    data,
+    headers: {
+      'is-demo': tempUserID,
+      'conversation-id': conversationID,
+      'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
+      'interactive-charts': 'true',
+      'use-dict': 'true',
+      'Gcs': 'true',
+      'Is-Native': 'false',
+      'sec-ch-ua-platform': 'Windows'
+    },
+    responseType: 'text'
+  });
+
+  return response.data.split('\n')
+    .filter(line => line.trim())
+    .map(line => {
+      try { return JSON.parse(line); } 
+      catch { return null; }
+    })
+    .filter(Boolean)
+    .reduce((acc, item) => acc + (item.content || ''), '');
 };
 
-app.all('*', async (req, res) => {
-  // 身份验证
-  const authToken = process.env.AUTH_TOKEN;
-  if (authToken) {
-    const requestToken = req.headers.authorization?.replace('Bearer ', '');
-    if (!requestToken || requestToken !== authToken) {
-      return res.status(401).send('Access Denied');
-    }
-  }
-
-  // 健康检查
-  if (req.path !== '/v1/chat/completions') {
-    return res.json({
-      status: "Julius2Api Service Running...",
-      message: "MoLoveSze..."
-    });
-  }
-
-  if (req.method !== 'POST') {
-    return res.status(405).send('Method not allowed');
-  }
-
+// 处理聊天请求
+app.post('/v1/chat/completions', authMiddleware, async (req, res) => {
   try {
-    const openAIReq = req.body;
-    const isStream = openAIReq.stream || false;
+    const { messages, model: originalModel, stream } = req.body;
+    const model = modelMapping[originalModel] || "GPT-4o mini";
+    const tempUserID = await getTempUserID();
+    const content = messages[messages.length - 1].content;
 
-    // 模型映射
-    const mappedModel = modelMapping[openAIReq.model] || 'GPT-4o mini';
-    openAIReq.model = mappedModel;
+    // 获取 Julius 响应
+    const juliusResp = await sendToJulius(tempUserID, content, model);
 
-    const tempUserID = await getTempUserID();
-    const juliusResp = await sendToJulius(tempUserID, 
-      openAIReq.messages[openAIReq.messages.length - 1].content, 
-      mappedModel
-    );
-
-    const respId = `chatcmpl-${tempUserID}`;
-    const created = getCurrentTimestamp();
-
-    if (isStream) {
-      res.writeHead(200, {
-        'Content-Type': 'text/event-stream',
-        'Cache-Control': 'no-cache',
-        'Connection': 'keep-alive'
-      });
+    // 流式响应
+    if (stream) {
+      res.setHeader('Content-Type', 'text/event-stream');
+      res.setHeader('Cache-Control', 'no-cache');
+      res.setHeader('Connection', 'keep-alive');
+
+      const respId = `chatcmpl-${tempUserID}`;
+      const created = Math.floor(Date.now() / 1000);
+      const chunks = splitIntoChunks(juliusResp, 50);
 
-      // 发送初始响应
-      const firstResponse = {
+      // 发送初始化消息
+      res.write(`data: ${JSON.stringify({
         id: respId,
         object: "chat.completion.chunk",
-        created: created,
-        model: mappedModel,
-        choices: [{
-          delta: { role: "assistant" },
-          index: 0
-        }]
-      };
-      res.write(`data: ${JSON.stringify(firstResponse)}\n\n`);
-
-      // 分块发送内容
-      const chunks = splitIntoChunks(juliusResp, 50);
+        created,
+        model,
+        choices: [{ delta: { role: "assistant" }, index: 0 }]
+      })}\n\n`);
+
+      // 发送内容块
       chunks.forEach((chunk, index) => {
-        const response = {
+        res.write(`data: ${JSON.stringify({
           id: respId,
           object: "chat.completion.chunk",
-          created: created,
-          model: mappedModel,
+          created,
+          model,
           choices: [{
             delta: { content: chunk },
             index: 0,
-            finish_reason: index === chunks.length - 1 ? 'stop' : null
+            finish_reason: index === chunks.length - 1 ? "stop" : null
           }]
-        };
-        res.write(`data: ${JSON.stringify(response)}\n\n`);
+        })}\n\n`);
       });
 
+      // 结束标志
       res.write('data: [DONE]\n\n');
-      res.end();
-    } else {
-      const response = {
-        id: respId,
-        object: "chat.completion",
-        created: created,
-        model: mappedModel,
-        choices: [{
-          message: {
-            role: "assistant",
-            content: juliusResp
-          },
-          finish_reason: "stop"
-        }]
-      };
-      res.json(response);
+      return res.end();
     }
+
+    // 普通响应
+    res.json({
+      id: `chatcmpl-${tempUserID}`,
+      object: "chat.completion",
+      created: Math.floor(Date.now() / 1000),
+      model,
+      choices: [{
+        message: {
+          role: "assistant",
+          content: juliusResp
+        },
+        finish_reason: "stop",
+        index: 0
+      }]
+    });
+
   } catch (error) {
-    console.error(error);
-    res.status(500).send(error.message);
+    console.error('Error:', error);
+    res.status(500).json({ error: error.message });
   }
 });
 
+// 根路径响应
+app.all('*', (req, res) => {
+  res.json({
+    status: "Julius2Api Service Running...",
+    message: "MoLoveSze..."
+  });
+});
+
+// 启动服务器
 const PORT = process.env.PORT || 3000;
 app.listen(PORT, () => {
   console.log(`Server running on port ${PORT}`);