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 |
+
|