Update api/utils.py

api/utils.py

@@ -148,12 +148,12 @@ async def process_streaming_response(request: ChatRequest):
                         # Yield the cleaned chunk as part of the stream
                         yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
 
-                # Ensure that the advertisement is added after the full response is processed
+                # After processing all chunks, add the advertisement once (at the end)
                 if ADVERTISEMENT_TEXT and not advertisement_added:
                     response_content += "\n\n" + ADVERTISEMENT_TEXT  # Add line break here
                     advertisement_added = True
 
-                # Yield the final chunk with the advertisement text appended at the end
+                # Yield the final chunk with the advertisement text appended at the end (this is the final response)
                 yield f"data: {json.dumps(create_chat_completion_data(response_content, request.model, timestamp, 'stop'))}\n\n"
 
                 # Add the final "done" marker
@@ -166,7 +166,6 @@ async def process_streaming_response(request: ChatRequest):
             logger.error(f"Error occurred during request for Request ID {request_id}: {e}")
             raise HTTPException(status_code=500, detail=str(e))
 
-
 # Process non-streaming response with headers from config.py
 async def process_non_streaming_response(request: ChatRequest):
     request_id = f"chatcmpl-{uuid.uuid4()}"