Niansuh commited on
Commit
b427211
·
verified ·
1 Parent(s): ca3090c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -252,7 +252,7 @@ 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 + ' NiansuhAI', request.model))}\n\n"
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 "NiansuhAI" to the final response content
268
- response_content += " NiansuhAI"
269
 
270
  return {
271
  "id": f"chatcmpl-{uuid.uuid4()}",
 
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")
 
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()}",