diff --git a/agixt/db/User.py b/agixt/db/User.py index 4bdee6f6623..182862f45c9 100644 --- a/agixt/db/User.py +++ b/agixt/db/User.py @@ -1,10 +1,13 @@ -from DBConnection import User +from DBConnection import User, get_session import os -def create_user(api_key: str, email: str, role: str) -> bool: +def create_user(api_key: str, email: str, role: str = "user") -> bool: if api_key != os.environ.get("AGIXT_API_KEY"): return {"error": "Invalid API key"}, 401 - user = User(email=email, role=role) - user.save() + session = get_session() + user = User(email=email, role=role.lower()) + session.add(user) + session.commit() + session.close() return {"status": "Success"}, 200