Fix for handling OpenAI ChatCompletionMessage

uwcirg · Oct 29, 2024 · dc1c506 · dc1c506
1 parent 86d7974
commit dc1c506
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -18,8 +18,8 @@ def chat():
         model="gpt-4o-mini-2024-07-18",
         messages=data['messages']
     )
-    # return jsonify(response.choices[0].message)
-    return response.choices[0].message
+    message_dict = response.choices[0].message.to_dict()
+    return jsonify(message_dict)
 
 @app.route('/api/go', methods=['GET'])
 def go():