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

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +2 -4
api/utils.py CHANGED
@@ -203,6 +203,7 @@ async def process_non_streaming_response(request: ChatRequest):
203
  response.raise_for_status()
204
  async for chunk in response.aiter_text():
205
  full_response += chunk
 
206
  except httpx.HTTPStatusError as e:
207
  logger.error(f"HTTP error occurred for Request ID {request_id}: {e}")
208
  raise HTTPException(status_code=e.response.status_code, detail=str(e))
@@ -215,12 +216,10 @@ async def process_non_streaming_response(request: ChatRequest):
215
 
216
  if BLOCKED_MESSAGE in full_response:
217
  full_response = full_response.replace(BLOCKED_MESSAGE, '').strip()
218
- if not full_response:
219
- raise HTTPException(status_code=500, detail="Blocked message detected in response.")
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,4 +237,3 @@ async def process_non_streaming_response(request: ChatRequest):
238
  ],
239
  "usage": None,
240
  }
241
-
 
203
  response.raise_for_status()
204
  async for chunk in response.aiter_text():
205
  full_response += chunk
206
+
207
  except httpx.HTTPStatusError as e:
208
  logger.error(f"HTTP error occurred for Request ID {request_id}: {e}")
209
  raise HTTPException(status_code=e.response.status_code, detail=str(e))
 
216
 
217
  if BLOCKED_MESSAGE in full_response:
218
  full_response = full_response.replace(BLOCKED_MESSAGE, '').strip()
 
 
219
 
220
  cleaned_full_response = strip_model_prefix(full_response, model_prefix)
221
 
222
+ # Append advertisement once
223
  if ADVERTISEMENT_TEXT:
224
  cleaned_full_response += "\n\n" + ADVERTISEMENT_TEXT
225
 
 
237
  ],
238
  "usage": None,
239
  }