Update main.py

main.py

@@ -17,10 +17,15 @@ from datetime import datetime
 from aiohttp import ClientSession, ClientTimeout, ClientError
 from fastapi import FastAPI, HTTPException, Request, Depends, Header
 from fastapi.responses import StreamingResponse, JSONResponse, RedirectResponse
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 from io import BytesIO
 import base64
 
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -30,10 +35,6 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 # Load environment variables
-from dotenv import load_dotenv
-
-load_dotenv()
-
 API_KEYS = os.getenv('API_KEYS', '').split(',')  # Comma-separated API keys
 RATE_LIMIT = int(os.getenv('RATE_LIMIT', '60'))  # Requests per minute
 AVAILABLE_MODELS = os.getenv('AVAILABLE_MODELS', '')  # Comma-separated available models
@@ -457,6 +458,12 @@ class Message(BaseModel):
     role: str
     content: str
 
+    @validator('content')
+    def content_must_be_string(cls, v):
+        if not isinstance(v, str):
+            raise ValueError('content must be a string')
+        return v
+
 class ChatRequest(BaseModel):
     model: str
     messages: List[Message]