ParthSadaria commited on
Commit
0881536
·
verified ·
1 Parent(s): 8842282

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -120,7 +120,7 @@ async def fetch_models():
120
  @app.post("/chat/completions")
121
  @app.post("/v1/chat/completions")
122
  async def get_completion(payload: Payload, request: Request):
123
- model_to_use = "gpt-4o-mini" if payload.model == "gpt-4o" else payload.model
124
  payload_dict = payload.dict()
125
  payload_dict["model"] = model_to_use
126
 
 
120
  @app.post("/chat/completions")
121
  @app.post("/v1/chat/completions")
122
  async def get_completion(payload: Payload, request: Request):
123
+ model_to_use = payload.model
124
  payload_dict = payload.dict()
125
  payload_dict["model"] = model_to_use
126