Update main.py

main.py

@@ -70,5 +70,27 @@ def chat_completions():
 
     return app.response_class(stream(), mimetype='text/event-stream')
 
+
+@app.route("/v1/chat/completions", methods=['POST'])
+def v_chat_completions():
+    streaming = request.json.get('stream', False)
+    model = request.json.get('model', 'gpt-3.5-turbo')
+    messages = request.json.get('messages')
+    response = ''
+
+    try:
+        response = closeai.ChatCompletion.create(model=model, stream=streaming, messages=messages)
+    except Exception as er:
+        return er
+    if not streaming:
+        return jsonify(response)
+
+    def stream():
+        for token in response:
+            completion_data = token
+            yield 'data: %s\n\n' % json.dumps(completion_data, separators=(',' ':'))
+    return app.response_class(stream(), mimetype='text/event-stream')
+
+
 if __name__ == '__main__':
     app.run(**settings['web'])