Fix user creation

Josh-XT · Dec 5, 2023 · 4868d44 · 4868d44
1 parent cc411aa
commit 4868d44
Showing 1 changed file with 7 additions and 4 deletions.
diff --git 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