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