diff options
| -rw-r--r-- | gn3/api/llm.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/gn3/api/llm.py b/gn3/api/llm.py index d22a42c..cb68526 100644 --- a/gn3/api/llm.py +++ b/gn3/api/llm.py @@ -16,6 +16,7 @@ from gn3.auth import db from redis import Redis +import os import json import sqlite3 from datetime import timedelta @@ -23,6 +24,9 @@ from datetime import timedelta GnQNA = Blueprint("GnQNA", __name__) + + + def handle_errors(func): @wraps(func) def decorated_function(*args, **kwargs): @@ -73,7 +77,8 @@ def rating(task_id): results.get("query"), results.get("answer"), results.get("weight", 0)) - with db.connection(current_app.config["DATA_DIR"]) as conn: + + with db.connection(os.path.join(current_app.config["DATA_DIR"],"/llm.db")) as conn: cursor = conn.cursor() create_table = """CREATE TABLE IF NOT EXISTS Rating( user_id INTEGER NOT NULL, |