Update api/utils.py
Browse files- api/utils.py +2 -2
api/utils.py
CHANGED
|
@@ -177,8 +177,8 @@ async def process_streaming_response(request: ChatRequest):
|
|
| 177 |
yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
|
| 178 |
response_content += cleaned_content # Collect the full response content
|
| 179 |
|
| 180 |
-
# After all chunks are processed, add the advertisement text at the end
|
| 181 |
-
if ADVERTISEMENT_TEXT:
|
| 182 |
response_content += "\n\n" + ADVERTISEMENT_TEXT
|
| 183 |
yield f"data: {json.dumps(create_chat_completion_data(response_content, request.model, timestamp, 'stop'))}\n\n"
|
| 184 |
|
|
|
|
| 177 |
yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
|
| 178 |
response_content += cleaned_content # Collect the full response content
|
| 179 |
|
| 180 |
+
# After all chunks are processed, add the advertisement text at the end only if it's not already present
|
| 181 |
+
if ADVERTISEMENT_TEXT and ADVERTISEMENT_TEXT not in response_content:
|
| 182 |
response_content += "\n\n" + ADVERTISEMENT_TEXT
|
| 183 |
yield f"data: {json.dumps(create_chat_completion_data(response_content, request.model, timestamp, 'stop'))}\n\n"
|
| 184 |
|