Th3BossC's picture
initial commit
23d152f
raw
history blame
546 Bytes
from flask import Blueprint, request
from flask_restful import Api, Resource
from chatBot.common.utils import getAnswerLlama, getAnswerGpt
resources = Blueprint('resources', __name__)
api = Api(resources)
class Gpt(Resource):
def get(self):
question = request.json['question']
return {'answer': getAnswerGpt(question)}
api.add_resource(Gpt, '/gpt')
class Llama(Resource):
def get(self):
question = request.json['question']
return {'answer': getAnswerLlama(question)}
api.add_resource(Llama, '/llama')