Update main.py

main.py

@@ -41,26 +41,56 @@ async def chat_completion(request: ChatRequest):
     }
 
     if request.stream:
-        def event_stream():
-            sent_done = False
-            with requests.post(
-                "https://www.chatwithmono.xyz/api/chat",
-                headers=headers,
-                json=payload,
-                stream=True,
-                timeout=120
-            ) as response:
-                for line in response.iter_lines(decode_unicode=True):
-                    if line.startswith("0:"):
-                        try:
-                            content_piece = json.loads(line[2:])
-                            yield f'data: {json.dumps({"choices": [{"delta": {"content": content_piece}, "finish_reason": None}]})}\n\n'
-                        except:
-                            continue
-                    elif line.startswith(("e:", "d:")) and not sent_done:
-                        sent_done = True
-                        yield "data: [DONE]\n\n"
-        return StreamingResponse(event_stream(), media_type="text/event-stream")
+    def event_stream():
+        chat_id = f"chatcmpl-{unix_id()}"
+        created = int(time.time())
+        sent_done = False
+
+        with requests.post(
+            "https://www.chatwithmono.xyz/api/chat",
+            headers=headers,
+            json=payload,
+            stream=True,
+            timeout=120
+        ) as response:
+            for line in response.iter_lines(decode_unicode=True):
+                if line.startswith("0:"):
+                    try:
+                        content_piece = json.loads(line[2:])
+                        chunk_data = {
+                            "id": chat_id,
+                            "object": "chat.completion.chunk",
+                            "created": created,
+                            "model": request.model,
+                            "choices": [
+                                {
+                                    "delta": {"content": content_piece},
+                                    "index": 0,
+                                    "finish_reason": None
+                                }
+                            ]
+                        }
+                        yield f"data: {json.dumps(chunk_data)}\n\n"
+                    except:
+                        continue
+                elif line.startswith(("e:", "d:")) and not sent_done:
+                    sent_done = True
+                    done_chunk = {
+                        "id": chat_id,
+                        "object": "chat.completion.chunk",
+                        "created": created,
+                        "model": request.model,
+                        "choices": [
+                            {
+                                "delta": {},
+                                "index": 0,
+                                "finish_reason": "stop"
+                            }
+                        ]
+                    }
+                    yield f"data: {json.dumps(done_chunk)}\n\ndata: [DONE]\n\n"
+
+    return StreamingResponse(event_stream(), media_type="text/event-stream")
 
     else:
         assistant_response = ""