diff --git a/examples/server/api_like_OAI.py b/examples/server/api_like_OAI.py index e07324ba7..aa325a03e 100755 --- a/examples/server/api_like_OAI.py +++ b/examples/server/api_like_OAI.py @@ -165,6 +165,7 @@ def chat_completions(): if (not stream): data = requests.request("POST", urllib.parse.urljoin(args.llama_api, "/completion"), data=json.dumps(postData)) + print(data.json()) resData = make_resData(data.json(), chat=True, promptToken=promptToken) return jsonify(resData) else: @@ -214,12 +215,5 @@ def completion(): yield 'data: {}\n'.format(json.dumps(resData)) return Response(generate(), mimetype='text/event-stream') - - - - - - - if __name__ == '__main__': app.run(args.host, port=args.port)