Update main.py
Browse files
main.py
CHANGED
@@ -252,7 +252,8 @@ async def chat_completions(request: ChatRequest):
|
|
252 |
image_markdown = f""
|
253 |
yield f"data: {json.dumps(create_response(image_markdown, request.model))}\n\n"
|
254 |
else:
|
255 |
-
|
|
|
256 |
yield "data: [DONE]\n\n"
|
257 |
|
258 |
return StreamingResponse(generate(), media_type="text/event-stream")
|
|
|
252 |
image_markdown = f""
|
253 |
yield f"data: {json.dumps(create_response(image_markdown, request.model))}\n\n"
|
254 |
else:
|
255 |
+
response_with_footer = f"{chunk}\nNiansuhAI" # Concatenating here
|
256 |
+
yield f"data: {json.dumps(create_response(response_with_footer, request.model))}\n\n"
|
257 |
yield "data: [DONE]\n\n"
|
258 |
|
259 |
return StreamingResponse(generate(), media_type="text/event-stream")
|