Update api/utils.py

api/utils.py

@@ -15,7 +15,6 @@ from api.config import (
     AGENT_MODE,
     TRENDING_AGENT_MODE,
     MODEL_PREFIXES,
-    MODEL_REFERERS
 )
 from api.models import ChatRequest
 from api.logger import setup_logger
@@ -124,17 +123,18 @@ async def process_streaming_response(request: ChatRequest):
 
                         cleaned_content = strip_model_prefix(content, model_prefix)
 
-                        # Add the chunk to the response content
+                        # Add chunk to the response content
                         response_content += cleaned_content
 
-                        # Yield each chunk of content
+                        # Only yield cleaned content in chunks
                         yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
 
-                # Append advertisement at the end
+                # Append advertisement at the end, ensuring it happens only once
                 if ADVERTISEMENT_TEXT and not advertisement_added:
                     response_content += "\n\n" + ADVERTISEMENT_TEXT
                     advertisement_added = True
 
+                # Final chunk with complete content
                 yield f"data: {json.dumps(create_chat_completion_data(response_content, request.model, timestamp, 'stop'))}\n\n"
                 yield "data: [DONE]\n\n"