jonathanjordan21 commited on
Commit
198a4f7
1 Parent(s): 28bce37

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +1 -1
apis/chat_api.py CHANGED
@@ -227,7 +227,7 @@ class ChatAPIApp:
227
  "model": data_response.get('model'),
228
  "created_at": data_response.get('created_at'),
229
  "response": data_response.get('choices')[-1].get('message').get('content'),
230
- "done": True if data_response.get('choices')[-1].get('finish_reason') != None else False,
231
  }
232
  return data_response
233
  except HfApiException as e:
 
227
  "model": data_response.get('model'),
228
  "created_at": data_response.get('created_at'),
229
  "response": data_response.get('choices')[-1].get('message').get('content'),
230
+ "done": True,
231
  }
232
  return data_response
233
  except HfApiException as e: