| 123456789101112131415161718192021222324252627 |
- import os
- from flask import Blueprint, jsonify, make_response, request
- from models import APIKey
- from paths import DB_DIRECTORY_OPEN_AI
- from embedchain import App
- sources_bp = Blueprint("sources", __name__)
- # API route to add data sources
- @sources_bp.route("/api/add_sources", methods=["POST"])
- def add_sources():
- try:
- embedding_model = request.json.get("embedding_model")
- name = request.json.get("name")
- value = request.json.get("value")
- if embedding_model == "open_ai":
- os.chdir(DB_DIRECTORY_OPEN_AI)
- api_key = APIKey.query.first().key
- os.environ["OPENAI_API_KEY"] = api_key
- chat_bot = App()
- chat_bot.add(name, value)
- return make_response(jsonify(message="Sources added successfully"), 200)
- except Exception as e:
- return make_response(jsonify(message=f"Error adding sources: {str(e)}"), 400)
|
