-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathapp.py
29 lines (24 loc) · 1.1 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import os
from flask import Flask
from flask_cors import CORS
from modules.routes.chat import chat_bp
from modules.routes.session import session_bp
from modules.routes.game import game_bp
from modules.routes.member import member_bp
app = Flask(__name__)
CORS(app, supports_credentials=True)
app.config["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
app.config["GPT_MODEL_ID"] = os.getenv("GPT_MODEL_ID", "ft:gpt-3.5-turbo-0125:personal::AN9MQ5gR1")
app.config["MODEL_ID"] = os.getenv("MODEL_ID", "MLP-KTLim/llama-3-Korean-Bllossom-8B")
app.config["MODEL_PATH"] = os.getenv("MODEL_PATH", "llama3/ft2_invest")
app.config["DEBUG"] = os.getenv("DEBUG", "False").lower() == "true"
app.config["PORT"] = int(os.getenv("PORT", 3309))
# Add blueprints
app.register_blueprint(chat_bp)
app.register_blueprint(session_bp)
app.register_blueprint(game_bp, url_prefix='/game')
app.register_blueprint(member_bp, url_prefix='/members')
if __name__ == "__main__":
if not app.config["OPENAI_API_KEY"]:
raise ValueError("OPENAI_API_KEY not set")
app.run(host="127.0.0.1", port=app.config["PORT"], debug=app.config["DEBUG"])