ParthSadaria commited on
Commit
c5cedd6
·
verified ·
1 Parent(s): 1d32d66

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -125,7 +125,9 @@ import cloudscraper
125
  @app.post("/chat/completions")
126
  @app.post("/v1/chat/completions")
127
  async def get_completion(payload: Payload, request: Request):
 
128
  payload_dict = payload.dict()
 
129
  # Select the appropriate endpoint
130
  endpoint = secret_api_endpoint_2 if model_to_use in alternate_models else secret_api_endpoint
131
  print(payload_dict)
 
125
  @app.post("/chat/completions")
126
  @app.post("/v1/chat/completions")
127
  async def get_completion(payload: Payload, request: Request):
128
+ model_to_use = payload.model
129
  payload_dict = payload.dict()
130
+ payload_dict["model"] = model_to_use
131
  # Select the appropriate endpoint
132
  endpoint = secret_api_endpoint_2 if model_to_use in alternate_models else secret_api_endpoint
133
  print(payload_dict)