Update main.py
Browse files
main.py
CHANGED
@@ -252,7 +252,7 @@ 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 |
-
yield f"data: {json.dumps(create_response(chunk + '
|
256 |
yield "data: [DONE]\n\n"
|
257 |
|
258 |
return StreamingResponse(generate(), media_type="text/event-stream")
|
@@ -264,8 +264,8 @@ async def chat_completions(request: ChatRequest):
|
|
264 |
else:
|
265 |
response_content += chunk
|
266 |
|
267 |
-
# Append "
|
268 |
-
response_content += "
|
269 |
|
270 |
return {
|
271 |
"id": f"chatcmpl-{uuid.uuid4()}",
|
|
|
252 |
image_markdown = f""
|
253 |
yield f"data: {json.dumps(create_response(image_markdown, request.model))}\n\n"
|
254 |
else:
|
255 |
+
yield f"data: {json.dumps(create_response(chunk + '\nNiansuhAI', request.model))}\n\n"
|
256 |
yield "data: [DONE]\n\n"
|
257 |
|
258 |
return StreamingResponse(generate(), media_type="text/event-stream")
|
|
|
264 |
else:
|
265 |
response_content += chunk
|
266 |
|
267 |
+
# Append "\nNiansuhAI" to the final response content
|
268 |
+
response_content += "\nNiansuhAI"
|
269 |
|
270 |
return {
|
271 |
"id": f"chatcmpl-{uuid.uuid4()}",
|