Update app.js

app.js

@@ -10,12 +10,107 @@ const NOTDIAMOND_HEADERS = {
   'Content-Type': 'application/json',
   'next-action': '8189eb37107121e024940f588629a394a594e6a4'
 };
-const AUTH_KEY = process.env.AUTH_KEY || 'sk-linuxdo';
+const AUTH_KEY = process.env.AUTH_KEY;
 
-// 默认模型，如果用户没有指定模型，就使用这个
 const DEFAULT_MODEL = 'gpt-4-turbo-2024-04-09';
 
-// 保留其他函数定义：createOpenAIChunk, streamNotdiamondResponse
+function createOpenAIChunk(content, model, finishReason = null) {
+  return {
+    id: `chatcmpl-${crypto.randomUUID()}`,
+    object: "chat.completion.chunk",
+    created: Math.floor(Date.now() / 1000),
+    model: model,
+    system_fingerprint: "fp_4e2b2da518",
+    choices: [
+      {
+        index: 0,
+        delta: content ? { content: content } : {},
+        logprobs: null,
+        finish_reason: finishReason
+      }
+    ],
+    usage: null
+  };
+}
+
+async function* streamNotdiamondResponse(response, model) {
+  const reader = response.body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  let lastContent = "";
+
+  function processDollars(str) {
+    return str.replace(/\${2,}/, match => match.slice(1));
+  }
+
+  while (true) {
+    const { done, value } = await reader.read();
+    if (done) break;
+
+    buffer += decoder.decode(value, { stream: true });
+    let lines = buffer.split('\n');
+    buffer = lines.pop() || "";
+
+    for (const line of lines) {
+      if (line.trim() === '') continue;
+      
+      try {
+        const jsonMatch = line.match(/\{.*\}/);
+        if (jsonMatch) {
+          const data = JSON.parse(jsonMatch[0]);
+          let content = '';
+          
+          if (data.output?.curr !== undefined) {
+            content = processDollars(data.output.curr);
+            lastContent = content;
+          } else if (data.curr !== undefined) {
+            content = processDollars(data.curr);
+            lastContent = content;
+          } else if (data.diff !== undefined && Array.isArray(data.diff) && data.diff.length > 1) {
+            const newContent = processDollars(data.diff[1]);
+            content = lastContent + newContent;
+            lastContent = content;
+          } else if (data.diff !== undefined && Array.isArray(data.diff) && data.diff.length === 1) {
+            content = lastContent;  // If diff is empty, use the last content
+          }
+          
+          if (content) {
+            yield createOpenAIChunk(content, model);
+          }
+        }
+      } catch (error) {
+        console.error('Error processing line:', error, 'Raw line:', line);
+      }
+    }
+  }
+
+  if (buffer.trim()) {
+    try {
+      const jsonMatch = buffer.match(/\{.*\}/);
+      if (jsonMatch) {
+        const data = JSON.parse(jsonMatch[0]);
+        let content = '';
+        if (data.output?.curr !== undefined) {
+          content = processDollars(data.output.curr);
+        } else if (data.curr !== undefined) {
+          content = processDollars(data.curr);
+        } else if (data.diff !== undefined && Array.isArray(data.diff) && data.diff.length > 1) {
+          const newContent = processDollars(data.diff[1]);
+          content = lastContent + newContent;
+        } else if (data.diff !== undefined && Array.isArray(data.diff) && data.diff.length === 1) {
+          content = lastContent;
+        }
+        if (content) {
+          yield createOpenAIChunk(content, model);
+        }
+      }
+    } catch (error) {
+      console.error('Error processing final buffer:', error);
+    }
+  }
+
+  yield createOpenAIChunk('', model, 'stop');
+}
 
 app.use(express.json());