Update main.py

main.py

@@ -9,7 +9,7 @@ import base64
 from aiohttp import ClientSession, ClientTimeout, ClientError
 from fastapi import FastAPI, HTTPException, Request
 from pydantic import BaseModel
-from typing import List, Dict, Any, Optional, AsyncGenerator
+from typing import List, Dict, Any, Optional, AsyncGenerator, Union
 from datetime import datetime
 from fastapi.responses import StreamingResponse
 
@@ -371,6 +371,20 @@ def create_response(content: str, model: str, finish_reason: Optional[str] = Non
 async def chat_completions(request: ChatRequest, req: Request):
     logger.info(f"Received chat completions request: model='{request.model}' messages={request.messages} stream={request.stream} webSearchMode={request.webSearchMode} image={request.image}")
     try:
+        # Validate that all messages have string content
+        for idx, msg in enumerate(request.messages):
+            if not isinstance(msg.content, str):
+                logger.error(f"Message at index {idx} has invalid content type: {type(msg.content)}")
+                raise HTTPException(
+                    status_code=422,
+                    detail=[{
+                        "type": "string_type",
+                        "loc": ["body", "messages", idx, "content"],
+                        "msg": "Input should be a valid string",
+                        "input": msg.content
+                    }]
+                )
+
         # Convert Pydantic messages to dicts
         messages = [{"role": msg.role, "content": msg.content} for msg in request.messages]