Update main.py

main.py

@@ -3,7 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import openai
-from typing import List, Optional
+from typing import List, Optional,Union
 import logging
 from itertools import cycle
 import asyncio