Update api/utils.py

api/utils.py

@@ -142,6 +142,8 @@ async def process_streaming_response(request: ChatRequest):
     }
 
     response_content = ""  # Variable to hold the full response content
+    advertisement_added = False  # Track if advertisement is added
+
     async with httpx.AsyncClient() as client:
         try:
             async with client.stream(
@@ -178,8 +180,9 @@ async def process_streaming_response(request: ChatRequest):
                         yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
 
                 # After all chunks are processed, append the advertisement text at the end if it's not already included
-                if ADVERTISEMENT_TEXT and ADVERTISEMENT_TEXT not in response_content:
+                if ADVERTISEMENT_TEXT and not advertisement_added:
                     response_content += "\n\n" + ADVERTISEMENT_TEXT
+                    advertisement_added = True
 
                 # Now yield the final chunk with the advertisement text appended at the end
                 yield f"data: {json.dumps(create_chat_completion_data(response_content, request.model, timestamp, 'stop'))}\n\n"