matthoffner commited on
Commit
65c273f
1 Parent(s): ebbbded

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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: ChatCompletionRequest, response_mode=None):
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