Update main.py

main.py

@@ -279,11 +279,25 @@ async def text_to_speech(
 # ============================================================================
 
 class PresentationChatModel(BaseModel):
-    prompt: str
-    conversation_id: Optional[str] = None
-    model_id: str
-    user_id: str
-    
+    prompt: str = Field(..., description="User's query or prompt")
+    model_id: ModelID = Field(
+        default="openai/gpt-4o-mini",
+        description="ID of the model to use for response generation"
+    )
+    conversation_id: Optional[str] = Field(None, description="Unique identifier for the conversation")
+    user_id: str = Field(..., description="Unique identifier for the user")
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "prompt": "Help me create a presentation for my healthy snacks startup",
+                "model_id": "openai/gpt-4o-mini",
+                "conversation_id": "123e4567-e89b-12d3-a456-426614174000",
+                "user_id": "user123"
+            }
+        }
+
+
 # Enum for output formats
 class OutputFormatEnum(str, Enum):
     html = "html"