Update index.js

index.js

@@ -12,7 +12,7 @@ dotenv.config();
 const CONFIG = {
   API: {
       BASE_URL: "wss://api.inkeep.com/graphql",
-      API_KEY: process.env.API_KEY || "sk-123456", // 仍然从环境变量获取，但保留默认值
+      API_KEY: process.env.API_KEY || "sk-123456",
   },
   MODELS: {
       'claude-3-5-sonnet-20241022': 'claude-3-5-sonnet-20241022',
@@ -123,7 +123,7 @@ class ResponseHandler {
       
       let index = 0;
       while (index < responseContent.length) {
-          const chunkSize = Math.floor(Math.random() * (30 - 16)) + 15; // 保留随机 chunk 大小
+          const chunkSize = Math.floor(Math.random() * (30 - 16)) + 15;
           const chunk = responseContent.slice(index, index + chunkSize);
           
           res.write(`data: ${JSON.stringify({
@@ -172,135 +172,127 @@ class ResponseHandler {
 
 // WebSocket工具类
 class WebSocketUtils {
-    // 生成WebSocket密钥
-    static generateWebSocketKey() {
-        return randomBytes(16).toString('base64');
-    }
-
-    // 创建WebSocket客户端
-    static async createWebSocketClient(requestPayload) {
-        const websocketKey = this.generateWebSocketKey();
-        const ws = new WebSocket(CONFIG.API.BASE_URL, 'graphql-transport-ws', { // 保留硬编码
-            headers: {
-                ...CONFIG.DEFAULT_HEADERS,
-                'Sec-WebSocket-Key': websocketKey,
-            }
-        });
-
-        let responseContent = '';
-        let isComplete = false;
-
-        try {
-            await new Promise((resolve, reject) => {
-                ws.on('open', () => {
-                    console.log('WebSocket连接已建立');
-                    const connectionInitMessage = {
-                        type: 'connection_init',
-                        payload: {
-                            headers: {
-                                // 从环境变量获取 API 密钥
-                                Authorization: `Bearer ${CONFIG.API.API_KEY}`
-                            }
-                        }
-                    };
-                    ws.send(JSON.stringify(connectionInitMessage));
-                });
-
-                ws.on('message', async (data) => {
-                    const message = data.toString();
-                    const parsedMessage = JSON.parse(message);
-
-                    switch (parsedMessage.type) {
-                        case 'connection_ack':
-                            console.log('WebSocket连接请求中');
-                            this.sendChatSubscription(ws, requestPayload);
-                            break;
-                        case 'next':
-                            const chatResponse = await this.handleChatResponse(parsedMessage);
-                            if (chatResponse) {
-                                responseContent = chatResponse;
-                            }
-                            break;
-                        case 'complete':
-                            isComplete = true;
-                            ws.close();
-                            resolve(responseContent);
-                            break;
-                    }
-                });
-
-                ws.on('error', (err) => {
-                    console.error('WebSocket错误:', err);
-                    // 除了记录错误，还可以尝试发送错误信息给客户端
-                    reject(err);
-                });
-
-                ws.on('close', (code, reason) => {
-                    console.log(`WebSocket连接关闭，代码：${code}，原因：${reason}`);
-                    if (!isComplete) {
-                        reject(new Error(`WebSocket closed unexpectedly (code: ${code}, reason: ${reason})`));
-                    } else {
-                        resolve(responseContent); // 在连接正常关闭时也需要 resolve
-                    }
-                });
-            });
-
-            return responseContent; // 返回最终的响应内容
-
-        } catch (error) {
-            console.error('WebSocket处理过程中发生错误:', error);
-            throw error; // 将错误继续抛出，以便上层处理
-        }
-    }
-
-    // 发送聊天订阅
-    static sendChatSubscription(ws, requestPayload) {
-        const subscribeMessage = {
-            id: uuidv4(),
-            type: 'subscribe',
-            payload: {
-                variables: {
-                    messageInput: requestPayload.contextMessages,
-                    messageContext: null,
-                    organizationId: 'org_JfjtEvzbwOikUEUn', // 保留硬编码
-                    integrationId: 'clwtqz9sq001izszu8ms5g4om', // 保留硬编码
-                    chatMode: 'AUTO',
-                    context: requestPayload.systemMessage,
-                    messageAttributes: {},
-                    includeAIAnnotations: false,
-                    environment: 'production'
-                },
-                extensions: {},
-                operationName: 'OnNewSessionChatResult',
-                query: `subscription OnNewSessionChatResult($messageInput: String!, $messageContext: String, $organizationId: ID!, $integrationId: ID, $chatMode: ChatMode, $filters: ChatFiltersInput, $messageAttributes: JSON, $tags: [String!], $workflowId: String, $context: String, $guidance: String, $includeAIAnnotations: Boolean!, $environment: String) {
-                    newSessionChatResult(
-                        input: {messageInput: $messageInput, messageContext: $messageContext, organizationId: $organizationId, integrationId: $integrationId, chatMode: $chatMode, filters: $filters, messageAttributes: $messageAttributes, tags: $tags, workflowId: $workflowId, context: $context, guidance: $guidance, environment: $environment}
-                    ) {
-                        isEnd
-                        sessionId
-                        message {
-                            id
-                            content
-                        }
-                        __typename
+  // 生成WebSocket密钥
+  static generateWebSocketKey() {
+      return randomBytes(16).toString('base64');
+  }
+
+  // 创建WebSocket客户端
+  static createWebSocketClient(requestPayload) {
+      return new Promise((resolve, reject) => {
+          const websocketKey = this.generateWebSocketKey();
+          const ws = new WebSocket(CONFIG.API.BASE_URL, 'graphql-transport-ws', {
+              headers: {
+                  ...CONFIG.DEFAULT_HEADERS,
+                  'Sec-WebSocket-Key': websocketKey,
+              }
+          });
+
+          let responseContent = '';
+          let isComplete = false;
+
+          ws.on('open', () => {
+              console.log('WebSocket连接已建立');
+              const connectionInitMessage = {
+                  type: 'connection_init',
+                  payload: {
+                      headers: {
+                          Authorization: 'Bearer ee5b7c15ed3553cd6abc407340aad09ac7cb3b9f76d8613a'
+                      }
+                  }
+              };
+              ws.send(JSON.stringify(connectionInitMessage));
+          });
+
+          ws.on('message', async (data) => {
+              const message = data.toString();
+              const parsedMessage = JSON.parse(message);
+              
+              switch (parsedMessage.type) {
+                  case 'connection_ack':
+                      console.log('WebSocket连接请求中');
+                      this.sendChatSubscription(ws, requestPayload);
+                      break;
+                  case 'next':
+                      const chatResponse = await this.handleChatResponse(parsedMessage);
+                      if (chatResponse) {
+                        responseContent = chatResponse;
+                        // 获取到响应后立即关闭连接
+                        isComplete = true;
+                        ws.close();
+                        resolve(responseContent);
                     }
-                }`
-            }
-        };
-
-        ws.send(JSON.stringify(subscribeMessage));
-    }
-
-    // 处理聊天响应
-    static async handleChatResponse(message) {
-        if (message.payload && message.payload.data) {
-            const chatResult = message.payload.data.newSessionChatResult;
-            if (chatResult && chatResult.isEnd == true && chatResult.message) {
-                return chatResult.message.content;
-            }
-        }
-        return null;
-    }
+                      break;
+                  case 'complete':
+                      isComplete = true;
+                      ws.close();
+                      resolve(responseContent);
+                      break;
+              }
+          });
+
+          ws.on('error', (err) => {
+              console.error('WebSocket错误:', err);
+              reject(err);
+          });
+
+          ws.on('close', (code, reason) => {
+              console.log('请求完毕，关闭连接');
+              if (!isComplete) {
+                  reject(new Error('WebSocket closed unexpectedly'));
+              }
+          });
+      });
+  }
+
+  // 发送聊天订阅
+  static sendChatSubscription(ws, requestPayload) {
+      const subscribeMessage = {
+          id: uuidv4(),
+          type: 'subscribe',
+          payload: {
+              variables: {
+                  messageInput: requestPayload.contextMessages,
+                  messageContext: null,
+                  organizationId: 'org_JfjtEvzbwOikUEUn',
+                  integrationId: 'clwtqz9sq001izszu8ms5g4om',
+                  chatMode: 'AUTO',
+                  context: requestPayload.systemMessage,
+                  messageAttributes: {},
+                  includeAIAnnotations: false,
+                  environment: 'production'
+              },
+              extensions: {},
+              operationName: 'OnNewSessionChatResult',
+              query: `subscription OnNewSessionChatResult($messageInput: String!, $messageContext: String, $organizationId: ID!, $integrationId: ID, $chatMode: ChatMode, $filters: ChatFiltersInput, $messageAttributes: JSON, $tags: [String!], $workflowId: String, $context: String, $guidance: String, $includeAIAnnotations: Boolean!, $environment: String) {
+                  newSessionChatResult(
+                      input: {messageInput: $messageInput, messageContext: $messageContext, organizationId: $organizationId, integrationId: $integrationId, chatMode: $chatMode, filters: $filters, messageAttributes: $messageAttributes, tags: $tags, workflowId: $workflowId, context: $context, guidance: $guidance, environment: $environment}
+                  ) {
+                      isEnd
+                      sessionId
+                      message {
+                          id
+                          content
+                      }
+                      __typename
+                  }
+              }`
+          }
+      };
+
+      ws.send(JSON.stringify(subscribeMessage));
+  }
+
+  // 处理聊天响应
+  static async handleChatResponse(message) {
+      if (message.payload && message.payload.data) {
+          const chatResult = message.payload.data.newSessionChatResult;
+          if (chatResult && chatResult.isEnd == true && chatResult.message) {
+              return chatResult.message.content;
+          }
+      }
+      return null;
+  }
 }
 
 // 创建Express应用
@@ -310,67 +302,61 @@ const app = express();
 app.use(express.json({ limit: CONFIG.SERVER.BODY_LIMIT }));
 app.use(express.urlencoded({ extended: true, limit: CONFIG.SERVER.BODY_LIMIT }));
 app.use(cors({
-    origin: '*',
-    methods: ['GET', 'POST', 'OPTIONS'],
-    allowedHeaders: ['Content-Type', 'Authorization']
+  origin: '*',
+  methods: ['GET', 'POST', 'OPTIONS'],
+  allowedHeaders: ['Content-Type', 'Authorization']
 }));
 
 // 获取模型列表路由
 app.get('/hf/v1/models', (req, res) => {
-    res.json({
-        object: "list",
-        data: [{
-            id: "claude-3-5-sonnet-20241022",
-            object: "model",
-            created: Math.floor(Date.now() / 1000),
-            owned_by: "claude",
-        }]
-    });
+  res.json({
+      object: "list",
+      data: [{
+          id: "claude-3-5-sonnet-20241022",
+          object: "model",
+          created: Math.floor(Date.now() / 1000),
+          owned_by: "claude",
+      }]
+  });
 });
 
 // 聊天完成路由
 app.post('/hf/v1/chat/completions', async (req, res) => {
-    try {
-        const { messages, model, stream } = req.body;
-        const authToken = req.headers.authorization?.replace('Bearer ', '');
-
-        if (authToken !== CONFIG.API.API_KEY) {
-            return res.status(401).json({ error: "Unauthorized" });
-        }
-
-        const apiClient = new AiApiClient(req.body.model);
-        const requestPayload = await apiClient.transformMessages(req.body);
-
-        const userMessage = messages.reverse().find(message => message.role === 'user')?.content;
-        if (!userMessage) {
-            return res.status(400).json({ error: "缺失用户消息" });
-        }
-
-        const responseContent = await WebSocketUtils.createWebSocketClient(requestPayload);
-
-        if (stream) {
-            await ResponseHandler.handleStreamResponse(responseContent, model, res);
-        } else {
-            await ResponseHandler.handleNormalResponse(userMessage, responseContent, model, res);
-        }
-    } catch (error) {
-        console.error('处理请求时发生错误:', error);
-        res.status(500).json({ error: "内部服务器错误", details: error.message });
-    }
+  try {
+      const { messages, model, stream } = req.body;
+      const authToken = req.headers.authorization?.replace('Bearer ', '');
+      
+      if (authToken !== CONFIG.API.API_KEY) {
+          return res.status(401).json({ error: "Unauthorized" });
+      }
+
+      const apiClient = new AiApiClient(req.body.model);
+      const requestPayload = await apiClient.transformMessages(req.body);
+
+      const userMessage = messages.reverse().find(message => message.role === 'user')?.content;
+      if (!userMessage) {
+          return res.status(400).json({ error: "缺失用户消息" });
+      }
+
+      const responseContent = await WebSocketUtils.createWebSocketClient(requestPayload);
+
+      if (stream) {
+          await ResponseHandler.handleStreamResponse(responseContent, model, res);
+      } else {
+          await ResponseHandler.handleNormalResponse(userMessage, responseContent, model, res);
+      }
+  } catch (error) {
+      console.error('处理请求时发生错误:', error);
+      res.status(500).json({ error: "内部服务器错误", details: error.message });
+  }
 });
 
 // 404处理
 app.use((req, res) => {
-    res.status(404).json({ message: "请使用正确请求路径" });
-});
-
-// 全局错误处理（新增）
-app.use((err, req, res, next) => {
-    console.error(err.stack);
-    res.status(500).json({ error: '服务器内部错误', details: err.message });
+  res.status(404).json({ message: "服务创建成功运行中，请根据规则使用正确请求路径" });
 });
 
 // 启动服务器
 app.listen(CONFIG.SERVER.PORT, () => {
-    console.log(`服务器运行在 http://localhost:${CONFIG.SERVER.PORT}`);
-});
+  console.log(`服务器运行在 http://localhost:${CONFIG.SERVER.PORT}`);
+});