Update api/utils.py
Browse files- api/utils.py +3 -1
api/utils.py
CHANGED
@@ -18,7 +18,7 @@ from api.config import (
|
|
18 |
)
|
19 |
from api.models import ChatRequest
|
20 |
from api.logger import setup_logger
|
21 |
-
from api.validate import getHid
|
22 |
|
23 |
logger = setup_logger(__name__)
|
24 |
|
@@ -220,6 +220,7 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
220 |
|
221 |
cleaned_full_response = strip_model_prefix(full_response, model_prefix)
|
222 |
|
|
|
223 |
if ADVERTISEMENT_TEXT:
|
224 |
cleaned_full_response += "\n\n" + ADVERTISEMENT_TEXT
|
225 |
|
@@ -237,3 +238,4 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
237 |
],
|
238 |
"usage": None,
|
239 |
}
|
|
|
|
18 |
)
|
19 |
from api.models import ChatRequest
|
20 |
from api.logger import setup_logger
|
21 |
+
from api.validate import getHid
|
22 |
|
23 |
logger = setup_logger(__name__)
|
24 |
|
|
|
220 |
|
221 |
cleaned_full_response = strip_model_prefix(full_response, model_prefix)
|
222 |
|
223 |
+
# Append advertisement only once
|
224 |
if ADVERTISEMENT_TEXT:
|
225 |
cleaned_full_response += "\n\n" + ADVERTISEMENT_TEXT
|
226 |
|
|
|
238 |
],
|
239 |
"usage": None,
|
240 |
}
|
241 |
+
|