ParthSadaria commited on
Commit
924ff1d
·
verified ·
1 Parent(s): 31eab42

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -148,7 +148,7 @@ available_model_ids = [
148
 
149
  @app.post("/chat/completions")
150
  @app.post("/v1/chat/completions")
151
- async def get_completion(payload: Payload):
152
  model_to_use = payload.model if payload.model else "gpt-4o-mini"
153
  # Validate model availability
154
  if model_to_use not in available_model_ids:
 
148
 
149
  @app.post("/chat/completions")
150
  @app.post("/v1/chat/completions")
151
+ async def get_completion(payload: Payload,request: Request):
152
  model_to_use = payload.model if payload.model else "gpt-4o-mini"
153
  # Validate model availability
154
  if model_to_use not in available_model_ids: