Update apis/chat_api.py

apis/chat_api.py

@@ -6,6 +6,7 @@ import uvicorn
 
 from pathlib import Path
 from fastapi import FastAPI, Depends
+from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from pydantic import BaseModel, Field
@@ -25,11 +26,19 @@ class ChatAPIApp:
             swagger_ui_parameters={"defaultModelsExpandDepth": -1},
             version="1.0",
         )
+        self.setup_cors()
         self.setup_routes()
 
+    def setup_cors(self):
+        self.app.add_middleware(
+            CORSMiddleware,
+            allow_origins=["*"],  # Adjust this to specify allowed origins
+            allow_credentials=True,
+            allow_methods=["*"],
+            allow_headers=["*"],
+        )
+
     def get_available_models(self):
-        # https://platform.openai.com/docs/api-reference/models/list
-        # ANCHOR[id=available-models]: Available models
         self.available_models = {
             "object": "list",
             "data": [
@@ -121,7 +130,6 @@ class ChatAPIApp:
         streamer = MessageStreamer(model=item.model)
         composer = MessageComposer(model=item.model)
         composer.merge(messages=item.messages)
-        # streamer.chat = stream_chat_mock
 
         stream_response = streamer.chat_response(
             prompt=composer.merged_str,
@@ -220,4 +228,4 @@ if __name__ == "__main__":
     # python -m apis.chat_api      # [Docker] on product mode
     # python -m apis.chat_api -d   # [Dev]    on develop mode
 
-os.system("python -m apis.chat_api")        
+os.system("python -m apis.chat_api")