open-webui/Fu-Jie_openwebui-extensions
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: add discovery docstrings

Browse Source 
Co-authored-by: Fu-Jie <33599649+Fu-Jie@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-01-11 08:31:10 +00:00
parent 3649d75539
commit 7e44dde979
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/filters/async-context-compression/async_context_compression.py
View File

@@ -277,6 +277,7 @@ from datetime import datetime
def _discover_owui_engine(db_module: Any) -> Optional[Engine]: def _discover_owui_engine(db_module: Any) -> Optional[Engine]:
"""Discover the Open WebUI SQLAlchemy engine via provided db module helpers."""
if db_module is None: if db_module is None:
return None return None
@@ -288,7 +289,7 @@ def _discover_owui_engine(db_module: Any) -> Optional[Engine]:
with db_context() as session: with db_context() as session:
try: try:
return session.get_bind() return session.get_bind()
except Exception: except AttributeError:
return getattr(session, "bind", None) or getattr( return getattr(session, "bind", None) or getattr(
session, "engine", None session, "engine", None
) )
@@ -304,6 +305,7 @@ def _discover_owui_engine(db_module: Any) -> Optional[Engine]:
def _discover_owui_schema(db_module: Any) -> Optional[str]: def _discover_owui_schema(db_module: Any) -> Optional[str]:
"""Discover the Open WebUI database schema name if configured."""
if db_module is None: if db_module is None:
return None return None
@@ -376,6 +378,7 @@ class Filter:
@contextlib.contextmanager @contextlib.contextmanager
def _db_session(self): def _db_session(self):
"""Yield a database session using Open WebUI helpers with graceful fallbacks."""
db_module = self._owui_db db_module = self._owui_db
db_context = None db_context = None
if db_module is not None: if db_module is not None: