From 7ded1318f75dccedc2c5977e4c0952083921ca31 Mon Sep 17 00:00:00 2001
From: multipleof4 <planetrenox@protonmail.com>
Date: Sat, 1 Nov 2025 14:08:12 -0700
Subject: [PATCH] Fix: Add newline after thoughts and enable for Google

---
 index.js | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/index.js b/index.js
index 96ca3ea..ca79e10 100644
--- a/index.js
+++ b/index.js
@@ -272,14 +272,26 @@ export class MyDurableObject {
     if (!resp.ok) throw new Error(`Google API error: ${resp.status} ${await resp.text()}`);
     const reader = resp.body.getReader();
     const decoder = new TextDecoder();
-    let buffer = '';
+    let buffer = '', hasReasoning = false, hasContent = false;
     while (this.phase === 'running') {
       const { done, value } = await reader.read();
       if (done) break;
       buffer += decoder.decode(value, { stream: true });
       for (const line of buffer.split('\n')) {
         if (!line.startsWith('data: ')) continue;
-        try { JSON.parse(line.substring(6))?.candidates?.[0]?.content?.parts?.forEach(p => p.text && this.queueDelta(p.text)); } catch {}
+        try {
+          JSON.parse(line.substring(6))?.candidates?.[0]?.content?.parts?.forEach(p => {
+            if (p.thought?.thought) {
+              this.queueDelta(p.thought.thought);
+              hasReasoning = true;
+            }
+            if (p.text) {
+              if (hasReasoning && !hasContent) this.queueDelta('\n');
+              this.queueDelta(p.text);
+              hasContent = true;
+            }
+          });
+        } catch {}
       }
       buffer = buffer.slice(buffer.lastIndexOf('\n') + 1);
     }
@@ -288,11 +300,19 @@ export class MyDurableObject {
   async streamOpenRouter({ apiKey, body }) {
     const client = new OpenAI({ apiKey, baseURL: 'https://openrouter.ai/api/v1' });
     const stream = await client.chat.completions.create({ ...body, stream: true }, { signal: this.controller.signal });
+    let hasReasoning = false, hasContent = false;
     for await (const chunk of stream) {
       if (this.phase !== 'running') break;
       const delta = chunk?.choices?.[0]?.delta;
-      if (delta?.reasoning && body.reasoning?.exclude !== true) this.queueDelta(delta.reasoning);
-      if (delta?.content) this.queueDelta(delta.content);
+      if (delta?.reasoning && body.reasoning?.exclude !== true) {
+        this.queueDelta(delta.reasoning);
+        hasReasoning = true;
+      }
+      if (delta?.content) {
+        if (hasReasoning && !hasContent) this.queueDelta('\n');
+        this.queueDelta(delta.content);
+        hasContent = true;
+      }
     }
   }