ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix

Browse source
This commit is contained in:
Timothy J. Baek 2024-07-03 23:39:16 -07:00
parent 37a5d2c06b
commit 8fe2a7bb75
2 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
backend/apps/webui/internal/db.py
View file

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

1
backend/apps/webui/models/tools.py
View file

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