bikin api
unknown
python
2 years ago
1.2 kB
13
Indexable
import json
import os
from embedchain import App
from flask import Flask, request, jsonify
os.environ["OPENAI_API_KEY"] = "sk-apikeyku"
app = Flask(__name__)
with open('data.json', 'r', encoding='utf-8') as file:
data = json.load(file)
marbot = App.from_config(config={
"llm": {
"provider": "openai",
"config": {
"model": "gpt-3.5-turbo",
"temperature": 0.0,
"top_p": 1,
"stream": False,
},
},
"embedder": {
"provider": "openai",
"config": {
"model": "text-embedding-ada-002",
}
},
})
marbot.add("data.json")
@app.route('/chat', methods=['POST'])
def chat():
data = request.json
query = data.get('query', '')
if not query:
return jsonify({"error": "No query provided"}), 400
try:
response = marbot.chat(query)
print(response)
return jsonify({"response": response})
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__":
app.run(debug=True,
host="0.0.0.0",
port=int(os.environ.get("PORT", 8000)))Editor is loading...
Leave a Comment