UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 13:07:25 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

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)
from contextlib import contextmanager
# Dependency
def get_db():
def get_session():
db = SessionLocal()
try:
yield db
finally:
db.close()
get_db = contextmanager(get_session)

3
backend/apps/webui/models/tools.py
View File

@ -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: