Niansuh commited on
Commit
8fedb8b
·
verified ·
1 Parent(s): dd3e05b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -4
main.py CHANGED
@@ -373,18 +373,21 @@ def create_response(content: str, model: str, finish_reason: Optional[str] = Non
373
 
374
  @app.post("/niansuhai/v1/chat/completions", dependencies=[Depends(rate_limiter)])
375
  async def chat_completions(request: ChatRequest, req: Request, api_key: str = Depends(get_api_key)):
376
- logger.info(f"Received chat completions request from API key: {api_key} | Model: {request.model}")
377
-
 
 
 
378
  try:
379
  # Validate that the requested model is available
380
  if request.model not in Blackbox.models and request.model not in Blackbox.model_aliases:
381
  logger.warning(f"Attempt to use unavailable model: {request.model}")
382
  raise HTTPException(status_code=400, detail="Requested model is not available.")
383
 
384
- # Process the request but do not log sensitive content
385
  async_generator = Blackbox.create_async_generator(
386
  model=request.model,
387
- messages=[{"role": msg.role, "content": "[redacted]"} for msg in request.messages], # Redact user messages in logs
388
  image=None,
389
  image_name=None,
390
  webSearchMode=request.webSearchMode
 
373
 
374
  @app.post("/niansuhai/v1/chat/completions", dependencies=[Depends(rate_limiter)])
375
  async def chat_completions(request: ChatRequest, req: Request, api_key: str = Depends(get_api_key)):
376
+ # Redact user messages only for logging purposes
377
+ redacted_messages = [{"role": msg.role, "content": "[redacted]"} for msg in request.messages]
378
+
379
+ logger.info(f"Received chat completions request from API key: {api_key} | Model: {request.model} | Messages: {redacted_messages}")
380
+
381
  try:
382
  # Validate that the requested model is available
383
  if request.model not in Blackbox.models and request.model not in Blackbox.model_aliases:
384
  logger.warning(f"Attempt to use unavailable model: {request.model}")
385
  raise HTTPException(status_code=400, detail="Requested model is not available.")
386
 
387
+ # Process the request with actual message content, but don't log it
388
  async_generator = Blackbox.create_async_generator(
389
  model=request.model,
390
+ messages=[{"role": msg.role, "content": msg.content} for msg in request.messages], # Actual message content used here
391
  image=None,
392
  image_name=None,
393
  webSearchMode=request.webSearchMode