Update api/utils.py

api/utils.py

@@ -148,9 +148,9 @@ 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"
 
-                # Only add the advertisement once, after the entire stream is processed
+                # Ensure that the advertisement is added after the full response is processed
                 if ADVERTISEMENT_TEXT and not advertisement_added:
-                    response_content += "\n\n" + ADVERTISEMENT_TEXT  # Line break added here
+                    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
@@ -166,6 +166,7 @@ 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()}"