File size: 546 Bytes
23d152f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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')