浏览代码

fix: tools update

Timothy J. Baek 10 月之前
父节点
当前提交
87f656b029
共有 1 个文件被更改,包括 5 次插入3 次删除
  1. 5 3
      backend/apps/webui/models/tools.py

+ 5 - 3
backend/apps/webui/models/tools.py

@@ -188,10 +188,12 @@ class ToolsTable:
     def update_tool_by_id(self, id: str, updated: dict) -> Optional[ToolModel]:
         try:
             with get_db() as db:
-                tool = db.get(Tool, id)
-                tool.update(**updated)
-                tool.updated_at = int(time.time())
+                db.query(Tool).filter_by(id=id).update(
+                    {**updated, "updated_at": int(time.time())}
+                )
                 db.commit()
+
+                tool = db.query(Tool).get(id)
                 db.refresh(tool)
                 return ToolModel.model_validate(tool)
         except: