matthoffner
commited on
Commit
•
65c273f
1
Parent(s):
ebbbded
Update main.py
Browse files
main.py
CHANGED
@@ -56,7 +56,7 @@ class ChatCompletionRequest(BaseModel):
|
|
56 |
max_tokens: int = 250
|
57 |
|
58 |
@app.post("/v1/completions")
|
59 |
-
async def completion(request:
|
60 |
response = llm(request.prompt)
|
61 |
return response
|
62 |
|
|
|
56 |
max_tokens: int = 250
|
57 |
|
58 |
@app.post("/v1/completions")
|
59 |
+
async def completion(request: ChatCompletionRequestV0, response_mode=None):
|
60 |
response = llm(request.prompt)
|
61 |
return response
|
62 |
|