Update main.py

main.py

@@ -240,8 +240,7 @@ async def chat_completions(request: ChatRequest):
             model=request.model,
             messages=messages,
             image=None,  # Pass the image if required
-            image_name=None
-  # Pass image name if required
+            image_name=None  # Pass image name if required
         )
     except ModelNotWorkingException as e:
         raise HTTPException(status_code=503, detail=str(e))
@@ -253,7 +252,7 @@ async def chat_completions(request: ChatRequest):
                     image_markdown = f"![image]({chunk.url})"
                     yield f"data: {json.dumps(create_response(image_markdown, request.model))}\n\n"
                 else:
-                    yield f"data: {json.dumps(create_response(chunk, request.model))}\n\n"
+                    yield f"data: {json.dumps(create_response(chunk + ' NiansuhAI', request.model))}\n\n"
             yield "data: [DONE]\n\n"
 
         return StreamingResponse(generate(), media_type="text/event-stream")
@@ -264,6 +263,9 @@ async def chat_completions(request: ChatRequest):
                 response_content += f"![image]({chunk.url})\n"
             else:
                 response_content += chunk
+        
+        # Append "NiansuhAI" to the final response content
+        response_content += " NiansuhAI"
 
         return {
             "id": f"chatcmpl-{uuid.uuid4()}",
@@ -283,6 +285,7 @@ async def chat_completions(request: ChatRequest):
             "usage": None,
         }
 
+
 @app.get("/niansuhai/v1/models")
 async def get_models():
     return {"models": Blackbox.models}