Niansuh commited on
Commit
d47936e
·
verified ·
1 Parent(s): b8e584a

Update api/utils.py

Browse files
Files changed (1) hide show
  1. 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 # Import the asynchronous getHid function
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
+