Update index.js

index.js

@@ -8,12 +8,12 @@ import dotenv from 'dotenv';
 // 配置加载
 dotenv.config();
 
-
 // 配置常量
 const CONFIG = {
   API: {
       BASE_URL: "wss://api.inkeep.com/graphql",
       API_KEY: process.env.API_KEY || "sk-123456",
+      SYSTEM_MESSAGE: process.env.SYSTEM_MESSAGE || null
   },
   MODELS: {
       'claude-3-5-sonnet-20241022': 'claude-3-5-sonnet-20241022',
@@ -71,7 +71,7 @@ class AiApiClient {
           
           if (currentContent === null) return acc;
 
-          const currentMessageRole = current.role === "system" || current.role === "user" ? "HUMAN" : "ASSISTANT";
+          const currentMessageRole = current.role === "system" || current.role === "user" ? "Human" : "Assistant";
 
           // 系统消息处理逻辑
           if (current.role === "system") {
@@ -186,7 +186,7 @@ class WebSocketUtils {
     static async createWebSocketClient(requestPayload) {
         // 检查当前连接数是否达到上限
         if (this.activeConnections.size >= this.MAX_CONNECTIONS) {
-            throw new Error(`当前连接数已达到上限 (${this.MAX_CONNECTIONS})，请稍后重试喵！`);
+            throw new Error(`当前连接数已达到上限 (${this.MAX_CONNECTIONS})，请稍后重试！`);
         }
 
         let timeoutId;
@@ -310,7 +310,7 @@ class WebSocketUtils {
                     organizationId: 'org_JfjtEvzbwOikUEUn',
                     integrationId: 'clwtqz9sq001izszu8ms5g4om',
                     chatMode: 'AUTO',
-                    context: requestPayload.systemMessage,
+                    context: requestPayload.systemMessage || CONFIG.API.SYSTEM_MESSAGE,
                     messageAttributes: {},
                     includeAIAnnotations: false,
                     environment: 'production'
@@ -333,6 +333,7 @@ class WebSocketUtils {
             }
         };
 
+        console.log(JSON.stringify(subscribeMessage,null,2));
         if (ws.readyState === WebSocket.OPEN) {
             ws.send(JSON.stringify(subscribeMessage));
         }
@@ -369,7 +370,7 @@ app.use(cors({
 
 
 // 获取模型列表路由
-app.get('/hf/v1/models', (req, res) => {
+app.get('/v1/models', (req, res) => {
   res.json({
       object: "list",
       data: [{
@@ -382,7 +383,7 @@ app.get('/hf/v1/models', (req, res) => {
 });
 
 // 聊天完成路由
-app.post('/hf/v1/chat/completions', async (req, res) => {
+app.post('/v1/chat/completions', async (req, res) => {
   try {
       const { messages, model, stream } = req.body;
       const authToken = req.headers.authorization?.replace('Bearer ', '');
@@ -420,4 +421,4 @@ app.use((req, res) => {
 // 启动服务器
 app.listen(CONFIG.SERVER.PORT, () => {
   console.log(`服务器运行在 http://localhost:${CONFIG.SERVER.PORT}`);
-});
+});