Update main.py
Browse files
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 |
-
|
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
|
385 |
async_generator = Blackbox.create_async_generator(
|
386 |
model=request.model,
|
387 |
-
messages=[{"role": msg.role, "content":
|
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
|