From c1d706dc5a47c64a3d7ea77c6a6817f12f5e6528 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sun, 7 Jul 2024 23:01:15 -0700
Subject: [PATCH] fix: db issues

---
 backend/apps/webui/models/auths.py | 2 --
 backend/apps/webui/models/chats.py | 5 +++--
 backend/apps/webui/models/users.py | 1 +
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/backend/apps/webui/models/auths.py b/backend/apps/webui/models/auths.py
index 616eb70f4..a17d92ff1 100644
--- a/backend/apps/webui/models/auths.py
+++ b/backend/apps/webui/models/auths.py
@@ -169,7 +169,6 @@ class AuthsTable:
     def update_user_password_by_id(self, id: str, new_password: str) -> bool:
         try:
             with get_db() as db:
-
                 result = (
                     db.query(Auth).filter_by(id=id).update({"password": new_password})
                 )
@@ -181,7 +180,6 @@ class AuthsTable:
     def update_email_by_id(self, id: str, email: str) -> bool:
         try:
             with get_db() as db:
-
                 result = db.query(Auth).filter_by(id=id).update({"email": email})
                 db.commit()
                 return True if result == 1 else False
diff --git a/backend/apps/webui/models/chats.py b/backend/apps/webui/models/chats.py
index ae63999a2..207bfab69 100644
--- a/backend/apps/webui/models/chats.py
+++ b/backend/apps/webui/models/chats.py
@@ -141,13 +141,14 @@ class ChatTable:
             db.add(shared_result)
             db.commit()
             db.refresh(shared_result)
+
             # Update the original chat with the share_id
             result = (
                 db.query(Chat)
                 .filter_by(id=chat_id)
                 .update({"share_id": shared_chat.id})
             )
-
+            db.commit()
             return shared_chat if (shared_result and result) else None
 
     def update_shared_chat_by_chat_id(self, chat_id: str) -> Optional[ChatModel]:
@@ -206,8 +207,8 @@ class ChatTable:
     def archive_all_chats_by_user_id(self, user_id: str) -> bool:
         try:
             with get_db() as db:
-
                 db.query(Chat).filter_by(user_id=user_id).update({"archived": True})
+                db.commit()
                 return True
         except:
             return False
diff --git a/backend/apps/webui/models/users.py b/backend/apps/webui/models/users.py
index 8e3b57bba..2f30cda02 100644
--- a/backend/apps/webui/models/users.py
+++ b/backend/apps/webui/models/users.py
@@ -212,6 +212,7 @@ class UsersTable:
         try:
             with get_db() as db:
                 db.query(User).filter_by(id=id).update({"oauth_sub": oauth_sub})
+                db.commit()
 
                 user = db.query(User).filter_by(id=id).first()
                 return UserModel.model_validate(user)