瀏覽代碼

fix: user valves

Timothy J. Baek 10 月之前
父節點
當前提交
9a6cbafdef
共有 2 個文件被更改,包括 4 次插入4 次删除
  1. 2 2
      backend/apps/webui/models/functions.py
  2. 2 2
      backend/apps/webui/models/tools.py

+ 2 - 2
backend/apps/webui/models/functions.py

@@ -185,7 +185,7 @@ class FunctionsTable:
     ) -> Optional[dict]:
     ) -> Optional[dict]:
         try:
         try:
             user = Users.get_user_by_id(user_id)
             user = Users.get_user_by_id(user_id)
-            user_settings = user.settings.model_dump()
+            user_settings = user.settings.model_dump() if user.settings else {}
 
 
             # Check if user has "functions" and "valves" settings
             # Check if user has "functions" and "valves" settings
             if "functions" not in user_settings:
             if "functions" not in user_settings:
@@ -203,7 +203,7 @@ class FunctionsTable:
     ) -> Optional[dict]:
     ) -> Optional[dict]:
         try:
         try:
             user = Users.get_user_by_id(user_id)
             user = Users.get_user_by_id(user_id)
-            user_settings = user.settings.model_dump()
+            user_settings = user.settings.model_dump() if user.settings else {}
 
 
             # Check if user has "functions" and "valves" settings
             # Check if user has "functions" and "valves" settings
             if "functions" not in user_settings:
             if "functions" not in user_settings:

+ 2 - 2
backend/apps/webui/models/tools.py

@@ -141,7 +141,7 @@ class ToolsTable:
     ) -> Optional[dict]:
     ) -> Optional[dict]:
         try:
         try:
             user = Users.get_user_by_id(user_id)
             user = Users.get_user_by_id(user_id)
-            user_settings = user.settings.model_dump()
+            user_settings = user.settings.model_dump() if user.settings else {}
 
 
             # Check if user has "tools" and "valves" settings
             # Check if user has "tools" and "valves" settings
             if "tools" not in user_settings:
             if "tools" not in user_settings:
@@ -159,7 +159,7 @@ class ToolsTable:
     ) -> Optional[dict]:
     ) -> Optional[dict]:
         try:
         try:
             user = Users.get_user_by_id(user_id)
             user = Users.get_user_by_id(user_id)
-            user_settings = user.settings.model_dump()
+            user_settings = user.settings.model_dump() if user.settings else {}
 
 
             # Check if user has "tools" and "valves" settings
             # Check if user has "tools" and "valves" settings
             if "tools" not in user_settings:
             if "tools" not in user_settings: