fix

backend/apps/webui/internal/db.py

@@ -62,10 +62,16 @@ Base = declarative_base()
 Session = scoped_session(SessionLocal)
 
 
+from contextlib import contextmanager
+
+
 # Dependency
-def get_db():
+def get_session():
     db = SessionLocal()
     try:
         yield db
     finally:
         db.close()
+
+
+get_db = contextmanager(get_session)

backend/apps/webui/models/tools.py

@@ -119,7 +119,8 @@ class ToolsTable:
             return None
 
     def get_tools(self) -> List[ToolModel]:
-        return [ToolModel.model_validate(tool) for tool in db.query(Tool).all()]
+        with get_db() as db:
+            return [ToolModel.model_validate(tool) for tool in db.query(Tool).all()]
 
     def get_tool_valves_by_id(self, id: str) -> Optional[dict]:
         try: