jonathanjordan21 commited on
Commit
c70c206
1 Parent(s): bd6a555

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +5 -1
apis/chat_api.py CHANGED
@@ -212,7 +212,11 @@ class ChatAPIApp:
212
  data_response = {
213
  "model": data_response.get('model'),
214
  "created_at": data_response.get('created'),
215
- "response": data_response["choices"][0]["message"]["content"],
 
 
 
 
216
  "done": True,
217
  }
218
  return data_response
 
212
  data_response = {
213
  "model": data_response.get('model'),
214
  "created_at": data_response.get('created'),
215
+ "message": {
216
+ "role": "assistant",
217
+ "content": data_response["choices"][0]["message"]["content"],
218
+ },
219
+ # "response": data_response["choices"][0]["message"]["content"],
220
  "done": True,
221
  }
222
  return data_response