jonathanjordan21 commited on
Commit
a4e2086
·
verified ·
1 Parent(s): bdd9aa8

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +2 -2
apis/chat_api.py CHANGED
@@ -241,8 +241,8 @@ class ChatAPIApp:
241
  data_response = streamer.chat_return_dict(stream_response)
242
  data_response = {
243
  "model": data_response.get('model'),
244
- "created_at": data_response.get('created_at'),
245
- "response": data_response.get('choices')[-1].get('message').get('content'),
246
  "done": True,
247
  }
248
  return data_response
 
241
  data_response = streamer.chat_return_dict(stream_response)
242
  data_response = {
243
  "model": data_response.get('model'),
244
+ "created_at": data_response.get('created'),
245
+ "response": data_response["choices"][0]["message"]["content"]
246
  "done": True,
247
  }
248
  return data_response