File size: 1,308 Bytes
b2841f8 a177e53 b2841f8 a177e53 b2841f8 a6d9c05 023be4e a6d9c05 023be4e e73e36b 6b3b5e6 e73e36b 023be4e 6b3b5e6 4f66887 a177e53 a6d9c05 023be4e a177e53 b2841f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from flask import Flask, request
from flask_restful import Resource, Api, reqparse
from flask_cors import CORS
from encoder_file import get_encoder
encoder = get_encoder()
app = Flask(__name__)
CORS(app)
api = Api(app)
parser = reqparse.RequestParser()
class Status(Resource):
def get(self):
return {'status': 'Server Running Successfully'}
DATASET_CONSTANT = {
"data": [
"dataset-1",
"dataset-2",
"dataset-3",
"dataset-4",
"dataset-5"
]
}
class Data(Resource):
def get(self):
return DATASET_CONSTANT
class Datas(Resource):
def post(self):
parser.add_argument('prompt', type=str, required=True)
args = parser.parse_args()
prompt_ = args['prompt']
try:
return_output = len(encoder.encode(prompt_))
# old_prompt + new_input
total_consumed = 561 + return_output
return {'status': '1', 'token_legth': return_output,
'char_length': len(prompt_),
'remaining_tokens': 4000 - total_consumed}
except Exception as e:
return {'status': '-1', 'e': e}
api.add_resource(Status, '/')
api.add_resource(Data, '/data')
api.add_resource(Datas, '/get_size')
if __name__ == '__main__':
app.run(debug=True) |