Niansuh commited on
Commit
11cade3
·
verified ·
1 Parent(s): 2f8aeba

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -252,7 +252,8 @@ async def chat_completions(request: ChatRequest):
252
  image_markdown = f"![image]({chunk.url})"
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")
 
252
  image_markdown = f"![image]({chunk.url})"
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")