Update utils/server/index.ts

utils/server/index.ts

@@ -73,38 +73,19 @@ export const LLMStream = async (
 
   const stream = new ReadableStream({
     async start(controller) {
-      const onParse = (event: any) => {
+      const onParse = (event: ParsedEvent | ReconnectInterval) => {
         if (event.type === 'event') {
           const data = event.data;
 
           try {
             const json = JSON.parse(data);
-
-            // New condition to handle Python API format
-            if (json.choices && json.choices[0].message) {
-              const text = json.choices[0].message.content;
-              const finishReason = json.choices[0].finish_reason;
-
-              if (finishReason === 'stop') {
-                controller.close();
-                return;
-              }
-              
-              const queue = encoder.encode(text);
-              controller.enqueue(queue);
-            }
-            // Old condition to handle existing format
-            else if (json.choices && json.choices[0].delta) {
-              if (json.choices[0].finish_reason != null) {
-                controller.close();
-                return;
-              }
-              
-              const text = json.choices[0].delta.content;
-              const queue = encoder.encode(text);
-              controller.enqueue(queue);
+            if (json.choices[0].finish_reason != null) {
+              controller.close();
+              return;
             }
-
+            const text = json.choices[0].delta.content;
+            const queue = encoder.encode(text);
+            controller.enqueue(queue);
           } catch (e) {
             controller.error(e);
           }
@@ -119,7 +100,5 @@ export const LLMStream = async (
     },
   });
 
-
-
   return stream;
 };