From 4b324015424c08bb21944d2ccf69abd6fcac9f08 Mon Sep 17 00:00:00 2001 From: Cyberes Date: Wed, 30 Aug 2023 20:24:55 -0600 Subject: [PATCH] oops wrong data strucutre --- llm_server/routes/v1/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llm_server/routes/v1/generate.py b/llm_server/routes/v1/generate.py index 668856a..b634151 100644 --- a/llm_server/routes/v1/generate.py +++ b/llm_server/routes/v1/generate.py @@ -8,7 +8,7 @@ from ..request_handler import OobaRequestHandler @bp.route('/generate', methods=['POST']) def generate(): request_valid_json, request_json_body = validate_json(request.data) - if not request_valid_json or not request_json_body.get('prompt') or not request_json_body.get('parameters'): + if not request_valid_json or not request_json_body.get('prompt'): return jsonify({'code': 400, 'msg': 'Invalid JSON'}), 400 else: handler = OobaRequestHandler(request)