Update main.py

main.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, List, Optional
 import httpx
 import uvicorn
 from dotenv import load_dotenv
-from fastapi import FastAPI, HTTPException, Depends
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from starlette.responses import StreamingResponse
 
@@ -19,11 +19,10 @@ logger = logging.getLogger(__name__)
 
 load_dotenv()
 app = FastAPI()
-APP_SECRET = os.getenv("APP_SECRET", "666")
 ACCESS_TOKEN = os.getenv("SD_ACCESS_TOKEN", "")
 headers = {
     'authorization': f'Bearer {ACCESS_TOKEN}',
-    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
+    'user-agent': 'Mozilla/5.0',
 }
 
 ALLOWED_MODELS = [
@@ -75,11 +74,12 @@ async def chat_completions(request: ChatRequest):
         'prompt': "\n".join(
             [f"{'User' if msg.role == 'user' else 'Assistant'}: {msg.content}" for msg in request.messages]
         ),
-        'stream': True,
+        'stream': request.stream,
         'model': request.model,
     }
     
-    logger.info(f"Sending request to external API with data: {json_data}")
+    logger.info(f"Sending request to external API with model: {request.model}")
+    logger.debug(f"Request payload: {json.dumps(json_data, indent=2)}")
 
     async def generate():
         async with httpx.AsyncClient() as client: