Spaces:
Running
Running
Update main.py
Browse files
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)
|