Update main.py

main.py

@@ -23,17 +23,13 @@ API_KEYS = [
     os.getenv("API_GEMINI_2"),
     os.getenv("API_GEMINI_3"), 
     os.getenv("API_GEMINI_4"), 
-    os.getenv("API_GEMINI_5") 
+    os.getenv("API_GEMINI_5"), 
 ]
 
 # Classi Pydantic di VALIDAZIONE Body
-class Message(BaseModel):
-    role: Any
-    content: Any
-
 class ChatCompletionRequest(BaseModel):
     model: str = "gemini-2.0-flash" 
-    messages: List[Message]
+    messages: Optional[Any]
     max_tokens: Optional[int] = 8196
     temperature: Optional[float] = 0.8
     stream: Optional[bool] = False
@@ -125,4 +121,8 @@ async def chat_completions(req: ChatCompletionRequest):
         else:
             return StreamingResponse(_resp_async_generator(req), media_type="application/x-ndjson")            
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)