embedchain

1
import os
2

3
from flask import Blueprint, jsonify, make_response, request
4
from models import APIKey
5
from paths import DB_DIRECTORY_OPEN_AI
6

7
from embedchain import App
8

9
chat_response_bp = Blueprint("chat_response", __name__)
10

11

12
# Chat Response for user query
13
@chat_response_bp.route("/api/get_answer", methods=["POST"])
14
def get_answer():
15
    try:
16
        data = request.get_json()
17
        query = data.get("query")
18
        embedding_model = data.get("embedding_model")
19
        app_type = data.get("app_type")
20

21
        if embedding_model == "open_ai":
22
            os.chdir(DB_DIRECTORY_OPEN_AI)
23
            api_key = APIKey.query.first().key
24
            os.environ["OPENAI_API_KEY"] = api_key
25
            if app_type == "app":
26
                chat_bot = App()
27

28
        response = chat_bot.chat(query)
29
        return make_response(jsonify({"response": response}), 200)
30

31
    except Exception as e:
32
        return make_response(jsonify({"error": str(e)}), 400)
33