diff --git a/LICENSE b/LICENSE
index faa0129c65..99f39e7fef 100644
--- a/LICENSE
+++ b/LICENSE
@@ -1,3 +1,5 @@
+Open WebUI License
+
Copyright (c) 2023- Open WebUI Inc. [Created by Timothy Jaeryang Baek]
All rights reserved.
@@ -15,11 +17,27 @@ modification, are permitted provided that the following conditions are met:
contributors may be used to endorse or promote products derived from
this software without specific prior written permission.
-4. Notwithstanding any other provision of this License, and as a material condition of the rights granted herein, licensees are strictly prohibited from altering, removing, obscuring, or replacing any "Open WebUI" branding, including but not limited to the name, logo, or any visual, textual, or symbolic identifiers that distinguish the software and its interfaces, in any deployment or distribution, regardless of the number of users, except as explicitly set forth in Clauses 5 and 6 below.
+4. Notwithstanding any other provision of this License, and as a material
+ condition of the rights granted herein, licensees are strictly prohibited
+ from altering, removing, obscuring, or replacing any "Open WebUI"
+ branding, including but not limited to the name, logo, or any visual,
+ textual, or symbolic identifiers that distinguish the software and its
+ interfaces, in any deployment or distribution, except in the following
+ circumstances: (i) deployments or distributions where the total number
+ of end users (defined as individual natural persons with direct access
+ to the application) does not exceed fifty (50) within any rolling
+ thirty (30) day period; (ii) the licensee has obtained specific prior
+ written permission from the copyright holder; or (iii) where the
+ licensee has obtained a duly executed enterprise license expressly
+ permitting such modification. For all other cases, any removal or
+ alteration of the "Open WebUI" branding shall constitute a material
+ breach of license.
-5. The branding restriction enumerated in Clause 4 shall not apply in the following limited circumstances: (i) deployments or distributions where the total number of end users (defined as individual natural persons with direct access to the application) does not exceed fifty (50) within any rolling thirty (30) day period; (ii) cases in which the licensee is an official contributor to the codebase—with a substantive code change successfully merged into the main branch of the official codebase maintained by the copyright holder—who has obtained specific prior written permission for branding adjustment from the copyright holder; or (iii) where the licensee has obtained a duly executed enterprise license expressly permitting such modification. For all other cases, any removal or alteration of the "Open WebUI" branding shall constitute a material breach of license.
+Materials governed by prior licenses retain those original license
+terms, as specified in LICENSE_HISTORY.
-6. All code, modifications, or derivative works incorporated into this project prior to the incorporation of this branding clause remain licensed under the BSD 3-Clause License, and prior contributors retain all BSD-3 rights therein; if any such contributor requests the removal of their BSD-3-licensed code, the copyright holder will do so, and any replacement code will be licensed under the project's primary license then in effect. By contributing after this clause's adoption, you agree to the project's Contributor License Agreement (CLA) and to these updated terms for all new contributions.
+By contributing to this project, you agree to the project's Contributor
+License Agreement (CONTRIBUTOR_LICENSE_AGREEMENT).
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py
index 3631ca8191..751bbb3153 100644
--- a/backend/open_webui/config.py
+++ b/backend/open_webui/config.py
@@ -1,3 +1,4 @@
+import asyncio
import json
import logging
import os
@@ -35,7 +36,7 @@ from open_webui.env import (
WEBUI_NAME,
log,
)
-from open_webui.internal.db import Base, get_db
+from open_webui.internal.db import Base, get_db, get_async_db
from open_webui.utils.redis import get_redis_connection
@@ -90,6 +91,7 @@ def load_json_config():
def save_to_db(data):
+ """Sync save — used ONLY at startup/import time."""
with get_db() as db:
existing_config = db.query(Config).first()
if not existing_config:
@@ -102,12 +104,39 @@ def save_to_db(data):
db.commit()
+async def async_save_to_db(data):
+ """Async save — used for ALL runtime config persistence."""
+ from sqlalchemy import select
+
+ async with get_async_db() as db:
+ result = await db.execute(select(Config).limit(1))
+ existing_config = result.scalars().first()
+ if not existing_config:
+ new_config = Config(data=data, version=0)
+ db.add(new_config)
+ else:
+ existing_config.data = data
+ existing_config.updated_at = datetime.now()
+ db.add(existing_config)
+ await db.commit()
+
+
def reset_config():
+ """Sync reset — used ONLY at startup."""
with get_db() as db:
db.query(Config).delete()
db.commit()
+async def async_reset_config():
+ """Async reset — used at runtime."""
+ from sqlalchemy import delete as sa_delete
+
+ async with get_async_db() as db:
+ await db.execute(sa_delete(Config))
+ await db.commit()
+
+
# When initializing, check if config.json exists and migrate it to the database
if os.path.exists(f'{DATA_DIR}/config.json'):
data = load_json_config()
@@ -144,6 +173,7 @@ PERSISTENT_CONFIG_REGISTRY = []
def save_config(config):
+ """Sync save — used ONLY at startup/import time."""
global CONFIG_DATA
global PERSISTENT_CONFIG_REGISTRY
try:
@@ -159,6 +189,23 @@ def save_config(config):
return True
+async def async_save_config(config):
+ """Async save — used for ALL runtime config persistence."""
+ global CONFIG_DATA
+ global PERSISTENT_CONFIG_REGISTRY
+ try:
+ await async_save_to_db(config)
+ CONFIG_DATA = config
+
+ # Trigger updates on all registered PersistentConfig entries
+ for config_item in PERSISTENT_CONFIG_REGISTRY:
+ config_item.update()
+ except Exception as e:
+ log.exception(e)
+ return False
+ return True
+
+
T = TypeVar('T')
ENABLE_PERSISTENT_CONFIG = os.environ.get('ENABLE_PERSISTENT_CONFIG', 'True').lower() == 'true'
@@ -202,6 +249,7 @@ class PersistentConfig(Generic[T]):
log.info(f'Updated {self.env_name} to new value {self.value}')
def save(self):
+ """Sync save — used ONLY at startup/import time."""
log.info(f"Saving '{self.env_name}' to the database")
path_parts = self.config_path.split('.')
sub_config = CONFIG_DATA
@@ -213,6 +261,19 @@ class PersistentConfig(Generic[T]):
save_to_db(CONFIG_DATA)
self.config_value = self.value
+ async def async_save(self):
+ """Async save — used for ALL runtime config persistence."""
+ log.info(f"Saving '{self.env_name}' to the database")
+ path_parts = self.config_path.split('.')
+ sub_config = CONFIG_DATA
+ for key in path_parts[:-1]:
+ if key not in sub_config:
+ sub_config[key] = {}
+ sub_config = sub_config[key]
+ sub_config[path_parts[-1]] = self.value
+ await async_save_to_db(CONFIG_DATA)
+ self.config_value = self.value
+
class AppConfig:
_redis: Union[redis.Redis, redis.cluster.RedisCluster] = None
@@ -246,12 +307,27 @@ class AppConfig:
self._state[key] = value
else:
self._state[key].value = value
- self._state[key].save()
+
+ # At runtime (inside the event loop) persist via the async engine
+ # to avoid blocking the loop and contending with the async DB pool.
+ # At startup/import time, fall back to sync.
+ try:
+ loop = asyncio.get_running_loop()
+ loop.create_task(self._async_persist(key))
+ except RuntimeError:
+ self._state[key].save()
if self._redis and ENABLE_PERSISTENT_CONFIG:
redis_key = f'{self._redis_key_prefix}:config:{key}'
self._redis.set(redis_key, json.dumps(self._state[key].value))
+ async def _async_persist(self, key):
+ """Persist a single config key via the async engine."""
+ try:
+ await self._state[key].async_save()
+ except Exception as e:
+ log.error(f'Failed to async-persist config key {key}: {e}')
+
def __getattr__(self, key):
if key not in self._state:
raise AttributeError(f"Config key '{key}' not found")
@@ -915,6 +991,7 @@ if CUSTOM_NAME:
####################################
STORAGE_PROVIDER = os.environ.get('STORAGE_PROVIDER', 'local') # defaults to local, s3
+STORAGE_LOCAL_CACHE = os.environ.get('STORAGE_LOCAL_CACHE', 'true').lower() == 'true'
S3_ACCESS_KEY_ID = os.environ.get('S3_ACCESS_KEY_ID', None)
S3_SECRET_ACCESS_KEY = os.environ.get('S3_SECRET_ACCESS_KEY', None)
@@ -1148,10 +1225,16 @@ ENABLE_SIGNUP = PersistentConfig(
ENABLE_LOGIN_FORM = PersistentConfig(
'ENABLE_LOGIN_FORM',
- 'ui.ENABLE_LOGIN_FORM',
+ 'ui.enable_login_form',
os.environ.get('ENABLE_LOGIN_FORM', 'True').lower() == 'true',
)
+ENABLE_PASSWORD_CHANGE_FORM = PersistentConfig(
+ 'ENABLE_PASSWORD_CHANGE_FORM',
+ 'ui.enable_password_change_form',
+ os.environ.get('ENABLE_PASSWORD_CHANGE_FORM', 'True').lower() == 'true',
+)
+
ENABLE_PASSWORD_AUTH = os.environ.get('ENABLE_PASSWORD_AUTH', 'True').lower() == 'true'
DEFAULT_LOCALE = PersistentConfig(
@@ -1541,6 +1624,18 @@ ENABLE_CHANNELS = PersistentConfig(
os.environ.get('ENABLE_CHANNELS', 'False').lower() == 'true',
)
+AUTOMATION_MAX_COUNT = PersistentConfig(
+ 'AUTOMATION_MAX_COUNT',
+ 'automations.max_count',
+ os.environ.get('AUTOMATION_MAX_COUNT', ''),
+)
+
+AUTOMATION_MIN_INTERVAL = PersistentConfig(
+ 'AUTOMATION_MIN_INTERVAL',
+ 'automations.min_interval',
+ os.environ.get('AUTOMATION_MIN_INTERVAL', ''),
+)
+
ENABLE_NOTES = PersistentConfig(
'ENABLE_NOTES',
'notes.enable',
diff --git a/backend/open_webui/constants.py b/backend/open_webui/constants.py
index c0c79fdf50..ad1bdf4a20 100644
--- a/backend/open_webui/constants.py
+++ b/backend/open_webui/constants.py
@@ -91,6 +91,17 @@ class ERROR_MESSAGES(str, Enum):
INVALID_PASSWORD = lambda err='': err if err else 'The password does not meet the required validation criteria.'
+ AUTOMATION_LIMIT_EXCEEDED = lambda size='': f'Automation limit reached ({size})'
+ AUTOMATION_TOO_FREQUENT = lambda interval='': f'Schedule too frequent. Minimum interval is {interval} seconds.'
+ AUTOMATION_INVALID_RRULE = lambda err='': f'Invalid RRULE: {err}'
+ AUTOMATION_NO_FUTURE_RUNS = 'RRULE has no future occurrences'
+
+ FEATURE_DISABLED = lambda name='': f'{name} is disabled'
+ INPUT_TOO_LONG = lambda size='': f'Input prompt exceeds maximum length of {size}'
+ SERVER_CONNECTION_ERROR = 'Open WebUI: Server Connection Error'
+ REQUIRED_FIELD_EMPTY = lambda name='': f'Required field {name} is empty'
+ OAUTH_NOT_CONFIGURED = lambda name='': f"Provider '{name}' is not configured"
+
class TASKS(str, Enum):
def __str__(self) -> str:
diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py
index 53737a1f2f..fbabd32361 100644
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@@ -426,6 +426,27 @@ try:
except ValueError:
REDIS_SOCKET_CONNECT_TIMEOUT = None
+# Whether to enable TCP SO_KEEPALIVE on Redis client sockets. Opt-in:
+# defaults to off so behavior is unchanged for existing deployments. When
+# enabled, the kernel sends TCP keepalive probes on idle connections so
+# half-closed sockets (e.g. after a silent firewall/LB reset or a NIC
+# flap) are detected before the next command lands on them.
+REDIS_SOCKET_KEEPALIVE = os.environ.get('REDIS_SOCKET_KEEPALIVE', 'False').lower() == 'true'
+
+# How often (in seconds) redis-py should PING an idle pooled connection
+# before reusing it. Opt-in: defaults to unset (empty string) so behavior
+# is unchanged for existing deployments. When set, should be shorter than
+# the Redis server `timeout` setting and any firewall/LB idle timeout on
+# the path to Redis, so stale connections are detected before a real
+# command lands on them. Set to 0 or empty to disable.
+REDIS_HEALTH_CHECK_INTERVAL = os.environ.get('REDIS_HEALTH_CHECK_INTERVAL', '')
+try:
+ REDIS_HEALTH_CHECK_INTERVAL = int(REDIS_HEALTH_CHECK_INTERVAL)
+ if REDIS_HEALTH_CHECK_INTERVAL <= 0:
+ REDIS_HEALTH_CHECK_INTERVAL = None
+except ValueError:
+ REDIS_HEALTH_CHECK_INTERVAL = None
+
REDIS_RECONNECT_DELAY = os.environ.get('REDIS_RECONNECT_DELAY', '')
if REDIS_RECONNECT_DELAY == '':
@@ -496,6 +517,16 @@ PASSWORD_VALIDATION_HINT = os.environ.get('PASSWORD_VALIDATION_HINT', '')
BYPASS_MODEL_ACCESS_CONTROL = os.environ.get('BYPASS_MODEL_ACCESS_CONTROL', 'False').lower() == 'true'
+# When enabled, skips pydub-based preprocessing (format conversion, compression,
+# and chunked splitting) before sending files to processing engines. Useful when
+# the upstream provider handles these steps or when ffmpeg is unavailable.
+BYPASS_PYDUB_PREPROCESSING = os.environ.get('BYPASS_PYDUB_PREPROCESSING', 'False').lower() == 'true'
+
+# When disabled (default), the OpenAI catch-all proxy endpoint (/{path:path})
+# is blocked. Enable only if you need direct passthrough to upstream OpenAI-
+# compatible APIs for endpoints not natively handled by Open WebUI.
+ENABLE_OPENAI_API_PASSTHROUGH = os.environ.get('ENABLE_OPENAI_API_PASSTHROUGH', 'False').lower() == 'true'
+
WEBUI_AUTH_SIGNOUT_REDIRECT_URL = os.environ.get('WEBUI_AUTH_SIGNOUT_REDIRECT_URL', None)
####################################
@@ -778,6 +809,36 @@ else:
AIOHTTP_CLIENT_TIMEOUT_TOOL_SERVER = AIOHTTP_CLIENT_TIMEOUT
+####################################
+# AIOHTTP Connection Pool
+####################################
+
+AIOHTTP_POOL_CONNECTIONS = os.environ.get('AIOHTTP_POOL_CONNECTIONS', '')
+if AIOHTTP_POOL_CONNECTIONS == '':
+ AIOHTTP_POOL_CONNECTIONS = None
+else:
+ try:
+ AIOHTTP_POOL_CONNECTIONS = int(AIOHTTP_POOL_CONNECTIONS)
+ except ValueError:
+ AIOHTTP_POOL_CONNECTIONS = None
+
+AIOHTTP_POOL_CONNECTIONS_PER_HOST = os.environ.get('AIOHTTP_POOL_CONNECTIONS_PER_HOST', '')
+if AIOHTTP_POOL_CONNECTIONS_PER_HOST == '':
+ AIOHTTP_POOL_CONNECTIONS_PER_HOST = None
+else:
+ try:
+ AIOHTTP_POOL_CONNECTIONS_PER_HOST = int(AIOHTTP_POOL_CONNECTIONS_PER_HOST)
+ except ValueError:
+ AIOHTTP_POOL_CONNECTIONS_PER_HOST = None
+
+AIOHTTP_POOL_DNS_TTL = os.environ.get('AIOHTTP_POOL_DNS_TTL', '300')
+try:
+ AIOHTTP_POOL_DNS_TTL = int(AIOHTTP_POOL_DNS_TTL)
+ if AIOHTTP_POOL_DNS_TTL < 0:
+ AIOHTTP_POOL_DNS_TTL = 300
+except ValueError:
+ AIOHTTP_POOL_DNS_TTL = 300
+
RAG_EMBEDDING_TIMEOUT = os.environ.get('RAG_EMBEDDING_TIMEOUT', '')
if RAG_EMBEDDING_TIMEOUT == '':
@@ -881,6 +942,9 @@ AUDIT_INCLUDED_PATHS = os.getenv('AUDIT_INCLUDED_PATHS', '').split(',')
AUDIT_INCLUDED_PATHS = [path.strip() for path in AUDIT_INCLUDED_PATHS]
AUDIT_INCLUDED_PATHS = [path.lstrip('/') for path in AUDIT_INCLUDED_PATHS if path]
+# When enabled, GET requests are also audited (disabled by default to avoid log noise)
+ENABLE_AUDIT_GET_REQUESTS = os.getenv('ENABLE_AUDIT_GET_REQUESTS', 'False').lower() == 'true'
+
####################################
# OPENTELEMETRY
diff --git a/backend/open_webui/functions.py b/backend/open_webui/functions.py
index 6a1fe22149..37a9011aab 100644
--- a/backend/open_webui/functions.py
+++ b/backend/open_webui/functions.py
@@ -53,12 +53,12 @@ logging.basicConfig(stream=sys.stdout, level=GLOBAL_LOG_LEVEL)
log = logging.getLogger(__name__)
-def get_function_module_by_id(request: Request, pipe_id: str):
- function_module, _, _ = get_function_module_from_cache(request, pipe_id)
+async def get_function_module_by_id(request: Request, pipe_id: str):
+ function_module, _, _ = await get_function_module_from_cache(request, pipe_id)
if hasattr(function_module, 'valves') and hasattr(function_module, 'Valves'):
Valves = function_module.Valves
- valves = Functions.get_function_valves_by_id(pipe_id)
+ valves = await Functions.get_function_valves_by_id(pipe_id)
if valves:
try:
@@ -73,12 +73,12 @@ def get_function_module_by_id(request: Request, pipe_id: str):
async def get_function_models(request):
- pipes = Functions.get_functions_by_type('pipe', active_only=True)
+ pipes = await Functions.get_functions_by_type('pipe', active_only=True)
pipe_models = []
for pipe in pipes:
try:
- function_module = get_function_module_by_id(request, pipe.id)
+ function_module = await get_function_module_by_id(request, pipe.id)
has_user_valves = False
if hasattr(function_module, 'UserValves'):
@@ -187,7 +187,7 @@ async def generate_function_chat_completion(request, form_data, user, models: di
pipe_id, _ = pipe_id.split('.', 1)
return pipe_id
- def get_function_params(function_module, form_data, user, extra_params=None):
+ async def get_function_params(function_module, form_data, user, extra_params=None):
if extra_params is None:
extra_params = {}
@@ -198,7 +198,7 @@ async def generate_function_chat_completion(request, form_data, user, models: di
params = {'body': form_data} | {k: v for k, v in extra_params.items() if k in sig.parameters}
if '__user__' in params and hasattr(function_module, 'UserValves'):
- user_valves = Functions.get_user_valves_by_id_and_user_id(pipe_id, user.id)
+ user_valves = await Functions.get_user_valves_by_id_and_user_id(pipe_id, user.id)
try:
params['__user__']['valves'] = function_module.UserValves(**user_valves)
except Exception as e:
@@ -208,7 +208,7 @@ async def generate_function_chat_completion(request, form_data, user, models: di
return params
model_id = form_data.get('model')
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
metadata = form_data.pop('metadata', {})
@@ -225,8 +225,8 @@ async def generate_function_chat_completion(request, form_data, user, models: di
if metadata:
if all(k in metadata for k in ('session_id', 'chat_id', 'message_id')):
- __event_emitter__ = get_event_emitter(metadata)
- __event_call__ = get_event_call(metadata)
+ __event_emitter__ = await get_event_emitter(metadata)
+ __event_call__ = await get_event_call(metadata)
__task__ = metadata.get('task', None)
__task_body__ = metadata.get('task_body', None)
@@ -268,10 +268,10 @@ async def generate_function_chat_completion(request, form_data, user, models: di
form_data = apply_system_prompt_to_body(system, form_data, metadata, user)
pipe_id = get_pipe_id(form_data)
- function_module = get_function_module_by_id(request, pipe_id)
+ function_module = await get_function_module_by_id(request, pipe_id)
pipe = function_module.pipe
- params = get_function_params(function_module, form_data, user, extra_params)
+ params = await get_function_params(function_module, form_data, user, extra_params)
if form_data.get('stream', False):
diff --git a/backend/open_webui/internal/db.py b/backend/open_webui/internal/db.py
index b0545255a6..3818543fc7 100644
--- a/backend/open_webui/internal/db.py
+++ b/backend/open_webui/internal/db.py
@@ -1,7 +1,7 @@
import os
import json
import logging
-from contextlib import contextmanager
+from contextlib import asynccontextmanager, contextmanager
from typing import Any, Optional
from open_webui.internal.wrappers import register_connection
@@ -19,6 +19,7 @@ from open_webui.env import (
)
from peewee_migrate import Router
from sqlalchemy import Dialect, create_engine, MetaData, event, types
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker, Session
from sqlalchemy.pool import QueuePool, NullPool
@@ -81,6 +82,32 @@ if ENABLE_DB_MIGRATIONS:
SQLALCHEMY_DATABASE_URL = DATABASE_URL
+
+def _make_async_url(url: str) -> str:
+ """Convert a sync database URL to its async driver equivalent."""
+ if url.startswith('sqlite+sqlcipher://'):
+ # SQLCipher has no async driver — not supported for async
+ raise ValueError(
+ 'sqlite+sqlcipher:// URLs are not supported with async engine. '
+ 'Use standard sqlite:// or postgresql:// instead.'
+ )
+ if url.startswith('sqlite:///') or url.startswith('sqlite://'):
+ return url.replace('sqlite://', 'sqlite+aiosqlite://', 1)
+ if url.startswith('postgresql+psycopg2://'):
+ return url.replace('postgresql+psycopg2://', 'postgresql+asyncpg://', 1)
+ if url.startswith('postgresql://'):
+ return url.replace('postgresql://', 'postgresql+asyncpg://', 1)
+ if url.startswith('postgres://'):
+ return url.replace('postgres://', 'postgresql+asyncpg://', 1)
+ # For other dialects, return as-is and let SQLAlchemy handle it
+ return url
+
+
+# ============================================================
+# SYNC ENGINE (used only for: startup migrations, config loading,
+# Alembic, peewee migration, health checks)
+# ============================================================
+
# Handle SQLCipher URLs
if SQLALCHEMY_DATABASE_URL.startswith('sqlite+sqlcipher://'):
database_password = os.environ.get('DATABASE_PASSWORD')
@@ -155,6 +182,7 @@ else:
engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True)
+# Sync session — used ONLY for startup config loading (config.py runs at import time)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine, expire_on_commit=False)
metadata_obj = MetaData(schema=DATABASE_SCHEMA)
Base = declarative_base(metadata=metadata_obj)
@@ -162,6 +190,7 @@ ScopedSession = scoped_session(SessionLocal)
def get_session():
+ """Sync session generator — used ONLY for startup/config operations."""
db = SessionLocal()
try:
yield db
@@ -172,10 +201,82 @@ def get_session():
get_db = contextmanager(get_session)
-@contextmanager
-def get_db_context(db: Optional[Session] = None):
- if isinstance(db, Session) and DATABASE_ENABLE_SESSION_SHARING:
+# ============================================================
+# ASYNC ENGINE (used for ALL runtime database operations)
+# ============================================================
+
+ASYNC_SQLALCHEMY_DATABASE_URL = _make_async_url(SQLALCHEMY_DATABASE_URL)
+
+if 'sqlite' in ASYNC_SQLALCHEMY_DATABASE_URL:
+ async_engine = create_async_engine(
+ ASYNC_SQLALCHEMY_DATABASE_URL,
+ connect_args={'check_same_thread': False},
+ )
+
+ if DATABASE_ENABLE_SQLITE_WAL:
+
+ @event.listens_for(async_engine.sync_engine, 'connect')
+ def _set_sqlite_wal(dbapi_connection, connection_record):
+ cursor = dbapi_connection.cursor()
+ cursor.execute('PRAGMA journal_mode=WAL')
+ cursor.close()
+else:
+ if isinstance(DATABASE_POOL_SIZE, int):
+ if DATABASE_POOL_SIZE > 0:
+ async_engine = create_async_engine(
+ ASYNC_SQLALCHEMY_DATABASE_URL,
+ pool_size=DATABASE_POOL_SIZE,
+ max_overflow=DATABASE_POOL_MAX_OVERFLOW,
+ pool_timeout=DATABASE_POOL_TIMEOUT,
+ pool_recycle=DATABASE_POOL_RECYCLE,
+ pool_pre_ping=True,
+ )
+ else:
+ async_engine = create_async_engine(
+ ASYNC_SQLALCHEMY_DATABASE_URL,
+ pool_pre_ping=True,
+ poolclass=NullPool,
+ )
+ else:
+ async_engine = create_async_engine(
+ ASYNC_SQLALCHEMY_DATABASE_URL,
+ pool_pre_ping=True,
+ )
+
+
+AsyncSessionLocal = async_sessionmaker(
+ bind=async_engine,
+ class_=AsyncSession,
+ autocommit=False,
+ autoflush=False,
+ expire_on_commit=False,
+)
+
+
+async def get_async_session():
+ """Async session generator for FastAPI Depends()."""
+ async with AsyncSessionLocal() as db:
+ try:
+ yield db
+ finally:
+ await db.close()
+
+
+@asynccontextmanager
+async def get_async_db():
+ """Async context manager for use outside of FastAPI dependency injection."""
+ async with AsyncSessionLocal() as db:
+ try:
+ yield db
+ finally:
+ await db.close()
+
+
+@asynccontextmanager
+async def get_async_db_context(db: Optional[AsyncSession] = None):
+ """Async context manager that reuses an existing session if provided and session sharing is enabled."""
+ if isinstance(db, AsyncSession) and DATABASE_ENABLE_SESSION_SHARING:
yield db
else:
- with get_db() as session:
+ async with get_async_db() as session:
yield session
diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py
index 1affa2088b..9d02875e7e 100644
--- a/backend/open_webui/main.py
+++ b/backend/open_webui/main.py
@@ -21,7 +21,7 @@ from typing import Optional
from aiocache import cached
import aiohttp
import anyio.to_thread
-import requests
+
from redis import Redis
@@ -46,7 +46,6 @@ from fastapi.staticfiles import StaticFiles
from starlette_compress import CompressMiddleware
from starlette.exceptions import HTTPException as StarletteHTTPException
-from starlette.middleware.base import BaseHTTPMiddleware
from starlette.middleware.sessions import SessionMiddleware
from starlette.responses import Response, StreamingResponse
from starlette.datastructures import Headers
@@ -58,8 +57,15 @@ from starsessions import (
from starsessions.stores.redis import RedisStore
from open_webui.utils import logger
+from open_webui.utils.asgi_middleware import (
+ AuthTokenMiddleware,
+ CommitSessionMiddleware,
+ RedirectMiddleware,
+ WebsocketUpgradeGuardMiddleware,
+)
from open_webui.utils.audit import AuditLevel, AuditLoggingMiddleware
from open_webui.utils.logger import start_logger
+from open_webui.utils.session_pool import get_session
from open_webui.socket.main import (
MODELS,
app as socket_app,
@@ -67,6 +73,7 @@ from open_webui.socket.main import (
periodic_session_pool_cleanup,
get_event_emitter,
get_models_in_use,
+ get_user_id_from_session_pool,
)
from open_webui.routers import (
analytics,
@@ -108,13 +115,13 @@ from open_webui.routers.retrieval import (
)
-from sqlalchemy.orm import Session
-from open_webui.internal.db import ScopedSession, engine, get_session
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import ScopedSession, engine, get_async_session
from open_webui.models.functions import Functions
from open_webui.models.models import Models
from open_webui.models.users import UserModel, Users
-from open_webui.models.chats import Chats
+from open_webui.models.chats import Chats, ChatForm
from open_webui.config import (
# Ollama
@@ -380,11 +387,14 @@ from open_webui.config import (
JWT_EXPIRES_IN,
ENABLE_SIGNUP,
ENABLE_LOGIN_FORM,
+ ENABLE_PASSWORD_CHANGE_FORM,
ENABLE_API_KEYS,
ENABLE_API_KEYS_ENDPOINT_RESTRICTIONS,
API_KEYS_ALLOWED_ENDPOINTS,
ENABLE_FOLDERS,
FOLDER_MAX_FILE_COUNT,
+ AUTOMATION_MAX_COUNT,
+ AUTOMATION_MIN_INTERVAL,
ENABLE_CHANNELS,
ENABLE_NOTES,
ENABLE_USER_STATUS,
@@ -469,12 +479,14 @@ from open_webui.config import (
AUTOCOMPLETE_GENERATION_INPUT_MAX_LENGTH,
AppConfig,
reset_config,
+ async_reset_config,
)
from open_webui.env import (
ENABLE_CUSTOM_MODEL_FALLBACK,
LICENSE_KEY,
AUDIT_EXCLUDED_PATHS,
AUDIT_INCLUDED_PATHS,
+ ENABLE_AUDIT_GET_REQUESTS,
AUDIT_LOG_LEVEL,
CHANGELOG,
REDIS_URL,
@@ -553,6 +565,7 @@ from open_webui.utils.oauth import (
get_oauth_client_info_with_static_credentials,
encrypt_data,
decrypt_data,
+ resolve_oauth_client_info,
OAuthManager,
OAuthClientManager,
OAuthClientInformationFull,
@@ -565,17 +578,18 @@ from open_webui.tasks import (
list_task_ids_by_item_id,
create_task,
stop_task,
+ stop_item_tasks,
list_tasks,
) # Import from tasks.py
from open_webui.utils.redis import get_sentinels_from_env
-from open_webui.constants import ERROR_MESSAGES
+from open_webui.constants import ERROR_MESSAGES, TASKS
if SAFE_MODE:
print('SAFE MODE ENABLED')
- Functions.deactivate_all_functions()
+ # Functions.deactivate_all_functions() is awaited in lifespan below
logging.basicConfig(stream=sys.stdout, level=GLOBAL_LOG_LEVEL)
log = logging.getLogger(__name__)
@@ -622,21 +636,24 @@ async def lifespan(app: FastAPI):
start_logger()
if RESET_CONFIG_ON_START:
- reset_config()
+ await async_reset_config()
if LICENSE_KEY:
get_license_data(app, LICENSE_KEY)
# Create admin account from env vars if specified and no users exist
if WEBUI_ADMIN_EMAIL and WEBUI_ADMIN_PASSWORD:
- if create_admin_user(WEBUI_ADMIN_EMAIL, WEBUI_ADMIN_PASSWORD, WEBUI_ADMIN_NAME):
+ if await create_admin_user(WEBUI_ADMIN_EMAIL, WEBUI_ADMIN_PASSWORD, WEBUI_ADMIN_NAME):
# Disable signup since we now have an admin
app.state.config.ENABLE_SIGNUP = False
+ if SAFE_MODE:
+ await Functions.deactivate_all_functions()
+
# This should be blocking (sync) so functions are not deactivated on first /get_models calls
# when the first user lands on the / route.
log.info('Installing external dependencies of functions and tools...')
- install_tool_and_function_dependencies()
+ await install_tool_and_function_dependencies()
app.state.redis = get_redis_connection(
redis_url=REDIS_URL,
@@ -715,6 +732,11 @@ async def lifespan(app: FastAPI):
yield
+ # Shutdown: clean up shared resources
+ from open_webui.utils.session_pool import close_session
+
+ await close_session()
+
if hasattr(app.state, 'redis_task_command_listener'):
app.state.redis_task_command_listener.cancel()
@@ -842,6 +864,7 @@ app.state.BASE_MODELS = []
app.state.config.WEBUI_URL = WEBUI_URL
app.state.config.ENABLE_SIGNUP = ENABLE_SIGNUP
app.state.config.ENABLE_LOGIN_FORM = ENABLE_LOGIN_FORM
+app.state.config.ENABLE_PASSWORD_CHANGE_FORM = ENABLE_PASSWORD_CHANGE_FORM
app.state.config.ENABLE_API_KEYS = ENABLE_API_KEYS
app.state.config.ENABLE_API_KEYS_ENDPOINT_RESTRICTIONS = ENABLE_API_KEYS_ENDPOINT_RESTRICTIONS
@@ -876,6 +899,8 @@ app.state.config.BANNERS = WEBUI_BANNERS
app.state.config.ENABLE_FOLDERS = ENABLE_FOLDERS
app.state.config.FOLDER_MAX_FILE_COUNT = FOLDER_MAX_FILE_COUNT
+app.state.config.AUTOMATION_MAX_COUNT = AUTOMATION_MAX_COUNT
+app.state.config.AUTOMATION_MIN_INTERVAL = AUTOMATION_MIN_INTERVAL
app.state.config.ENABLE_CHANNELS = ENABLE_CHANNELS
app.state.config.ENABLE_NOTES = ENABLE_NOTES
app.state.config.ENABLE_COMMUNITY_SHARING = ENABLE_COMMUNITY_SHARING
@@ -1343,149 +1368,19 @@ if ENABLE_COMPRESSION_MIDDLEWARE:
app.add_middleware(CompressMiddleware)
-class RedirectMiddleware(BaseHTTPMiddleware):
- async def dispatch(self, request: Request, call_next):
- # Check if the request is a GET request
- if request.method == 'GET':
- path = request.url.path
- query_params = dict(parse_qs(urlparse(str(request.url)).query))
-
- redirect_params = {}
-
- # Check for the specific watch path and the presence of 'v' parameter
- if path.endswith('/watch') and 'v' in query_params:
- # Extract the first 'v' parameter
- youtube_video_id = query_params['v'][0]
- redirect_params['youtube'] = youtube_video_id
-
- if 'shared' in query_params and len(query_params['shared']) > 0:
- # PWA share_target support
-
- text = query_params['shared'][0]
- if text:
- urls = re.match(r'https://\S+', text)
- if urls:
- from open_webui.retrieval.loaders.youtube import _parse_video_id
-
- if youtube_video_id := _parse_video_id(urls[0]):
- redirect_params['youtube'] = youtube_video_id
- else:
- redirect_params['load-url'] = urls[0]
- else:
- redirect_params['q'] = text
-
- if redirect_params:
- redirect_url = f'/?{urlencode(redirect_params)}'
- return RedirectResponse(url=redirect_url)
-
- # Proceed with the normal flow of other requests
- response = await call_next(request)
- return response
-
-
+# All HTTP middlewares below are pure-ASGI implementations. The previous
+# `BaseHTTPMiddleware` / `@app.middleware('http')` versions wrapped the
+# downstream app in an anyio task group whose cancel scope cancelled
+# in-flight DB calls (and any other awaits) on client disconnect /
+# response completion — which surfaced as noisy SQLAlchemy
+# `terminate_force_close` tracebacks under aiosqlite and as random
+# CancelledError storms across the request path. See
+# `open_webui.utils.asgi_middleware` for the rationale.
app.add_middleware(RedirectMiddleware)
app.add_middleware(SecurityHeadersMiddleware)
-
-
-class APIKeyRestrictionMiddleware:
- def __init__(self, app):
- self.app = app
-
- async def __call__(self, scope, receive, send):
- if scope['type'] == 'http':
- request = Request(scope)
- auth_header = request.headers.get('Authorization')
- token = None
-
- if auth_header:
- parts = auth_header.split(' ', 1)
- if len(parts) == 2:
- token = parts[1]
-
- # Only apply restrictions if an sk- API key is used
- if token and token.startswith('sk-'):
- # Check if restrictions are enabled
- if app.state.config.ENABLE_API_KEYS_ENDPOINT_RESTRICTIONS:
- allowed_paths = [
- path.strip()
- for path in str(app.state.config.API_KEYS_ALLOWED_ENDPOINTS).split(',')
- if path.strip()
- ]
-
- request_path = request.url.path
-
- # Match exact path or prefix path
- is_allowed = any(
- request_path == allowed or request_path.startswith(allowed + '/') for allowed in allowed_paths
- )
-
- if not is_allowed:
- await JSONResponse(
- status_code=status.HTTP_403_FORBIDDEN,
- content={'detail': 'API key not allowed to access this endpoint.'},
- )(scope, receive, send)
- return
-
- await self.app(scope, receive, send)
-
-
-app.add_middleware(APIKeyRestrictionMiddleware)
-
-
-@app.middleware('http')
-async def commit_session_after_request(request: Request, call_next):
- response = await call_next(request)
- # log.debug("Commit session after request")
- try:
- ScopedSession.commit()
- finally:
- # CRITICAL: remove() returns the connection to the pool.
- # Without this, connections remain "checked out" and accumulate
- # as "idle in transaction" in PostgreSQL.
- ScopedSession.remove()
- return response
-
-
-@app.middleware('http')
-async def check_url(request: Request, call_next):
- start_time = int(time.time())
- request.state.token = get_http_authorization_cred(request.headers.get('Authorization'))
- # Fallback to cookie token for browser sessions
- if request.state.token is None and request.cookies.get('token'):
- from fastapi.security import HTTPAuthorizationCredentials
-
- request.state.token = HTTPAuthorizationCredentials(scheme='Bearer', credentials=request.cookies.get('token'))
-
- # Fallback to x-api-key header for Anthropic Messages API routes
- if request.state.token is None and request.headers.get('x-api-key'):
- request_path = request.url.path
- if request_path in ('/api/message', '/api/v1/messages') or request_path.startswith('/ollama/v1/messages'):
- from fastapi.security import HTTPAuthorizationCredentials
-
- request.state.token = HTTPAuthorizationCredentials(
- scheme='Bearer', credentials=request.headers.get('x-api-key')
- )
-
- request.state.enable_api_keys = app.state.config.ENABLE_API_KEYS
- response = await call_next(request)
- process_time = int(time.time()) - start_time
- response.headers['X-Process-Time'] = str(process_time)
- return response
-
-
-@app.middleware('http')
-async def inspect_websocket(request: Request, call_next):
- if '/ws/socket.io' in request.url.path and request.query_params.get('transport') == 'websocket':
- upgrade = (request.headers.get('Upgrade') or '').lower()
- connection = (request.headers.get('Connection') or '').lower().split(',')
- # Check that there's the correct headers for an upgrade, else reject the connection
- # This is to work around this upstream issue: https://github.com/miguelgrinberg/python-engineio/issues/367
- if upgrade != 'websocket' or 'upgrade' not in connection:
- return JSONResponse(
- status_code=status.HTTP_400_BAD_REQUEST,
- content={'detail': 'Invalid WebSocket upgrade request'},
- )
- return await call_next(request)
+app.add_middleware(CommitSessionMiddleware)
+app.add_middleware(AuthTokenMiddleware, fastapi_app=app)
+app.add_middleware(WebsocketUpgradeGuardMiddleware)
app.add_middleware(
@@ -1557,6 +1452,7 @@ if audit_level != AuditLevel.NONE:
audit_level=audit_level,
excluded_paths=AUDIT_EXCLUDED_PATHS,
included_paths=AUDIT_INCLUDED_PATHS,
+ audit_get_requests=ENABLE_AUDIT_GET_REQUESTS,
max_body_size=MAX_BODY_LOG_SIZE,
)
##################################
@@ -1605,7 +1501,7 @@ async def get_models(request: Request, refresh: bool = False, user=Depends(get_v
)
)
- models = get_filtered_models(models, user)
+ models = await get_filtered_models(models, user)
log.debug(
f'/api/models returned filtered models accessible to the user: {json.dumps([model.get("id") for model in models])}'
@@ -1671,12 +1567,12 @@ async def chat_completion(
raise Exception('Model not found')
model = request.app.state.MODELS[model_id]
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
# Check if user has access to the model
if not BYPASS_MODEL_ACCESS_CONTROL and (user.role != 'admin' or not BYPASS_ADMIN_ACCESS_CONTROL):
try:
- check_model_access(user, model)
+ await check_model_access(user, model)
except Exception as e:
raise e
else:
@@ -1724,13 +1620,30 @@ async def chat_completion(
if model_info_params.get('reasoning_tags') is not None:
reasoning_tags = model_info_params.get('reasoning_tags')
+ # parent_id signals intent:
+ # null → new chat (root message, no parent)
+ # value → follow-up (user message's parentId = prev assistant)
+ # absent → legacy caller, no chat management
+ is_new_chat = 'parent_id' in form_data and form_data['parent_id'] is None and not form_data.get('chat_id')
+ parent_id = form_data.pop('parent_id', None)
+ form_data.pop('new_chat', None) # Legacy field
+
+ # Multi-model: {model_id: assistant_message_id}
+ # Single-model fallback: built from 'model' + 'id'
+ message_ids = form_data.pop('message_ids', None)
+ if not message_ids:
+ message_ids = {model_id: form_data.pop('id', None)}
+ else:
+ form_data.pop('id', None)
+
+ user_message = form_data.pop('user_message', None) or form_data.pop('parent_message', None)
metadata = {
'user_id': user.id,
'chat_id': form_data.pop('chat_id', None),
- 'message_id': form_data.pop('id', None),
- 'parent_message': form_data.pop('parent_message', None),
- 'parent_message_id': form_data.pop('parent_id', None),
+ 'user_message': user_message,
+ 'user_message_id': user_message.get('id') if user_message else None,
'session_id': form_data.pop('session_id', None),
+ 'folder_id': form_data.pop('folder_id', None),
'filter_ids': form_data.pop('filter_ids', []),
'tool_ids': form_data.get('tool_ids', None),
'tool_servers': form_data.pop('tool_servers', None),
@@ -1753,36 +1666,160 @@ async def chat_completion(
},
}
+ if is_new_chat:
+ metadata['chat_id'] = str(uuid4())
+
if metadata.get('chat_id') and user:
- if not metadata['chat_id'].startswith('local:'): # temporary chats are not stored
- # Verify chat ownership — lightweight EXISTS check avoids
- # deserializing the full chat JSON blob just to confirm the row exists
- if (
- not Chats.is_chat_owner(metadata['chat_id'], user.id) and user.role != 'admin'
- ): # admins can access any chat
- raise HTTPException(
- status_code=status.HTTP_404_NOT_FOUND,
- detail=ERROR_MESSAGES.DEFAULT(),
+ chat_id = metadata['chat_id']
+ if not chat_id.startswith('local:'): # temporary chats are not stored
+ if is_new_chat:
+ # Build the full history upfront with ALL assistant placeholders
+ user_message = metadata.get('user_message') or {}
+ user_message_id = user_message.get('id') if user_message else None
+
+ history_messages = {}
+ all_assistant_ids = [assistant_id for assistant_id in message_ids.values() if assistant_id]
+
+ if user_message_id and user_message:
+ user_message['childrenIds'] = all_assistant_ids
+ history_messages[user_message_id] = user_message
+
+ for target_model_id, assistant_message_id in message_ids.items():
+ if assistant_message_id:
+ history_messages[assistant_message_id] = {
+ 'id': assistant_message_id,
+ 'parentId': user_message_id,
+ 'childrenIds': [],
+ 'role': 'assistant',
+ 'content': '',
+ 'done': False,
+ 'model': target_model_id,
+ 'timestamp': int(time.time()),
+ }
+
+ await Chats.insert_new_chat(
+ chat_id,
+ user.id,
+ ChatForm(
+ chat={
+ 'id': chat_id,
+ 'title': 'New Chat',
+ 'models': list(message_ids.keys()),
+ 'history': {
+ 'currentId': all_assistant_ids[0] if all_assistant_ids else user_message_id,
+ 'messages': history_messages,
+ },
+ 'messages': [
+ {'role': 'user', 'content': user_message.get('content', '')},
+ ]
+ if user_message_id
+ else [],
+ 'tags': [],
+ 'timestamp': int(time.time() * 1000),
+ },
+ folder_id=metadata.get('folder_id'),
+ ),
)
- # Insert chat files from parent message if any
- parent_message = metadata.get('parent_message') or {}
- parent_message_files = parent_message.get('files', [])
- if parent_message_files:
- try:
- Chats.insert_chat_files(
- metadata['chat_id'],
- parent_message.get('id'),
- [
- file_item.get('id')
- for file_item in parent_message_files
- if file_item.get('type') == 'file'
- ],
- user.id,
+ # Insert chat files from user message if any
+ user_message_files = user_message.get('files', [])
+ if user_message_files:
+ try:
+ await Chats.insert_chat_files(
+ chat_id,
+ user_message_id,
+ [
+ file_item.get('id')
+ for file_item in user_message_files
+ if file_item.get('type') == 'file'
+ ],
+ user.id,
+ )
+ except Exception as e:
+ log.debug(f'Error inserting chat files: {e}')
+ pass
+ else:
+ # Existing chat — verify ownership
+ if not await Chats.is_chat_owner(chat_id, user.id) and user.role != 'admin':
+ raise HTTPException(
+ status_code=status.HTTP_404_NOT_FOUND,
+ detail=ERROR_MESSAGES.DEFAULT(),
)
- except Exception as e:
- log.debug(f'Error inserting chat files: {e}')
- pass
+
+ # Save user message to DB
+ user_message = metadata.get('user_message') or {}
+ if user_message and user_message.get('id'):
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
+ chat_id,
+ user_message['id'],
+ user_message,
+ )
+
+ # Link grandparent → user message (childrenIds)
+ grandparent_id = user_message.get('parentId')
+ if grandparent_id:
+ grandparent = await Chats.get_message_by_id_and_message_id(chat_id, grandparent_id)
+ if grandparent:
+ child_ids = grandparent.get('childrenIds', [])
+ if user_message['id'] not in child_ids:
+ child_ids.append(user_message['id'])
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
+ chat_id, grandparent_id, {'childrenIds': child_ids}
+ )
+
+ # Insert chat files from user message if any
+ user_message_files = user_message.get('files', [])
+ if user_message_files:
+ try:
+ await Chats.insert_chat_files(
+ chat_id,
+ user_message.get('id'),
+ [
+ file_item.get('id')
+ for file_item in user_message_files
+ if file_item.get('type') == 'file'
+ ],
+ user.id,
+ )
+ except Exception as e:
+ log.debug(f'Error inserting chat files: {e}')
+ pass
+
+ # Save ALL assistant placeholders
+ user_message_id = metadata.get('user_message_id')
+ all_assistant_ids = [assistant_id for assistant_id in message_ids.values() if assistant_id]
+
+ # Link user message → all assistant messages (childrenIds)
+ if user_message_id and all_assistant_ids:
+ existing_user_message = await Chats.get_message_by_id_and_message_id(chat_id, user_message_id)
+ if existing_user_message:
+ child_ids = existing_user_message.get('childrenIds', [])
+ for assistant_id in all_assistant_ids:
+ if assistant_id not in child_ids:
+ child_ids.append(assistant_id)
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
+ chat_id,
+ user_message_id,
+ {'childrenIds': child_ids},
+ )
+
+ # Save each assistant placeholder
+ for target_model_id, assistant_message_id in message_ids.items():
+ if assistant_message_id:
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
+ chat_id,
+ assistant_message_id,
+ {
+ 'id': assistant_message_id,
+ 'parentId': user_message_id,
+ 'childrenIds': [],
+ 'role': 'assistant',
+ 'content': '',
+ 'done': False,
+ 'model': target_model_id,
+ 'timestamp': int(time.time()),
+ },
+ )
request.state.metadata = metadata
form_data['metadata'] = metadata
@@ -1794,66 +1831,70 @@ async def chat_completion(
detail=str(e),
)
- async def process_chat(request, form_data, user, metadata, model):
+ async def process_chat(request, form_data, user, metadata, model, tasks=None):
try:
form_data, metadata, events = await process_chat_payload(request, form_data, user, metadata, model)
response = await chat_completion_handler(request, form_data, user)
- if metadata.get('chat_id') and metadata.get('message_id'):
- try:
- if not metadata['chat_id'].startswith('local:'):
- Chats.upsert_message_to_chat_by_id_and_message_id(
- metadata['chat_id'],
- metadata['message_id'],
- {
- 'parentId': metadata.get('parent_message_id', None),
- 'model': model_id,
- },
- )
- except Exception:
- pass
- ctx = build_chat_response_context(request, form_data, user, model, metadata, tasks, events)
+ # When the upstream provider returns an error (e.g. HTTP 400
+ # content-filter, quota exceeded), generate_chat_completion
+ # returns a JSONResponse instead of raising. Detect this and
+ # raise so the except-block below emits chat:message:error +
+ # chat:tasks:cancel, unblocking the frontend.
+ if isinstance(response, JSONResponse) and response.status_code >= 400:
+ try:
+ error_body = json.loads(response.body.decode('utf-8', 'replace'))
+ detail = error_body.get('error', error_body) if isinstance(error_body, dict) else error_body
+ if isinstance(detail, dict):
+ detail = detail.get('message', detail.get('detail', str(detail)))
+ except Exception:
+ detail = f'Provider returned HTTP {response.status_code}'
+ raise Exception(detail)
+
+ ctx = await build_chat_response_context(request, form_data, user, model, metadata, tasks, events)
return await process_chat_response(response, ctx)
except asyncio.CancelledError:
log.info('Chat processing was cancelled')
try:
- event_emitter = get_event_emitter(metadata)
- await asyncio.shield(
- event_emitter(
- {'type': 'chat:tasks:cancel'},
+ event_emitter = await get_event_emitter(metadata)
+ if event_emitter:
+ await asyncio.shield(
+ event_emitter(
+ {'type': 'chat:tasks:cancel'},
+ )
)
- )
except Exception as e:
pass
finally:
raise # re-raise to ensure proper task cancellation handling
except Exception as e:
- log.debug(f'Error processing chat payload: {e}')
+ log.error('Error processing chat payload: %s', e)
if metadata.get('chat_id') and metadata.get('message_id'):
# Update the chat message with the error
try:
if not metadata['chat_id'].startswith('local:'):
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
- 'parentId': metadata.get('parent_message_id', None),
+ 'parentId': metadata.get('user_message_id', None),
'error': {'content': str(e)},
},
)
- event_emitter = get_event_emitter(metadata)
- await event_emitter(
- {
- 'type': 'chat:message:error',
- 'data': {'error': {'content': str(e)}},
- }
- )
- await event_emitter(
- {'type': 'chat:tasks:cancel'},
- )
+ event_emitter = await get_event_emitter(metadata)
+ if event_emitter:
+ await event_emitter(
+ {
+ 'type': 'chat:message:error',
+ 'data': {'error': {'content': str(e)}},
+ }
+ )
+ await event_emitter(
+ {'type': 'chat:tasks:cancel'},
+ )
except Exception:
pass
@@ -1883,26 +1924,78 @@ async def chat_completion(
# Emit chat:active=false when task completes
try:
if metadata.get('chat_id'):
- event_emitter = get_event_emitter(metadata, update_db=False)
+ event_emitter = await get_event_emitter(metadata, update_db=False)
if event_emitter:
await event_emitter({'type': 'chat:active', 'data': {'active': False}})
except Exception as e:
log.debug(f'Error emitting chat:active: {e}')
- if metadata.get('session_id') and metadata.get('chat_id') and metadata.get('message_id'):
- # Asynchronous Chat Processing
- task_id, _ = await create_task(
- request.app.state.redis,
- process_chat(request, form_data, user, metadata, model),
- id=metadata['chat_id'],
- )
- # Emit chat:active=true when task starts
- event_emitter = get_event_emitter(metadata, update_db=False)
- if event_emitter:
- await event_emitter({'type': 'chat:active', 'data': {'active': True}})
- return {'status': True, 'task_id': task_id}
+ # Fan out: one task per model
+ if metadata.get('session_id') and metadata.get('chat_id'):
+ task_ids = []
+ chat_id = metadata['chat_id']
+
+ for idx, (target_model_id, assistant_message_id) in enumerate(message_ids.items()):
+ if not assistant_message_id:
+ continue
+
+ # Per-model metadata: own message_id + model
+ per_model_metadata = {
+ **metadata,
+ 'message_id': assistant_message_id,
+ }
+
+ # Per-model form_data: own model
+ model_form_data = {
+ **form_data,
+ 'model': target_model_id,
+ 'metadata': per_model_metadata,
+ }
+
+ # Resolve the model object for this specific model
+ resolved_model = request.app.state.MODELS.get(target_model_id, model)
+
+ # Only the first model runs title/tags generation;
+ # subsequent models only run follow-ups.
+ task_id, _ = await create_task(
+ request.app.state.redis,
+ process_chat(
+ request,
+ model_form_data,
+ user,
+ per_model_metadata,
+ resolved_model,
+ tasks
+ if idx == 0
+ else {
+ k: v
+ for k, v in (tasks or {}).items()
+ if k not in (TASKS.TITLE_GENERATION, TASKS.TAGS_GENERATION)
+ }
+ or None,
+ ),
+ id=chat_id,
+ )
+ task_ids.append(task_id)
+
+ # Emit chat:active=true
+ if task_ids:
+ event_emitter = await get_event_emitter(
+ {**metadata, 'message_id': list(message_ids.values())[0]},
+ update_db=False,
+ )
+ if event_emitter:
+ await event_emitter({'type': 'chat:active', 'data': {'active': True}})
+
+ return {
+ 'status': True,
+ 'task_ids': task_ids,
+ 'chat_id': chat_id,
+ }
else:
- return await process_chat(request, form_data, user, metadata, model)
+ # Legacy/direct: single model, synchronous
+ metadata['message_id'] = list(message_ids.values())[0]
+ return await process_chat(request, form_data, user, metadata, model, tasks)
# Alias for chat_completion (Legacy)
@@ -1976,6 +2069,8 @@ async def generate_messages(
@app.post('/api/chat/completed')
async def chat_completed(request: Request, form_data: dict, user=Depends(get_verified_user)):
+ """Deprecated: outlet filters now run inline during chat completion.
+ Kept for backward compatibility with external integrations."""
try:
model_item = form_data.pop('model_item', {})
@@ -2009,7 +2104,7 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user=De
@app.post('/api/tasks/stop/{task_id}')
-async def stop_task_endpoint(request: Request, task_id: str, user=Depends(get_verified_user)):
+async def stop_task_endpoint(request: Request, task_id: str, user=Depends(get_admin_user)):
try:
result = await stop_task(request.app.state.redis, task_id)
return result
@@ -2018,15 +2113,21 @@ async def stop_task_endpoint(request: Request, task_id: str, user=Depends(get_ve
@app.get('/api/tasks')
-async def list_tasks_endpoint(request: Request, user=Depends(get_verified_user)):
+async def list_tasks_endpoint(request: Request, user=Depends(get_admin_user)):
return {'tasks': await list_tasks(request.app.state.redis)}
-@app.get('/api/tasks/chat/{chat_id}')
+@app.get('/api/tasks/chat/{chat_id:path}')
async def list_tasks_by_chat_id_endpoint(request: Request, chat_id: str, user=Depends(get_verified_user)):
- chat = Chats.get_chat_by_id(chat_id)
- if chat is None or chat.user_id != user.id:
- return {'task_ids': []}
+ if chat_id.startswith('local:'):
+ socket_id = chat_id[len('local:') :]
+ owner_id = get_user_id_from_session_pool(socket_id)
+ if owner_id != user.id and user.role != 'admin':
+ return {'task_ids': []}
+ else:
+ chat = await Chats.get_chat_by_id(chat_id)
+ if chat is None or (chat.user_id != user.id and user.role != 'admin'):
+ return {'task_ids': []}
task_ids = await list_task_ids_by_item_id(request.app.state.redis, chat_id)
@@ -2034,6 +2135,21 @@ async def list_tasks_by_chat_id_endpoint(request: Request, chat_id: str, user=De
return {'task_ids': task_ids}
+@app.post('/api/tasks/chat/{chat_id:path}/stop')
+async def stop_tasks_by_chat_id_endpoint(request: Request, chat_id: str, user=Depends(get_verified_user)):
+ if chat_id.startswith('local:'):
+ socket_id = chat_id[len('local:') :]
+ owner_id = get_user_id_from_session_pool(socket_id)
+ if owner_id != user.id and user.role != 'admin':
+ raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
+ else:
+ chat = await Chats.get_chat_by_id(chat_id)
+ if chat is None or (chat.user_id != user.id and user.role != 'admin'):
+ raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
+ result = await stop_item_tasks(request.app.state.redis, chat_id)
+ return result
+
+
##################################
#
# Config Endpoints
@@ -2065,9 +2181,9 @@ async def get_app_config(request: Request):
detail='Invalid token',
)
if data is not None and 'id' in data:
- user = Users.get_user_by_id(data['id'])
+ user = await Users.get_user_by_id(data['id'])
- user_count = Users.get_num_users()
+ user_count = await Users.get_num_users()
onboarding = False
if user is None:
@@ -2088,6 +2204,7 @@ async def get_app_config(request: Request):
'enable_api_keys': app.state.config.ENABLE_API_KEYS,
'enable_signup': app.state.config.ENABLE_SIGNUP,
'enable_login_form': app.state.config.ENABLE_LOGIN_FORM,
+ 'enable_password_change_form': app.state.config.ENABLE_PASSWORD_CHANGE_FORM,
'enable_websocket': ENABLE_WEBSOCKET_SUPPORT,
'enable_version_update_check': ENABLE_VERSION_UPDATE_CHECK,
'enable_public_active_users_count': ENABLE_PUBLIC_ACTIVE_USERS_COUNT,
@@ -2276,7 +2393,7 @@ async def get_current_usage(user=Depends(get_verified_user)):
return {
'model_ids': get_models_in_use(),
- 'user_count': Users.get_active_user_count(),
+ 'user_count': await Users.get_active_user_count(),
}
except HTTPException:
raise
@@ -2298,10 +2415,8 @@ if len(app.state.config.TOOL_SERVER_CONNECTIONS) > 0:
auth_type = tool_server_connection.get('auth_type', 'none')
if server_id and auth_type in ('oauth_2.1', 'oauth_2.1_static'):
- oauth_client_info = tool_server_connection.get('info', {}).get('oauth_client_info', '')
-
try:
- oauth_client_info = decrypt_data(oauth_client_info)
+ oauth_client_info = resolve_oauth_client_info(tool_server_connection)
app.state.oauth_client_manager.add_client(
f'mcp:{server_id}',
OAuthClientInformationFull(**oauth_client_info),
@@ -2362,18 +2477,25 @@ async def register_client(request, client_id: str) -> bool:
try:
if auth_type == 'oauth_2.1_static':
- # Static credentials: rebuild from stored credentials + fresh metadata
- existing_client_info = connection.get('info', {}).get('oauth_client_info', '')
- if not existing_client_info:
- log.error(f'No stored OAuth client info for static client {client_id}')
- return False
- existing_data = decrypt_data(existing_client_info)
+ # Static credentials: rebuild from admin-provided credentials + fresh metadata
+ info = connection.get('info', {})
+ oauth_client_id = info.get('oauth_client_id') or ''
+ oauth_client_secret = info.get('oauth_client_secret') or ''
+ if not oauth_client_id or not oauth_client_secret:
+ # Fall back to blob for backward compatibility
+ existing_client_info = info.get('oauth_client_info', '')
+ if not existing_client_info:
+ log.error(f'No stored OAuth client info for static client {client_id}')
+ return False
+ existing_data = decrypt_data(existing_client_info)
+ oauth_client_id = oauth_client_id or existing_data.get('client_id', '')
+ oauth_client_secret = oauth_client_secret or existing_data.get('client_secret', '')
oauth_client_info = await get_oauth_client_info_with_static_credentials(
request,
client_id,
server_url,
- oauth_client_id=existing_data.get('client_id', ''),
- oauth_client_secret=existing_data.get('client_secret', ''),
+ oauth_client_id=oauth_client_id,
+ oauth_client_secret=oauth_client_secret,
)
else:
oauth_client_info = await get_oauth_client_info_with_dynamic_client_registration(
@@ -2483,7 +2605,7 @@ async def oauth_login_callback(
provider: str,
request: Request,
response: Response,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
return await oauth_manager.handle_callback(request, provider, response, db=db)
@@ -2496,7 +2618,7 @@ async def oauth_login_callback(
@app.post('/oauth/backchannel-logout')
async def oauth_backchannel_logout(
request: Request,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not ENABLE_OAUTH_BACKCHANNEL_LOGOUT:
raise HTTPException(status_code=404)
@@ -2506,7 +2628,13 @@ async def oauth_backchannel_logout(
@app.get('/manifest.json')
async def get_manifest_json():
if app.state.EXTERNAL_PWA_MANIFEST_URL:
- return requests.get(app.state.EXTERNAL_PWA_MANIFEST_URL).json()
+ session = await get_session()
+ async with session.get(
+ app.state.EXTERNAL_PWA_MANIFEST_URL,
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ return await r.json()
else:
return {
'name': app.state.WEBUI_NAME,
diff --git a/backend/open_webui/migrations/versions/e1f2a3b4c5d6_add_is_pinned_to_note.py b/backend/open_webui/migrations/versions/e1f2a3b4c5d6_add_is_pinned_to_note.py
new file mode 100644
index 0000000000..0d80558746
--- /dev/null
+++ b/backend/open_webui/migrations/versions/e1f2a3b4c5d6_add_is_pinned_to_note.py
@@ -0,0 +1,23 @@
+"""Add is_pinned to note table
+
+Revision ID: e1f2a3b4c5d6
+Revises: b7c8d9e0f1a2
+Create Date: 2026-04-14 22:00:00.000000
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+revision = 'e1f2a3b4c5d6'
+down_revision = 'b7c8d9e0f1a2'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+ op.add_column('note', sa.Column('is_pinned', sa.Boolean(), nullable=True))
+
+
+def downgrade():
+ op.drop_column('note', 'is_pinned')
diff --git a/backend/open_webui/models/access_grants.py b/backend/open_webui/models/access_grants.py
index 20601fd30e..f031495912 100644
--- a/backend/open_webui/models/access_grants.py
+++ b/backend/open_webui/models/access_grants.py
@@ -3,8 +3,9 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db_context
+from sqlalchemy import select, delete
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
from pydantic import BaseModel, ConfigDict
from sqlalchemy import BigInteger, Column, Text, UniqueConstraint, or_, and_
@@ -281,29 +282,28 @@ def grants_to_access_control(grants: list) -> Optional[dict]:
class AccessGrantsTable:
- def grant_access(
+ async def grant_access(
self,
resource_type: str,
resource_id: str,
principal_type: str,
principal_id: str,
permission: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[AccessGrantModel]:
"""Add a single access grant. Idempotent (ignores duplicates)."""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Check for existing grant
- existing = (
- db.query(AccessGrant)
- .filter_by(
+ result = await db.execute(
+ select(AccessGrant).filter_by(
resource_type=resource_type,
resource_id=resource_id,
principal_type=principal_type,
principal_id=principal_id,
permission=permission,
)
- .first()
)
+ existing = result.scalars().first()
if existing:
return AccessGrantModel.model_validate(existing)
@@ -317,71 +317,69 @@ class AccessGrantsTable:
created_at=int(time.time()),
)
db.add(grant)
- db.commit()
- db.refresh(grant)
+ await db.commit()
+ await db.refresh(grant)
return AccessGrantModel.model_validate(grant)
- def revoke_access(
+ async def revoke_access(
self,
resource_type: str,
resource_id: str,
principal_type: str,
principal_id: str,
permission: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
"""Remove a single access grant."""
- with get_db_context(db) as db:
- deleted = (
- db.query(AccessGrant)
- .filter_by(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ delete(AccessGrant).filter_by(
resource_type=resource_type,
resource_id=resource_id,
principal_type=principal_type,
principal_id=principal_id,
permission=permission,
)
- .delete()
)
- db.commit()
- return deleted > 0
+ await db.commit()
+ return result.rowcount > 0
- def revoke_all_access(
+ async def revoke_all_access(
self,
resource_type: str,
resource_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> int:
"""Remove all access grants for a resource."""
- with get_db_context(db) as db:
- deleted = (
- db.query(AccessGrant)
- .filter_by(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ delete(AccessGrant).filter_by(
resource_type=resource_type,
resource_id=resource_id,
)
- .delete()
)
- db.commit()
- return deleted
+ await db.commit()
+ return result.rowcount
- def set_access_control(
+ async def set_access_control(
self,
resource_type: str,
resource_id: str,
access_control: Optional[dict],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[AccessGrantModel]:
"""
Replace all grants for a resource from an access_control JSON dict.
This is the primary bridge for backward compat with the frontend.
"""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Delete all existing grants for this resource
- db.query(AccessGrant).filter_by(
- resource_type=resource_type,
- resource_id=resource_id,
- ).delete()
+ await db.execute(
+ delete(AccessGrant).filter_by(
+ resource_type=resource_type,
+ resource_id=resource_id,
+ )
+ )
# Convert JSON to grant dicts
grant_dicts = access_control_to_grants(resource_type, resource_id, access_control)
@@ -397,25 +395,27 @@ class AccessGrantsTable:
db.add(grant)
results.append(grant)
- db.commit()
+ await db.commit()
return [AccessGrantModel.model_validate(g) for g in results]
- def set_access_grants(
+ async def set_access_grants(
self,
resource_type: str,
resource_id: str,
access_grants: Optional[list],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[AccessGrantModel]:
"""
Replace all grants for a resource from a direct access_grants list.
"""
- with get_db_context(db) as db:
- db.query(AccessGrant).filter_by(
- resource_type=resource_type,
- resource_id=resource_id,
- ).delete()
+ async with get_async_db_context(db) as db:
+ await db.execute(
+ delete(AccessGrant).filter_by(
+ resource_type=resource_type,
+ resource_id=resource_id,
+ )
+ )
normalized_grants = normalize_access_grants(access_grants)
@@ -433,80 +433,77 @@ class AccessGrantsTable:
db.add(grant)
results.append(grant)
- db.commit()
+ await db.commit()
return [AccessGrantModel.model_validate(g) for g in results]
- def get_access_control(
+ async def get_access_control(
self,
resource_type: str,
resource_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[dict]:
"""
Reconstruct the old-style access_control JSON dict from grants.
For backward compat with the frontend.
"""
- with get_db_context(db) as db:
- grants = (
- db.query(AccessGrant)
- .filter_by(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(AccessGrant).filter_by(
resource_type=resource_type,
resource_id=resource_id,
)
- .all()
)
+ grants = result.scalars().all()
grant_models = [AccessGrantModel.model_validate(g) for g in grants]
return grants_to_access_control(grant_models)
- def get_grants_by_resource(
+ async def get_grants_by_resource(
self,
resource_type: str,
resource_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[AccessGrantModel]:
"""Get all grants for a specific resource."""
- with get_db_context(db) as db:
- grants = (
- db.query(AccessGrant)
- .filter_by(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(AccessGrant).filter_by(
resource_type=resource_type,
resource_id=resource_id,
)
- .all()
)
+ grants = result.scalars().all()
return [AccessGrantModel.model_validate(g) for g in grants]
- def get_grants_by_resources(
+ async def get_grants_by_resources(
self,
resource_type: str,
resource_ids: list[str],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, list[AccessGrantModel]]:
"""Batch-fetch grants for multiple resources. Returns {resource_id: [grants]}."""
if not resource_ids:
return {}
- with get_db_context(db) as db:
- grants = (
- db.query(AccessGrant)
- .filter(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(AccessGrant).filter(
AccessGrant.resource_type == resource_type,
AccessGrant.resource_id.in_(resource_ids),
)
- .all()
)
- result: dict[str, list[AccessGrantModel]] = {rid: [] for rid in resource_ids}
+ grants = result.scalars().all()
+ result_dict: dict[str, list[AccessGrantModel]] = {rid: [] for rid in resource_ids}
for g in grants:
- result[g.resource_id].append(AccessGrantModel.model_validate(g))
- return result
+ result_dict[g.resource_id].append(AccessGrantModel.model_validate(g))
+ return result_dict
- def has_access(
+ async def has_access(
self,
user_id: str,
resource_type: str,
resource_id: str,
permission: str = 'read',
user_group_ids: Optional[set[str]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
"""
Check if a user has the specified permission on a resource.
@@ -516,7 +513,7 @@ class AccessGrantsTable:
- There's a grant for the specific user with the requested permission
- There's a grant for any of the user's groups with the requested permission
"""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Build conditions for matching grants
conditions = [
# Public access
@@ -535,7 +532,7 @@ class AccessGrantsTable:
if user_group_ids is None:
from open_webui.models.groups import Groups
- user_groups = Groups.get_groups_by_member_id(user_id, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
user_group_ids = {group.id for group in user_groups}
if user_group_ids:
@@ -546,26 +543,27 @@ class AccessGrantsTable:
)
)
- exists = (
- db.query(AccessGrant)
+ result = await db.execute(
+ select(AccessGrant)
.filter(
AccessGrant.resource_type == resource_type,
AccessGrant.resource_id == resource_id,
AccessGrant.permission == permission,
or_(*conditions),
)
- .first()
+ .limit(1)
)
- return exists is not None
+ grant = result.scalars().first()
+ return grant is not None
- def get_accessible_resource_ids(
+ async def get_accessible_resource_ids(
self,
user_id: str,
resource_type: str,
resource_ids: list[str],
permission: str = 'read',
user_group_ids: Optional[set[str]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> set[str]:
"""
Batch check: return the subset of resource_ids that the user can access.
@@ -575,7 +573,7 @@ class AccessGrantsTable:
if not resource_ids:
return set()
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
conditions = [
and_(
AccessGrant.principal_type == 'user',
@@ -590,7 +588,7 @@ class AccessGrantsTable:
if user_group_ids is None:
from open_webui.models.groups import Groups
- user_groups = Groups.get_groups_by_member_id(user_id, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
user_group_ids = {group.id for group in user_groups}
if user_group_ids:
@@ -601,8 +599,8 @@ class AccessGrantsTable:
)
)
- rows = (
- db.query(AccessGrant.resource_id)
+ result = await db.execute(
+ select(AccessGrant.resource_id)
.filter(
AccessGrant.resource_type == resource_type,
AccessGrant.resource_id.in_(resource_ids),
@@ -610,16 +608,16 @@ class AccessGrantsTable:
or_(*conditions),
)
.distinct()
- .all()
)
+ rows = result.all()
return {row[0] for row in rows}
- def get_users_with_access(
+ async def get_users_with_access(
self,
resource_type: str,
resource_id: str,
permission: str = 'read',
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list:
"""
Get all users who have the specified permission on a resource.
@@ -628,21 +626,20 @@ class AccessGrantsTable:
from open_webui.models.users import Users, UserModel
from open_webui.models.groups import Groups
- with get_db_context(db) as db:
- grants = (
- db.query(AccessGrant)
- .filter_by(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(AccessGrant).filter_by(
resource_type=resource_type,
resource_id=resource_id,
permission=permission,
)
- .all()
)
+ grants = result.scalars().all()
# Check for public access
for grant in grants:
if grant.principal_type == 'user' and grant.principal_id == '*':
- result = Users.get_users(filter={'roles': ['!pending']}, db=db)
+ result = await Users.get_users(filter={'roles': ['!pending']}, db=db)
return result.get('users', [])
user_ids_with_access = set()
@@ -651,14 +648,14 @@ class AccessGrantsTable:
if grant.principal_type == 'user':
user_ids_with_access.add(grant.principal_id)
elif grant.principal_type == 'group':
- group_user_ids = Groups.get_group_user_ids_by_id(grant.principal_id, db=db)
+ group_user_ids = await Groups.get_group_user_ids_by_id(grant.principal_id, db=db)
if group_user_ids:
user_ids_with_access.update(group_user_ids)
if not user_ids_with_access:
return []
- return Users.get_users_by_user_ids(list(user_ids_with_access), db=db)
+ return await Users.get_users_by_user_ids(list(user_ids_with_access), db=db)
def has_permission_filter(
self,
@@ -673,6 +670,10 @@ class AccessGrantsTable:
Apply access control filtering to a SQLAlchemy query by JOINing with access_grant.
This replaces the old JSON-column-based filtering with a proper relational JOIN.
+
+ Note: This method builds SQLAlchemy expressions and does NOT perform I/O itself,
+ so it remains synchronous. The caller is responsible for executing the query
+ asynchronously with `await db.execute(...)`.
"""
group_ids = filter.get('group_ids', [])
user_id = filter.get('user_id')
@@ -718,7 +719,7 @@ class AccessGrantsTable:
# LEFT JOIN access_grant and filter
# We use a subquery approach to avoid duplicates from multiple matching grants
- from sqlalchemy import exists as sa_exists, select
+ from sqlalchemy import exists as sa_exists
grant_exists = (
select(AccessGrant.id)
@@ -776,11 +777,15 @@ class AccessGrantsTable:
"""
Filter for items where user has read BUT NOT write access.
Public items are NOT considered read_only.
+
+ Note: This method builds SQLAlchemy expressions and does NOT perform I/O itself,
+ so it remains synchronous. The caller is responsible for executing the query
+ asynchronously with `await db.execute(...)`.
"""
group_ids = filter.get('group_ids', [])
user_id = filter.get('user_id')
- from sqlalchemy import exists as sa_exists, select
+ from sqlalchemy import exists as sa_exists
# Has read grant (not public)
read_grant_exists = (
diff --git a/backend/open_webui/models/auths.py b/backend/open_webui/models/auths.py
index 1a1b164c12..2c8c6ba99f 100644
--- a/backend/open_webui/models/auths.py
+++ b/backend/open_webui/models/auths.py
@@ -2,8 +2,9 @@ import logging
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.users import User, UserModel, UserProfileImageResponse, Users
from open_webui.utils.validate import validate_profile_image_url
from pydantic import BaseModel, field_validator
@@ -88,7 +89,7 @@ class AddUserForm(SignupForm):
class AuthsTable:
- def insert_new_auth(
+ async def insert_new_auth(
self,
email: str,
password: str,
@@ -96,9 +97,9 @@ class AuthsTable:
profile_image_url: str = '/user.png',
role: str = 'pending',
oauth: Optional[dict] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[UserModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
log.info('insert_new_auth')
id = str(uuid.uuid4())
@@ -107,28 +108,29 @@ class AuthsTable:
result = Auth(**auth.model_dump())
db.add(result)
- user = Users.insert_new_user(id, name, email, profile_image_url, role, oauth=oauth, db=db)
+ user = await Users.insert_new_user(id, name, email, profile_image_url, role, oauth=oauth, db=db)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result and user:
return user
else:
return None
- def authenticate_user(
- self, email: str, verify_password: callable, db: Optional[Session] = None
+ async def authenticate_user(
+ self, email: str, verify_password: callable, db: Optional[AsyncSession] = None
) -> Optional[UserModel]:
log.info(f'authenticate_user: {email}')
- user = Users.get_user_by_email(email, db=db)
+ user = await Users.get_user_by_email(email, db=db)
if not user:
return None
try:
- with get_db_context(db) as db:
- auth = db.query(Auth).filter_by(id=user.id, active=True).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Auth).filter_by(id=user.id, active=True))
+ auth = result.scalars().first()
if auth:
if verify_password(auth.password):
return user
@@ -139,66 +141,66 @@ class AuthsTable:
except Exception:
return None
- def authenticate_user_by_api_key(self, api_key: str, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def authenticate_user_by_api_key(
+ self, api_key: str, db: Optional[AsyncSession] = None
+ ) -> Optional[UserModel]:
log.info(f'authenticate_user_by_api_key')
# if no api_key, return None
if not api_key:
return None
try:
- user = Users.get_user_by_api_key(api_key, db=db)
+ user = await Users.get_user_by_api_key(api_key, db=db)
return user if user else None
except Exception:
return False
- def authenticate_user_by_email(self, email: str, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def authenticate_user_by_email(self, email: str, db: Optional[AsyncSession] = None) -> Optional[UserModel]:
log.info(f'authenticate_user_by_email: {email}')
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Single JOIN query instead of two separate queries
- result = (
- db.query(Auth, User)
- .join(User, Auth.id == User.id)
- .filter(Auth.email == email, Auth.active == True)
- .first()
+ result = await db.execute(
+ select(Auth, User).join(User, Auth.id == User.id).filter(Auth.email == email, Auth.active == True)
)
- if result:
- _, user = result
+ row = result.first()
+ if row:
+ _, user = row
return UserModel.model_validate(user)
return None
except Exception:
return None
- def update_user_password_by_id(self, id: str, new_password: str, db: Optional[Session] = None) -> bool:
+ async def update_user_password_by_id(self, id: str, new_password: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- result = db.query(Auth).filter_by(id=id).update({'password': new_password})
- db.commit()
- return True if result == 1 else False
+ async with get_async_db_context(db) as db:
+ result = await db.execute(update(Auth).filter_by(id=id).values(password=new_password))
+ await db.commit()
+ return True if result.rowcount == 1 else False
except Exception:
return False
- def update_email_by_id(self, id: str, email: str, db: Optional[Session] = None) -> bool:
+ async def update_email_by_id(self, id: str, email: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- result = db.query(Auth).filter_by(id=id).update({'email': email})
- db.commit()
- if result == 1:
- Users.update_user_by_id(id, {'email': email}, db=db)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(update(Auth).filter_by(id=id).values(email=email))
+ await db.commit()
+ if result.rowcount == 1:
+ await Users.update_user_by_id(id, {'email': email}, db=db)
return True
return False
except Exception:
return False
- def delete_auth_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_auth_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Delete User
- result = Users.delete_user_by_id(id, db=db)
+ result = await Users.delete_user_by_id(id, db=db)
if result:
- db.query(Auth).filter_by(id=id).delete()
- db.commit()
+ await db.execute(delete(Auth).filter_by(id=id))
+ await db.commit()
return True
else:
diff --git a/backend/open_webui/models/automations.py b/backend/open_webui/models/automations.py
index 485f097d5f..c891c3204e 100644
--- a/backend/open_webui/models/automations.py
+++ b/backend/open_webui/models/automations.py
@@ -4,10 +4,10 @@ from typing import Optional
from uuid import uuid4
from pydantic import BaseModel, ConfigDict
-from sqlalchemy import Column, Text, JSON, Boolean, BigInteger, Index, select, or_, func, cast, String
-from sqlalchemy.orm import Session
+from sqlalchemy import Column, Text, JSON, Boolean, BigInteger, Index, select, or_, func, cast, String, delete, update
+from sqlalchemy.ext.asyncio import AsyncSession
-from open_webui.internal.db import Base, get_db, get_db_context
+from open_webui.internal.db import Base, get_async_db_context
log = logging.getLogger(__name__)
@@ -45,7 +45,10 @@ class AutomationRun(Base):
error = Column(Text, nullable=True)
created_at = Column(BigInteger, nullable=False)
- __table_args__ = (Index('ix_automation_run_automation_id', 'automation_id'),)
+ __table_args__ = (
+ Index('ix_automation_run_automation_id', 'automation_id'),
+ Index('ix_automation_run_aid_created', 'automation_id', 'created_at'),
+ )
####################
@@ -115,14 +118,14 @@ class AutomationListResponse(BaseModel):
class AutomationTable:
- def insert(
+ async def insert(
self,
user_id: str,
form: AutomationForm,
next_run_at: int,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> AutomationModel:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
now = int(time.time_ns())
row = Automation(
id=str(uuid4()),
@@ -136,31 +139,36 @@ class AutomationTable:
updated_at=now,
)
db.add(row)
- db.commit()
- db.refresh(row)
+ await db.commit()
+ await db.refresh(row)
return AutomationModel.model_validate(row)
- def get_by_id(self, id: str, db: Optional[Session] = None) -> Optional[AutomationModel]:
- with get_db_context(db) as db:
- row = db.get(Automation, id)
+ async def count_by_user(self, user_id: str, db: Optional[AsyncSession] = None) -> int:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(func.count()).select_from(Automation).filter_by(user_id=user_id))
+ return result.scalar()
+
+ async def get_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[AutomationModel]:
+ async with get_async_db_context(db) as db:
+ row = await db.get(Automation, id)
return AutomationModel.model_validate(row) if row else None
- def search_automations(
+ async def search_automations(
self,
user_id: str,
query: Optional[str] = None,
status: Optional[str] = None,
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> 'AutomationListResponse':
- with get_db_context(db) as db:
- q = db.query(Automation).filter_by(user_id=user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Automation).filter_by(user_id=user_id)
if query:
search = f'%{query}%'
# Search in name and prompt inside JSON data
- q = q.filter(
+ stmt = stmt.filter(
or_(
Automation.name.ilike(search),
cast(Automation.data, String).ilike(search),
@@ -168,34 +176,37 @@ class AutomationTable:
)
if status == 'active':
- q = q.filter(Automation.is_active == True)
+ stmt = stmt.filter(Automation.is_active == True)
elif status == 'paused':
- q = q.filter(Automation.is_active == False)
+ stmt = stmt.filter(Automation.is_active == False)
- q = q.order_by(Automation.created_at.desc())
+ stmt = stmt.order_by(Automation.created_at.desc())
- total = q.count()
+ # Get total count
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- q = q.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- q = q.limit(limit)
+ stmt = stmt.limit(limit)
- rows = q.all()
+ result = await db.execute(stmt)
+ rows = result.scalars().all()
return AutomationListResponse(
items=[AutomationModel.model_validate(r) for r in rows],
total=total,
)
- def update_by_id(
+ async def update_by_id(
self,
id: str,
form: AutomationForm,
next_run_at: int,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[AutomationModel]:
- with get_db_context(db) as db:
- row = db.get(Automation, id)
+ async with get_async_db_context(db) as db:
+ row = await db.get(Automation, id)
if not row:
return None
row.name = form.name
@@ -205,37 +216,37 @@ class AutomationTable:
row.is_active = form.is_active
row.next_run_at = next_run_at
row.updated_at = int(time.time_ns())
- db.commit()
- db.refresh(row)
+ await db.commit()
+ await db.refresh(row)
return AutomationModel.model_validate(row)
- def toggle(
+ async def toggle(
self,
id: str,
next_run_at: Optional[int],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[AutomationModel]:
- with get_db_context(db) as db:
- row = db.get(Automation, id)
+ async with get_async_db_context(db) as db:
+ row = await db.get(Automation, id)
if not row:
return None
row.is_active = not row.is_active
row.next_run_at = next_run_at if row.is_active else None
row.updated_at = int(time.time_ns())
- db.commit()
- db.refresh(row)
+ await db.commit()
+ await db.refresh(row)
return AutomationModel.model_validate(row)
- def delete(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- row = db.get(Automation, id)
+ async def delete(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ row = await db.get(Automation, id)
if not row:
return False
- db.delete(row)
- db.commit()
+ await db.delete(row)
+ await db.commit()
return True
- def claim_due(self, now_ns: int, limit: int = 10, db: Optional[Session] = None) -> list[AutomationModel]:
+ async def claim_due(self, now_ns: int, limit: int = 10, db: Optional[AsyncSession] = None) -> list[AutomationModel]:
"""
Atomically claim due automations for execution.
@@ -243,7 +254,7 @@ class AutomationTable:
double-claimed. On PostgreSQL, uses FOR UPDATE SKIP LOCKED
for zero-contention distributed work claiming.
"""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
stmt = (
select(Automation)
.where(
@@ -257,7 +268,8 @@ class AutomationTable:
if db.bind.dialect.name == 'postgresql':
stmt = stmt.with_for_update(skip_locked=True)
- rows = db.execute(stmt).scalars().all()
+ result = await db.execute(stmt)
+ rows = result.scalars().all()
from open_webui.utils.automations import next_run_ns
@@ -265,7 +277,7 @@ class AutomationTable:
row.last_run_at = now_ns
row.next_run_at = next_run_ns(row.data.get('rrule', ''))
- db.commit()
+ await db.commit()
return [AutomationModel.model_validate(r) for r in rows]
@@ -276,15 +288,15 @@ class AutomationTable:
class AutomationRunTable:
- def insert(
+ async def insert(
self,
automation_id: str,
status: str,
chat_id: Optional[str] = None,
error: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> AutomationRunModel:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
row = AutomationRun(
id=str(uuid4()),
automation_id=automation_id,
@@ -294,43 +306,71 @@ class AutomationRunTable:
created_at=int(time.time_ns()),
)
db.add(row)
- db.commit()
- db.refresh(row)
+ await db.commit()
+ await db.refresh(row)
return AutomationRunModel.model_validate(row)
- def get_latest(self, automation_id: str, db: Optional[Session] = None) -> Optional[AutomationRunModel]:
- with get_db_context(db) as db:
- row = (
- db.query(AutomationRun)
+ async def get_latest(self, automation_id: str, db: Optional[AsyncSession] = None) -> Optional[AutomationRunModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(AutomationRun)
.filter_by(automation_id=automation_id)
.order_by(AutomationRun.created_at.desc())
- .first()
+ .limit(1)
)
+ row = result.scalars().first()
return AutomationRunModel.model_validate(row) if row else None
- def get_by_automation(
+ async def get_latest_batch(
+ self, automation_ids: list[str], db: Optional[AsyncSession] = None
+ ) -> dict[str, AutomationRunModel]:
+ """Fetch the latest run for each automation in a single query."""
+ if not automation_ids:
+ return {}
+ async with get_async_db_context(db) as db:
+ # Subquery: max created_at per automation_id
+ subq = (
+ select(
+ AutomationRun.automation_id,
+ func.max(AutomationRun.created_at).label('max_created'),
+ )
+ .filter(AutomationRun.automation_id.in_(automation_ids))
+ .group_by(AutomationRun.automation_id)
+ .subquery()
+ )
+ result = await db.execute(
+ select(AutomationRun).join(
+ subq,
+ (AutomationRun.automation_id == subq.c.automation_id)
+ & (AutomationRun.created_at == subq.c.max_created),
+ )
+ )
+ rows = result.scalars().all()
+ return {row.automation_id: AutomationRunModel.model_validate(row) for row in rows}
+
+ async def get_by_automation(
self,
automation_id: str,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[AutomationRunModel]:
- with get_db_context(db) as db:
- rows = (
- db.query(AutomationRun)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(AutomationRun)
.filter_by(automation_id=automation_id)
.order_by(AutomationRun.created_at.desc())
.offset(skip)
.limit(limit)
- .all()
)
+ rows = result.scalars().all()
return [AutomationRunModel.model_validate(r) for r in rows]
- def delete_by_automation(self, automation_id: str, db: Optional[Session] = None) -> int:
- with get_db_context(db) as db:
- count = db.query(AutomationRun).filter_by(automation_id=automation_id).delete()
- db.commit()
- return count
+ async def delete_by_automation(self, automation_id: str, db: Optional[AsyncSession] = None) -> int:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(delete(AutomationRun).filter_by(automation_id=automation_id))
+ await db.commit()
+ return result.rowcount
Automations = AutomationTable()
diff --git a/backend/open_webui/models/channels.py b/backend/open_webui/models/channels.py
index 4d773491d5..942c06d6b3 100644
--- a/backend/open_webui/models/channels.py
+++ b/backend/open_webui/models/channels.py
@@ -4,8 +4,9 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update, func, case, or_, and_
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.groups import Groups
from open_webui.models.access_grants import (
AccessGrantModel,
@@ -25,11 +26,7 @@ from sqlalchemy import (
Text,
JSON,
UniqueConstraint,
- case,
- cast,
)
-from sqlalchemy import or_, func, select, and_, text
-from sqlalchemy.sql import exists
####################
# Channel DB Schema
@@ -249,22 +246,22 @@ class ChannelWebhookForm(BaseModel):
class ChannelTable:
- def _get_access_grants(self, channel_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('channel', channel_id, db=db)
+ async def _get_access_grants(self, channel_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('channel', channel_id, db=db)
- def _to_channel_model(
+ async def _to_channel_model(
self,
channel: Channel,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> ChannelModel:
channel_data = ChannelModel.model_validate(channel).model_dump(exclude={'access_grants'})
channel_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(channel_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(channel_data['id'], db=db)
)
return ChannelModel.model_validate(channel_data)
- def _collect_unique_user_ids(
+ async def _collect_unique_user_ids(
self,
invited_by: str,
user_ids: Optional[list[str]] = None,
@@ -281,7 +278,8 @@ class ChannelTable:
users.add(invited_by)
for group_id in group_ids or []:
- users.update(Groups.get_group_user_ids_by_id(group_id))
+ group_user_ids = await Groups.get_group_user_ids_by_id(group_id)
+ users.update(group_user_ids)
return users
@@ -321,10 +319,20 @@ class ChannelTable:
return memberships
- def insert_new_channel(
- self, form_data: CreateChannelForm, user_id: str, db: Optional[Session] = None
+ def _has_permission(self, db, query, filter: dict, permission: str = 'read'):
+ return AccessGrants.has_permission_filter(
+ db=db,
+ query=query,
+ DocumentModel=Channel,
+ filter=filter,
+ resource_type='channel',
+ permission=permission,
+ )
+
+ async def insert_new_channel(
+ self, form_data: CreateChannelForm, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[ChannelModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
channel = ChannelModel(
**{
**form_data.model_dump(exclude={'access_grants'}),
@@ -340,7 +348,7 @@ class ChannelTable:
new_channel = Channel(**channel.model_dump(exclude={'access_grants'}))
if form_data.type in ['group', 'dm']:
- users = self._collect_unique_user_ids(
+ users = await self._collect_unique_user_ids(
invited_by=user_id,
user_ids=form_data.user_ids,
group_ids=form_data.group_ids,
@@ -353,17 +361,18 @@ class ChannelTable:
db.add_all(memberships)
db.add(new_channel)
- db.commit()
- AccessGrants.set_access_grants('channel', new_channel.id, form_data.access_grants, db=db)
- return self._to_channel_model(new_channel, db=db)
+ await db.commit()
+ await AccessGrants.set_access_grants('channel', new_channel.id, form_data.access_grants, db=db)
+ return await self._to_channel_model(new_channel, db=db)
- def get_channels(self, db: Optional[Session] = None) -> list[ChannelModel]:
- with get_db_context(db) as db:
- channels = db.query(Channel).all()
+ async def get_channels(self, db: Optional[AsyncSession] = None) -> list[ChannelModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Channel))
+ channels = result.scalars().all()
channel_ids = [channel.id for channel in channels]
- grants_map = AccessGrants.get_grants_by_resources('channel', channel_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('channel', channel_ids, db=db)
return [
- self._to_channel_model(
+ await self._to_channel_model(
channel,
access_grants=grants_map.get(channel.id, []),
db=db,
@@ -371,22 +380,12 @@ class ChannelTable:
for channel in channels
]
- def _has_permission(self, db, query, filter: dict, permission: str = 'read'):
- return AccessGrants.has_permission_filter(
- db=db,
- query=query,
- DocumentModel=Channel,
- filter=filter,
- resource_type='channel',
- permission=permission,
- )
+ async def get_channels_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[ChannelModel]:
+ async with get_async_db_context(db) as db:
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id, db=db)]
- def get_channels_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[ChannelModel]:
- with get_db_context(db) as db:
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id, db=db)]
-
- membership_channels = (
- db.query(Channel)
+ result = await db.execute(
+ select(Channel)
.join(ChannelMember, Channel.id == ChannelMember.channel_id)
.filter(
Channel.deleted_at.is_(None),
@@ -395,10 +394,10 @@ class ChannelTable:
ChannelMember.user_id == user_id,
ChannelMember.is_active.is_(True),
)
- .all()
)
+ membership_channels = result.scalars().all()
- query = db.query(Channel).filter(
+ stmt = select(Channel).filter(
Channel.deleted_at.is_(None),
Channel.archived_at.is_(None),
or_(
@@ -407,17 +406,22 @@ class ChannelTable:
and_(Channel.type != 'group', Channel.type != 'dm'),
),
)
- query = self._has_permission(db, query, {'user_id': user_id, 'group_ids': user_group_ids})
+ stmt = self._has_permission(db, stmt, {'user_id': user_id, 'group_ids': user_group_ids})
- standard_channels = query.all()
+ result = await db.execute(stmt)
+ standard_channels = result.scalars().all()
- all_channels = membership_channels + standard_channels
+ all_channels = list(membership_channels) + list(standard_channels)
channel_ids = [c.id for c in all_channels]
- grants_map = AccessGrants.get_grants_by_resources('channel', channel_ids, db=db)
- return [self._to_channel_model(c, access_grants=grants_map.get(c.id, []), db=db) for c in all_channels]
+ grants_map = await AccessGrants.get_grants_by_resources('channel', channel_ids, db=db)
+ return [
+ await self._to_channel_model(c, access_grants=grants_map.get(c.id, []), db=db) for c in all_channels
+ ]
- def get_dm_channel_by_user_ids(self, user_ids: list[str], db: Optional[Session] = None) -> Optional[ChannelModel]:
- with get_db_context(db) as db:
+ async def get_dm_channel_by_user_ids(
+ self, user_ids: list[str], db: Optional[AsyncSession] = None
+ ) -> Optional[ChannelModel]:
+ async with get_async_db_context(db) as db:
# Ensure uniqueness in case a list with duplicates is passed
unique_user_ids = list(set(user_ids))
@@ -429,7 +433,7 @@ class ChannelTable:
)
subquery = (
- db.query(ChannelMember.channel_id)
+ select(ChannelMember.channel_id)
.group_by(ChannelMember.channel_id)
# 1. Channel must have exactly len(user_ids) members
.having(func.count(ChannelMember.user_id) == len(unique_user_ids))
@@ -438,33 +442,32 @@ class ChannelTable:
.subquery()
)
- channel = (
- db.query(Channel)
+ result = await db.execute(
+ select(Channel)
.filter(
- Channel.id.in_(subquery),
+ Channel.id.in_(select(subquery.c.channel_id)),
Channel.type == 'dm',
)
- .first()
+ .limit(1)
)
+ channel = result.scalars().first()
- return self._to_channel_model(channel, db=db) if channel else None
+ return await self._to_channel_model(channel, db=db) if channel else None
- def add_members_to_channel(
+ async def add_members_to_channel(
self,
channel_id: str,
invited_by: str,
user_ids: Optional[list[str]] = None,
group_ids: Optional[list[str]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChannelMemberModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# 1. Collect all user_ids including groups + inviter
- requested_users = self._collect_unique_user_ids(invited_by, user_ids, group_ids)
+ requested_users = await self._collect_unique_user_ids(invited_by, user_ids, group_ids)
- existing_users = {
- row.user_id
- for row in db.query(ChannelMember.user_id).filter(ChannelMember.channel_id == channel_id).all()
- }
+ result = await db.execute(select(ChannelMember.user_id).filter(ChannelMember.channel_id == channel_id))
+ existing_users = {row[0] for row in result.all()}
new_user_ids = requested_users - existing_users
if not new_user_ids:
@@ -473,58 +476,56 @@ class ChannelTable:
new_memberships = self._create_membership_models(channel_id, invited_by, new_user_ids)
db.add_all(new_memberships)
- db.commit()
+ await db.commit()
return [ChannelMemberModel.model_validate(membership) for membership in new_memberships]
- def remove_members_from_channel(
+ async def remove_members_from_channel(
self,
channel_id: str,
user_ids: list[str],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> int:
- with get_db_context(db) as db:
- result = (
- db.query(ChannelMember)
- .filter(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ delete(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id.in_(user_ids),
)
- .delete(synchronize_session=False)
)
- db.commit()
- return result # number of rows deleted
+ await db.commit()
+ return result.rowcount # number of rows deleted
- def is_user_channel_manager(self, channel_id: str, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- # Check if the user is the creator of the channel
- # or has a 'manager' role in ChannelMember
- channel = db.query(Channel).filter(Channel.id == channel_id).first()
+ async def is_user_channel_manager(self, channel_id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Channel).filter(Channel.id == channel_id))
+ channel = result.scalars().first()
if channel and channel.user_id == user_id:
return True
- membership = (
- db.query(ChannelMember)
- .filter(
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
+ ChannelMember.is_active.is_(True),
ChannelMember.role == 'manager',
)
- .first()
)
+ membership = result.scalars().first()
return membership is not None
- def join_channel(self, channel_id: str, user_id: str, db: Optional[Session] = None) -> Optional[ChannelMemberModel]:
- with get_db_context(db) as db:
+ async def join_channel(
+ self, channel_id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[ChannelMemberModel]:
+ async with get_async_db_context(db) as db:
# Check if the membership already exists
- existing_membership = (
- db.query(ChannelMember)
- .filter(
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
)
- .first()
)
+ existing_membership = result.scalars().first()
if existing_membership:
return ChannelMemberModel.model_validate(existing_membership)
@@ -548,19 +549,18 @@ class ChannelTable:
new_membership = ChannelMember(**channel_member.model_dump())
db.add(new_membership)
- db.commit()
+ await db.commit()
return channel_member
- def leave_channel(self, channel_id: str, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- membership = (
- db.query(ChannelMember)
- .filter(
+ async def leave_channel(self, channel_id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
)
- .first()
)
+ membership = result.scalars().first()
if not membership:
return False
@@ -569,125 +569,131 @@ class ChannelTable:
membership.left_at = int(time.time_ns())
membership.updated_at = int(time.time_ns())
- db.commit()
+ await db.commit()
return True
- def get_member_by_channel_and_user_id(
- self, channel_id: str, user_id: str, db: Optional[Session] = None
+ async def get_member_by_channel_and_user_id(
+ self, channel_id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[ChannelMemberModel]:
- with get_db_context(db) as db:
- membership = (
- db.query(ChannelMember)
- .filter(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
)
- .first()
)
+ membership = result.scalars().first()
return ChannelMemberModel.model_validate(membership) if membership else None
- def get_members_by_channel_id(self, channel_id: str, db: Optional[Session] = None) -> list[ChannelMemberModel]:
- with get_db_context(db) as db:
- memberships = db.query(ChannelMember).filter(ChannelMember.channel_id == channel_id).all()
+ async def get_members_by_channel_id(
+ self, channel_id: str, db: Optional[AsyncSession] = None
+ ) -> list[ChannelMemberModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelMember).filter(ChannelMember.channel_id == channel_id))
+ memberships = result.scalars().all()
return [ChannelMemberModel.model_validate(membership) for membership in memberships]
- def pin_channel(
+ async def pin_channel(
self,
channel_id: str,
user_id: str,
is_pinned: bool,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
- with get_db_context(db) as db:
- membership = (
- db.query(ChannelMember)
- .filter(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
)
- .first()
)
+ membership = result.scalars().first()
if not membership:
return False
membership.is_channel_pinned = is_pinned
membership.updated_at = int(time.time_ns())
- db.commit()
+ await db.commit()
return True
- def update_member_last_read_at(self, channel_id: str, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- membership = (
- db.query(ChannelMember)
- .filter(
+ async def update_member_last_read_at(
+ self, channel_id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
)
- .first()
)
+ membership = result.scalars().first()
if not membership:
return False
membership.last_read_at = int(time.time_ns())
membership.updated_at = int(time.time_ns())
- db.commit()
+ await db.commit()
return True
- def update_member_active_status(
+ async def update_member_active_status(
self,
channel_id: str,
user_id: str,
is_active: bool,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
- with get_db_context(db) as db:
- membership = (
- db.query(ChannelMember)
- .filter(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelMember).filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
)
- .first()
)
+ membership = result.scalars().first()
if not membership:
return False
membership.is_active = is_active
membership.updated_at = int(time.time_ns())
- db.commit()
+ await db.commit()
return True
- def is_user_channel_member(self, channel_id: str, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- membership = (
- db.query(ChannelMember)
+ async def is_user_channel_member(self, channel_id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelMember)
.filter(
ChannelMember.channel_id == channel_id,
ChannelMember.user_id == user_id,
+ ChannelMember.is_active.is_(True),
)
- .first()
+ .limit(1)
)
+ membership = result.scalars().first()
return membership is not None
- def get_channel_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ChannelModel]:
+ async def get_channel_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ChannelModel]:
try:
- with get_db_context(db) as db:
- channel = db.query(Channel).filter(Channel.id == id).first()
- return self._to_channel_model(channel, db=db) if channel else None
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Channel).filter(Channel.id == id))
+ channel = result.scalars().first()
+ return await self._to_channel_model(channel, db=db) if channel else None
except Exception:
return None
- def get_channels_by_file_id(self, file_id: str, db: Optional[Session] = None) -> list[ChannelModel]:
- with get_db_context(db) as db:
- channel_files = db.query(ChannelFile).filter(ChannelFile.file_id == file_id).all()
+ async def get_channels_by_file_id(self, file_id: str, db: Optional[AsyncSession] = None) -> list[ChannelModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelFile).filter(ChannelFile.file_id == file_id))
+ channel_files = result.scalars().all()
channel_ids = [cf.channel_id for cf in channel_files]
- channels = db.query(Channel).filter(Channel.id.in_(channel_ids)).all()
- grants_map = AccessGrants.get_grants_by_resources('channel', channel_ids, db=db)
+ result = await db.execute(select(Channel).filter(Channel.id.in_(channel_ids)))
+ channels = result.scalars().all()
+ grants_map = await AccessGrants.get_grants_by_resources('channel', channel_ids, db=db)
return [
- self._to_channel_model(
+ await self._to_channel_model(
channel,
access_grants=grants_map.get(channel.id, []),
db=db,
@@ -695,123 +701,127 @@ class ChannelTable:
for channel in channels
]
- def get_channels_by_file_id_and_user_id(
- self, file_id: str, user_id: str, db: Optional[Session] = None
+ async def get_channels_by_file_id_and_user_id(
+ self, file_id: str, user_id: str, db: Optional[AsyncSession] = None
) -> list[ChannelModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# 1. Determine which channels have this file
- channel_file_rows = db.query(ChannelFile).filter(ChannelFile.file_id == file_id).all()
+ result = await db.execute(select(ChannelFile).filter(ChannelFile.file_id == file_id))
+ channel_file_rows = result.scalars().all()
channel_ids = [row.channel_id for row in channel_file_rows]
if not channel_ids:
return []
# 2. Load all channel rows that still exist
- channels = (
- db.query(Channel)
- .filter(
+ result = await db.execute(
+ select(Channel).filter(
Channel.id.in_(channel_ids),
Channel.deleted_at.is_(None),
Channel.archived_at.is_(None),
)
- .all()
)
+ channels = result.scalars().all()
if not channels:
return []
# Preload user's group membership
- user_group_ids = [g.id for g in Groups.get_groups_by_member_id(user_id, db=db)]
+ user_group_ids = [g.id for g in await Groups.get_groups_by_member_id(user_id, db=db)]
allowed_channels = []
for channel in channels:
# --- Case A: group or dm => user must be an active member ---
if channel.type in ['group', 'dm']:
- membership = (
- db.query(ChannelMember)
+ result = await db.execute(
+ select(ChannelMember)
.filter(
ChannelMember.channel_id == channel.id,
ChannelMember.user_id == user_id,
ChannelMember.is_active.is_(True),
)
- .first()
+ .limit(1)
)
+ membership = result.scalars().first()
if membership:
- allowed_channels.append(self._to_channel_model(channel, db=db))
+ allowed_channels.append(await self._to_channel_model(channel, db=db))
continue
# --- Case B: standard channel => rely on ACL permissions ---
- query = db.query(Channel).filter(Channel.id == channel.id)
+ stmt = select(Channel).filter(Channel.id == channel.id)
- query = self._has_permission(
+ stmt = self._has_permission(
db,
- query,
+ stmt,
{'user_id': user_id, 'group_ids': user_group_ids},
permission='read',
)
- allowed = query.first()
+ result = await db.execute(stmt)
+ allowed = result.scalars().first()
if allowed:
- allowed_channels.append(self._to_channel_model(allowed, db=db))
+ allowed_channels.append(await self._to_channel_model(allowed, db=db))
return allowed_channels
- def get_channel_by_id_and_user_id(
- self, id: str, user_id: str, db: Optional[Session] = None
+ async def get_channel_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[ChannelModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Fetch the channel
- channel: Channel = (
- db.query(Channel)
- .filter(
+ result = await db.execute(
+ select(Channel).filter(
Channel.id == id,
Channel.deleted_at.is_(None),
Channel.archived_at.is_(None),
)
- .first()
)
+ channel = result.scalars().first()
if not channel:
return None
# If the channel is a group or dm, read access requires membership (active)
if channel.type in ['group', 'dm']:
- membership = (
- db.query(ChannelMember)
+ result = await db.execute(
+ select(ChannelMember)
.filter(
ChannelMember.channel_id == id,
ChannelMember.user_id == user_id,
ChannelMember.is_active.is_(True),
)
- .first()
+ .limit(1)
)
+ membership = result.scalars().first()
if membership:
- return self._to_channel_model(channel, db=db)
+ return await self._to_channel_model(channel, db=db)
else:
return None
# For channels that are NOT group/dm, fall back to ACL-based read access
- query = db.query(Channel).filter(Channel.id == id)
+ stmt = select(Channel).filter(Channel.id == id)
# Determine user groups
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id, db=db)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id, db=db)]
# Apply ACL rules
- query = self._has_permission(
+ stmt = self._has_permission(
db,
- query,
+ stmt,
{'user_id': user_id, 'group_ids': user_group_ids},
permission='read',
)
- channel_allowed = query.first()
- return self._to_channel_model(channel_allowed, db=db) if channel_allowed else None
+ result = await db.execute(stmt)
+ channel_allowed = result.scalars().first()
+ return await self._to_channel_model(channel_allowed, db=db) if channel_allowed else None
- def update_channel_by_id(
- self, id: str, form_data: ChannelForm, db: Optional[Session] = None
+ async def update_channel_by_id(
+ self, id: str, form_data: ChannelForm, db: Optional[AsyncSession] = None
) -> Optional[ChannelModel]:
- with get_db_context(db) as db:
- channel = db.query(Channel).filter(Channel.id == id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Channel).filter(Channel.id == id))
+ channel = result.scalars().first()
if not channel:
return None
@@ -823,16 +833,16 @@ class ChannelTable:
channel.meta = form_data.meta
if form_data.access_grants is not None:
- AccessGrants.set_access_grants('channel', id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('channel', id, form_data.access_grants, db=db)
channel.updated_at = int(time.time_ns())
- db.commit()
- return self._to_channel_model(channel, db=db) if channel else None
+ await db.commit()
+ return await self._to_channel_model(channel, db=db) if channel else None
- def add_file_to_channel_by_id(
- self, channel_id: str, file_id: str, user_id: str, db: Optional[Session] = None
+ async def add_file_to_channel_by_id(
+ self, channel_id: str, file_id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[ChannelFileModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
channel_file = ChannelFileModel(
**{
'id': str(uuid.uuid4()),
@@ -847,8 +857,8 @@ class ChannelTable:
try:
result = ChannelFile(**channel_file.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return ChannelFileModel.model_validate(result)
else:
@@ -856,55 +866,58 @@ class ChannelTable:
except Exception:
return None
- def set_file_message_id_in_channel_by_id(
+ async def set_file_message_id_in_channel_by_id(
self,
channel_id: str,
file_id: str,
message_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
try:
- with get_db_context(db) as db:
- channel_file = db.query(ChannelFile).filter_by(channel_id=channel_id, file_id=file_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelFile).filter_by(channel_id=channel_id, file_id=file_id))
+ channel_file = result.scalars().first()
if not channel_file:
return False
channel_file.message_id = message_id
channel_file.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return True
except Exception:
return False
- def remove_file_from_channel_by_id(self, channel_id: str, file_id: str, db: Optional[Session] = None) -> bool:
+ async def remove_file_from_channel_by_id(
+ self, channel_id: str, file_id: str, db: Optional[AsyncSession] = None
+ ) -> bool:
try:
- with get_db_context(db) as db:
- db.query(ChannelFile).filter_by(channel_id=channel_id, file_id=file_id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(ChannelFile).filter_by(channel_id=channel_id, file_id=file_id))
+ await db.commit()
return True
except Exception:
return False
- def delete_channel_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- AccessGrants.revoke_all_access('channel', id, db=db)
- db.query(Channel).filter(Channel.id == id).delete()
- db.commit()
+ async def delete_channel_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ await AccessGrants.revoke_all_access('channel', id, db=db)
+ await db.execute(delete(Channel).filter(Channel.id == id))
+ await db.commit()
return True
####################
# Webhook Methods
####################
- def insert_webhook(
+ async def insert_webhook(
self,
channel_id: str,
user_id: str,
form_data: ChannelWebhookForm,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[ChannelWebhookModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
webhook = ChannelWebhookModel(
id=str(uuid.uuid4()),
channel_id=channel_id,
@@ -917,63 +930,70 @@ class ChannelTable:
updated_at=int(time.time_ns()),
)
db.add(ChannelWebhook(**webhook.model_dump()))
- db.commit()
+ await db.commit()
return webhook
- def get_webhooks_by_channel_id(self, channel_id: str, db: Optional[Session] = None) -> list[ChannelWebhookModel]:
- with get_db_context(db) as db:
- webhooks = db.query(ChannelWebhook).filter(ChannelWebhook.channel_id == channel_id).all()
+ async def get_webhooks_by_channel_id(
+ self, channel_id: str, db: Optional[AsyncSession] = None
+ ) -> list[ChannelWebhookModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelWebhook).filter(ChannelWebhook.channel_id == channel_id))
+ webhooks = result.scalars().all()
return [ChannelWebhookModel.model_validate(w) for w in webhooks]
- def get_webhook_by_id(self, webhook_id: str, db: Optional[Session] = None) -> Optional[ChannelWebhookModel]:
- with get_db_context(db) as db:
- webhook = db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).first()
+ async def get_webhook_by_id(
+ self, webhook_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[ChannelWebhookModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelWebhook).filter(ChannelWebhook.id == webhook_id))
+ webhook = result.scalars().first()
return ChannelWebhookModel.model_validate(webhook) if webhook else None
- def get_webhook_by_id_and_token(
- self, webhook_id: str, token: str, db: Optional[Session] = None
+ async def get_webhook_by_id_and_token(
+ self, webhook_id: str, token: str, db: Optional[AsyncSession] = None
) -> Optional[ChannelWebhookModel]:
- with get_db_context(db) as db:
- webhook = (
- db.query(ChannelWebhook)
- .filter(
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChannelWebhook).filter(
ChannelWebhook.id == webhook_id,
ChannelWebhook.token == token,
)
- .first()
)
+ webhook = result.scalars().first()
return ChannelWebhookModel.model_validate(webhook) if webhook else None
- def update_webhook_by_id(
+ async def update_webhook_by_id(
self,
webhook_id: str,
form_data: ChannelWebhookForm,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[ChannelWebhookModel]:
- with get_db_context(db) as db:
- webhook = db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelWebhook).filter(ChannelWebhook.id == webhook_id))
+ webhook = result.scalars().first()
if not webhook:
return None
webhook.name = form_data.name
webhook.profile_image_url = form_data.profile_image_url
webhook.updated_at = int(time.time_ns())
- db.commit()
+ await db.commit()
return ChannelWebhookModel.model_validate(webhook)
- def update_webhook_last_used_at(self, webhook_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- webhook = db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).first()
+ async def update_webhook_last_used_at(self, webhook_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ChannelWebhook).filter(ChannelWebhook.id == webhook_id))
+ webhook = result.scalars().first()
if not webhook:
return False
webhook.last_used_at = int(time.time_ns())
- db.commit()
+ await db.commit()
return True
- def delete_webhook_by_id(self, webhook_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- result = db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).delete()
- db.commit()
- return result > 0
+ async def delete_webhook_by_id(self, webhook_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(delete(ChannelWebhook).filter(ChannelWebhook.id == webhook_id))
+ await db.commit()
+ return result.rowcount > 0
Channels = ChannelTable()
diff --git a/backend/open_webui/models/chat_messages.py b/backend/open_webui/models/chat_messages.py
index b37c04037e..bd9c720fa4 100644
--- a/backend/open_webui/models/chat_messages.py
+++ b/backend/open_webui/models/chat_messages.py
@@ -3,8 +3,9 @@ import time
import uuid
from typing import Any, Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db_context
+from sqlalchemy import select, delete, func, cast, Integer
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
from open_webui.utils.response import normalize_usage
from pydantic import BaseModel, ConfigDict
@@ -16,7 +17,6 @@ from sqlalchemy import (
Text,
JSON,
Index,
- func,
)
####################
@@ -129,23 +129,23 @@ class ChatMessageModel(BaseModel):
class ChatMessageTable:
- def upsert_message(
+ async def upsert_message(
self,
message_id: str,
chat_id: str,
user_id: str,
data: dict,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[ChatMessageModel]:
"""Insert or update a chat message."""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
now = int(time.time())
timestamp = data.get('timestamp', now)
# Use composite ID: {chat_id}-{message_id}
composite_id = f'{chat_id}-{message_id}'
- existing = db.get(ChatMessage, composite_id)
+ existing = await db.get(ChatMessage, composite_id)
if existing:
# Update existing
if 'role' in data:
@@ -178,8 +178,8 @@ class ChatMessageTable:
# from accidentally clearing the primary response's token counts
existing.usage = {**(existing.usage or {}), **usage}
existing.updated_at = now
- db.commit()
- db.refresh(existing)
+ await db.commit()
+ await db.refresh(existing)
return ChatMessageModel.model_validate(existing)
else:
# Insert new
@@ -205,143 +205,149 @@ class ChatMessageTable:
updated_at=now,
)
db.add(message)
- db.commit()
- db.refresh(message)
+ await db.commit()
+ await db.refresh(message)
return ChatMessageModel.model_validate(message)
- def get_message_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ChatMessageModel]:
- with get_db_context(db) as db:
- message = db.get(ChatMessage, id)
+ async def get_message_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ChatMessageModel]:
+ async with get_async_db_context(db) as db:
+ message = await db.get(ChatMessage, id)
return ChatMessageModel.model_validate(message) if message else None
- def get_messages_by_chat_id(self, chat_id: str, db: Optional[Session] = None) -> list[ChatMessageModel]:
- with get_db_context(db) as db:
- messages = db.query(ChatMessage).filter_by(chat_id=chat_id).order_by(ChatMessage.created_at.asc()).all()
+ async def get_messages_by_chat_id(self, chat_id: str, db: Optional[AsyncSession] = None) -> list[ChatMessageModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChatMessage).filter_by(chat_id=chat_id).order_by(ChatMessage.created_at.asc())
+ )
+ messages = result.scalars().all()
return [ChatMessageModel.model_validate(message) for message in messages]
- def get_messages_by_user_id(
+ async def get_messages_by_user_id(
self,
user_id: str,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatMessageModel]:
- with get_db_context(db) as db:
- messages = (
- db.query(ChatMessage)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChatMessage)
.filter_by(user_id=user_id)
.order_by(ChatMessage.created_at.desc())
.offset(skip)
.limit(limit)
- .all()
)
+ messages = result.scalars().all()
return [ChatMessageModel.model_validate(message) for message in messages]
- def get_messages_by_model_id(
+ async def get_messages_by_model_id(
self,
model_id: str,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
skip: int = 0,
limit: int = 100,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatMessageModel]:
- with get_db_context(db) as db:
- query = db.query(ChatMessage).filter_by(model_id=model_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(ChatMessage).filter_by(model_id=model_id)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
- messages = query.order_by(ChatMessage.created_at.desc()).offset(skip).limit(limit).all()
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.order_by(ChatMessage.created_at.desc()).offset(skip).limit(limit)
+ result = await db.execute(stmt)
+ messages = result.scalars().all()
return [ChatMessageModel.model_validate(message) for message in messages]
- def get_chat_ids_by_model_id(
+ async def get_chat_ids_by_model_id(
self,
model_id: str,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[str]:
"""Get distinct chat_ids that used a specific model."""
- with get_db_context(db) as db:
- query = db.query(
+ async with get_async_db_context(db) as db:
+ stmt = select(
ChatMessage.chat_id,
func.max(ChatMessage.created_at).label('last_message_at'),
).filter(ChatMessage.model_id == model_id)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
# Group by chat_id and order by most recent message in each chat
# Secondary sort on chat_id ensures deterministic pagination
- # (prevents duplicates across pages when timestamps tie)
- chat_ids = (
- query.group_by(ChatMessage.chat_id)
+ stmt = (
+ stmt.group_by(ChatMessage.chat_id)
.order_by(func.max(ChatMessage.created_at).desc(), ChatMessage.chat_id)
.offset(skip)
.limit(limit)
- .all()
)
+ result = await db.execute(stmt)
+ chat_ids = result.all()
return [chat_id for chat_id, _ in chat_ids]
- def delete_messages_by_chat_id(self, chat_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- db.query(ChatMessage).filter_by(chat_id=chat_id).delete()
- db.commit()
+ async def delete_messages_by_chat_id(self, chat_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(ChatMessage).filter_by(chat_id=chat_id))
+ await db.commit()
return True
# Analytics methods
- def get_message_count_by_model(
+ async def get_message_count_by_model(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
group_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, int]:
- with get_db_context(db) as db:
- from sqlalchemy import func
+ async with get_async_db_context(db) as db:
from open_webui.models.groups import GroupMember
- query = db.query(ChatMessage.model_id, func.count(ChatMessage.id).label('count')).filter(
+ stmt = select(ChatMessage.model_id, func.count(ChatMessage.id).label('count')).filter(
ChatMessage.role == 'assistant',
ChatMessage.model_id.isnot(None),
~ChatMessage.user_id.like('shared-%'),
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
if group_id:
- group_users = db.query(GroupMember.user_id).filter(GroupMember.group_id == group_id).subquery()
- query = query.filter(ChatMessage.user_id.in_(group_users))
+ group_users = select(GroupMember.user_id).filter(GroupMember.group_id == group_id).scalar_subquery()
+ stmt = stmt.filter(ChatMessage.user_id.in_(group_users))
- results = query.group_by(ChatMessage.model_id).all()
- return {row.model_id: row.count for row in results}
+ stmt = stmt.group_by(ChatMessage.model_id)
+ result = await db.execute(stmt)
+ return {row.model_id: row.count for row in result.all()}
- def get_token_usage_by_model(
+ async def get_token_usage_by_model(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
group_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, dict]:
"""Aggregate token usage by model using database-level aggregation."""
- with get_db_context(db) as db:
- from sqlalchemy import func, cast, Integer
+ async with get_async_db_context(db) as db:
from open_webui.models.groups import GroupMember
- dialect = db.bind.dialect.name
+ # We need the dialect to determine JSON extraction syntax
+ # For async sessions, access via get_bind()
+ bind = await db.connection()
+ dialect = bind.dialect.name
if dialect == 'sqlite':
input_tokens = cast(func.json_extract(ChatMessage.usage, '$.input_tokens'), Integer)
output_tokens = cast(func.json_extract(ChatMessage.usage, '$.output_tokens'), Integer)
elif dialect == 'postgresql':
- # Use json_extract_path_text for PostgreSQL JSON columns
input_tokens = cast(
func.json_extract_path_text(ChatMessage.usage, 'input_tokens'),
Integer,
@@ -353,7 +359,7 @@ class ChatMessageTable:
else:
raise NotImplementedError(f'Unsupported dialect: {dialect}')
- query = db.query(
+ stmt = select(
ChatMessage.model_id,
func.coalesce(func.sum(input_tokens), 0).label('input_tokens'),
func.coalesce(func.sum(output_tokens), 0).label('output_tokens'),
@@ -366,14 +372,15 @@ class ChatMessageTable:
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
if group_id:
- group_users = db.query(GroupMember.user_id).filter(GroupMember.group_id == group_id).subquery()
- query = query.filter(ChatMessage.user_id.in_(group_users))
+ group_users = select(GroupMember.user_id).filter(GroupMember.group_id == group_id).scalar_subquery()
+ stmt = stmt.filter(ChatMessage.user_id.in_(group_users))
- results = query.group_by(ChatMessage.model_id).all()
+ stmt = stmt.group_by(ChatMessage.model_id)
+ result = await db.execute(stmt)
return {
row.model_id: {
@@ -382,28 +389,27 @@ class ChatMessageTable:
'total_tokens': row.input_tokens + row.output_tokens,
'message_count': row.message_count,
}
- for row in results
+ for row in result.all()
}
- def get_token_usage_by_user(
+ async def get_token_usage_by_user(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
group_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, dict]:
"""Aggregate token usage by user using database-level aggregation."""
- with get_db_context(db) as db:
- from sqlalchemy import func, cast, Integer
+ async with get_async_db_context(db) as db:
from open_webui.models.groups import GroupMember
- dialect = db.bind.dialect.name
+ bind = await db.connection()
+ dialect = bind.dialect.name
if dialect == 'sqlite':
input_tokens = cast(func.json_extract(ChatMessage.usage, '$.input_tokens'), Integer)
output_tokens = cast(func.json_extract(ChatMessage.usage, '$.output_tokens'), Integer)
elif dialect == 'postgresql':
- # Use json_extract_path_text for PostgreSQL JSON columns
input_tokens = cast(
func.json_extract_path_text(ChatMessage.usage, 'input_tokens'),
Integer,
@@ -415,7 +421,7 @@ class ChatMessageTable:
else:
raise NotImplementedError(f'Unsupported dialect: {dialect}')
- query = db.query(
+ stmt = select(
ChatMessage.user_id,
func.coalesce(func.sum(input_tokens), 0).label('input_tokens'),
func.coalesce(func.sum(output_tokens), 0).label('output_tokens'),
@@ -428,14 +434,15 @@ class ChatMessageTable:
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
if group_id:
- group_users = db.query(GroupMember.user_id).filter(GroupMember.group_id == group_id).subquery()
- query = query.filter(ChatMessage.user_id.in_(group_users))
+ group_users = select(GroupMember.user_id).filter(GroupMember.group_id == group_id).scalar_subquery()
+ stmt = stmt.filter(ChatMessage.user_id.in_(group_users))
- results = query.group_by(ChatMessage.user_id).all()
+ stmt = stmt.group_by(ChatMessage.user_id)
+ result = await db.execute(stmt)
return {
row.user_id: {
@@ -444,88 +451,89 @@ class ChatMessageTable:
'total_tokens': row.input_tokens + row.output_tokens,
'message_count': row.message_count,
}
- for row in results
+ for row in result.all()
}
- def get_message_count_by_user(
+ async def get_message_count_by_user(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
group_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, int]:
- with get_db_context(db) as db:
- from sqlalchemy import func
+ async with get_async_db_context(db) as db:
from open_webui.models.groups import GroupMember
- query = db.query(ChatMessage.user_id, func.count(ChatMessage.id).label('count')).filter(
+ stmt = select(ChatMessage.user_id, func.count(ChatMessage.id).label('count')).filter(
~ChatMessage.user_id.like('shared-%')
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
if group_id:
- group_users = db.query(GroupMember.user_id).filter(GroupMember.group_id == group_id).subquery()
- query = query.filter(ChatMessage.user_id.in_(group_users))
+ group_users = select(GroupMember.user_id).filter(GroupMember.group_id == group_id).scalar_subquery()
+ stmt = stmt.filter(ChatMessage.user_id.in_(group_users))
- results = query.group_by(ChatMessage.user_id).all()
- return {row.user_id: row.count for row in results}
+ stmt = stmt.group_by(ChatMessage.user_id)
+ result = await db.execute(stmt)
+ return {row.user_id: row.count for row in result.all()}
- def get_message_count_by_chat(
+ async def get_message_count_by_chat(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
group_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, int]:
- with get_db_context(db) as db:
- from sqlalchemy import func
+ async with get_async_db_context(db) as db:
from open_webui.models.groups import GroupMember
- query = db.query(ChatMessage.chat_id, func.count(ChatMessage.id).label('count')).filter(
+ stmt = select(ChatMessage.chat_id, func.count(ChatMessage.id).label('count')).filter(
~ChatMessage.user_id.like('shared-%')
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
if group_id:
- group_users = db.query(GroupMember.user_id).filter(GroupMember.group_id == group_id).subquery()
- query = query.filter(ChatMessage.user_id.in_(group_users))
+ group_users = select(GroupMember.user_id).filter(GroupMember.group_id == group_id).scalar_subquery()
+ stmt = stmt.filter(ChatMessage.user_id.in_(group_users))
- results = query.group_by(ChatMessage.chat_id).all()
- return {row.chat_id: row.count for row in results}
+ stmt = stmt.group_by(ChatMessage.chat_id)
+ result = await db.execute(stmt)
+ return {row.chat_id: row.count for row in result.all()}
- def get_daily_message_counts_by_model(
+ async def get_daily_message_counts_by_model(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
group_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, dict[str, int]]:
"""Get message counts grouped by day and model."""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
from datetime import datetime, timedelta
from open_webui.models.groups import GroupMember
- query = db.query(ChatMessage.created_at, ChatMessage.model_id).filter(
+ stmt = select(ChatMessage.created_at, ChatMessage.model_id).filter(
ChatMessage.role == 'assistant',
ChatMessage.model_id.isnot(None),
~ChatMessage.user_id.like('shared-%'),
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
if group_id:
- group_users = db.query(GroupMember.user_id).filter(GroupMember.group_id == group_id).subquery()
- query = query.filter(ChatMessage.user_id.in_(group_users))
+ group_users = select(GroupMember.user_id).filter(GroupMember.group_id == group_id).scalar_subquery()
+ stmt = stmt.filter(ChatMessage.user_id.in_(group_users))
- results = query.all()
+ result = await db.execute(stmt)
+ results = result.all()
# Group by date -> model -> count
daily_counts: dict[str, dict[str, int]] = {}
@@ -547,28 +555,29 @@ class ChatMessageTable:
return daily_counts
- def get_hourly_message_counts_by_model(
+ async def get_hourly_message_counts_by_model(
self,
start_date: Optional[int] = None,
end_date: Optional[int] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict[str, dict[str, int]]:
"""Get message counts grouped by hour and model."""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
from datetime import datetime, timedelta
- query = db.query(ChatMessage.created_at, ChatMessage.model_id).filter(
+ stmt = select(ChatMessage.created_at, ChatMessage.model_id).filter(
ChatMessage.role == 'assistant',
ChatMessage.model_id.isnot(None),
~ChatMessage.user_id.like('shared-%'),
)
if start_date:
- query = query.filter(ChatMessage.created_at >= start_date)
+ stmt = stmt.filter(ChatMessage.created_at >= start_date)
if end_date:
- query = query.filter(ChatMessage.created_at <= end_date)
+ stmt = stmt.filter(ChatMessage.created_at <= end_date)
- results = query.all()
+ result = await db.execute(stmt)
+ results = result.all()
# Group by hour -> model -> count
hourly_counts: dict[str, dict[str, int]] = {}
diff --git a/backend/open_webui/models/chats.py b/backend/open_webui/models/chats.py
index 5f53e741e4..3bcfdce03f 100644
--- a/backend/open_webui/models/chats.py
+++ b/backend/open_webui/models/chats.py
@@ -4,8 +4,11 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update, func, or_, and_, text
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.sql import exists
+from sqlalchemy.sql.expression import bindparam
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.tags import TagModel, Tag, Tags
from open_webui.models.folders import Folders
from open_webui.models.chat_messages import ChatMessage, ChatMessages
@@ -24,9 +27,6 @@ from sqlalchemy import (
Index,
UniqueConstraint,
)
-from sqlalchemy import or_, func, select, and_, text
-from sqlalchemy.sql import exists
-from sqlalchemy.sql.expression import bindparam
####################
# Chat DB Schema
@@ -62,15 +62,10 @@ class Chat(Base):
__table_args__ = (
# Performance indexes for common queries
- # WHERE folder_id = ...
Index('folder_id_idx', 'folder_id'),
- # WHERE user_id = ... AND pinned = ...
Index('user_id_pinned_idx', 'user_id', 'pinned'),
- # WHERE user_id = ... AND archived = ...
Index('user_id_archived_idx', 'user_id', 'archived'),
- # WHERE user_id = ... ORDER BY updated_at DESC
Index('updated_at_user_id_idx', 'updated_at', 'user_id'),
- # WHERE folder_id = ... AND user_id = ...
Index('folder_id_user_id_idx', 'folder_id', 'user_id'),
)
@@ -297,9 +292,10 @@ class ChatTable:
return changed
- def insert_new_chat(self, user_id: str, form_data: ChatForm, db: Optional[Session] = None) -> Optional[ChatModel]:
- with get_db_context(db) as db:
- id = str(uuid.uuid4())
+ async def insert_new_chat(
+ self, id: str, user_id: str, form_data: ChatForm, db: Optional[AsyncSession] = None
+ ) -> Optional[ChatModel]:
+ async with get_async_db_context(db) as db:
chat = ChatModel(
**{
'id': id,
@@ -316,8 +312,8 @@ class ChatTable:
chat_item = Chat(**chat.model_dump())
db.add(chat_item)
- db.commit()
- db.refresh(chat_item)
+ await db.commit()
+ await db.refresh(chat_item)
# Dual-write initial messages to chat_message table
try:
@@ -325,7 +321,7 @@ class ChatTable:
messages = history.get('messages', {})
for message_id, message in messages.items():
if isinstance(message, dict) and message.get('role'):
- ChatMessages.upsert_message(
+ await ChatMessages.upsert_message(
message_id=message_id,
chat_id=id,
user_id=user_id,
@@ -353,13 +349,13 @@ class ChatTable:
)
return chat
- def import_chats(
+ async def import_chats(
self,
user_id: str,
chat_import_forms: list[ChatImportForm],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
chats = []
for form_data in chat_import_forms:
@@ -367,7 +363,7 @@ class ChatTable:
chats.append(Chat(**chat.model_dump()))
db.add_all(chats)
- db.commit()
+ await db.commit()
# Dual-write messages to chat_message table
try:
@@ -376,7 +372,7 @@ class ChatTable:
messages = history.get('messages', {})
for message_id, message in messages.items():
if isinstance(message, dict) and message.get('role'):
- ChatMessages.upsert_message(
+ await ChatMessages.upsert_message(
message_id=message_id,
chat_id=chat_obj.id,
user_id=user_id,
@@ -387,53 +383,53 @@ class ChatTable:
return [ChatModel.model_validate(chat) for chat in chats]
- def update_chat_by_id(self, id: str, chat: dict, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def update_chat_by_id(self, id: str, chat: dict, db: Optional[AsyncSession] = None) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat_item = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat_item = await db.get(Chat, id)
chat_item.chat = self._clean_null_bytes(chat)
chat_item.title = self._clean_null_bytes(chat['title']) if 'title' in chat else 'New Chat'
chat_item.updated_at = int(time.time())
- db.commit()
- db.refresh(chat_item)
+ await db.commit()
+ await db.refresh(chat_item)
return ChatModel.model_validate(chat_item)
except Exception:
return None
- def update_chat_last_read_at_by_id(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
+ async def update_chat_last_read_at_by_id(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
if chat and chat.user_id == user_id:
chat.last_read_at = int(time.time())
- db.commit()
+ await db.commit()
return True
return False
except Exception:
return False
- def update_chat_title_by_id(self, id: str, title: str) -> Optional[ChatModel]:
+ async def update_chat_title_by_id(self, id: str, title: str) -> Optional[ChatModel]:
try:
- with get_db_context() as db:
- chat_item = db.get(Chat, id)
+ async with get_async_db_context() as db:
+ chat_item = await db.get(Chat, id)
if chat_item is None:
return None
clean_title = self._clean_null_bytes(title)
chat_item.title = clean_title
chat_item.chat = {**(chat_item.chat or {}), 'title': clean_title}
chat_item.updated_at = int(time.time())
- db.commit()
- db.refresh(chat_item)
+ await db.commit()
+ await db.refresh(chat_item)
return ChatModel.model_validate(chat_item)
except Exception:
return None
- def update_chat_tags_by_id(self, id: str, tags: list[str], user) -> Optional[ChatModel]:
- with get_db_context() as db:
- chat = db.get(Chat, id)
+ async def update_chat_tags_by_id(self, id: str, tags: list[str], user) -> Optional[ChatModel]:
+ async with get_async_db_context() as db:
+ chat = await db.get(Chat, id)
if chat is None:
return None
@@ -443,44 +439,45 @@ class ChatTable:
# Single meta update
chat.meta = {**chat.meta, 'tags': new_tag_ids}
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
# Batch-create any missing tag rows
- Tags.ensure_tags_exist(new_tags, user.id, db=db)
+ await Tags.ensure_tags_exist(new_tags, user.id, db=db)
# Clean up orphaned old tags in one query
removed = set(old_tags) - set(new_tag_ids)
if removed:
- self.delete_orphan_tags_for_user(list(removed), user.id, db=db)
+ await self.delete_orphan_tags_for_user(list(removed), user.id, db=db)
return ChatModel.model_validate(chat)
- def get_chat_title_by_id(self, id: str) -> Optional[str]:
- with get_db_context() as db:
- result = db.query(Chat.title).filter_by(id=id).first()
- if result is None:
+ async def get_chat_title_by_id(self, id: str) -> Optional[str]:
+ async with get_async_db_context() as db:
+ result = await db.execute(select(Chat.title).filter_by(id=id))
+ row = result.first()
+ if row is None:
return None
- return result[0] or 'New Chat'
+ return row[0] or 'New Chat'
- def get_messages_map_by_chat_id(self, id: str) -> Optional[dict]:
- chat = self.get_chat_by_id(id)
+ async def get_messages_map_by_chat_id(self, id: str) -> Optional[dict]:
+ chat = await self.get_chat_by_id(id)
if chat is None:
return None
return chat.chat.get('history', {}).get('messages', {}) or {}
- def get_message_by_id_and_message_id(self, id: str, message_id: str) -> Optional[dict]:
- chat = self.get_chat_by_id(id)
+ async def get_message_by_id_and_message_id(self, id: str, message_id: str) -> Optional[dict]:
+ chat = await self.get_chat_by_id(id)
if chat is None:
return None
return chat.chat.get('history', {}).get('messages', {}).get(message_id, {})
- def upsert_message_to_chat_by_id_and_message_id(
+ async def upsert_message_to_chat_by_id_and_message_id(
self, id: str, message_id: str, message: dict
) -> Optional[ChatModel]:
- chat = self.get_chat_by_id(id)
+ chat = await self.get_chat_by_id(id)
if chat is None:
return None
@@ -506,7 +503,7 @@ class ChatTable:
# Dual-write to chat_message table
try:
- ChatMessages.upsert_message(
+ await ChatMessages.upsert_message(
message_id=message_id,
chat_id=id,
user_id=user_id,
@@ -515,12 +512,12 @@ class ChatTable:
except Exception as e:
log.warning(f'Failed to write to chat_message table: {e}')
- return self.update_chat_by_id(id, chat)
+ return await self.update_chat_by_id(id, chat)
- def add_message_status_to_chat_by_id_and_message_id(
+ async def add_message_status_to_chat_by_id_and_message_id(
self, id: str, message_id: str, status: dict
) -> Optional[ChatModel]:
- chat = self.get_chat_by_id(id)
+ chat = await self.get_chat_by_id(id)
if chat is None:
return None
@@ -533,11 +530,11 @@ class ChatTable:
history['messages'][message_id]['statusHistory'] = status_history
chat['history'] = history
- return self.update_chat_by_id(id, chat)
+ return await self.update_chat_by_id(id, chat)
- def add_message_files_by_id_and_message_id(self, id: str, message_id: str, files: list[dict]) -> list[dict]:
- with get_db_context() as db:
- chat = self.get_chat_by_id(id, db=db)
+ async def add_message_files_by_id_and_message_id(self, id: str, message_id: str, files: list[dict]) -> list[dict]:
+ async with get_async_db_context() as db:
+ chat = await self.get_chat_by_id(id, db=db)
if chat is None:
return None
@@ -552,19 +549,21 @@ class ChatTable:
history['messages'][message_id]['files'] = message_files
chat['history'] = history
- self.update_chat_by_id(id, chat, db=db)
+ await self.update_chat_by_id(id, chat, db=db)
return message_files
- def insert_shared_chat_by_chat_id(self, chat_id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
- with get_db_context(db) as db:
+ async def insert_shared_chat_by_chat_id(
+ self, chat_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[ChatModel]:
+ async with get_async_db_context(db) as db:
# Get the existing chat to share
- chat = db.get(Chat, chat_id)
+ chat = await db.get(Chat, chat_id)
# Check if chat exists
if not chat:
return None
# Check if the chat is already shared
if chat.share_id:
- return self.get_chat_by_id_and_user_id(chat.share_id, 'shared', db=db)
+ return await self.get_chat_by_id_and_user_id(chat.share_id, 'shared', db=db)
# Create a new chat with the same data, but with a new ID
shared_chat = ChatModel(
**{
@@ -581,22 +580,25 @@ class ChatTable:
)
shared_result = Chat(**shared_chat.model_dump())
db.add(shared_result)
- db.commit()
- db.refresh(shared_result)
+ await db.commit()
+ await db.refresh(shared_result)
# Update the original chat with the share_id
- result = db.query(Chat).filter_by(id=chat_id).update({'share_id': shared_chat.id})
- db.commit()
- return shared_chat if (shared_result and result) else None
+ await db.execute(update(Chat).filter_by(id=chat_id).values(share_id=shared_chat.id))
+ await db.commit()
+ return shared_chat if shared_result else None
- def update_shared_chat_by_chat_id(self, chat_id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def update_shared_chat_by_chat_id(
+ self, chat_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, chat_id)
- shared_chat = db.query(Chat).filter_by(user_id=f'shared-{chat_id}').first()
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, chat_id)
+ result = await db.execute(select(Chat).filter_by(user_id=f'shared-{chat_id}'))
+ shared_chat = result.scalars().first()
if shared_chat is None:
- return self.insert_shared_chat_by_chat_id(chat_id, db=db)
+ return await self.insert_shared_chat_by_chat_id(chat_id, db=db)
shared_chat.title = chat.title
shared_chat.chat = chat.chat
@@ -604,99 +606,102 @@ class ChatTable:
shared_chat.pinned = chat.pinned
shared_chat.folder_id = chat.folder_id
shared_chat.updated_at = int(time.time())
- db.commit()
- db.refresh(shared_chat)
+ await db.commit()
+ await db.refresh(shared_chat)
return ChatModel.model_validate(shared_chat)
except Exception:
return None
- def delete_shared_chat_by_chat_id(self, chat_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_shared_chat_by_chat_id(self, chat_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- # Use subquery to delete chat_messages for shared chats
- shared_chat_id_subquery = db.query(Chat.id).filter_by(user_id=f'shared-{chat_id}').scalar_subquery()
- db.query(ChatMessage).filter(ChatMessage.chat_id.in_(shared_chat_id_subquery)).delete(
- synchronize_session=False
- )
- db.query(Chat).filter_by(user_id=f'shared-{chat_id}').delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ # Get shared chat IDs
+ result = await db.execute(select(Chat.id).filter_by(user_id=f'shared-{chat_id}'))
+ shared_ids = [row[0] for row in result.all()]
+
+ if shared_ids:
+ await db.execute(delete(ChatMessage).filter(ChatMessage.chat_id.in_(shared_ids)))
+ await db.execute(delete(Chat).filter_by(user_id=f'shared-{chat_id}'))
+ await db.commit()
return True
except Exception:
return False
- def unarchive_all_chats_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
+ async def unarchive_all_chats_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(Chat).filter_by(user_id=user_id).update({'archived': False})
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(update(Chat).filter_by(user_id=user_id).values(archived=False))
+ await db.commit()
return True
except Exception:
return False
- def update_chat_share_id_by_id(
- self, id: str, share_id: Optional[str], db: Optional[Session] = None
+ async def update_chat_share_id_by_id(
+ self, id: str, share_id: Optional[str], db: Optional[AsyncSession] = None
) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
chat.share_id = share_id
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
return ChatModel.model_validate(chat)
except Exception:
return None
- def toggle_chat_pinned_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def toggle_chat_pinned_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
chat.pinned = not chat.pinned
chat.updated_at = int(time.time())
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
return ChatModel.model_validate(chat)
except Exception:
return None
- def toggle_chat_archive_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def toggle_chat_archive_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
chat.archived = not chat.archived
chat.folder_id = None
chat.updated_at = int(time.time())
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
return ChatModel.model_validate(chat)
except Exception:
return None
- def archive_all_chats_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
+ async def archive_all_chats_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(Chat).filter_by(user_id=user_id).update({'archived': True})
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(update(Chat).filter_by(user_id=user_id).values(archived=True))
+ await db.commit()
return True
except Exception:
return False
- def get_archived_chat_list_by_user_id(
+ async def get_archived_chat_list_by_user_id(
self,
user_id: str,
filter: Optional[dict] = None,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatTitleIdResponse]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id, archived=True)
+ async with get_async_db_context(db) as db:
+ stmt = select(Chat.id, Chat.title, Chat.updated_at, Chat.created_at).filter_by(
+ user_id=user_id, archived=True
+ )
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(Chat.title.ilike(f'%{query_key}%'))
+ stmt = stmt.filter(Chat.title.ilike(f'%{query_key}%'))
order_by = filter.get('order_by')
direction = filter.get('direction')
@@ -706,22 +711,21 @@ class ChatTable:
raise ValueError('Invalid order_by field')
if direction.lower() == 'asc':
- query = query.order_by(getattr(Chat, order_by).asc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).asc(), Chat.id)
elif direction.lower() == 'desc':
- query = query.order_by(getattr(Chat, order_by).desc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).desc(), Chat.id)
else:
raise ValueError('Invalid direction for ordering')
else:
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
-
- query = query.with_entities(Chat.id, Chat.title, Chat.updated_at, Chat.created_at)
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.all()
return [
ChatTitleIdResponse.model_validate(
{
@@ -734,21 +738,25 @@ class ChatTable:
for chat in all_chats
]
- def get_shared_chat_list_by_user_id(
+ async def get_shared_chat_list_by_user_id(
self,
user_id: str,
filter: Optional[dict] = None,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[SharedChatResponse]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id).filter(Chat.share_id.isnot(None))
+ async with get_async_db_context(db) as db:
+ stmt = (
+ select(Chat.id, Chat.title, Chat.share_id, Chat.updated_at, Chat.created_at)
+ .filter_by(user_id=user_id)
+ .filter(Chat.share_id.isnot(None))
+ )
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(Chat.title.ilike(f'%{query_key}%'))
+ stmt = stmt.filter(Chat.title.ilike(f'%{query_key}%'))
order_by = filter.get('order_by')
direction = filter.get('direction')
@@ -758,30 +766,21 @@ class ChatTable:
raise ValueError('Invalid order_by field')
if direction.lower() == 'asc':
- query = query.order_by(getattr(Chat, order_by).asc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).asc(), Chat.id)
elif direction.lower() == 'desc':
- query = query.order_by(getattr(Chat, order_by).desc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).desc(), Chat.id)
else:
raise ValueError('Invalid direction for ordering')
else:
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
-
- # Select only the columns needed for SharedChatResponse
- # to avoid loading the heavy chat JSON blob
- query = query.with_entities(
- Chat.id,
- Chat.title,
- Chat.share_id,
- Chat.updated_at,
- Chat.created_at,
- )
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.all()
return [
SharedChatResponse.model_validate(
{
@@ -795,46 +794,47 @@ class ChatTable:
for chat in all_chats
]
- def get_chat_list_by_user_id(
+ async def get_chat_list_by_user_id(
self,
user_id: str,
include_archived: bool = False,
filter: Optional[dict] = None,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatTitleIdResponse]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at).filter_by(
+ user_id=user_id
+ )
if not include_archived:
- query = query.filter_by(archived=False)
+ stmt = stmt.filter_by(archived=False)
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(Chat.title.ilike(f'%{query_key}%'))
+ stmt = stmt.filter(Chat.title.ilike(f'%{query_key}%'))
order_by = filter.get('order_by')
direction = filter.get('direction')
if order_by and direction and getattr(Chat, order_by):
if direction.lower() == 'asc':
- query = query.order_by(getattr(Chat, order_by).asc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).asc(), Chat.id)
elif direction.lower() == 'desc':
- query = query.order_by(getattr(Chat, order_by).desc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).desc(), Chat.id)
else:
raise ValueError('Invalid direction for ordering')
else:
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
-
- query = query.with_entities(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at)
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.all()
return [
ChatTitleIdResponse.model_validate(
{
@@ -848,7 +848,7 @@ class ChatTable:
for chat in all_chats
]
- def get_chat_title_id_list_by_user_id(
+ async def get_chat_title_id_list_by_user_id(
self,
user_id: str,
include_archived: bool = False,
@@ -856,32 +856,32 @@ class ChatTable:
include_pinned: bool = False,
skip: Optional[int] = None,
limit: Optional[int] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatTitleIdResponse]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id)
-
- if not include_folders:
- query = query.filter_by(folder_id=None)
-
- if not include_pinned:
- query = query.filter(or_(Chat.pinned == False, Chat.pinned == None))
-
- if not include_archived:
- query = query.filter_by(archived=False)
-
- query = query.order_by(Chat.updated_at.desc(), Chat.id).with_entities(
- Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at
+ async with get_async_db_context(db) as db:
+ stmt = select(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at).filter_by(
+ user_id=user_id
)
+ if not include_folders:
+ stmt = stmt.filter_by(folder_id=None)
+
+ if not include_pinned:
+ stmt = stmt.filter(or_(Chat.pinned == False, Chat.pinned == None))
+
+ if not include_archived:
+ stmt = stmt.filter_by(archived=False)
+
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
+
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.all()
- # result has to be destructured from sqlalchemy `row` and mapped to a dict since the `ChatModel`is not the returned dataclass.
return [
ChatTitleIdResponse.model_validate(
{
@@ -895,106 +895,104 @@ class ChatTable:
for chat in all_chats
]
- def get_chat_list_by_chat_ids(
+ async def get_chat_list_by_chat_ids(
self,
chat_ids: list[str],
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatModel]:
- with get_db_context(db) as db:
- all_chats = (
- db.query(Chat)
- .filter(Chat.id.in_(chat_ids))
- .filter_by(archived=False)
- .order_by(Chat.updated_at.desc())
- .all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Chat).filter(Chat.id.in_(chat_ids)).filter_by(archived=False).order_by(Chat.updated_at.desc())
)
+ all_chats = result.scalars().all()
return [ChatModel.model_validate(chat) for chat in all_chats]
- def get_chat_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def get_chat_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat_item = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat_item = await db.get(Chat, id)
if chat_item is None:
return None
if self._sanitize_chat_row(chat_item):
- db.commit()
- db.refresh(chat_item)
+ await db.commit()
+ await db.refresh(chat_item)
return ChatModel.model_validate(chat_item)
except Exception:
return None
- def get_chat_by_share_id(self, id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def get_chat_by_share_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- # it is possible that the shared link was deleted. hence,
- # we check if the chat is still shared by checking if a chat with the share_id exists
- chat = db.query(Chat).filter_by(share_id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Chat).filter_by(share_id=id))
+ chat = result.scalars().first()
if chat:
- return self.get_chat_by_id(id, db=db)
+ return await self.get_chat_by_id(id, db=db)
else:
return None
except Exception:
return None
- def get_chat_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> Optional[ChatModel]:
+ async def get_chat_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat = db.query(Chat).filter_by(id=id, user_id=user_id).first()
- return ChatModel.model_validate(chat)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Chat).filter_by(id=id, user_id=user_id))
+ chat = result.scalars().first()
+ return ChatModel.model_validate(chat) if chat else None
except Exception:
return None
- def is_chat_owner(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
+ async def is_chat_owner(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
"""
Lightweight ownership check — uses EXISTS subquery instead of loading
the full Chat row (which includes the potentially large JSON blob).
"""
try:
- with get_db_context(db) as db:
- return db.query(exists().where(and_(Chat.id == id, Chat.user_id == user_id))).scalar()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(exists().where(and_(Chat.id == id, Chat.user_id == user_id))))
+ return result.scalar()
except Exception:
return False
- def get_chat_folder_id(self, id: str, user_id: str, db: Optional[Session] = None) -> Optional[str]:
+ async def get_chat_folder_id(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> Optional[str]:
"""
Fetch only the folder_id column for a chat, without loading the full
JSON blob. Returns None if chat doesn't exist or doesn't belong to user.
"""
try:
- with get_db_context(db) as db:
- result = db.query(Chat.folder_id).filter_by(id=id, user_id=user_id).first()
- return result[0] if result else None
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Chat.folder_id).filter_by(id=id, user_id=user_id))
+ row = result.first()
+ return row[0] if row else None
except Exception:
return None
- def get_chats(self, skip: int = 0, limit: int = 50, db: Optional[Session] = None) -> list[ChatModel]:
- with get_db_context(db) as db:
- all_chats = (
- db.query(Chat)
- # .limit(limit).offset(skip)
- .order_by(Chat.updated_at.desc())
- )
+ async def get_chats(self, skip: int = 0, limit: int = 50, db: Optional[AsyncSession] = None) -> list[ChatModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Chat).order_by(Chat.updated_at.desc()))
+ all_chats = result.scalars().all()
return [ChatModel.model_validate(chat) for chat in all_chats]
- def get_chats_by_user_id(
+ async def get_chats_by_user_id(
self,
user_id: str,
filter: Optional[dict] = None,
skip: Optional[int] = None,
limit: Optional[int] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> ChatListResponse:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Chat).filter_by(user_id=user_id)
if filter:
if filter.get('updated_at'):
- query = query.filter(Chat.updated_at > filter.get('updated_at'))
+ stmt = stmt.filter(Chat.updated_at > filter.get('updated_at'))
order_by = filter.get('order_by')
direction = filter.get('direction')
@@ -1002,23 +1000,25 @@ class ChatTable:
if order_by and direction:
if hasattr(Chat, order_by):
if direction.lower() == 'asc':
- query = query.order_by(getattr(Chat, order_by).asc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).asc(), Chat.id)
elif direction.lower() == 'desc':
- query = query.order_by(getattr(Chat, order_by).desc(), Chat.id)
+ stmt = stmt.order_by(getattr(Chat, order_by).desc(), Chat.id)
else:
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
else:
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip is not None:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit is not None:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.scalars().all()
return ChatListResponse(
**{
@@ -1027,14 +1027,16 @@ class ChatTable:
}
)
- def get_pinned_chats_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[ChatTitleIdResponse]:
- with get_db_context(db) as db:
- all_chats = (
- db.query(Chat)
+ async def get_pinned_chats_by_user_id(
+ self, user_id: str, db: Optional[AsyncSession] = None
+ ) -> list[ChatTitleIdResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at)
.filter_by(user_id=user_id, pinned=True, archived=False)
.order_by(Chat.updated_at.desc())
- .with_entities(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at)
)
+ all_chats = result.all()
return [
ChatTitleIdResponse.model_validate(
{
@@ -1048,19 +1050,21 @@ class ChatTable:
for chat in all_chats
]
- def get_archived_chats_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[ChatModel]:
- with get_db_context(db) as db:
- all_chats = db.query(Chat).filter_by(user_id=user_id, archived=True).order_by(Chat.updated_at.desc())
- return [ChatModel.model_validate(chat) for chat in all_chats]
+ async def get_archived_chats_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[ChatModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Chat).filter_by(user_id=user_id, archived=True).order_by(Chat.updated_at.desc())
+ )
+ return [ChatModel.model_validate(chat) for chat in result.scalars().all()]
- def get_chats_by_user_id_and_search_text(
+ async def get_chats_by_user_id_and_search_text(
self,
user_id: str,
search_text: str,
include_archived: bool = False,
skip: int = 0,
limit: int = 60,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatModel]:
"""
Filters chats based on a search query using Python, allowing pagination using skip and limit.
@@ -1068,17 +1072,19 @@ class ChatTable:
search_text = sanitize_text_for_db(search_text).lower().strip()
if not search_text:
- return self.get_chat_list_by_user_id(user_id, include_archived, filter={}, skip=skip, limit=limit, db=db)
+ return await self.get_chat_list_by_user_id(
+ user_id, include_archived, filter={}, skip=skip, limit=limit, db=db
+ )
search_text_words = search_text.split(' ')
- # search_text might contain 'tag:tag_name' format so we need to extract the tag_name, split the search_text and remove the tags
+ # search_text might contain 'tag:tag_name' format so we need to extract the tag_name
tag_ids = [
word.replace('tag:', '').replace(' ', '_').lower() for word in search_text_words if word.startswith('tag:')
]
- # Extract folder names - handle spaces and case insensitivity
- folders = Folders.search_folders_by_names(
+ # Extract folder names
+ folders = await Folders.search_folders_by_names(
user_id,
[word.replace('folder:', '') for word in search_text_words if word.startswith('folder:')],
)
@@ -1116,30 +1122,31 @@ class ChatTable:
search_text = ' '.join(search_text_words)
- with get_db_context(db) as db:
- query = db.query(Chat).filter(Chat.user_id == user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Chat).filter(Chat.user_id == user_id)
if is_archived is not None:
- query = query.filter(Chat.archived == is_archived)
+ stmt = stmt.filter(Chat.archived == is_archived)
elif not include_archived:
- query = query.filter(Chat.archived == False)
+ stmt = stmt.filter(Chat.archived == False)
if is_pinned is not None:
- query = query.filter(Chat.pinned == is_pinned)
+ stmt = stmt.filter(Chat.pinned == is_pinned)
if is_shared is not None:
if is_shared:
- query = query.filter(Chat.share_id.isnot(None))
+ stmt = stmt.filter(Chat.share_id.isnot(None))
else:
- query = query.filter(Chat.share_id.is_(None))
+ stmt = stmt.filter(Chat.share_id.is_(None))
if folder_ids:
- query = query.filter(Chat.folder_id.in_(folder_ids))
+ stmt = stmt.filter(Chat.folder_id.in_(folder_ids))
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
# Check if the database dialect is either 'sqlite' or 'postgresql'
- dialect_name = db.bind.dialect.name
+ bind = await db.connection()
+ dialect_name = bind.dialect.name
if dialect_name == 'sqlite':
# SQLite case: using JSON1 extension for JSON searching
sqlite_content_sql = (
@@ -1150,15 +1157,15 @@ class ChatTable:
')'
)
sqlite_content_clause = text(sqlite_content_sql)
- query = query.filter(
+ stmt = stmt.filter(
or_(Chat.title.ilike(bindparam('title_key')), sqlite_content_clause).params(
title_key=f'%{search_text}%', content_key=search_text
)
)
- # Check if there are any tags to filter, it should have all the tags
+ # Check if there are any tags to filter
if 'none' in tag_ids:
- query = query.filter(
+ stmt = stmt.filter(
text("""
NOT EXISTS (
SELECT 1
@@ -1167,7 +1174,7 @@ class ChatTable:
""")
)
elif tag_ids:
- query = query.filter(
+ stmt = stmt.filter(
and_(
*[
text(f"""
@@ -1183,14 +1190,11 @@ class ChatTable:
)
elif dialect_name == 'postgresql':
- # PostgreSQL doesn't allow null bytes in text. We filter those out by checking
- # the JSON representation for \u0000 before attempting text extraction
-
# Safety filter: JSON field must not contain \u0000
- query = query.filter(text("Chat.chat::text NOT LIKE '%\\\\u0000%'"))
+ stmt = stmt.filter(text("Chat.chat::text NOT LIKE '%\\\\u0000%'"))
# Safety filter: title must not contain actual null bytes
- query = query.filter(text("Chat.title::text NOT LIKE '%\\x00%'"))
+ stmt = stmt.filter(text("Chat.title::text NOT LIKE '%\\x00%'"))
postgres_content_sql = """
EXISTS (
@@ -1203,16 +1207,15 @@ class ChatTable:
postgres_content_clause = text(postgres_content_sql)
- query = query.filter(
+ stmt = stmt.filter(
or_(
Chat.title.ilike(bindparam('title_key')),
postgres_content_clause,
)
).params(title_key=f'%{search_text}%', content_key=search_text.lower())
- # Check if there are any tags to filter, it should have all the tags
if 'none' in tag_ids:
- query = query.filter(
+ stmt = stmt.filter(
text("""
NOT EXISTS (
SELECT 1
@@ -1221,7 +1224,7 @@ class ChatTable:
""")
)
elif tag_ids:
- query = query.filter(
+ stmt = stmt.filter(
and_(
*[
text(f"""
@@ -1236,39 +1239,42 @@ class ChatTable:
)
)
else:
- raise NotImplementedError(f'Unsupported dialect: {db.bind.dialect.name}')
+ raise NotImplementedError(f'Unsupported dialect: {dialect_name}')
# Perform pagination at the SQL level
- all_chats = query.offset(skip).limit(limit).all()
+ stmt = stmt.offset(skip).limit(limit)
+ result = await db.execute(stmt)
+ all_chats = result.scalars().all()
log.info(f'The number of chats: {len(all_chats)}')
# Validate and return chats
return [ChatModel.model_validate(chat) for chat in all_chats]
- def get_chats_by_folder_id_and_user_id(
+ async def get_chats_by_folder_id_and_user_id(
self,
folder_id: str,
user_id: str,
skip: int = 0,
limit: int = 60,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatTitleIdResponse]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(folder_id=folder_id, user_id=user_id)
- query = query.filter(or_(Chat.pinned == False, Chat.pinned == None))
- query = query.filter_by(archived=False)
-
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
-
- query = query.with_entities(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at)
+ async with get_async_db_context(db) as db:
+ stmt = (
+ select(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at)
+ .filter_by(folder_id=folder_id, user_id=user_id)
+ .filter(or_(Chat.pinned == False, Chat.pinned == None))
+ .filter_by(archived=False)
+ .order_by(Chat.updated_at.desc(), Chat.id)
+ )
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.all()
return [
ChatTitleIdResponse.model_validate(
{
@@ -1282,76 +1288,82 @@ class ChatTable:
for chat in all_chats
]
- def get_chats_by_folder_ids_and_user_id(
- self, folder_ids: list[str], user_id: str, db: Optional[Session] = None
+ async def get_chats_by_folder_ids_and_user_id(
+ self, folder_ids: list[str], user_id: str, db: Optional[AsyncSession] = None
) -> list[ChatModel]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter(Chat.folder_id.in_(folder_ids), Chat.user_id == user_id)
- query = query.filter(or_(Chat.pinned == False, Chat.pinned == None))
- query = query.filter_by(archived=False)
+ async with get_async_db_context(db) as db:
+ stmt = (
+ select(Chat)
+ .filter(Chat.folder_id.in_(folder_ids), Chat.user_id == user_id)
+ .filter(or_(Chat.pinned == False, Chat.pinned == None))
+ .filter_by(archived=False)
+ .order_by(Chat.updated_at.desc())
+ )
- query = query.order_by(Chat.updated_at.desc())
-
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.scalars().all()
return [ChatModel.model_validate(chat) for chat in all_chats]
- def update_chat_folder_id_by_id_and_user_id(
- self, id: str, user_id: str, folder_id: str, db: Optional[Session] = None
+ async def update_chat_folder_id_by_id_and_user_id(
+ self, id: str, user_id: str, folder_id: str, db: Optional[AsyncSession] = None
) -> Optional[ChatModel]:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
chat.folder_id = folder_id
chat.updated_at = int(time.time())
chat.pinned = False
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
return ChatModel.model_validate(chat)
except Exception:
return None
- def get_chat_tags_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> list[TagModel]:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async def get_chat_tags_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> list[TagModel]:
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
tag_ids = chat.meta.get('tags', [])
- return Tags.get_tags_by_ids_and_user_id(tag_ids, user_id, db=db)
+ return await Tags.get_tags_by_ids_and_user_id(tag_ids, user_id, db=db)
- def get_chat_list_by_user_id_and_tag_name(
+ async def get_chat_list_by_user_id_and_tag_name(
self,
user_id: str,
tag_name: str,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ChatTitleIdResponse]:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at).filter_by(
+ user_id=user_id
+ )
tag_id = tag_name.replace(' ', '_').lower()
- log.info(f'DB dialect name: {db.bind.dialect.name}')
- if db.bind.dialect.name == 'sqlite':
- # SQLite JSON1 querying for tags within the meta JSON field
- query = query.filter(
+ bind = await db.connection()
+ dialect_name = bind.dialect.name
+ log.info(f'DB dialect name: {dialect_name}')
+ if dialect_name == 'sqlite':
+ stmt = stmt.filter(
text(f"EXISTS (SELECT 1 FROM json_each(Chat.meta, '$.tags') WHERE json_each.value = :tag_id)")
).params(tag_id=tag_id)
- elif db.bind.dialect.name == 'postgresql':
- # PostgreSQL JSON query for tags within the meta JSON field (for `json` type)
- query = query.filter(
+ elif dialect_name == 'postgresql':
+ stmt = stmt.filter(
text("EXISTS (SELECT 1 FROM json_array_elements_text(Chat.meta->'tags') elem WHERE elem = :tag_id)")
).params(tag_id=tag_id)
else:
- raise NotImplementedError(f'Unsupported dialect: {db.bind.dialect.name}')
+ raise NotImplementedError(f'Unsupported dialect: {dialect_name}')
- query = query.order_by(Chat.updated_at.desc(), Chat.id)
-
- query = query.with_entities(Chat.id, Chat.title, Chat.updated_at, Chat.created_at, Chat.last_read_at)
+ stmt = stmt.order_by(Chat.updated_at.desc(), Chat.id)
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- all_chats = query.all()
+ result = await db.execute(stmt)
+ all_chats = result.all()
return [
ChatTitleIdResponse.model_validate(
{
@@ -1365,49 +1377,54 @@ class ChatTable:
for chat in all_chats
]
- def add_chat_tag_by_id_and_user_id_and_tag_name(
- self, id: str, user_id: str, tag_name: str, db: Optional[Session] = None
+ async def add_chat_tag_by_id_and_user_id_and_tag_name(
+ self, id: str, user_id: str, tag_name: str, db: Optional[AsyncSession] = None
) -> Optional[ChatModel]:
tag_id = tag_name.replace(' ', '_').lower()
- Tags.ensure_tags_exist([tag_name], user_id, db=db)
+ await Tags.ensure_tags_exist([tag_name], user_id, db=db)
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
if tag_id not in chat.meta.get('tags', []):
chat.meta = {
**chat.meta,
'tags': list(set(chat.meta.get('tags', []) + [tag_id])),
}
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
return ChatModel.model_validate(chat)
except Exception:
return None
- def count_chats_by_tag_name_and_user_id(self, tag_name: str, user_id: str, db: Optional[Session] = None) -> int:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id, archived=False)
+ async def count_chats_by_tag_name_and_user_id(
+ self, tag_name: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> int:
+ async with get_async_db_context(db) as db:
+ stmt = select(func.count(Chat.id)).filter_by(user_id=user_id, archived=False)
tag_id = tag_name.replace(' ', '_').lower()
- if db.bind.dialect.name == 'sqlite':
- query = query.filter(
+ bind = await db.connection()
+ dialect_name = bind.dialect.name
+ if dialect_name == 'sqlite':
+ stmt = stmt.filter(
text("EXISTS (SELECT 1 FROM json_each(Chat.meta, '$.tags') WHERE json_each.value = :tag_id)")
).params(tag_id=tag_id)
- elif db.bind.dialect.name == 'postgresql':
- query = query.filter(
+ elif dialect_name == 'postgresql':
+ stmt = stmt.filter(
text("EXISTS (SELECT 1 FROM json_array_elements_text(Chat.meta->'tags') elem WHERE elem = :tag_id)")
).params(tag_id=tag_id)
else:
- raise NotImplementedError(f'Unsupported dialect: {db.bind.dialect.name}')
+ raise NotImplementedError(f'Unsupported dialect: {dialect_name}')
- return query.count()
+ result = await db.execute(stmt)
+ return result.scalar()
- def delete_orphan_tags_for_user(
+ async def delete_orphan_tags_for_user(
self,
tag_ids: list[str],
user_id: str,
threshold: int = 0,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> None:
"""Delete tag rows from *tag_ids* that appear in at most *threshold*
non-archived chats for *user_id*. One query to find orphans, one to
@@ -1419,30 +1436,30 @@ class ChatTable:
"""
if not tag_ids:
return
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
orphans = []
for tag_id in tag_ids:
- count = self.count_chats_by_tag_name_and_user_id(tag_id, user_id, db=db)
+ count = await self.count_chats_by_tag_name_and_user_id(tag_id, user_id, db=db)
if count <= threshold:
orphans.append(tag_id)
- Tags.delete_tags_by_ids_and_user_id(orphans, user_id, db=db)
+ await Tags.delete_tags_by_ids_and_user_id(orphans, user_id, db=db)
- def count_chats_by_folder_id_and_user_id(self, folder_id: str, user_id: str, db: Optional[Session] = None) -> int:
- with get_db_context(db) as db:
- query = db.query(Chat).filter_by(user_id=user_id)
-
- query = query.filter_by(folder_id=folder_id)
- count = query.count()
+ async def count_chats_by_folder_id_and_user_id(
+ self, folder_id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> int:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(func.count(Chat.id)).filter_by(user_id=user_id, folder_id=folder_id))
+ count = result.scalar()
log.info(f"Count of chats for folder '{folder_id}': {count}")
return count
- def delete_tag_by_id_and_user_id_and_tag_name(
- self, id: str, user_id: str, tag_name: str, db: Optional[Session] = None
+ async def delete_tag_by_id_and_user_id_and_tag_name(
+ self, id: str, user_id: str, tag_name: str, db: Optional[AsyncSession] = None
) -> bool:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
tags = chat.meta.get('tags', [])
tag_id = tag_name.replace(' ', '_').lower()
@@ -1451,134 +1468,138 @@ class ChatTable:
**chat.meta,
'tags': list(set(tags)),
}
- db.commit()
+ await db.commit()
return True
except Exception:
return False
- def delete_all_tags_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_all_tags_by_id_and_user_id(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context(db) as db:
+ chat = await db.get(Chat, id)
chat.meta = {
**chat.meta,
'tags': [],
}
- db.commit()
+ await db.commit()
return True
except Exception:
return False
- def delete_chat_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_chat_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(AutomationRun).filter_by(chat_id=id).update(
- {AutomationRun.chat_id: None}, synchronize_session=False
- )
- db.query(ChatMessage).filter_by(chat_id=id).delete()
- db.query(Chat).filter_by(id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(update(AutomationRun).filter_by(chat_id=id).values(chat_id=None))
+ await db.execute(delete(ChatMessage).filter_by(chat_id=id))
+ await db.execute(delete(Chat).filter_by(id=id))
+ await db.commit()
- return True and self.delete_shared_chat_by_chat_id(id, db=db)
+ return True and await self.delete_shared_chat_by_chat_id(id, db=db)
except Exception:
return False
- def delete_chat_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_chat_by_id_and_user_id(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(AutomationRun).filter_by(chat_id=id).update(
- {AutomationRun.chat_id: None}, synchronize_session=False
- )
- db.query(ChatMessage).filter_by(chat_id=id).delete()
- db.query(Chat).filter_by(id=id, user_id=user_id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(update(AutomationRun).filter_by(chat_id=id).values(chat_id=None))
+ await db.execute(delete(ChatMessage).filter_by(chat_id=id))
+ await db.execute(delete(Chat).filter_by(id=id, user_id=user_id))
+ await db.commit()
- return True and self.delete_shared_chat_by_chat_id(id, db=db)
+ return True and await self.delete_shared_chat_by_chat_id(id, db=db)
except Exception:
return False
- def delete_chats_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_chats_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- self.delete_shared_chats_by_user_id(user_id, db=db)
+ async with get_async_db_context(db) as db:
+ await self.delete_shared_chats_by_user_id(user_id, db=db)
- chat_id_subquery = db.query(Chat.id).filter_by(user_id=user_id).subquery()
- db.query(AutomationRun).filter(AutomationRun.chat_id.in_(chat_id_subquery)).update(
- {AutomationRun.chat_id: None}, synchronize_session=False
+ chat_id_subquery = select(Chat.id).filter_by(user_id=user_id).scalar_subquery()
+ await db.execute(
+ update(AutomationRun)
+ .filter(AutomationRun.chat_id.in_(select(Chat.id).filter_by(user_id=user_id)))
+ .values(chat_id=None)
)
- db.query(ChatMessage).filter(ChatMessage.chat_id.in_(chat_id_subquery)).delete(
- synchronize_session=False
+ await db.execute(
+ delete(ChatMessage).filter(ChatMessage.chat_id.in_(select(Chat.id).filter_by(user_id=user_id)))
)
- db.query(Chat).filter_by(user_id=user_id).delete()
- db.commit()
+ await db.execute(delete(Chat).filter_by(user_id=user_id))
+ await db.commit()
return True
except Exception:
return False
- def delete_chats_by_user_id_and_folder_id(self, user_id: str, folder_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_chats_by_user_id_and_folder_id(
+ self, user_id: str, folder_id: str, db: Optional[AsyncSession] = None
+ ) -> bool:
try:
- with get_db_context(db) as db:
- chat_id_subquery = db.query(Chat.id).filter_by(user_id=user_id, folder_id=folder_id).subquery()
- db.query(AutomationRun).filter(AutomationRun.chat_id.in_(chat_id_subquery)).update(
- {AutomationRun.chat_id: None}, synchronize_session=False
+ async with get_async_db_context(db) as db:
+ chat_ids_stmt = select(Chat.id).filter_by(user_id=user_id, folder_id=folder_id)
+ await db.execute(
+ update(AutomationRun).filter(AutomationRun.chat_id.in_(chat_ids_stmt)).values(chat_id=None)
)
- db.query(ChatMessage).filter(ChatMessage.chat_id.in_(chat_id_subquery)).delete(
- synchronize_session=False
- )
- db.query(Chat).filter_by(user_id=user_id, folder_id=folder_id).delete()
- db.commit()
+ await db.execute(delete(ChatMessage).filter(ChatMessage.chat_id.in_(chat_ids_stmt)))
+ await db.execute(delete(Chat).filter_by(user_id=user_id, folder_id=folder_id))
+ await db.commit()
return True
except Exception:
return False
- def move_chats_by_user_id_and_folder_id(
+ async def move_chats_by_user_id_and_folder_id(
self,
user_id: str,
folder_id: str,
new_folder_id: Optional[str],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
try:
- with get_db_context(db) as db:
- db.query(Chat).filter_by(user_id=user_id, folder_id=folder_id).update({'folder_id': new_folder_id})
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(
+ update(Chat).filter_by(user_id=user_id, folder_id=folder_id).values(folder_id=new_folder_id)
+ )
+ await db.commit()
return True
except Exception:
return False
- def delete_shared_chats_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_shared_chats_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- id_rows = db.query(Chat.id).filter_by(user_id=user_id).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Chat.id).filter_by(user_id=user_id))
+ id_rows = result.all()
shared_chat_ids = [f'shared-{row[0]}' for row in id_rows]
- # Use subquery to delete chat_messages for shared chats
- shared_id_subq = db.query(Chat.id).filter(Chat.user_id.in_(shared_chat_ids)).subquery()
- db.query(ChatMessage).filter(ChatMessage.chat_id.in_(shared_id_subq)).delete(synchronize_session=False)
- db.query(Chat).filter(Chat.user_id.in_(shared_chat_ids)).delete()
- db.commit()
+ if shared_chat_ids:
+ # Get shared chat IDs to delete associated messages
+ shared_result = await db.execute(select(Chat.id).filter(Chat.user_id.in_(shared_chat_ids)))
+ shared_ids = [row[0] for row in shared_result.all()]
+ if shared_ids:
+ await db.execute(delete(ChatMessage).filter(ChatMessage.chat_id.in_(shared_ids)))
+ await db.execute(delete(Chat).filter(Chat.user_id.in_(shared_chat_ids)))
+ await db.commit()
return True
except Exception:
return False
- def insert_chat_files(
+ async def insert_chat_files(
self,
chat_id: str,
message_id: str,
file_ids: list[str],
user_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[list[ChatFileModel]]:
if not file_ids:
return None
chat_message_file_ids = [
- item.id for item in self.get_chat_files_by_chat_id_and_message_id(chat_id, message_id, db=db)
+ item.id for item in await self.get_chat_files_by_chat_id_and_message_id(chat_id, message_id, db=db)
]
# Remove duplicates and existing file_ids
file_ids = list(set([file_id for file_id in file_ids if file_id and file_id not in chat_message_file_ids]))
@@ -1586,7 +1607,7 @@ class ChatTable:
return None
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
now = int(time.time())
chat_files = [
@@ -1605,66 +1626,64 @@ class ChatTable:
results = [ChatFile(**chat_file.model_dump()) for chat_file in chat_files]
db.add_all(results)
- db.commit()
+ await db.commit()
return chat_files
except Exception:
return None
- def get_chat_files_by_chat_id_and_message_id(
- self, chat_id: str, message_id: str, db: Optional[Session] = None
+ async def get_chat_files_by_chat_id_and_message_id(
+ self, chat_id: str, message_id: str, db: Optional[AsyncSession] = None
) -> list[ChatFileModel]:
- with get_db_context(db) as db:
- all_chat_files = (
- db.query(ChatFile)
- .filter_by(chat_id=chat_id, message_id=message_id)
- .order_by(ChatFile.created_at.asc())
- .all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(ChatFile).filter_by(chat_id=chat_id, message_id=message_id).order_by(ChatFile.created_at.asc())
)
+ all_chat_files = result.scalars().all()
return [ChatFileModel.model_validate(chat_file) for chat_file in all_chat_files]
- def delete_chat_file(self, chat_id: str, file_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_chat_file(self, chat_id: str, file_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(ChatFile).filter_by(chat_id=chat_id, file_id=file_id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(ChatFile).filter_by(chat_id=chat_id, file_id=file_id))
+ await db.commit()
return True
except Exception:
return False
- def get_shared_chats_by_file_id(self, file_id: str, db: Optional[Session] = None) -> list[ChatModel]:
- with get_db_context(db) as db:
- # Join Chat and ChatFile tables to get shared chats associated with the file_id
- all_chats = (
- db.query(Chat)
+ async def get_shared_chats_by_file_id(self, file_id: str, db: Optional[AsyncSession] = None) -> list[ChatModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Chat)
.join(ChatFile, Chat.id == ChatFile.chat_id)
.filter(ChatFile.file_id == file_id, Chat.share_id.isnot(None))
- .all()
)
+ all_chats = result.scalars().all()
return [ChatModel.model_validate(chat) for chat in all_chats]
- def update_chat_tasks_by_id(self, id: str, tasks: list[dict]) -> Optional[ChatModel]:
+ async def update_chat_tasks_by_id(self, id: str, tasks: list[dict]) -> Optional[ChatModel]:
"""Update the tasks list on a chat."""
try:
- with get_db_context() as db:
- chat = db.get(Chat, id)
+ async with get_async_db_context() as db:
+ chat = await db.get(Chat, id)
if chat is None:
return None
chat.tasks = tasks
- db.commit()
- db.refresh(chat)
+ await db.commit()
+ await db.refresh(chat)
return ChatModel.model_validate(chat)
except Exception:
return None
- def get_chat_tasks_by_id(self, id: str) -> list[dict]:
+ async def get_chat_tasks_by_id(self, id: str) -> list[dict]:
"""Read the tasks list from a chat (lightweight column query)."""
- with get_db_context() as db:
- result = db.query(Chat.tasks).filter_by(id=id).first()
- if result is None or result[0] is None:
+ async with get_async_db_context() as db:
+ result = await db.execute(select(Chat.tasks).filter_by(id=id))
+ row = result.first()
+ if row is None or row[0] is None:
return []
- return result[0]
+ return row[0]
Chats = ChatTable()
diff --git a/backend/open_webui/models/feedbacks.py b/backend/open_webui/models/feedbacks.py
index 9172e2ba8e..02f61f82ee 100644
--- a/backend/open_webui/models/feedbacks.py
+++ b/backend/open_webui/models/feedbacks.py
@@ -3,9 +3,10 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
-from open_webui.models.users import User
+from sqlalchemy import select, delete, func
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
+from open_webui.models.users import User, UserModel
from pydantic import BaseModel, ConfigDict
from sqlalchemy import BigInteger, Column, Text, JSON, Boolean
@@ -139,10 +140,10 @@ class ModelHistoryResponse(BaseModel):
class FeedbackTable:
- def insert_new_feedback(
- self, user_id: str, form_data: FeedbackForm, db: Optional[Session] = None
+ async def insert_new_feedback(
+ self, user_id: str, form_data: FeedbackForm, db: Optional[AsyncSession] = None
) -> Optional[FeedbackModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
id = str(uuid.uuid4())
feedback = FeedbackModel(
**{
@@ -157,8 +158,8 @@ class FeedbackTable:
try:
result = Feedback(**feedback.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return FeedbackModel.model_validate(result)
else:
@@ -167,97 +168,99 @@ class FeedbackTable:
log.exception(f'Error creating a new feedback: {e}')
return None
- def get_feedback_by_id(self, id: str, db: Optional[Session] = None) -> Optional[FeedbackModel]:
+ async def get_feedback_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[FeedbackModel]:
try:
- with get_db_context(db) as db:
- feedback = db.query(Feedback).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(id=id))
+ feedback = result.scalars().first()
if not feedback:
return None
return FeedbackModel.model_validate(feedback)
except Exception:
return None
- def get_feedback_by_id_and_user_id(
- self, id: str, user_id: str, db: Optional[Session] = None
+ async def get_feedback_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[FeedbackModel]:
try:
- with get_db_context(db) as db:
- feedback = db.query(Feedback).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(id=id, user_id=user_id))
+ feedback = result.scalars().first()
if not feedback:
return None
return FeedbackModel.model_validate(feedback)
except Exception:
return None
- def get_feedbacks_by_chat_id(self, chat_id: str, db: Optional[Session] = None) -> list[FeedbackModel]:
+ async def get_feedbacks_by_chat_id(self, chat_id: str, db: Optional[AsyncSession] = None) -> list[FeedbackModel]:
"""Get all feedbacks for a specific chat."""
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# meta.chat_id stores the chat reference
- feedbacks = (
- db.query(Feedback)
+ result = await db.execute(
+ select(Feedback)
.filter(Feedback.meta['chat_id'].as_string() == chat_id)
.order_by(Feedback.created_at.desc())
- .all()
)
+ feedbacks = result.scalars().all()
return [FeedbackModel.model_validate(fb) for fb in feedbacks]
except Exception:
return []
- def get_feedback_items(
+ async def get_feedback_items(
self,
filter: dict = {},
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> FeedbackListResponse:
- with get_db_context(db) as db:
- query = db.query(Feedback, User).join(User, Feedback.user_id == User.id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Feedback, User).join(User, Feedback.user_id == User.id)
if filter:
# Apply model_id filter (exact match)
model_id = filter.get('model_id')
if model_id:
- query = query.filter(Feedback.data['model_id'].as_string() == model_id)
+ stmt = stmt.filter(Feedback.data['model_id'].as_string() == model_id)
order_by = filter.get('order_by')
direction = filter.get('direction')
if order_by == 'username':
if direction == 'asc':
- query = query.order_by(User.name.asc())
+ stmt = stmt.order_by(User.name.asc())
else:
- query = query.order_by(User.name.desc())
+ stmt = stmt.order_by(User.name.desc())
elif order_by == 'model_id':
- # it's stored in feedback.data['model_id']
if direction == 'asc':
- query = query.order_by(Feedback.data['model_id'].as_string().asc())
+ stmt = stmt.order_by(Feedback.data['model_id'].as_string().asc())
else:
- query = query.order_by(Feedback.data['model_id'].as_string().desc())
+ stmt = stmt.order_by(Feedback.data['model_id'].as_string().desc())
elif order_by == 'rating':
- # it's stored in feedback.data['rating']
if direction == 'asc':
- query = query.order_by(Feedback.data['rating'].as_string().asc())
+ stmt = stmt.order_by(Feedback.data['rating'].as_string().asc())
else:
- query = query.order_by(Feedback.data['rating'].as_string().desc())
+ stmt = stmt.order_by(Feedback.data['rating'].as_string().desc())
elif order_by == 'updated_at':
if direction == 'asc':
- query = query.order_by(Feedback.updated_at.asc())
+ stmt = stmt.order_by(Feedback.updated_at.asc())
else:
- query = query.order_by(Feedback.updated_at.desc())
+ stmt = stmt.order_by(Feedback.updated_at.desc())
else:
- query = query.order_by(Feedback.created_at.desc())
+ stmt = stmt.order_by(Feedback.created_at.desc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
feedbacks = []
for feedback, user in items:
@@ -267,15 +270,18 @@ class FeedbackTable:
return FeedbackListResponse(items=feedbacks, total=total)
- def get_all_feedbacks(self, db: Optional[Session] = None) -> list[FeedbackModel]:
- with get_db_context(db) as db:
- return [
- FeedbackModel.model_validate(feedback)
- for feedback in db.query(Feedback).order_by(Feedback.updated_at.desc()).all()
- ]
+ async def get_all_feedbacks(self, db: Optional[AsyncSession] = None) -> list[FeedbackModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).order_by(Feedback.updated_at.desc()))
+ return [FeedbackModel.model_validate(feedback) for feedback in result.scalars().all()]
- def get_all_feedback_ids(self, db: Optional[Session] = None) -> list[FeedbackIdResponse]:
- with get_db_context(db) as db:
+ async def get_all_feedback_ids(self, db: Optional[AsyncSession] = None) -> list[FeedbackIdResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Feedback.id, Feedback.user_id, Feedback.created_at, Feedback.updated_at).order_by(
+ Feedback.updated_at.desc()
+ )
+ )
return [
FeedbackIdResponse(
id=row.id,
@@ -283,36 +289,28 @@ class FeedbackTable:
created_at=row.created_at,
updated_at=row.updated_at,
)
- for row in db.query(
- Feedback.id,
- Feedback.user_id,
- Feedback.created_at,
- Feedback.updated_at,
- )
- .order_by(Feedback.updated_at.desc())
- .all()
+ for row in result.all()
]
- def get_distinct_model_ids(self, db: Optional[Session] = None) -> list[str]:
+ async def get_distinct_model_ids(self, db: Optional[AsyncSession] = None) -> list[str]:
"""Get distinct model_ids from feedback data for filter dropdowns."""
- with get_db_context(db) as db:
- rows = (
- db.query(Feedback.data['model_id'].as_string())
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Feedback.data['model_id'].as_string())
.filter(Feedback.data['model_id'].as_string().isnot(None))
.distinct()
- .all()
)
+ rows = result.all()
return sorted([row[0] for row in rows if row[0]])
- def get_feedbacks_for_leaderboard(self, db: Optional[Session] = None) -> list[LeaderboardFeedbackData]:
+ async def get_feedbacks_for_leaderboard(self, db: Optional[AsyncSession] = None) -> list[LeaderboardFeedbackData]:
"""Fetch only id and data for leaderboard computation (excludes snapshot/meta)."""
- with get_db_context(db) as db:
- return [
- LeaderboardFeedbackData(id=row.id, data=row.data) for row in db.query(Feedback.id, Feedback.data).all()
- ]
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback.id, Feedback.data))
+ return [LeaderboardFeedbackData(id=row.id, data=row.data) for row in result.all()]
- def get_model_evaluation_history(
- self, model_id: str, days: int = 30, db: Optional[Session] = None
+ async def get_model_evaluation_history(
+ self, model_id: str, days: int = 30, db: Optional[AsyncSession] = None
) -> list[ModelHistoryEntry]:
"""
Get daily wins/losses for a specific model over the past N days.
@@ -322,13 +320,16 @@ class FeedbackTable:
from datetime import datetime, timedelta
from collections import defaultdict
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
if days == 0:
# All time - no cutoff
- rows = db.query(Feedback.created_at, Feedback.data).all()
+ result = await db.execute(select(Feedback.created_at, Feedback.data))
else:
cutoff = int(time.time()) - (days * 86400)
- rows = db.query(Feedback.created_at, Feedback.data).filter(Feedback.created_at >= cutoff).all()
+ result = await db.execute(
+ select(Feedback.created_at, Feedback.data).filter(Feedback.created_at >= cutoff)
+ )
+ rows = result.all()
daily_counts = defaultdict(lambda: {'won': 0, 'lost': 0})
first_date = None
@@ -374,25 +375,22 @@ class FeedbackTable:
return result
- def get_feedbacks_by_type(self, type: str, db: Optional[Session] = None) -> list[FeedbackModel]:
- with get_db_context(db) as db:
- return [
- FeedbackModel.model_validate(feedback)
- for feedback in db.query(Feedback).filter_by(type=type).order_by(Feedback.updated_at.desc()).all()
- ]
+ async def get_feedbacks_by_type(self, type: str, db: Optional[AsyncSession] = None) -> list[FeedbackModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(type=type).order_by(Feedback.updated_at.desc()))
+ return [FeedbackModel.model_validate(feedback) for feedback in result.scalars().all()]
- def get_feedbacks_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[FeedbackModel]:
- with get_db_context(db) as db:
- return [
- FeedbackModel.model_validate(feedback)
- for feedback in db.query(Feedback).filter_by(user_id=user_id).order_by(Feedback.updated_at.desc()).all()
- ]
+ async def get_feedbacks_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[FeedbackModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(user_id=user_id).order_by(Feedback.updated_at.desc()))
+ return [FeedbackModel.model_validate(feedback) for feedback in result.scalars().all()]
- def update_feedback_by_id(
- self, id: str, form_data: FeedbackForm, db: Optional[Session] = None
+ async def update_feedback_by_id(
+ self, id: str, form_data: FeedbackForm, db: Optional[AsyncSession] = None
) -> Optional[FeedbackModel]:
- with get_db_context(db) as db:
- feedback = db.query(Feedback).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(id=id))
+ feedback = result.scalars().first()
if not feedback:
return None
@@ -405,18 +403,19 @@ class FeedbackTable:
feedback.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FeedbackModel.model_validate(feedback)
- def update_feedback_by_id_and_user_id(
+ async def update_feedback_by_id_and_user_id(
self,
id: str,
user_id: str,
form_data: FeedbackForm,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[FeedbackModel]:
- with get_db_context(db) as db:
- feedback = db.query(Feedback).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(id=id, user_id=user_id))
+ feedback = result.scalars().first()
if not feedback:
return None
@@ -429,38 +428,40 @@ class FeedbackTable:
feedback.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FeedbackModel.model_validate(feedback)
- def delete_feedback_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- feedback = db.query(Feedback).filter_by(id=id).first()
+ async def delete_feedback_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(id=id))
+ feedback = result.scalars().first()
if not feedback:
return False
- db.delete(feedback)
- db.commit()
+ await db.delete(feedback)
+ await db.commit()
return True
- def delete_feedback_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- feedback = db.query(Feedback).filter_by(id=id, user_id=user_id).first()
+ async def delete_feedback_by_id_and_user_id(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Feedback).filter_by(id=id, user_id=user_id))
+ feedback = result.scalars().first()
if not feedback:
return False
- db.delete(feedback)
- db.commit()
+ await db.delete(feedback)
+ await db.commit()
return True
- def delete_feedbacks_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- result = db.query(Feedback).filter_by(user_id=user_id).delete()
- db.commit()
- return result > 0
+ async def delete_feedbacks_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(delete(Feedback).filter_by(user_id=user_id))
+ await db.commit()
+ return result.rowcount > 0
- def delete_all_feedbacks(self, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- result = db.query(Feedback).delete()
- db.commit()
- return result > 0
+ async def delete_all_feedbacks(self, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(delete(Feedback))
+ await db.commit()
+ return result.rowcount > 0
Feedbacks = FeedbackTable()
diff --git a/backend/open_webui/models/files.py b/backend/open_webui/models/files.py
index 7a9f77a3b0..cfdcfbc2d9 100644
--- a/backend/open_webui/models/files.py
+++ b/backend/open_webui/models/files.py
@@ -2,8 +2,9 @@ import logging
import time
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, func
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.utils.misc import sanitize_metadata
from pydantic import BaseModel, ConfigDict, model_validator
from sqlalchemy import BigInteger, Column, String, Text, JSON
@@ -124,8 +125,10 @@ class FileUpdateForm(BaseModel):
class FilesTable:
- def insert_new_file(self, user_id: str, form_data: FileForm, db: Optional[Session] = None) -> Optional[FileModel]:
- with get_db_context(db) as db:
+ async def insert_new_file(
+ self, user_id: str, form_data: FileForm, db: Optional[AsyncSession] = None
+ ) -> Optional[FileModel]:
+ async with get_async_db_context(db) as db:
file_data = form_data.model_dump()
# Sanitize meta to remove non-JSON-serializable objects
@@ -145,8 +148,8 @@ class FilesTable:
try:
result = File(**file.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return FileModel.model_validate(result)
else:
@@ -155,21 +158,24 @@ class FilesTable:
log.exception(f'Error inserting a new file: {e}')
return None
- def get_file_by_id(self, id: str, db: Optional[Session] = None) -> Optional[FileModel]:
+ async def get_file_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[FileModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
- file = db.get(File, id)
- return FileModel.model_validate(file)
+ file = await db.get(File, id)
+ return FileModel.model_validate(file) if file else None
except Exception:
return None
except Exception:
return None
- def get_file_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> Optional[FileModel]:
- with get_db_context(db) as db:
+ async def get_file_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[FileModel]:
+ async with get_async_db_context(db) as db:
try:
- file = db.query(File).filter_by(id=id, user_id=user_id).first()
+ result = await db.execute(select(File).filter_by(id=id, user_id=user_id))
+ file = result.scalars().first()
if file:
return FileModel.model_validate(file)
else:
@@ -177,10 +183,14 @@ class FilesTable:
except Exception:
return None
- def get_file_metadata_by_id(self, id: str, db: Optional[Session] = None) -> Optional[FileMetadataResponse]:
- with get_db_context(db) as db:
+ async def get_file_metadata_by_id(
+ self, id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[FileMetadataResponse]:
+ async with get_async_db_context(db) as db:
try:
- file = db.get(File, id)
+ file = await db.get(File, id)
+ if not file:
+ return None
return FileMetadataResponse(
id=file.id,
hash=file.hash,
@@ -191,12 +201,13 @@ class FilesTable:
except Exception:
return None
- def get_files(self, db: Optional[Session] = None) -> list[FileModel]:
- with get_db_context(db) as db:
- return [FileModel.model_validate(file) for file in db.query(File).all()]
+ async def get_files(self, db: Optional[AsyncSession] = None) -> list[FileModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(File))
+ return [FileModel.model_validate(file) for file in result.scalars().all()]
- def check_access_by_user_id(self, id, user_id, permission='write', db: Optional[Session] = None) -> bool:
- file = self.get_file_by_id(id, db=db)
+ async def check_access_by_user_id(self, id, user_id, permission='write', db: Optional[AsyncSession] = None) -> bool:
+ file = await self.get_file_by_id(id, db=db)
if not file:
return False
if file.user_id == user_id:
@@ -204,51 +215,53 @@ class FilesTable:
# Implement additional access control logic here as needed
return False
- def get_files_by_ids(self, ids: list[str], db: Optional[Session] = None) -> list[FileModel]:
- with get_db_context(db) as db:
- return [
- FileModel.model_validate(file)
- for file in db.query(File).filter(File.id.in_(ids)).order_by(File.updated_at.desc()).all()
- ]
+ async def get_files_by_ids(self, ids: list[str], db: Optional[AsyncSession] = None) -> list[FileModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(File).filter(File.id.in_(ids)).order_by(File.updated_at.desc()))
+ return [FileModel.model_validate(file) for file in result.scalars().all()]
- def get_file_metadatas_by_ids(self, ids: list[str], db: Optional[Session] = None) -> list[FileMetadataResponse]:
- with get_db_context(db) as db:
- return [
- FileMetadataResponse(
- id=file.id,
- hash=file.hash,
- meta=file.meta,
- created_at=file.created_at,
- updated_at=file.updated_at,
- )
- for file in db.query(File.id, File.hash, File.meta, File.created_at, File.updated_at)
+ async def get_file_metadatas_by_ids(
+ self, ids: list[str], db: Optional[AsyncSession] = None
+ ) -> list[FileMetadataResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(File.id, File.hash, File.meta, File.created_at, File.updated_at)
.filter(File.id.in_(ids))
.order_by(File.updated_at.desc())
- .all()
+ )
+ return [
+ FileMetadataResponse(
+ id=row.id,
+ hash=row.hash,
+ meta=row.meta,
+ created_at=row.created_at,
+ updated_at=row.updated_at,
+ )
+ for row in result.all()
]
- def get_files_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[FileModel]:
- with get_db_context(db) as db:
- return [FileModel.model_validate(file) for file in db.query(File).filter_by(user_id=user_id).all()]
+ async def get_files_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[FileModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(File).filter_by(user_id=user_id))
+ return [FileModel.model_validate(file) for file in result.scalars().all()]
- def get_file_list(
+ async def get_file_list(
self,
user_id: Optional[str] = None,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> 'FileListResponse':
- with get_db_context(db) as db:
- query = db.query(File)
+ async with get_async_db_context(db) as db:
+ stmt = select(File)
if user_id:
- query = query.filter_by(user_id=user_id)
+ stmt = stmt.filter_by(user_id=user_id)
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
- items = [
- FileModelResponse.model_validate(file, from_attributes=True)
- for file in query.order_by(File.updated_at.desc(), File.id.desc()).offset(skip).limit(limit).all()
- ]
+ result = await db.execute(stmt.order_by(File.updated_at.desc(), File.id.desc()).offset(skip).limit(limit))
+ items = [FileModelResponse.model_validate(file, from_attributes=True) for file in result.scalars().all()]
return FileListResponse(items=items, total=total)
@@ -275,13 +288,13 @@ class FilesTable:
pattern = pattern.replace('?', '_')
return pattern
- def search_files(
+ async def search_files(
self,
user_id: Optional[str] = None,
filename: str = '*',
skip: int = 0,
limit: int = 100,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[FileModel]:
"""
Search files with glob pattern matching, optional user filter, and pagination.
@@ -296,27 +309,26 @@ class FilesTable:
Returns:
List of matching FileModel objects, ordered by created_at descending.
"""
- with get_db_context(db) as db:
- query = db.query(File)
+ async with get_async_db_context(db) as db:
+ stmt = select(File)
if user_id:
- query = query.filter_by(user_id=user_id)
+ stmt = stmt.filter_by(user_id=user_id)
pattern = self._glob_to_like_pattern(filename)
if pattern != '%':
- query = query.filter(File.filename.ilike(pattern, escape='\\'))
+ stmt = stmt.filter(File.filename.ilike(pattern, escape='\\'))
- return [
- FileModel.model_validate(file)
- for file in query.order_by(File.created_at.desc(), File.id.desc()).offset(skip).limit(limit).all()
- ]
+ result = await db.execute(stmt.order_by(File.created_at.desc(), File.id.desc()).offset(skip).limit(limit))
+ return [FileModel.model_validate(file) for file in result.scalars().all()]
- def update_file_by_id(
- self, id: str, form_data: FileUpdateForm, db: Optional[Session] = None
+ async def update_file_by_id(
+ self, id: str, form_data: FileUpdateForm, db: Optional[AsyncSession] = None
) -> Optional[FileModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
- file = db.query(File).filter_by(id=id).first()
+ result = await db.execute(select(File).filter_by(id=id))
+ file = result.scalars().first()
if form_data.hash is not None:
file.hash = form_data.hash
@@ -328,63 +340,70 @@ class FilesTable:
file.meta = {**(file.meta if file.meta else {}), **form_data.meta}
file.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FileModel.model_validate(file)
except Exception as e:
log.exception(f'Error updating file completely by id: {e}')
return None
- def update_file_hash_by_id(self, id: str, hash: Optional[str], db: Optional[Session] = None) -> Optional[FileModel]:
- with get_db_context(db) as db:
+ async def update_file_hash_by_id(
+ self, id: str, hash: Optional[str], db: Optional[AsyncSession] = None
+ ) -> Optional[FileModel]:
+ async with get_async_db_context(db) as db:
try:
- file = db.query(File).filter_by(id=id).first()
+ result = await db.execute(select(File).filter_by(id=id))
+ file = result.scalars().first()
file.hash = hash
file.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FileModel.model_validate(file)
except Exception:
return None
- def update_file_data_by_id(self, id: str, data: dict, db: Optional[Session] = None) -> Optional[FileModel]:
- with get_db_context(db) as db:
+ async def update_file_data_by_id(
+ self, id: str, data: dict, db: Optional[AsyncSession] = None
+ ) -> Optional[FileModel]:
+ async with get_async_db_context(db) as db:
try:
- file = db.query(File).filter_by(id=id).first()
+ result = await db.execute(select(File).filter_by(id=id))
+ file = result.scalars().first()
file.data = {**(file.data if file.data else {}), **data}
file.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FileModel.model_validate(file)
except Exception as e:
return None
- def update_file_metadata_by_id(self, id: str, meta: dict, db: Optional[Session] = None) -> Optional[FileModel]:
- with get_db_context(db) as db:
+ async def update_file_metadata_by_id(
+ self, id: str, meta: dict, db: Optional[AsyncSession] = None
+ ) -> Optional[FileModel]:
+ async with get_async_db_context(db) as db:
try:
- file = db.query(File).filter_by(id=id).first()
+ result = await db.execute(select(File).filter_by(id=id))
+ file = result.scalars().first()
file.meta = {**(file.meta if file.meta else {}), **meta}
file.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FileModel.model_validate(file)
except Exception:
return None
- return False
-
- def delete_file_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_file_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- db.query(File).filter_by(id=id).delete()
- db.commit()
+ await db.execute(delete(File).filter_by(id=id))
+ await db.commit()
return True
except Exception:
return False
- def delete_all_files(self, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_all_files(self, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- db.query(File).delete()
- db.commit()
+ await db.execute(delete(File))
+ await db.commit()
return True
except Exception:
diff --git a/backend/open_webui/models/folders.py b/backend/open_webui/models/folders.py
index cd9c9bbc67..c553239482 100644
--- a/backend/open_webui/models/folders.py
+++ b/backend/open_webui/models/folders.py
@@ -6,10 +6,10 @@ import re
from pydantic import BaseModel, ConfigDict
-from sqlalchemy import BigInteger, Column, Text, JSON, Boolean, func
-from sqlalchemy.orm import Session
+from sqlalchemy import BigInteger, Column, Text, JSON, Boolean, func, select, delete
+from sqlalchemy.ext.asyncio import AsyncSession
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from open_webui.internal.db import Base, JSONField, get_async_db_context
log = logging.getLogger(__name__)
@@ -74,25 +74,25 @@ class FolderForm(BaseModel):
data: Optional[dict] = None
meta: Optional[dict] = None
parent_id: Optional[str] = None
- model_config = ConfigDict(extra='allow')
+ model_config = ConfigDict(extra='forbid')
class FolderUpdateForm(BaseModel):
name: Optional[str] = None
data: Optional[dict] = None
meta: Optional[dict] = None
- model_config = ConfigDict(extra='allow')
+ model_config = ConfigDict(extra='forbid')
class FolderTable:
- def insert_new_folder(
+ async def insert_new_folder(
self,
user_id: str,
form_data: FolderForm,
parent_id: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[FolderModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
id = str(uuid.uuid4())
folder = FolderModel(
**{
@@ -107,8 +107,8 @@ class FolderTable:
try:
result = Folder(**folder.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return FolderModel.model_validate(result)
else:
@@ -117,12 +117,13 @@ class FolderTable:
log.exception(f'Error inserting a new folder: {e}')
return None
- def get_folder_by_id_and_user_id(
- self, id: str, user_id: str, db: Optional[Session] = None
+ async def get_folder_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[FolderModel]:
try:
- with get_db_context(db) as db:
- folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(id=id, user_id=user_id))
+ folder = result.scalars().first()
if not folder:
return None
@@ -131,48 +132,48 @@ class FolderTable:
except Exception:
return None
- def get_children_folders_by_id_and_user_id(
- self, id: str, user_id: str, db: Optional[Session] = None
+ async def get_children_folders_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[list[FolderModel]]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
folders = []
- def get_children(folder):
- children = self.get_folders_by_parent_id_and_user_id(folder.id, user_id, db=db)
+ async def get_children(folder):
+ children = await self.get_folders_by_parent_id_and_user_id(folder.id, user_id, db=db)
for child in children:
- get_children(child)
+ await get_children(child)
folders.append(child)
- folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
+ result = await db.execute(select(Folder).filter_by(id=id, user_id=user_id))
+ folder = result.scalars().first()
if not folder:
return None
- get_children(folder)
+ await get_children(folder)
return folders
except Exception:
return None
- def get_folders_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[FolderModel]:
- with get_db_context(db) as db:
- return [FolderModel.model_validate(folder) for folder in db.query(Folder).filter_by(user_id=user_id).all()]
+ async def get_folders_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[FolderModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(user_id=user_id))
+ return [FolderModel.model_validate(folder) for folder in result.scalars().all()]
- def get_folder_by_parent_id_and_user_id_and_name(
+ async def get_folder_by_parent_id_and_user_id_and_name(
self,
parent_id: Optional[str],
user_id: str,
name: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[FolderModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Check if folder exists
- folder = (
- db.query(Folder)
- .filter_by(parent_id=parent_id, user_id=user_id)
- .filter(Folder.name.ilike(name))
- .first()
+ result = await db.execute(
+ select(Folder).filter_by(parent_id=parent_id, user_id=user_id).filter(Folder.name.ilike(name))
)
+ folder = result.scalars().first()
if not folder:
return None
@@ -182,25 +183,24 @@ class FolderTable:
log.error(f'get_folder_by_parent_id_and_user_id_and_name: {e}')
return None
- def get_folders_by_parent_id_and_user_id(
- self, parent_id: Optional[str], user_id: str, db: Optional[Session] = None
+ async def get_folders_by_parent_id_and_user_id(
+ self, parent_id: Optional[str], user_id: str, db: Optional[AsyncSession] = None
) -> list[FolderModel]:
- with get_db_context(db) as db:
- return [
- FolderModel.model_validate(folder)
- for folder in db.query(Folder).filter_by(parent_id=parent_id, user_id=user_id).all()
- ]
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(parent_id=parent_id, user_id=user_id))
+ return [FolderModel.model_validate(folder) for folder in result.scalars().all()]
- def update_folder_parent_id_by_id_and_user_id(
+ async def update_folder_parent_id_by_id_and_user_id(
self,
id: str,
user_id: str,
parent_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[FolderModel]:
try:
- with get_db_context(db) as db:
- folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(id=id, user_id=user_id))
+ folder = result.scalars().first()
if not folder:
return None
@@ -208,38 +208,38 @@ class FolderTable:
folder.parent_id = parent_id
folder.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FolderModel.model_validate(folder)
except Exception as e:
log.error(f'update_folder: {e}')
return
- def update_folder_by_id_and_user_id(
+ async def update_folder_by_id_and_user_id(
self,
id: str,
user_id: str,
form_data: FolderUpdateForm,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[FolderModel]:
try:
- with get_db_context(db) as db:
- folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(id=id, user_id=user_id))
+ folder = result.scalars().first()
if not folder:
return None
form_data = form_data.model_dump(exclude_unset=True)
- existing_folder = (
- db.query(Folder)
- .filter_by(
+ existing_result = await db.execute(
+ select(Folder).filter_by(
name=form_data.get('name'),
parent_id=folder.parent_id,
user_id=user_id,
)
- .first()
)
+ existing_folder = existing_result.scalars().first()
if existing_folder and existing_folder.id != id:
return None
@@ -258,19 +258,20 @@ class FolderTable:
}
folder.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FolderModel.model_validate(folder)
except Exception as e:
log.error(f'update_folder: {e}')
return
- def update_folder_is_expanded_by_id_and_user_id(
- self, id: str, user_id: str, is_expanded: bool, db: Optional[Session] = None
+ async def update_folder_is_expanded_by_id_and_user_id(
+ self, id: str, user_id: str, is_expanded: bool, db: Optional[AsyncSession] = None
) -> Optional[FolderModel]:
try:
- with get_db_context(db) as db:
- folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(id=id, user_id=user_id))
+ folder = result.scalars().first()
if not folder:
return None
@@ -278,37 +279,41 @@ class FolderTable:
folder.is_expanded = is_expanded
folder.updated_at = int(time.time())
- db.commit()
+ await db.commit()
return FolderModel.model_validate(folder)
except Exception as e:
log.error(f'update_folder: {e}')
return
- def delete_folder_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> list[str]:
+ async def delete_folder_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> list[str]:
try:
folder_ids = []
- with get_db_context(db) as db:
- folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(id=id, user_id=user_id))
+ folder = result.scalars().first()
if not folder:
return folder_ids
folder_ids.append(folder.id)
# Delete all children folders
- def delete_children(folder):
- folder_children = self.get_folders_by_parent_id_and_user_id(folder.id, user_id, db=db)
+ async def delete_children(folder):
+ folder_children = await self.get_folders_by_parent_id_and_user_id(folder.id, user_id, db=db)
for folder_child in folder_children:
- delete_children(folder_child)
+ await delete_children(folder_child)
folder_ids.append(folder_child.id)
- folder = db.query(Folder).filter_by(id=folder_child.id).first()
- db.delete(folder)
- db.commit()
+ child_result = await db.execute(select(Folder).filter_by(id=folder_child.id))
+ child_folder = child_result.scalars().first()
+ await db.delete(child_folder)
+ await db.commit()
- delete_children(folder)
- db.delete(folder)
- db.commit()
+ await delete_children(folder)
+ await db.delete(folder)
+ await db.commit()
return folder_ids
except Exception as e:
log.error(f'delete_folder: {e}')
@@ -319,8 +324,8 @@ class FolderTable:
name = re.sub(r'[\s_]+', ' ', name)
return name.strip().lower()
- def search_folders_by_names(
- self, user_id: str, queries: list[str], db: Optional[Session] = None
+ async def search_folders_by_names(
+ self, user_id: str, queries: list[str], db: Optional[AsyncSession] = None
) -> list[FolderModel]:
"""
Search for folders for a user where the name matches any of the queries, treating _ and space as equivalent, case-insensitive.
@@ -330,16 +335,18 @@ class FolderTable:
return []
results = {}
- with get_db_context(db) as db:
- folders = db.query(Folder).filter_by(user_id=user_id).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(user_id=user_id))
+ folders = result.scalars().all()
for folder in folders:
if self.normalize_folder_name(folder.name) in normalized_queries:
results[folder.id] = FolderModel.model_validate(folder)
# get children folders
- children = self.get_children_folders_by_id_and_user_id(folder.id, user_id, db=db)
- for child in children:
- results[child.id] = child
+ children = await self.get_children_folders_by_id_and_user_id(folder.id, user_id, db=db)
+ if children:
+ for child in children:
+ results[child.id] = child
# Return the results as a list
if not results:
@@ -348,16 +355,17 @@ class FolderTable:
results = list(results.values())
return results
- def search_folders_by_name_contains(
- self, user_id: str, query: str, db: Optional[Session] = None
+ async def search_folders_by_name_contains(
+ self, user_id: str, query: str, db: Optional[AsyncSession] = None
) -> list[FolderModel]:
"""
Partial match: normalized name contains (as substring) the normalized query.
"""
normalized_query = self.normalize_folder_name(query)
results = []
- with get_db_context(db) as db:
- folders = db.query(Folder).filter_by(user_id=user_id).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Folder).filter_by(user_id=user_id))
+ folders = result.scalars().all()
for folder in folders:
norm_name = self.normalize_folder_name(folder.name)
if normalized_query in norm_name:
diff --git a/backend/open_webui/models/functions.py b/backend/open_webui/models/functions.py
index f9761e947a..ddac317863 100644
--- a/backend/open_webui/models/functions.py
+++ b/backend/open_webui/models/functions.py
@@ -2,8 +2,9 @@ import logging
import time
from typing import Optional
-from sqlalchemy.orm import Session, defer
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.users import Users, UserModel, UserResponse
from pydantic import BaseModel, ConfigDict
from sqlalchemy import BigInteger, Boolean, Column, String, Text, Index
@@ -107,12 +108,12 @@ class FunctionValves(BaseModel):
class FunctionsTable:
- def insert_new_function(
+ async def insert_new_function(
self,
user_id: str,
type: str,
form_data: FunctionForm,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[FunctionModel]:
function = FunctionModel(
**{
@@ -125,11 +126,11 @@ class FunctionsTable:
)
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
result = Function(**function.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return FunctionModel.model_validate(result)
else:
@@ -138,17 +139,18 @@ class FunctionsTable:
log.exception(f'Error creating a new function: {e}')
return None
- def sync_functions(
+ async def sync_functions(
self,
user_id: str,
functions: list[FunctionWithValvesModel],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[FunctionWithValvesModel]:
# Synchronize functions for a user by updating existing ones, inserting new ones, and removing those that are no longer present.
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Get existing functions
- existing_functions = db.query(Function).all()
+ result = await db.execute(select(Function))
+ existing_functions = result.scalars().all()
existing_ids = {func.id for func in existing_functions}
# Prepare a set of new function IDs
@@ -157,12 +159,14 @@ class FunctionsTable:
# Update or insert functions
for func in functions:
if func.id in existing_ids:
- db.query(Function).filter_by(id=func.id).update(
- {
+ await db.execute(
+ update(Function)
+ .filter_by(id=func.id)
+ .values(
**func.model_dump(),
- 'user_id': user_id,
- 'updated_at': int(time.time()),
- }
+ user_id=user_id,
+ updated_at=int(time.time()),
+ )
)
else:
new_func = Function(
@@ -177,24 +181,25 @@ class FunctionsTable:
# Remove functions that are no longer present
for func in existing_functions:
if func.id not in new_function_ids:
- db.delete(func)
+ await db.delete(func)
- db.commit()
+ await db.commit()
- return [FunctionModel.model_validate(func) for func in db.query(Function).all()]
+ result = await db.execute(select(Function))
+ return [FunctionModel.model_validate(func) for func in result.scalars().all()]
except Exception as e:
log.exception(f'Error syncing functions for user {user_id}: {e}')
return []
- def get_function_by_id(self, id: str, db: Optional[Session] = None) -> Optional[FunctionModel]:
+ async def get_function_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[FunctionModel]:
try:
- with get_db_context(db) as db:
- function = db.get(Function, id)
- return FunctionModel.model_validate(function)
+ async with get_async_db_context(db) as db:
+ function = await db.get(Function, id)
+ return FunctionModel.model_validate(function) if function else None
except Exception:
return None
- def get_functions_by_ids(self, ids: list[str], db: Optional[Session] = None) -> list[FunctionModel]:
+ async def get_functions_by_ids(self, ids: list[str], db: Optional[AsyncSession] = None) -> list[FunctionModel]:
"""
Batch fetch multiple functions by their IDs in a single query.
Returns functions in the same order as the input IDs (None entries filtered out).
@@ -202,8 +207,9 @@ class FunctionsTable:
if not ids:
return []
try:
- with get_db_context(db) as db:
- functions = db.query(Function).filter(Function.id.in_(ids)).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Function).filter(Function.id.in_(ids)))
+ functions = result.scalars().all()
# Create a dict for O(1) lookup
func_dict = {f.id: FunctionModel.model_validate(f) for f in functions}
# Return in original order, filtering out any not found
@@ -211,27 +217,29 @@ class FunctionsTable:
except Exception:
return []
- def get_functions(
- self, active_only=False, include_valves=False, db: Optional[Session] = None
+ async def get_functions(
+ self, active_only=False, include_valves=False, db: Optional[AsyncSession] = None
) -> list[FunctionModel | FunctionWithValvesModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
if active_only:
- functions = db.query(Function).filter_by(is_active=True).all()
-
+ result = await db.execute(select(Function).filter_by(is_active=True))
else:
- functions = db.query(Function).all()
+ result = await db.execute(select(Function))
+
+ functions = result.scalars().all()
if include_valves:
return [FunctionWithValvesModel.model_validate(function) for function in functions]
else:
return [FunctionModel.model_validate(function) for function in functions]
- def get_function_list(self, db: Optional[Session] = None) -> list[FunctionUserResponse]:
- with get_db_context(db) as db:
- functions = db.query(Function).options(defer(Function.content)).order_by(Function.updated_at.desc()).all()
+ async def get_function_list(self, db: Optional[AsyncSession] = None) -> list[FunctionUserResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Function).order_by(Function.updated_at.desc()))
+ functions = result.scalars().all()
user_ids = list(set(func.user_id for func in functions))
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
return [
@@ -253,42 +261,36 @@ class FunctionsTable:
for func in functions
]
- def get_functions_by_type(self, type: str, active_only=False, db: Optional[Session] = None) -> list[FunctionModel]:
- with get_db_context(db) as db:
+ async def get_functions_by_type(
+ self, type: str, active_only=False, db: Optional[AsyncSession] = None
+ ) -> list[FunctionModel]:
+ async with get_async_db_context(db) as db:
if active_only:
- return [
- FunctionModel.model_validate(function)
- for function in db.query(Function).filter_by(type=type, is_active=True).all()
- ]
+ result = await db.execute(select(Function).filter_by(type=type, is_active=True))
else:
- return [
- FunctionModel.model_validate(function) for function in db.query(Function).filter_by(type=type).all()
- ]
+ result = await db.execute(select(Function).filter_by(type=type))
+ return [FunctionModel.model_validate(function) for function in result.scalars().all()]
- def get_global_filter_functions(self, db: Optional[Session] = None) -> list[FunctionModel]:
- with get_db_context(db) as db:
- return [
- FunctionModel.model_validate(function)
- for function in db.query(Function).filter_by(type='filter', is_active=True, is_global=True).all()
- ]
+ async def get_global_filter_functions(self, db: Optional[AsyncSession] = None) -> list[FunctionModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Function).filter_by(type='filter', is_active=True, is_global=True))
+ return [FunctionModel.model_validate(function) for function in result.scalars().all()]
- def get_global_action_functions(self, db: Optional[Session] = None) -> list[FunctionModel]:
- with get_db_context(db) as db:
- return [
- FunctionModel.model_validate(function)
- for function in db.query(Function).filter_by(type='action', is_active=True, is_global=True).all()
- ]
+ async def get_global_action_functions(self, db: Optional[AsyncSession] = None) -> list[FunctionModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Function).filter_by(type='action', is_active=True, is_global=True))
+ return [FunctionModel.model_validate(function) for function in result.scalars().all()]
- def get_function_valves_by_id(self, id: str, db: Optional[Session] = None) -> Optional[dict]:
- with get_db_context(db) as db:
+ async def get_function_valves_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[dict]:
+ async with get_async_db_context(db) as db:
try:
- function = db.get(Function, id)
+ function = await db.get(Function, id)
return function.valves if function.valves else {}
except Exception as e:
log.exception(f'Error getting function valves by id {id}: {e}')
return None
- def get_function_valves_by_ids(self, ids: list[str], db: Optional[Session] = None) -> dict[str, dict]:
+ async def get_function_valves_by_ids(self, ids: list[str], db: Optional[AsyncSession] = None) -> dict[str, dict]:
"""
Batch fetch valves for multiple functions in a single query.
Returns a dict mapping function_id -> valves dict.
@@ -297,33 +299,34 @@ class FunctionsTable:
if not ids:
return {}
try:
- with get_db_context(db) as db:
- functions = db.query(Function.id, Function.valves).filter(Function.id.in_(ids)).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Function.id, Function.valves).filter(Function.id.in_(ids)))
+ functions = result.all()
return {f.id: (f.valves if f.valves else {}) for f in functions}
except Exception as e:
log.exception(f'Error batch-fetching function valves: {e}')
return {}
- def update_function_valves_by_id(
- self, id: str, valves: dict, db: Optional[Session] = None
+ async def update_function_valves_by_id(
+ self, id: str, valves: dict, db: Optional[AsyncSession] = None
) -> Optional[FunctionValves]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
- function = db.get(Function, id)
+ function = await db.get(Function, id)
function.valves = valves
function.updated_at = int(time.time())
- db.commit()
- db.refresh(function)
+ await db.commit()
+ await db.refresh(function)
return FunctionModel.model_validate(function)
except Exception:
return None
- def update_function_metadata_by_id(
- self, id: str, metadata: dict, db: Optional[Session] = None
+ async def update_function_metadata_by_id(
+ self, id: str, metadata: dict, db: Optional[AsyncSession] = None
) -> Optional[FunctionModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
- function = db.get(Function, id)
+ function = await db.get(Function, id)
if function:
if function.meta:
@@ -332,8 +335,8 @@ class FunctionsTable:
function.meta = metadata
function.updated_at = int(time.time())
- db.commit()
- db.refresh(function)
+ await db.commit()
+ await db.refresh(function)
return FunctionModel.model_validate(function)
else:
return None
@@ -341,9 +344,11 @@ class FunctionsTable:
log.exception(f'Error updating function metadata by id {id}: {e}')
return None
- def get_user_valves_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> Optional[dict]:
+ async def get_user_valves_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[dict]:
try:
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
user_settings = user.settings.model_dump() if user.settings else {}
# Check if user has "functions" and "valves" settings
@@ -357,11 +362,11 @@ class FunctionsTable:
log.exception(f'Error getting user values by id {id} and user id {user_id}')
return None
- def update_user_valves_by_id_and_user_id(
- self, id: str, user_id: str, valves: dict, db: Optional[Session] = None
+ async def update_user_valves_by_id_and_user_id(
+ self, id: str, user_id: str, valves: dict, db: Optional[AsyncSession] = None
) -> Optional[dict]:
try:
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
user_settings = user.settings.model_dump() if user.settings else {}
# Check if user has "functions" and "valves" settings
@@ -373,47 +378,51 @@ class FunctionsTable:
user_settings['functions']['valves'][id] = valves
# Update the user settings in the database
- Users.update_user_by_id(user_id, {'settings': user_settings}, db=db)
+ await Users.update_user_by_id(user_id, {'settings': user_settings}, db=db)
return user_settings['functions']['valves'][id]
except Exception as e:
log.exception(f'Error updating user valves by id {id} and user_id {user_id}: {e}')
return None
- def update_function_by_id(self, id: str, updated: dict, db: Optional[Session] = None) -> Optional[FunctionModel]:
- with get_db_context(db) as db:
+ async def update_function_by_id(
+ self, id: str, updated: dict, db: Optional[AsyncSession] = None
+ ) -> Optional[FunctionModel]:
+ async with get_async_db_context(db) as db:
try:
- db.query(Function).filter_by(id=id).update(
- {
+ await db.execute(
+ update(Function)
+ .filter_by(id=id)
+ .values(
**updated,
- 'updated_at': int(time.time()),
- }
+ updated_at=int(time.time()),
+ )
)
- db.commit()
- function = db.get(Function, id)
+ await db.commit()
+ function = await db.get(Function, id)
return FunctionModel.model_validate(function) if function else None
except Exception:
return None
- def deactivate_all_functions(self, db: Optional[Session] = None) -> Optional[bool]:
- with get_db_context(db) as db:
+ async def deactivate_all_functions(self, db: Optional[AsyncSession] = None) -> Optional[bool]:
+ async with get_async_db_context(db) as db:
try:
- db.query(Function).update(
- {
- 'is_active': False,
- 'updated_at': int(time.time()),
- }
+ await db.execute(
+ update(Function).values(
+ is_active=False,
+ updated_at=int(time.time()),
+ )
)
- db.commit()
+ await db.commit()
return True
except Exception:
return None
- def delete_function_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_function_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- db.query(Function).filter_by(id=id).delete()
- db.commit()
+ await db.execute(delete(Function).filter_by(id=id))
+ await db.commit()
return True
except Exception:
diff --git a/backend/open_webui/models/groups.py b/backend/open_webui/models/groups.py
index fc4cfb0d31..bc199fac5b 100644
--- a/backend/open_webui/models/groups.py
+++ b/backend/open_webui/models/groups.py
@@ -4,8 +4,9 @@ import time
from typing import Optional
import uuid
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update, func, and_, or_, cast, String
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.env import DEFAULT_GROUP_SHARE_PERMISSION
from open_webui.models.files import FileMetadataResponse
@@ -15,15 +16,9 @@ from pydantic import BaseModel, ConfigDict
from sqlalchemy import (
BigInteger,
Column,
- String,
Text,
JSON,
- and_,
- func,
ForeignKey,
- cast,
- or_,
- select,
)
log = logging.getLogger(__name__)
@@ -143,10 +138,10 @@ class GroupTable:
group_data['data']['config']['share'] = DEFAULT_GROUP_SHARE_PERMISSION
return group_data
- def insert_new_group(
- self, user_id: str, form_data: GroupForm, db: Optional[Session] = None
+ async def insert_new_group(
+ self, user_id: str, form_data: GroupForm, db: Optional[AsyncSession] = None
) -> Optional[GroupModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
group_data = self._ensure_default_share_config(form_data.model_dump(exclude_none=True))
group = GroupModel(
**{
@@ -161,8 +156,8 @@ class GroupTable:
try:
result = Group(**group.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return GroupModel.model_validate(result)
else:
@@ -171,18 +166,20 @@ class GroupTable:
except Exception:
return None
- def get_all_groups(self, db: Optional[Session] = None) -> list[GroupModel]:
- with get_db_context(db) as db:
- groups = db.query(Group).order_by(Group.updated_at.desc()).all()
+ async def get_all_groups(self, db: Optional[AsyncSession] = None) -> list[GroupModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Group).order_by(Group.updated_at.desc()))
+ groups = result.scalars().all()
return [GroupModel.model_validate(group) for group in groups]
- def get_group_by_name(self, name: str, db: Optional[Session] = None) -> Optional[GroupModel]:
- with get_db_context(db) as db:
- group = db.query(Group).filter(Group.name == name).first()
+ async def get_group_by_name(self, name: str, db: Optional[AsyncSession] = None) -> Optional[GroupModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Group).filter(Group.name == name))
+ group = result.scalars().first()
return GroupModel.model_validate(group) if group else None
- def get_groups(self, filter, db: Optional[Session] = None) -> list[GroupResponse]:
- with get_db_context(db) as db:
+ async def get_groups(self, filter, db: Optional[AsyncSession] = None) -> list[GroupResponse]:
+ async with get_async_db_context(db) as db:
member_count = (
select(func.count(GroupMember.user_id))
.where(GroupMember.group_id == Group.id)
@@ -190,11 +187,11 @@ class GroupTable:
.scalar_subquery()
.label('member_count')
)
- query = db.query(Group, member_count)
+ stmt = select(Group, member_count)
if filter:
if 'query' in filter:
- query = query.filter(Group.name.ilike(f'%{filter["query"]}%'))
+ stmt = stmt.filter(Group.name.ilike(f'%{filter["query"]}%'))
# When share filter is present, member check is handled in the share logic
if 'share' in filter:
@@ -218,20 +215,21 @@ class GroupTable:
json_share_lower == 'members',
Group.id.in_(member_groups_select),
)
- query = query.filter(or_(anyone_can_share, members_only_and_is_member))
+ stmt = stmt.filter(or_(anyone_can_share, members_only_and_is_member))
else:
- query = query.filter(anyone_can_share)
+ stmt = stmt.filter(anyone_can_share)
else:
- query = query.filter(and_(Group.data.isnot(None), json_share_lower == 'false'))
+ stmt = stmt.filter(and_(Group.data.isnot(None), json_share_lower == 'false'))
else:
# Only apply member_id filter when share filter is NOT present
if 'member_id' in filter:
- query = query.filter(
+ stmt = stmt.filter(
Group.id.in_(select(GroupMember.group_id).where(GroupMember.user_id == filter['member_id']))
)
- results = query.order_by(Group.updated_at.desc()).all()
+ result = await db.execute(stmt.order_by(Group.updated_at.desc()))
+ rows = result.all()
return [
GroupResponse.model_validate(
@@ -240,32 +238,34 @@ class GroupTable:
'member_count': count or 0,
}
)
- for group, count in results
+ for group, count in rows
]
- def search_groups(
+ async def search_groups(
self,
filter: Optional[dict] = None,
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> GroupListResponse:
- with get_db_context(db) as db:
- query = db.query(Group)
+ async with get_async_db_context(db) as db:
+ stmt = select(Group)
if filter:
if 'query' in filter:
- query = query.filter(Group.name.ilike(f'%{filter["query"]}%'))
+ stmt = stmt.filter(Group.name.ilike(f'%{filter["query"]}%'))
if 'member_id' in filter:
- query = query.filter(
+ stmt = stmt.filter(
Group.id.in_(select(GroupMember.group_id).where(GroupMember.user_id == filter['member_id']))
)
if 'share' in filter:
share_value = filter['share']
- query = query.filter(Group.data.op('->>')('share') == str(share_value))
+ stmt = stmt.filter(Group.data.op('->>')('share') == str(share_value))
- total = query.count()
+ # Get total count
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
member_count = (
select(func.count(GroupMember.user_id))
@@ -274,7 +274,14 @@ class GroupTable:
.scalar_subquery()
.label('member_count')
)
- results = query.add_columns(member_count).order_by(Group.updated_at.desc()).offset(skip).limit(limit).all()
+ result = await db.execute(
+ select(Group, member_count)
+ .where(Group.id.in_(select(stmt.subquery().c.id)))
+ .order_by(Group.updated_at.desc())
+ .offset(skip)
+ .limit(limit)
+ )
+ rows = result.all()
return {
'items': [
@@ -284,65 +291,69 @@ class GroupTable:
'member_count': count or 0,
}
)
- for group, count in results
+ for group, count in rows
],
'total': total,
}
- def get_groups_by_member_id(self, user_id: str, db: Optional[Session] = None) -> list[GroupModel]:
- with get_db_context(db) as db:
- return [
- GroupModel.model_validate(group)
- for group in db.query(Group)
+ async def get_groups_by_member_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[GroupModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Group)
.join(GroupMember, GroupMember.group_id == Group.id)
.filter(GroupMember.user_id == user_id)
.order_by(Group.updated_at.desc())
- .all()
- ]
+ )
+ return [GroupModel.model_validate(group) for group in result.scalars().all()]
- def get_groups_by_member_ids(
- self, user_ids: list[str], db: Optional[Session] = None
+ async def get_groups_by_member_ids(
+ self, user_ids: list[str], db: Optional[AsyncSession] = None
) -> dict[str, list[GroupModel]]:
"""Fetch groups for multiple users in a single query to avoid N+1."""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Query GroupMember joined with Group, filtering by user_ids
- results = (
- db.query(GroupMember.user_id, Group)
+ result = await db.execute(
+ select(GroupMember.user_id, Group)
.join(Group, Group.id == GroupMember.group_id)
.filter(GroupMember.user_id.in_(user_ids))
.order_by(Group.updated_at.desc())
- .all()
)
+ rows = result.all()
# Group groups by user_id
user_groups: dict[str, list[GroupModel]] = {uid: [] for uid in user_ids}
- for user_id, group in results:
+ for user_id, group in rows:
user_groups[user_id].append(GroupModel.model_validate(group))
return user_groups
- def get_group_by_id(self, id: str, db: Optional[Session] = None) -> Optional[GroupModel]:
+ async def get_group_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[GroupModel]:
try:
- with get_db_context(db) as db:
- group = db.query(Group).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Group).filter_by(id=id))
+ group = result.scalars().first()
return GroupModel.model_validate(group) if group else None
except Exception:
return None
- def get_group_user_ids_by_id(self, id: str, db: Optional[Session] = None) -> list[str]:
- with get_db_context(db) as db:
- members = db.query(GroupMember.user_id).filter(GroupMember.group_id == id).all()
+ async def get_group_user_ids_by_id(self, id: str, db: Optional[AsyncSession] = None) -> list[str]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(GroupMember.user_id).filter(GroupMember.group_id == id))
+ members = result.all()
if not members:
return []
return [m[0] for m in members]
- def get_group_user_ids_by_ids(self, group_ids: list[str], db: Optional[Session] = None) -> dict[str, list[str]]:
- with get_db_context(db) as db:
- members = (
- db.query(GroupMember.group_id, GroupMember.user_id).filter(GroupMember.group_id.in_(group_ids)).all()
+ async def get_group_user_ids_by_ids(
+ self, group_ids: list[str], db: Optional[AsyncSession] = None
+ ) -> dict[str, list[str]]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(GroupMember.group_id, GroupMember.user_id).filter(GroupMember.group_id.in_(group_ids))
)
+ members = result.all()
group_user_ids: dict[str, list[str]] = {group_id: [] for group_id in group_ids}
@@ -351,10 +362,12 @@ class GroupTable:
return group_user_ids
- def set_group_user_ids_by_id(self, group_id: str, user_ids: list[str], db: Optional[Session] = None) -> None:
- with get_db_context(db) as db:
+ async def set_group_user_ids_by_id(
+ self, group_id: str, user_ids: list[str], db: Optional[AsyncSession] = None
+ ) -> None:
+ async with get_async_db_context(db) as db:
# Delete existing members
- db.query(GroupMember).filter(GroupMember.group_id == group_id).delete()
+ await db.execute(delete(GroupMember).filter(GroupMember.group_id == group_id))
# Insert new members
now = int(time.time())
@@ -370,101 +383,104 @@ class GroupTable:
]
db.add_all(new_members)
- db.commit()
+ await db.commit()
- def get_group_member_count_by_id(self, id: str, db: Optional[Session] = None) -> int:
- with get_db_context(db) as db:
- count = db.query(func.count(GroupMember.user_id)).filter(GroupMember.group_id == id).scalar()
+ async def get_group_member_count_by_id(self, id: str, db: Optional[AsyncSession] = None) -> int:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(func.count(GroupMember.user_id)).filter(GroupMember.group_id == id))
+ count = result.scalar()
return count if count else 0
- def get_group_member_counts_by_ids(self, ids: list[str], db: Optional[Session] = None) -> dict[str, int]:
+ async def get_group_member_counts_by_ids(self, ids: list[str], db: Optional[AsyncSession] = None) -> dict[str, int]:
if not ids:
return {}
- with get_db_context(db) as db:
- rows = (
- db.query(GroupMember.group_id, func.count(GroupMember.user_id))
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(GroupMember.group_id, func.count(GroupMember.user_id))
.filter(GroupMember.group_id.in_(ids))
.group_by(GroupMember.group_id)
- .all()
)
+ rows = result.all()
return {group_id: count for group_id, count in rows}
- def update_group_by_id(
+ async def update_group_by_id(
self,
id: str,
form_data: GroupUpdateForm,
overwrite: bool = False,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[GroupModel]:
try:
- with get_db_context(db) as db:
- db.query(Group).filter_by(id=id).update(
- {
+ async with get_async_db_context(db) as db:
+ await db.execute(
+ update(Group)
+ .filter_by(id=id)
+ .values(
**form_data.model_dump(exclude_none=True),
- 'updated_at': int(time.time()),
- }
+ updated_at=int(time.time()),
+ )
)
- db.commit()
- return self.get_group_by_id(id=id, db=db)
+ await db.commit()
+ return await self.get_group_by_id(id=id, db=db)
except Exception as e:
log.exception(e)
return None
- def delete_group_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_group_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(Group).filter_by(id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(Group).filter_by(id=id))
+ await db.commit()
return True
except Exception:
return False
- def delete_all_groups(self, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_all_groups(self, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- db.query(Group).delete()
- db.commit()
+ await db.execute(delete(Group))
+ await db.commit()
return True
except Exception:
return False
- def remove_user_from_all_groups(self, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def remove_user_from_all_groups(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
# Find all groups the user belongs to
- groups = (
- db.query(Group)
+ result = await db.execute(
+ select(Group)
.join(GroupMember, GroupMember.group_id == Group.id)
.filter(GroupMember.user_id == user_id)
- .all()
)
+ groups = result.scalars().all()
# Remove the user from each group
for group in groups:
- db.query(GroupMember).filter(
- GroupMember.group_id == group.id, GroupMember.user_id == user_id
- ).delete()
+ await db.execute(
+ delete(GroupMember).filter(GroupMember.group_id == group.id, GroupMember.user_id == user_id)
+ )
- db.query(Group).filter_by(id=group.id).update({'updated_at': int(time.time())})
+ await db.execute(update(Group).filter_by(id=group.id).values(updated_at=int(time.time())))
- db.commit()
+ await db.commit()
return True
except Exception:
- db.rollback()
+ await db.rollback()
return False
- def create_groups_by_group_names(
- self, user_id: str, group_names: list[str], db: Optional[Session] = None
+ async def create_groups_by_group_names(
+ self, user_id: str, group_names: list[str], db: Optional[AsyncSession] = None
) -> list[GroupModel]:
# check for existing groups
- existing_groups = self.get_all_groups(db=db)
+ existing_groups = await self.get_all_groups(db=db)
existing_group_names = {group.name for group in existing_groups}
new_groups = []
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
for group_name in group_names:
if group_name not in existing_group_names:
new_group = GroupModel(
@@ -483,31 +499,33 @@ class GroupTable:
try:
result = Group(**new_group.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
new_groups.append(GroupModel.model_validate(result))
except Exception as e:
log.exception(e)
continue
return new_groups
- def sync_groups_by_group_names(self, user_id: str, group_names: list[str], db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def sync_groups_by_group_names(
+ self, user_id: str, group_names: list[str], db: Optional[AsyncSession] = None
+ ) -> bool:
+ async with get_async_db_context(db) as db:
try:
now = int(time.time())
# 1. Groups that SHOULD contain the user
- target_groups = db.query(Group).filter(Group.name.in_(group_names)).all()
+ result = await db.execute(select(Group).filter(Group.name.in_(group_names)))
+ target_groups = result.scalars().all()
target_group_ids = {g.id for g in target_groups}
# 2. Groups the user is CURRENTLY in
- existing_group_ids = {
- g.id
- for g in db.query(Group)
+ result = await db.execute(
+ select(Group)
.join(GroupMember, GroupMember.group_id == Group.id)
.filter(GroupMember.user_id == user_id)
- .all()
- }
+ )
+ existing_group_ids = {g.id for g in result.scalars().all()}
# 3. Determine adds + removals
groups_to_add = target_group_ids - existing_group_ids
@@ -515,15 +533,15 @@ class GroupTable:
# 4. Remove in one bulk delete
if groups_to_remove:
- db.query(GroupMember).filter(
- GroupMember.user_id == user_id,
- GroupMember.group_id.in_(groups_to_remove),
- ).delete(synchronize_session=False)
-
- db.query(Group).filter(Group.id.in_(groups_to_remove)).update(
- {'updated_at': now}, synchronize_session=False
+ await db.execute(
+ delete(GroupMember).filter(
+ GroupMember.user_id == user_id,
+ GroupMember.group_id.in_(groups_to_remove),
+ )
)
+ await db.execute(update(Group).filter(Group.id.in_(groups_to_remove)).values(updated_at=now))
+
# 5. Bulk insert missing memberships
for group_id in groups_to_add:
db.add(
@@ -537,27 +555,26 @@ class GroupTable:
)
if groups_to_add:
- db.query(Group).filter(Group.id.in_(groups_to_add)).update(
- {'updated_at': now}, synchronize_session=False
- )
+ await db.execute(update(Group).filter(Group.id.in_(groups_to_add)).values(updated_at=now))
- db.commit()
+ await db.commit()
return True
except Exception as e:
log.exception(e)
- db.rollback()
+ await db.rollback()
return False
- def add_users_to_group(
+ async def add_users_to_group(
self,
id: str,
user_ids: Optional[list[str]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[GroupModel]:
try:
- with get_db_context(db) as db:
- group = db.query(Group).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Group).filter_by(id=id))
+ group = result.scalars().first()
if not group:
return None
@@ -574,15 +591,14 @@ class GroupTable:
updated_at=now,
)
)
- db.flush() # Detect unique constraint violation early
+ await db.flush() # Detect unique constraint violation early
except Exception:
- db.rollback() # Clear failed INSERT
- db.begin() # Start a new transaction
+ await db.rollback() # Clear failed INSERT
continue # Duplicate → ignore
group.updated_at = now
- db.commit()
- db.refresh(group)
+ await db.commit()
+ await db.refresh(group)
return GroupModel.model_validate(group)
@@ -590,15 +606,16 @@ class GroupTable:
log.exception(e)
return None
- def remove_users_from_group(
+ async def remove_users_from_group(
self,
id: str,
user_ids: Optional[list[str]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[GroupModel]:
try:
- with get_db_context(db) as db:
- group = db.query(Group).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Group).filter_by(id=id))
+ group = result.scalars().first()
if not group:
return None
@@ -606,15 +623,15 @@ class GroupTable:
return GroupModel.model_validate(group)
# Remove users from group_member in batch
- db.query(GroupMember).filter(GroupMember.group_id == id, GroupMember.user_id.in_(user_ids)).delete(
- synchronize_session=False
+ await db.execute(
+ delete(GroupMember).filter(GroupMember.group_id == id, GroupMember.user_id.in_(user_ids))
)
# Update group timestamp
group.updated_at = int(time.time())
- db.commit()
- db.refresh(group)
+ await db.commit()
+ await db.refresh(group)
return GroupModel.model_validate(group)
except Exception as e:
diff --git a/backend/open_webui/models/knowledge.py b/backend/open_webui/models/knowledge.py
index 30510221fb..2750ef6058 100644
--- a/backend/open_webui/models/knowledge.py
+++ b/backend/open_webui/models/knowledge.py
@@ -4,8 +4,9 @@ import time
from typing import Optional
import uuid
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update, or_, func
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.files import (
File,
@@ -27,7 +28,6 @@ from sqlalchemy import (
Text,
JSON,
UniqueConstraint,
- or_,
)
log = logging.getLogger(__name__)
@@ -134,25 +134,25 @@ class KnowledgeFileListResponse(BaseModel):
class KnowledgeTable:
- def _get_access_grants(self, knowledge_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('knowledge', knowledge_id, db=db)
+ async def _get_access_grants(self, knowledge_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('knowledge', knowledge_id, db=db)
- def _to_knowledge_model(
+ async def _to_knowledge_model(
self,
knowledge: Knowledge,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> KnowledgeModel:
knowledge_data = KnowledgeModel.model_validate(knowledge).model_dump(exclude={'access_grants'})
knowledge_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(knowledge_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(knowledge_data['id'], db=db)
)
return KnowledgeModel.model_validate(knowledge_data)
- def insert_new_knowledge(
- self, user_id: str, form_data: KnowledgeForm, db: Optional[Session] = None
+ async def insert_new_knowledge(
+ self, user_id: str, form_data: KnowledgeForm, db: Optional[AsyncSession] = None
) -> Optional[KnowledgeModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
knowledge = KnowledgeModel(
**{
**form_data.model_dump(exclude={'access_grants'}),
@@ -167,27 +167,28 @@ class KnowledgeTable:
try:
result = Knowledge(**knowledge.model_dump(exclude={'access_grants'}))
db.add(result)
- db.commit()
- db.refresh(result)
- AccessGrants.set_access_grants('knowledge', result.id, form_data.access_grants, db=db)
+ await db.commit()
+ await db.refresh(result)
+ await AccessGrants.set_access_grants('knowledge', result.id, form_data.access_grants, db=db)
if result:
- return self._to_knowledge_model(result, db=db)
+ return await self._to_knowledge_model(result, db=db)
else:
return None
except Exception:
return None
- def get_knowledge_bases(
- self, skip: int = 0, limit: int = 30, db: Optional[Session] = None
+ async def get_knowledge_bases(
+ self, skip: int = 0, limit: int = 30, db: Optional[AsyncSession] = None
) -> list[KnowledgeUserModel]:
- with get_db_context(db) as db:
- all_knowledge = db.query(Knowledge).order_by(Knowledge.updated_at.desc()).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Knowledge).order_by(Knowledge.updated_at.desc()))
+ all_knowledge = result.scalars().all()
user_ids = list(set(knowledge.user_id for knowledge in all_knowledge))
knowledge_ids = [knowledge.id for knowledge in all_knowledge]
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
- grants_map = AccessGrants.get_grants_by_resources('knowledge', knowledge_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('knowledge', knowledge_ids, db=db)
knowledge_bases = []
for knowledge in all_knowledge:
@@ -195,10 +196,12 @@ class KnowledgeTable:
knowledge_bases.append(
KnowledgeUserModel.model_validate(
{
- **self._to_knowledge_model(
- knowledge,
- access_grants=grants_map.get(knowledge.id, []),
- db=db,
+ **(
+ await self._to_knowledge_model(
+ knowledge,
+ access_grants=grants_map.get(knowledge.id, []),
+ db=db,
+ )
).model_dump(),
'user': user.model_dump() if user else None,
}
@@ -206,22 +209,22 @@ class KnowledgeTable:
)
return knowledge_bases
- def search_knowledge_bases(
+ async def search_knowledge_bases(
self,
user_id: str,
filter: dict,
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> KnowledgeListResponse:
try:
- with get_db_context(db) as db:
- query = db.query(Knowledge, User).outerjoin(User, User.id == Knowledge.user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Knowledge, User).outerjoin(User, User.id == Knowledge.user_id)
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(
+ stmt = stmt.filter(
or_(
Knowledge.name.ilike(f'%{query_key}%'),
Knowledge.description.ilike(f'%{query_key}%'),
@@ -233,41 +236,45 @@ class KnowledgeTable:
view_option = filter.get('view_option')
if view_option == 'created':
- query = query.filter(Knowledge.user_id == user_id)
+ stmt = stmt.filter(Knowledge.user_id == user_id)
elif view_option == 'shared':
- query = query.filter(Knowledge.user_id != user_id)
+ stmt = stmt.filter(Knowledge.user_id != user_id)
- query = AccessGrants.has_permission_filter(
+ stmt = AccessGrants.has_permission_filter(
db=db,
- query=query,
+ query=stmt,
DocumentModel=Knowledge,
filter=filter,
resource_type='knowledge',
permission='read',
)
- query = query.order_by(Knowledge.updated_at.desc(), Knowledge.id.asc())
+ stmt = stmt.order_by(Knowledge.updated_at.desc(), Knowledge.id.asc())
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
knowledge_ids = [kb.id for kb, _ in items]
- grants_map = AccessGrants.get_grants_by_resources('knowledge', knowledge_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('knowledge', knowledge_ids, db=db)
knowledge_bases = []
for knowledge_base, user in items:
knowledge_bases.append(
KnowledgeUserModel.model_validate(
{
- **self._to_knowledge_model(
- knowledge_base,
- access_grants=grants_map.get(knowledge_base.id, []),
- db=db,
+ **(
+ await self._to_knowledge_model(
+ knowledge_base,
+ access_grants=grants_map.get(knowledge_base.id, []),
+ db=db,
+ )
).model_dump(),
'user': (UserModel.model_validate(user).model_dump() if user else None),
}
@@ -279,28 +286,27 @@ class KnowledgeTable:
print(e)
return KnowledgeListResponse(items=[], total=0)
- def search_knowledge_files(
- self, filter: dict, skip: int = 0, limit: int = 30, db: Optional[Session] = None
+ async def search_knowledge_files(
+ self, filter: dict, skip: int = 0, limit: int = 30, db: Optional[AsyncSession] = None
) -> KnowledgeFileListResponse:
"""
Scalable version: search files across all knowledge bases the user has
READ access to, without loading all KBs or using large IN() lists.
"""
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Base query: join Knowledge → KnowledgeFile → File
- query = (
- db.query(File, User, Knowledge)
+ stmt = (
+ select(File, User, Knowledge)
.join(KnowledgeFile, File.id == KnowledgeFile.file_id)
.join(Knowledge, KnowledgeFile.knowledge_id == Knowledge.id)
.outerjoin(User, User.id == KnowledgeFile.user_id)
)
# Apply access-control directly to the joined query
- # This makes the database handle filtering, even with 10k+ KBs
- query = AccessGrants.has_permission_filter(
+ stmt = AccessGrants.has_permission_filter(
db=db,
- query=query,
+ query=stmt,
DocumentModel=Knowledge,
filter=filter,
resource_type='knowledge',
@@ -311,20 +317,22 @@ class KnowledgeTable:
if filter:
q = filter.get('query')
if q:
- query = query.filter(File.filename.ilike(f'%{q}%'))
+ stmt = stmt.filter(File.filename.ilike(f'%{q}%'))
# Order by file changes
- query = query.order_by(File.updated_at.desc(), File.id.asc())
+ stmt = stmt.order_by(File.updated_at.desc(), File.id.asc())
# Count before pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- rows = query.all()
+ result = await db.execute(stmt)
+ rows = result.all()
items = []
for file, user, knowledge in rows:
@@ -332,7 +340,7 @@ class KnowledgeTable:
FileUserResponse(
**FileModel.model_validate(file).model_dump(),
user=(UserResponse(**UserModel.model_validate(user).model_dump()) if user else None),
- collection=self._to_knowledge_model(knowledge, db=db).model_dump(),
+ collection=(await self._to_knowledge_model(knowledge, db=db)).model_dump(),
)
)
@@ -342,14 +350,15 @@ class KnowledgeTable:
print('search_knowledge_files error:', e)
return KnowledgeFileListResponse(items=[], total=0)
- def check_access_by_user_id(self, id, user_id, permission='write', db: Optional[Session] = None) -> bool:
- knowledge = self.get_knowledge_by_id(id, db=db)
+ async def check_access_by_user_id(self, id, user_id, permission='write', db: Optional[AsyncSession] = None) -> bool:
+ knowledge = await self.get_knowledge_by_id(id, db=db)
if not knowledge:
return False
if knowledge.user_id == user_id:
return True
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
- return AccessGrants.has_access(
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
+ return await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -358,45 +367,50 @@ class KnowledgeTable:
db=db,
)
- def get_knowledge_bases_by_user_id(
- self, user_id: str, permission: str = 'write', db: Optional[Session] = None
+ async def get_knowledge_bases_by_user_id(
+ self, user_id: str, permission: str = 'write', db: Optional[AsyncSession] = None
) -> list[KnowledgeUserModel]:
- knowledge_bases = self.get_knowledge_bases(db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
- return [
- knowledge_base
- for knowledge_base in knowledge_bases
- if knowledge_base.user_id == user_id
- or AccessGrants.has_access(
+ knowledge_bases = await self.get_knowledge_bases(db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
+
+ result = []
+ for knowledge_base in knowledge_bases:
+ if knowledge_base.user_id == user_id:
+ result.append(knowledge_base)
+ elif await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge_base.id,
permission=permission,
user_group_ids=user_group_ids,
db=db,
- )
- ]
+ ):
+ result.append(knowledge_base)
+ return result
- def get_knowledge_by_id(self, id: str, db: Optional[Session] = None) -> Optional[KnowledgeModel]:
+ async def get_knowledge_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[KnowledgeModel]:
try:
- with get_db_context(db) as db:
- knowledge = db.query(Knowledge).filter_by(id=id).first()
- return self._to_knowledge_model(knowledge, db=db) if knowledge else None
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Knowledge).filter_by(id=id))
+ knowledge = result.scalars().first()
+ return await self._to_knowledge_model(knowledge, db=db) if knowledge else None
except Exception:
return None
- def get_knowledge_by_id_and_user_id(
- self, id: str, user_id: str, db: Optional[Session] = None
+ async def get_knowledge_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[KnowledgeModel]:
- knowledge = self.get_knowledge_by_id(id, db=db)
+ knowledge = await self.get_knowledge_by_id(id, db=db)
if not knowledge:
return None
if knowledge.user_id == user_id:
return knowledge
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
- if AccessGrants.has_access(
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
+ if await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -407,19 +421,19 @@ class KnowledgeTable:
return knowledge
return None
- def get_knowledges_by_file_id(self, file_id: str, db: Optional[Session] = None) -> list[KnowledgeModel]:
+ async def get_knowledges_by_file_id(self, file_id: str, db: Optional[AsyncSession] = None) -> list[KnowledgeModel]:
try:
- with get_db_context(db) as db:
- knowledges = (
- db.query(Knowledge)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Knowledge)
.join(KnowledgeFile, Knowledge.id == KnowledgeFile.knowledge_id)
.filter(KnowledgeFile.file_id == file_id)
- .all()
)
+ knowledges = result.scalars().all()
knowledge_ids = [k.id for k in knowledges]
- grants_map = AccessGrants.get_grants_by_resources('knowledge', knowledge_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('knowledge', knowledge_ids, db=db)
return [
- self._to_knowledge_model(
+ await self._to_knowledge_model(
knowledge,
access_grants=grants_map.get(knowledge.id, []),
db=db,
@@ -429,19 +443,19 @@ class KnowledgeTable:
except Exception:
return []
- def search_files_by_id(
+ async def search_files_by_id(
self,
knowledge_id: str,
user_id: str,
filter: dict,
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> KnowledgeFileListResponse:
try:
- with get_db_context(db) as db:
- query = (
- db.query(File, User)
+ async with get_async_db_context(db) as db:
+ stmt = (
+ select(File, User)
.join(KnowledgeFile, File.id == KnowledgeFile.file_id)
.outerjoin(User, User.id == KnowledgeFile.user_id)
.filter(KnowledgeFile.knowledge_id == knowledge_id)
@@ -453,13 +467,13 @@ class KnowledgeTable:
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(or_(File.filename.ilike(f'%{query_key}%')))
+ stmt = stmt.filter(or_(File.filename.ilike(f'%{query_key}%')))
view_option = filter.get('view_option')
if view_option == 'created':
- query = query.filter(KnowledgeFile.user_id == user_id)
+ stmt = stmt.filter(KnowledgeFile.user_id == user_id)
elif view_option == 'shared':
- query = query.filter(KnowledgeFile.user_id != user_id)
+ stmt = stmt.filter(KnowledgeFile.user_id != user_id)
order_by = filter.get('order_by')
direction = filter.get('direction')
@@ -473,17 +487,19 @@ class KnowledgeTable:
primary_sort = File.updated_at.asc() if is_asc else File.updated_at.desc()
# Apply sort with secondary key for deterministic pagination
- query = query.order_by(primary_sort, File.id.asc())
+ stmt = stmt.order_by(primary_sort, File.id.asc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
files = []
for file, user in items:
@@ -499,35 +515,36 @@ class KnowledgeTable:
print(e)
return KnowledgeFileListResponse(items=[], total=0)
- def get_files_by_id(self, knowledge_id: str, db: Optional[Session] = None) -> list[FileModel]:
+ async def get_files_by_id(self, knowledge_id: str, db: Optional[AsyncSession] = None) -> list[FileModel]:
try:
- with get_db_context(db) as db:
- files = (
- db.query(File)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(File)
.join(KnowledgeFile, File.id == KnowledgeFile.file_id)
.filter(KnowledgeFile.knowledge_id == knowledge_id)
- .all()
)
+ files = result.scalars().all()
return [FileModel.model_validate(file) for file in files]
except Exception:
return []
- def get_file_metadatas_by_id(self, knowledge_id: str, db: Optional[Session] = None) -> list[FileMetadataResponse]:
+ async def get_file_metadatas_by_id(
+ self, knowledge_id: str, db: Optional[AsyncSession] = None
+ ) -> list[FileMetadataResponse]:
try:
- with get_db_context(db) as db:
- files = self.get_files_by_id(knowledge_id, db=db)
- return [FileMetadataResponse(**file.model_dump()) for file in files]
+ files = await self.get_files_by_id(knowledge_id, db=db)
+ return [FileMetadataResponse(**file.model_dump()) for file in files]
except Exception:
return []
- def add_file_to_knowledge_by_id(
+ async def add_file_to_knowledge_by_id(
self,
knowledge_id: str,
file_id: str,
user_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[KnowledgeFileModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
knowledge_file = KnowledgeFileModel(
**{
'id': str(uuid.uuid4()),
@@ -542,8 +559,8 @@ class KnowledgeTable:
try:
result = KnowledgeFile(**knowledge_file.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return KnowledgeFileModel.model_validate(result)
else:
@@ -551,103 +568,107 @@ class KnowledgeTable:
except Exception:
return None
- def has_file(self, knowledge_id: str, file_id: str, db: Optional[Session] = None) -> bool:
+ async def has_file(self, knowledge_id: str, file_id: str, db: Optional[AsyncSession] = None) -> bool:
"""Check whether a file belongs to a knowledge base."""
try:
- with get_db_context(db) as db:
- return db.query(KnowledgeFile).filter_by(knowledge_id=knowledge_id, file_id=file_id).first() is not None
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(KnowledgeFile).filter_by(knowledge_id=knowledge_id, file_id=file_id).limit(1)
+ )
+ return result.scalars().first() is not None
except Exception:
return False
- def remove_file_from_knowledge_by_id(self, knowledge_id: str, file_id: str, db: Optional[Session] = None) -> bool:
+ async def remove_file_from_knowledge_by_id(
+ self, knowledge_id: str, file_id: str, db: Optional[AsyncSession] = None
+ ) -> bool:
try:
- with get_db_context(db) as db:
- db.query(KnowledgeFile).filter_by(knowledge_id=knowledge_id, file_id=file_id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(KnowledgeFile).filter_by(knowledge_id=knowledge_id, file_id=file_id))
+ await db.commit()
return True
except Exception:
return False
- def reset_knowledge_by_id(self, id: str, db: Optional[Session] = None) -> Optional[KnowledgeModel]:
+ async def reset_knowledge_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[KnowledgeModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Delete all knowledge_file entries for this knowledge_id
- db.query(KnowledgeFile).filter_by(knowledge_id=id).delete()
- db.commit()
+ await db.execute(delete(KnowledgeFile).filter_by(knowledge_id=id))
+ await db.commit()
# Update the knowledge entry's updated_at timestamp
- db.query(Knowledge).filter_by(id=id).update(
- {
- 'updated_at': int(time.time()),
- }
- )
- db.commit()
+ await db.execute(update(Knowledge).filter_by(id=id).values(updated_at=int(time.time())))
+ await db.commit()
- return self.get_knowledge_by_id(id=id, db=db)
+ return await self.get_knowledge_by_id(id=id, db=db)
except Exception as e:
log.exception(e)
return None
- def update_knowledge_by_id(
+ async def update_knowledge_by_id(
self,
id: str,
form_data: KnowledgeForm,
overwrite: bool = False,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[KnowledgeModel]:
try:
- with get_db_context(db) as db:
- knowledge = self.get_knowledge_by_id(id=id, db=db)
- db.query(Knowledge).filter_by(id=id).update(
- {
+ async with get_async_db_context(db) as db:
+ await db.execute(
+ update(Knowledge)
+ .filter_by(id=id)
+ .values(
**form_data.model_dump(exclude={'access_grants'}),
- 'updated_at': int(time.time()),
- }
+ updated_at=int(time.time()),
+ )
)
- db.commit()
+ await db.commit()
if form_data.access_grants is not None:
- AccessGrants.set_access_grants('knowledge', id, form_data.access_grants, db=db)
- return self.get_knowledge_by_id(id=id, db=db)
+ await AccessGrants.set_access_grants('knowledge', id, form_data.access_grants, db=db)
+ return await self.get_knowledge_by_id(id=id, db=db)
except Exception as e:
log.exception(e)
return None
- def update_knowledge_data_by_id(
- self, id: str, data: dict, db: Optional[Session] = None
+ async def update_knowledge_data_by_id(
+ self, id: str, data: dict, db: Optional[AsyncSession] = None
) -> Optional[KnowledgeModel]:
try:
- with get_db_context(db) as db:
- knowledge = self.get_knowledge_by_id(id=id, db=db)
- db.query(Knowledge).filter_by(id=id).update(
- {
- 'data': data,
- 'updated_at': int(time.time()),
- }
+ async with get_async_db_context(db) as db:
+ await db.execute(
+ update(Knowledge)
+ .filter_by(id=id)
+ .values(
+ data=data,
+ updated_at=int(time.time()),
+ )
)
- db.commit()
- return self.get_knowledge_by_id(id=id, db=db)
+ await db.commit()
+ return await self.get_knowledge_by_id(id=id, db=db)
except Exception as e:
log.exception(e)
return None
- def delete_knowledge_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_knowledge_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- AccessGrants.revoke_all_access('knowledge', id, db=db)
- db.query(Knowledge).filter_by(id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await AccessGrants.revoke_all_access('knowledge', id, db=db)
+ await db.execute(delete(Knowledge).filter_by(id=id))
+ await db.commit()
return True
except Exception:
return False
- def delete_all_knowledge(self, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_all_knowledge(self, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- knowledge_ids = [row[0] for row in db.query(Knowledge.id).all()]
+ result = await db.execute(select(Knowledge.id))
+ knowledge_ids = [row[0] for row in result.all()]
for knowledge_id in knowledge_ids:
- AccessGrants.revoke_all_access('knowledge', knowledge_id, db=db)
- db.query(Knowledge).delete()
- db.commit()
+ await AccessGrants.revoke_all_access('knowledge', knowledge_id, db=db)
+ await db.execute(delete(Knowledge))
+ await db.commit()
return True
except Exception:
diff --git a/backend/open_webui/models/memories.py b/backend/open_webui/models/memories.py
index 7c34de9f07..e956826800 100644
--- a/backend/open_webui/models/memories.py
+++ b/backend/open_webui/models/memories.py
@@ -2,8 +2,9 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db, get_db_context
+from sqlalchemy import select, delete
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
from pydantic import BaseModel, ConfigDict
from sqlalchemy import BigInteger, Column, String, Text
@@ -40,13 +41,13 @@ class MemoryModel(BaseModel):
class MemoriesTable:
- def insert_new_memory(
+ async def insert_new_memory(
self,
user_id: str,
content: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[MemoryModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
id = str(uuid.uuid4())
memory = MemoryModel(
@@ -60,90 +61,92 @@ class MemoriesTable:
)
result = Memory(**memory.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return MemoryModel.model_validate(result)
else:
return None
- def update_memory_by_id_and_user_id(
+ async def update_memory_by_id_and_user_id(
self,
id: str,
user_id: str,
content: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[MemoryModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
- memory = db.get(Memory, id)
+ memory = await db.get(Memory, id)
if not memory or memory.user_id != user_id:
return None
memory.content = content
memory.updated_at = int(time.time())
- db.commit()
- db.refresh(memory)
+ await db.commit()
+ await db.refresh(memory)
return MemoryModel.model_validate(memory)
except Exception:
return None
- def get_memories(self, db: Optional[Session] = None) -> list[MemoryModel]:
- with get_db_context(db) as db:
+ async def get_memories(self, db: Optional[AsyncSession] = None) -> list[MemoryModel]:
+ async with get_async_db_context(db) as db:
try:
- memories = db.query(Memory).all()
+ result = await db.execute(select(Memory))
+ memories = result.scalars().all()
return [MemoryModel.model_validate(memory) for memory in memories]
except Exception:
return None
- def get_memories_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[MemoryModel]:
- with get_db_context(db) as db:
+ async def get_memories_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[MemoryModel]:
+ async with get_async_db_context(db) as db:
try:
- memories = db.query(Memory).filter_by(user_id=user_id).all()
+ result = await db.execute(select(Memory).filter_by(user_id=user_id))
+ memories = result.scalars().all()
return [MemoryModel.model_validate(memory) for memory in memories]
except Exception:
return None
- def get_memory_by_id(self, id: str, db: Optional[Session] = None) -> Optional[MemoryModel]:
- with get_db_context(db) as db:
+ async def get_memory_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[MemoryModel]:
+ async with get_async_db_context(db) as db:
try:
- memory = db.get(Memory, id)
- return MemoryModel.model_validate(memory)
+ memory = await db.get(Memory, id)
+ return MemoryModel.model_validate(memory) if memory else None
except Exception:
return None
- def delete_memory_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_memory_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- db.query(Memory).filter_by(id=id).delete()
- db.commit()
+ await db.execute(delete(Memory).filter_by(id=id))
+ await db.commit()
return True
except Exception:
return False
- def delete_memories_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_memories_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- db.query(Memory).filter_by(user_id=user_id).delete()
- db.commit()
+ await db.execute(delete(Memory).filter_by(user_id=user_id))
+ await db.commit()
return True
except Exception:
return False
- def delete_memory_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
+ async def delete_memory_by_id_and_user_id(self, id: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
try:
- memory = db.get(Memory, id)
+ memory = await db.get(Memory, id)
if not memory or memory.user_id != user_id:
return None
# Delete the memory
- db.delete(memory)
- db.commit()
+ await db.delete(memory)
+ await db.commit()
return True
except Exception:
diff --git a/backend/open_webui/models/messages.py b/backend/open_webui/models/messages.py
index 034eaac160..7f33a72eff 100644
--- a/backend/open_webui/models/messages.py
+++ b/backend/open_webui/models/messages.py
@@ -3,8 +3,9 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, func
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.tags import TagModel, Tag, Tags
from open_webui.models.users import Users, User, UserNameResponse
from open_webui.models.channels import Channels, ChannelMember
@@ -12,7 +13,7 @@ from open_webui.models.channels import Channels, ChannelMember
from pydantic import BaseModel, ConfigDict, field_validator
from sqlalchemy import BigInteger, Boolean, Column, String, Text, JSON
-from sqlalchemy import or_, func, select, and_, text
+from sqlalchemy import or_, func, and_, text
from sqlalchemy.sql import exists
####################
@@ -137,15 +138,15 @@ class MessageResponse(MessageReplyToResponse):
class MessageTable:
- def insert_new_message(
+ async def insert_new_message(
self,
form_data: MessageForm,
channel_id: str,
user_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[MessageModel]:
- with get_db_context(db) as db:
- channel_member = Channels.join_channel(channel_id, user_id)
+ async with get_async_db_context(db) as db:
+ channel_member = await Channels.join_channel(channel_id, user_id)
id = str(uuid.uuid4())
ts = int(time.time_ns())
@@ -170,38 +171,38 @@ class MessageTable:
result = Message(**message.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
return MessageModel.model_validate(result) if result else None
- def get_message_by_id(
+ async def get_message_by_id(
self,
id: str,
include_thread_replies: Optional[bool] = True,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[MessageResponse]:
- with get_db_context(db) as db:
- message = db.get(Message, id)
+ async with get_async_db_context(db) as db:
+ message = await db.get(Message, id)
if not message:
return None
reply_to_message = (
- self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
+ await self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
if message.reply_to_id
else None
)
- reactions = self.get_reactions_by_message_id(id, db=db)
+ reactions = await self.get_reactions_by_message_id(id, db=db)
thread_replies = []
if include_thread_replies:
- thread_replies = self.get_thread_replies_by_message_id(id, db=db)
+ thread_replies = await self.get_thread_replies_by_message_id(id, db=db)
# Check if message was sent by webhook (webhook info in meta takes precedence)
webhook_info = message.meta.get('webhook') if message.meta else None
if webhook_info and webhook_info.get('id'):
# Look up webhook by ID to get current name
- webhook = Channels.get_webhook_by_id(webhook_info.get('id'), db=db)
+ webhook = await Channels.get_webhook_by_id(webhook_info.get('id'), db=db)
if webhook:
user_info = {
'id': webhook.id,
@@ -216,7 +217,7 @@ class MessageTable:
'role': 'webhook',
}
else:
- user = Users.get_user_by_id(message.user_id, db=db)
+ user = await Users.get_user_by_id(message.user_id, db=db)
user_info = user.model_dump() if user else None
return MessageResponse.model_validate(
@@ -230,34 +231,41 @@ class MessageTable:
}
)
- def get_thread_replies_by_message_id(self, id: str, db: Optional[Session] = None) -> list[MessageReplyToResponse]:
- with get_db_context(db) as db:
- all_messages = db.query(Message).filter_by(parent_id=id).order_by(Message.created_at.desc()).all()
+ async def _resolve_user_info(self, message: Message, db: AsyncSession) -> Optional[dict]:
+ """Resolve user info from message, handling webhook messages."""
+ webhook_info = message.meta.get('webhook') if message.meta else None
+ if webhook_info and webhook_info.get('id'):
+ webhook = await Channels.get_webhook_by_id(webhook_info.get('id'), db=db)
+ if webhook:
+ return {
+ 'id': webhook.id,
+ 'name': webhook.name,
+ 'role': 'webhook',
+ }
+ else:
+ return {
+ 'id': webhook_info.get('id'),
+ 'name': 'Deleted Webhook',
+ 'role': 'webhook',
+ }
+ return None
+
+ async def get_thread_replies_by_message_id(
+ self, id: str, db: Optional[AsyncSession] = None
+ ) -> list[MessageReplyToResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Message).filter_by(parent_id=id).order_by(Message.created_at.desc()))
+ all_messages = result.scalars().all()
messages = []
for message in all_messages:
reply_to_message = (
- self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
+ await self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
if message.reply_to_id
else None
)
- webhook_info = message.meta.get('webhook') if message.meta else None
- user_info = None
- if webhook_info and webhook_info.get('id'):
- webhook = Channels.get_webhook_by_id(webhook_info.get('id'), db=db)
- if webhook:
- user_info = {
- 'id': webhook.id,
- 'name': webhook.name,
- 'role': 'webhook',
- }
- else:
- user_info = {
- 'id': webhook_info.get('id'),
- 'name': 'Deleted Webhook',
- 'role': 'webhook',
- }
+ user_info = await self._resolve_user_info(message, db)
messages.append(
MessageReplyToResponse.model_validate(
@@ -270,51 +278,37 @@ class MessageTable:
)
return messages
- def get_reply_user_ids_by_message_id(self, id: str, db: Optional[Session] = None) -> list[str]:
- with get_db_context(db) as db:
- return [message.user_id for message in db.query(Message).filter_by(parent_id=id).all()]
+ async def get_reply_user_ids_by_message_id(self, id: str, db: Optional[AsyncSession] = None) -> list[str]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Message.user_id).filter_by(parent_id=id))
+ return [row[0] for row in result.all()]
- def get_messages_by_channel_id(
+ async def get_messages_by_channel_id(
self,
channel_id: str,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[MessageReplyToResponse]:
- with get_db_context(db) as db:
- all_messages = (
- db.query(Message)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Message)
.filter_by(channel_id=channel_id, parent_id=None)
.order_by(Message.created_at.desc())
.offset(skip)
.limit(limit)
- .all()
)
+ all_messages = result.scalars().all()
messages = []
for message in all_messages:
reply_to_message = (
- self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
+ await self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
if message.reply_to_id
else None
)
- webhook_info = message.meta.get('webhook') if message.meta else None
- user_info = None
- if webhook_info and webhook_info.get('id'):
- webhook = Channels.get_webhook_by_id(webhook_info.get('id'), db=db)
- if webhook:
- user_info = {
- 'id': webhook.id,
- 'name': webhook.name,
- 'role': 'webhook',
- }
- else:
- user_info = {
- 'id': webhook_info.get('id'),
- 'name': 'Deleted Webhook',
- 'role': 'webhook',
- }
+ user_info = await self._resolve_user_info(message, db)
messages.append(
MessageReplyToResponse.model_validate(
@@ -327,28 +321,28 @@ class MessageTable:
)
return messages
- def get_messages_by_parent_id(
+ async def get_messages_by_parent_id(
self,
channel_id: str,
parent_id: str,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[MessageReplyToResponse]:
- with get_db_context(db) as db:
- message = db.get(Message, parent_id)
+ async with get_async_db_context(db) as db:
+ message = await db.get(Message, parent_id)
if not message:
return []
- all_messages = (
- db.query(Message)
+ result = await db.execute(
+ select(Message)
.filter_by(channel_id=channel_id, parent_id=parent_id)
.order_by(Message.created_at.desc())
.offset(skip)
.limit(limit)
- .all()
)
+ all_messages = list(result.scalars().all())
# If length of all_messages is less than limit, then add the parent message
if len(all_messages) < limit:
@@ -357,27 +351,12 @@ class MessageTable:
messages = []
for message in all_messages:
reply_to_message = (
- self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
+ await self.get_message_by_id(message.reply_to_id, include_thread_replies=False, db=db)
if message.reply_to_id
else None
)
- webhook_info = message.meta.get('webhook') if message.meta else None
- user_info = None
- if webhook_info and webhook_info.get('id'):
- webhook = Channels.get_webhook_by_id(webhook_info.get('id'), db=db)
- if webhook:
- user_info = {
- 'id': webhook.id,
- 'name': webhook.name,
- 'role': 'webhook',
- }
- else:
- user_info = {
- 'id': webhook_info.get('id'),
- 'name': 'Deleted Webhook',
- 'role': 'webhook',
- }
+ user_info = await self._resolve_user_info(message, db)
messages.append(
MessageReplyToResponse.model_validate(
@@ -390,34 +369,39 @@ class MessageTable:
)
return messages
- def get_last_message_by_channel_id(self, channel_id: str, db: Optional[Session] = None) -> Optional[MessageModel]:
- with get_db_context(db) as db:
- message = db.query(Message).filter_by(channel_id=channel_id).order_by(Message.created_at.desc()).first()
+ async def get_last_message_by_channel_id(
+ self, channel_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[MessageModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Message).filter_by(channel_id=channel_id).order_by(Message.created_at.desc()).limit(1)
+ )
+ message = result.scalars().first()
return MessageModel.model_validate(message) if message else None
- def get_pinned_messages_by_channel_id(
+ async def get_pinned_messages_by_channel_id(
self,
channel_id: str,
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[MessageModel]:
- with get_db_context(db) as db:
- all_messages = (
- db.query(Message)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Message)
.filter_by(channel_id=channel_id, is_pinned=True)
.order_by(Message.pinned_at.desc())
.offset(skip)
.limit(limit)
- .all()
)
+ all_messages = result.scalars().all()
return [MessageModel.model_validate(message) for message in all_messages]
- def update_message_by_id(
- self, id: str, form_data: MessageForm, db: Optional[Session] = None
+ async def update_message_by_id(
+ self, id: str, form_data: MessageForm, db: Optional[AsyncSession] = None
) -> Optional[MessageModel]:
- with get_db_context(db) as db:
- message = db.get(Message, id)
+ async with get_async_db_context(db) as db:
+ message = await db.get(Message, id)
message.content = form_data.content
message.data = {
**(message.data if message.data else {}),
@@ -428,49 +412,51 @@ class MessageTable:
**(form_data.meta if form_data.meta else {}),
}
message.updated_at = int(time.time_ns())
- db.commit()
- db.refresh(message)
+ await db.commit()
+ await db.refresh(message)
return MessageModel.model_validate(message) if message else None
- def update_is_pinned_by_id(
+ async def update_is_pinned_by_id(
self,
id: str,
is_pinned: bool,
pinned_by: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[MessageModel]:
- with get_db_context(db) as db:
- message = db.get(Message, id)
+ async with get_async_db_context(db) as db:
+ message = await db.get(Message, id)
message.is_pinned = is_pinned
message.pinned_at = int(time.time_ns()) if is_pinned else None
message.pinned_by = pinned_by if is_pinned else None
- db.commit()
- db.refresh(message)
+ await db.commit()
+ await db.refresh(message)
return MessageModel.model_validate(message) if message else None
- def get_unread_message_count(
+ async def get_unread_message_count(
self,
channel_id: str,
user_id: str,
last_read_at: Optional[int] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> int:
- with get_db_context(db) as db:
- query = db.query(Message).filter(
+ async with get_async_db_context(db) as db:
+ stmt = select(func.count(Message.id)).filter(
Message.channel_id == channel_id,
Message.parent_id == None, # only count top-level messages
Message.created_at > (last_read_at if last_read_at else 0),
)
if user_id:
- query = query.filter(Message.user_id != user_id)
- return query.count()
+ stmt = stmt.filter(Message.user_id != user_id)
+ result = await db.execute(stmt)
+ return result.scalar()
- def add_reaction_to_message(
- self, id: str, user_id: str, name: str, db: Optional[Session] = None
+ async def add_reaction_to_message(
+ self, id: str, user_id: str, name: str, db: Optional[AsyncSession] = None
) -> Optional[MessageReactionModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# check for existing reaction
- existing_reaction = db.query(MessageReaction).filter_by(message_id=id, user_id=user_id, name=name).first()
+ result = await db.execute(select(MessageReaction).filter_by(message_id=id, user_id=user_id, name=name))
+ existing_reaction = result.scalars().first()
if existing_reaction:
return MessageReactionModel.model_validate(existing_reaction)
@@ -484,19 +470,19 @@ class MessageTable:
)
result = MessageReaction(**reaction.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
return MessageReactionModel.model_validate(result) if result else None
- def get_reactions_by_message_id(self, id: str, db: Optional[Session] = None) -> list[Reactions]:
- with get_db_context(db) as db:
+ async def get_reactions_by_message_id(self, id: str, db: Optional[AsyncSession] = None) -> list[Reactions]:
+ async with get_async_db_context(db) as db:
# JOIN User so all user info is fetched in one query
- results = (
- db.query(MessageReaction, User)
+ result = await db.execute(
+ select(MessageReaction, User)
.join(User, MessageReaction.user_id == User.id)
.filter(MessageReaction.message_id == id)
- .all()
)
+ results = result.all()
reactions = {}
@@ -518,58 +504,60 @@ class MessageTable:
return [Reactions(**reaction) for reaction in reactions.values()]
- def remove_reaction_by_id_and_user_id_and_name(
- self, id: str, user_id: str, name: str, db: Optional[Session] = None
+ async def remove_reaction_by_id_and_user_id_and_name(
+ self, id: str, user_id: str, name: str, db: Optional[AsyncSession] = None
) -> bool:
- with get_db_context(db) as db:
- db.query(MessageReaction).filter_by(message_id=id, user_id=user_id, name=name).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(MessageReaction).filter_by(message_id=id, user_id=user_id, name=name))
+ await db.commit()
return True
- def delete_reactions_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- db.query(MessageReaction).filter_by(message_id=id).delete()
- db.commit()
+ async def delete_reactions_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(MessageReaction).filter_by(message_id=id))
+ await db.commit()
return True
- def delete_replies_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- db.query(Message).filter_by(parent_id=id).delete()
- db.commit()
+ async def delete_replies_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(Message).filter_by(parent_id=id))
+ await db.commit()
return True
- def delete_message_by_id(self, id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- db.query(Message).filter_by(id=id).delete()
+ async def delete_message_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(Message).filter_by(id=id))
# Delete all reactions to this message
- db.query(MessageReaction).filter_by(message_id=id).delete()
+ await db.execute(delete(MessageReaction).filter_by(message_id=id))
- db.commit()
+ await db.commit()
return True
- def search_messages_by_channel_ids(
+ async def search_messages_by_channel_ids(
self,
channel_ids: list[str],
query: str,
start_timestamp: Optional[int] = None,
end_timestamp: Optional[int] = None,
limit: int = 10,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[MessageModel]:
"""Search messages in specified channels by content."""
- with get_db_context(db) as db:
- query_builder = db.query(Message).filter(
+ async with get_async_db_context(db) as db:
+ stmt = select(Message).filter(
Message.channel_id.in_(channel_ids),
Message.content.ilike(f'%{query}%'),
)
if start_timestamp:
- query_builder = query_builder.filter(Message.created_at >= start_timestamp)
+ stmt = stmt.filter(Message.created_at >= start_timestamp)
if end_timestamp:
- query_builder = query_builder.filter(Message.created_at <= end_timestamp)
+ stmt = stmt.filter(Message.created_at <= end_timestamp)
- messages = query_builder.order_by(Message.created_at.desc()).limit(limit).all()
+ stmt = stmt.order_by(Message.created_at.desc()).limit(limit)
+ result = await db.execute(stmt)
+ messages = result.scalars().all()
return [MessageModel.model_validate(msg) for msg in messages]
diff --git a/backend/open_webui/models/models.py b/backend/open_webui/models/models.py
index 7cab2c830e..9bd3f888c1 100755
--- a/backend/open_webui/models/models.py
+++ b/backend/open_webui/models/models.py
@@ -1,9 +1,11 @@
+import json
import logging
import time
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update, or_, func, String, cast
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.groups import Groups
from open_webui.models.users import User, UserModel, Users, UserResponse
@@ -12,9 +14,6 @@ from open_webui.models.access_grants import AccessGrantModel, AccessGrants
from pydantic import BaseModel, ConfigDict, Field, model_validator
-from sqlalchemy import String, cast, or_, and_, func
-from sqlalchemy.dialects import postgresql, sqlite
-
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy import BigInteger, Column, Text, Boolean
@@ -154,26 +153,26 @@ class ModelForm(BaseModel):
class ModelsTable:
- def _get_access_grants(self, model_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('model', model_id, db=db)
+ async def _get_access_grants(self, model_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('model', model_id, db=db)
- def _to_model_model(
+ async def _to_model_model(
self,
model: Model,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> ModelModel:
model_data = ModelModel.model_validate(model).model_dump(exclude={'access_grants'})
model_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(model_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(model_data['id'], db=db)
)
return ModelModel.model_validate(model_data)
- def insert_new_model(
- self, form_data: ModelForm, user_id: str, db: Optional[Session] = None
+ async def insert_new_model(
+ self, form_data: ModelForm, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[ModelModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
result = Model(
**{
**form_data.model_dump(exclude={'access_grants'}),
@@ -183,37 +182,40 @@ class ModelsTable:
}
)
db.add(result)
- db.commit()
- db.refresh(result)
- AccessGrants.set_access_grants('model', result.id, form_data.access_grants, db=db)
+ await db.commit()
+ await db.refresh(result)
+ await AccessGrants.set_access_grants('model', result.id, form_data.access_grants, db=db)
if result:
- return self._to_model_model(result, db=db)
+ return await self._to_model_model(result, db=db)
else:
return None
except Exception as e:
log.exception(f'Failed to insert a new model: {e}')
return None
- def get_all_models(self, db: Optional[Session] = None) -> list[ModelModel]:
- with get_db_context(db) as db:
- all_models = db.query(Model).all()
+ async def get_all_models(self, db: Optional[AsyncSession] = None) -> list[ModelModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Model))
+ all_models = result.scalars().all()
model_ids = [model.id for model in all_models]
- grants_map = AccessGrants.get_grants_by_resources('model', model_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('model', model_ids, db=db)
return [
- self._to_model_model(model, access_grants=grants_map.get(model.id, []), db=db) for model in all_models
+ await self._to_model_model(model, access_grants=grants_map.get(model.id, []), db=db)
+ for model in all_models
]
- def get_models(self, db: Optional[Session] = None) -> list[ModelUserResponse]:
- with get_db_context(db) as db:
- all_models = db.query(Model).filter(Model.base_model_id != None).all()
+ async def get_models(self, db: Optional[AsyncSession] = None) -> list[ModelUserResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Model).filter(Model.base_model_id != None))
+ all_models = result.scalars().all()
user_ids = list(set(model.user_id for model in all_models))
model_ids = [model.id for model in all_models]
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
- grants_map = AccessGrants.get_grants_by_resources('model', model_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('model', model_ids, db=db)
models = []
for model in all_models:
@@ -221,10 +223,12 @@ class ModelsTable:
models.append(
ModelUserResponse.model_validate(
{
- **self._to_model_model(
- model,
- access_grants=grants_map.get(model.id, []),
- db=db,
+ **(
+ await self._to_model_model(
+ model,
+ access_grants=grants_map.get(model.id, []),
+ db=db,
+ )
).model_dump(),
'user': user.model_dump() if user else None,
}
@@ -232,33 +236,38 @@ class ModelsTable:
)
return models
- def get_base_models(self, db: Optional[Session] = None) -> list[ModelModel]:
- with get_db_context(db) as db:
- all_models = db.query(Model).filter(Model.base_model_id == None).all()
+ async def get_base_models(self, db: Optional[AsyncSession] = None) -> list[ModelModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Model).filter(Model.base_model_id == None))
+ all_models = result.scalars().all()
model_ids = [model.id for model in all_models]
- grants_map = AccessGrants.get_grants_by_resources('model', model_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('model', model_ids, db=db)
return [
- self._to_model_model(model, access_grants=grants_map.get(model.id, []), db=db) for model in all_models
+ await self._to_model_model(model, access_grants=grants_map.get(model.id, []), db=db)
+ for model in all_models
]
- def get_models_by_user_id(
- self, user_id: str, permission: str = 'write', db: Optional[Session] = None
+ async def get_models_by_user_id(
+ self, user_id: str, permission: str = 'write', db: Optional[AsyncSession] = None
) -> list[ModelUserResponse]:
- models = self.get_models(db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
- return [
- model
- for model in models
- if model.user_id == user_id
- or AccessGrants.has_access(
+ models = await self.get_models(db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
+
+ result = []
+ for model in models:
+ if model.user_id == user_id:
+ result.append(model)
+ elif await AccessGrants.has_access(
user_id=user_id,
resource_type='model',
resource_id=model.id,
permission=permission,
user_group_ids=user_group_ids,
db=db,
- )
- ]
+ ):
+ result.append(model)
+ return result
def _has_permission(self, db, query, filter: dict, permission: str = 'read'):
return AccessGrants.has_permission_filter(
@@ -270,23 +279,22 @@ class ModelsTable:
permission=permission,
)
- def search_models(
+ async def search_models(
self,
user_id: str,
filter: dict = {},
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> ModelListResponse:
- with get_db_context(db) as db:
- # Join GroupMember so we can order by group_id when requested
- query = db.query(Model, User).outerjoin(User, User.id == Model.user_id)
- query = query.filter(Model.base_model_id != None)
+ async with get_async_db_context(db) as db:
+ stmt = select(Model, User).outerjoin(User, User.id == Model.user_id)
+ stmt = stmt.filter(Model.base_model_id != None)
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(
+ stmt = stmt.filter(
or_(
Model.name.ilike(f'%{query_key}%'),
Model.base_model_id.ilike(f'%{query_key}%'),
@@ -298,69 +306,82 @@ class ModelsTable:
view_option = filter.get('view_option')
if view_option == 'created':
- query = query.filter(Model.user_id == user_id)
+ stmt = stmt.filter(Model.user_id == user_id)
elif view_option == 'shared':
- query = query.filter(Model.user_id != user_id)
+ stmt = stmt.filter(Model.user_id != user_id)
# Apply access control filtering
- query = self._has_permission(
+ stmt = self._has_permission(
db,
- query,
+ stmt,
filter,
permission='read',
)
tag = filter.get('tag')
if tag:
- # TODO: This is a simple implementation and should be improved for performance
- like_pattern = f'%"{tag.lower()}"%' # `"tag"` inside JSON array
- meta_text = func.lower(cast(Model.meta, String))
-
- query = query.filter(meta_text.like(like_pattern))
+ # SQLite stores JSON text via json.dumps(ensure_ascii=True),
+ # so non-ASCII chars are \uXXXX-escaped. PostgreSQL native JSONB
+ # stores literal Unicode. Use the right pattern for each.
+ if db.bind.dialect.name == 'sqlite':
+ if tag.isascii():
+ meta_text = func.lower(cast(Model.meta, String))
+ pattern = f'%{json.dumps(tag.lower())}%'
+ else:
+ meta_text = cast(Model.meta, String)
+ pattern = f'%{json.dumps(tag)}%'
+ else:
+ meta_text = func.lower(cast(Model.meta, String))
+ pattern = f'%{json.dumps(tag.lower(), ensure_ascii=False)}%'
+ stmt = stmt.filter(meta_text.like(pattern))
order_by = filter.get('order_by')
direction = filter.get('direction')
if order_by == 'name':
if direction == 'asc':
- query = query.order_by(Model.name.asc())
+ stmt = stmt.order_by(Model.name.asc())
else:
- query = query.order_by(Model.name.desc())
+ stmt = stmt.order_by(Model.name.desc())
elif order_by == 'created_at':
if direction == 'asc':
- query = query.order_by(Model.created_at.asc())
+ stmt = stmt.order_by(Model.created_at.asc())
else:
- query = query.order_by(Model.created_at.desc())
+ stmt = stmt.order_by(Model.created_at.desc())
elif order_by == 'updated_at':
if direction == 'asc':
- query = query.order_by(Model.updated_at.asc())
+ stmt = stmt.order_by(Model.updated_at.asc())
else:
- query = query.order_by(Model.updated_at.desc())
+ stmt = stmt.order_by(Model.updated_at.desc())
else:
- query = query.order_by(Model.created_at.desc())
+ stmt = stmt.order_by(Model.created_at.desc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
model_ids = [model.id for model, _ in items]
- grants_map = AccessGrants.get_grants_by_resources('model', model_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('model', model_ids, db=db)
models = []
for model, user in items:
models.append(
ModelUserResponse(
- **self._to_model_model(
- model,
- access_grants=grants_map.get(model.id, []),
- db=db,
+ **(
+ await self._to_model_model(
+ model,
+ access_grants=grants_map.get(model.id, []),
+ db=db,
+ )
).model_dump(),
user=(UserResponse(**UserModel.model_validate(user).model_dump()) if user else None),
)
@@ -368,22 +389,23 @@ class ModelsTable:
return ModelListResponse(items=models, total=total)
- def get_model_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ModelModel]:
+ async def get_model_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ModelModel]:
try:
- with get_db_context(db) as db:
- model = db.get(Model, id)
- return self._to_model_model(model, db=db) if model else None
+ async with get_async_db_context(db) as db:
+ model = await db.get(Model, id)
+ return await self._to_model_model(model, db=db) if model else None
except Exception:
return None
- def get_models_by_ids(self, ids: list[str], db: Optional[Session] = None) -> list[ModelModel]:
+ async def get_models_by_ids(self, ids: list[str], db: Optional[AsyncSession] = None) -> list[ModelModel]:
try:
- with get_db_context(db) as db:
- models = db.query(Model).filter(Model.id.in_(ids)).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Model).filter(Model.id.in_(ids)))
+ models = result.scalars().all()
model_ids = [model.id for model in models]
- grants_map = AccessGrants.get_grants_by_resources('model', model_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('model', model_ids, db=db)
return [
- self._to_model_model(
+ await self._to_model_model(
model,
access_grants=grants_map.get(model.id, []),
db=db,
@@ -393,82 +415,90 @@ class ModelsTable:
except Exception:
return []
- def toggle_model_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ModelModel]:
- with get_db_context(db) as db:
+ async def toggle_model_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ModelModel]:
+ async with get_async_db_context(db) as db:
try:
- model = db.query(Model).filter_by(id=id).first()
+ result = await db.execute(select(Model).filter_by(id=id))
+ model = result.scalars().first()
if not model:
return None
model.is_active = not model.is_active
model.updated_at = int(time.time())
- db.commit()
- db.refresh(model)
+ await db.commit()
+ await db.refresh(model)
- return self._to_model_model(model, db=db)
+ return await self._to_model_model(model, db=db)
except Exception:
return None
- def update_model_by_id(self, id: str, model: ModelForm, db: Optional[Session] = None) -> Optional[ModelModel]:
+ async def update_model_by_id(
+ self, id: str, model: ModelForm, db: Optional[AsyncSession] = None
+ ) -> Optional[ModelModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# update only the fields that are present in the model
data = model.model_dump(exclude={'id', 'access_grants'})
data['updated_at'] = int(time.time())
- result = db.query(Model).filter_by(id=id).update(data)
+ await db.execute(update(Model).filter_by(id=id).values(**data))
- db.commit()
+ await db.commit()
if model.access_grants is not None:
- AccessGrants.set_access_grants('model', id, model.access_grants, db=db)
+ await AccessGrants.set_access_grants('model', id, model.access_grants, db=db)
- return self.get_model_by_id(id, db=db)
+ return await self.get_model_by_id(id, db=db)
except Exception as e:
log.exception(f'Failed to update the model by id {id}: {e}')
return None
- def update_model_updated_at_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ModelModel]:
+ async def update_model_updated_at_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ModelModel]:
try:
- with get_db_context(db) as db:
- result = db.query(Model).filter_by(id=id).first()
- if not result:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Model).filter_by(id=id))
+ model_obj = result.scalars().first()
+ if not model_obj:
return None
- result.updated_at = int(time.time())
- db.commit()
- db.refresh(result)
- return self._to_model_model(result, db=db)
+ model_obj.updated_at = int(time.time())
+ await db.commit()
+ await db.refresh(model_obj)
+ return await self._to_model_model(model_obj, db=db)
except Exception as e:
log.exception(f'Failed to update the model updated_at by id {id}: {e}')
return None
- def delete_model_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_model_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- AccessGrants.revoke_all_access('model', id, db=db)
- db.query(Model).filter_by(id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await AccessGrants.revoke_all_access('model', id, db=db)
+ await db.execute(delete(Model).filter_by(id=id))
+ await db.commit()
return True
except Exception:
return False
- def delete_all_models(self, db: Optional[Session] = None) -> bool:
+ async def delete_all_models(self, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- model_ids = [row[0] for row in db.query(Model.id).all()]
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Model.id))
+ model_ids = [row[0] for row in result.all()]
for model_id in model_ids:
- AccessGrants.revoke_all_access('model', model_id, db=db)
- db.query(Model).delete()
- db.commit()
+ await AccessGrants.revoke_all_access('model', model_id, db=db)
+ await db.execute(delete(Model))
+ await db.commit()
return True
except Exception:
return False
- def sync_models(self, user_id: str, models: list[ModelModel], db: Optional[Session] = None) -> list[ModelModel]:
+ async def sync_models(
+ self, user_id: str, models: list[ModelModel], db: Optional[AsyncSession] = None
+ ) -> list[ModelModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Get existing models
- existing_models = db.query(Model).all()
+ result = await db.execute(select(Model))
+ existing_models = result.scalars().all()
existing_ids = {model.id for model in existing_models}
# Prepare a set of new model IDs
@@ -477,12 +507,14 @@ class ModelsTable:
# Update or insert models
for model in models:
if model.id in existing_ids:
- db.query(Model).filter_by(id=model.id).update(
- {
+ await db.execute(
+ update(Model)
+ .filter_by(id=model.id)
+ .values(
**model.model_dump(exclude={'access_grants'}),
- 'user_id': user_id,
- 'updated_at': int(time.time()),
- }
+ user_id=user_id,
+ updated_at=int(time.time()),
+ )
)
else:
new_model = Model(
@@ -493,21 +525,22 @@ class ModelsTable:
}
)
db.add(new_model)
- AccessGrants.set_access_grants('model', model.id, model.access_grants, db=db)
+ await AccessGrants.set_access_grants('model', model.id, model.access_grants, db=db)
# Remove models that are no longer present
for model in existing_models:
if model.id not in new_model_ids:
- AccessGrants.revoke_all_access('model', model.id, db=db)
- db.delete(model)
+ await AccessGrants.revoke_all_access('model', model.id, db=db)
+ await db.delete(model)
- db.commit()
+ await db.commit()
- all_models = db.query(Model).all()
+ result = await db.execute(select(Model))
+ all_models = result.scalars().all()
model_ids = [model.id for model in all_models]
- grants_map = AccessGrants.get_grants_by_resources('model', model_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('model', model_ids, db=db)
return [
- self._to_model_model(
+ await self._to_model_model(
model,
access_grants=grants_map.get(model.id, []),
db=db,
diff --git a/backend/open_webui/models/notes.py b/backend/open_webui/models/notes.py
index 34749f5f6c..1a34750a7d 100644
--- a/backend/open_webui/models/notes.py
+++ b/backend/open_webui/models/notes.py
@@ -4,8 +4,9 @@ import uuid
from typing import Optional
from functools import lru_cache
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db, get_db_context
+from sqlalchemy import Boolean, select, delete, update, or_, func, cast
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
from open_webui.models.groups import Groups
from open_webui.models.users import User, UserModel, Users, UserResponse
from open_webui.models.access_grants import AccessGrantModel, AccessGrants
@@ -13,7 +14,6 @@ from open_webui.models.access_grants import AccessGrantModel, AccessGrants
from pydantic import BaseModel, ConfigDict, Field
from sqlalchemy import BigInteger, Column, Text, JSON
-from sqlalchemy import or_, func, cast
####################
# Note DB Schema
@@ -29,6 +29,7 @@ class Note(Base):
title = Column(Text)
data = Column(JSON, nullable=True)
meta = Column(JSON, nullable=True)
+ is_pinned = Column(Boolean, default=False, nullable=True)
created_at = Column(BigInteger)
updated_at = Column(BigInteger)
@@ -43,6 +44,7 @@ class NoteModel(BaseModel):
title: str
data: Optional[dict] = None
meta: Optional[dict] = None
+ is_pinned: Optional[bool] = False
access_grants: list[AccessGrantModel] = Field(default_factory=list)
@@ -77,6 +79,7 @@ class NoteItemResponse(BaseModel):
id: str
title: str
data: Optional[dict]
+ is_pinned: Optional[bool] = False
updated_at: int
created_at: int
user: Optional[UserResponse] = None
@@ -88,18 +91,18 @@ class NoteListResponse(BaseModel):
class NoteTable:
- def _get_access_grants(self, note_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('note', note_id, db=db)
+ async def _get_access_grants(self, note_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('note', note_id, db=db)
- def _to_note_model(
+ async def _to_note_model(
self,
note: Note,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> NoteModel:
note_data = NoteModel.model_validate(note).model_dump(exclude={'access_grants'})
note_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(note_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(note_data['id'], db=db)
)
return NoteModel.model_validate(note_data)
@@ -113,8 +116,10 @@ class NoteTable:
permission=permission,
)
- def insert_new_note(self, user_id: str, form_data: NoteForm, db: Optional[Session] = None) -> Optional[NoteModel]:
- with get_db_context(db) as db:
+ async def insert_new_note(
+ self, user_id: str, form_data: NoteForm, db: Optional[AsyncSession] = None
+ ) -> Optional[NoteModel]:
+ async with get_async_db_context(db) as db:
note = NoteModel(
**{
'id': str(uuid.uuid4()),
@@ -129,53 +134,58 @@ class NoteTable:
new_note = Note(**note.model_dump(exclude={'access_grants'}))
db.add(new_note)
- db.commit()
- AccessGrants.set_access_grants('note', note.id, form_data.access_grants, db=db)
- return self._to_note_model(new_note, db=db)
+ await db.commit()
+ await AccessGrants.set_access_grants('note', note.id, form_data.access_grants, db=db)
+ return await self._to_note_model(new_note, db=db)
- def get_notes(self, skip: int = 0, limit: int = 50, db: Optional[Session] = None) -> list[NoteModel]:
- with get_db_context(db) as db:
- query = db.query(Note).order_by(Note.updated_at.desc())
+ async def get_notes(self, skip: int = 0, limit: int = 50, db: Optional[AsyncSession] = None) -> list[NoteModel]:
+ async with get_async_db_context(db) as db:
+ stmt = select(Note).order_by(Note.updated_at.desc())
if skip is not None:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit is not None:
- query = query.limit(limit)
- notes = query.all()
+ stmt = stmt.limit(limit)
+ result = await db.execute(stmt)
+ notes = result.scalars().all()
note_ids = [note.id for note in notes]
- grants_map = AccessGrants.get_grants_by_resources('note', note_ids, db=db)
- return [self._to_note_model(note, access_grants=grants_map.get(note.id, []), db=db) for note in notes]
+ grants_map = await AccessGrants.get_grants_by_resources('note', note_ids, db=db)
+ return [await self._to_note_model(note, access_grants=grants_map.get(note.id, []), db=db) for note in notes]
- def search_notes(
+ async def search_notes(
self,
user_id: str,
filter: dict = {},
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> NoteListResponse:
- with get_db_context(db) as db:
- query = db.query(Note, User).outerjoin(User, User.id == Note.user_id)
+ async with get_async_db_context(db) as db:
+ stmt = select(Note, User).outerjoin(User, User.id == Note.user_id)
if filter:
query_key = filter.get('query')
if query_key:
- # Normalize search by removing hyphens and spaces (e.g., "todo" matches "to-do" and "to do")
- normalized_query = query_key.replace('-', '').replace(' ', '')
- query = query.filter(
- or_(
- func.replace(func.replace(Note.title, '-', ''), ' ', '').ilike(f'%{normalized_query}%'),
- func.replace(
- func.replace(cast(Note.data['content']['md'], Text), '-', ''),
- ' ',
- '',
- ).ilike(f'%{normalized_query}%'),
+ # Split query into individual words and normalize each
+ # (strip hyphens so "todo" matches "to-do").
+ # All words must match somewhere in title OR content (AND semantics).
+ search_words = query_key.split()
+ normalized_words = [w.replace('-', '') for w in search_words if w.replace('-', '')]
+ for word in normalized_words:
+ stmt = stmt.filter(
+ or_(
+ func.replace(func.replace(Note.title, '-', ''), ' ', '').ilike(f'%{word}%'),
+ func.replace(
+ func.replace(cast(Note.data['content']['md'], Text), '-', ''),
+ ' ',
+ '',
+ ).ilike(f'%{word}%'),
+ )
)
- )
view_option = filter.get('view_option')
if view_option == 'created':
- query = query.filter(Note.user_id == user_id)
+ stmt = stmt.filter(Note.user_id == user_id)
elif view_option == 'shared':
- query = query.filter(Note.user_id != user_id)
+ stmt = stmt.filter(Note.user_id != user_id)
# Apply access control filtering
if 'permission' in filter:
@@ -183,9 +193,9 @@ class NoteTable:
else:
permission = 'write'
- query = self._has_permission(
+ stmt = self._has_permission(
db,
- query,
+ stmt,
filter,
permission=permission,
)
@@ -195,46 +205,50 @@ class NoteTable:
if order_by == 'name':
if direction == 'asc':
- query = query.order_by(Note.title.asc())
+ stmt = stmt.order_by(Note.title.asc())
else:
- query = query.order_by(Note.title.desc())
+ stmt = stmt.order_by(Note.title.desc())
elif order_by == 'created_at':
if direction == 'asc':
- query = query.order_by(Note.created_at.asc())
+ stmt = stmt.order_by(Note.created_at.asc())
else:
- query = query.order_by(Note.created_at.desc())
+ stmt = stmt.order_by(Note.created_at.desc())
elif order_by == 'updated_at':
if direction == 'asc':
- query = query.order_by(Note.updated_at.asc())
+ stmt = stmt.order_by(Note.updated_at.asc())
else:
- query = query.order_by(Note.updated_at.desc())
+ stmt = stmt.order_by(Note.updated_at.desc())
else:
- query = query.order_by(Note.updated_at.desc())
+ stmt = stmt.order_by(Note.updated_at.desc())
else:
- query = query.order_by(Note.updated_at.desc())
+ stmt = stmt.order_by(Note.updated_at.desc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
note_ids = [note.id for note, _ in items]
- grants_map = AccessGrants.get_grants_by_resources('note', note_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('note', note_ids, db=db)
notes = []
for note, user in items:
notes.append(
NoteUserResponse(
- **self._to_note_model(
- note,
- access_grants=grants_map.get(note.id, []),
- db=db,
+ **(
+ await self._to_note_model(
+ note,
+ access_grants=grants_map.get(note.id, []),
+ db=db,
+ )
).model_dump(),
user=(UserResponse(**UserModel.model_validate(user).model_dump()) if user else None),
)
@@ -242,40 +256,44 @@ class NoteTable:
return NoteListResponse(items=notes, total=total)
- def get_notes_by_user_id(
+ async def get_notes_by_user_id(
self,
user_id: str,
permission: str = 'read',
skip: int = 0,
limit: int = 50,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[NoteModel]:
- with get_db_context(db) as db:
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id, db=db)]
+ async with get_async_db_context(db) as db:
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = [group.id for group in user_groups]
- query = db.query(Note).order_by(Note.updated_at.desc())
- query = self._has_permission(db, query, {'user_id': user_id, 'group_ids': user_group_ids}, permission)
+ stmt = select(Note).order_by(Note.updated_at.desc())
+ stmt = self._has_permission(db, stmt, {'user_id': user_id, 'group_ids': user_group_ids}, permission)
if skip is not None:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit is not None:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- notes = query.all()
+ result = await db.execute(stmt)
+ notes = result.scalars().all()
note_ids = [note.id for note in notes]
- grants_map = AccessGrants.get_grants_by_resources('note', note_ids, db=db)
- return [self._to_note_model(note, access_grants=grants_map.get(note.id, []), db=db) for note in notes]
+ grants_map = await AccessGrants.get_grants_by_resources('note', note_ids, db=db)
+ return [await self._to_note_model(note, access_grants=grants_map.get(note.id, []), db=db) for note in notes]
- def get_note_by_id(self, id: str, db: Optional[Session] = None) -> Optional[NoteModel]:
- with get_db_context(db) as db:
- note = db.query(Note).filter(Note.id == id).first()
- return self._to_note_model(note, db=db) if note else None
+ async def get_note_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[NoteModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Note).filter(Note.id == id))
+ note = result.scalars().first()
+ return await self._to_note_model(note, db=db) if note else None
- def update_note_by_id(
- self, id: str, form_data: NoteUpdateForm, db: Optional[Session] = None
+ async def update_note_by_id(
+ self, id: str, form_data: NoteUpdateForm, db: Optional[AsyncSession] = None
) -> Optional[NoteModel]:
- with get_db_context(db) as db:
- note = db.query(Note).filter(Note.id == id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Note).filter(Note.id == id))
+ note = result.scalars().first()
if not note:
return None
@@ -289,19 +307,52 @@ class NoteTable:
note.meta = {**note.meta, **form_data['meta']}
if 'access_grants' in form_data:
- AccessGrants.set_access_grants('note', id, form_data['access_grants'], db=db)
+ await AccessGrants.set_access_grants('note', id, form_data['access_grants'], db=db)
note.updated_at = int(time.time_ns())
- db.commit()
- return self._to_note_model(note, db=db) if note else None
+ await db.commit()
+ return await self._to_note_model(note, db=db) if note else None
- def delete_note_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def toggle_note_pinned_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[NoteModel]:
try:
- with get_db_context(db) as db:
- AccessGrants.revoke_all_access('note', id, db=db)
- db.query(Note).filter(Note.id == id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Note).filter(Note.id == id))
+ note = result.scalars().first()
+ if not note:
+ return None
+ note.is_pinned = not note.is_pinned
+ note.updated_at = int(time.time_ns())
+ await db.commit()
+ return await self._to_note_model(note, db=db)
+ except Exception:
+ return None
+
+ async def get_pinned_notes_by_user_id(
+ self,
+ user_id: str,
+ permission: str = 'read',
+ db: Optional[AsyncSession] = None,
+ ) -> list[NoteModel]:
+ async with get_async_db_context(db) as db:
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = [group.id for group in user_groups]
+
+ stmt = select(Note).filter(Note.is_pinned == True).order_by(Note.updated_at.desc())
+ stmt = self._has_permission(db, stmt, {'user_id': user_id, 'group_ids': user_group_ids}, permission)
+
+ result = await db.execute(stmt)
+ notes = result.scalars().all()
+ note_ids = [note.id for note in notes]
+ grants_map = await AccessGrants.get_grants_by_resources('note', note_ids, db=db)
+ return [await self._to_note_model(note, access_grants=grants_map.get(note.id, []), db=db) for note in notes]
+
+ async def delete_note_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
+ try:
+ async with get_async_db_context(db) as db:
+ await AccessGrants.revoke_all_access('note', id, db=db)
+ await db.execute(delete(Note).filter(Note.id == id))
+ await db.commit()
return True
except Exception:
return False
diff --git a/backend/open_webui/models/oauth_sessions.py b/backend/open_webui/models/oauth_sessions.py
index 868216164a..050a50d486 100644
--- a/backend/open_webui/models/oauth_sessions.py
+++ b/backend/open_webui/models/oauth_sessions.py
@@ -8,8 +8,9 @@ import json
from cryptography.fernet import Fernet
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db, get_db_context
+from sqlalchemy import select, delete, update
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
from open_webui.env import OAUTH_SESSION_TOKEN_ENCRYPTION_KEY
from pydantic import BaseModel, ConfigDict
@@ -103,16 +104,16 @@ class OAuthSessionTable:
log.error(f'Error decrypting tokens: {type(e).__name__}: {e}')
raise
- def create_session(
+ async def create_session(
self,
user_id: str,
provider: str,
token: dict,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[OAuthSessionModel]:
"""Create a new OAuth session"""
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
current_time = int(time.time())
id = str(uuid.uuid4())
@@ -122,98 +123,137 @@ class OAuthSessionTable:
'user_id': user_id,
'provider': provider,
'token': self._encrypt_token(token),
- 'expires_at': token.get('expires_at'),
+ 'expires_at': token.get('expires_at') or int(time.time() + 3600),
'created_at': current_time,
'updated_at': current_time,
}
)
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
- db.expunge(result) # Detach so dict swap is never flushed
- result.token = token # Return decrypted token
- return OAuthSessionModel.model_validate(result)
+ # Make a copy of the model data before closing session
+ model = OAuthSessionModel(
+ id=result.id,
+ user_id=result.user_id,
+ provider=result.provider,
+ token=token, # Return decrypted token
+ expires_at=result.expires_at,
+ created_at=result.created_at,
+ updated_at=result.updated_at,
+ )
+ return model
else:
return None
except Exception as e:
log.error(f'Error creating OAuth session: {e}')
return None
- def get_session_by_id(self, session_id: str, db: Optional[Session] = None) -> Optional[OAuthSessionModel]:
+ async def get_session_by_id(
+ self, session_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[OAuthSessionModel]:
"""Get OAuth session by ID"""
try:
- with get_db_context(db) as db:
- session = db.query(OAuthSession).filter_by(id=session_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(OAuthSession).filter_by(id=session_id))
+ session = result.scalars().first()
if session:
- db.expunge(session)
- session.token = self._decrypt_token(session.token)
- return OAuthSessionModel.model_validate(session)
+ return OAuthSessionModel(
+ id=session.id,
+ user_id=session.user_id,
+ provider=session.provider,
+ token=self._decrypt_token(session.token),
+ expires_at=session.expires_at,
+ created_at=session.created_at,
+ updated_at=session.updated_at,
+ )
return None
except Exception as e:
log.error(f'Error getting OAuth session by ID: {e}')
return None
- def get_session_by_id_and_user_id(
- self, session_id: str, user_id: str, db: Optional[Session] = None
+ async def get_session_by_id_and_user_id(
+ self, session_id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[OAuthSessionModel]:
"""Get OAuth session by ID and user ID"""
try:
- with get_db_context(db) as db:
- session = db.query(OAuthSession).filter_by(id=session_id, user_id=user_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(OAuthSession).filter_by(id=session_id, user_id=user_id))
+ session = result.scalars().first()
if session:
- db.expunge(session)
- session.token = self._decrypt_token(session.token)
- return OAuthSessionModel.model_validate(session)
+ return OAuthSessionModel(
+ id=session.id,
+ user_id=session.user_id,
+ provider=session.provider,
+ token=self._decrypt_token(session.token),
+ expires_at=session.expires_at,
+ created_at=session.created_at,
+ updated_at=session.updated_at,
+ )
return None
except Exception as e:
log.error(f'Error getting OAuth session by ID: {e}')
return None
- def get_session_by_provider_and_user_id(
- self, provider: str, user_id: str, db: Optional[Session] = None
+ async def get_session_by_provider_and_user_id(
+ self, provider: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[OAuthSessionModel]:
"""Get OAuth session by provider and user ID"""
try:
- with get_db_context(db) as db:
- session = (
- db.query(OAuthSession)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(OAuthSession)
.filter_by(provider=provider, user_id=user_id)
.order_by(OAuthSession.created_at.desc())
- .first()
)
+ session = result.scalars().first()
if session:
- db.expunge(session)
- session.token = self._decrypt_token(session.token)
- return OAuthSessionModel.model_validate(session)
+ return OAuthSessionModel(
+ id=session.id,
+ user_id=session.user_id,
+ provider=session.provider,
+ token=self._decrypt_token(session.token),
+ expires_at=session.expires_at,
+ created_at=session.created_at,
+ updated_at=session.updated_at,
+ )
return None
except Exception as e:
log.error(f'Error getting OAuth session by provider and user ID: {e}')
return None
- def get_sessions_by_user_id(self, user_id: str, db: Optional[Session] = None) -> List[OAuthSessionModel]:
+ async def get_sessions_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> List[OAuthSessionModel]:
"""Get all OAuth sessions for a user"""
try:
- with get_db_context(db) as db:
- sessions = db.query(OAuthSession).filter_by(user_id=user_id).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(OAuthSession).filter_by(user_id=user_id))
+ sessions = result.scalars().all()
results = []
for session in sessions:
try:
- db.expunge(session)
- session.token = self._decrypt_token(session.token)
- results.append(OAuthSessionModel.model_validate(session))
+ results.append(
+ OAuthSessionModel(
+ id=session.id,
+ user_id=session.user_id,
+ provider=session.provider,
+ token=self._decrypt_token(session.token),
+ expires_at=session.expires_at,
+ created_at=session.created_at,
+ updated_at=session.updated_at,
+ )
+ )
except Exception as e:
log.warning(
f'Skipping OAuth session {session.id} due to decryption failure, deleting corrupted session: {type(e).__name__}: {e}'
)
- db.query(OAuthSession).filter_by(id=session.id).delete()
- db.commit()
+ await db.execute(delete(OAuthSession).filter_by(id=session.id))
+ await db.commit()
return results
@@ -221,62 +261,71 @@ class OAuthSessionTable:
log.error(f'Error getting OAuth sessions by user ID: {e}')
return []
- def update_session_by_id(
- self, session_id: str, token: dict, db: Optional[Session] = None
+ async def update_session_by_id(
+ self, session_id: str, token: dict, db: Optional[AsyncSession] = None
) -> Optional[OAuthSessionModel]:
"""Update OAuth session tokens"""
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
current_time = int(time.time())
- db.query(OAuthSession).filter_by(id=session_id).update(
- {
- 'token': self._encrypt_token(token),
- 'expires_at': token.get('expires_at'),
- 'updated_at': current_time,
- }
+ await db.execute(
+ update(OAuthSession)
+ .filter_by(id=session_id)
+ .values(
+ token=self._encrypt_token(token),
+ expires_at=token.get('expires_at') or int(time.time() + 3600),
+ updated_at=current_time,
+ )
)
- db.commit()
- session = db.query(OAuthSession).filter_by(id=session_id).first()
+ await db.commit()
+ result = await db.execute(select(OAuthSession).filter_by(id=session_id))
+ session = result.scalars().first()
if session:
- db.expunge(session)
- session.token = self._decrypt_token(session.token)
- return OAuthSessionModel.model_validate(session)
+ return OAuthSessionModel(
+ id=session.id,
+ user_id=session.user_id,
+ provider=session.provider,
+ token=self._decrypt_token(session.token),
+ expires_at=session.expires_at,
+ created_at=session.created_at,
+ updated_at=session.updated_at,
+ )
return None
except Exception as e:
log.error(f'Error updating OAuth session tokens: {e}')
return None
- def delete_session_by_id(self, session_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_session_by_id(self, session_id: str, db: Optional[AsyncSession] = None) -> bool:
"""Delete an OAuth session"""
try:
- with get_db_context(db) as db:
- result = db.query(OAuthSession).filter_by(id=session_id).delete()
- db.commit()
- return result > 0
+ async with get_async_db_context(db) as db:
+ result = await db.execute(delete(OAuthSession).filter_by(id=session_id))
+ await db.commit()
+ return result.rowcount > 0
except Exception as e:
log.error(f'Error deleting OAuth session: {e}')
return False
- def delete_sessions_by_user_id(self, user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_sessions_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
"""Delete all OAuth sessions for a user"""
try:
- with get_db_context(db) as db:
- result = db.query(OAuthSession).filter_by(user_id=user_id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(OAuthSession).filter_by(user_id=user_id))
+ await db.commit()
return True
except Exception as e:
log.error(f'Error deleting OAuth sessions by user ID: {e}')
return False
- def delete_sessions_by_provider(self, provider: str, db: Optional[Session] = None) -> bool:
+ async def delete_sessions_by_provider(self, provider: str, db: Optional[AsyncSession] = None) -> bool:
"""Delete all OAuth sessions for a provider"""
try:
- with get_db_context(db) as db:
- db.query(OAuthSession).filter_by(provider=provider).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(OAuthSession).filter_by(provider=provider))
+ await db.commit()
return True
except Exception as e:
log.error(f'Error deleting OAuth sessions by provider {provider}: {e}')
diff --git a/backend/open_webui/models/prompt_history.py b/backend/open_webui/models/prompt_history.py
index d42b4bfa24..5d0f4a65b2 100644
--- a/backend/open_webui/models/prompt_history.py
+++ b/backend/open_webui/models/prompt_history.py
@@ -6,8 +6,9 @@ from typing import Optional
import json
import difflib
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db_context
+from sqlalchemy import select, delete, func
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
from open_webui.models.users import Users, UserResponse
from pydantic import BaseModel, ConfigDict
@@ -49,17 +50,17 @@ class PromptHistoryResponse(PromptHistoryModel):
class PromptHistoryTable:
- def create_history_entry(
+ async def create_history_entry(
self,
prompt_id: str,
snapshot: dict,
user_id: str,
parent_id: Optional[str] = None,
commit_message: Optional[str] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptHistoryModel]:
"""Create a new history entry (commit) for a prompt."""
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
history = PromptHistory(
id=str(uuid.uuid4()),
prompt_id=prompt_id,
@@ -70,31 +71,31 @@ class PromptHistoryTable:
created_at=int(time.time()),
)
db.add(history)
- db.commit()
- db.refresh(history)
+ await db.commit()
+ await db.refresh(history)
return PromptHistoryModel.model_validate(history)
- def get_history_by_prompt_id(
+ async def get_history_by_prompt_id(
self,
prompt_id: str,
limit: int = 50,
offset: int = 0,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[PromptHistoryResponse]:
"""Get all history entries for a prompt, ordered by created_at desc."""
- with get_db_context(db) as db:
- entries = (
- db.query(PromptHistory)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(PromptHistory)
.filter(PromptHistory.prompt_id == prompt_id)
.order_by(PromptHistory.created_at.desc())
.offset(offset)
.limit(limit)
- .all()
)
+ entries = result.scalars().all()
# Get user info for each entry
user_ids = list(set(e.user_id for e in entries))
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
return [
@@ -105,54 +106,61 @@ class PromptHistoryTable:
for entry in entries
]
- def get_history_entry_by_id(
+ async def get_history_entry_by_id(
self,
history_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptHistoryModel]:
"""Get a specific history entry by ID."""
- with get_db_context(db) as db:
- entry = db.query(PromptHistory).filter(PromptHistory.id == history_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(PromptHistory).filter(PromptHistory.id == history_id))
+ entry = result.scalars().first()
if entry:
return PromptHistoryModel.model_validate(entry)
return None
- def get_latest_history_entry(
+ async def get_latest_history_entry(
self,
prompt_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptHistoryModel]:
"""Get the most recent history entry for a prompt."""
- with get_db_context(db) as db:
- entry = (
- db.query(PromptHistory)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(PromptHistory)
.filter(PromptHistory.prompt_id == prompt_id)
.order_by(PromptHistory.created_at.desc())
- .first()
+ .limit(1)
)
+ entry = result.scalars().first()
if entry:
return PromptHistoryModel.model_validate(entry)
return None
- def get_history_count(
+ async def get_history_count(
self,
prompt_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> int:
"""Get the number of history entries for a prompt."""
- with get_db_context(db) as db:
- return db.query(PromptHistory).filter(PromptHistory.prompt_id == prompt_id).count()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(func.count()).select_from(PromptHistory).filter(PromptHistory.prompt_id == prompt_id)
+ )
+ return result.scalar()
- def compute_diff(
+ async def compute_diff(
self,
from_id: str,
to_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[dict]:
"""Compute diff between two history entries."""
- with get_db_context(db) as db:
- from_entry = db.query(PromptHistory).filter(PromptHistory.id == from_id).first()
- to_entry = db.query(PromptHistory).filter(PromptHistory.id == to_id).first()
+ async with get_async_db_context(db) as db:
+ result_from = await db.execute(select(PromptHistory).filter(PromptHistory.id == from_id))
+ from_entry = result_from.scalars().first()
+ result_to = await db.execute(select(PromptHistory).filter(PromptHistory.id == to_id))
+ to_entry = result_to.scalars().first()
if not from_entry or not to_entry:
return None
@@ -183,37 +191,39 @@ class PromptHistoryTable:
'name_changed': from_snapshot.get('name') != to_snapshot.get('name'),
}
- def delete_history_by_prompt_id(
+ async def delete_history_by_prompt_id(
self,
prompt_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
"""Delete all history entries for a prompt."""
- with get_db_context(db) as db:
- db.query(PromptHistory).filter(PromptHistory.prompt_id == prompt_id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(PromptHistory).filter(PromptHistory.prompt_id == prompt_id))
+ await db.commit()
return True
- def delete_history_entry(
+ async def delete_history_entry(
self,
history_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
"""Delete a history entry and reparent its children to grandparent."""
- with get_db_context(db) as db:
- entry = db.query(PromptHistory).filter_by(id=history_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(PromptHistory).filter_by(id=history_id))
+ entry = result.scalars().first()
if not entry:
return False
# Find children that reference this entry as parent
- children = db.query(PromptHistory).filter_by(parent_id=history_id).all()
+ children_result = await db.execute(select(PromptHistory).filter_by(parent_id=history_id))
+ children = children_result.scalars().all()
# Reparent children to grandparent
for child in children:
child.parent_id = entry.parent_id
- db.delete(entry)
- db.commit()
+ await db.delete(entry)
+ await db.commit()
return True
diff --git a/backend/open_webui/models/prompts.py b/backend/open_webui/models/prompts.py
index bb77f32f31..fbf5401203 100644
--- a/backend/open_webui/models/prompts.py
+++ b/backend/open_webui/models/prompts.py
@@ -1,17 +1,19 @@
+import json
import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update, or_, func, cast, String
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.groups import Groups
-from open_webui.models.users import Users, UserResponse
+from open_webui.models.users import Users, User, UserModel, UserResponse
from open_webui.models.prompt_history import PromptHistories
from open_webui.models.access_grants import AccessGrantModel, AccessGrants
from pydantic import BaseModel, ConfigDict, Field
-from sqlalchemy import BigInteger, Boolean, Column, String, Text, JSON, or_, func, cast
+from sqlalchemy import BigInteger, Boolean, Column, Text, JSON
####################
# Prompts DB Schema
@@ -92,23 +94,23 @@ class PromptForm(BaseModel):
class PromptsTable:
- def _get_access_grants(self, prompt_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('prompt', prompt_id, db=db)
+ async def _get_access_grants(self, prompt_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('prompt', prompt_id, db=db)
- def _to_prompt_model(
+ async def _to_prompt_model(
self,
prompt: Prompt,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> PromptModel:
prompt_data = PromptModel.model_validate(prompt).model_dump(exclude={'access_grants'})
prompt_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(prompt_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(prompt_data['id'], db=db)
)
return PromptModel.model_validate(prompt_data)
- def insert_new_prompt(
- self, user_id: str, form_data: PromptForm, db: Optional[Session] = None
+ async def insert_new_prompt(
+ self, user_id: str, form_data: PromptForm, db: Optional[AsyncSession] = None
) -> Optional[PromptModel]:
now = int(time.time())
prompt_id = str(uuid.uuid4())
@@ -129,15 +131,15 @@ class PromptsTable:
)
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
result = Prompt(**prompt.model_dump(exclude={'access_grants'}))
db.add(result)
- db.commit()
- db.refresh(result)
- AccessGrants.set_access_grants('prompt', prompt_id, form_data.access_grants, db=db)
+ await db.commit()
+ await db.refresh(result)
+ await AccessGrants.set_access_grants('prompt', prompt_id, form_data.access_grants, db=db)
if result:
- current_access_grants = self._get_access_grants(prompt_id, db=db)
+ current_access_grants = await self._get_access_grants(prompt_id, db=db)
snapshot = {
'name': form_data.name,
'content': form_data.content,
@@ -148,7 +150,7 @@ class PromptsTable:
'access_grants': [grant.model_dump() for grant in current_access_grants],
}
- history_entry = PromptHistories.create_history_entry(
+ history_entry = await PromptHistories.create_history_entry(
prompt_id=prompt_id,
snapshot=snapshot,
user_id=user_id,
@@ -160,46 +162,51 @@ class PromptsTable:
# Set the initial version as the production version
if history_entry:
result.version_id = history_entry.id
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
- return self._to_prompt_model(result, db=db)
+ return await self._to_prompt_model(result, db=db)
else:
return None
except Exception:
return None
- def get_prompt_by_id(self, prompt_id: str, db: Optional[Session] = None) -> Optional[PromptModel]:
+ async def get_prompt_by_id(self, prompt_id: str, db: Optional[AsyncSession] = None) -> Optional[PromptModel]:
"""Get prompt by UUID."""
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(id=prompt_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(id=prompt_id))
+ prompt = result.scalars().first()
if prompt:
- return self._to_prompt_model(prompt, db=db)
+ return await self._to_prompt_model(prompt, db=db)
return None
except Exception:
return None
- def get_prompt_by_command(self, command: str, db: Optional[Session] = None) -> Optional[PromptModel]:
+ async def get_prompt_by_command(self, command: str, db: Optional[AsyncSession] = None) -> Optional[PromptModel]:
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(command=command).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(command=command))
+ prompt = result.scalars().first()
if prompt:
- return self._to_prompt_model(prompt, db=db)
+ return await self._to_prompt_model(prompt, db=db)
return None
except Exception:
return None
- def get_prompts(self, db: Optional[Session] = None) -> list[PromptUserResponse]:
- with get_db_context(db) as db:
- all_prompts = db.query(Prompt).filter(Prompt.is_active == True).order_by(Prompt.updated_at.desc()).all()
+ async def get_prompts(self, db: Optional[AsyncSession] = None) -> list[PromptUserResponse]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(Prompt).filter(Prompt.is_active == True).order_by(Prompt.updated_at.desc())
+ )
+ all_prompts = result.scalars().all()
user_ids = list(set(prompt.user_id for prompt in all_prompts))
prompt_ids = [prompt.id for prompt in all_prompts]
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
- grants_map = AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=db)
prompts = []
for prompt in all_prompts:
@@ -207,10 +214,12 @@ class PromptsTable:
prompts.append(
PromptUserResponse.model_validate(
{
- **self._to_prompt_model(
- prompt,
- access_grants=grants_map.get(prompt.id, []),
- db=db,
+ **(
+ await self._to_prompt_model(
+ prompt,
+ access_grants=grants_map.get(prompt.id, []),
+ db=db,
+ )
).model_dump(),
'user': user.model_dump() if user else None,
}
@@ -219,44 +228,44 @@ class PromptsTable:
return prompts
- def get_prompts_by_user_id(
- self, user_id: str, permission: str = 'write', db: Optional[Session] = None
+ async def get_prompts_by_user_id(
+ self, user_id: str, permission: str = 'write', db: Optional[AsyncSession] = None
) -> list[PromptUserResponse]:
- prompts = self.get_prompts(db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
+ prompts = await self.get_prompts(db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
- return [
- prompt
- for prompt in prompts
- if prompt.user_id == user_id
- or AccessGrants.has_access(
+ result = []
+ for prompt in prompts:
+ if prompt.user_id == user_id:
+ result.append(prompt)
+ elif await AccessGrants.has_access(
user_id=user_id,
resource_type='prompt',
resource_id=prompt.id,
permission=permission,
user_group_ids=user_group_ids,
db=db,
- )
- ]
+ ):
+ result.append(prompt)
+ return result
- def search_prompts(
+ async def search_prompts(
self,
user_id: str,
filter: dict = {},
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> PromptListResponse:
- with get_db_context(db) as db:
- from open_webui.models.users import User, UserModel
-
+ async with get_async_db_context(db) as db:
# Join with User table for user filtering and sorting
- query = db.query(Prompt, User).outerjoin(User, User.id == Prompt.user_id)
+ stmt = select(Prompt, User).outerjoin(User, User.id == Prompt.user_id)
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(
+ stmt = stmt.filter(
or_(
Prompt.name.ilike(f'%{query_key}%'),
Prompt.command.ilike(f'%{query_key}%'),
@@ -268,14 +277,14 @@ class PromptsTable:
view_option = filter.get('view_option')
if view_option == 'created':
- query = query.filter(Prompt.user_id == user_id)
+ stmt = stmt.filter(Prompt.user_id == user_id)
elif view_option == 'shared':
- query = query.filter(Prompt.user_id != user_id)
+ stmt = stmt.filter(Prompt.user_id != user_id)
# Apply access grant filtering
- query = AccessGrants.has_permission_filter(
+ stmt = AccessGrants.has_permission_filter(
db=db,
- query=query,
+ query=stmt,
DocumentModel=Prompt,
filter=filter,
resource_type='prompt',
@@ -284,55 +293,71 @@ class PromptsTable:
tag = filter.get('tag')
if tag:
- # Search for tag in JSON array field
- like_pattern = f'%"{tag.lower()}"%'
- tags_text = func.lower(cast(Prompt.tags, String))
- query = query.filter(tags_text.like(like_pattern))
+ # SQLite stores JSON text via json.dumps(ensure_ascii=True),
+ # so non-ASCII chars are \uXXXX-escaped. PostgreSQL native JSONB
+ # stores literal Unicode. Use the right pattern for each.
+ if db.bind.dialect.name == 'sqlite':
+ if tag.isascii():
+ tags_text = func.lower(cast(Prompt.tags, String))
+ pattern = f'%{json.dumps(tag.lower())}%'
+ else:
+ # LOWER() is ASCII-only; non-ASCII codepoints would
+ # produce different \uXXXX escapes when lowered.
+ tags_text = cast(Prompt.tags, String)
+ pattern = f'%{json.dumps(tag)}%'
+ else:
+ tags_text = func.lower(cast(Prompt.tags, String))
+ pattern = f'%{json.dumps(tag.lower(), ensure_ascii=False)}%'
+ stmt = stmt.filter(tags_text.like(pattern))
order_by = filter.get('order_by')
direction = filter.get('direction')
if order_by == 'name':
if direction == 'asc':
- query = query.order_by(Prompt.name.asc())
+ stmt = stmt.order_by(Prompt.name.asc())
else:
- query = query.order_by(Prompt.name.desc())
+ stmt = stmt.order_by(Prompt.name.desc())
elif order_by == 'created_at':
if direction == 'asc':
- query = query.order_by(Prompt.created_at.asc())
+ stmt = stmt.order_by(Prompt.created_at.asc())
else:
- query = query.order_by(Prompt.created_at.desc())
+ stmt = stmt.order_by(Prompt.created_at.desc())
elif order_by == 'updated_at':
if direction == 'asc':
- query = query.order_by(Prompt.updated_at.asc())
+ stmt = stmt.order_by(Prompt.updated_at.asc())
else:
- query = query.order_by(Prompt.updated_at.desc())
+ stmt = stmt.order_by(Prompt.updated_at.desc())
else:
- query = query.order_by(Prompt.updated_at.desc())
+ stmt = stmt.order_by(Prompt.updated_at.desc())
else:
- query = query.order_by(Prompt.updated_at.desc())
+ stmt = stmt.order_by(Prompt.updated_at.desc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
prompt_ids = [prompt.id for prompt, _ in items]
- grants_map = AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=db)
prompts = []
for prompt, user in items:
prompts.append(
PromptUserResponse(
- **self._to_prompt_model(
- prompt,
- access_grants=grants_map.get(prompt.id, []),
- db=db,
+ **(
+ await self._to_prompt_model(
+ prompt,
+ access_grants=grants_map.get(prompt.id, []),
+ db=db,
+ )
).model_dump(),
user=(UserResponse(**UserModel.model_validate(user).model_dump()) if user else None),
)
@@ -340,22 +365,23 @@ class PromptsTable:
return PromptListResponse(items=prompts, total=total)
- def update_prompt_by_command(
+ async def update_prompt_by_command(
self,
command: str,
form_data: PromptForm,
user_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptModel]:
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(command=command).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(command=command))
+ prompt = result.scalars().first()
if not prompt:
return None
- latest_history = PromptHistories.get_latest_history_entry(prompt.id, db=db)
+ latest_history = await PromptHistories.get_latest_history_entry(prompt.id, db=db)
parent_id = latest_history.id if latest_history else None
- current_access_grants = self._get_access_grants(prompt.id, db=db)
+ current_access_grants = await self._get_access_grants(prompt.id, db=db)
# Check if content changed to decide on history creation
content_changed = (
@@ -371,10 +397,10 @@ class PromptsTable:
prompt.meta = form_data.meta or prompt.meta
prompt.updated_at = int(time.time())
if form_data.access_grants is not None:
- AccessGrants.set_access_grants('prompt', prompt.id, form_data.access_grants, db=db)
- current_access_grants = self._get_access_grants(prompt.id, db=db)
+ await AccessGrants.set_access_grants('prompt', prompt.id, form_data.access_grants, db=db)
+ current_access_grants = await self._get_access_grants(prompt.id, db=db)
- db.commit()
+ await db.commit()
# Create history entry only if content changed
if content_changed:
@@ -387,7 +413,7 @@ class PromptsTable:
'access_grants': [grant.model_dump() for grant in current_access_grants],
}
- history_entry = PromptHistories.create_history_entry(
+ history_entry = await PromptHistories.create_history_entry(
prompt_id=prompt.id,
snapshot=snapshot,
user_id=user_id,
@@ -399,28 +425,29 @@ class PromptsTable:
# Set as production if flag is True (default)
if form_data.is_production and history_entry:
prompt.version_id = history_entry.id
- db.commit()
+ await db.commit()
- return self._to_prompt_model(prompt, db=db)
+ return await self._to_prompt_model(prompt, db=db)
except Exception:
return None
- def update_prompt_by_id(
+ async def update_prompt_by_id(
self,
prompt_id: str,
form_data: PromptForm,
user_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptModel]:
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(id=prompt_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(id=prompt_id))
+ prompt = result.scalars().first()
if not prompt:
return None
- latest_history = PromptHistories.get_latest_history_entry(prompt.id, db=db)
+ latest_history = await PromptHistories.get_latest_history_entry(prompt.id, db=db)
parent_id = latest_history.id if latest_history else None
- current_access_grants = self._get_access_grants(prompt.id, db=db)
+ current_access_grants = await self._get_access_grants(prompt.id, db=db)
# Check if content changed to decide on history creation
content_changed = (
@@ -442,12 +469,12 @@ class PromptsTable:
prompt.tags = form_data.tags
if form_data.access_grants is not None:
- AccessGrants.set_access_grants('prompt', prompt.id, form_data.access_grants, db=db)
- current_access_grants = self._get_access_grants(prompt.id, db=db)
+ await AccessGrants.set_access_grants('prompt', prompt.id, form_data.access_grants, db=db)
+ current_access_grants = await self._get_access_grants(prompt.id, db=db)
prompt.updated_at = int(time.time())
- db.commit()
+ await db.commit()
# Create history entry only if content changed
if content_changed:
@@ -461,7 +488,7 @@ class PromptsTable:
'access_grants': [grant.model_dump() for grant in current_access_grants],
}
- history_entry = PromptHistories.create_history_entry(
+ history_entry = await PromptHistories.create_history_entry(
prompt_id=prompt.id,
snapshot=snapshot,
user_id=user_id,
@@ -473,24 +500,25 @@ class PromptsTable:
# Set as production if flag is True (default)
if form_data.is_production and history_entry:
prompt.version_id = history_entry.id
- db.commit()
+ await db.commit()
- return self._to_prompt_model(prompt, db=db)
+ return await self._to_prompt_model(prompt, db=db)
except Exception:
return None
- def update_prompt_metadata(
+ async def update_prompt_metadata(
self,
prompt_id: str,
name: str,
command: str,
tags: Optional[list[str]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptModel]:
"""Update only name, command, and tags (no history created)."""
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(id=prompt_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(id=prompt_id))
+ prompt = result.scalars().first()
if not prompt:
return None
@@ -501,26 +529,27 @@ class PromptsTable:
prompt.tags = tags
prompt.updated_at = int(time.time())
- db.commit()
+ await db.commit()
- return self._to_prompt_model(prompt, db=db)
+ return await self._to_prompt_model(prompt, db=db)
except Exception:
return None
- def update_prompt_version(
+ async def update_prompt_version(
self,
prompt_id: str,
version_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[PromptModel]:
"""Set the active version of a prompt and restore content from that version's snapshot."""
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(id=prompt_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(id=prompt_id))
+ prompt = result.scalars().first()
if not prompt:
return None
- history_entry = PromptHistories.get_history_entry_by_id(version_id, db=db)
+ history_entry = await PromptHistories.get_history_entry_by_id(version_id, db=db)
if not history_entry:
return None
@@ -537,63 +566,67 @@ class PromptsTable:
prompt.version_id = version_id
prompt.updated_at = int(time.time())
- db.commit()
+ await db.commit()
- return self._to_prompt_model(prompt, db=db)
+ return await self._to_prompt_model(prompt, db=db)
except Exception:
return None
- def toggle_prompt_active(self, prompt_id: str, db: Optional[Session] = None) -> Optional[PromptModel]:
+ async def toggle_prompt_active(self, prompt_id: str, db: Optional[AsyncSession] = None) -> Optional[PromptModel]:
"""Toggle the is_active flag on a prompt."""
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(id=prompt_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(id=prompt_id))
+ prompt = result.scalars().first()
if prompt:
prompt.is_active = not prompt.is_active
prompt.updated_at = int(time.time())
- db.commit()
- db.refresh(prompt)
- return self._to_prompt_model(prompt, db=db)
+ await db.commit()
+ await db.refresh(prompt)
+ return await self._to_prompt_model(prompt, db=db)
return None
except Exception:
return None
- def delete_prompt_by_command(self, command: str, db: Optional[Session] = None) -> bool:
+ async def delete_prompt_by_command(self, command: str, db: Optional[AsyncSession] = None) -> bool:
"""Permanently delete a prompt and its history."""
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(command=command).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(command=command))
+ prompt = result.scalars().first()
if prompt:
- PromptHistories.delete_history_by_prompt_id(prompt.id, db=db)
- AccessGrants.revoke_all_access('prompt', prompt.id, db=db)
+ await PromptHistories.delete_history_by_prompt_id(prompt.id, db=db)
+ await AccessGrants.revoke_all_access('prompt', prompt.id, db=db)
- db.delete(prompt)
- db.commit()
+ await db.delete(prompt)
+ await db.commit()
return True
return False
except Exception:
return False
- def delete_prompt_by_id(self, prompt_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_prompt_by_id(self, prompt_id: str, db: Optional[AsyncSession] = None) -> bool:
"""Permanently delete a prompt and its history."""
try:
- with get_db_context(db) as db:
- prompt = db.query(Prompt).filter_by(id=prompt_id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(id=prompt_id))
+ prompt = result.scalars().first()
if prompt:
- PromptHistories.delete_history_by_prompt_id(prompt.id, db=db)
- AccessGrants.revoke_all_access('prompt', prompt.id, db=db)
+ await PromptHistories.delete_history_by_prompt_id(prompt.id, db=db)
+ await AccessGrants.revoke_all_access('prompt', prompt.id, db=db)
- db.delete(prompt)
- db.commit()
+ await db.delete(prompt)
+ await db.commit()
return True
return False
except Exception:
return False
- def get_tags(self, db: Optional[Session] = None) -> list[str]:
+ async def get_tags(self, db: Optional[AsyncSession] = None) -> list[str]:
try:
- with get_db_context(db) as db:
- prompts = db.query(Prompt).filter_by(is_active=True).all()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Prompt).filter_by(is_active=True))
+ prompts = result.scalars().all()
tags = set()
for prompt in prompts:
if prompt.tags:
diff --git a/backend/open_webui/models/skills.py b/backend/open_webui/models/skills.py
index cdf8ecaea4..0fc6dfc52d 100644
--- a/backend/open_webui/models/skills.py
+++ b/backend/open_webui/models/skills.py
@@ -2,14 +2,15 @@ import logging
import time
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, get_db, get_db_context
-from open_webui.models.users import Users, UserResponse
+from sqlalchemy import select, delete, update, or_
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, get_async_db_context
+from open_webui.models.users import Users, User, UserModel, UserResponse
from open_webui.models.groups import Groups
from open_webui.models.access_grants import AccessGrantModel, AccessGrants
from pydantic import BaseModel, ConfigDict, Field
-from sqlalchemy import JSON, BigInteger, Boolean, Column, String, Text, or_
+from sqlalchemy import JSON, BigInteger, Boolean, Column, String, Text, func
log = logging.getLogger(__name__)
@@ -105,28 +106,28 @@ class SkillAccessListResponse(BaseModel):
class SkillsTable:
- def _get_access_grants(self, skill_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('skill', skill_id, db=db)
+ async def _get_access_grants(self, skill_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('skill', skill_id, db=db)
- def _to_skill_model(
+ async def _to_skill_model(
self,
skill: Skill,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> SkillModel:
skill_data = SkillModel.model_validate(skill).model_dump(exclude={'access_grants'})
skill_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(skill_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(skill_data['id'], db=db)
)
return SkillModel.model_validate(skill_data)
- def insert_new_skill(
+ async def insert_new_skill(
self,
user_id: str,
form_data: SkillForm,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[SkillModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
result = Skill(
**{
@@ -137,43 +138,45 @@ class SkillsTable:
}
)
db.add(result)
- db.commit()
- db.refresh(result)
- AccessGrants.set_access_grants('skill', result.id, form_data.access_grants, db=db)
+ await db.commit()
+ await db.refresh(result)
+ await AccessGrants.set_access_grants('skill', result.id, form_data.access_grants, db=db)
if result:
- return self._to_skill_model(result, db=db)
+ return await self._to_skill_model(result, db=db)
else:
return None
except Exception as e:
log.exception(f'Error creating a new skill: {e}')
return None
- def get_skill_by_id(self, id: str, db: Optional[Session] = None) -> Optional[SkillModel]:
+ async def get_skill_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[SkillModel]:
try:
- with get_db_context(db) as db:
- skill = db.get(Skill, id)
- return self._to_skill_model(skill, db=db) if skill else None
+ async with get_async_db_context(db) as db:
+ skill = await db.get(Skill, id)
+ return await self._to_skill_model(skill, db=db) if skill else None
except Exception:
return None
- def get_skill_by_name(self, name: str, db: Optional[Session] = None) -> Optional[SkillModel]:
+ async def get_skill_by_name(self, name: str, db: Optional[AsyncSession] = None) -> Optional[SkillModel]:
try:
- with get_db_context(db) as db:
- skill = db.query(Skill).filter_by(name=name).first()
- return self._to_skill_model(skill, db=db) if skill else None
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Skill).filter_by(name=name))
+ skill = result.scalars().first()
+ return await self._to_skill_model(skill, db=db) if skill else None
except Exception:
return None
- def get_skills(self, db: Optional[Session] = None) -> list[SkillUserModel]:
- with get_db_context(db) as db:
- all_skills = db.query(Skill).order_by(Skill.updated_at.desc()).all()
+ async def get_skills(self, db: Optional[AsyncSession] = None) -> list[SkillUserModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Skill).order_by(Skill.updated_at.desc()))
+ all_skills = result.scalars().all()
user_ids = list(set(skill.user_id for skill in all_skills))
skill_ids = [skill.id for skill in all_skills]
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
- grants_map = AccessGrants.get_grants_by_resources('skill', skill_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('skill', skill_ids, db=db)
skills = []
for skill in all_skills:
@@ -181,10 +184,12 @@ class SkillsTable:
skills.append(
SkillUserModel.model_validate(
{
- **self._to_skill_model(
- skill,
- access_grants=grants_map.get(skill.id, []),
- db=db,
+ **(
+ await self._to_skill_model(
+ skill,
+ access_grants=grants_map.get(skill.id, []),
+ db=db,
+ )
).model_dump(),
'user': user.model_dump() if user else None,
}
@@ -192,45 +197,45 @@ class SkillsTable:
)
return skills
- def get_skills_by_user_id(
- self, user_id: str, permission: str = 'write', db: Optional[Session] = None
+ async def get_skills_by_user_id(
+ self, user_id: str, permission: str = 'write', db: Optional[AsyncSession] = None
) -> list[SkillUserModel]:
- skills = self.get_skills(db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
+ skills = await self.get_skills(db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
- return [
- skill
- for skill in skills
- if skill.user_id == user_id
- or AccessGrants.has_access(
+ result = []
+ for skill in skills:
+ if skill.user_id == user_id:
+ result.append(skill)
+ elif await AccessGrants.has_access(
user_id=user_id,
resource_type='skill',
resource_id=skill.id,
permission=permission,
user_group_ids=user_group_ids,
db=db,
- )
- ]
+ ):
+ result.append(skill)
+ return result
- def search_skills(
+ async def search_skills(
self,
user_id: str,
filter: dict = {},
skip: int = 0,
limit: int = 30,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> SkillListResponse:
try:
- with get_db_context(db) as db:
- from open_webui.models.users import User, UserModel
-
+ async with get_async_db_context(db) as db:
# Join with User table for user filtering
- query = db.query(Skill, User).outerjoin(User, User.id == Skill.user_id)
+ stmt = select(Skill, User).outerjoin(User, User.id == Skill.user_id)
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(
+ stmt = stmt.filter(
or_(
Skill.name.ilike(f'%{query_key}%'),
Skill.description.ilike(f'%{query_key}%'),
@@ -242,43 +247,47 @@ class SkillsTable:
view_option = filter.get('view_option')
if view_option == 'created':
- query = query.filter(Skill.user_id == user_id)
+ stmt = stmt.filter(Skill.user_id == user_id)
elif view_option == 'shared':
- query = query.filter(Skill.user_id != user_id)
+ stmt = stmt.filter(Skill.user_id != user_id)
# Apply access grant filtering
- query = AccessGrants.has_permission_filter(
+ stmt = AccessGrants.has_permission_filter(
db=db,
- query=query,
+ query=stmt,
DocumentModel=Skill,
filter=filter,
resource_type='skill',
permission='read',
)
- query = query.order_by(Skill.updated_at.desc())
+ stmt = stmt.order_by(Skill.updated_at.desc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
if skip:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- items = query.all()
+ result = await db.execute(stmt)
+ items = result.all()
skill_ids = [skill.id for skill, _ in items]
- grants_map = AccessGrants.get_grants_by_resources('skill', skill_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('skill', skill_ids, db=db)
skills = []
for skill, user in items:
skills.append(
SkillUserResponse(
- **self._to_skill_model(
- skill,
- access_grants=grants_map.get(skill.id, []),
- db=db,
+ **(
+ await self._to_skill_model(
+ skill,
+ access_grants=grants_map.get(skill.id, []),
+ db=db,
+ )
).model_dump(),
user=(UserResponse(**UserModel.model_validate(user).model_dump()) if user else None),
)
@@ -289,43 +298,46 @@ class SkillsTable:
log.exception(f'Error searching skills: {e}')
return SkillListResponse(items=[], total=0)
- def update_skill_by_id(self, id: str, updated: dict, db: Optional[Session] = None) -> Optional[SkillModel]:
+ async def update_skill_by_id(
+ self, id: str, updated: dict, db: Optional[AsyncSession] = None
+ ) -> Optional[SkillModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
access_grants = updated.pop('access_grants', None)
- db.query(Skill).filter_by(id=id).update({**updated, 'updated_at': int(time.time())})
- db.commit()
+ await db.execute(update(Skill).filter_by(id=id).values(**updated, updated_at=int(time.time())))
+ await db.commit()
if access_grants is not None:
- AccessGrants.set_access_grants('skill', id, access_grants, db=db)
+ await AccessGrants.set_access_grants('skill', id, access_grants, db=db)
- skill = db.query(Skill).get(id)
- db.refresh(skill)
- return self._to_skill_model(skill, db=db)
+ skill = await db.get(Skill, id)
+ await db.refresh(skill)
+ return await self._to_skill_model(skill, db=db)
except Exception:
return None
- def toggle_skill_by_id(self, id: str, db: Optional[Session] = None) -> Optional[SkillModel]:
- with get_db_context(db) as db:
+ async def toggle_skill_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[SkillModel]:
+ async with get_async_db_context(db) as db:
try:
- skill = db.query(Skill).filter_by(id=id).first()
+ result = await db.execute(select(Skill).filter_by(id=id))
+ skill = result.scalars().first()
if not skill:
return None
skill.is_active = not skill.is_active
skill.updated_at = int(time.time())
- db.commit()
- db.refresh(skill)
+ await db.commit()
+ await db.refresh(skill)
- return self._to_skill_model(skill, db=db)
+ return await self._to_skill_model(skill, db=db)
except Exception:
return None
- def delete_skill_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_skill_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- AccessGrants.revoke_all_access('skill', id, db=db)
- db.query(Skill).filter_by(id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await AccessGrants.revoke_all_access('skill', id, db=db)
+ await db.execute(delete(Skill).filter_by(id=id))
+ await db.commit()
return True
except Exception:
diff --git a/backend/open_webui/models/tags.py b/backend/open_webui/models/tags.py
index b60220bc23..ee2baefc01 100644
--- a/backend/open_webui/models/tags.py
+++ b/backend/open_webui/models/tags.py
@@ -3,8 +3,9 @@ import time
import uuid
from typing import Optional
-from sqlalchemy.orm import Session
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from pydantic import BaseModel, ConfigDict
@@ -53,15 +54,15 @@ class TagChatIdForm(BaseModel):
class TagTable:
- def insert_new_tag(self, name: str, user_id: str, db: Optional[Session] = None) -> Optional[TagModel]:
- with get_db_context(db) as db:
+ async def insert_new_tag(self, name: str, user_id: str, db: Optional[AsyncSession] = None) -> Optional[TagModel]:
+ async with get_async_db_context(db) as db:
id = name.replace(' ', '_').lower()
tag = TagModel(**{'id': id, 'user_id': user_id, 'name': name})
try:
result = Tag(**tag.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return TagModel.model_validate(result)
else:
@@ -70,64 +71,71 @@ class TagTable:
log.exception(f'Error inserting a new tag: {e}')
return None
- def get_tag_by_name_and_user_id(self, name: str, user_id: str, db: Optional[Session] = None) -> Optional[TagModel]:
+ async def get_tag_by_name_and_user_id(
+ self, name: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[TagModel]:
try:
id = name.replace(' ', '_').lower()
- with get_db_context(db) as db:
- tag = db.query(Tag).filter_by(id=id, user_id=user_id).first()
- return TagModel.model_validate(tag)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Tag).filter_by(id=id, user_id=user_id))
+ tag = result.scalars().first()
+ return TagModel.model_validate(tag) if tag else None
except Exception:
return None
- def get_tags_by_user_id(self, user_id: str, db: Optional[Session] = None) -> list[TagModel]:
- with get_db_context(db) as db:
- return [TagModel.model_validate(tag) for tag in (db.query(Tag).filter_by(user_id=user_id).all())]
+ async def get_tags_by_user_id(self, user_id: str, db: Optional[AsyncSession] = None) -> list[TagModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Tag).filter_by(user_id=user_id))
+ return [TagModel.model_validate(tag) for tag in result.scalars().all()]
- def get_tags_by_ids_and_user_id(self, ids: list[str], user_id: str, db: Optional[Session] = None) -> list[TagModel]:
- with get_db_context(db) as db:
- return [
- TagModel.model_validate(tag)
- for tag in (db.query(Tag).filter(Tag.id.in_(ids), Tag.user_id == user_id).all())
- ]
+ async def get_tags_by_ids_and_user_id(
+ self, ids: list[str], user_id: str, db: Optional[AsyncSession] = None
+ ) -> list[TagModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Tag).filter(Tag.id.in_(ids), Tag.user_id == user_id))
+ return [TagModel.model_validate(tag) for tag in result.scalars().all()]
- def delete_tag_by_name_and_user_id(self, name: str, user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_tag_by_name_and_user_id(self, name: str, user_id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
id = name.replace(' ', '_').lower()
- res = db.query(Tag).filter_by(id=id, user_id=user_id).delete()
- log.debug(f'res: {res}')
- db.commit()
+ result = await db.execute(delete(Tag).filter_by(id=id, user_id=user_id))
+ log.debug(f'res: {result.rowcount}')
+ await db.commit()
return True
except Exception as e:
log.error(f'delete_tag: {e}')
return False
- def delete_tags_by_ids_and_user_id(self, ids: list[str], user_id: str, db: Optional[Session] = None) -> bool:
+ async def delete_tags_by_ids_and_user_id(
+ self, ids: list[str], user_id: str, db: Optional[AsyncSession] = None
+ ) -> bool:
"""Delete all tags whose id is in *ids* for the given user, in one query."""
if not ids:
return True
try:
- with get_db_context(db) as db:
- db.query(Tag).filter(Tag.id.in_(ids), Tag.user_id == user_id).delete(synchronize_session=False)
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(Tag).filter(Tag.id.in_(ids), Tag.user_id == user_id))
+ await db.commit()
return True
except Exception as e:
log.error(f'delete_tags_by_ids: {e}')
return False
- def ensure_tags_exist(self, names: list[str], user_id: str, db: Optional[Session] = None) -> None:
+ async def ensure_tags_exist(self, names: list[str], user_id: str, db: Optional[AsyncSession] = None) -> None:
"""Create tag rows for any *names* that don't already exist for *user_id*."""
if not names:
return
ids = [n.replace(' ', '_').lower() for n in names]
- with get_db_context(db) as db:
- existing = {t.id for t in db.query(Tag.id).filter(Tag.id.in_(ids), Tag.user_id == user_id).all()}
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(Tag.id).filter(Tag.id.in_(ids), Tag.user_id == user_id))
+ existing = {row[0] for row in result.all()}
new_tags = [
Tag(id=tag_id, name=name, user_id=user_id) for tag_id, name in zip(ids, names) if tag_id not in existing
]
if new_tags:
db.add_all(new_tags)
- db.commit()
+ await db.commit()
Tags = TagTable()
diff --git a/backend/open_webui/models/tools.py b/backend/open_webui/models/tools.py
index f89b98c5e7..70035121aa 100644
--- a/backend/open_webui/models/tools.py
+++ b/backend/open_webui/models/tools.py
@@ -2,8 +2,9 @@ import logging
import time
from typing import Optional
-from sqlalchemy.orm import Session, defer
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
+from sqlalchemy import select, delete, update
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.models.users import Users, UserResponse
from open_webui.models.groups import Groups
from open_webui.models.access_grants import AccessGrantModel, AccessGrants
@@ -97,29 +98,29 @@ class ToolValves(BaseModel):
class ToolsTable:
- def _get_access_grants(self, tool_id: str, db: Optional[Session] = None) -> list[AccessGrantModel]:
- return AccessGrants.get_grants_by_resource('tool', tool_id, db=db)
+ async def _get_access_grants(self, tool_id: str, db: Optional[AsyncSession] = None) -> list[AccessGrantModel]:
+ return await AccessGrants.get_grants_by_resource('tool', tool_id, db=db)
- def _to_tool_model(
+ async def _to_tool_model(
self,
tool: Tool,
access_grants: Optional[list[AccessGrantModel]] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> ToolModel:
tool_data = ToolModel.model_validate(tool).model_dump(exclude={'access_grants'})
tool_data['access_grants'] = (
- access_grants if access_grants is not None else self._get_access_grants(tool_data['id'], db=db)
+ access_grants if access_grants is not None else await self._get_access_grants(tool_data['id'], db=db)
)
return ToolModel.model_validate(tool_data)
- def insert_new_tool(
+ async def insert_new_tool(
self,
user_id: str,
form_data: ToolForm,
specs: list[dict],
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[ToolModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
try:
result = Tool(
**{
@@ -131,38 +132,39 @@ class ToolsTable:
}
)
db.add(result)
- db.commit()
- db.refresh(result)
- AccessGrants.set_access_grants('tool', result.id, form_data.access_grants, db=db)
+ await db.commit()
+ await db.refresh(result)
+ await AccessGrants.set_access_grants('tool', result.id, form_data.access_grants, db=db)
if result:
- return self._to_tool_model(result, db=db)
+ return await self._to_tool_model(result, db=db)
else:
return None
except Exception as e:
log.exception(f'Error creating a new tool: {e}')
return None
- def get_tool_by_id(self, id: str, db: Optional[Session] = None) -> Optional[ToolModel]:
+ async def get_tool_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[ToolModel]:
try:
- with get_db_context(db) as db:
- tool = db.get(Tool, id)
- return self._to_tool_model(tool, db=db) if tool else None
+ async with get_async_db_context(db) as db:
+ tool = await db.get(Tool, id)
+ return await self._to_tool_model(tool, db=db) if tool else None
except Exception:
return None
- def get_tools(self, defer_content: bool = False, db: Optional[Session] = None) -> list[ToolUserModel]:
- with get_db_context(db) as db:
- query = db.query(Tool).order_by(Tool.updated_at.desc())
+ async def get_tools(self, defer_content: bool = False, db: Optional[AsyncSession] = None) -> list[ToolUserModel]:
+ async with get_async_db_context(db) as db:
+ stmt = select(Tool).order_by(Tool.updated_at.desc())
if defer_content:
- query = query.options(defer(Tool.content), defer(Tool.specs))
- all_tools = query.all()
+ stmt = stmt
+ result = await db.execute(stmt)
+ all_tools = result.scalars().all()
user_ids = list(set(tool.user_id for tool in all_tools))
tool_ids = [tool.id for tool in all_tools]
- users = Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
+ users = await Users.get_users_by_user_ids(user_ids, db=db) if user_ids else []
users_dict = {user.id: user for user in users}
- grants_map = AccessGrants.get_grants_by_resources('tool', tool_ids, db=db)
+ grants_map = await AccessGrants.get_grants_by_resources('tool', tool_ids, db=db)
tools = []
for tool in all_tools:
@@ -170,10 +172,12 @@ class ToolsTable:
tools.append(
ToolUserModel.model_validate(
{
- **self._to_tool_model(
- tool,
- access_grants=grants_map.get(tool.id, []),
- db=db,
+ **(
+ await self._to_tool_model(
+ tool,
+ access_grants=grants_map.get(tool.id, []),
+ db=db,
+ )
).model_dump(),
'user': user.model_dump() if user else None,
}
@@ -181,51 +185,57 @@ class ToolsTable:
)
return tools
- def get_tools_by_user_id(
+ async def get_tools_by_user_id(
self,
user_id: str,
permission: str = 'write',
defer_content: bool = False,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> list[ToolUserModel]:
- tools = self.get_tools(defer_content=defer_content, db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user_id, db=db)}
+ tools = await self.get_tools(defer_content=defer_content, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
+ user_group_ids = {group.id for group in user_groups}
- return [
- tool
- for tool in tools
- if tool.user_id == user_id
- or AccessGrants.has_access(
+ result = []
+ for tool in tools:
+ if tool.user_id == user_id:
+ result.append(tool)
+ elif await AccessGrants.has_access(
user_id=user_id,
resource_type='tool',
resource_id=tool.id,
permission=permission,
user_group_ids=user_group_ids,
db=db,
- )
- ]
+ ):
+ result.append(tool)
+ return result
- def get_tool_valves_by_id(self, id: str, db: Optional[Session] = None) -> Optional[dict]:
+ async def get_tool_valves_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[dict]:
try:
- with get_db_context(db) as db:
- tool = db.get(Tool, id)
+ async with get_async_db_context(db) as db:
+ tool = await db.get(Tool, id)
return tool.valves if tool.valves else {}
except Exception as e:
log.exception(f'Error getting tool valves by id {id}')
return None
- def update_tool_valves_by_id(self, id: str, valves: dict, db: Optional[Session] = None) -> Optional[ToolValves]:
+ async def update_tool_valves_by_id(
+ self, id: str, valves: dict, db: Optional[AsyncSession] = None
+ ) -> Optional[ToolValves]:
try:
- with get_db_context(db) as db:
- db.query(Tool).filter_by(id=id).update({'valves': valves, 'updated_at': int(time.time())})
- db.commit()
- return self.get_tool_by_id(id, db=db)
+ async with get_async_db_context(db) as db:
+ await db.execute(update(Tool).filter_by(id=id).values(valves=valves, updated_at=int(time.time())))
+ await db.commit()
+ return await self.get_tool_by_id(id, db=db)
except Exception:
return None
- def get_user_valves_by_id_and_user_id(self, id: str, user_id: str, db: Optional[Session] = None) -> Optional[dict]:
+ async def get_user_valves_by_id_and_user_id(
+ self, id: str, user_id: str, db: Optional[AsyncSession] = None
+ ) -> Optional[dict]:
try:
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
user_settings = user.settings.model_dump() if user.settings else {}
# Check if user has "tools" and "valves" settings
@@ -239,11 +249,11 @@ class ToolsTable:
log.exception(f'Error getting user values by id {id} and user_id {user_id}: {e}')
return None
- def update_user_valves_by_id_and_user_id(
- self, id: str, user_id: str, valves: dict, db: Optional[Session] = None
+ async def update_user_valves_by_id_and_user_id(
+ self, id: str, user_id: str, valves: dict, db: Optional[AsyncSession] = None
) -> Optional[dict]:
try:
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
user_settings = user.settings.model_dump() if user.settings else {}
# Check if user has "tools" and "valves" settings
@@ -255,34 +265,34 @@ class ToolsTable:
user_settings['tools']['valves'][id] = valves
# Update the user settings in the database
- Users.update_user_by_id(user_id, {'settings': user_settings}, db=db)
+ await Users.update_user_by_id(user_id, {'settings': user_settings}, db=db)
return user_settings['tools']['valves'][id]
except Exception as e:
log.exception(f'Error updating user valves by id {id} and user_id {user_id}: {e}')
return None
- def update_tool_by_id(self, id: str, updated: dict, db: Optional[Session] = None) -> Optional[ToolModel]:
+ async def update_tool_by_id(self, id: str, updated: dict, db: Optional[AsyncSession] = None) -> Optional[ToolModel]:
try:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
access_grants = updated.pop('access_grants', None)
- db.query(Tool).filter_by(id=id).update({**updated, 'updated_at': int(time.time())})
- db.commit()
+ await db.execute(update(Tool).filter_by(id=id).values(**updated, updated_at=int(time.time())))
+ await db.commit()
if access_grants is not None:
- AccessGrants.set_access_grants('tool', id, access_grants, db=db)
+ await AccessGrants.set_access_grants('tool', id, access_grants, db=db)
- tool = db.query(Tool).get(id)
- db.refresh(tool)
- return self._to_tool_model(tool, db=db)
+ tool = await db.get(Tool, id)
+ await db.refresh(tool)
+ return await self._to_tool_model(tool, db=db)
except Exception:
return None
- def delete_tool_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_tool_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- AccessGrants.revoke_all_access('tool', id, db=db)
- db.query(Tool).filter_by(id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await AccessGrants.revoke_all_access('tool', id, db=db)
+ await db.execute(delete(Tool).filter_by(id=id))
+ await db.commit()
return True
except Exception:
diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py
index ef90745efe..025e79bd8a 100644
--- a/backend/open_webui/models/users.py
+++ b/backend/open_webui/models/users.py
@@ -1,20 +1,15 @@
import time
from typing import Optional
-from sqlalchemy.orm import Session, defer
-from open_webui.internal.db import Base, JSONField, get_db, get_db_context
-
+from sqlalchemy import select, delete, update, func, or_, case, exists
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import Base, JSONField, get_async_db_context
from open_webui.env import DATABASE_USER_ACTIVE_STATUS_UPDATE_INTERVAL
-from open_webui.models.chats import Chats
-from open_webui.models.groups import Groups, GroupMember
-from open_webui.models.channels import ChannelMember
-
from open_webui.utils.misc import throttle
from open_webui.utils.validate import validate_profile_image_url
-
from pydantic import BaseModel, ConfigDict, field_validator, model_validator
from sqlalchemy import (
BigInteger,
@@ -24,11 +19,8 @@ from sqlalchemy import (
Boolean,
Text,
Date,
- exists,
- select,
cast,
)
-from sqlalchemy import or_, case, func
from sqlalchemy.dialects.postgresql import JSONB
import datetime
@@ -247,20 +239,22 @@ class UserRoleUpdateForm(BaseModel):
class UserUpdateForm(BaseModel):
- role: str
- name: str
- email: str
- profile_image_url: str
+ role: Optional[str] = None
+ name: Optional[str] = None
+ email: Optional[str] = None
+ profile_image_url: Optional[str] = None
password: Optional[str] = None
- @field_validator('profile_image_url')
+ @field_validator('profile_image_url', mode='before')
@classmethod
- def check_profile_image_url(cls, v: str) -> str:
+ def check_profile_image_url(cls, v: Optional[str]) -> Optional[str]:
+ if v is None:
+ return v
return validate_profile_image_url(v)
class UsersTable:
- def insert_new_user(
+ async def insert_new_user(
self,
id: str,
name: str,
@@ -269,9 +263,9 @@ class UsersTable:
role: str = 'pending',
username: Optional[str] = None,
oauth: Optional[dict] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[UserModel]:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
user = UserModel(
**{
'id': id,
@@ -288,87 +282,100 @@ class UsersTable:
)
result = User(**user.model_dump())
db.add(result)
- db.commit()
- db.refresh(result)
+ await db.commit()
+ await db.refresh(result)
if result:
return user
else:
return None
- def get_user_by_id(self, id: str, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def get_user_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
- return UserModel.model_validate(user)
- except Exception:
- return None
-
- def get_user_by_api_key(self, api_key: str, db: Optional[Session] = None) -> Optional[UserModel]:
- try:
- with get_db_context(db) as db:
- user = db.query(User).join(ApiKey, User.id == ApiKey.user_id).filter(ApiKey.key == api_key).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
return UserModel.model_validate(user) if user else None
except Exception:
return None
- def get_user_by_email(self, email: str, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def get_user_by_api_key(self, api_key: str, db: Optional[AsyncSession] = None) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter(func.lower(User.email) == email.lower()).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(
+ select(User).join(ApiKey, User.id == ApiKey.user_id).filter(ApiKey.key == api_key)
+ )
+ user = result.scalars().first()
return UserModel.model_validate(user) if user else None
except Exception:
return None
- def get_user_by_oauth_sub(self, provider: str, sub: str, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def get_user_by_email(self, email: str, db: Optional[AsyncSession] = None) -> Optional[UserModel]:
try:
- with get_db_context(db) as db: # type: Session
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter(func.lower(User.email) == email.lower()))
+ user = result.scalars().first()
+ return UserModel.model_validate(user) if user else None
+ except Exception:
+ return None
+
+ async def get_user_by_oauth_sub(
+ self, provider: str, sub: str, db: Optional[AsyncSession] = None
+ ) -> Optional[UserModel]:
+ try:
+ async with get_async_db_context(db) as db:
dialect_name = db.bind.dialect.name
- query = db.query(User)
+ stmt = select(User)
if dialect_name == 'sqlite':
- query = query.filter(User.oauth.contains({provider: {'sub': sub}}))
+ stmt = stmt.filter(User.oauth.contains({provider: {'sub': sub}}))
elif dialect_name == 'postgresql':
- query = query.filter(User.oauth[provider].cast(JSONB)['sub'].astext == sub)
+ stmt = stmt.filter(User.oauth[provider].cast(JSONB)['sub'].astext == sub)
- user = query.first()
+ result = await db.execute(stmt)
+ user = result.scalars().first()
return UserModel.model_validate(user) if user else None
except Exception as e:
# You may want to log the exception here
return None
- def get_user_by_scim_external_id(
- self, provider: str, external_id: str, db: Optional[Session] = None
+ async def get_user_by_scim_external_id(
+ self, provider: str, external_id: str, db: Optional[AsyncSession] = None
) -> Optional[UserModel]:
try:
- with get_db_context(db) as db: # type: Session
+ async with get_async_db_context(db) as db:
dialect_name = db.bind.dialect.name
- query = db.query(User)
+ stmt = select(User)
if dialect_name == 'sqlite':
- query = query.filter(User.scim.contains({provider: {'external_id': external_id}}))
+ stmt = stmt.filter(User.scim.contains({provider: {'external_id': external_id}}))
elif dialect_name == 'postgresql':
- query = query.filter(User.scim[provider].cast(JSONB)['external_id'].astext == external_id)
+ stmt = stmt.filter(User.scim[provider].cast(JSONB)['external_id'].astext == external_id)
- user = query.first()
+ result = await db.execute(stmt)
+ user = result.scalars().first()
return UserModel.model_validate(user) if user else None
except Exception:
return None
- def get_users(
+ async def get_users(
self,
filter: Optional[dict] = None,
skip: Optional[int] = None,
limit: Optional[int] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> dict:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
+ # Import here to avoid circular imports
+ from open_webui.models.groups import GroupMember
+ from open_webui.models.channels import ChannelMember
+
# Join GroupMember so we can order by group_id when requested
- query = db.query(User).options(defer(User.profile_image_url))
+ stmt = select(User)
if filter:
query_key = filter.get('query')
if query_key:
- query = query.filter(
+ stmt = stmt.filter(
or_(
User.name.ilike(f'%{query_key}%'),
User.email.ilike(f'%{query_key}%'),
@@ -377,7 +384,7 @@ class UsersTable:
channel_id = filter.get('channel_id')
if channel_id:
- query = query.filter(
+ stmt = stmt.filter(
exists(
select(ChannelMember.id).where(
ChannelMember.user_id == User.id,
@@ -395,10 +402,10 @@ class UsersTable:
return {'users': [], 'total': 0}
if user_ids:
- query = query.filter(User.id.in_(user_ids))
+ stmt = stmt.filter(User.id.in_(user_ids))
if group_ids:
- query = query.filter(
+ stmt = stmt.filter(
exists(
select(GroupMember.id).where(
GroupMember.user_id == User.id,
@@ -413,9 +420,9 @@ class UsersTable:
exclude_roles = [role[1:] for role in roles if role.startswith('!')]
if include_roles:
- query = query.filter(User.role.in_(include_roles))
+ stmt = stmt.filter(User.role.in_(include_roles))
if exclude_roles:
- query = query.filter(~User.role.in_(exclude_roles))
+ stmt = stmt.filter(~User.role.in_(exclude_roles))
order_by = filter.get('order_by')
direction = filter.get('direction')
@@ -435,99 +442,107 @@ class UsersTable:
group_sort = case((membership_exists, 1), else_=0)
if direction == 'asc':
- query = query.order_by(group_sort.asc(), User.name.asc())
+ stmt = stmt.order_by(group_sort.asc(), User.name.asc())
else:
- query = query.order_by(group_sort.desc(), User.name.asc())
+ stmt = stmt.order_by(group_sort.desc(), User.name.asc())
elif order_by == 'name':
if direction == 'asc':
- query = query.order_by(User.name.asc())
+ stmt = stmt.order_by(User.name.asc())
else:
- query = query.order_by(User.name.desc())
+ stmt = stmt.order_by(User.name.desc())
elif order_by == 'email':
if direction == 'asc':
- query = query.order_by(User.email.asc())
+ stmt = stmt.order_by(User.email.asc())
else:
- query = query.order_by(User.email.desc())
+ stmt = stmt.order_by(User.email.desc())
elif order_by == 'created_at':
if direction == 'asc':
- query = query.order_by(User.created_at.asc())
+ stmt = stmt.order_by(User.created_at.asc())
else:
- query = query.order_by(User.created_at.desc())
+ stmt = stmt.order_by(User.created_at.desc())
elif order_by == 'last_active_at':
if direction == 'asc':
- query = query.order_by(User.last_active_at.asc())
+ stmt = stmt.order_by(User.last_active_at.asc())
else:
- query = query.order_by(User.last_active_at.desc())
+ stmt = stmt.order_by(User.last_active_at.desc())
elif order_by == 'updated_at':
if direction == 'asc':
- query = query.order_by(User.updated_at.asc())
+ stmt = stmt.order_by(User.updated_at.asc())
else:
- query = query.order_by(User.updated_at.desc())
+ stmt = stmt.order_by(User.updated_at.desc())
elif order_by == 'role':
if direction == 'asc':
- query = query.order_by(User.role.asc())
+ stmt = stmt.order_by(User.role.asc())
else:
- query = query.order_by(User.role.desc())
+ stmt = stmt.order_by(User.role.desc())
else:
- query = query.order_by(User.created_at.desc())
+ stmt = stmt.order_by(User.created_at.desc())
# Count BEFORE pagination
- total = query.count()
+ count_result = await db.execute(select(func.count()).select_from(stmt.subquery()))
+ total = count_result.scalar()
# correct pagination logic
if skip is not None:
- query = query.offset(skip)
+ stmt = stmt.offset(skip)
if limit is not None:
- query = query.limit(limit)
+ stmt = stmt.limit(limit)
- users = query.all()
+ result = await db.execute(stmt)
+ users = result.scalars().all()
return {
'users': [UserModel.model_validate(user) for user in users],
'total': total,
}
- def get_users_by_group_id(self, group_id: str, db: Optional[Session] = None) -> list[UserModel]:
- with get_db_context(db) as db:
- users = (
- db.query(User)
- .options(defer(User.profile_image_url))
- .join(GroupMember, User.id == GroupMember.user_id)
- .filter(GroupMember.group_id == group_id)
- .all()
+ async def get_users_by_group_id(self, group_id: str, db: Optional[AsyncSession] = None) -> list[UserModel]:
+ async with get_async_db_context(db) as db:
+ from open_webui.models.groups import GroupMember
+
+ result = await db.execute(
+ select(User).join(GroupMember, User.id == GroupMember.user_id).filter(GroupMember.group_id == group_id)
)
+ users = result.scalars().all()
return [UserModel.model_validate(user) for user in users]
- def get_users_by_user_ids(self, user_ids: list[str], db: Optional[Session] = None) -> list[UserStatusModel]:
- with get_db_context(db) as db:
- users = db.query(User).options(defer(User.profile_image_url)).filter(User.id.in_(user_ids)).all()
+ async def get_users_by_user_ids(
+ self, user_ids: list[str], db: Optional[AsyncSession] = None
+ ) -> list[UserStatusModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter(User.id.in_(user_ids)))
+ users = result.scalars().all()
return [UserModel.model_validate(user) for user in users]
- def get_num_users(self, db: Optional[Session] = None) -> Optional[int]:
- with get_db_context(db) as db:
- return db.query(User).count()
+ async def get_num_users(self, db: Optional[AsyncSession] = None) -> Optional[int]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(func.count()).select_from(User))
+ return result.scalar()
- def has_users(self, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- return db.query(db.query(User).exists()).scalar()
+ async def has_users(self, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(exists(select(User))))
+ return result.scalar()
- def get_first_user(self, db: Optional[Session] = None) -> UserModel:
+ async def get_first_user(self, db: Optional[AsyncSession] = None) -> UserModel:
try:
- with get_db_context(db) as db:
- user = db.query(User).order_by(User.created_at).first()
- return UserModel.model_validate(user)
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).order_by(User.created_at).limit(1))
+ user = result.scalars().first()
+ return UserModel.model_validate(user) if user else None
except Exception:
return None
- def get_user_webhook_url_by_id(self, id: str, db: Optional[Session] = None) -> Optional[str]:
+ async def get_user_webhook_url_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[str]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if user.settings is None:
return None
@@ -536,68 +551,75 @@ class UsersTable:
except Exception:
return None
- def get_num_users_active_today(self, db: Optional[Session] = None) -> Optional[int]:
- with get_db_context(db) as db:
+ async def get_num_users_active_today(self, db: Optional[AsyncSession] = None) -> Optional[int]:
+ async with get_async_db_context(db) as db:
current_timestamp = int(datetime.datetime.now().timestamp())
today_midnight_timestamp = current_timestamp - (current_timestamp % 86400)
- query = db.query(User).filter(User.last_active_at > today_midnight_timestamp)
- return query.count()
+ result = await db.execute(
+ select(func.count()).select_from(User).filter(User.last_active_at > today_midnight_timestamp)
+ )
+ return result.scalar()
- def update_user_role_by_id(self, id: str, role: str, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def update_user_role_by_id(
+ self, id: str, role: str, db: Optional[AsyncSession] = None
+ ) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
user.role = role
- db.commit()
- db.refresh(user)
+ await db.commit()
+ await db.refresh(user)
return UserModel.model_validate(user)
except Exception:
return None
- def update_user_status_by_id(
- self, id: str, form_data: UserStatus, db: Optional[Session] = None
+ async def update_user_status_by_id(
+ self, id: str, form_data: UserStatus, db: Optional[AsyncSession] = None
) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
for key, value in form_data.model_dump(exclude_none=True).items():
setattr(user, key, value)
- db.commit()
- db.refresh(user)
+ await db.commit()
+ await db.refresh(user)
return UserModel.model_validate(user)
except Exception:
return None
- def update_user_profile_image_url_by_id(
- self, id: str, profile_image_url: str, db: Optional[Session] = None
+ async def update_user_profile_image_url_by_id(
+ self, id: str, profile_image_url: str, db: Optional[AsyncSession] = None
) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
user.profile_image_url = profile_image_url
- db.commit()
- db.refresh(user)
+ await db.commit()
+ await db.refresh(user)
return UserModel.model_validate(user)
except Exception:
return None
@throttle(DATABASE_USER_ACTIVE_STATUS_UPDATE_INTERVAL)
- def update_last_active_by_id(self, id: str, db: Optional[Session] = None) -> None:
+ async def update_last_active_by_id(self, id: str, db: Optional[AsyncSession] = None) -> None:
try:
- with get_db_context(db) as db:
- db.query(User).filter_by(id=id).update({'last_active_at': int(time.time())})
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(update(User).filter_by(id=id).values(last_active_at=int(time.time())))
+ await db.commit()
except Exception:
pass
- def update_user_oauth_by_id(
- self, id: str, provider: str, sub: str, db: Optional[Session] = None
+ async def update_user_oauth_by_id(
+ self, id: str, provider: str, sub: str, db: Optional[AsyncSession] = None
) -> Optional[UserModel]:
"""
Update or insert an OAuth provider/sub pair into the user's oauth JSON field.
@@ -608,8 +630,9 @@ class UsersTable:
}
"""
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
@@ -620,20 +643,20 @@ class UsersTable:
oauth[provider] = {'sub': sub}
# Persist updated JSON
- db.query(User).filter_by(id=id).update({'oauth': oauth})
- db.commit()
+ await db.execute(update(User).filter_by(id=id).values(oauth=oauth))
+ await db.commit()
return UserModel.model_validate(user)
except Exception:
return None
- def update_user_scim_by_id(
+ async def update_user_scim_by_id(
self,
id: str,
provider: str,
external_id: str,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> Optional[UserModel]:
"""
Update or insert a SCIM provider/external_id pair into the user's scim JSON field.
@@ -644,41 +667,46 @@ class UsersTable:
}
"""
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
scim = user.scim or {}
scim[provider] = {'external_id': external_id}
- db.query(User).filter_by(id=id).update({'scim': scim})
- db.commit()
+ await db.execute(update(User).filter_by(id=id).values(scim=scim))
+ await db.commit()
return UserModel.model_validate(user)
except Exception:
return None
- def update_user_by_id(self, id: str, updated: dict, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def update_user_by_id(self, id: str, updated: dict, db: Optional[AsyncSession] = None) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
for key, value in updated.items():
setattr(user, key, value)
- db.commit()
- db.refresh(user)
+ await db.commit()
+ await db.refresh(user)
return UserModel.model_validate(user)
except Exception as e:
print(e)
return None
- def update_user_settings_by_id(self, id: str, updated: dict, db: Optional[Session] = None) -> Optional[UserModel]:
+ async def update_user_settings_by_id(
+ self, id: str, updated: dict, db: Optional[AsyncSession] = None
+ ) -> Optional[UserModel]:
try:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
if not user:
return None
@@ -689,26 +717,30 @@ class UsersTable:
user_settings.update(updated)
- db.query(User).filter_by(id=id).update({'settings': user_settings})
- db.commit()
+ await db.execute(update(User).filter_by(id=id).values(settings=user_settings))
+ await db.commit()
- user = db.query(User).filter_by(id=id).first()
+ result = await db.execute(select(User).filter_by(id=id))
+ user = result.scalars().first()
return UserModel.model_validate(user)
except Exception:
return None
- def delete_user_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_user_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
+ from open_webui.models.groups import Groups
+ from open_webui.models.chats import Chats
+
# Remove User from Groups
- Groups.remove_user_from_all_groups(id)
+ await Groups.remove_user_from_all_groups(id)
# Delete User Chats
- result = Chats.delete_chats_by_user_id(id, db=db)
+ result = await Chats.delete_chats_by_user_id(id, db=db)
if result:
- with get_db_context(db) as db:
+ async with get_async_db_context(db) as db:
# Delete User
- db.query(User).filter_by(id=id).delete()
- db.commit()
+ await db.execute(delete(User).filter_by(id=id))
+ await db.commit()
return True
else:
@@ -716,19 +748,20 @@ class UsersTable:
except Exception:
return False
- def get_user_api_key_by_id(self, id: str, db: Optional[Session] = None) -> Optional[str]:
+ async def get_user_api_key_by_id(self, id: str, db: Optional[AsyncSession] = None) -> Optional[str]:
try:
- with get_db_context(db) as db:
- api_key = db.query(ApiKey).filter_by(user_id=id).first()
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(ApiKey).filter_by(user_id=id))
+ api_key = result.scalars().first()
return api_key.key if api_key else None
except Exception:
return None
- def update_user_api_key_by_id(self, id: str, api_key: str, db: Optional[Session] = None) -> bool:
+ async def update_user_api_key_by_id(self, id: str, api_key: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(ApiKey).filter_by(user_id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(ApiKey).filter_by(user_id=id))
+ await db.commit()
now = int(time.time())
new_api_key = ApiKey(
@@ -739,41 +772,45 @@ class UsersTable:
updated_at=now,
)
db.add(new_api_key)
- db.commit()
+ await db.commit()
return True
except Exception:
return False
- def delete_user_api_key_by_id(self, id: str, db: Optional[Session] = None) -> bool:
+ async def delete_user_api_key_by_id(self, id: str, db: Optional[AsyncSession] = None) -> bool:
try:
- with get_db_context(db) as db:
- db.query(ApiKey).filter_by(user_id=id).delete()
- db.commit()
+ async with get_async_db_context(db) as db:
+ await db.execute(delete(ApiKey).filter_by(user_id=id))
+ await db.commit()
return True
except Exception:
return False
- def get_valid_user_ids(self, user_ids: list[str], db: Optional[Session] = None) -> list[str]:
- with get_db_context(db) as db:
- users = db.query(User).filter(User.id.in_(user_ids)).all()
+ async def get_valid_user_ids(self, user_ids: list[str], db: Optional[AsyncSession] = None) -> list[str]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter(User.id.in_(user_ids)))
+ users = result.scalars().all()
return [user.id for user in users]
- def get_super_admin_user(self, db: Optional[Session] = None) -> Optional[UserModel]:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(role='admin').first()
+ async def get_super_admin_user(self, db: Optional[AsyncSession] = None) -> Optional[UserModel]:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(role='admin').limit(1))
+ user = result.scalars().first()
if user:
return UserModel.model_validate(user)
else:
return None
- def get_active_user_count(self, db: Optional[Session] = None) -> int:
- with get_db_context(db) as db:
+ async def get_active_user_count(self, db: Optional[AsyncSession] = None) -> int:
+ async with get_async_db_context(db) as db:
# Consider user active if last_active_at within the last 3 minutes
three_minutes_ago = int(time.time()) - 180
- count = db.query(User).filter(User.last_active_at >= three_minutes_ago).count()
- return count
+ result = await db.execute(
+ select(func.count()).select_from(User).filter(User.last_active_at >= three_minutes_ago)
+ )
+ return result.scalar()
@staticmethod
def is_active(user: UserModel) -> bool:
@@ -783,9 +820,10 @@ class UsersTable:
return user.last_active_at >= three_minutes_ago
return False
- def is_user_active(self, user_id: str, db: Optional[Session] = None) -> bool:
- with get_db_context(db) as db:
- user = db.query(User).filter_by(id=user_id).first()
+ async def is_user_active(self, user_id: str, db: Optional[AsyncSession] = None) -> bool:
+ async with get_async_db_context(db) as db:
+ result = await db.execute(select(User).filter_by(id=user_id))
+ user = result.scalars().first()
if user and user.last_active_at:
# Consider user active if last_active_at within the last 3 minutes
three_minutes_ago = int(time.time()) - 180
diff --git a/backend/open_webui/retrieval/loaders/main.py b/backend/open_webui/retrieval/loaders/main.py
index 747c3ea771..7e5b5774d7 100644
--- a/backend/open_webui/retrieval/loaders/main.py
+++ b/backend/open_webui/retrieval/loaders/main.py
@@ -1,3 +1,4 @@
+import asyncio
import requests
import logging
import ftfy
@@ -238,6 +239,18 @@ class Loader:
return [Document(page_content=ftfy.fix_text(doc.page_content), metadata=doc.metadata) for doc in docs]
+ async def aload(self, filename: str, file_content_type: str, file_path: str) -> list[Document]:
+ """
+ Async wrapper around `load`.
+
+ Document loaders dispatched by `_get_loader` (PyMuPDF, Unstructured,
+ python-docx, Tika, etc.) are uniformly synchronous and CPU/IO-bound.
+ Calling `load` directly from an async handler would block the event
+ loop for the entire parse — minutes for large PDFs. This offloads
+ the work to a worker thread so the loop stays responsive.
+ """
+ return await asyncio.to_thread(self.load, filename, file_content_type, file_path)
+
def _is_text_file(self, file_ext: str, file_content_type: str) -> bool:
return file_ext in known_source_ext or (
file_content_type
diff --git a/backend/open_webui/retrieval/utils.py b/backend/open_webui/retrieval/utils.py
index 4ab8bdf7c0..3638409eee 100644
--- a/backend/open_webui/retrieval/utils.py
+++ b/backend/open_webui/retrieval/utils.py
@@ -20,6 +20,7 @@ from langchain_community.retrievers import BM25Retriever
from langchain_core.documents import Document
from open_webui.config import VECTOR_DB
+from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
@@ -121,7 +122,7 @@ class VectorSearchRetriever(BaseRetriever):
run_manager: CallbackManagerForRetrieverRun,
) -> list[Document]:
embedding = await self.embedding_function(query, RAG_EMBEDDING_QUERY_PREFIX)
- result = VECTOR_DB_CLIENT.search(
+ result = await ASYNC_VECTOR_DB_CLIENT.search(
collection_name=self.collection_name,
vectors=[embedding],
limit=self.top_k,
@@ -488,16 +489,24 @@ async def query_collection_with_hybrid_search(
) -> dict:
results = []
error = False
- # Fetch collection data once per collection sequentially
- # Avoid fetching the same data multiple times later
- collection_results = {}
- for collection_name in collection_names:
+ # Fetch every collection's contents once up front so the
+ # per-query/per-document loop below can reuse them. Each fetch
+ # offloads to a worker thread, so run them concurrently with
+ # `asyncio.gather` instead of awaiting them serially — otherwise
+ # latency scales linearly with `len(collection_names)`.
+ log.debug(
+ 'query_collection_with_hybrid_search: prefetching %d collections',
+ len(collection_names),
+ )
+
+ async def _fetch_collection(name: str):
try:
- log.debug(f'query_collection_with_hybrid_search:VECTOR_DB_CLIENT.get:collection {collection_name}')
- collection_results[collection_name] = VECTOR_DB_CLIENT.get(collection_name=collection_name)
+ return name, await ASYNC_VECTOR_DB_CLIENT.get(collection_name=name)
except Exception as e:
- log.exception(f'Failed to fetch collection {collection_name}: {e}')
- collection_results[collection_name] = None
+ log.exception(f'Failed to fetch collection {name}: {e}')
+ return name, None
+
+ collection_results = dict(await asyncio.gather(*(_fetch_collection(name) for name in collection_names)))
log.info(f'Starting hybrid search for {len(queries)} queries in {len(collection_names)} collections...')
@@ -978,12 +987,12 @@ async def get_sources_from_items(
elif item.get('type') == 'note':
# Note Attached
- note = Notes.get_note_by_id(item.get('id'))
+ note = await Notes.get_note_by_id(item.get('id'))
if note and (
user.role == 'admin'
or note.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -998,7 +1007,7 @@ async def get_sources_from_items(
elif item.get('type') == 'chat':
# Chat Attached
- chat = Chats.get_chat_by_id(item.get('id'))
+ chat = await Chats.get_chat_by_id(item.get('id'))
if chat and (user.role == 'admin' or chat.user_id == user.id):
messages_map = chat.chat.get('history', {}).get('messages', {})
@@ -1042,11 +1051,11 @@ async def get_sources_from_items(
],
}
elif item.get('id'):
- file_object = Files.get_file_by_id(item.get('id'))
+ file_object = await Files.get_file_by_id(item.get('id'))
if file_object and (
user.role == 'admin'
or file_object.user_id == user.id
- or has_access_to_file(item.get('id'), 'read', user)
+ or await has_access_to_file(item.get('id'), 'read', user)
):
query_result = {
'documents': [[file_object.data.get('content', '')]],
@@ -1069,12 +1078,12 @@ async def get_sources_from_items(
elif item.get('type') == 'collection':
# Manual Full Mode Toggle for Collection
- knowledge_base = Knowledges.get_knowledge_by_id(item.get('id'))
+ knowledge_base = await Knowledges.get_knowledge_by_id(item.get('id'))
if knowledge_base and (
user.role == 'admin'
or knowledge_base.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge_base.id,
@@ -1085,14 +1094,14 @@ async def get_sources_from_items(
if knowledge_base and (
user.role == 'admin'
or knowledge_base.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge_base.id,
permission='read',
)
):
- files = Knowledges.get_files_by_id(knowledge_base.id)
+ files = await Knowledges.get_files_by_id(knowledge_base.id)
documents = []
metadatas = []
@@ -1140,7 +1149,9 @@ async def get_sources_from_items(
try:
if full_context:
- query_result = get_all_items_from_collections(collection_names)
+ # Sync helper makes blocking VECTOR_DB_CLIENT calls;
+ # offload so the async caller's event loop stays free.
+ query_result = await asyncio.to_thread(get_all_items_from_collections, collection_names)
else:
query_result = await query_collection(
request,
diff --git a/backend/open_webui/retrieval/vector/async_client.py b/backend/open_webui/retrieval/vector/async_client.py
new file mode 100644
index 0000000000..0bea6696a9
--- /dev/null
+++ b/backend/open_webui/retrieval/vector/async_client.py
@@ -0,0 +1,129 @@
+"""
+Async facade over the synchronous VECTOR_DB_CLIENT.
+
+The vector DB backends bundled with Open WebUI (Chroma, pgvector, Qdrant,
+Milvus, OpenSearch, Pinecone, Weaviate, …) all expose a uniformly
+synchronous API. Each method performs blocking network or disk I/O — and
+some, like `insert`/`upsert`, can run for several seconds.
+
+When such a sync method is awaited from an async route handler, it blocks
+the event loop for its entire duration, freezing every other in-flight
+HTTP request, websocket message and background task.
+
+This module wraps the sync client in an `AsyncVectorDBClient` that
+transparently dispatches each call to a worker thread via
+`asyncio.to_thread`. Async callers can `await ASYNC_VECTOR_DB_CLIENT.x(...)`
+in place of `VECTOR_DB_CLIENT.x(...)` and the loop stays responsive.
+
+The original `VECTOR_DB_CLIENT` is unchanged, so callers already running
+inside `run_in_threadpool` (e.g. `save_docs_to_vector_db`) are not
+affected.
+
+Thread-safety expectations
+--------------------------
+Every async caller now invokes `VECTOR_DB_CLIENT` from a worker thread
+rather than the event-loop thread, and many can run concurrently. The
+sync client (and its underlying backend driver) is therefore expected
+to be safe for concurrent use across threads, which is the standard
+contract for the bundled drivers (chroma, pgvector via SQLAlchemy
+pool, qdrant-client, opensearch-py, …). This is *not* a new exposure
+introduced by this facade — `save_docs_to_vector_db` already called
+the sync client from `run_in_threadpool`, so concurrent threaded
+access has always been a requirement of the codebase. Adding a global
+serialization lock here would defeat the responsiveness this facade
+exists to provide; any backend that genuinely cannot tolerate
+concurrent access should grow its own internal serialization.
+
+API surface
+-----------
+Method signatures mirror `VectorDBBase` exactly. This is deliberate:
+permissive `*args/**kwargs` forwarding hides typos at the call site
+(an earlier revision of this file shipped that, and a `metadata=`
+typo silently broke an entire endpoint until explicit signatures
+surfaced it). Callers that need a backend-specific parameter not on
+`VectorDBBase` should reach for the `.sync` escape hatch and wrap
+their own `asyncio.to_thread`, e.g. ::
+
+ await asyncio.to_thread(
+ ASYNC_VECTOR_DB_CLIENT.sync.some_backend_specific_op,
+ collection_name, special_kwarg=value,
+ )
+"""
+
+from __future__ import annotations
+
+import asyncio
+from typing import Dict, List, Optional, Union
+
+from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+from open_webui.retrieval.vector.main import (
+ GetResult,
+ SearchResult,
+ VectorDBBase,
+ VectorItem,
+)
+
+
+class AsyncVectorDBClient:
+ """Awaitable mirror of `VectorDBBase` that off-loads each call to a thread.
+
+ Method signatures mirror `VectorDBBase` exactly so static analysis
+ catches bad kwargs at the call site instead of letting them surface
+ deep inside the worker thread (where the resulting ``TypeError`` is
+ typically swallowed by surrounding ``try/except``).
+ """
+
+ def __init__(self, sync_client: VectorDBBase) -> None:
+ self._sync = sync_client
+
+ @property
+ def sync(self) -> VectorDBBase:
+ """Escape hatch for code that must call the sync client directly
+ (e.g. already inside a worker thread)."""
+ return self._sync
+
+ async def has_collection(self, collection_name: str) -> bool:
+ return await asyncio.to_thread(self._sync.has_collection, collection_name)
+
+ async def delete_collection(self, collection_name: str) -> None:
+ return await asyncio.to_thread(self._sync.delete_collection, collection_name)
+
+ async def insert(self, collection_name: str, items: List[VectorItem]) -> None:
+ return await asyncio.to_thread(self._sync.insert, collection_name, items)
+
+ async def upsert(self, collection_name: str, items: List[VectorItem]) -> None:
+ return await asyncio.to_thread(self._sync.upsert, collection_name, items)
+
+ async def search(
+ self,
+ collection_name: str,
+ vectors: List[List[Union[float, int]]],
+ filter: Optional[Dict] = None,
+ limit: int = 10,
+ ) -> Optional[SearchResult]:
+ return await asyncio.to_thread(self._sync.search, collection_name, vectors, filter, limit)
+
+ async def query(
+ self,
+ collection_name: str,
+ filter: Dict,
+ limit: Optional[int] = None,
+ ) -> Optional[GetResult]:
+ return await asyncio.to_thread(self._sync.query, collection_name, filter, limit)
+
+ async def get(self, collection_name: str) -> Optional[GetResult]:
+ return await asyncio.to_thread(self._sync.get, collection_name)
+
+ async def delete(
+ self,
+ collection_name: str,
+ ids: Optional[List[str]] = None,
+ filter: Optional[Dict] = None,
+ ) -> None:
+ return await asyncio.to_thread(self._sync.delete, collection_name, ids, filter)
+
+ async def reset(self) -> None:
+ return await asyncio.to_thread(self._sync.reset)
+
+
+ASYNC_VECTOR_DB_CLIENT = AsyncVectorDBClient(VECTOR_DB_CLIENT)
diff --git a/backend/open_webui/retrieval/vector/dbs/pgvector.py b/backend/open_webui/retrieval/vector/dbs/pgvector.py
index 4775ff21f4..90e65b9ad0 100644
--- a/backend/open_webui/retrieval/vector/dbs/pgvector.py
+++ b/backend/open_webui/retrieval/vector/dbs/pgvector.py
@@ -34,6 +34,7 @@ from open_webui.retrieval.vector.main import (
SearchResult,
GetResult,
)
+from open_webui.utils.misc import sanitize_text_for_db
from open_webui.config import (
PGVECTOR_DB_URL,
PGVECTOR_INITIALIZE_MAX_VECTOR_LENGTH,
@@ -289,7 +290,9 @@ class PgvectorClient(VectorDBBase):
vector = self.adjust_vector_length(item['vector'])
# Use raw SQL for BYTEA/pgcrypto
# Ensure metadata is converted to its JSON text representation
- json_metadata = json.dumps(item['metadata'])
+ # Sanitize to strip null bytes / surrogates that PostgreSQL cannot store
+ json_metadata = sanitize_text_for_db(json.dumps(item['metadata']))
+ item_text = sanitize_text_for_db(item['text'])
self.session.execute(
text("""
INSERT INTO document_chunk
@@ -305,7 +308,7 @@ class PgvectorClient(VectorDBBase):
'id': item['id'],
'vector': vector,
'collection_name': collection_name,
- 'text': item['text'],
+ 'text': item_text,
'metadata_text': json_metadata,
'key': PGVECTOR_PGCRYPTO_KEY,
},
@@ -338,7 +341,9 @@ class PgvectorClient(VectorDBBase):
if PGVECTOR_PGCRYPTO:
for item in items:
vector = self.adjust_vector_length(item['vector'])
- json_metadata = json.dumps(item['metadata'])
+ # Sanitize to strip null bytes / surrogates that PostgreSQL cannot store
+ json_metadata = sanitize_text_for_db(json.dumps(item['metadata']))
+ item_text = sanitize_text_for_db(item['text'])
self.session.execute(
text("""
INSERT INTO document_chunk
@@ -358,7 +363,7 @@ class PgvectorClient(VectorDBBase):
'id': item['id'],
'vector': vector,
'collection_name': collection_name,
- 'text': item['text'],
+ 'text': item_text,
'metadata_text': json_metadata,
'key': PGVECTOR_PGCRYPTO_KEY,
},
diff --git a/backend/open_webui/retrieval/vector/utils.py b/backend/open_webui/retrieval/vector/utils.py
index b2e2fed762..4915b024c3 100644
--- a/backend/open_webui/retrieval/vector/utils.py
+++ b/backend/open_webui/retrieval/vector/utils.py
@@ -1,5 +1,7 @@
from datetime import datetime
+from open_webui.utils.misc import sanitize_text_for_db
+
KEYS_TO_EXCLUDE = ['content', 'pages', 'tables', 'paragraphs', 'sections', 'figures']
@@ -12,7 +14,8 @@ def filter_metadata(metadata: dict[str, any]) -> dict[str, any]:
def process_metadata(
metadata: dict[str, any],
) -> dict[str, any]:
- # Removes large fields and converts non-serializable types (datetime, list, dict) to strings.
+ # Removes large fields, converts non-serializable types (datetime, list, dict) to strings,
+ # and sanitizes strings for database storage (strips null bytes and invalid surrogates).
result = {}
for key, value in metadata.items():
# Skip large fields
@@ -20,7 +23,7 @@ def process_metadata(
continue
# Convert non-serializable fields to strings
if isinstance(value, (datetime, list, dict)):
- result[key] = str(value)
+ result[key] = sanitize_text_for_db(str(value))
else:
- result[key] = value
+ result[key] = sanitize_text_for_db(value)
return result
diff --git a/backend/open_webui/retrieval/web/firecrawl.py b/backend/open_webui/retrieval/web/firecrawl.py
index 4bb23e3797..4af302c0de 100644
--- a/backend/open_webui/retrieval/web/firecrawl.py
+++ b/backend/open_webui/retrieval/web/firecrawl.py
@@ -1,6 +1,7 @@
import logging
from typing import Optional, List
+import requests
from open_webui.retrieval.web.main import SearchResult, get_filtered_results
log = logging.getLogger(__name__)
@@ -14,23 +15,38 @@ def search_firecrawl(
filter_list: Optional[List[str]] = None,
) -> List[SearchResult]:
try:
- from firecrawl import FirecrawlApp
+ url = firecrawl_url.rstrip('/')
+ response = requests.post(
+ f'{url}/v1/search',
+ headers={
+ 'Content-Type': 'application/json',
+ 'Authorization': f'Bearer {firecrawl_api_key}',
+ },
+ json={
+ 'query': query,
+ 'limit': count,
+ 'timeout': count * 3000,
+ },
+ timeout=count * 3 + 10,
+ )
+ response.raise_for_status()
+ data = response.json().get('data', {})
- firecrawl = FirecrawlApp(api_key=firecrawl_api_key, api_url=firecrawl_url)
- response = firecrawl.search(query=query, limit=count, ignore_invalid_urls=True, timeout=count * 3)
- results = response.web
- if filter_list:
- results = get_filtered_results(results, filter_list)
results = [
SearchResult(
- link=result.url,
- title=result.title,
- snippet=result.description,
+ link=r.get('url', ''),
+ title=r.get('title', ''),
+ snippet=r.get('description', ''),
)
- for result in results[:count]
+ for r in data.get('web', [])
]
- log.info(f'External search results: {results}')
+
+ if filter_list:
+ results = get_filtered_results(results, filter_list)
+
+ results = results[:count]
+ log.info(f'FireCrawl search results: {results}')
return results
except Exception as e:
- log.error(f'Error in External search: {e}')
+ log.error(f'Error in FireCrawl search: {e}')
return []
diff --git a/backend/open_webui/retrieval/web/utils.py b/backend/open_webui/retrieval/web/utils.py
index c9442f208b..cfe0f71b85 100644
--- a/backend/open_webui/retrieval/web/utils.py
+++ b/backend/open_webui/retrieval/web/utils.py
@@ -1,4 +1,5 @@
import asyncio
+import ipaddress
import logging
import socket
import ssl
@@ -84,11 +85,9 @@ def validate_url(url: Union[str, Sequence[str]]):
ipv4_addresses, ipv6_addresses = resolve_hostname(parsed_url.hostname)
# Check if any of the resolved addresses are private
# This is technically still vulnerable to DNS rebinding attacks, as we don't control WebBaseLoader
- for ip in ipv4_addresses:
- if validators.ipv4(ip, private=True):
- raise ValueError(ERROR_MESSAGES.INVALID_URL)
- for ip in ipv6_addresses:
- if validators.ipv6(ip, private=True):
+ for ip in ipv4_addresses + ipv6_addresses:
+ addr = ipaddress.ip_address(ip)
+ if not addr.is_global:
raise ValueError(ERROR_MESSAGES.INVALID_URL)
return True
elif isinstance(url, Sequence):
@@ -193,27 +192,6 @@ class SafeFireCrawlLoader(BaseLoader, RateLimitMixin, URLProcessingMixin):
proxy: Optional[Dict[str, str]] = None,
params: Optional[Dict] = None,
):
- """Concurrent document loader for FireCrawl operations.
-
- Executes multiple FireCrawlLoader instances concurrently using thread pooling
- to improve bulk processing efficiency.
- Args:
- web_paths: List of URLs/paths to process.
- verify_ssl: If True, verify SSL certificates.
- trust_env: If True, use proxy settings from environment variables.
- requests_per_second: Number of requests per second to limit to.
- continue_on_failure (bool): If True, continue loading other URLs on failure.
- api_key: API key for FireCrawl service. Defaults to None
- (uses FIRE_CRAWL_API_KEY environment variable if not provided).
- api_url: Base URL for FireCrawl API. Defaults to official API endpoint.
- mode: Operation mode selection:
- - 'crawl': Website crawling mode
- - 'scrape': Direct page scraping (default)
- - 'map': Site map generation
- proxy: Proxy override settings for the FireCrawl API.
- params: The parameters to pass to the Firecrawl API.
- For more details, visit: https://docs.firecrawl.dev/sdks/python#batch-scrape
- """
proxy_server = proxy.get('server') if proxy else None
if trust_env and not proxy_server:
env_proxies = urllib.request.getproxies()
@@ -230,44 +208,43 @@ class SafeFireCrawlLoader(BaseLoader, RateLimitMixin, URLProcessingMixin):
self.trust_env = trust_env
self.continue_on_failure = continue_on_failure
self.api_key = api_key
- self.api_url = api_url
+ self.api_url = (api_url or 'https://api.firecrawl.dev').rstrip('/')
self.timeout = timeout
self.mode = mode
self.params = params or {}
def lazy_load(self) -> Iterator[Document]:
- """Load documents using FireCrawl batch_scrape."""
- log.debug(
- 'Starting FireCrawl batch scrape for %d URLs, mode: %s, params: %s',
- len(self.web_paths),
- self.mode,
- self.params,
- )
try:
- from firecrawl import FirecrawlApp
+ headers = {
+ 'Content-Type': 'application/json',
+ 'Authorization': f'Bearer {self.api_key}',
+ }
- firecrawl = FirecrawlApp(api_key=self.api_key, api_url=self.api_url)
- result = firecrawl.batch_scrape(
- self.web_paths,
- formats=['markdown'],
- skip_tls_verification=not self.verify_ssl,
- ignore_invalid_urls=True,
- remove_base64_images=True,
- max_age=300000, # 5 minutes https://docs.firecrawl.dev/features/fast-scraping#common-maxage-values
- wait_timeout=self.timeout if self.timeout else len(self.web_paths) * 3,
- **self.params,
- )
+ for url in self.web_paths:
+ payload = {
+ 'url': url,
+ 'formats': ['markdown'],
+ **self.params,
+ }
+ if self.timeout:
+ payload['timeout'] = self.timeout * 1000
- if result.status != 'completed':
- raise RuntimeError(f'FireCrawl batch scrape did not complete successfully. result: {result}')
-
- for data in result.data:
- metadata = data.metadata or {}
- yield Document(
- page_content=data.markdown or '',
- metadata={'source': metadata.url or metadata.source_url or ''},
+ response = requests.post(
+ f'{self.api_url}/v1/scrape',
+ headers=headers,
+ json=payload,
+ timeout=self.timeout or 60,
+ verify=self.verify_ssl,
)
+ response.raise_for_status()
+ data = response.json().get('data', {})
+ metadata = data.get('metadata', {})
+ source = metadata.get('url') or metadata.get('sourceURL') or url
+ yield Document(
+ page_content=data.get('markdown', ''),
+ metadata={'source': source},
+ )
except Exception as e:
if self.continue_on_failure:
log.exception(f'Error extracting content from URLs: {e}')
@@ -275,38 +252,10 @@ class SafeFireCrawlLoader(BaseLoader, RateLimitMixin, URLProcessingMixin):
raise e
async def alazy_load(self):
- """Async version of lazy_load."""
- log.debug(
- 'Starting FireCrawl batch scrape for %d URLs, mode: %s, params: %s',
- len(self.web_paths),
- self.mode,
- self.params,
- )
try:
- from firecrawl import FirecrawlApp
-
- firecrawl = FirecrawlApp(api_key=self.api_key, api_url=self.api_url)
- result = firecrawl.batch_scrape(
- self.web_paths,
- formats=['markdown'],
- skip_tls_verification=not self.verify_ssl,
- ignore_invalid_urls=True,
- remove_base64_images=True,
- max_age=300000, # 5 minutes https://docs.firecrawl.dev/features/fast-scraping#common-maxage-values
- wait_timeout=self.timeout if self.timeout else len(self.web_paths) * 3,
- **self.params,
- )
-
- if result.status != 'completed':
- raise RuntimeError(f'FireCrawl batch scrape did not complete successfully. result: {result}')
-
- for data in result.data:
- metadata = data.metadata or {}
- yield Document(
- page_content=data.markdown or '',
- metadata={'source': metadata.url or metadata.source_url or ''},
- )
-
+ docs = await run_in_threadpool(lambda: list(self.lazy_load()))
+ for doc in docs:
+ yield doc
except Exception as e:
if self.continue_on_failure:
log.exception(f'Error extracting content from URLs: {e}')
diff --git a/backend/open_webui/routers/analytics.py b/backend/open_webui/routers/analytics.py
index 790c134295..fd045f79e7 100644
--- a/backend/open_webui/routers/analytics.py
+++ b/backend/open_webui/routers/analytics.py
@@ -11,8 +11,8 @@ from open_webui.models.groups import Groups
from open_webui.models.users import Users
from open_webui.models.feedbacks import Feedbacks
from open_webui.utils.auth import get_admin_user
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
@@ -59,10 +59,12 @@ async def get_model_analytics(
end_date: Optional[int] = Query(None, description='End timestamp (epoch)'),
group_id: Optional[str] = Query(None, description='Filter by user group ID'),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get message counts per model."""
- counts = ChatMessages.get_message_count_by_model(start_date=start_date, end_date=end_date, group_id=group_id, db=db)
+ counts = await ChatMessages.get_message_count_by_model(
+ start_date=start_date, end_date=end_date, group_id=group_id, db=db
+ )
models = [
ModelAnalyticsEntry(model_id=model_id, count=count)
for model_id, count in sorted(counts.items(), key=lambda x: -x[1])
@@ -77,17 +79,19 @@ async def get_user_analytics(
group_id: Optional[str] = Query(None, description='Filter by user group ID'),
limit: int = Query(50, description='Max users to return'),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get message counts and token usage per user with user info."""
- counts = ChatMessages.get_message_count_by_user(start_date=start_date, end_date=end_date, group_id=group_id, db=db)
- token_usage = ChatMessages.get_token_usage_by_user(
+ counts = await ChatMessages.get_message_count_by_user(
+ start_date=start_date, end_date=end_date, group_id=group_id, db=db
+ )
+ token_usage = await ChatMessages.get_token_usage_by_user(
start_date=start_date, end_date=end_date, group_id=group_id, db=db
)
# Get user info for top users
top_user_ids = [uid for uid, _ in sorted(counts.items(), key=lambda x: -x[1])[:limit]]
- user_info = {u.id: u for u in Users.get_users_by_user_ids(top_user_ids, db=db)}
+ user_info = {u.id: u for u in await Users.get_users_by_user_ids(top_user_ids, db=db)}
users = []
for user_id in top_user_ids:
@@ -118,13 +122,13 @@ async def get_messages(
skip: int = Query(0),
limit: int = Query(50, le=100),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Query messages with filters."""
if chat_id:
- return ChatMessages.get_messages_by_chat_id(chat_id=chat_id, db=db)
+ return await ChatMessages.get_messages_by_chat_id(chat_id=chat_id, db=db)
elif model_id:
- return ChatMessages.get_messages_by_model_id(
+ return await ChatMessages.get_messages_by_model_id(
model_id=model_id,
start_date=start_date,
end_date=end_date,
@@ -133,7 +137,7 @@ async def get_messages(
db=db,
)
elif user_id:
- return ChatMessages.get_messages_by_user_id(user_id=user_id, skip=skip, limit=limit, db=db)
+ return await ChatMessages.get_messages_by_user_id(user_id=user_id, skip=skip, limit=limit, db=db)
else:
# Return empty if no filter specified
return []
@@ -152,16 +156,16 @@ async def get_summary(
end_date: Optional[int] = Query(None, description='End timestamp (epoch)'),
group_id: Optional[str] = Query(None, description='Filter by user group ID'),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get summary statistics for the dashboard."""
- model_counts = ChatMessages.get_message_count_by_model(
+ model_counts = await ChatMessages.get_message_count_by_model(
start_date=start_date, end_date=end_date, group_id=group_id, db=db
)
- user_counts = ChatMessages.get_message_count_by_user(
+ user_counts = await ChatMessages.get_message_count_by_user(
start_date=start_date, end_date=end_date, group_id=group_id, db=db
)
- chat_counts = ChatMessages.get_message_count_by_chat(
+ chat_counts = await ChatMessages.get_message_count_by_chat(
start_date=start_date, end_date=end_date, group_id=group_id, db=db
)
@@ -189,13 +193,13 @@ async def get_daily_stats(
group_id: Optional[str] = Query(None, description='Filter by user group ID'),
granularity: str = Query('daily', description="Granularity: 'hourly' or 'daily'"),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get message counts grouped by model for time-series chart."""
if granularity == 'hourly':
- counts = ChatMessages.get_hourly_message_counts_by_model(start_date=start_date, end_date=end_date, db=db)
+ counts = await ChatMessages.get_hourly_message_counts_by_model(start_date=start_date, end_date=end_date, db=db)
else:
- counts = ChatMessages.get_daily_message_counts_by_model(
+ counts = await ChatMessages.get_daily_message_counts_by_model(
start_date=start_date, end_date=end_date, group_id=group_id, db=db
)
return DailyStatsResponse(
@@ -224,10 +228,12 @@ async def get_token_usage(
end_date: Optional[int] = Query(None),
group_id: Optional[str] = Query(None, description='Filter by user group ID'),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get token usage aggregated by model."""
- usage = ChatMessages.get_token_usage_by_model(start_date=start_date, end_date=end_date, group_id=group_id, db=db)
+ usage = await ChatMessages.get_token_usage_by_model(
+ start_date=start_date, end_date=end_date, group_id=group_id, db=db
+ )
models = [
TokenUsageEntry(model_id=model_id, **data)
@@ -271,12 +277,12 @@ async def get_model_chats(
skip: int = Query(0),
limit: int = Query(50, le=100),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get chats that used a specific model, with preview and feedback info."""
# Get chat IDs that used this model
- chat_ids = ChatMessages.get_chat_ids_by_model_id(
+ chat_ids = await ChatMessages.get_chat_ids_by_model_id(
model_id=model_id,
start_date=start_date,
end_date=end_date,
@@ -291,7 +297,7 @@ async def get_model_chats(
# Get chat details from messages only
chats_data = []
for chat_id in chat_ids:
- messages = ChatMessages.get_messages_by_chat_id(chat_id, db=db)
+ messages = await ChatMessages.get_messages_by_chat_id(chat_id, db=db)
if not messages:
continue
@@ -312,7 +318,7 @@ async def get_model_chats(
# Get user info
user_name = None
if user_id:
- user_info = Users.get_user_by_id(user_id, db=db)
+ user_info = await Users.get_user_by_id(user_id, db=db)
user_name = user_info.name if user_info else None
# Timestamps from messages
@@ -357,12 +363,12 @@ async def get_model_overview(
model_id: str,
days: int = Query(30, description='Number of days of history (0 for all)'),
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get model overview with feedback history and chat tags."""
# Get chat IDs that used this model
- chat_ids = ChatMessages.get_chat_ids_by_model_id(
+ chat_ids = await ChatMessages.get_chat_ids_by_model_id(
model_id=model_id,
start_date=None,
end_date=None,
@@ -381,7 +387,7 @@ async def get_model_overview(
start_dt = now - timedelta(days=days)
for chat_id in chat_ids:
- feedbacks = Feedbacks.get_feedbacks_by_chat_id(chat_id, db=db)
+ feedbacks = await Feedbacks.get_feedbacks_by_chat_id(chat_id, db=db)
for fb in feedbacks:
if fb.data and 'rating' in fb.data:
rating = fb.data['rating']
@@ -425,7 +431,7 @@ async def get_model_overview(
# Get chat tags
tag_counts: dict[str, int] = defaultdict(int)
for chat_id in chat_ids:
- chat = Chats.get_chat_by_id(chat_id, db=db)
+ chat = await Chats.get_chat_by_id(chat_id, db=db)
if chat and chat.meta:
for tag in chat.meta.get('tags', []):
tag_counts[tag] += 1
diff --git a/backend/open_webui/routers/audio.py b/backend/open_webui/routers/audio.py
index 9d8938b419..499c876261 100644
--- a/backend/open_webui/routers/audio.py
+++ b/backend/open_webui/routers/audio.py
@@ -54,6 +54,7 @@ from open_webui.env import (
AIOHTTP_CLIENT_SESSION_SSL,
AIOHTTP_CLIENT_TIMEOUT,
AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST,
+ BYPASS_PYDUB_PREPROCESSING,
DEVICE_TYPE,
ENABLE_FORWARD_USER_INFO_HEADERS,
)
@@ -330,7 +331,9 @@ async def speech(request: Request, user=Depends(get_verified_user)):
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if user.role != 'admin' and not has_permission(user.id, 'chat.tts', request.app.state.config.USER_PERMISSIONS):
+ if user.role != 'admin' and not await has_permission(
+ user.id, 'chat.tts', request.app.state.config.USER_PERMISSIONS
+ ):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -630,6 +633,7 @@ async def speech(request: Request, user=Depends(get_verified_user)):
detail=detail,
)
+
def transcription_handler(request, file_path, metadata, user=None):
filename = os.path.basename(file_path)
file_dir = os.path.dirname(file_path)
@@ -660,7 +664,7 @@ def transcription_handler(request, file_path, metadata, user=None):
data = {'text': transcript.strip()}
# save the transcript to a json file
- transcript_file = f'{file_dir}/{id}.json'
+ transcript_file = os.path.join(file_dir, f'{id}.json')
with open(transcript_file, 'w') as f:
json.dump(data, f)
@@ -698,7 +702,7 @@ def transcription_handler(request, file_path, metadata, user=None):
data = r.json()
# save the transcript to a json file
- transcript_file = f'{file_dir}/{id}.json'
+ transcript_file = os.path.join(file_dir, f'{id}.json')
with open(transcript_file, 'w') as f:
json.dump(data, f)
@@ -767,7 +771,7 @@ def transcription_handler(request, file_path, metadata, user=None):
data = {'text': transcript.strip()}
# Save transcript
- transcript_file = f'{file_dir}/{id}.json'
+ transcript_file = os.path.join(file_dir, f'{id}.json')
with open(transcript_file, 'w') as f:
json.dump(data, f)
@@ -874,7 +878,7 @@ def transcription_handler(request, file_path, metadata, user=None):
data = {'text': transcript}
# Save transcript to json file (consistent with other providers)
- transcript_file = f'{file_dir}/{id}.json'
+ transcript_file = os.path.join(file_dir, f'{id}.json')
with open(transcript_file, 'w') as f:
json.dump(data, f)
@@ -1059,7 +1063,7 @@ def transcription_handler(request, file_path, metadata, user=None):
data = {'text': transcript}
# Save transcript to json file (consistent with other providers)
- transcript_file = f'{file_dir}/{id}.json'
+ transcript_file = os.path.join(file_dir, f'{id}.json')
with open(transcript_file, 'w') as f:
json.dump(data, f)
@@ -1095,24 +1099,28 @@ def transcription_handler(request, file_path, metadata, user=None):
def transcribe(request: Request, file_path: str, metadata: Optional[dict] = None, user=None):
log.info(f'transcribe: {file_path} {metadata}')
- if is_audio_conversion_required(file_path):
- file_path = convert_audio_to_mp3(file_path)
+ if BYPASS_PYDUB_PREPROCESSING:
+ log.info('Bypassing pydub preprocessing (BYPASS_PYDUB_PREPROCESSING=true)')
+ chunk_paths = [file_path]
+ else:
+ if is_audio_conversion_required(file_path):
+ file_path = convert_audio_to_mp3(file_path)
- try:
- file_path = compress_audio(file_path)
- except Exception as e:
- log.exception(e)
+ try:
+ file_path = compress_audio(file_path)
+ except Exception as e:
+ log.exception(e)
- # Always produce a list of chunk paths (could be one entry if small)
- try:
- chunk_paths = split_audio(file_path, MAX_FILE_SIZE)
- print(f'Chunk paths: {chunk_paths}')
- except Exception as e:
- log.exception(e)
- raise HTTPException(
- status_code=status.HTTP_400_BAD_REQUEST,
- detail=ERROR_MESSAGES.DEFAULT(e),
- )
+ # Always produce a list of chunk paths (could be one entry if small)
+ try:
+ chunk_paths = split_audio(file_path, MAX_FILE_SIZE)
+ print(f'Chunk paths: {chunk_paths}')
+ except Exception as e:
+ log.exception(e)
+ raise HTTPException(
+ status_code=status.HTTP_400_BAD_REQUEST,
+ detail=ERROR_MESSAGES.DEFAULT(e),
+ )
results = []
try:
@@ -1208,13 +1216,15 @@ def split_audio(file_path, max_bytes, format='mp3', bitrate='32k'):
@router.post('/transcriptions')
-def transcription(
+async def transcription(
request: Request,
file: UploadFile = File(...),
language: Optional[str] = Form(None),
user=Depends(get_verified_user),
):
- if user.role != 'admin' and not has_permission(user.id, 'chat.stt', request.app.state.config.USER_PERMISSIONS):
+ if user.role != 'admin' and not await has_permission(
+ user.id, 'chat.stt', request.app.state.config.USER_PERMISSIONS
+ ):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -1237,9 +1247,9 @@ def transcription(
filename = f'{id}.{ext}'
contents = file.file.read()
- file_dir = f'{CACHE_DIR}/audio/transcriptions'
+ file_dir = os.path.join(CACHE_DIR, 'audio', 'transcriptions')
os.makedirs(file_dir, exist_ok=True)
- file_path = f'{file_dir}/{filename}'
+ file_path = os.path.join(file_dir, filename)
# Defense-in-depth: ensure resolved path stays within intended directory
if not os.path.realpath(file_path).startswith(os.path.realpath(file_dir)):
diff --git a/backend/open_webui/routers/auths.py b/backend/open_webui/routers/auths.py
index 88f0fe69fb..651e123b64 100644
--- a/backend/open_webui/routers/auths.py
+++ b/backend/open_webui/routers/auths.py
@@ -54,6 +54,7 @@ from open_webui.config import (
OAUTH_PROVIDERS,
OAUTH_MERGE_ACCOUNTS_BY_EMAIL,
)
+from open_webui.utils.oauth import auth_manager_config
from pydantic import BaseModel
from open_webui.utils.misc import parse_duration, validate_email_format
@@ -70,8 +71,8 @@ from open_webui.utils.auth import (
get_password_hash,
get_http_authorization_cred,
)
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from open_webui.utils.webhook import post_webhook
from open_webui.utils.access_control import get_permissions, has_permission
from open_webui.utils.groups import apply_default_group_assignment
@@ -96,7 +97,9 @@ log = logging.getLogger(__name__)
signin_rate_limiter = RateLimiter(redis_client=get_redis_client(), limit=5 * 3, window=60 * 3)
-def create_session_response(request: Request, user, db, response: Response = None, set_cookie: bool = False) -> dict:
+async def create_session_response(
+ request: Request, user, db, response: Response = None, set_cookie: bool = False
+) -> dict:
"""
Create JWT token and build session response for a user.
Shared helper for signin, signup, ldap_auth, add_user, and token_exchange endpoints.
@@ -131,7 +134,7 @@ def create_session_response(request: Request, user, db, response: Response = Non
**({'max_age': max_age} if max_age is not None else {}),
)
- user_permissions = get_permissions(user.id, request.app.state.config.USER_PERMISSIONS, db=db)
+ user_permissions = await get_permissions(user.id, request.app.state.config.USER_PERMISSIONS, db=db)
return {
'token': token,
@@ -167,7 +170,7 @@ async def get_session_user(
request: Request,
response: Response,
user=Depends(get_current_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
auth_header = request.headers.get('Authorization')
auth_token = get_http_authorization_cred(auth_header)
@@ -197,7 +200,7 @@ async def get_session_user(
**({'max_age': max_age} if max_age is not None else {}),
)
- user_permissions = get_permissions(user.id, request.app.state.config.USER_PERMISSIONS, db=db)
+ user_permissions = await get_permissions(user.id, request.app.state.config.USER_PERMISSIONS, db=db)
return {
'token': token,
@@ -227,10 +230,10 @@ async def get_session_user(
async def update_profile(
form_data: UpdateProfileForm,
session_user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if session_user:
- user = Users.update_user_by_id(
+ user = await Users.update_user_by_id(
session_user.id,
form_data.model_dump(),
db=db,
@@ -256,10 +259,10 @@ class UpdateTimezoneForm(BaseModel):
async def update_timezone(
form_data: UpdateTimezoneForm,
session_user=Depends(get_current_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if session_user:
- Users.update_user_by_id(
+ await Users.update_user_by_id(
session_user.id,
{'timezone': form_data.timezone},
db=db,
@@ -278,12 +281,12 @@ async def update_timezone(
async def update_password(
form_data: UpdatePasswordForm,
session_user=Depends(get_current_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if WEBUI_AUTH_TRUSTED_EMAIL_HEADER:
raise HTTPException(400, detail=ERROR_MESSAGES.ACTION_PROHIBITED)
if session_user:
- user = Auths.authenticate_user(
+ user = await Auths.authenticate_user(
session_user.email,
lambda pw: verify_password(form_data.password, pw),
db=db,
@@ -295,7 +298,7 @@ async def update_password(
except Exception as e:
raise HTTPException(400, detail=str(e))
hashed = get_password_hash(form_data.new_password)
- return Auths.update_user_password_by_id(user.id, hashed, db=db)
+ return await Auths.update_user_password_by_id(user.id, hashed, db=db)
else:
raise HTTPException(400, detail=ERROR_MESSAGES.INCORRECT_PASSWORD)
else:
@@ -310,7 +313,7 @@ async def ldap_auth(
request: Request,
response: Response,
form_data: LdapForm,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
# Security checks FIRST - before loading any config
if not request.app.state.config.ENABLE_LDAP:
@@ -322,6 +325,14 @@ async def ldap_auth(
detail=ERROR_MESSAGES.ACTION_PROHIBITED,
)
+ # Reject empty passwords before attempting the LDAP bind.
+ # Per RFC 4513 §5.1.2, a Simple Bind with a non-empty DN but empty
+ # password is "unauthenticated simple authentication" — many LDAP
+ # servers (OpenLDAP default, some AD configs) return success for these,
+ # which would grant access without valid credentials.
+ if not form_data.password or not form_data.password.strip():
+ raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)
+
# NOW load LDAP config variables
LDAP_SERVER_LABEL = request.app.state.config.LDAP_SERVER_LABEL
LDAP_SERVER_HOST = request.app.state.config.LDAP_SERVER_HOST
@@ -476,23 +487,29 @@ async def ldap_auth(
if not await asyncio.to_thread(connection_user.bind):
raise HTTPException(400, 'Authentication failed.')
- user = Users.get_user_by_email(email, db=db)
+ user = await Users.get_user_by_email(email, db=db)
if not user:
try:
- role = 'admin' if not Users.has_users(db=db) else request.app.state.config.DEFAULT_USER_ROLE
-
- user = Auths.insert_new_auth(
+ # Insert with default role first to avoid TOCTOU race on
+ # first-user registration. Matches signup_handler pattern.
+ user = await Auths.insert_new_auth(
email=email,
password=str(uuid.uuid4()),
name=cn,
- role=role,
+ role=request.app.state.config.DEFAULT_USER_ROLE,
db=db,
)
if not user:
raise HTTPException(500, detail=ERROR_MESSAGES.CREATE_USER_ERROR)
- apply_default_group_assignment(
+ # Atomically check if this is the only user *after* the
+ # insert. Only the single user present should become admin.
+ if await Users.get_num_users(db=db) == 1:
+ await Users.update_user_role_by_id(user.id, 'admin', db=db)
+ user = await Users.get_user_by_id(user.id, db=db)
+
+ await apply_default_group_assignment(
request.app.state.config.DEFAULT_GROUP_ID,
user.id,
db=db,
@@ -504,19 +521,19 @@ async def ldap_auth(
log.error(f'LDAP user creation error: {str(err)}')
raise HTTPException(500, detail='Internal error occurred during LDAP user creation.')
- user = Auths.authenticate_user_by_email(email, db=db)
+ user = await Auths.authenticate_user_by_email(email, db=db)
if user:
if ENABLE_LDAP_GROUP_MANAGEMENT and user_groups:
if ENABLE_LDAP_GROUP_CREATION:
- Groups.create_groups_by_group_names(user.id, user_groups, db=db)
+ await Groups.create_groups_by_group_names(user.id, user_groups, db=db)
try:
- Groups.sync_groups_by_group_names(user.id, user_groups, db=db)
+ await Groups.sync_groups_by_group_names(user.id, user_groups, db=db)
log.info(f'Successfully synced groups for user {user.id}: {user_groups}')
except Exception as e:
log.error(f'Failed to sync groups for user {user.id}: {e}')
- return create_session_response(request, user, db, response, set_cookie=True)
+ return await create_session_response(request, user, db, response, set_cookie=True)
else:
raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)
else:
@@ -536,7 +553,7 @@ async def signin(
request: Request,
response: Response,
form_data: SigninForm,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not ENABLE_PASSWORD_AUTH:
raise HTTPException(
@@ -558,7 +575,7 @@ async def signin(
except Exception as e:
pass
- if not Users.get_user_by_email(email.lower(), db=db):
+ if not await Users.get_user_by_email(email.lower(), db=db):
await signup_handler(
request,
email,
@@ -567,20 +584,20 @@ async def signin(
db=db,
)
- user = Auths.authenticate_user_by_email(email, db=db)
+ user = await Auths.authenticate_user_by_email(email, db=db)
if user:
if WEBUI_AUTH_TRUSTED_GROUPS_HEADER:
group_names = request.headers.get(WEBUI_AUTH_TRUSTED_GROUPS_HEADER, '').split(',')
group_names = [name.strip() for name in group_names if name.strip()]
if group_names:
- Groups.sync_groups_by_group_names(user.id, group_names, db=db)
+ await Groups.sync_groups_by_group_names(user.id, group_names, db=db)
if WEBUI_AUTH_TRUSTED_ROLE_HEADER:
trusted_role = request.headers.get(WEBUI_AUTH_TRUSTED_ROLE_HEADER, '').lower().strip()
if trusted_role in {'admin', 'user', 'pending'}:
if user.role != trusted_role:
- Users.update_user_role_by_id(user.id, trusted_role, db=db)
+ await Users.update_user_role_by_id(user.id, trusted_role, db=db)
elif trusted_role:
log.warning(f'Ignoring invalid trusted role header value: {trusted_role}')
@@ -588,14 +605,14 @@ async def signin(
admin_email = 'admin@localhost'
admin_password = 'admin'
- if Users.get_user_by_email(admin_email.lower(), db=db):
- user = Auths.authenticate_user(
+ if await Users.get_user_by_email(admin_email.lower(), db=db):
+ user = await Auths.authenticate_user(
admin_email.lower(),
lambda pw: verify_password(admin_password, pw),
db=db,
)
else:
- if Users.has_users(db=db):
+ if await Users.has_users(db=db):
raise HTTPException(400, detail=ERROR_MESSAGES.EXISTING_USERS)
await signup_handler(
@@ -606,7 +623,7 @@ async def signin(
db=db,
)
- user = Auths.authenticate_user(
+ user = await Auths.authenticate_user(
admin_email.lower(),
lambda pw: verify_password(admin_password, pw),
db=db,
@@ -627,14 +644,14 @@ async def signin(
# decode safely — ignore incomplete UTF-8 sequences
form_data.password = password_bytes.decode('utf-8', errors='ignore')
- user = Auths.authenticate_user(
+ user = await Auths.authenticate_user(
form_data.email.lower(),
lambda pw: verify_password(form_data.password, pw),
db=db,
)
if user:
- return create_session_response(request, user, db, response, set_cookie=True)
+ return await create_session_response(request, user, db, response, set_cookie=True)
else:
raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)
@@ -651,7 +668,7 @@ async def signup_handler(
name: str,
profile_image_url: str = '/user.png',
*,
- db: Session,
+ db: AsyncSession,
) -> UserModel:
"""
Core user-creation logic shared by the signup endpoint and
@@ -665,7 +682,7 @@ async def signup_handler(
# first-user registration can all see an empty table and each get admin.
hashed = get_password_hash(password)
- user = Auths.insert_new_auth(
+ user = await Auths.insert_new_auth(
email=email.lower(),
password=hashed,
name=name,
@@ -678,9 +695,9 @@ async def signup_handler(
# Atomically check if this is the only user *after* the insert.
# Only the single user present at this point should become admin.
- if Users.get_num_users(db=db) == 1:
- Users.update_user_role_by_id(user.id, 'admin', db=db)
- user = Users.get_user_by_id(user.id, db=db)
+ if await Users.get_num_users(db=db) == 1:
+ await Users.update_user_role_by_id(user.id, 'admin', db=db)
+ user = await Users.get_user_by_id(user.id, db=db)
request.app.state.config.ENABLE_SIGNUP = False
if request.app.state.config.WEBHOOK_URL:
@@ -695,7 +712,7 @@ async def signup_handler(
},
)
- apply_default_group_assignment(
+ await apply_default_group_assignment(
request.app.state.config.DEFAULT_GROUP_ID,
user.id,
db=db,
@@ -709,9 +726,9 @@ async def signup(
request: Request,
response: Response,
form_data: SignupForm,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- has_users = Users.has_users(db=db)
+ has_users = await Users.has_users(db=db)
if WEBUI_AUTH:
if not request.app.state.config.ENABLE_SIGNUP or not request.app.state.config.ENABLE_LOGIN_FORM:
@@ -724,7 +741,7 @@ async def signup(
if not validate_email_format(form_data.email.lower()):
raise HTTPException(status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.INVALID_EMAIL_FORMAT)
- if Users.get_user_by_email(form_data.email.lower(), db=db):
+ if await Users.get_user_by_email(form_data.email.lower(), db=db):
raise HTTPException(400, detail=ERROR_MESSAGES.EMAIL_TAKEN)
try:
@@ -741,7 +758,7 @@ async def signup(
form_data.profile_image_url,
db=db,
)
- return create_session_response(request, user, db, response, set_cookie=True)
+ return await create_session_response(request, user, db, response, set_cookie=True)
except HTTPException:
raise
except Exception as err:
@@ -750,7 +767,7 @@ async def signup(
@router.get('/signout')
-async def signout(request: Request, response: Response, db: Session = Depends(get_session)):
+async def signout(request: Request, response: Response, db: AsyncSession = Depends(get_async_session)):
# get auth token from headers or cookies
token = None
auth_header = request.headers.get('Authorization')
@@ -771,7 +788,7 @@ async def signout(request: Request, response: Response, db: Session = Depends(ge
if oauth_session_id:
response.delete_cookie('oauth_session_id')
- session = OAuthSessions.get_session_by_id(oauth_session_id, db=db)
+ session = await OAuthSessions.get_session_by_id(oauth_session_id, db=db)
# If a custom end_session_endpoint is configured (e.g. AWS Cognito), redirect
# there directly instead of attempting OIDC discovery.
@@ -846,12 +863,12 @@ async def add_user(
request: Request,
form_data: AddUserForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not validate_email_format(form_data.email.lower()):
raise HTTPException(status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.INVALID_EMAIL_FORMAT)
- if Users.get_user_by_email(form_data.email.lower(), db=db):
+ if await Users.get_user_by_email(form_data.email.lower(), db=db):
raise HTTPException(400, detail=ERROR_MESSAGES.EMAIL_TAKEN)
try:
@@ -861,7 +878,7 @@ async def add_user(
raise HTTPException(400, detail=str(e))
hashed = get_password_hash(form_data.password)
- user = Auths.insert_new_auth(
+ user = await Auths.insert_new_auth(
form_data.email.lower(),
hashed,
form_data.name,
@@ -871,13 +888,14 @@ async def add_user(
)
if user:
- apply_default_group_assignment(
+ await apply_default_group_assignment(
request.app.state.config.DEFAULT_GROUP_ID,
user.id,
db=db,
)
- token = create_token(data={'id': user.id})
+ expires_delta = parse_duration(request.app.state.config.JWT_EXPIRES_IN)
+ token = create_token(data={'id': user.id}, expires_delta=expires_delta)
return {
'token': token,
'token_type': 'Bearer',
@@ -902,7 +920,9 @@ async def add_user(
@router.get('/admin/details')
-async def get_admin_details(request: Request, user=Depends(get_current_user), db: Session = Depends(get_session)):
+async def get_admin_details(
+ request: Request, user=Depends(get_current_user), db: AsyncSession = Depends(get_async_session)
+):
if request.app.state.config.SHOW_ADMIN_DETAILS:
admin_email = request.app.state.config.ADMIN_EMAIL
admin_name = None
@@ -910,11 +930,11 @@ async def get_admin_details(request: Request, user=Depends(get_current_user), db
log.info(f'Admin details - Email: {admin_email}, Name: {admin_name}')
if admin_email:
- admin = Users.get_user_by_email(admin_email, db=db)
+ admin = await Users.get_user_by_email(admin_email, db=db)
if admin:
admin_name = admin.name
else:
- admin = Users.get_first_user(db=db)
+ admin = await Users.get_first_user(db=db)
if admin:
admin_email = admin.email
admin_name = admin.name
@@ -949,6 +969,8 @@ async def get_admin_config(request: Request, user=Depends(get_admin_user)):
'ENABLE_MESSAGE_RATING': request.app.state.config.ENABLE_MESSAGE_RATING,
'ENABLE_FOLDERS': request.app.state.config.ENABLE_FOLDERS,
'FOLDER_MAX_FILE_COUNT': request.app.state.config.FOLDER_MAX_FILE_COUNT,
+ 'AUTOMATION_MAX_COUNT': request.app.state.config.AUTOMATION_MAX_COUNT,
+ 'AUTOMATION_MIN_INTERVAL': request.app.state.config.AUTOMATION_MIN_INTERVAL,
'ENABLE_CHANNELS': request.app.state.config.ENABLE_CHANNELS,
'ENABLE_MEMORIES': request.app.state.config.ENABLE_MEMORIES,
'ENABLE_NOTES': request.app.state.config.ENABLE_NOTES,
@@ -975,6 +997,8 @@ class AdminConfig(BaseModel):
ENABLE_MESSAGE_RATING: bool
ENABLE_FOLDERS: bool
FOLDER_MAX_FILE_COUNT: Optional[int | str] = None
+ AUTOMATION_MAX_COUNT: Optional[int | str] = None
+ AUTOMATION_MIN_INTERVAL: Optional[int | str] = None
ENABLE_CHANNELS: bool
ENABLE_MEMORIES: bool
ENABLE_NOTES: bool
@@ -1000,6 +1024,12 @@ async def update_admin_config(request: Request, form_data: AdminConfig, user=Dep
request.app.state.config.FOLDER_MAX_FILE_COUNT = (
int(form_data.FOLDER_MAX_FILE_COUNT) if form_data.FOLDER_MAX_FILE_COUNT else ''
)
+ request.app.state.config.AUTOMATION_MAX_COUNT = (
+ int(form_data.AUTOMATION_MAX_COUNT) if form_data.AUTOMATION_MAX_COUNT else ''
+ )
+ request.app.state.config.AUTOMATION_MIN_INTERVAL = (
+ int(form_data.AUTOMATION_MIN_INTERVAL) if form_data.AUTOMATION_MIN_INTERVAL else ''
+ )
request.app.state.config.ENABLE_CHANNELS = form_data.ENABLE_CHANNELS
request.app.state.config.ENABLE_MEMORIES = form_data.ENABLE_MEMORIES
request.app.state.config.ENABLE_NOTES = form_data.ENABLE_NOTES
@@ -1041,6 +1071,8 @@ async def update_admin_config(request: Request, form_data: AdminConfig, user=Dep
'ENABLE_MESSAGE_RATING': request.app.state.config.ENABLE_MESSAGE_RATING,
'ENABLE_FOLDERS': request.app.state.config.ENABLE_FOLDERS,
'FOLDER_MAX_FILE_COUNT': request.app.state.config.FOLDER_MAX_FILE_COUNT,
+ 'AUTOMATION_MAX_COUNT': request.app.state.config.AUTOMATION_MAX_COUNT,
+ 'AUTOMATION_MIN_INTERVAL': request.app.state.config.AUTOMATION_MIN_INTERVAL,
'ENABLE_CHANNELS': request.app.state.config.ENABLE_CHANNELS,
'ENABLE_MEMORIES': request.app.state.config.ENABLE_MEMORIES,
'ENABLE_NOTES': request.app.state.config.ENABLE_NOTES,
@@ -1099,7 +1131,7 @@ async def update_ldap_server(request: Request, form_data: LdapServerConfig, user
for key in required_fields:
value = getattr(form_data, key)
if not value:
- raise HTTPException(400, detail=f'Required field {key} is empty')
+ raise HTTPException(400, detail=ERROR_MESSAGES.REQUIRED_FIELD_EMPTY(key))
request.app.state.config.LDAP_SERVER_LABEL = form_data.label
request.app.state.config.LDAP_SERVER_HOST = form_data.host
@@ -1154,10 +1186,12 @@ async def update_ldap_config(request: Request, form_data: LdapConfigForm, user=D
# create api key
@router.post('/api_key', response_model=ApiKey)
-async def generate_api_key(request: Request, user=Depends(get_current_user), db: Session = Depends(get_session)):
+async def generate_api_key(
+ request: Request, user=Depends(get_current_user), db: AsyncSession = Depends(get_async_session)
+):
if not request.app.state.config.ENABLE_API_KEYS or (
user.role != 'admin'
- and not has_permission(user.id, 'features.api_keys', request.app.state.config.USER_PERMISSIONS)
+ and not await has_permission(user.id, 'features.api_keys', request.app.state.config.USER_PERMISSIONS)
):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
@@ -1165,7 +1199,7 @@ async def generate_api_key(request: Request, user=Depends(get_current_user), db:
)
api_key = create_api_key()
- success = Users.update_user_api_key_by_id(user.id, api_key, db=db)
+ success = await Users.update_user_api_key_by_id(user.id, api_key, db=db)
if success:
return {
@@ -1177,14 +1211,14 @@ async def generate_api_key(request: Request, user=Depends(get_current_user), db:
# delete api key
@router.delete('/api_key', response_model=bool)
-async def delete_api_key(user=Depends(get_current_user), db: Session = Depends(get_session)):
- return Users.delete_user_api_key_by_id(user.id, db=db)
+async def delete_api_key(user=Depends(get_current_user), db: AsyncSession = Depends(get_async_session)):
+ return await Users.delete_user_api_key_by_id(user.id, db=db)
# get api key
@router.get('/api_key', response_model=ApiKey)
-async def get_api_key(user=Depends(get_current_user), db: Session = Depends(get_session)):
- api_key = Users.get_user_api_key_by_id(user.id, db=db)
+async def get_api_key(user=Depends(get_current_user), db: AsyncSession = Depends(get_async_session)):
+ api_key = await Users.get_user_api_key_by_id(user.id, db=db)
if api_key:
return {
'api_key': api_key,
@@ -1208,7 +1242,7 @@ async def token_exchange(
response: Response,
provider: str,
form_data: TokenExchangeForm,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""
Exchange an external OAuth provider token for an OpenWebUI JWT.
@@ -1226,7 +1260,7 @@ async def token_exchange(
if provider not in OAUTH_PROVIDERS:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail=f"Provider '{provider}' is not configured",
+ detail=ERROR_MESSAGES.OAUTH_NOT_CONFIGURED(provider),
)
# Get the OAuth client for this provider
oauth_manager = request.app.state.oauth_manager
@@ -1234,7 +1268,7 @@ async def token_exchange(
if not client:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail=f"OAuth client for '{provider}' not found",
+ detail=ERROR_MESSAGES.OAUTH_NOT_CONFIGURED(provider),
)
# Validate the token by calling the userinfo endpoint
@@ -1276,15 +1310,26 @@ async def token_exchange(
)
email = email.lower()
+ # Enforce domain allowlist — same check as the normal OAuth callback
+ if (
+ '*' not in auth_manager_config.OAUTH_ALLOWED_DOMAINS
+ and email.split('@')[-1] not in auth_manager_config.OAUTH_ALLOWED_DOMAINS
+ ):
+ log.warning(f'Token exchange denied: email domain not in allowed domains list')
+ raise HTTPException(
+ status_code=status.HTTP_403_FORBIDDEN,
+ detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
+ )
+
# Try to find the user by OAuth sub
- user = Users.get_user_by_oauth_sub(provider, sub, db=db)
+ user = await Users.get_user_by_oauth_sub(provider, sub, db=db)
if not user and OAUTH_MERGE_ACCOUNTS_BY_EMAIL.value:
# Try to find by email if merge is enabled
- user = Users.get_user_by_email(email, db=db)
+ user = await Users.get_user_by_email(email, db=db)
if user:
# Link the OAuth sub to this user
- Users.update_user_oauth_by_id(user.id, provider, sub, db=db)
+ await Users.update_user_oauth_by_id(user.id, provider, sub, db=db)
if not user:
raise HTTPException(
@@ -1292,4 +1337,4 @@ async def token_exchange(
detail='User not found. Please sign in via the web interface first.',
)
- return create_session_response(request, user, db)
+ return await create_session_response(request, user, db)
diff --git a/backend/open_webui/routers/automations.py b/backend/open_webui/routers/automations.py
index 803f59a6f2..d68bd8e2c6 100644
--- a/backend/open_webui/routers/automations.py
+++ b/backend/open_webui/routers/automations.py
@@ -3,7 +3,7 @@ import logging
from typing import Optional
from fastapi import APIRouter, Depends, HTTPException, Request, status
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
from open_webui.models.automations import (
Automations,
@@ -19,10 +19,11 @@ from open_webui.utils.automations import (
next_run_ns,
next_n_runs_ns,
execute_automation,
+ rrule_interval_seconds,
)
from open_webui.utils.auth import get_verified_user, get_admin_user
from open_webui.utils.access_control import has_permission
-from open_webui.internal.db import get_session
+from open_webui.internal.db import get_async_session
from open_webui.constants import ERROR_MESSAGES
log = logging.getLogger(__name__)
@@ -37,8 +38,8 @@ PAGE_ITEM_COUNT = 30
############################
-def check_automations_permission(request, user):
- if user.role != 'admin' and not has_permission(
+async def check_automations_permission(request, user):
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.automations', request.app.state.config.USER_PERMISSIONS
):
raise HTTPException(
@@ -60,8 +61,38 @@ def check_automation_access(automation, user):
)
-def enrich_automation(automation: AutomationModel, db: Session, tz: str = None) -> AutomationResponse:
- last_run = AutomationRuns.get_latest(automation.id, db=db)
+async def check_automation_limits(request, user, rrule_str: str, db, is_create: bool = False):
+ """Enforce global automation limits. Admins bypass all checks."""
+ if user.role == 'admin':
+ return
+
+ # Max count (create only)
+ if is_create:
+ max_count = request.app.state.config.AUTOMATION_MAX_COUNT
+ if max_count:
+ max_count = int(max_count)
+ if max_count > 0 and await Automations.count_by_user(user.id, db=db) >= max_count:
+ raise HTTPException(
+ status_code=status.HTTP_403_FORBIDDEN,
+ detail=ERROR_MESSAGES.AUTOMATION_LIMIT_EXCEEDED(max_count),
+ )
+
+ # Min interval (create + update)
+ min_interval = request.app.state.config.AUTOMATION_MIN_INTERVAL
+ if min_interval:
+ min_interval = int(min_interval)
+ if min_interval > 0:
+ interval = rrule_interval_seconds(rrule_str)
+ if interval is not None and interval < min_interval:
+ raise HTTPException(
+ status_code=status.HTTP_400_BAD_REQUEST,
+ detail=ERROR_MESSAGES.AUTOMATION_TOO_FREQUENT(min_interval),
+ )
+
+
+async def enrich_automation(automation: AutomationModel, db: AsyncSession, tz: str = None) -> AutomationResponse:
+ """Full enrichment for single-item views (includes next_runs computation)."""
+ last_run = await AutomationRuns.get_latest(automation.id, db=db)
return AutomationResponse(
**automation.model_dump(),
last_run=last_run,
@@ -81,14 +112,14 @@ async def get_automation_items(
status: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
+ await check_automations_permission(request, user)
limit = PAGE_ITEM_COUNT
page = max(1, page)
skip = (page - 1) * limit
- result = Automations.search_automations(
+ result = await Automations.search_automations(
user_id=user.id,
query=query,
status=status,
@@ -97,8 +128,18 @@ async def get_automation_items(
db=db,
)
+ # Batch-fetch latest runs in a single query instead of N+1
+ ids = [item.id for item in result.items]
+ latest_runs = await AutomationRuns.get_latest_batch(ids, db=db) if ids else {}
+
return {
- 'items': [enrich_automation(item, db, tz=user.timezone) for item in result.items],
+ 'items': [
+ AutomationResponse(
+ **item.model_dump(),
+ last_run=latest_runs.get(item.id),
+ )
+ for item in result.items
+ ],
'total': result.total,
}
@@ -113,9 +154,9 @@ async def create_new_automation(
request: Request,
form_data: AutomationForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
+ await check_automations_permission(request, user)
try:
validate_rrule(form_data.data.rrule)
except ValueError as e:
@@ -124,18 +165,11 @@ async def create_new_automation(
detail=str(e),
)
- # Validate terminal server exists if linked
- if form_data.data.terminal and form_data.data.terminal.server_id:
- connections = request.app.state.config.TERMINAL_SERVER_CONNECTIONS or []
- if not any(c.get('id') == form_data.data.terminal.server_id for c in connections):
- raise HTTPException(
- status_code=status.HTTP_400_BAD_REQUEST,
- detail='Terminal server not found',
- )
+ await check_automation_limits(request, user, form_data.data.rrule, db, is_create=True)
tz = user.timezone
- automation = Automations.insert(user.id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db)
- return enrich_automation(automation, db, tz=tz)
+ automation = await Automations.insert(user.id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db)
+ return await enrich_automation(automation, db, tz=tz)
############################
@@ -148,12 +182,12 @@ async def get_automation_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
- automation = Automations.get_by_id(id, db=db)
+ await check_automations_permission(request, user)
+ automation = await Automations.get_by_id(id, db=db)
check_automation_access(automation, user)
- return enrich_automation(automation, db, tz=user.timezone)
+ return await enrich_automation(automation, db, tz=user.timezone)
############################
@@ -167,10 +201,10 @@ async def update_automation_by_id(
id: str,
form_data: AutomationForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
- automation = Automations.get_by_id(id, db=db)
+ await check_automations_permission(request, user)
+ automation = await Automations.get_by_id(id, db=db)
check_automation_access(automation, user)
try:
@@ -181,18 +215,11 @@ async def update_automation_by_id(
detail=str(e),
)
- # Validate terminal server exists if linked
- if form_data.data.terminal and form_data.data.terminal.server_id:
- connections = request.app.state.config.TERMINAL_SERVER_CONNECTIONS or []
- if not any(c.get('id') == form_data.data.terminal.server_id for c in connections):
- raise HTTPException(
- status_code=status.HTTP_400_BAD_REQUEST,
- detail='Terminal server not found',
- )
+ await check_automation_limits(request, user, form_data.data.rrule, db, is_create=False)
tz = user.timezone
- updated = Automations.update_by_id(id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db)
- return enrich_automation(updated, db, tz=tz)
+ updated = await Automations.update_by_id(id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db)
+ return await enrich_automation(updated, db, tz=tz)
############################
@@ -205,13 +232,13 @@ async def toggle_automation_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
- automation = Automations.get_by_id(id, db=db)
+ await check_automations_permission(request, user)
+ automation = await Automations.get_by_id(id, db=db)
check_automation_access(automation, user)
- toggled = Automations.toggle(id, next_run_ns(automation.data['rrule'], tz=user.timezone), db=db)
- return enrich_automation(toggled, db, tz=user.timezone)
+ toggled = await Automations.toggle(id, next_run_ns(automation.data['rrule'], tz=user.timezone), db=db)
+ return await enrich_automation(toggled, db, tz=user.timezone)
############################
@@ -224,13 +251,13 @@ async def run_automation_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
- automation = Automations.get_by_id(id, db=db)
+ await check_automations_permission(request, user)
+ automation = await Automations.get_by_id(id, db=db)
check_automation_access(automation, user)
asyncio.create_task(execute_automation(request.app, automation))
- return enrich_automation(automation, db, tz=user.timezone)
+ return await enrich_automation(automation, db, tz=user.timezone)
############################
@@ -243,13 +270,13 @@ async def delete_automation_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
- automation = Automations.get_by_id(id, db=db)
+ await check_automations_permission(request, user)
+ automation = await Automations.get_by_id(id, db=db)
check_automation_access(automation, user)
- AutomationRuns.delete_by_automation(id, db=db)
- return Automations.delete(id, db=db)
+ await AutomationRuns.delete_by_automation(id, db=db)
+ return await Automations.delete(id, db=db)
############################
@@ -264,9 +291,9 @@ async def get_automation_runs(
skip: int = 0,
limit: int = 50,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_automations_permission(request, user)
- automation = Automations.get_by_id(id, db=db)
+ await check_automations_permission(request, user)
+ automation = await Automations.get_by_id(id, db=db)
check_automation_access(automation, user)
- return AutomationRuns.get_by_automation(id, skip=skip, limit=limit, db=db)
+ return await AutomationRuns.get_by_automation(id, skip=skip, limit=limit, db=db)
diff --git a/backend/open_webui/routers/channels.py b/backend/open_webui/routers/channels.py
index aa1ea52662..a771b95920 100644
--- a/backend/open_webui/routers/channels.py
+++ b/backend/open_webui/routers/channels.py
@@ -61,25 +61,25 @@ from open_webui.utils.chat import generate_chat_completion
from open_webui.utils.auth import get_admin_user, get_verified_user
-from open_webui.utils.access_control import has_permission
+from open_webui.utils.access_control import has_permission, filter_allowed_access_grants
from open_webui.utils.webhook import post_webhook
from open_webui.utils.channels import extract_mentions, replace_mentions
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
router = APIRouter()
-def channel_has_access(
+async def channel_has_access(
user_id: str,
channel: ChannelModel,
permission: str = 'read',
strict: bool = True,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
) -> bool:
- if AccessGrants.has_access(
+ if await AccessGrants.has_access(
user_id=user_id,
resource_type='channel',
resource_id=channel.id,
@@ -94,8 +94,10 @@ def channel_has_access(
return False
-def get_channel_users_with_access(channel: ChannelModel, permission: str = 'read', db: Optional[Session] = None):
- return AccessGrants.get_users_with_access(
+async def get_channel_users_with_access(
+ channel: ChannelModel, permission: str = 'read', db: Optional[AsyncSession] = None
+):
+ return await AccessGrants.get_users_with_access(
resource_type='channel',
resource_id=channel.id,
permission=permission,
@@ -133,16 +135,16 @@ def get_channel_permitted_group_and_user_ids(
############################
-def check_channels_access(request: Request, user: Optional[UserModel] = None):
+async def check_channels_access(request: Request, user: Optional[UserModel] = None):
"""Dependency to ensure channels are globally enabled."""
if not request.app.state.config.ENABLE_CHANNELS:
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
- detail='Channels are not enabled',
+ detail=ERROR_MESSAGES.FEATURE_DISABLED('Channels'),
)
if user:
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.channels', request.app.state.config.USER_PERMISSIONS
):
raise HTTPException(
@@ -168,19 +170,19 @@ class ChannelListItemResponse(ChannelModel):
async def get_channels(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
- channels = Channels.get_channels_by_user_id(user.id, db=db)
+ channels = await Channels.get_channels_by_user_id(user.id, db=db)
channel_list = []
for channel in channels:
- last_message = Messages.get_last_message_by_channel_id(channel.id, db=db)
+ last_message = await Messages.get_last_message_by_channel_id(channel.id, db=db)
last_message_at = last_message.created_at if last_message else None
- channel_member = Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
+ channel_member = await Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
unread_count = (
- Messages.get_unread_message_count(channel.id, user.id, channel_member.last_read_at, db=db)
+ await Messages.get_unread_message_count(channel.id, user.id, channel_member.last_read_at, db=db)
if channel_member
else 0
)
@@ -188,15 +190,15 @@ async def get_channels(
user_ids = None
users = None
if channel.type == 'dm':
- user_ids = [member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)]
+ user_ids = [member.user_id for member in await Channels.get_members_by_channel_id(channel.id, db=db)]
users = [
UserIdNameStatusResponse(
**{
- **user.model_dump(),
- 'is_active': Users.is_active(user),
+ **u.model_dump(),
+ 'is_active': Users.is_active(u),
}
)
- for user in Users.get_users_by_user_ids(user_ids, db=db)
+ for u in await Users.get_users_by_user_ids(user_ids, db=db)
]
channel_list.append(
@@ -216,12 +218,12 @@ async def get_channels(
async def get_all_channels(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
if user.role == 'admin':
- return Channels.get_channels(db=db)
- return Channels.get_channels_by_user_id(user.id, db=db)
+ return await Channels.get_channels(db=db)
+ return await Channels.get_channels_by_user_id(user.id, db=db)
############################
@@ -234,14 +236,14 @@ async def get_dm_channel_by_user_id(
request: Request,
user_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
try:
- existing_channel = Channels.get_dm_channel_by_user_ids([user.id, user_id], db=db)
+ existing_channel = await Channels.get_dm_channel_by_user_ids([user.id, user_id], db=db)
if existing_channel:
participant_ids = [
- member.user_id for member in Channels.get_members_by_channel_id(existing_channel.id, db=db)
+ member.user_id for member in await Channels.get_members_by_channel_id(existing_channel.id, db=db)
]
await emit_to_users(
@@ -251,10 +253,10 @@ async def get_dm_channel_by_user_id(
)
await enter_room_for_users(f'channel:{existing_channel.id}', participant_ids)
- Channels.update_member_active_status(existing_channel.id, user.id, True, db=db)
+ await Channels.update_member_active_status(existing_channel.id, user.id, True, db=db)
return ChannelModel(**existing_channel.model_dump())
- channel = Channels.insert_new_channel(
+ channel = await Channels.insert_new_channel(
CreateChannelForm(
type='dm',
name='',
@@ -265,7 +267,7 @@ async def get_dm_channel_by_user_id(
)
if channel:
- participant_ids = [member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)]
+ participant_ids = [member.user_id for member in await Channels.get_members_by_channel_id(channel.id, db=db)]
await emit_to_users(
'events:channel',
@@ -292,9 +294,9 @@ async def create_new_channel(
request: Request,
form_data: CreateChannelForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
if form_data.type not in ['group', 'dm'] and user.role != 'admin':
# Only admins can create standard channels (joined by default)
@@ -303,12 +305,20 @@ async def create_new_channel(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
+ form_data.access_grants = filter_allowed_access_grants(
+ request.app.state.config.USER_PERMISSIONS,
+ user.id,
+ user.role,
+ form_data.access_grants,
+ 'sharing.public_channels',
+ )
+
try:
if form_data.type == 'dm':
- existing_channel = Channels.get_dm_channel_by_user_ids([user.id, *form_data.user_ids], db=db)
+ existing_channel = await Channels.get_dm_channel_by_user_ids([user.id, *form_data.user_ids], db=db)
if existing_channel:
participant_ids = [
- member.user_id for member in Channels.get_members_by_channel_id(existing_channel.id, db=db)
+ member.user_id for member in await Channels.get_members_by_channel_id(existing_channel.id, db=db)
]
await emit_to_users(
'events:channel',
@@ -317,13 +327,13 @@ async def create_new_channel(
)
await enter_room_for_users(f'channel:{existing_channel.id}', participant_ids)
- Channels.update_member_active_status(existing_channel.id, user.id, True, db=db)
+ await Channels.update_member_active_status(existing_channel.id, user.id, True, db=db)
return ChannelModel(**existing_channel.model_dump())
- channel = Channels.insert_new_channel(form_data, user.id, db=db)
+ channel = await Channels.insert_new_channel(form_data, user.id, db=db)
if channel:
- participant_ids = [member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)]
+ participant_ids = [member.user_id for member in await Channels.get_members_by_channel_id(channel.id, db=db)]
await emit_to_users(
'events:channel',
@@ -358,10 +368,10 @@ async def get_channel_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -369,23 +379,23 @@ async def get_channel_by_id(
users = None
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- user_ids = [member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)]
+ user_ids = [member.user_id for member in await Channels.get_members_by_channel_id(channel.id, db=db)]
users = [
UserIdNameStatusResponse(
**{
- **user.model_dump(),
- 'is_active': Users.is_active(user),
+ **u.model_dump(),
+ 'is_active': Users.is_active(u),
}
)
- for user in Users.get_users_by_user_ids(user_ids, db=db)
+ for u in await Users.get_users_by_user_ids(user_ids, db=db)
]
- channel_member = Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
- unread_count = Messages.get_unread_message_count(
+ channel_member = await Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
+ unread_count = await Messages.get_unread_message_count(
channel.id, user.id, channel_member.last_read_at if channel_member else None
)
@@ -394,7 +404,7 @@ async def get_channel_by_id(
**channel.model_dump(),
'user_ids': user_ids,
'users': users,
- 'is_manager': Channels.is_user_channel_manager(channel.id, user.id, db=db),
+ 'is_manager': await Channels.is_user_channel_manager(channel.id, user.id, db=db),
'write_access': True,
'user_count': len(user_ids),
'last_read_at': channel_member.last_read_at if channel_member else None,
@@ -402,10 +412,10 @@ async def get_channel_by_id(
}
)
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- write_access = channel_has_access(
+ write_access = await channel_has_access(
user.id,
channel,
permission='write',
@@ -413,10 +423,10 @@ async def get_channel_by_id(
db=db,
)
- user_count = len(get_channel_users_with_access(channel, 'read', db=db))
+ user_count = len(await get_channel_users_with_access(channel, 'read', db=db))
- channel_member = Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
- unread_count = Messages.get_unread_message_count(
+ channel_member = await Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
+ unread_count = await Messages.get_unread_message_count(
channel.id, user.id, channel_member.last_read_at if channel_member else None
)
@@ -425,7 +435,7 @@ async def get_channel_by_id(
**channel.model_dump(),
'user_ids': user_ids,
'users': users,
- 'is_manager': Channels.is_user_channel_manager(channel.id, user.id, db=db),
+ 'is_manager': await Channels.is_user_channel_manager(channel.id, user.id, db=db),
'write_access': write_access or user.role == 'admin',
'user_count': user_count,
'last_read_at': channel_member.last_read_at if channel_member else None,
@@ -451,11 +461,11 @@ async def get_channel_members_by_id(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -465,16 +475,19 @@ async def get_channel_members_by_id(
skip = (page - 1) * limit
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
+ raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
+ else:
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
if channel.type == 'dm':
- user_ids = [member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)]
- users = Users.get_users_by_user_ids(user_ids, db=db)
- total = len(users)
+ user_ids = [member.user_id for member in await Channels.get_members_by_channel_id(channel.id, db=db)]
+ fetched_users = await Users.get_users_by_user_ids(user_ids, db=db)
+ total = len(fetched_users)
return {
- 'users': [UserModelResponse(**user.model_dump(), is_active=Users.is_active(user)) for user in users],
+ 'users': [UserModelResponse(**u.model_dump(), is_active=Users.is_active(u)) for u in fetched_users],
'total': total,
}
else:
@@ -496,13 +509,13 @@ async def get_channel_members_by_id(
filter['user_ids'] = permitted_ids.get('user_ids')
filter['group_ids'] = permitted_ids.get('group_ids')
- result = Users.get_users(filter=filter, skip=skip, limit=limit, db=db)
+ result = await Users.get_users(filter=filter, skip=skip, limit=limit, db=db)
- users = result['users']
+ fetched_users = result['users']
total = result['total']
return {
- 'users': [UserModelResponse(**user.model_dump(), is_active=Users.is_active(user)) for user in users],
+ 'users': [UserModelResponse(**u.model_dump(), is_active=Users.is_active(u)) for u in fetched_users],
'total': total,
}
@@ -522,17 +535,17 @@ async def update_is_active_member_by_id_and_user_id(
id: str,
form_data: UpdateActiveMemberForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
- Channels.update_member_active_status(channel.id, user.id, form_data.is_active, db=db)
+ await Channels.update_member_active_status(channel.id, user.id, form_data.is_active, db=db)
return True
@@ -552,10 +565,10 @@ async def add_members_by_id(
id: str,
form_data: UpdateMembersForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -563,7 +576,7 @@ async def add_members_by_id(
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- memberships = Channels.add_members_to_channel(
+ memberships = await Channels.add_members_to_channel(
channel.id, user.id, form_data.user_ids, form_data.group_ids, db=db
)
@@ -588,11 +601,11 @@ async def remove_members_by_id(
id: str,
form_data: RemoveMembersForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -600,7 +613,7 @@ async def remove_members_by_id(
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- deleted = Channels.remove_members_from_channel(channel.id, form_data.user_ids, db=db)
+ deleted = await Channels.remove_members_from_channel(channel.id, form_data.user_ids, db=db)
return deleted
except Exception as e:
@@ -619,19 +632,27 @@ async def update_channel_by_id(
id: str,
form_data: ChannelForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.user_id != user.id and user.role != 'admin':
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
+ form_data.access_grants = filter_allowed_access_grants(
+ request.app.state.config.USER_PERMISSIONS,
+ user.id,
+ user.role,
+ form_data.access_grants,
+ 'sharing.public_channels',
+ )
+
try:
- channel = Channels.update_channel_by_id(id, form_data, db=db)
+ channel = await Channels.update_channel_by_id(id, form_data, db=db)
return ChannelModel(**channel.model_dump())
except Exception as e:
log.exception(e)
@@ -648,11 +669,11 @@ async def delete_channel_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -660,7 +681,7 @@ async def delete_channel_by_id(
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- Channels.delete_channel_by_id(id, db=db)
+ await Channels.delete_channel_by_id(id, db=db)
return True
except Exception as e:
log.exception(e)
@@ -692,40 +713,40 @@ async def get_channel_messages(
skip: int = 0,
limit: int = 50,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- channel_member = Channels.join_channel(id, user.id, db=db) # Ensure user is a member of the channel
+ channel_member = await Channels.join_channel(id, user.id, db=db) # Ensure user is a member of the channel
- message_list = Messages.get_messages_by_channel_id(id, skip, limit, db=db)
+ message_list = await Messages.get_messages_by_channel_id(id, skip, limit, db=db)
if not message_list:
return []
# Batch fetch all users in a single query (fixes N+1 problem)
user_ids = list(set(m.user_id for m in message_list))
- users = {u.id: u for u in Users.get_users_by_user_ids(user_ids, db=db)}
+ fetched_users = {u.id: u for u in await Users.get_users_by_user_ids(user_ids, db=db)}
messages = []
for message in message_list:
- thread_replies = Messages.get_thread_replies_by_message_id(message.id, db=db)
+ thread_replies = await Messages.get_thread_replies_by_message_id(message.id, db=db)
latest_thread_reply_at = thread_replies[0].created_at if thread_replies else None
# Use message.user if present (for webhooks), otherwise look up by user_id
user_info = message.user
- if user_info is None and message.user_id in users:
- user_info = UserNameResponse(**users[message.user_id].model_dump())
+ if user_info is None and message.user_id in fetched_users:
+ user_info = UserNameResponse(**fetched_users[message.user_id].model_dump())
messages.append(
MessageUserResponse(
@@ -733,7 +754,7 @@ async def get_channel_messages(
**message.model_dump(),
'reply_count': len(thread_replies),
'latest_reply_at': latest_thread_reply_at,
- 'reactions': Messages.get_reactions_by_message_id(message.id, db=db),
+ 'reactions': await Messages.get_reactions_by_message_id(message.id, db=db),
'user': user_info,
}
)
@@ -755,32 +776,32 @@ async def get_pinned_channel_messages(
id: str,
page: int = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
page = max(1, page)
skip = (page - 1) * PAGE_ITEM_COUNT_PINNED
limit = PAGE_ITEM_COUNT_PINNED
- message_list = Messages.get_pinned_messages_by_channel_id(id, skip, limit, db=db)
+ message_list = await Messages.get_pinned_messages_by_channel_id(id, skip, limit, db=db)
if not message_list:
return []
# Batch fetch all users in a single query (fixes N+1 problem)
user_ids = list(set(m.user_id for m in message_list))
- users = {u.id: u for u in Users.get_users_by_user_ids(user_ids, db=db)}
+ fetched_users = {u.id: u for u in await Users.get_users_by_user_ids(user_ids, db=db)}
messages = []
for message in message_list:
@@ -788,12 +809,12 @@ async def get_pinned_channel_messages(
webhook_info = message.meta.get('webhook') if message.meta else None
if webhook_info:
user_info = UserNameResponse(
- id=webhook_info.get('id'),
- name=webhook_info.get('name'),
+ id=webhook_info.get('id') or '',
+ name=webhook_info.get('name') or 'Webhook',
role='webhook',
)
- elif message.user_id in users:
- user_info = UserNameResponse(**users[message.user_id].model_dump())
+ elif message.user_id in fetched_users:
+ user_info = UserNameResponse(**fetched_users[message.user_id].model_dump())
else:
user_info = None
@@ -801,7 +822,7 @@ async def get_pinned_channel_messages(
MessageWithReactionsResponse(
**{
**message.model_dump(),
- 'reactions': Messages.get_reactions_by_message_id(message.id, db=db),
+ 'reactions': await Messages.get_reactions_by_message_id(message.id, db=db),
'user': user_info,
}
)
@@ -820,12 +841,12 @@ async def send_notification(request, channel, message, active_user_ids, db=None)
webui_url = request.app.state.config.WEBUI_URL
enable_user_webhooks = request.app.state.config.ENABLE_USER_WEBHOOKS
- users = get_channel_users_with_access(channel, 'read', db=db)
+ users = await get_channel_users_with_access(channel, 'read', db=db)
- for user in users:
- if (user.id not in active_user_ids) and Channels.is_user_channel_member(channel.id, user.id, db=db):
- if enable_user_webhooks and user.settings:
- webhook_url = user.settings.ui.get('notifications', {}).get('webhook_url', None)
+ for u in users:
+ if (u.id not in active_user_ids) and await Channels.is_user_channel_member(channel.id, u.id, db=db):
+ if enable_user_webhooks and u.settings:
+ webhook_url = u.settings.ui.get('notifications', {}).get('webhook_url', None)
if webhook_url:
await post_webhook(
name,
@@ -843,7 +864,7 @@ async def send_notification(request, channel, message, active_user_ids, db=None)
async def model_response_handler(request, channel, message, user, db=None):
- MODELS = {model['id']: model for model in get_filtered_models(await get_all_models(request, user=user), user)}
+ MODELS = {model['id']: model for model in await get_filtered_models(await get_all_models(request, user=user), user)}
mentions = extract_mentions(message.content)
message_content = replace_mentions(message.content)
@@ -874,10 +895,12 @@ async def model_response_handler(request, channel, message, user, db=None):
if model:
try:
# reverse to get in chronological order
- thread_messages = Messages.get_messages_by_parent_id(
- channel.id,
- message.parent_id if message.parent_id else message.id,
- db=db,
+ thread_messages = (
+ await Messages.get_messages_by_parent_id(
+ channel.id,
+ message.parent_id if message.parent_id else message.id,
+ db=db,
+ )
)[::-1]
response_message, channel = await new_message_handler(
@@ -905,7 +928,7 @@ async def model_response_handler(request, channel, message, user, db=None):
for thread_message in thread_messages:
message_user = None
if thread_message.user_id not in message_users:
- message_user = Users.get_user_by_id(thread_message.user_id, db=db)
+ message_user = await Users.get_user_by_id(thread_message.user_id, db=db)
message_users[thread_message.user_id] = message_user
else:
message_user = message_users[thread_message.user_id]
@@ -925,7 +948,7 @@ async def model_response_handler(request, channel, message, user, db=None):
if file.get('type', '') == 'image':
images.append(file.get('url', ''))
elif file.get('content_type', '').startswith('image/'):
- image = get_image_base64_from_file_id(file.get('id', ''))
+ image = await get_image_base64_from_file_id(file.get('id', ''))
if image:
images.append(image)
@@ -1014,15 +1037,15 @@ async def model_response_handler(request, channel, message, user, db=None):
async def new_message_handler(request: Request, id: str, form_data: MessageForm, user, db):
- channel = Channels.get_channel_by_id(id, db=db)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(
+ if user.role != 'admin' and not await channel_has_access(
user.id,
channel,
permission='write',
@@ -1032,15 +1055,15 @@ async def new_message_handler(request: Request, id: str, form_data: MessageForm,
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- message = Messages.insert_new_message(form_data, channel.id, user.id, db=db)
+ message = await Messages.insert_new_message(form_data, channel.id, user.id, db=db)
if message:
if channel.type in ['group', 'dm']:
- members = Channels.get_members_by_channel_id(channel.id, db=db)
+ members = await Channels.get_members_by_channel_id(channel.id, db=db)
for member in members:
if not member.is_active:
- Channels.update_member_active_status(channel.id, member.user_id, True, db=db)
+ await Channels.update_member_active_status(channel.id, member.user_id, True, db=db)
- message = Messages.get_message_by_id(message.id, db=db)
+ message = await Messages.get_message_by_id(message.id, db=db)
event_data = {
'channel_id': channel.id,
'message_id': message.id,
@@ -1060,7 +1083,7 @@ async def new_message_handler(request: Request, id: str, form_data: MessageForm,
if message.parent_id:
# If this message is a reply, emit to the parent message as well
- parent_message = Messages.get_message_by_id(message.parent_id, db=db)
+ parent_message = await Messages.get_message_by_id(message.parent_id, db=db)
if parent_message:
await sio.emit(
@@ -1092,16 +1115,18 @@ async def post_new_message(
form_data: MessageForm,
background_tasks: BackgroundTasks,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
+ await check_channels_access(request, user)
try:
message, channel = await new_message_handler(request, id, form_data, user, db)
try:
if files := message.data.get('files', []):
for file in files:
- Channels.set_file_message_id_in_channel_by_id(channel.id, file.get('id', ''), message.id, db=db)
+ await Channels.set_file_message_id_in_channel_by_id(
+ channel.id, file.get('id', ''), message.id, db=db
+ )
except Exception as e:
log.debug(e)
@@ -1141,31 +1166,32 @@ async def get_channel_message(
id: str,
message_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if message.channel_id != id:
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
+ message_user = await Users.get_user_by_id(message.user_id, db=db)
return MessageResponse(
**{
**message.model_dump(),
- 'user': UserNameResponse(**Users.get_user_by_id(message.user_id, db=db).model_dump()),
+ 'user': UserNameResponse(**message_user.model_dump()) if message_user else None,
}
)
@@ -1181,21 +1207,21 @@ async def get_channel_message_data(
id: str,
message_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1221,21 +1247,21 @@ async def pin_channel_message(
message_id: str,
form_data: PinMessageForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1243,12 +1269,13 @@ async def pin_channel_message(
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
try:
- Messages.update_is_pinned_by_id(message_id, form_data.is_pinned, user.id, db=db)
- message = Messages.get_message_by_id(message_id, db=db)
+ await Messages.update_is_pinned_by_id(message_id, form_data.is_pinned, user.id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
+ message_user = await Users.get_user_by_id(message.user_id, db=db)
return MessageUserResponse(
**{
**message.model_dump(),
- 'user': UserNameResponse(**Users.get_user_by_id(message.user_id, db=db).model_dump()),
+ 'user': UserNameResponse(**message_user.model_dump()) if message_user else None,
}
)
except Exception as e:
@@ -1269,35 +1296,35 @@ async def get_channel_thread_messages(
skip: int = 0,
limit: int = 50,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(user.id, channel, permission='read', db=db):
+ if user.role != 'admin' and not await channel_has_access(user.id, channel, permission='read', db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- message_list = Messages.get_messages_by_parent_id(id, message_id, skip, limit, db=db)
+ message_list = await Messages.get_messages_by_parent_id(id, message_id, skip, limit, db=db)
if not message_list:
return []
# Batch fetch all users in a single query (fixes N+1 problem)
user_ids = list(set(m.user_id for m in message_list))
- users = {u.id: u for u in Users.get_users_by_user_ids(user_ids, db=db)}
+ fetched_users = {u.id: u for u in await Users.get_users_by_user_ids(user_ids, db=db)}
messages = []
for message in message_list:
# Use message.user if present (for webhooks), otherwise look up by user_id
user_info = message.user
- if user_info is None and message.user_id in users:
- user_info = UserNameResponse(**users[message.user_id].model_dump())
+ if user_info is None and message.user_id in fetched_users:
+ user_info = UserNameResponse(**fetched_users[message.user_id].model_dump())
messages.append(
MessageUserResponse(
@@ -1305,7 +1332,7 @@ async def get_channel_thread_messages(
**message.model_dump(),
'reply_count': 0,
'latest_reply_at': None,
- 'reactions': Messages.get_reactions_by_message_id(message.id, db=db),
+ 'reactions': await Messages.get_reactions_by_message_id(message.id, db=db),
'user': user_info,
}
)
@@ -1326,14 +1353,14 @@ async def update_message_by_id(
message_id: str,
form_data: MessageForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1341,19 +1368,19 @@ async def update_message_by_id(
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
if (
user.role != 'admin'
and message.user_id != user.id
- and not channel_has_access(user.id, channel, permission='write', strict=False, db=db)
+ and not await channel_has_access(user.id, channel, permission='write', strict=False, db=db)
):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- message = Messages.update_message_by_id(message_id, form_data, db=db)
- message = Messages.get_message_by_id(message_id, db=db)
+ await Messages.update_message_by_id(message_id, form_data, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if message:
await sio.emit(
@@ -1393,18 +1420,18 @@ async def add_reaction_to_message(
message_id: str,
form_data: ReactionForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(
+ if user.role != 'admin' and not await channel_has_access(
user.id,
channel,
permission='write',
@@ -1413,7 +1440,7 @@ async def add_reaction_to_message(
):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1421,8 +1448,8 @@ async def add_reaction_to_message(
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
try:
- Messages.add_reaction_to_message(message_id, user.id, form_data.name, db=db)
- message = Messages.get_message_by_id(message_id, db=db)
+ await Messages.add_reaction_to_message(message_id, user.id, form_data.name, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
await sio.emit(
'events:channel',
@@ -1460,18 +1487,18 @@ async def remove_reaction_by_id_and_user_id_and_name(
message_id: str,
form_data: ReactionForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
- if user.role != 'admin' and not channel_has_access(
+ if user.role != 'admin' and not await channel_has_access(
user.id,
channel,
permission='write',
@@ -1480,7 +1507,7 @@ async def remove_reaction_by_id_and_user_id_and_name(
):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1488,9 +1515,9 @@ async def remove_reaction_by_id_and_user_id_and_name(
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
try:
- Messages.remove_reaction_by_id_and_user_id_and_name(message_id, user.id, form_data.name, db=db)
+ await Messages.remove_reaction_by_id_and_user_id_and_name(message_id, user.id, form_data.name, db=db)
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
await sio.emit(
'events:channel',
@@ -1527,14 +1554,14 @@ async def delete_message_by_id(
id: str,
message_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
- message = Messages.get_message_by_id(message_id, db=db)
+ message = await Messages.get_message_by_id(message_id, db=db)
if not message:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1542,13 +1569,13 @@ async def delete_message_by_id(
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
if channel.type in ['group', 'dm']:
- if not Channels.is_user_channel_member(channel.id, user.id, db=db):
+ if not await Channels.is_user_channel_member(channel.id, user.id, db=db):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
else:
if (
user.role != 'admin'
and message.user_id != user.id
- and not channel_has_access(
+ and not await channel_has_access(
user.id,
channel,
permission='write',
@@ -1559,7 +1586,7 @@ async def delete_message_by_id(
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- Messages.delete_message_by_id(message_id, db=db)
+ await Messages.delete_message_by_id(message_id, db=db)
await sio.emit(
'events:channel',
{
@@ -1580,7 +1607,7 @@ async def delete_message_by_id(
if message.parent_id:
# If this message is a reply, emit to the parent message as well
- parent_message = Messages.get_message_by_id(message.parent_id, db=db)
+ parent_message = await Messages.get_message_by_id(message.parent_id, db=db)
if parent_message:
await sio.emit(
@@ -1610,9 +1637,9 @@ async def delete_message_by_id(
@router.get('/webhooks/{webhook_id}/profile/image')
-def get_webhook_profile_image(webhook_id: str, user=Depends(get_verified_user)):
+async def get_webhook_profile_image(webhook_id: str, user=Depends(get_verified_user)):
"""Get webhook profile image by webhook ID."""
- webhook = Channels.get_webhook_by_id(webhook_id)
+ webhook = await Channels.get_webhook_by_id(webhook_id)
if not webhook:
# Return default favicon if webhook not found
return FileResponse(f'{STATIC_DIR}/favicon.png')
@@ -1648,18 +1675,18 @@ async def get_channel_webhooks(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
# Only channel managers can view webhooks
- if not Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
+ if not await Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.UNAUTHORIZED)
- return Channels.get_webhooks_by_channel_id(id, db=db)
+ return await Channels.get_webhooks_by_channel_id(id, db=db)
@router.post('/{id}/webhooks/create', response_model=ChannelWebhookModel)
@@ -1668,18 +1695,18 @@ async def create_channel_webhook(
id: str,
form_data: ChannelWebhookForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
# Only channel managers can create webhooks
- if not Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
+ if not await Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.UNAUTHORIZED)
- webhook = Channels.insert_webhook(id, user.id, form_data, db=db)
+ webhook = await Channels.insert_webhook(id, user.id, form_data, db=db)
if not webhook:
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
@@ -1693,22 +1720,22 @@ async def update_channel_webhook(
webhook_id: str,
form_data: ChannelWebhookForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
# Only channel managers can update webhooks
- if not Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
+ if not await Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.UNAUTHORIZED)
- webhook = Channels.get_webhook_by_id(webhook_id, db=db)
+ webhook = await Channels.get_webhook_by_id(webhook_id, db=db)
if not webhook or webhook.channel_id != id:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
- updated = Channels.update_webhook_by_id(webhook_id, form_data, db=db)
+ updated = await Channels.update_webhook_by_id(webhook_id, form_data, db=db)
if not updated:
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
@@ -1721,22 +1748,22 @@ async def delete_channel_webhook(
id: str,
webhook_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- check_channels_access(request, user)
- channel = Channels.get_channel_by_id(id, db=db)
+ await check_channels_access(request, user)
+ channel = await Channels.get_channel_by_id(id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
# Only channel managers can delete webhooks
- if not Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
+ if not await Channels.is_user_channel_manager(channel.id, user.id, db=db) and user.role != 'admin':
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.UNAUTHORIZED)
- webhook = Channels.get_webhook_by_id(webhook_id, db=db)
+ webhook = await Channels.get_webhook_by_id(webhook_id, db=db)
if not webhook or webhook.channel_id != id:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
- return Channels.delete_webhook_by_id(webhook_id, db=db)
+ return await Channels.delete_webhook_by_id(webhook_id, db=db)
############################
@@ -1754,25 +1781,25 @@ async def post_webhook_message(
webhook_id: str,
token: str,
form_data: WebhookMessageForm,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Public endpoint to post messages via webhook. No authentication required."""
- check_channels_access(request)
+ await check_channels_access(request)
# Validate webhook
- webhook = Channels.get_webhook_by_id_and_token(webhook_id, token, db=db)
+ webhook = await Channels.get_webhook_by_id_and_token(webhook_id, token, db=db)
if not webhook:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
- detail='Invalid webhook URL',
+ detail=ERROR_MESSAGES.INVALID_URL,
)
- channel = Channels.get_channel_by_id(webhook.channel_id, db=db)
+ channel = await Channels.get_channel_by_id(webhook.channel_id, db=db)
if not channel:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
# Create message with webhook identity stored in meta
- message = Messages.insert_new_message(
+ message = await Messages.insert_new_message(
MessageForm(content=form_data.content, meta={'webhook': {'id': webhook.id}}),
webhook.channel_id,
webhook.user_id, # Required for DB but webhook info in meta takes precedence
@@ -1782,14 +1809,14 @@ async def post_webhook_message(
if not message:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail='Failed to create message',
+ detail=ERROR_MESSAGES.DEFAULT('Failed to create message'),
)
# Update last_used_at
- Channels.update_webhook_last_used_at(webhook_id, db=db)
+ await Channels.update_webhook_last_used_at(webhook_id, db=db)
# Get full message and emit event
- message = Messages.get_message_by_id(message.id, db=db)
+ message = await Messages.get_message_by_id(message.id, db=db)
event_data = {
'channel_id': channel.id,
diff --git a/backend/open_webui/routers/chats.py b/backend/open_webui/routers/chats.py
index 2d12e02523..1980d22362 100644
--- a/backend/open_webui/routers/chats.py
+++ b/backend/open_webui/routers/chats.py
@@ -1,7 +1,8 @@
import json
import logging
from typing import Optional
-from sqlalchemy.orm import Session
+from uuid import uuid4
+from sqlalchemy.ext.asyncio import AsyncSession
import asyncio
from fastapi.responses import StreamingResponse
@@ -25,7 +26,7 @@ from open_webui.models.chats import (
)
from open_webui.models.tags import TagModel, Tags
from open_webui.models.folders import Folders
-from open_webui.internal.db import get_session
+from open_webui.internal.db import get_async_session
from open_webui.config import ENABLE_ADMIN_CHAT_ACCESS, ENABLE_ADMIN_EXPORT
from open_webui.constants import ERROR_MESSAGES
@@ -49,19 +50,19 @@ router = APIRouter()
@router.get('/', response_model=list[ChatTitleIdResponse])
@router.get('/list', response_model=list[ChatTitleIdResponse])
-def get_session_user_chat_list(
+async def get_session_user_chat_list(
user=Depends(get_verified_user),
page: Optional[int] = None,
include_pinned: Optional[bool] = False,
include_folders: Optional[bool] = False,
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
if page is not None:
limit = 60
skip = (page - 1) * limit
- return Chats.get_chat_title_id_list_by_user_id(
+ return await Chats.get_chat_title_id_list_by_user_id(
user.id,
include_folders=include_folders,
include_pinned=include_pinned,
@@ -70,7 +71,7 @@ def get_session_user_chat_list(
db=db,
)
else:
- return Chats.get_chat_title_id_list_by_user_id(
+ return await Chats.get_chat_title_id_list_by_user_id(
user.id,
include_folders=include_folders,
include_pinned=include_pinned,
@@ -88,17 +89,17 @@ def get_session_user_chat_list(
@router.get('/stats/usage', response_model=ChatUsageStatsListResponse)
-def get_session_user_chat_usage_stats(
+async def get_session_user_chat_usage_stats(
items_per_page: Optional[int] = 50,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
limit = items_per_page
skip = (page - 1) * limit
- result = Chats.get_chats_by_user_id(user.id, skip=skip, limit=limit, db=db)
+ result = await Chats.get_chats_by_user_id(user.id, skip=skip, limit=limit, db=db)
chats = result.items
total = result.total
@@ -332,11 +333,11 @@ def _process_chat_for_export(chat) -> Optional[ChatStatsExport]:
return None
-def calculate_chat_stats(user_id, skip=0, limit=10, filter=None):
+async def calculate_chat_stats(user_id, skip=0, limit=10, filter=None):
if filter is None:
filter = {}
- result = Chats.get_chats_by_user_id(
+ result = await Chats.get_chats_by_user_id(
user_id,
skip=skip,
limit=limit,
@@ -352,12 +353,12 @@ def calculate_chat_stats(user_id, skip=0, limit=10, filter=None):
return chat_stats_export_list, result.total
-def generate_chat_stats_jsonl_generator(user_id, filter):
+async def generate_chat_stats_jsonl_generator(user_id, filter):
"""
- Synchronous generator for streaming chat stats export.
+ Async generator for streaming chat stats export.
NOTE: We intentionally do NOT pass a shared db session here. Instead, we let
- each batch create its own short-lived session via get_db_context(None).
+ each batch create its own short-lived session via get_async_db_context(None).
This is critical for SQLite in low-resource environments because:
1. SQLite uses file-level locking
2. Holding a session open for the entire streaming duration blocks other requests
@@ -368,12 +369,12 @@ def generate_chat_stats_jsonl_generator(user_id, filter):
while True:
# Each batch gets its own session that closes after the query
- result = Chats.get_chats_by_user_id(
+ result = await Chats.get_chats_by_user_id(
user_id,
filter=filter,
skip=skip,
limit=limit,
- db=None, # Let get_db_context create a fresh session per batch
+ db=None, # Let get_async_db_context create a fresh session per batch
)
if not result.items:
break
@@ -421,7 +422,7 @@ async def export_chat_stats(
limit = CHAT_EXPORT_PAGE_ITEM_COUNT
skip = (page - 1) * limit
- chat_stats_export_list, total = await asyncio.to_thread(calculate_chat_stats, user.id, skip, limit, filter)
+ chat_stats_export_list, total = await calculate_chat_stats(user.id, skip, limit, filter)
return ChatStatsExportList(items=chat_stats_export_list, total=total, page=page)
@@ -440,7 +441,7 @@ async def export_single_chat_stats(
request: Request,
chat_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""
Export stats for exactly one chat by ID.
@@ -454,7 +455,7 @@ async def export_single_chat_stats(
)
try:
- chat = Chats.get_chat_by_id(chat_id, db=db)
+ chat = await Chats.get_chat_by_id(chat_id, db=db)
if not chat:
raise HTTPException(
@@ -469,8 +470,8 @@ async def export_single_chat_stats(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- # Process the chat for export
- chat_stats = await asyncio.to_thread(_process_chat_for_export, chat)
+ # Process the chat for export (pure computation, no DB)
+ chat_stats = _process_chat_for_export(chat)
if not chat_stats:
raise HTTPException(
@@ -491,15 +492,17 @@ async def export_single_chat_stats(
async def delete_all_user_chats(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role == 'user' and not has_permission(user.id, 'chat.delete', request.app.state.config.USER_PERMISSIONS):
+ if user.role == 'user' and not await has_permission(
+ user.id, 'chat.delete', request.app.state.config.USER_PERMISSIONS
+ ):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- result = Chats.delete_chats_by_user_id(user.id, db=db)
+ result = await Chats.delete_chats_by_user_id(user.id, db=db)
return result
@@ -516,7 +519,7 @@ async def get_user_chat_list_by_user_id(
order_by: Optional[str] = None,
direction: Optional[str] = None,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not ENABLE_ADMIN_CHAT_ACCESS:
raise HTTPException(
@@ -538,7 +541,9 @@ async def get_user_chat_list_by_user_id(
if direction:
filter['direction'] = direction
- return Chats.get_chat_list_by_user_id(user_id, include_archived=True, filter=filter, skip=skip, limit=limit, db=db)
+ return await Chats.get_chat_list_by_user_id(
+ user_id, include_archived=True, filter=filter, skip=skip, limit=limit, db=db
+ )
############################
@@ -550,10 +555,10 @@ async def get_user_chat_list_by_user_id(
async def create_new_chat(
form_data: ChatForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
- chat = Chats.insert_new_chat(user.id, form_data, db=db)
+ chat = await Chats.insert_new_chat(str(uuid4()), user.id, form_data, db=db)
return ChatResponse(**chat.model_dump())
except Exception as e:
log.exception(e)
@@ -569,10 +574,10 @@ async def create_new_chat(
async def import_chats(
form_data: ChatsImportForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
- chats = Chats.import_chats(user.id, form_data.chats, db=db)
+ chats = await Chats.import_chats(user.id, form_data.chats, db=db)
return chats
except Exception as e:
log.exception(e)
@@ -585,11 +590,11 @@ async def import_chats(
@router.get('/search', response_model=list[ChatTitleIdResponse])
-def search_user_chats(
+async def search_user_chats(
text: str,
page: Optional[int] = None,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if page is None:
page = 1
@@ -599,7 +604,7 @@ def search_user_chats(
chat_list = [
ChatTitleIdResponse(**chat.model_dump())
- for chat in Chats.get_chats_by_user_id_and_search_text(user.id, text, skip=skip, limit=limit, db=db)
+ for chat in await Chats.get_chats_by_user_id_and_search_text(user.id, text, skip=skip, limit=limit, db=db)
]
# Delete tag if no chat is found
@@ -607,9 +612,9 @@ def search_user_chats(
if page == 1 and len(words) == 1 and words[0].startswith('tag:'):
tag_id = words[0].replace('tag:', '')
if len(chat_list) == 0:
- if Tags.get_tag_by_name_and_user_id(tag_id, user.id, db=db):
+ if await Tags.get_tag_by_name_and_user_id(tag_id, user.id, db=db):
log.debug(f'deleting tag: {tag_id}')
- Tags.delete_tag_by_name_and_user_id(tag_id, user.id, db=db)
+ await Tags.delete_tag_by_name_and_user_id(tag_id, user.id, db=db)
return chat_list
@@ -620,15 +625,17 @@ def search_user_chats(
@router.get('/folder/{folder_id}', response_model=list[ChatResponse])
-async def get_chats_by_folder_id(folder_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_chats_by_folder_id(
+ folder_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
folder_ids = [folder_id]
- children_folders = Folders.get_children_folders_by_id_and_user_id(folder_id, user.id, db=db)
+ children_folders = await Folders.get_children_folders_by_id_and_user_id(folder_id, user.id, db=db)
if children_folders:
folder_ids.extend([folder.id for folder in children_folders])
return [
ChatResponse(**chat.model_dump())
- for chat in Chats.get_chats_by_folder_ids_and_user_id(folder_ids, user.id, db=db)
+ for chat in await Chats.get_chats_by_folder_ids_and_user_id(folder_ids, user.id, db=db)
]
@@ -637,13 +644,13 @@ async def get_chat_list_by_folder_id(
folder_id: str,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
limit = 10
skip = (page - 1) * limit
- chats = Chats.get_chats_by_folder_id_and_user_id(folder_id, user.id, skip=skip, limit=limit, db=db)
+ chats = await Chats.get_chats_by_folder_id_and_user_id(folder_id, user.id, skip=skip, limit=limit, db=db)
return [
{'title': chat.title, 'id': chat.id, 'updated_at': chat.updated_at, 'last_read_at': chat.last_read_at}
for chat in chats
@@ -660,8 +667,8 @@ async def get_chat_list_by_folder_id(
@router.get('/pinned', response_model=list[ChatTitleIdResponse])
-async def get_user_pinned_chats(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- return Chats.get_pinned_chats_by_user_id(user.id, db=db)
+async def get_user_pinned_chats(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ return await Chats.get_pinned_chats_by_user_id(user.id, db=db)
############################
@@ -670,8 +677,8 @@ async def get_user_pinned_chats(user=Depends(get_verified_user), db: Session = D
@router.get('/all', response_model=list[ChatResponse])
-async def get_user_chats(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- result = Chats.get_chats_by_user_id(user.id, db=db)
+async def get_user_chats(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ result = await Chats.get_chats_by_user_id(user.id, db=db)
return [ChatResponse(**chat.model_dump()) for chat in result.items]
@@ -681,8 +688,8 @@ async def get_user_chats(user=Depends(get_verified_user), db: Session = Depends(
@router.get('/all/archived', response_model=list[ChatResponse])
-async def get_user_archived_chats(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- return [ChatResponse(**chat.model_dump()) for chat in Chats.get_archived_chats_by_user_id(user.id, db=db)]
+async def get_user_archived_chats(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ return [ChatResponse(**chat.model_dump()) for chat in await Chats.get_archived_chats_by_user_id(user.id, db=db)]
############################
@@ -691,9 +698,9 @@ async def get_user_archived_chats(user=Depends(get_verified_user), db: Session =
@router.get('/all/tags', response_model=list[TagModel])
-async def get_all_user_tags(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_all_user_tags(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
try:
- tags = Tags.get_tags_by_user_id(user.id, db=db)
+ tags = await Tags.get_tags_by_user_id(user.id, db=db)
return tags
except Exception as e:
log.exception(e)
@@ -706,13 +713,13 @@ async def get_all_user_tags(user=Depends(get_verified_user), db: Session = Depen
@router.get('/all/db', response_model=list[ChatResponse])
-async def get_all_user_chats_in_db(user=Depends(get_admin_user), db: Session = Depends(get_session)):
+async def get_all_user_chats_in_db(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
if not ENABLE_ADMIN_EXPORT:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- return [ChatResponse(**chat.model_dump()) for chat in Chats.get_chats(db=db)]
+ return [ChatResponse(**chat.model_dump()) for chat in await Chats.get_chats(db=db)]
############################
@@ -727,7 +734,7 @@ async def get_archived_session_user_chat_list(
order_by: Optional[str] = None,
direction: Optional[str] = None,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if page is None:
page = 1
@@ -743,7 +750,7 @@ async def get_archived_session_user_chat_list(
if direction:
filter['direction'] = direction
- return Chats.get_archived_chat_list_by_user_id(
+ return await Chats.get_archived_chat_list_by_user_id(
user.id,
filter=filter,
skip=skip,
@@ -758,8 +765,8 @@ async def get_archived_session_user_chat_list(
@router.post('/archive/all', response_model=bool)
-async def archive_all_chats(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- return Chats.archive_all_chats_by_user_id(user.id, db=db)
+async def archive_all_chats(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ return await Chats.archive_all_chats_by_user_id(user.id, db=db)
############################
@@ -768,8 +775,8 @@ async def archive_all_chats(user=Depends(get_verified_user), db: Session = Depen
@router.post('/unarchive/all', response_model=bool)
-async def unarchive_all_chats(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- return Chats.unarchive_all_chats_by_user_id(user.id, db=db)
+async def unarchive_all_chats(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ return await Chats.unarchive_all_chats_by_user_id(user.id, db=db)
############################
@@ -784,7 +791,7 @@ async def get_shared_session_user_chat_list(
order_by: Optional[str] = None,
direction: Optional[str] = None,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if page is None:
page = 1
@@ -800,7 +807,7 @@ async def get_shared_session_user_chat_list(
if direction:
filter['direction'] = direction
- return Chats.get_shared_chat_list_by_user_id(
+ return await Chats.get_shared_chat_list_by_user_id(
user.id,
filter=filter,
skip=skip,
@@ -815,14 +822,16 @@ async def get_shared_session_user_chat_list(
@router.get('/share/{share_id}', response_model=Optional[ChatResponse])
-async def get_shared_chat_by_id(share_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_shared_chat_by_id(
+ share_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
if user.role == 'pending':
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND)
if user.role == 'user' or (user.role == 'admin' and not ENABLE_ADMIN_CHAT_ACCESS):
- chat = Chats.get_chat_by_share_id(share_id, db=db)
+ chat = await Chats.get_chat_by_share_id(share_id, db=db)
elif user.role == 'admin' and ENABLE_ADMIN_CHAT_ACCESS:
- chat = Chats.get_chat_by_id(share_id, db=db)
+ chat = await Chats.get_chat_by_id(share_id, db=db)
if chat:
return ChatResponse(**chat.model_dump())
@@ -849,11 +858,13 @@ class TagFilterForm(TagForm):
async def get_user_chat_list_by_tag_name(
form_data: TagFilterForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chats = Chats.get_chat_list_by_user_id_and_tag_name(user.id, form_data.name, form_data.skip, form_data.limit, db=db)
+ chats = await Chats.get_chat_list_by_user_id_and_tag_name(
+ user.id, form_data.name, form_data.skip, form_data.limit, db=db
+ )
if len(chats) == 0:
- Tags.delete_tag_by_name_and_user_id(form_data.name, user.id, db=db)
+ await Tags.delete_tag_by_name_and_user_id(form_data.name, user.id, db=db)
return chats
@@ -864,8 +875,8 @@ async def get_user_chat_list_by_tag_name(
@router.get('/{id}', response_model=Optional[ChatResponse])
-async def get_chat_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def get_chat_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
return ChatResponse(**chat.model_dump())
@@ -884,12 +895,12 @@ async def update_chat_by_id(
id: str,
form_data: ChatForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
updated_chat = {**chat.chat, **form_data.chat}
- chat = Chats.update_chat_by_id(id, updated_chat, db=db)
+ chat = await Chats.update_chat_by_id(id, updated_chat, db=db)
return ChatResponse(**chat.model_dump())
else:
raise HTTPException(
@@ -911,9 +922,9 @@ async def update_chat_message_by_id(
message_id: str,
form_data: MessageForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id(id, db=db)
+ chat = await Chats.get_chat_by_id(id, db=db)
if not chat:
raise HTTPException(
@@ -927,7 +938,7 @@ async def update_chat_message_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- chat = Chats.upsert_message_to_chat_by_id_and_message_id(
+ chat = await Chats.upsert_message_to_chat_by_id_and_message_id(
id,
message_id,
{
@@ -935,7 +946,7 @@ async def update_chat_message_by_id(
},
)
- event_emitter = get_event_emitter(
+ event_emitter = await get_event_emitter(
{
'user_id': user.id,
'chat_id': id,
@@ -973,9 +984,9 @@ async def send_chat_message_event_by_id(
message_id: str,
form_data: EventForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id(id, db=db)
+ chat = await Chats.get_chat_by_id(id, db=db)
if not chat:
raise HTTPException(
@@ -989,7 +1000,7 @@ async def send_chat_message_event_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- event_emitter = get_event_emitter(
+ event_emitter = await get_event_emitter(
{
'user_id': user.id,
'chat_id': id,
@@ -1017,36 +1028,36 @@ async def delete_chat_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if user.role == 'admin':
- chat = Chats.get_chat_by_id(id, db=db)
+ chat = await Chats.get_chat_by_id(id, db=db)
if not chat:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=ERROR_MESSAGES.NOT_FOUND,
)
- Chats.delete_orphan_tags_for_user(chat.meta.get('tags', []), user.id, threshold=1, db=db)
+ await Chats.delete_orphan_tags_for_user(chat.meta.get('tags', []), user.id, threshold=1, db=db)
- result = Chats.delete_chat_by_id(id, db=db)
+ result = await Chats.delete_chat_by_id(id, db=db)
return result
else:
- if not has_permission(user.id, 'chat.delete', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'chat.delete', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if not chat:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=ERROR_MESSAGES.NOT_FOUND,
)
- Chats.delete_orphan_tags_for_user(chat.meta.get('tags', []), user.id, threshold=1, db=db)
+ await Chats.delete_orphan_tags_for_user(chat.meta.get('tags', []), user.id, threshold=1, db=db)
- result = Chats.delete_chat_by_id_and_user_id(id, user.id, db=db)
+ result = await Chats.delete_chat_by_id_and_user_id(id, user.id, db=db)
return result
@@ -1056,8 +1067,10 @@ async def delete_chat_by_id(
@router.get('/{id}/pinned', response_model=Optional[bool])
-async def get_pinned_status_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def get_pinned_status_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
return chat.pinned
else:
@@ -1070,10 +1083,10 @@ async def get_pinned_status_by_id(id: str, user=Depends(get_verified_user), db:
@router.post('/{id}/pin', response_model=Optional[ChatResponse])
-async def pin_chat_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def pin_chat_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
- chat = Chats.toggle_chat_pinned_by_id(id, db=db)
+ chat = await Chats.toggle_chat_pinned_by_id(id, db=db)
return chat
else:
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT())
@@ -1093,9 +1106,9 @@ async def clone_chat_by_id(
form_data: CloneForm,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
updated_chat = {
**chat.chat,
@@ -1104,7 +1117,7 @@ async def clone_chat_by_id(
'title': form_data.title if form_data.title else f'Clone of {chat.title}',
}
- chats = Chats.import_chats(
+ chats = await Chats.import_chats(
user.id,
[
ChatImportForm(
@@ -1137,11 +1150,13 @@ async def clone_chat_by_id(
@router.post('/{id}/clone/shared', response_model=Optional[ChatResponse])
-async def clone_shared_chat_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def clone_shared_chat_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
if user.role == 'admin':
- chat = Chats.get_chat_by_id(id, db=db)
+ chat = await Chats.get_chat_by_id(id, db=db)
else:
- chat = Chats.get_chat_by_share_id(id, db=db)
+ chat = await Chats.get_chat_by_share_id(id, db=db)
if chat:
updated_chat = {
@@ -1151,7 +1166,7 @@ async def clone_shared_chat_by_id(id: str, user=Depends(get_verified_user), db:
'title': f'Clone of {chat.title}',
}
- chats = Chats.import_chats(
+ chats = await Chats.import_chats(
user.id,
[
ChatImportForm(
@@ -1184,18 +1199,18 @@ async def clone_shared_chat_by_id(id: str, user=Depends(get_verified_user), db:
@router.post('/{id}/archive', response_model=Optional[ChatResponse])
-async def archive_chat_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def archive_chat_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
- chat = Chats.toggle_chat_archive_by_id(id, db=db)
+ chat = await Chats.toggle_chat_archive_by_id(id, db=db)
tag_ids = chat.meta.get('tags', [])
if chat.archived:
# Archived chats are excluded from count — clean up orphans
- Chats.delete_orphan_tags_for_user(tag_ids, user.id, db=db)
+ await Chats.delete_orphan_tags_for_user(tag_ids, user.id, db=db)
else:
# Unarchived — ensure tag rows exist
- Tags.ensure_tags_exist(tag_ids, user.id, db=db)
+ await Tags.ensure_tags_exist(tag_ids, user.id, db=db)
return ChatResponse(**chat.model_dump())
else:
@@ -1212,24 +1227,24 @@ async def share_chat_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if (user.role != 'admin') and (
- not has_permission(user.id, 'chat.share', request.app.state.config.USER_PERMISSIONS)
+ not await has_permission(user.id, 'chat.share', request.app.state.config.USER_PERMISSIONS)
):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
if chat.share_id:
- shared_chat = Chats.update_shared_chat_by_chat_id(chat.id, db=db)
+ shared_chat = await Chats.update_shared_chat_by_chat_id(chat.id, db=db)
return ChatResponse(**shared_chat.model_dump())
- shared_chat = Chats.insert_shared_chat_by_chat_id(chat.id, db=db)
+ shared_chat = await Chats.insert_shared_chat_by_chat_id(chat.id, db=db)
if not shared_chat:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -1250,14 +1265,16 @@ async def share_chat_by_id(
@router.delete('/{id}/share', response_model=Optional[bool])
-async def delete_shared_chat_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def delete_shared_chat_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
if not chat.share_id:
return False
- result = Chats.delete_shared_chat_by_chat_id(id, db=db)
- update_result = Chats.update_chat_share_id_by_id(id, None, db=db)
+ result = await Chats.delete_shared_chat_by_chat_id(id, db=db)
+ update_result = await Chats.update_chat_share_id_by_id(id, None, db=db)
return result and update_result != None
else:
@@ -1281,11 +1298,11 @@ async def update_chat_folder_id_by_id(
id: str,
form_data: ChatFolderIdForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
- chat = Chats.update_chat_folder_id_by_id_and_user_id(id, user.id, form_data.folder_id, db=db)
+ chat = await Chats.update_chat_folder_id_by_id_and_user_id(id, user.id, form_data.folder_id, db=db)
return ChatResponse(**chat.model_dump())
else:
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT())
@@ -1297,11 +1314,11 @@ async def update_chat_folder_id_by_id(
@router.get('/{id}/tags', response_model=list[TagModel])
-async def get_chat_tags_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def get_chat_tags_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
tags = chat.meta.get('tags', [])
- return Tags.get_tags_by_ids_and_user_id(tags, user.id, db=db)
+ return await Tags.get_tags_by_ids_and_user_id(tags, user.id, db=db)
else:
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1316,9 +1333,9 @@ async def add_tag_by_id_and_tag_name(
id: str,
form_data: TagForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
tags = chat.meta.get('tags', [])
tag_id = form_data.name.replace(' ', '_').lower()
@@ -1330,11 +1347,11 @@ async def add_tag_by_id_and_tag_name(
)
if tag_id not in tags:
- Chats.add_chat_tag_by_id_and_user_id_and_tag_name(id, user.id, form_data.name, db=db)
+ await Chats.add_chat_tag_by_id_and_user_id_and_tag_name(id, user.id, form_data.name, db=db)
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
tags = chat.meta.get('tags', [])
- return Tags.get_tags_by_ids_and_user_id(tags, user.id, db=db)
+ return await Tags.get_tags_by_ids_and_user_id(tags, user.id, db=db)
else:
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT())
@@ -1349,18 +1366,18 @@ async def delete_tag_by_id_and_tag_name(
id: str,
form_data: TagForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
- Chats.delete_tag_by_id_and_user_id_and_tag_name(id, user.id, form_data.name, db=db)
+ await Chats.delete_tag_by_id_and_user_id_and_tag_name(id, user.id, form_data.name, db=db)
- if Chats.count_chats_by_tag_name_and_user_id(form_data.name, user.id, db=db) == 0:
- Tags.delete_tag_by_name_and_user_id(form_data.name, user.id, db=db)
+ if await Chats.count_chats_by_tag_name_and_user_id(form_data.name, user.id, db=db) == 0:
+ await Tags.delete_tag_by_name_and_user_id(form_data.name, user.id, db=db)
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
tags = chat.meta.get('tags', [])
- return Tags.get_tags_by_ids_and_user_id(tags, user.id, db=db)
+ return await Tags.get_tags_by_ids_and_user_id(tags, user.id, db=db)
else:
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -1371,12 +1388,14 @@ async def delete_tag_by_id_and_tag_name(
@router.delete('/{id}/tags/all', response_model=Optional[bool])
-async def delete_all_tags_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- chat = Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
+async def delete_all_tags_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ chat = await Chats.get_chat_by_id_and_user_id(id, user.id, db=db)
if chat:
old_tags = chat.meta.get('tags', [])
- Chats.delete_all_tags_by_id_and_user_id(id, user.id, db=db)
- Chats.delete_orphan_tags_for_user(old_tags, user.id, db=db)
+ await Chats.delete_all_tags_by_id_and_user_id(id, user.id, db=db)
+ await Chats.delete_orphan_tags_for_user(old_tags, user.id, db=db)
return True
else:
diff --git a/backend/open_webui/routers/configs.py b/backend/open_webui/routers/configs.py
index 041c4ad935..7c54c09039 100644
--- a/backend/open_webui/routers/configs.py
+++ b/backend/open_webui/routers/configs.py
@@ -8,7 +8,7 @@ from typing import Optional
from open_webui.env import AIOHTTP_CLIENT_TIMEOUT
from open_webui.utils.auth import get_admin_user, get_verified_user
-from open_webui.config import get_config, save_config
+from open_webui.config import get_config, save_config, async_save_config
from open_webui.config import BannerModel
from open_webui.utils.tools import (
@@ -27,6 +27,7 @@ from open_webui.utils.oauth import (
get_oauth_client_info_with_static_credentials,
encrypt_data,
decrypt_data,
+ resolve_oauth_client_info,
OAuthClientInformationFull,
)
from mcp.shared.auth import OAuthMetadata
@@ -49,7 +50,7 @@ class ImportConfigForm(BaseModel):
@router.post('/import', response_model=dict)
async def import_config(form_data: ImportConfigForm, user=Depends(get_admin_user)):
- save_config(form_data.config)
+ await async_save_config(form_data.config)
return get_config()
@@ -203,9 +204,7 @@ async def set_tool_servers_config(
if auth_type in ('oauth_2.1', 'oauth_2.1_static') and server_id:
try:
- oauth_client_info = connection.get('info', {}).get('oauth_client_info', '')
- oauth_client_info = decrypt_data(oauth_client_info)
-
+ oauth_client_info = resolve_oauth_client_info(connection)
request.app.state.oauth_client_manager.add_client(
f'{server_type}:{server_id}',
OAuthClientInformationFull(**oauth_client_info),
diff --git a/backend/open_webui/routers/evaluations.py b/backend/open_webui/routers/evaluations.py
index 9805f2ece2..072c7fa732 100644
--- a/backend/open_webui/routers/evaluations.py
+++ b/backend/open_webui/routers/evaluations.py
@@ -20,8 +20,8 @@ from open_webui.models.feedbacks import (
from open_webui.constants import ERROR_MESSAGES
from open_webui.utils.auth import get_admin_user, get_verified_user
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
@@ -208,10 +208,10 @@ class LeaderboardResponse(BaseModel):
async def get_leaderboard(
query: Optional[str] = None,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get model leaderboard with Elo ratings. Query filters by tag similarity."""
- feedbacks = Feedbacks.get_feedbacks_for_leaderboard(db=db)
+ feedbacks = await Feedbacks.get_feedbacks_for_leaderboard(db=db)
similarities = None
if query and query.strip():
@@ -244,10 +244,10 @@ async def get_model_history(
model_id: str,
days: int = 30,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get daily win/loss history for a specific model."""
- history = Feedbacks.get_model_evaluation_history(model_id=model_id, days=days, db=db)
+ history = await Feedbacks.get_model_evaluation_history(model_id=model_id, days=days, db=db)
return ModelHistoryResponse(model_id=model_id, history=history)
@@ -292,24 +292,24 @@ async def update_config(
@router.get('/feedbacks/models', response_model=list[str])
-async def get_feedback_model_ids(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- return Feedbacks.get_distinct_model_ids(db=db)
+async def get_feedback_model_ids(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ return await Feedbacks.get_distinct_model_ids(db=db)
@router.get('/feedbacks/all', response_model=list[FeedbackResponse])
-async def get_all_feedbacks(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- feedbacks = Feedbacks.get_all_feedbacks(db=db)
+async def get_all_feedbacks(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ feedbacks = await Feedbacks.get_all_feedbacks(db=db)
return feedbacks
@router.get('/feedbacks/all/ids', response_model=list[FeedbackIdResponse])
-async def get_all_feedback_ids(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- return Feedbacks.get_all_feedback_ids(db=db)
+async def get_all_feedback_ids(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ return await Feedbacks.get_all_feedback_ids(db=db)
@router.delete('/feedbacks/all')
-async def delete_all_feedbacks(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- success = Feedbacks.delete_all_feedbacks(db=db)
+async def delete_all_feedbacks(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ success = await Feedbacks.delete_all_feedbacks(db=db)
return success
@@ -317,23 +317,23 @@ async def delete_all_feedbacks(user=Depends(get_admin_user), db: Session = Depen
async def export_all_feedbacks(
model_id: Optional[str] = None,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- feedbacks = Feedbacks.get_all_feedbacks(db=db)
+ feedbacks = await Feedbacks.get_all_feedbacks(db=db)
if model_id:
feedbacks = [f for f in feedbacks if f.data and f.data.get('model_id') == model_id]
return feedbacks
@router.get('/feedbacks/user', response_model=list[FeedbackUserResponse])
-async def get_feedbacks(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- feedbacks = Feedbacks.get_feedbacks_by_user_id(user.id, db=db)
+async def get_feedbacks(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ feedbacks = await Feedbacks.get_feedbacks_by_user_id(user.id, db=db)
return feedbacks
@router.delete('/feedbacks', response_model=bool)
-async def delete_feedbacks(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- success = Feedbacks.delete_feedbacks_by_user_id(user.id, db=db)
+async def delete_feedbacks(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ success = await Feedbacks.delete_feedbacks_by_user_id(user.id, db=db)
return success
@@ -347,7 +347,7 @@ async def get_feedbacks(
page: Optional[int] = 1,
model_id: Optional[str] = None,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
limit = PAGE_ITEM_COUNT
@@ -362,7 +362,7 @@ async def get_feedbacks(
if model_id:
filter['model_id'] = model_id
- result = Feedbacks.get_feedback_items(filter=filter, skip=skip, limit=limit, db=db)
+ result = await Feedbacks.get_feedback_items(filter=filter, skip=skip, limit=limit, db=db)
return result
@@ -371,9 +371,9 @@ async def create_feedback(
request: Request,
form_data: FeedbackForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- feedback = Feedbacks.insert_new_feedback(user_id=user.id, form_data=form_data, db=db)
+ feedback = await Feedbacks.insert_new_feedback(user_id=user.id, form_data=form_data, db=db)
if not feedback:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -384,11 +384,11 @@ async def create_feedback(
@router.get('/feedback/{id}', response_model=FeedbackModel)
-async def get_feedback_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_feedback_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
if user.role == 'admin':
- feedback = Feedbacks.get_feedback_by_id(id=id, db=db)
+ feedback = await Feedbacks.get_feedback_by_id(id=id, db=db)
else:
- feedback = Feedbacks.get_feedback_by_id_and_user_id(id=id, user_id=user.id, db=db)
+ feedback = await Feedbacks.get_feedback_by_id_and_user_id(id=id, user_id=user.id, db=db)
if not feedback:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -401,12 +401,12 @@ async def update_feedback_by_id(
id: str,
form_data: FeedbackForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if user.role == 'admin':
- feedback = Feedbacks.update_feedback_by_id(id=id, form_data=form_data, db=db)
+ feedback = await Feedbacks.update_feedback_by_id(id=id, form_data=form_data, db=db)
else:
- feedback = Feedbacks.update_feedback_by_id_and_user_id(id=id, user_id=user.id, form_data=form_data, db=db)
+ feedback = await Feedbacks.update_feedback_by_id_and_user_id(id=id, user_id=user.id, form_data=form_data, db=db)
if not feedback:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
@@ -415,11 +415,13 @@ async def update_feedback_by_id(
@router.delete('/feedback/{id}')
-async def delete_feedback_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def delete_feedback_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
if user.role == 'admin':
- success = Feedbacks.delete_feedback_by_id(id=id, db=db)
+ success = await Feedbacks.delete_feedback_by_id(id=id, db=db)
else:
- success = Feedbacks.delete_feedback_by_id_and_user_id(id=id, user_id=user.id, db=db)
+ success = await Feedbacks.delete_feedback_by_id_and_user_id(id=id, user_id=user.id, db=db)
if not success:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
diff --git a/backend/open_webui/routers/files.py b/backend/open_webui/routers/files.py
index 48172e744e..7ca1c2e73f 100644
--- a/backend/open_webui/routers/files.py
+++ b/backend/open_webui/routers/files.py
@@ -21,11 +21,11 @@ from fastapi import (
)
from fastapi.responses import FileResponse, StreamingResponse
-from sqlalchemy.orm import Session
-from open_webui.internal.db import get_session, SessionLocal
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import get_async_session, get_async_db_context
from open_webui.constants import ERROR_MESSAGES
-from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
from open_webui.models.channels import Channels
from open_webui.models.users import Users
@@ -48,7 +48,7 @@ from open_webui.routers.audio import transcribe
from open_webui.storage.provider import Storage
-from open_webui.config import BYPASS_ADMIN_ACCESS_CONTROL
+from open_webui.config import BYPASS_ADMIN_ACCESS_CONTROL, STORAGE_LOCAL_CACHE, STORAGE_PROVIDER, UPLOAD_DIR
from open_webui.utils.auth import get_admin_user, get_verified_user
from open_webui.utils.misc import strict_match_mime_type
from pydantic import BaseModel
@@ -88,16 +88,30 @@ def _is_text_file(file_path: str, chunk_size: int = 8192) -> bool:
return False
-def process_uploaded_file(
+def _cleanup_local_cache(file_path: str) -> None:
+ """Remove the local cached copy of a cloud-stored file after processing."""
+ if STORAGE_LOCAL_CACHE or STORAGE_PROVIDER == 'local':
+ return
+ try:
+ local_filename = os.path.basename(file_path)
+ local_path = os.path.join(UPLOAD_DIR, local_filename)
+ if os.path.isfile(local_path):
+ os.remove(local_path)
+ log.debug(f'Cleaned up local cache: {local_path}')
+ except OSError as e:
+ log.warning(f'Failed to clean up local cache for {file_path}: {e}')
+
+
+async def process_uploaded_file(
request,
file,
file_path,
file_item,
file_metadata,
user,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
):
- def _process_handler(db_session):
+ async def _process_handler(db_session):
try:
content_type = file.content_type
@@ -110,10 +124,10 @@ def process_uploaded_file(
stt_supported_content_types = getattr(request.app.state.config, 'STT_SUPPORTED_CONTENT_TYPES', [])
if strict_match_mime_type(stt_supported_content_types, content_type):
- file_path_processed = Storage.get_file(file_path)
+ file_path_processed = await asyncio.to_thread(Storage.get_file, file_path)
result = transcribe(request, file_path_processed, file_metadata, user)
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=file_item.id, content=result.get('text', '')),
user=user,
@@ -122,7 +136,7 @@ def process_uploaded_file(
elif (not content_type.startswith(('image/', 'video/'))) or (
request.app.state.config.CONTENT_EXTRACTION_ENGINE == 'external'
):
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=file_item.id),
user=user,
@@ -132,7 +146,7 @@ def process_uploaded_file(
raise Exception(f'File type {content_type} is not supported for processing')
else:
log.info(f'File type {file.content_type} is not provided, but trying to process anyway')
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=file_item.id),
user=user,
@@ -141,7 +155,7 @@ def process_uploaded_file(
except Exception as e:
log.error(f'Error processing file: {file_item.id}')
- Files.update_file_data_by_id(
+ await Files.update_file_data_by_id(
file_item.id,
{
'status': 'failed',
@@ -150,15 +164,18 @@ def process_uploaded_file(
db=db_session,
)
- if db:
- _process_handler(db)
- else:
- with SessionLocal() as db_session:
- _process_handler(db_session)
+ try:
+ if db:
+ await _process_handler(db)
+ else:
+ async with get_async_db_context() as db_session:
+ await _process_handler(db_session)
+ finally:
+ _cleanup_local_cache(file_path)
@router.post('/', response_model=FileModelResponse)
-def upload_file(
+async def upload_file(
request: Request,
background_tasks: BackgroundTasks,
file: UploadFile = File(...),
@@ -166,9 +183,9 @@ def upload_file(
process: bool = Query(True),
process_in_background: bool = Query(True),
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- return upload_file_handler(
+ return await upload_file_handler(
request,
file=file,
metadata=metadata,
@@ -180,7 +197,7 @@ def upload_file(
)
-def upload_file_handler(
+async def upload_file_handler(
request: Request,
file: UploadFile = File(...),
metadata: Optional[dict | str] = Form(None),
@@ -188,7 +205,7 @@ def upload_file_handler(
process_in_background: bool = Query(True),
user=Depends(get_verified_user),
background_tasks: Optional[BackgroundTasks] = None,
- db: Optional[Session] = None,
+ db: Optional[AsyncSession] = None,
):
log.info(f'file.content_type: {file.content_type} {process}')
@@ -225,7 +242,8 @@ def upload_file_handler(
id = str(uuid.uuid4())
name = filename
filename = f'{id}_{filename}'
- contents, file_path = Storage.upload_file(
+ contents, file_path = await asyncio.to_thread(
+ Storage.upload_file,
file.file,
filename,
{
@@ -236,7 +254,7 @@ def upload_file_handler(
},
)
- file_item = Files.insert_new_file(
+ file_item = await Files.insert_new_file(
user.id,
FileForm(
**{
@@ -258,9 +276,9 @@ def upload_file_handler(
)
if 'channel_id' in file_metadata:
- channel = Channels.get_channel_by_id_and_user_id(file_metadata['channel_id'], user.id, db=db)
+ channel = await Channels.get_channel_by_id_and_user_id(file_metadata['channel_id'], user.id, db=db)
if channel:
- Channels.add_file_to_channel_by_id(channel.id, file_item.id, user.id, db=db)
+ await Channels.add_file_to_channel_by_id(channel.id, file_item.id, user.id, db=db)
if process:
if background_tasks and process_in_background:
@@ -275,7 +293,7 @@ def upload_file_handler(
)
return {'status': True, **file_item.model_dump()}
else:
- process_uploaded_file(
+ await process_uploaded_file(
request,
file,
file_path,
@@ -317,12 +335,12 @@ async def list_files(
user=Depends(get_verified_user),
page: int = Query(1, ge=1, description='Page number (1-indexed)'),
content: bool = Query(True),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
skip = (page - 1) * PAGE_SIZE
user_id = None if (user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL) else user.id
- result = Files.get_file_list(user_id=user_id, skip=skip, limit=PAGE_SIZE, db=db)
+ result = await Files.get_file_list(user_id=user_id, skip=skip, limit=PAGE_SIZE, db=db)
if not content:
for file in result.items:
@@ -347,7 +365,7 @@ async def search_files(
skip: int = Query(0, ge=0, description='Number of files to skip'),
limit: int = Query(100, ge=1, le=1000, description='Maximum number of files to return'),
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""
Search for files by filename with support for wildcard patterns.
@@ -357,7 +375,7 @@ async def search_files(
user_id = None if (user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL) else user.id
# Use optimized database query with pagination
- files = Files.search_files(
+ files = await Files.search_files(
user_id=user_id,
filename=filename,
skip=skip,
@@ -385,12 +403,12 @@ async def search_files(
@router.delete('/all')
-async def delete_all_files(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- result = Files.delete_all_files(db=db)
+async def delete_all_files(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ result = await Files.delete_all_files(db=db)
if result:
try:
- Storage.delete_all_files()
- VECTOR_DB_CLIENT.reset()
+ await asyncio.to_thread(Storage.delete_all_files)
+ await ASYNC_VECTOR_DB_CLIENT.reset()
except Exception as e:
log.exception(e)
log.error('Error deleting files')
@@ -412,8 +430,8 @@ async def delete_all_files(user=Depends(get_admin_user), db: Session = Depends(g
@router.get('/{id}', response_model=Optional[FileModel])
-async def get_file_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- file = Files.get_file_by_id(id, db=db)
+async def get_file_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -421,7 +439,7 @@ async def get_file_by_id(id: str, user=Depends(get_verified_user), db: Session =
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'read', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'read', user, db=db):
return file
else:
raise HTTPException(
@@ -435,9 +453,9 @@ async def get_file_process_status(
id: str,
stream: bool = Query(False),
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- file = Files.get_file_by_id(id, db=db)
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -445,7 +463,7 @@ async def get_file_process_status(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'read', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'read', user, db=db):
if stream:
MAX_FILE_PROCESSING_DURATION = 3600 * 2
@@ -454,7 +472,7 @@ async def get_file_process_status(
# Each poll creates its own short-lived session to avoid holding a
# connection for hours. A WebSocket push would be more efficient.
for _ in range(MAX_FILE_PROCESSING_DURATION):
- file_item = Files.get_file_by_id(file_id) # Creates own session
+ file_item = await Files.get_file_by_id(file_id) # Creates own session
if file_item:
data = file_item.model_dump().get('data', {})
status = data.get('status')
@@ -495,8 +513,10 @@ async def get_file_process_status(
@router.get('/{id}/data/content')
-async def get_file_data_content_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- file = Files.get_file_by_id(id, db=db)
+async def get_file_data_content_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -504,7 +524,7 @@ async def get_file_data_content_by_id(id: str, user=Depends(get_verified_user),
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'read', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'read', user, db=db):
return {'content': file.data.get('content', '')}
else:
raise HTTPException(
@@ -523,14 +543,14 @@ class ContentForm(BaseModel):
@router.post('/{id}/data/content/update')
-def update_file_data_content_by_id(
+async def update_file_data_content_by_id(
request: Request,
id: str,
form_data: ContentForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- file = Files.get_file_by_id(id, db=db)
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -538,15 +558,15 @@ def update_file_data_content_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'write', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'write', user, db=db):
try:
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=id, content=form_data.content),
user=user,
db=db,
)
- file = Files.get_file_by_id(id=id, db=db)
+ file = await Files.get_file_by_id(id=id, db=db)
except Exception as e:
log.exception(e)
log.error(f'Error processing file: {file.id}')
@@ -554,13 +574,13 @@ def update_file_data_content_by_id(
# Propagate content change to all knowledge collections referencing
# this file. Without this the old embeddings remain in the knowledge
# collection and RAG returns both stale and current data (#20558).
- knowledges = Knowledges.get_knowledges_by_file_id(id, db=db)
+ knowledges = await Knowledges.get_knowledges_by_file_id(id, db=db)
for knowledge in knowledges:
try:
# Remove old embeddings for this file from the KB collection
- VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': id})
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': id})
# Re-add from the now-updated file-{file_id} collection
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=id, collection_name=knowledge.id),
user=user,
@@ -587,9 +607,9 @@ async def get_file_content_by_id(
id: str,
user=Depends(get_verified_user),
attachment: bool = Query(False),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- file = Files.get_file_by_id(id, db=db)
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -597,9 +617,9 @@ async def get_file_content_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'read', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'read', user, db=db):
try:
- file_path = Storage.get_file(file.path)
+ file_path = await asyncio.to_thread(Storage.get_file, file.path)
file_path = Path(file_path)
# Check if the file already exists in the cache
@@ -646,8 +666,10 @@ async def get_file_content_by_id(
@router.get('/{id}/content/html')
-async def get_html_file_content_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- file = Files.get_file_by_id(id, db=db)
+async def get_html_file_content_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -655,16 +677,16 @@ async def get_html_file_content_by_id(id: str, user=Depends(get_verified_user),
detail=ERROR_MESSAGES.NOT_FOUND,
)
- file_user = Users.get_user_by_id(file.user_id, db=db)
- if not file_user.role == 'admin':
+ file_user = await Users.get_user_by_id(file.user_id, db=db)
+ if not file_user or file_user.role != 'admin':
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'read', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'read', user, db=db):
try:
- file_path = Storage.get_file(file.path)
+ file_path = await asyncio.to_thread(Storage.get_file, file.path)
file_path = Path(file_path)
# Check if the file already exists in the cache
@@ -693,8 +715,10 @@ async def get_html_file_content_by_id(id: str, user=Depends(get_verified_user),
@router.get('/{id}/content/{file_name}')
-async def get_file_content_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- file = Files.get_file_by_id(id, db=db)
+async def get_file_content_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -702,7 +726,7 @@ async def get_file_content_by_id(id: str, user=Depends(get_verified_user), db: S
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'read', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'read', user, db=db):
file_path = file.path
# Handle Unicode filenames
@@ -711,7 +735,7 @@ async def get_file_content_by_id(id: str, user=Depends(get_verified_user), db: S
headers = {'Content-Disposition': f"attachment; filename*=UTF-8''{encoded_filename}"}
if file_path:
- file_path = Storage.get_file(file_path)
+ file_path = await asyncio.to_thread(Storage.get_file, file_path)
file_path = Path(file_path)
# Check if the file already exists in the cache
@@ -749,8 +773,8 @@ async def get_file_content_by_id(id: str, user=Depends(get_verified_user), db: S
@router.delete('/{id}')
-async def delete_file_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- file = Files.get_file_by_id(id, db=db)
+async def delete_file_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ file = await Files.get_file_by_id(id, db=db)
if not file:
raise HTTPException(
@@ -758,25 +782,25 @@ async def delete_file_by_id(id: str, user=Depends(get_verified_user), db: Sessio
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if file.user_id == user.id or user.role == 'admin' or has_access_to_file(id, 'write', user, db=db):
+ if file.user_id == user.id or user.role == 'admin' or await has_access_to_file(id, 'write', user, db=db):
# Clean up KB associations and embeddings before deleting
- knowledges = Knowledges.get_knowledges_by_file_id(id, db=db)
+ knowledges = await Knowledges.get_knowledges_by_file_id(id, db=db)
for knowledge in knowledges:
# Remove KB-file relationship
- Knowledges.remove_file_from_knowledge_by_id(knowledge.id, id, db=db)
+ await Knowledges.remove_file_from_knowledge_by_id(knowledge.id, id, db=db)
# Clean KB embeddings (same logic as /knowledge/{id}/file/remove)
try:
- VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': id})
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': id})
if file.hash:
- VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'hash': file.hash})
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'hash': file.hash})
except Exception as e:
log.debug(f'KB embedding cleanup for {knowledge.id}: {e}')
- result = Files.delete_file_by_id(id, db=db)
+ result = await Files.delete_file_by_id(id, db=db)
if result:
try:
- Storage.delete_file(file.path)
- VECTOR_DB_CLIENT.delete(collection_name=f'file-{id}')
+ await asyncio.to_thread(Storage.delete_file, file.path)
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=f'file-{id}')
except Exception as e:
log.exception(e)
log.error('Error deleting files')
diff --git a/backend/open_webui/routers/folders.py b/backend/open_webui/routers/folders.py
index 0bf5a87f1e..ebd0c0cb17 100644
--- a/backend/open_webui/routers/folders.py
+++ b/backend/open_webui/routers/folders.py
@@ -22,8 +22,8 @@ from open_webui.models.knowledge import Knowledges
from open_webui.config import UPLOAD_DIR
from open_webui.constants import ERROR_MESSAGES
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from fastapi import APIRouter, Depends, File, HTTPException, UploadFile, status, Request
@@ -48,7 +48,7 @@ router = APIRouter()
async def get_folders(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if request.app.state.config.ENABLE_FOLDERS is False:
raise HTTPException(
@@ -56,7 +56,7 @@ async def get_folders(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id,
'features.folders',
request.app.state.config.USER_PERMISSIONS,
@@ -67,29 +67,31 @@ async def get_folders(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- folders = Folders.get_folders_by_user_id(user.id, db=db)
+ folders = await Folders.get_folders_by_user_id(user.id, db=db)
# Verify folder data integrity
folder_list = []
for folder in folders:
- if folder.parent_id and not Folders.get_folder_by_id_and_user_id(folder.parent_id, user.id, db=db):
- folder = Folders.update_folder_parent_id_by_id_and_user_id(folder.id, user.id, None, db=db)
+ if folder.parent_id and not await Folders.get_folder_by_id_and_user_id(folder.parent_id, user.id, db=db):
+ folder = await Folders.update_folder_parent_id_by_id_and_user_id(folder.id, user.id, None, db=db)
if folder.data:
if 'files' in folder.data:
valid_files = []
for file in folder.data['files']:
if file.get('type') == 'file':
- if Files.check_access_by_user_id(file.get('id'), user.id, 'read', db=db):
+ if await Files.check_access_by_user_id(file.get('id'), user.id, 'read', db=db):
valid_files.append(file)
elif file.get('type') == 'collection':
- if Knowledges.check_access_by_user_id(file.get('id'), user.id, 'read', db=db):
+ if await Knowledges.check_access_by_user_id(file.get('id'), user.id, 'read', db=db):
valid_files.append(file)
else:
valid_files.append(file)
folder.data['files'] = valid_files
- Folders.update_folder_by_id_and_user_id(folder.id, user.id, FolderUpdateForm(data=folder.data), db=db)
+ await Folders.update_folder_by_id_and_user_id(
+ folder.id, user.id, FolderUpdateForm(data=folder.data), db=db
+ )
folder_list.append(FolderNameIdResponse(**folder.model_dump()))
@@ -102,12 +104,14 @@ async def get_folders(
@router.post('/')
-def create_folder(
+async def create_folder(
form_data: FolderForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- folder = Folders.get_folder_by_parent_id_and_user_id_and_name(form_data.parent_id, user.id, form_data.name, db=db)
+ folder = await Folders.get_folder_by_parent_id_and_user_id_and_name(
+ form_data.parent_id, user.id, form_data.name, db=db
+ )
if folder:
raise HTTPException(
@@ -116,7 +120,7 @@ def create_folder(
)
try:
- folder = Folders.insert_new_folder(user.id, form_data, form_data.parent_id, db=db)
+ folder = await Folders.insert_new_folder(user.id, form_data, form_data.parent_id, db=db)
return folder
except Exception as e:
log.exception(e)
@@ -133,8 +137,8 @@ def create_folder(
@router.get('/{id}', response_model=Optional[FolderModel])
-async def get_folder_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
+async def get_folder_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ folder = await Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
return folder
else:
@@ -154,13 +158,13 @@ async def update_folder_name_by_id(
id: str,
form_data: FolderUpdateForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
+ folder = await Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
if form_data.name is not None:
# Check if folder with same name exists
- existing_folder = Folders.get_folder_by_parent_id_and_user_id_and_name(
+ existing_folder = await Folders.get_folder_by_parent_id_and_user_id_and_name(
folder.parent_id, user.id, form_data.name, db=db
)
if existing_folder and existing_folder.id != id:
@@ -170,7 +174,7 @@ async def update_folder_name_by_id(
)
try:
- folder = Folders.update_folder_by_id_and_user_id(id, user.id, form_data, db=db)
+ folder = await Folders.update_folder_by_id_and_user_id(id, user.id, form_data, db=db)
return folder
except Exception as e:
log.exception(e)
@@ -200,11 +204,11 @@ async def update_folder_parent_id_by_id(
id: str,
form_data: FolderParentIdForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
+ folder = await Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
- existing_folder = Folders.get_folder_by_parent_id_and_user_id_and_name(
+ existing_folder = await Folders.get_folder_by_parent_id_and_user_id_and_name(
form_data.parent_id, user.id, folder.name, db=db
)
@@ -215,7 +219,7 @@ async def update_folder_parent_id_by_id(
)
try:
- folder = Folders.update_folder_parent_id_by_id_and_user_id(id, user.id, form_data.parent_id, db=db)
+ folder = await Folders.update_folder_parent_id_by_id_and_user_id(id, user.id, form_data.parent_id, db=db)
return folder
except Exception as e:
log.exception(e)
@@ -245,12 +249,14 @@ async def update_folder_is_expanded_by_id(
id: str,
form_data: FolderIsExpandedForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
+ folder = await Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
try:
- folder = Folders.update_folder_is_expanded_by_id_and_user_id(id, user.id, form_data.is_expanded, db=db)
+ folder = await Folders.update_folder_is_expanded_by_id_and_user_id(
+ id, user.id, form_data.is_expanded, db=db
+ )
return folder
except Exception as e:
log.exception(e)
@@ -277,10 +283,10 @@ async def delete_folder_by_id(
id: str,
delete_contents: Optional[bool] = True,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if Chats.count_chats_by_folder_id_and_user_id(id, user.id, db=db):
- chat_delete_permission = has_permission(
+ if await Chats.count_chats_by_folder_id_and_user_id(id, user.id, db=db):
+ chat_delete_permission = await has_permission(
user.id, 'chat.delete', request.app.state.config.USER_PERMISSIONS, db=db
)
if user.role != 'admin' and not chat_delete_permission:
@@ -290,18 +296,18 @@ async def delete_folder_by_id(
)
folders = []
- folders.append(Folders.get_folder_by_id_and_user_id(id, user.id, db=db))
+ folders.append(await Folders.get_folder_by_id_and_user_id(id, user.id, db=db))
while folders:
folder = folders.pop()
if folder:
try:
- folder_ids = Folders.delete_folder_by_id_and_user_id(folder.id, user.id, db=db)
+ folder_ids = await Folders.delete_folder_by_id_and_user_id(folder.id, user.id, db=db)
for folder_id in folder_ids:
if delete_contents:
- Chats.delete_chats_by_user_id_and_folder_id(user.id, folder_id, db=db)
+ await Chats.delete_chats_by_user_id_and_folder_id(user.id, folder_id, db=db)
else:
- Chats.move_chats_by_user_id_and_folder_id(user.id, folder_id, None, db=db)
+ await Chats.move_chats_by_user_id_and_folder_id(user.id, folder_id, None, db=db)
return True
except Exception as e:
@@ -313,7 +319,7 @@ async def delete_folder_by_id(
)
finally:
# Get all subfolders
- subfolders = Folders.get_folders_by_parent_id_and_user_id(folder.id, user.id, db=db)
+ subfolders = await Folders.get_folders_by_parent_id_and_user_id(folder.id, user.id, db=db)
folders.extend(subfolders)
else:
diff --git a/backend/open_webui/routers/functions.py b/backend/open_webui/routers/functions.py
index 01bcbc411c..1d0f0342d2 100644
--- a/backend/open_webui/routers/functions.py
+++ b/backend/open_webui/routers/functions.py
@@ -26,8 +26,8 @@ from open_webui.constants import ERROR_MESSAGES
from fastapi import APIRouter, Depends, HTTPException, Request, status
from open_webui.utils.auth import get_admin_user, get_verified_user
from pydantic import BaseModel, HttpUrl
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
@@ -42,13 +42,13 @@ router = APIRouter()
@router.get('/', response_model=list[FunctionResponse])
-async def get_functions(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- return Functions.get_functions(db=db)
+async def get_functions(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ return await Functions.get_functions(db=db)
@router.get('/list', response_model=list[FunctionUserResponse])
-async def get_function_list(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- return Functions.get_function_list(db=db)
+async def get_function_list(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ return await Functions.get_function_list(db=db)
############################
@@ -60,9 +60,9 @@ async def get_function_list(user=Depends(get_admin_user), db: Session = Depends(
async def get_functions(
include_valves: bool = False,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- return Functions.get_functions(include_valves=include_valves, db=db)
+ return await Functions.get_functions(include_valves=include_valves, db=db)
############################
@@ -128,7 +128,7 @@ async def load_function_from_url(request: Request, form_data: LoadUrlForm, user=
'content': data,
}
except Exception as e:
- raise HTTPException(status_code=500, detail=f'Error importing function: {e}')
+ raise HTTPException(status_code=500, detail=ERROR_MESSAGES.DEFAULT(e))
############################
@@ -145,12 +145,12 @@ async def sync_functions(
request: Request,
form_data: SyncFunctionsForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
for function in form_data.functions:
function.content = replace_imports(function.content)
- function_module, function_type, frontmatter = load_function_module_by_id(
+ function_module, function_type, frontmatter = await load_function_module_by_id(
function.id,
content=function.content,
)
@@ -163,7 +163,7 @@ async def sync_functions(
log.exception(f'Error validating valves for function {function.id}: {e}')
raise e
- return Functions.sync_functions(user.id, form_data.functions, db=db)
+ return await Functions.sync_functions(user.id, form_data.functions, db=db)
except Exception as e:
log.exception(f'Failed to load a function: {e}')
raise HTTPException(
@@ -182,7 +182,7 @@ async def create_new_function(
request: Request,
form_data: FunctionForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not form_data.id.isidentifier():
raise HTTPException(
@@ -192,11 +192,11 @@ async def create_new_function(
form_data.id = form_data.id.lower()
- function = Functions.get_function_by_id(form_data.id, db=db)
+ function = await Functions.get_function_by_id(form_data.id, db=db)
if function is None:
try:
form_data.content = replace_imports(form_data.content)
- function_module, function_type, frontmatter = load_function_module_by_id(
+ function_module, function_type, frontmatter = await load_function_module_by_id(
form_data.id,
content=form_data.content,
)
@@ -205,13 +205,13 @@ async def create_new_function(
FUNCTIONS = request.app.state.FUNCTIONS
FUNCTIONS[form_data.id] = function_module
- function = Functions.insert_new_function(user.id, function_type, form_data, db=db)
+ function = await Functions.insert_new_function(user.id, function_type, form_data, db=db)
function_cache_dir = CACHE_DIR / 'functions' / form_data.id
function_cache_dir.mkdir(parents=True, exist_ok=True)
if function_type == 'filter' and getattr(function_module, 'toggle', None):
- Functions.update_function_metadata_by_id(form_data.id, {'toggle': True}, db=db)
+ await Functions.update_function_metadata_by_id(form_data.id, {'toggle': True}, db=db)
if function:
return function
@@ -239,8 +239,8 @@ async def create_new_function(
@router.get('/id/{id}', response_model=Optional[FunctionModel])
-async def get_function_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- function = Functions.get_function_by_id(id, db=db)
+async def get_function_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ function = await Functions.get_function_by_id(id, db=db)
if function:
return function
@@ -257,10 +257,10 @@ async def get_function_by_id(id: str, user=Depends(get_admin_user), db: Session
@router.post('/id/{id}/toggle', response_model=Optional[FunctionModel])
-async def toggle_function_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- function = Functions.get_function_by_id(id, db=db)
+async def toggle_function_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ function = await Functions.get_function_by_id(id, db=db)
if function:
- function = Functions.update_function_by_id(id, {'is_active': not function.is_active}, db=db)
+ function = await Functions.update_function_by_id(id, {'is_active': not function.is_active}, db=db)
if function:
return function
@@ -282,10 +282,10 @@ async def toggle_function_by_id(id: str, user=Depends(get_admin_user), db: Sessi
@router.post('/id/{id}/toggle/global', response_model=Optional[FunctionModel])
-async def toggle_global_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- function = Functions.get_function_by_id(id, db=db)
+async def toggle_global_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ function = await Functions.get_function_by_id(id, db=db)
if function:
- function = Functions.update_function_by_id(id, {'is_global': not function.is_global}, db=db)
+ function = await Functions.update_function_by_id(id, {'is_global': not function.is_global}, db=db)
if function:
return function
@@ -312,11 +312,11 @@ async def update_function_by_id(
id: str,
form_data: FunctionForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
form_data.content = replace_imports(form_data.content)
- function_module, function_type, frontmatter = load_function_module_by_id(id, content=form_data.content)
+ function_module, function_type, frontmatter = await load_function_module_by_id(id, content=form_data.content)
form_data.meta.manifest = frontmatter
FUNCTIONS = request.app.state.FUNCTIONS
@@ -325,10 +325,10 @@ async def update_function_by_id(
updated = {**form_data.model_dump(exclude={'id'}), 'type': function_type}
log.debug(updated)
- function = Functions.update_function_by_id(id, updated, db=db)
+ function = await Functions.update_function_by_id(id, updated, db=db)
if function_type == 'filter' and getattr(function_module, 'toggle', None):
- Functions.update_function_metadata_by_id(id, {'toggle': True}, db=db)
+ await Functions.update_function_metadata_by_id(id, {'toggle': True}, db=db)
if function:
return function
@@ -355,9 +355,9 @@ async def delete_function_by_id(
request: Request,
id: str,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- result = Functions.delete_function_by_id(id, db=db)
+ result = await Functions.delete_function_by_id(id, db=db)
if result:
FUNCTIONS = request.app.state.FUNCTIONS
@@ -373,11 +373,13 @@ async def delete_function_by_id(
@router.get('/id/{id}/valves', response_model=Optional[dict])
-async def get_function_valves_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- function = Functions.get_function_by_id(id, db=db)
+async def get_function_valves_by_id(
+ id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)
+):
+ function = await Functions.get_function_by_id(id, db=db)
if function:
try:
- valves = Functions.get_function_valves_by_id(id, db=db)
+ valves = await Functions.get_function_valves_by_id(id, db=db)
return valves
except Exception as e:
raise HTTPException(
@@ -401,11 +403,11 @@ async def get_function_valves_spec_by_id(
request: Request,
id: str,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- function = Functions.get_function_by_id(id, db=db)
+ function = await Functions.get_function_by_id(id, db=db)
if function:
- function_module, function_type, frontmatter = get_function_module_from_cache(request, id)
+ function_module, function_type, frontmatter = await get_function_module_from_cache(request, id)
if hasattr(function_module, 'Valves'):
Valves = function_module.Valves
@@ -432,11 +434,11 @@ async def update_function_valves_by_id(
id: str,
form_data: dict,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- function = Functions.get_function_by_id(id, db=db)
+ function = await Functions.get_function_by_id(id, db=db)
if function:
- function_module, function_type, frontmatter = get_function_module_from_cache(request, id)
+ function_module, function_type, frontmatter = await get_function_module_from_cache(request, id)
if hasattr(function_module, 'Valves'):
Valves = function_module.Valves
@@ -446,7 +448,7 @@ async def update_function_valves_by_id(
valves = Valves(**form_data)
valves_dict = valves.model_dump(exclude_unset=True)
- Functions.update_function_valves_by_id(id, valves_dict, db=db)
+ await Functions.update_function_valves_by_id(id, valves_dict, db=db)
return valves_dict
except Exception as e:
log.exception(f'Error updating function values by id {id}: {e}')
@@ -473,11 +475,13 @@ async def update_function_valves_by_id(
@router.get('/id/{id}/valves/user', response_model=Optional[dict])
-async def get_function_user_valves_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- function = Functions.get_function_by_id(id, db=db)
+async def get_function_user_valves_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ function = await Functions.get_function_by_id(id, db=db)
if function:
try:
- user_valves = Functions.get_user_valves_by_id_and_user_id(id, user.id, db=db)
+ user_valves = await Functions.get_user_valves_by_id_and_user_id(id, user.id, db=db)
return user_valves
except Exception as e:
raise HTTPException(
@@ -496,11 +500,11 @@ async def get_function_user_valves_spec_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- function = Functions.get_function_by_id(id, db=db)
+ function = await Functions.get_function_by_id(id, db=db)
if function:
- function_module, function_type, frontmatter = get_function_module_from_cache(request, id)
+ function_module, function_type, frontmatter = await get_function_module_from_cache(request, id)
if hasattr(function_module, 'UserValves'):
UserValves = function_module.UserValves
@@ -522,12 +526,12 @@ async def update_function_user_valves_by_id(
id: str,
form_data: dict,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- function = Functions.get_function_by_id(id, db=db)
+ function = await Functions.get_function_by_id(id, db=db)
if function:
- function_module, function_type, frontmatter = get_function_module_from_cache(request, id)
+ function_module, function_type, frontmatter = await get_function_module_from_cache(request, id)
if hasattr(function_module, 'UserValves'):
UserValves = function_module.UserValves
@@ -536,7 +540,7 @@ async def update_function_user_valves_by_id(
form_data = {k: v for k, v in form_data.items() if v is not None}
user_valves = UserValves(**form_data)
user_valves_dict = user_valves.model_dump(exclude_unset=True)
- Functions.update_user_valves_by_id_and_user_id(id, user.id, user_valves_dict, db=db)
+ await Functions.update_user_valves_by_id_and_user_id(id, user.id, user_valves_dict, db=db)
return user_valves_dict
except Exception as e:
log.exception(f'Error updating function user valves by id {id}: {e}')
diff --git a/backend/open_webui/routers/groups.py b/backend/open_webui/routers/groups.py
index 4e9688c3d8..c45690fc3a 100755
--- a/backend/open_webui/routers/groups.py
+++ b/backend/open_webui/routers/groups.py
@@ -17,8 +17,8 @@ from open_webui.config import CACHE_DIR
from open_webui.constants import ERROR_MESSAGES
from fastapi import APIRouter, Depends, HTTPException, Request, status
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from open_webui.utils.auth import get_admin_user, get_verified_user
@@ -35,7 +35,7 @@ router = APIRouter()
async def get_groups(
share: Optional[bool] = None,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
filter = {}
@@ -45,7 +45,7 @@ async def get_groups(
if share is not None:
filter['share'] = share
- groups = Groups.get_groups(filter=filter, db=db)
+ groups = await Groups.get_groups(filter=filter, db=db)
return groups
@@ -59,14 +59,14 @@ async def get_groups(
async def create_new_group(
form_data: GroupForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
- group = Groups.insert_new_group(user.id, form_data, db=db)
+ group = await Groups.insert_new_group(user.id, form_data, db=db)
if group:
return GroupResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -87,12 +87,12 @@ async def create_new_group(
@router.get('/id/{id}', response_model=Optional[GroupResponse])
-async def get_group_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- group = Groups.get_group_by_id(id, db=db)
+async def get_group_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ group = await Groups.get_group_by_id(id, db=db)
if group:
return GroupResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -102,12 +102,12 @@ async def get_group_by_id(id: str, user=Depends(get_admin_user), db: Session = D
@router.get('/id/{id}/info', response_model=Optional[GroupInfoResponse])
-async def get_group_info_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- group = Groups.get_group_by_id(id, db=db)
+async def get_group_info_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ group = await Groups.get_group_by_id(id, db=db)
if group:
return GroupInfoResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -127,13 +127,13 @@ class GroupExportResponse(GroupResponse):
@router.get('/id/{id}/export', response_model=Optional[GroupExportResponse])
-async def export_group_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- group = Groups.get_group_by_id(id, db=db)
+async def export_group_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ group = await Groups.get_group_by_id(id, db=db)
if group:
return GroupExportResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
- user_ids=Groups.get_group_user_ids_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
+ user_ids=await Groups.get_group_user_ids_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -148,9 +148,9 @@ async def export_group_by_id(id: str, user=Depends(get_admin_user), db: Session
@router.post('/id/{id}/users', response_model=list[UserInfoResponse])
-async def get_users_in_group(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
+async def get_users_in_group(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
try:
- users = Users.get_users_by_group_id(id, db=db)
+ users = await Users.get_users_by_group_id(id, db=db)
return users
except Exception as e:
log.exception(f'Error adding users to group {id}: {e}')
@@ -170,14 +170,14 @@ async def update_group_by_id(
id: str,
form_data: GroupUpdateForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
- group = Groups.update_group_by_id(id, form_data, db=db)
+ group = await Groups.update_group_by_id(id, form_data, db=db)
if group:
return GroupResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -202,17 +202,17 @@ async def add_user_to_group(
id: str,
form_data: UserIdsForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
if form_data.user_ids:
- form_data.user_ids = Users.get_valid_user_ids(form_data.user_ids, db=db)
+ form_data.user_ids = await Users.get_valid_user_ids(form_data.user_ids, db=db)
- group = Groups.add_users_to_group(id, form_data.user_ids, db=db)
+ group = await Groups.add_users_to_group(id, form_data.user_ids, db=db)
if group:
return GroupResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -232,14 +232,14 @@ async def remove_users_from_group(
id: str,
form_data: UserIdsForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
- group = Groups.remove_users_from_group(id, form_data.user_ids, db=db)
+ group = await Groups.remove_users_from_group(id, form_data.user_ids, db=db)
if group:
return GroupResponse(
**group.model_dump(),
- member_count=Groups.get_group_member_count_by_id(group.id, db=db),
+ member_count=await Groups.get_group_member_count_by_id(group.id, db=db),
)
else:
raise HTTPException(
@@ -260,9 +260,9 @@ async def remove_users_from_group(
@router.delete('/id/{id}/delete', response_model=bool)
-async def delete_group_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
+async def delete_group_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
try:
- result = Groups.delete_group_by_id(id, db=db)
+ result = await Groups.delete_group_by_id(id, db=db)
if result:
return result
else:
diff --git a/backend/open_webui/routers/images.py b/backend/open_webui/routers/images.py
index dca9a58a7a..37dda2eb1b 100644
--- a/backend/open_webui/routers/images.py
+++ b/backend/open_webui/routers/images.py
@@ -10,6 +10,7 @@ from pathlib import Path
from typing import Optional
from urllib.parse import quote
+import aiohttp
import requests
from fastapi import APIRouter, Depends, HTTPException, Request, UploadFile
from fastapi.responses import FileResponse
@@ -21,15 +22,16 @@ from open_webui.config import (
)
from open_webui.constants import ERROR_MESSAGES
from open_webui.retrieval.web.utils import validate_url
-from open_webui.env import ENABLE_FORWARD_USER_INFO_HEADERS
+from open_webui.env import AIOHTTP_CLIENT_SESSION_SSL, ENABLE_FORWARD_USER_INFO_HEADERS
+from open_webui.utils.session_pool import get_session
from open_webui.models.chats import Chats
from open_webui.routers.files import upload_file_handler, get_file_content_by_id
from open_webui.utils.auth import get_admin_user, get_verified_user
from open_webui.utils.access_control import has_permission
from open_webui.utils.headers import include_user_info_headers
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from open_webui.utils.images.comfyui import (
ComfyUICreateImageForm,
ComfyUIEditImageForm,
@@ -313,12 +315,14 @@ def get_automatic1111_api_auth(request: Request):
async def verify_url(request: Request, user=Depends(get_admin_user)):
if request.app.state.config.IMAGE_GENERATION_ENGINE == 'automatic1111':
try:
- r = requests.get(
+ session = await get_session()
+ async with session.get(
url=f'{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/options',
headers={'authorization': get_automatic1111_api_auth(request)},
- )
- r.raise_for_status()
- return True
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ return True
except Exception:
request.app.state.config.ENABLE_IMAGE_GENERATION = False
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.INVALID_URL)
@@ -327,12 +331,14 @@ async def verify_url(request: Request, user=Depends(get_admin_user)):
if request.app.state.config.COMFYUI_API_KEY:
headers = {'Authorization': f'Bearer {request.app.state.config.COMFYUI_API_KEY}'}
try:
- r = requests.get(
+ session = await get_session()
+ async with session.get(
url=f'{request.app.state.config.COMFYUI_BASE_URL}/object_info',
headers=headers,
- )
- r.raise_for_status()
- return True
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ return True
except Exception:
request.app.state.config.ENABLE_IMAGE_GENERATION = False
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.INVALID_URL)
@@ -341,7 +347,7 @@ async def verify_url(request: Request, user=Depends(get_admin_user)):
@router.get('/models')
-def get_models(request: Request, user=Depends(get_verified_user)):
+async def get_models(request: Request, user=Depends(get_verified_user)):
try:
if request.app.state.config.IMAGE_GENERATION_ENGINE == 'openai':
return [
@@ -357,11 +363,13 @@ def get_models(request: Request, user=Depends(get_verified_user)):
elif request.app.state.config.IMAGE_GENERATION_ENGINE == 'comfyui':
# TODO - get models from comfyui
headers = {'Authorization': f'Bearer {request.app.state.config.COMFYUI_API_KEY}'}
- r = requests.get(
+ session = await get_session()
+ async with session.get(
url=f'{request.app.state.config.COMFYUI_BASE_URL}/object_info',
headers=headers,
- )
- info = r.json()
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ info = await r.json()
workflow = json.loads(request.app.state.config.COMFYUI_WORKFLOW)
model_node_id = None
@@ -399,11 +407,13 @@ def get_models(request: Request, user=Depends(get_verified_user)):
request.app.state.config.IMAGE_GENERATION_ENGINE == 'automatic1111'
or request.app.state.config.IMAGE_GENERATION_ENGINE == ''
):
- r = requests.get(
+ session = await get_session()
+ async with session.get(
url=f'{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/sd-models',
headers={'authorization': get_automatic1111_api_auth(request)},
- )
- models = r.json()
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ models = await r.json()
return list(
map(
lambda model: {'id': model['title'], 'name': model['model_name']},
@@ -456,7 +466,7 @@ def get_image_data(data: str, headers=None):
return None, None
-def upload_image(request, image_data, content_type, metadata, user, db=None):
+async def upload_image(request, image_data, content_type, metadata, user, db=None):
image_format = mimetypes.guess_extension(content_type)
file = UploadFile(
file=io.BytesIO(image_data),
@@ -465,7 +475,7 @@ def upload_image(request, image_data, content_type, metadata, user, db=None):
'content-type': content_type,
},
)
- file_item = upload_file_handler(
+ file_item = await upload_file_handler(
request,
file=file,
metadata=metadata,
@@ -479,7 +489,7 @@ def upload_image(request, image_data, content_type, metadata, user, db=None):
message_id = metadata.get('message_id')
if chat_id and message_id:
- Chats.insert_chat_files(
+ await Chats.insert_chat_files(
chat_id=chat_id,
message_id=message_id,
file_ids=[file_item.id],
@@ -499,7 +509,7 @@ async def generate_images(request: Request, form_data: CreateImageForm, user=Dep
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.image_generation', request.app.state.config.USER_PERMISSIONS
):
raise HTTPException(
@@ -533,7 +543,6 @@ async def image_generations(
model = get_image_model(request)
- r = None
try:
if request.app.state.config.IMAGE_GENERATION_ENGINE == 'openai':
headers = {
@@ -552,7 +561,11 @@ async def image_generations(
'model': model,
'prompt': form_data.prompt,
'n': form_data.n,
- 'size': (form_data.size if form_data.size else request.app.state.config.IMAGE_SIZE),
+ **(
+ {'size': form_data.size or request.app.state.config.IMAGE_SIZE}
+ if (form_data.size or request.app.state.config.IMAGE_SIZE)
+ else {}
+ ),
**(
{}
if re.match(
@@ -568,16 +581,15 @@ async def image_generations(
),
}
- # Use asyncio.to_thread for the requests.post call
- r = await asyncio.to_thread(
- requests.post,
+ session = await get_session()
+ async with session.post(
url=url,
json=data,
headers=headers,
- )
-
- r.raise_for_status()
- res = r.json()
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ res = await r.json()
images = []
@@ -590,7 +602,7 @@ async def image_generations(
else:
image_data, content_type = get_image_data(image['b64_json'])
- _, url = upload_image(request, image_data, content_type, {**data, **metadata}, user)
+ _, url = await upload_image(request, image_data, content_type, {**data, **metadata}, user)
images.append({'url': url})
return images
@@ -619,30 +631,29 @@ async def image_generations(
model = f'{model}:generateContent'
data = {'contents': [{'parts': [{'text': form_data.prompt}]}]}
- # Use asyncio.to_thread for the requests.post call
- r = await asyncio.to_thread(
- requests.post,
+ session = await get_session()
+ async with session.post(
url=f'{request.app.state.config.IMAGES_GEMINI_API_BASE_URL}/models/{model}',
json=data,
headers=headers,
- )
-
- r.raise_for_status()
- res = r.json()
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ res = await r.json()
images = []
if model.endswith(':predict'):
for image in res['predictions']:
image_data, content_type = get_image_data(image['bytesBase64Encoded'])
- _, url = upload_image(request, image_data, content_type, {**data, **metadata}, user)
+ _, url = await upload_image(request, image_data, content_type, {**data, **metadata}, user)
images.append({'url': url})
elif model.endswith(':generateContent'):
for image in res['candidates']:
for part in image['content']['parts']:
if part.get('inlineData', {}).get('data'):
image_data, content_type = get_image_data(part['inlineData']['data'])
- _, url = upload_image(
+ _, url = await upload_image(
request,
image_data,
content_type,
@@ -681,7 +692,7 @@ async def image_generations(
res = await comfyui_create_image(
model,
form_data,
- user.id,
+ str(uuid.uuid4()),
request.app.state.config.COMFYUI_BASE_URL,
request.app.state.config.COMFYUI_API_KEY,
)
@@ -695,7 +706,7 @@ async def image_generations(
headers = {'Authorization': f'Bearer {request.app.state.config.COMFYUI_API_KEY}'}
image_data, content_type = get_image_data(image['url'], headers)
- _, url = upload_image(
+ _, url = await upload_image(
request,
image_data,
content_type,
@@ -727,22 +738,21 @@ async def image_generations(
if request.app.state.config.AUTOMATIC1111_PARAMS:
data = {**data, **request.app.state.config.AUTOMATIC1111_PARAMS}
- # Use asyncio.to_thread for the requests.post call
- r = await asyncio.to_thread(
- requests.post,
+ session = await get_session()
+ async with session.post(
url=f'{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/txt2img',
json=data,
headers={'authorization': get_automatic1111_api_auth(request)},
- )
-
- res = r.json()
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ res = await r.json()
log.debug(f'res: {res}')
images = []
for image in res['images']:
image_data, content_type = get_image_data(image)
- _, url = upload_image(
+ _, url = await upload_image(
request,
image_data,
content_type,
@@ -753,10 +763,8 @@ async def image_generations(
return images
except Exception as e:
error = e
- if r != None:
- data = r.json()
- if 'error' in data:
- error = data['error']['message']
+ if isinstance(e, aiohttp.ClientResponseError):
+ error = e.message
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(error))
@@ -798,11 +806,12 @@ async def image_edits(
if data.startswith('http://') or data.startswith('https://'):
# Validate URL to prevent SSRF attacks against local/private networks
validate_url(data)
- r = await asyncio.to_thread(requests.get, data)
- r.raise_for_status()
+ session = await get_session()
+ async with session.get(data, ssl=AIOHTTP_CLIENT_SESSION_SSL) as r:
+ r.raise_for_status()
- image_data = base64.b64encode(r.content).decode('utf-8')
- return f'data:{r.headers["content-type"]};base64,{image_data}'
+ image_data = base64.b64encode(await r.read()).decode('utf-8')
+ return f'data:{r.headers["content-type"]};base64,{image_data}'
else:
file_id = None
@@ -846,7 +855,6 @@ async def image_edits(
),
)
- r = None
try:
if request.app.state.config.IMAGE_EDIT_ENGINE == 'openai':
headers = {
@@ -883,17 +891,30 @@ async def image_edits(
if request.app.state.config.IMAGES_EDIT_OPENAI_API_VERSION:
url_search_params += f'?api-version={request.app.state.config.IMAGES_EDIT_OPENAI_API_VERSION}'
- # Use asyncio.to_thread for the requests.post call
- r = await asyncio.to_thread(
- requests.post,
+ # Build multipart form data for aiohttp
+ form = aiohttp.FormData()
+ for key, value in data.items():
+ if isinstance(value, dict):
+ form.add_field(key, json.dumps(value))
+ else:
+ form.add_field(key, str(value))
+ for param_name, (filename, file_obj, content_type_val) in files:
+ form.add_field(
+ param_name,
+ file_obj,
+ filename=filename,
+ content_type=content_type_val,
+ )
+
+ session = await get_session()
+ async with session.post(
url=f'{request.app.state.config.IMAGES_EDIT_OPENAI_API_BASE_URL}/images/edits{url_search_params}',
headers=headers,
- files=files,
- data=data,
- )
-
- r.raise_for_status()
- res = r.json()
+ data=form,
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ res = await r.json()
images = []
for image in res['data']:
@@ -905,7 +926,7 @@ async def image_edits(
else:
image_data, content_type = get_image_data(image['b64_json'])
- _, url = upload_image(request, image_data, content_type, {**data, **metadata}, user)
+ _, url = await upload_image(request, image_data, content_type, {**data, **metadata}, user)
images.append({'url': url})
return images
@@ -940,23 +961,22 @@ async def image_edits(
]
)
- # Use asyncio.to_thread for the requests.post call
- r = await asyncio.to_thread(
- requests.post,
+ session = await get_session()
+ async with session.post(
url=f'{request.app.state.config.IMAGES_EDIT_GEMINI_API_BASE_URL}/models/{model}',
json=data,
headers=headers,
- )
-
- r.raise_for_status()
- res = r.json()
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ ) as r:
+ r.raise_for_status()
+ res = await r.json()
images = []
for image in res['candidates']:
for part in image['content']['parts']:
if part.get('inlineData', {}).get('data'):
image_data, content_type = get_image_data(part['inlineData']['data'])
- _, url = upload_image(
+ _, url = await upload_image(
request,
image_data,
content_type,
@@ -1011,7 +1031,7 @@ async def image_edits(
res = await comfyui_edit_image(
model,
form_data,
- user.id,
+ str(uuid.uuid4()),
request.app.state.config.IMAGES_EDIT_COMFYUI_BASE_URL,
request.app.state.config.IMAGES_EDIT_COMFYUI_API_KEY,
)
@@ -1036,7 +1056,7 @@ async def image_edits(
headers = {'Authorization': f'Bearer {request.app.state.config.IMAGES_EDIT_COMFYUI_API_KEY}'}
image_data, content_type = get_image_data(image_url, headers)
- _, url = upload_image(
+ _, url = await upload_image(
request,
image_data,
content_type,
@@ -1048,13 +1068,7 @@ async def image_edits(
return images
except Exception as e:
error = e
- if r != None:
- data = r.text
- try:
- data = json.loads(data)
- if 'error' in data:
- error = data['error']['message']
- except Exception:
- error = data
+ if isinstance(e, aiohttp.ClientResponseError):
+ error = e.message
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(error))
diff --git a/backend/open_webui/routers/knowledge.py b/backend/open_webui/routers/knowledge.py
index ead782cdbf..77b72cacf0 100644
--- a/backend/open_webui/routers/knowledge.py
+++ b/backend/open_webui/routers/knowledge.py
@@ -2,14 +2,14 @@ from typing import List, Optional
from pydantic import BaseModel
from fastapi import APIRouter, Depends, HTTPException, status, Request, Query
from fastapi.responses import StreamingResponse
-from fastapi.concurrency import run_in_threadpool
+
import logging
import io
import zipfile
from urllib.parse import quote
-from sqlalchemy.orm import Session
-from open_webui.internal.db import get_session
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import get_async_session
from open_webui.models.groups import Groups
from open_webui.models.knowledge import (
KnowledgeFileListResponse,
@@ -19,7 +19,7 @@ from open_webui.models.knowledge import (
KnowledgeUserResponse,
)
from open_webui.models.files import Files, FileModel, FileMetadataResponse
-from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
from open_webui.routers.retrieval import (
process_file,
ProcessFileForm,
@@ -66,7 +66,7 @@ async def embed_knowledge_base_metadata(
try:
content = f'{name}\n\n{description}' if description else name
embedding = await request.app.state.EMBEDDING_FUNCTION(content)
- VECTOR_DB_CLIENT.upsert(
+ await ASYNC_VECTOR_DB_CLIENT.upsert(
collection_name=KNOWLEDGE_BASES_COLLECTION,
items=[
{
@@ -85,10 +85,10 @@ async def embed_knowledge_base_metadata(
return False
-def remove_knowledge_base_metadata_embedding(knowledge_base_id: str) -> bool:
+async def remove_knowledge_base_metadata_embedding(knowledge_base_id: str) -> bool:
"""Remove knowledge base embedding."""
try:
- VECTOR_DB_CLIENT.delete(
+ await ASYNC_VECTOR_DB_CLIENT.delete(
collection_name=KNOWLEDGE_BASES_COLLECTION,
ids=[knowledge_base_id],
)
@@ -111,14 +111,14 @@ class KnowledgeAccessListResponse(BaseModel):
async def get_knowledge_bases(
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
page = max(page, 1)
limit = PAGE_ITEM_COUNT
skip = (page - 1) * limit
filter = {}
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
user_group_ids = {group.id for group in groups}
if not user.role == 'admin' or not BYPASS_ADMIN_ACCESS_CONTROL:
@@ -127,11 +127,11 @@ async def get_knowledge_bases(
filter['user_id'] = user.id
- result = Knowledges.search_knowledge_bases(user.id, filter=filter, skip=skip, limit=limit, db=db)
+ result = await Knowledges.search_knowledge_bases(user.id, filter=filter, skip=skip, limit=limit, db=db)
# Batch-fetch writable knowledge IDs in a single query instead of N has_access calls
knowledge_base_ids = [knowledge_base.id for knowledge_base in result.items]
- writable_knowledge_base_ids = AccessGrants.get_accessible_resource_ids(
+ writable_knowledge_base_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='knowledge',
resource_ids=knowledge_base_ids,
@@ -162,7 +162,7 @@ async def search_knowledge_bases(
view_option: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
page = max(page, 1)
limit = PAGE_ITEM_COUNT
@@ -174,7 +174,7 @@ async def search_knowledge_bases(
if view_option:
filter['view_option'] = view_option
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
user_group_ids = {group.id for group in groups}
if not user.role == 'admin' or not BYPASS_ADMIN_ACCESS_CONTROL:
@@ -183,11 +183,11 @@ async def search_knowledge_bases(
filter['user_id'] = user.id
- result = Knowledges.search_knowledge_bases(user.id, filter=filter, skip=skip, limit=limit, db=db)
+ result = await Knowledges.search_knowledge_bases(user.id, filter=filter, skip=skip, limit=limit, db=db)
# Batch-fetch writable knowledge IDs in a single query instead of N has_access calls
knowledge_base_ids = [knowledge_base.id for knowledge_base in result.items]
- writable_knowledge_base_ids = AccessGrants.get_accessible_resource_ids(
+ writable_knowledge_base_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='knowledge',
resource_ids=knowledge_base_ids,
@@ -217,7 +217,7 @@ async def search_knowledge_files(
query: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
page = max(page, 1)
limit = PAGE_ITEM_COUNT
@@ -227,13 +227,13 @@ async def search_knowledge_files(
if query:
filter['query'] = query
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
if groups:
filter['group_ids'] = [group.id for group in groups]
filter['user_id'] = user.id
- return Knowledges.search_knowledge_files(filter=filter, skip=skip, limit=limit, db=db)
+ return await Knowledges.search_knowledge_files(filter=filter, skip=skip, limit=limit, db=db)
############################
@@ -247,11 +247,11 @@ async def create_new_knowledge(
form_data: KnowledgeForm,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (has_permission, filter_allowed_access_grants, insert_new_knowledge) manage their own sessions.
# This prevents holding a connection during embed_knowledge_base_metadata()
# which makes external embedding API calls (1-5+ seconds).
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'workspace.knowledge', request.app.state.config.USER_PERMISSIONS
):
raise HTTPException(
@@ -259,7 +259,7 @@ async def create_new_knowledge(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -267,7 +267,7 @@ async def create_new_knowledge(
'sharing.public_knowledge',
)
- knowledge = Knowledges.insert_new_knowledge(user.id, form_data)
+ knowledge = await Knowledges.insert_new_knowledge(user.id, form_data)
if knowledge:
# Embed knowledge base for semantic search
@@ -294,7 +294,7 @@ async def create_new_knowledge(
async def reindex_knowledge_files(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if user.role != 'admin':
raise HTTPException(
@@ -302,16 +302,16 @@ async def reindex_knowledge_files(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- knowledge_bases = Knowledges.get_knowledge_bases(db=db)
+ knowledge_bases = await Knowledges.get_knowledge_bases(db=db)
log.info(f'Starting reindexing for {len(knowledge_bases)} knowledge bases')
for knowledge_base in knowledge_bases:
try:
- files = Knowledges.get_files_by_id(knowledge_base.id, db=db)
+ files = await Knowledges.get_files_by_id(knowledge_base.id, db=db)
try:
- if VECTOR_DB_CLIENT.has_collection(collection_name=knowledge_base.id):
- VECTOR_DB_CLIENT.delete_collection(collection_name=knowledge_base.id)
+ if await ASYNC_VECTOR_DB_CLIENT.has_collection(collection_name=knowledge_base.id):
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=knowledge_base.id)
except Exception as e:
log.error(f'Error deleting collection {knowledge_base.id}: {str(e)}')
continue # Skip, don't raise
@@ -319,8 +319,7 @@ async def reindex_knowledge_files(
failed_files = []
for file in files:
try:
- await run_in_threadpool(
- process_file,
+ await process_file(
request,
ProcessFileForm(file_id=file.id, collection_name=knowledge_base.id),
user=user,
@@ -357,12 +356,12 @@ async def reindex_knowledge_base_metadata_embeddings(
):
"""Batch embed all existing knowledge bases. Admin only.
- NOTE: We intentionally do NOT use Depends(get_session) here.
+ NOTE: We intentionally do NOT use Depends(get_async_session) here.
This endpoint loops through ALL knowledge bases and calls embed_knowledge_base_metadata()
for each one, making N external embedding API calls. Holding a session during
this entire operation would exhaust the connection pool.
"""
- knowledge_bases = Knowledges.get_knowledge_bases()
+ knowledge_bases = await Knowledges.get_knowledge_bases()
log.info(f'Reindexing embeddings for {len(knowledge_bases)} knowledge bases')
success_count = 0
@@ -385,14 +384,14 @@ class KnowledgeFilesResponse(KnowledgeResponse):
@router.get('/{id}', response_model=Optional[KnowledgeFilesResponse])
-async def get_knowledge_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+async def get_knowledge_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if knowledge:
if (
user.role == 'admin'
or knowledge.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -405,7 +404,7 @@ async def get_knowledge_by_id(id: str, user=Depends(get_verified_user), db: Sess
write_access=(
user.id == knowledge.user_id
or (user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -438,11 +437,11 @@ async def update_knowledge_by_id(
form_data: KnowledgeForm,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations manage their own short-lived sessions internally.
# This prevents holding a connection during embed_knowledge_base_metadata()
# which makes external embedding API calls (1-5+ seconds).
- knowledge = Knowledges.get_knowledge_by_id(id=id)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -451,7 +450,7 @@ async def update_knowledge_by_id(
# Is the user the original creator, in a group with write access, or an admin
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -464,7 +463,7 @@ async def update_knowledge_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -472,7 +471,7 @@ async def update_knowledge_by_id(
'sharing.public_knowledge',
)
- knowledge = Knowledges.update_knowledge_by_id(id=id, form_data=form_data)
+ knowledge = await Knowledges.update_knowledge_by_id(id=id, form_data=form_data)
if knowledge:
# Re-embed knowledge base for semantic search
await embed_knowledge_base_metadata(
@@ -483,7 +482,7 @@ async def update_knowledge_by_id(
)
return KnowledgeFilesResponse(
**knowledge.model_dump(),
- files=Knowledges.get_file_metadatas_by_id(knowledge.id),
+ files=await Knowledges.get_file_metadatas_by_id(knowledge.id),
)
else:
raise HTTPException(
@@ -507,9 +506,9 @@ async def update_knowledge_access_by_id(
id: str,
form_data: KnowledgeAccessGrantsForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -518,7 +517,7 @@ async def update_knowledge_access_by_id(
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -532,7 +531,7 @@ async def update_knowledge_access_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -540,11 +539,11 @@ async def update_knowledge_access_by_id(
'sharing.public_knowledge',
)
- AccessGrants.set_access_grants('knowledge', id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('knowledge', id, form_data.access_grants, db=db)
return KnowledgeFilesResponse(
- **Knowledges.get_knowledge_by_id(id=id, db=db).model_dump(),
- files=Knowledges.get_file_metadatas_by_id(id, db=db),
+ **(await Knowledges.get_knowledge_by_id(id=id, db=db)).model_dump(),
+ files=await Knowledges.get_file_metadatas_by_id(id, db=db),
)
@@ -562,9 +561,9 @@ async def get_knowledge_files_by_id(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -574,7 +573,7 @@ async def get_knowledge_files_by_id(
if not (
user.role == 'admin'
or knowledge.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -602,7 +601,7 @@ async def get_knowledge_files_by_id(
if direction:
filter['direction'] = direction
- return Knowledges.search_files_by_id(id, user.id, filter=filter, skip=skip, limit=limit, db=db)
+ return await Knowledges.search_files_by_id(id, user.id, filter=filter, skip=skip, limit=limit, db=db)
############################
@@ -615,14 +614,14 @@ class KnowledgeFileIdForm(BaseModel):
@router.post('/{id}/file/add', response_model=Optional[KnowledgeFilesResponse])
-def add_file_to_knowledge_by_id(
+async def add_file_to_knowledge_by_id(
request: Request,
id: str,
form_data: KnowledgeFileIdForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -631,7 +630,7 @@ def add_file_to_knowledge_by_id(
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -645,7 +644,7 @@ def add_file_to_knowledge_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- file = Files.get_file_by_id(form_data.file_id, db=db)
+ file = await Files.get_file_by_id(form_data.file_id, db=db)
if not file:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -659,7 +658,7 @@ def add_file_to_knowledge_by_id(
# Add content to the vector database
try:
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=form_data.file_id, collection_name=id),
user=user,
@@ -667,7 +666,7 @@ def add_file_to_knowledge_by_id(
)
# Add file to knowledge base
- Knowledges.add_file_to_knowledge_by_id(knowledge_id=id, file_id=form_data.file_id, user_id=user.id, db=db)
+ await Knowledges.add_file_to_knowledge_by_id(knowledge_id=id, file_id=form_data.file_id, user_id=user.id, db=db)
except Exception as e:
log.debug(e)
raise HTTPException(
@@ -678,7 +677,7 @@ def add_file_to_knowledge_by_id(
if knowledge:
return KnowledgeFilesResponse(
**knowledge.model_dump(),
- files=Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
+ files=await Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
)
else:
raise HTTPException(
@@ -688,14 +687,14 @@ def add_file_to_knowledge_by_id(
@router.post('/{id}/file/update', response_model=Optional[KnowledgeFilesResponse])
-def update_file_from_knowledge_by_id(
+async def update_file_from_knowledge_by_id(
request: Request,
id: str,
form_data: KnowledgeFileIdForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -704,7 +703,7 @@ def update_file_from_knowledge_by_id(
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -718,7 +717,7 @@ def update_file_from_knowledge_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- file = Files.get_file_by_id(form_data.file_id, db=db)
+ file = await Files.get_file_by_id(form_data.file_id, db=db)
if not file:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -726,18 +725,18 @@ def update_file_from_knowledge_by_id(
)
# Validate the file actually belongs to this knowledge base
- if not Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
+ if not await Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.NOT_FOUND,
)
# Remove content from the vector database
- VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': form_data.file_id})
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=knowledge.id, filter={'file_id': form_data.file_id})
# Add content to the vector database
try:
- process_file(
+ await process_file(
request,
ProcessFileForm(file_id=form_data.file_id, collection_name=id),
user=user,
@@ -752,7 +751,7 @@ def update_file_from_knowledge_by_id(
if knowledge:
return KnowledgeFilesResponse(
**knowledge.model_dump(),
- files=Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
+ files=await Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
)
else:
raise HTTPException(
@@ -767,14 +766,14 @@ def update_file_from_knowledge_by_id(
@router.post('/{id}/file/remove', response_model=Optional[KnowledgeFilesResponse])
-def remove_file_from_knowledge_by_id(
+async def remove_file_from_knowledge_by_id(
id: str,
form_data: KnowledgeFileIdForm,
delete_file: bool = Query(True),
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -783,7 +782,7 @@ def remove_file_from_knowledge_by_id(
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -797,7 +796,7 @@ def remove_file_from_knowledge_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- file = Files.get_file_by_id(form_data.file_id, db=db)
+ file = await Files.get_file_by_id(form_data.file_id, db=db)
if not file:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -805,21 +804,21 @@ def remove_file_from_knowledge_by_id(
)
# Validate the file actually belongs to this knowledge base
- if not Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
+ if not await Knowledges.has_file(knowledge_id=id, file_id=form_data.file_id, db=db):
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.NOT_FOUND,
)
- Knowledges.remove_file_from_knowledge_by_id(knowledge_id=id, file_id=form_data.file_id, db=db)
+ await Knowledges.remove_file_from_knowledge_by_id(knowledge_id=id, file_id=form_data.file_id, db=db)
# Remove content from the vector database
try:
- VECTOR_DB_CLIENT.delete(
+ await ASYNC_VECTOR_DB_CLIENT.delete(
collection_name=knowledge.id, filter={'file_id': form_data.file_id}
) # Remove by file_id first
- VECTOR_DB_CLIENT.delete(
+ await ASYNC_VECTOR_DB_CLIENT.delete(
collection_name=knowledge.id, filter={'hash': file.hash}
) # Remove by hash as well in case of duplicates
except Exception as e:
@@ -831,20 +830,20 @@ def remove_file_from_knowledge_by_id(
try:
# Remove the file's collection from vector database
file_collection = f'file-{form_data.file_id}'
- if VECTOR_DB_CLIENT.has_collection(collection_name=file_collection):
- VECTOR_DB_CLIENT.delete_collection(collection_name=file_collection)
+ if await ASYNC_VECTOR_DB_CLIENT.has_collection(collection_name=file_collection):
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=file_collection)
except Exception as e:
log.debug('This was most likely caused by bypassing embedding processing')
log.debug(e)
pass
# Delete file from database
- Files.delete_file_by_id(form_data.file_id, db=db)
+ await Files.delete_file_by_id(form_data.file_id, db=db)
if knowledge:
return KnowledgeFilesResponse(
**knowledge.model_dump(),
- files=Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
+ files=await Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
)
else:
raise HTTPException(
@@ -859,8 +858,10 @@ def remove_file_from_knowledge_by_id(
@router.delete('/{id}/delete', response_model=bool)
-async def delete_knowledge_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+async def delete_knowledge_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -869,7 +870,7 @@ async def delete_knowledge_by_id(id: str, user=Depends(get_verified_user), db: S
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -886,7 +887,7 @@ async def delete_knowledge_by_id(id: str, user=Depends(get_verified_user), db: S
log.info(f'Deleting knowledge base: {id} (name: {knowledge.name})')
# Get all models
- models = Models.get_all_models(db=db)
+ models = await Models.get_all_models(db=db)
log.info(f'Found {len(models)} models to check for knowledge base {id}')
# Update models that reference this knowledge base
@@ -910,19 +911,19 @@ async def delete_knowledge_by_id(id: str, user=Depends(get_verified_user), db: S
access_grants=model.access_grants,
is_active=model.is_active,
)
- Models.update_model_by_id(model.id, model_form, db=db)
+ await Models.update_model_by_id(model.id, model_form, db=db)
# Clean up vector DB
try:
- VECTOR_DB_CLIENT.delete_collection(collection_name=id)
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=id)
except Exception as e:
log.debug(e)
pass
# Remove knowledge base embedding
- remove_knowledge_base_metadata_embedding(id)
+ await remove_knowledge_base_metadata_embedding(id)
- result = Knowledges.delete_knowledge_by_id(id=id, db=db)
+ result = await Knowledges.delete_knowledge_by_id(id=id, db=db)
return result
@@ -932,8 +933,10 @@ async def delete_knowledge_by_id(id: str, user=Depends(get_verified_user), db: S
@router.post('/{id}/reset', response_model=Optional[KnowledgeResponse])
-async def reset_knowledge_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+async def reset_knowledge_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -942,7 +945,7 @@ async def reset_knowledge_by_id(id: str, user=Depends(get_verified_user), db: Se
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -957,12 +960,12 @@ async def reset_knowledge_by_id(id: str, user=Depends(get_verified_user), db: Se
)
try:
- VECTOR_DB_CLIENT.delete_collection(collection_name=id)
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=id)
except Exception as e:
log.debug(e)
pass
- knowledge = Knowledges.reset_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.reset_knowledge_by_id(id=id, db=db)
return knowledge
@@ -977,12 +980,12 @@ async def add_files_to_knowledge_batch(
id: str,
form_data: list[KnowledgeFileIdForm],
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""
Add multiple files to a knowledge base
"""
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -991,7 +994,7 @@ async def add_files_to_knowledge_batch(
if (
knowledge.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -1008,7 +1011,7 @@ async def add_files_to_knowledge_batch(
# Batch-fetch all files to avoid N+1 queries
log.info(f'files/batch/add - {len(form_data)} files')
file_ids = [form.file_id for form in form_data]
- files = Files.get_files_by_ids(file_ids, db=db)
+ files = await Files.get_files_by_ids(file_ids, db=db)
# Verify all requested files were found
found_ids = {file.id for file in files}
@@ -1034,14 +1037,14 @@ async def add_files_to_knowledge_batch(
# Only add files that were successfully processed
successful_file_ids = [r.file_id for r in result.results if r.status == 'completed']
for file_id in successful_file_ids:
- Knowledges.add_file_to_knowledge_by_id(knowledge_id=id, file_id=file_id, user_id=user.id, db=db)
+ await Knowledges.add_file_to_knowledge_by_id(knowledge_id=id, file_id=file_id, user_id=user.id, db=db)
# If there were any errors, include them in the response
if result.errors:
error_details = [f'{err.file_id}: {err.error}' for err in result.errors]
return KnowledgeFilesResponse(
**knowledge.model_dump(),
- files=Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
+ files=await Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
warnings={
'message': 'Some files failed to process',
'errors': error_details,
@@ -1050,7 +1053,7 @@ async def add_files_to_knowledge_batch(
return KnowledgeFilesResponse(
**knowledge.model_dump(),
- files=Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
+ files=await Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
)
@@ -1060,20 +1063,20 @@ async def add_files_to_knowledge_batch(
@router.get('/{id}/export')
-async def export_knowledge_by_id(id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
+async def export_knowledge_by_id(id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
"""
Export a knowledge base as a zip file containing .txt files.
Admin only.
"""
- knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
+ knowledge = await Knowledges.get_knowledge_by_id(id=id, db=db)
if not knowledge:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=ERROR_MESSAGES.NOT_FOUND,
)
- files = Knowledges.get_files_by_id(id, db=db)
+ files = await Knowledges.get_files_by_id(id, db=db)
# Create zip file in memory
zip_buffer = io.BytesIO()
diff --git a/backend/open_webui/routers/memories.py b/backend/open_webui/routers/memories.py
index 4557f0c44d..3a42801d01 100644
--- a/backend/open_webui/routers/memories.py
+++ b/backend/open_webui/routers/memories.py
@@ -5,10 +5,10 @@ import asyncio
from typing import Optional
from open_webui.models.memories import Memories, MemoryModel
-from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
from open_webui.utils.auth import get_verified_user
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from open_webui.utils.access_control import has_permission
from open_webui.constants import ERROR_MESSAGES
@@ -29,7 +29,7 @@ router = APIRouter()
async def get_memories(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not request.app.state.config.ENABLE_MEMORIES:
raise HTTPException(
@@ -37,13 +37,13 @@ async def get_memories(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- return Memories.get_memories_by_user_id(user.id, db=db)
+ return await Memories.get_memories_by_user_id(user.id, db=db)
############################
@@ -65,7 +65,7 @@ async def add_memory(
form_data: AddMemoryForm,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (insert_new_memory) manage their own short-lived sessions.
# This prevents holding a connection during EMBEDDING_FUNCTION()
# which makes external embedding API calls (1-5+ seconds).
@@ -75,17 +75,17 @@ async def add_memory(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- memory = Memories.insert_new_memory(user.id, form_data.content)
+ memory = await Memories.insert_new_memory(user.id, form_data.content)
vector = await request.app.state.EMBEDDING_FUNCTION(memory.content, user=user)
- VECTOR_DB_CLIENT.upsert(
+ await ASYNC_VECTOR_DB_CLIENT.upsert(
collection_name=f'user-memory-{user.id}',
items=[
{
@@ -116,7 +116,7 @@ async def query_memory(
form_data: QueryMemoryForm,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (get_memories_by_user_id) manage their own short-lived sessions.
# This prevents holding a connection during EMBEDDING_FUNCTION()
# which makes external embedding API calls (1-5+ seconds).
@@ -126,19 +126,19 @@ async def query_memory(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- memories = Memories.get_memories_by_user_id(user.id)
+ memories = await Memories.get_memories_by_user_id(user.id)
if not memories:
raise HTTPException(status_code=404, detail='No memories found for user')
vector = await request.app.state.EMBEDDING_FUNCTION(form_data.content, user=user)
- results = VECTOR_DB_CLIENT.search(
+ results = await ASYNC_VECTOR_DB_CLIENT.search(
collection_name=f'user-memory-{user.id}',
vectors=[vector],
limit=form_data.k,
@@ -157,7 +157,7 @@ async def reset_memory_from_vector_db(
):
"""Reset user's memory vector embeddings.
- CRITICAL: We intentionally do NOT use Depends(get_session) here.
+ CRITICAL: We intentionally do NOT use Depends(get_async_session) here.
This endpoint generates embeddings for ALL user memories in parallel using
asyncio.gather(). A user with 100 memories would trigger 100 embedding API
calls simultaneously. With a session held, this could block a connection
@@ -169,22 +169,22 @@ async def reset_memory_from_vector_db(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- VECTOR_DB_CLIENT.delete_collection(f'user-memory-{user.id}')
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(f'user-memory-{user.id}')
- memories = Memories.get_memories_by_user_id(user.id)
+ memories = await Memories.get_memories_by_user_id(user.id)
# Generate vectors in parallel
vectors = await asyncio.gather(
*[request.app.state.EMBEDDING_FUNCTION(memory.content, user=user) for memory in memories]
)
- VECTOR_DB_CLIENT.upsert(
+ await ASYNC_VECTOR_DB_CLIENT.upsert(
collection_name=f'user-memory-{user.id}',
items=[
{
@@ -212,7 +212,7 @@ async def reset_memory_from_vector_db(
async def delete_memory_by_user_id(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not request.app.state.config.ENABLE_MEMORIES:
raise HTTPException(
@@ -220,17 +220,17 @@ async def delete_memory_by_user_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- result = Memories.delete_memories_by_user_id(user.id, db=db)
+ result = await Memories.delete_memories_by_user_id(user.id, db=db)
if result:
try:
- VECTOR_DB_CLIENT.delete_collection(f'user-memory-{user.id}')
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(f'user-memory-{user.id}')
except Exception as e:
log.error(e)
return True
@@ -250,7 +250,7 @@ async def update_memory_by_id(
form_data: MemoryUpdateModel,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (update_memory_by_id_and_user_id) manage their own
# short-lived sessions. This prevents holding a connection during
# EMBEDDING_FUNCTION() which makes external API calls (1-5+ seconds).
@@ -260,20 +260,20 @@ async def update_memory_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- memory = Memories.update_memory_by_id_and_user_id(memory_id, user.id, form_data.content)
+ memory = await Memories.update_memory_by_id_and_user_id(memory_id, user.id, form_data.content)
if memory is None:
- raise HTTPException(status_code=404, detail='Memory not found')
+ raise HTTPException(status_code=404, detail=ERROR_MESSAGES.NOT_FOUND)
if form_data.content is not None:
vector = await request.app.state.EMBEDDING_FUNCTION(memory.content, user=user)
- VECTOR_DB_CLIENT.upsert(
+ await ASYNC_VECTOR_DB_CLIENT.upsert(
collection_name=f'user-memory-{user.id}',
items=[
{
@@ -301,7 +301,7 @@ async def delete_memory_by_id(
memory_id: str,
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not request.app.state.config.ENABLE_MEMORIES:
raise HTTPException(
@@ -309,16 +309,16 @@ async def delete_memory_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
- if not has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
+ if not await has_permission(user.id, 'features.memories', request.app.state.config.USER_PERMISSIONS):
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- result = Memories.delete_memory_by_id_and_user_id(memory_id, user.id, db=db)
+ result = await Memories.delete_memory_by_id_and_user_id(memory_id, user.id, db=db)
if result:
- VECTOR_DB_CLIENT.delete(collection_name=f'user-memory-{user.id}', ids=[memory_id])
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=f'user-memory-{user.id}', ids=[memory_id])
return True
return False
diff --git a/backend/open_webui/routers/models.py b/backend/open_webui/routers/models.py
index 6f7b3d48df..b7d31321ff 100644
--- a/backend/open_webui/routers/models.py
+++ b/backend/open_webui/routers/models.py
@@ -35,8 +35,8 @@ from fastapi.responses import FileResponse, StreamingResponse
from open_webui.utils.auth import get_admin_user, get_verified_user
from open_webui.utils.access_control import has_permission, filter_allowed_access_grants
from open_webui.config import BYPASS_ADMIN_ACCESS_CONTROL, STATIC_DIR
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
@@ -66,7 +66,7 @@ async def get_models(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
limit = PAGE_ITEM_COUNT
@@ -86,7 +86,7 @@ async def get_models(
filter['direction'] = direction
# Pre-fetch user group IDs once - used for both filter and write_access check
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
user_group_ids = {group.id for group in groups}
if not user.role == 'admin' or not BYPASS_ADMIN_ACCESS_CONTROL:
@@ -95,11 +95,11 @@ async def get_models(
filter['user_id'] = user.id
- result = Models.search_models(user.id, filter=filter, skip=skip, limit=limit, db=db)
+ result = await Models.search_models(user.id, filter=filter, skip=skip, limit=limit, db=db)
# Batch-fetch writable model IDs in a single query instead of N has_access calls
model_ids = [model.id for model in result.items]
- writable_model_ids = AccessGrants.get_accessible_resource_ids(
+ writable_model_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='model',
resource_ids=model_ids,
@@ -130,8 +130,8 @@ async def get_models(
@router.get('/base', response_model=list[ModelResponse])
-async def get_base_models(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- return Models.get_base_models(db=db)
+async def get_base_models(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ return await Models.get_base_models(db=db)
###########################
@@ -140,11 +140,11 @@ async def get_base_models(user=Depends(get_admin_user), db: Session = Depends(ge
@router.get('/tags', response_model=list[str])
-async def get_model_tags(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_model_tags(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- models = Models.get_models(db=db)
+ models = await Models.get_models(db=db)
else:
- models = Models.get_models_by_user_id(user.id, db=db)
+ models = await Models.get_models_by_user_id(user.id, db=db)
tags_set = set()
for model in models:
@@ -172,9 +172,9 @@ async def create_new_model(
request: Request,
form_data: ModelForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'workspace.models', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -182,7 +182,7 @@ async def create_new_model(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- model = Models.get_model_by_id(form_data.id, db=db)
+ model = await Models.get_model_by_id(form_data.id, db=db)
if model:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -196,7 +196,7 @@ async def create_new_model(
)
else:
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -204,7 +204,7 @@ async def create_new_model(
'sharing.public_models',
)
- model = Models.insert_new_model(form_data, user.id, db=db)
+ model = await Models.insert_new_model(form_data, user.id, db=db)
if model:
return model
else:
@@ -223,9 +223,9 @@ async def create_new_model(
async def export_models(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id,
'workspace.models_export',
request.app.state.config.USER_PERMISSIONS,
@@ -237,9 +237,9 @@ async def export_models(
)
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- return Models.get_models(db=db)
+ return await Models.get_models(db=db)
else:
- return Models.get_models_by_user_id(user.id, db=db)
+ return await Models.get_models_by_user_id(user.id, db=db)
############################
@@ -256,9 +256,9 @@ async def import_models(
request: Request,
user=Depends(get_verified_user),
form_data: ModelsImportForm = (...),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id,
'workspace.models_import',
request.app.state.config.USER_PERMISSIONS,
@@ -278,7 +278,7 @@ async def import_models(
if model_data.get('id') and is_valid_model_id(model_data.get('id'))
]
existing_models = {
- model.id: model for model in (Models.get_models_by_ids(model_ids, db=db) if model_ids else [])
+ model.id: model for model in (await Models.get_models_by_ids(model_ids, db=db) if model_ids else [])
}
for model_data in data:
@@ -293,13 +293,13 @@ async def import_models(
model_data['params'] = model_data.get('params', {})
updated_model = ModelForm(**{**existing_model.model_dump(), **model_data})
- Models.update_model_by_id(model_id, updated_model, db=db)
+ await Models.update_model_by_id(model_id, updated_model, db=db)
else:
# Insert new model
model_data['meta'] = model_data.get('meta', {})
model_data['params'] = model_data.get('params', {})
new_model = ModelForm(**model_data)
- Models.insert_new_model(user_id=user.id, form_data=new_model, db=db)
+ await Models.insert_new_model(user_id=user.id, form_data=new_model, db=db)
return True
else:
raise HTTPException(status_code=400, detail='Invalid JSON format')
@@ -322,9 +322,9 @@ async def sync_models(
request: Request,
form_data: SyncModelsForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- return Models.sync_models(user.id, form_data.models, db=db)
+ return await Models.sync_models(user.id, form_data.models, db=db)
###########################
@@ -338,13 +338,13 @@ class ModelIdForm(BaseModel):
# Note: We're not using the typical url path param here, but instead using a query parameter to allow '/' in the id
@router.get('/model', response_model=Optional[ModelAccessResponse])
-async def get_model_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- model = Models.get_model_by_id(id, db=db)
+async def get_model_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ model = await Models.get_model_by_id(id, db=db)
if model:
if (
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or model.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model.id,
@@ -357,7 +357,7 @@ async def get_model_by_id(id: str, user=Depends(get_verified_user), db: Session
write_access=(
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == model.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model.id,
@@ -384,8 +384,8 @@ async def get_model_by_id(id: str, user=Depends(get_verified_user), db: Session
@router.get('/model/profile/image')
-def get_model_profile_image(id: str, user=Depends(get_verified_user)):
- model = Models.get_model_by_id(id)
+async def get_model_profile_image(id: str, user=Depends(get_verified_user)):
+ model = await Models.get_model_by_id(id)
if model:
etag = f'"{model.updated_at}"' if model.updated_at else None
@@ -426,13 +426,13 @@ def get_model_profile_image(id: str, user=Depends(get_verified_user)):
@router.post('/model/toggle', response_model=Optional[ModelResponse])
-async def toggle_model_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- model = Models.get_model_by_id(id, db=db)
+async def toggle_model_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ model = await Models.get_model_by_id(id, db=db)
if model:
if (
user.role == 'admin'
or model.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model.id,
@@ -440,7 +440,7 @@ async def toggle_model_by_id(id: str, user=Depends(get_verified_user), db: Sessi
db=db,
)
):
- model = Models.toggle_model_by_id(id, db=db)
+ model = await Models.toggle_model_by_id(id, db=db)
if model:
return model
@@ -471,9 +471,9 @@ async def update_model_by_id(
request: Request,
form_data: ModelForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- model = Models.get_model_by_id(form_data.id, db=db)
+ model = await Models.get_model_by_id(form_data.id, db=db)
if not model:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -482,7 +482,7 @@ async def update_model_by_id(
if (
model.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model.id,
@@ -496,7 +496,7 @@ async def update_model_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -504,7 +504,7 @@ async def update_model_by_id(
'sharing.public_models',
)
- model = Models.update_model_by_id(form_data.id, ModelForm(**form_data.model_dump()), db=db)
+ model = await Models.update_model_by_id(form_data.id, ModelForm(**form_data.model_dump()), db=db)
return model
@@ -524,9 +524,9 @@ async def update_model_access_by_id(
request: Request,
form_data: ModelAccessGrantsForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- model = Models.get_model_by_id(form_data.id, db=db)
+ model = await Models.get_model_by_id(form_data.id, db=db)
# Non-preset models (e.g. direct Ollama/OpenAI models) may not have a DB
# entry yet. Create a minimal one so access grants can be stored.
@@ -536,7 +536,7 @@ async def update_model_access_by_id(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- model = Models.insert_new_model(
+ model = await Models.insert_new_model(
ModelForm(
id=form_data.id,
name=form_data.name or form_data.id,
@@ -554,7 +554,7 @@ async def update_model_access_by_id(
if (
model.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model.id,
@@ -568,7 +568,7 @@ async def update_model_access_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -576,11 +576,11 @@ async def update_model_access_by_id(
'sharing.public_models',
)
- AccessGrants.set_access_grants('model', form_data.id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('model', form_data.id, form_data.access_grants, db=db)
- Models.update_model_updated_at_by_id(form_data.id, db=db)
+ await Models.update_model_updated_at_by_id(form_data.id, db=db)
- return Models.get_model_by_id(form_data.id, db=db)
+ return await Models.get_model_by_id(form_data.id, db=db)
############################
@@ -592,9 +592,9 @@ async def update_model_access_by_id(
async def delete_model_by_id(
form_data: ModelIdForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- model = Models.get_model_by_id(form_data.id, db=db)
+ model = await Models.get_model_by_id(form_data.id, db=db)
if not model:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -604,7 +604,7 @@ async def delete_model_by_id(
if (
user.role != 'admin'
and model.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model.id,
@@ -617,11 +617,11 @@ async def delete_model_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- result = Models.delete_model_by_id(form_data.id, db=db)
+ result = await Models.delete_model_by_id(form_data.id, db=db)
return result
@router.delete('/delete/all', response_model=bool)
-async def delete_all_models(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- result = Models.delete_all_models(db=db)
+async def delete_all_models(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ result = await Models.delete_all_models(db=db)
return result
diff --git a/backend/open_webui/routers/notes.py b/backend/open_webui/routers/notes.py
index 0eec88a251..4fbdd09993 100644
--- a/backend/open_webui/routers/notes.py
+++ b/backend/open_webui/routers/notes.py
@@ -34,8 +34,8 @@ from open_webui.utils.access_control import (
filter_allowed_access_grants,
)
from open_webui.models.access_grants import AccessGrants
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
@@ -58,6 +58,7 @@ class NoteItemResponse(BaseModel):
id: str
title: str
data: Optional[dict]
+ is_pinned: Optional[bool] = False
updated_at: int
created_at: int
user: Optional[UserResponse] = None
@@ -68,9 +69,9 @@ async def get_notes(
request: Request,
page: Optional[int] = None,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -84,12 +85,51 @@ async def get_notes(
limit = 60
skip = (page - 1) * limit
- notes = Notes.get_notes_by_user_id(user.id, 'read', skip=skip, limit=limit, db=db)
+ notes = await Notes.get_notes_by_user_id(user.id, 'read', skip=skip, limit=limit, db=db)
if not notes:
return []
user_ids = list(set(note.user_id for note in notes))
- users = {user.id: user for user in Users.get_users_by_user_ids(user_ids, db=db)}
+ users = {user.id: user for user in await Users.get_users_by_user_ids(user_ids, db=db)}
+
+ return [
+ NoteUserResponse(
+ **{
+ **note.model_dump(),
+ 'data': _truncate_note_data(note.data),
+ 'user': UserResponse(**users[note.user_id].model_dump()),
+ }
+ )
+ for note in notes
+ if note.user_id in users
+ ]
+
+
+############################
+# GetPinnedNotes
+############################
+
+
+@router.get('/pinned', response_model=list[NoteItemResponse])
+async def get_pinned_notes(
+ request: Request,
+ user=Depends(get_verified_user),
+ db: AsyncSession = Depends(get_async_session),
+):
+ if user.role != 'admin' and not await has_permission(
+ user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
+ ):
+ raise HTTPException(
+ status_code=status.HTTP_401_UNAUTHORIZED,
+ detail=ERROR_MESSAGES.UNAUTHORIZED,
+ )
+
+ notes = await Notes.get_pinned_notes_by_user_id(user.id, 'read', db=db)
+ if not notes:
+ return []
+
+ user_ids = list(set(note.user_id for note in notes))
+ users = {user.id: user for user in await Users.get_users_by_user_ids(user_ids, db=db)}
return [
NoteUserResponse(
@@ -114,9 +154,9 @@ async def search_notes(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -143,13 +183,13 @@ async def search_notes(
filter['direction'] = direction
if not user.role == 'admin' or not BYPASS_ADMIN_ACCESS_CONTROL:
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
if groups:
filter['group_ids'] = [group.id for group in groups]
filter['user_id'] = user.id
- result = Notes.search_notes(user.id, filter, skip=skip, limit=limit, db=db)
+ result = await Notes.search_notes(user.id, filter, skip=skip, limit=limit, db=db)
for note in result.items:
note.data = _truncate_note_data(note.data)
return result
@@ -165,9 +205,9 @@ async def create_new_note(
request: Request,
form_data: NoteForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -175,7 +215,7 @@ async def create_new_note(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -185,7 +225,7 @@ async def create_new_note(
)
try:
- note = Notes.insert_new_note(user.id, form_data, db=db)
+ note = await Notes.insert_new_note(user.id, form_data, db=db)
return note
except Exception as e:
log.exception(e)
@@ -206,9 +246,9 @@ async def get_note_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -216,14 +256,14 @@ async def get_note_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- note = Notes.get_note_by_id(id, db=db)
+ note = await Notes.get_note_by_id(id, db=db)
if not note:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if user.role != 'admin' and (
user.id != note.user_id
and (
- not AccessGrants.has_access(
+ not await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -237,7 +277,7 @@ async def get_note_by_id(
write_access = (
user.role == 'admin'
or (user.id == note.user_id)
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -261,9 +301,9 @@ async def update_note_by_id(
id: str,
form_data: NoteForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -271,13 +311,13 @@ async def update_note_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- note = Notes.get_note_by_id(id, db=db)
+ note = await Notes.get_note_by_id(id, db=db)
if not note:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if user.role != 'admin' and (
user.id != note.user_id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -287,7 +327,7 @@ async def update_note_by_id(
):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -297,7 +337,7 @@ async def update_note_by_id(
)
try:
- note = Notes.update_note_by_id(id, form_data, db=db)
+ note = await Notes.update_note_by_id(id, form_data, db=db)
await sio.emit(
'note-events',
note.model_dump(),
@@ -325,9 +365,9 @@ async def update_note_access_by_id(
id: str,
form_data: NoteAccessGrantsForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -335,13 +375,13 @@ async def update_note_access_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- note = Notes.get_note_by_id(id, db=db)
+ note = await Notes.get_note_by_id(id, db=db)
if not note:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if user.role != 'admin' and (
user.id != note.user_id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -351,7 +391,7 @@ async def update_note_access_by_id(
):
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -359,9 +399,49 @@ async def update_note_access_by_id(
'sharing.public_notes',
)
- AccessGrants.set_access_grants('note', id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('note', id, form_data.access_grants, db=db)
- return Notes.get_note_by_id(id, db=db)
+ return await Notes.get_note_by_id(id, db=db)
+
+
+############################
+# PinNoteById
+############################
+
+
+@router.post('/{id}/pin', response_model=Optional[NoteModel])
+async def pin_note_by_id(
+ request: Request,
+ id: str,
+ user=Depends(get_verified_user),
+ db: AsyncSession = Depends(get_async_session),
+):
+ if user.role != 'admin' and not await has_permission(
+ user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
+ ):
+ raise HTTPException(
+ status_code=status.HTTP_401_UNAUTHORIZED,
+ detail=ERROR_MESSAGES.UNAUTHORIZED,
+ )
+
+ note = await Notes.get_note_by_id(id, db=db)
+ if not note:
+ raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
+
+ if user.role != 'admin' and (
+ user.id != note.user_id
+ and not await AccessGrants.has_access(
+ user_id=user.id,
+ resource_type='note',
+ resource_id=note.id,
+ permission='read',
+ db=db,
+ )
+ ):
+ raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
+
+ note = await Notes.toggle_note_pinned_by_id(id, db=db)
+ return note
############################
@@ -374,9 +454,9 @@ async def delete_note_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.notes', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -384,13 +464,13 @@ async def delete_note_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- note = Notes.get_note_by_id(id, db=db)
+ note = await Notes.get_note_by_id(id, db=db)
if not note:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND)
if user.role != 'admin' and (
user.id != note.user_id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -401,7 +481,7 @@ async def delete_note_by_id(
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT())
try:
- note = Notes.delete_note_by_id(id, db=db)
+ note = await Notes.delete_note_by_id(id, db=db)
return True
except Exception as e:
log.exception(e)
diff --git a/backend/open_webui/routers/ollama.py b/backend/open_webui/routers/ollama.py
index 93745440c4..9272db1e6a 100644
--- a/backend/open_webui/routers/ollama.py
+++ b/backend/open_webui/routers/ollama.py
@@ -39,17 +39,21 @@ from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import StreamingResponse
from pydantic import BaseModel, ConfigDict, validator
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
-from open_webui.internal.db import get_session
+from open_webui.internal.db import get_async_session
from open_webui.models.models import Models
from open_webui.models.access_grants import AccessGrants
from open_webui.models.groups import Groups
+from open_webui.utils.access_control import check_model_access
from open_webui.utils.misc import (
calculate_sha256,
+)
+from open_webui.utils.session_pool import (
cleanup_response,
+ get_session,
stream_wrapper,
)
from open_webui.utils.payload import (
@@ -121,10 +125,7 @@ async def send_request(
r = None
streaming = False
try:
- session = aiohttp.ClientSession(
- trust_env=True,
- timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
- )
+ session = await get_session()
headers = {
'Content-Type': 'application/json',
@@ -137,8 +138,12 @@ async def send_request(
headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = metadata.get('chat_id')
r = await session.request(
- method, url, data=payload, headers=headers,
+ method,
+ url,
+ data=payload,
+ headers=headers,
ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
)
if not r.ok:
@@ -152,7 +157,7 @@ async def send_request(
log.error(f'Failed to parse error response: {e}')
raise HTTPException(
status_code=r.status,
- detail='Open WebUI: Server Connection Error',
+ detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR,
)
r.raise_for_status()
@@ -164,7 +169,7 @@ async def send_request(
streaming = True
return StreamingResponse(
- stream_wrapper(r, session),
+ stream_wrapper(r),
status_code=r.status,
headers=response_headers,
)
@@ -179,11 +184,11 @@ async def send_request(
except Exception as e:
raise HTTPException(
status_code=r.status if r else 500,
- detail=f'Ollama: {e}' if str(e) else 'Open WebUI: Server Connection Error',
+ detail=f'Ollama: {e}' if str(e) else ERROR_MESSAGES.SERVER_CONNECTION_ERROR,
)
finally:
if not streaming:
- await cleanup_response(r, session)
+ await cleanup_response(r)
def get_api_key(idx, url, configs):
@@ -246,7 +251,7 @@ async def verify_connection(form_data: ConnectionVerificationForm, user=Depends(
return data
except aiohttp.ClientError as e:
log.exception(f'Client error: {str(e)}')
- raise HTTPException(status_code=500, detail='Open WebUI: Server Connection Error')
+ raise HTTPException(status_code=500, detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR)
except Exception as e:
log.exception(f'Unexpected error: {e}')
error_detail = f'Unexpected error: {str(e)}'
@@ -397,11 +402,11 @@ async def get_all_models(request: Request, user: UserModel = None):
async def get_filtered_models(models, user, db=None):
# Filter models based on user access control
model_ids = [model['model'] for model in models.get('models', [])]
- model_infos = {model_info.id: model_info for model_info in Models.get_models_by_ids(model_ids, db=db)}
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
+ model_infos = {model_info.id: model_info for model_info in await Models.get_models_by_ids(model_ids, db=db)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
# Batch-fetch accessible resource IDs in a single query instead of N has_access calls
- accessible_model_ids = AccessGrants.get_accessible_resource_ids(
+ accessible_model_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='model',
resource_ids=list(model_infos.keys()),
@@ -423,7 +428,7 @@ async def get_filtered_models(models, user, db=None):
@router.get('/api/tags/{url_idx}')
async def get_ollama_tags(request: Request, url_idx: Optional[int] = None, user=Depends(get_verified_user)):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
models = []
@@ -616,7 +621,7 @@ async def pull_model(
user=Depends(get_admin_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
form_data = form_data.model_dump(exclude_none=True)
form_data['model'] = form_data.get('model', form_data.get('name'))
@@ -651,7 +656,7 @@ async def push_model(
user=Depends(get_admin_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
if url_idx is None:
await get_all_models(request, user=user)
@@ -694,7 +699,7 @@ async def create_model(
user=Depends(get_admin_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
log.debug(f'form_data: {form_data}')
url = request.app.state.config.OLLAMA_BASE_URLS[url_idx]
@@ -722,7 +727,7 @@ async def copy_model(
user=Depends(get_admin_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
if url_idx is None:
await get_all_models(request, user=user)
@@ -757,7 +762,7 @@ async def delete_model(
user=Depends(get_admin_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
form_data = form_data.model_dump(exclude_none=True)
form_data['model'] = form_data.get('model', form_data.get('name'))
@@ -780,7 +785,8 @@ async def delete_model(
key = get_api_key(url_idx, url, request.app.state.config.OLLAMA_API_CONFIGS)
await send_request(
- f'{url}/api/delete', 'DELETE',
+ f'{url}/api/delete',
+ 'DELETE',
payload=json.dumps(form_data),
key=key,
user=user,
@@ -791,16 +797,19 @@ async def delete_model(
@router.post('/api/show')
async def show_model_info(request: Request, form_data: ModelNameForm, user=Depends(get_verified_user)):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
form_data = form_data.model_dump(exclude_none=True)
form_data['model'] = form_data.get('model', form_data.get('name'))
+ model = form_data.get('model')
+
+ # Enforce per-model access control
+ await check_model_access(user, await Models.get_model_by_id(model), BYPASS_MODEL_ACCESS_CONTROL)
+
await get_all_models(request, user=user)
models = request.app.state.OLLAMA_MODELS
- model = form_data.get('model')
-
if model not in models:
raise HTTPException(
status_code=400,
@@ -841,10 +850,13 @@ async def embed(
user=Depends(get_verified_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
log.info(f'generate_ollama_batch_embeddings {form_data}')
+ # Enforce per-model access control
+ await check_model_access(user, await Models.get_model_by_id(form_data.model), BYPASS_MODEL_ACCESS_CONTROL)
+
if url_idx is None:
model = form_data.model
@@ -897,10 +909,13 @@ async def embeddings(
user=Depends(get_verified_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
log.info(f'generate_ollama_embeddings {form_data}')
+ # Enforce per-model access control
+ await check_model_access(user, await Models.get_model_by_id(form_data.model), BYPASS_MODEL_ACCESS_CONTROL)
+
if url_idx is None:
model = form_data.model
@@ -961,13 +976,17 @@ async def generate_completion(
user=Depends(get_verified_user),
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
+
+ # Enforce per-model access control
+ await check_model_access(user, await Models.get_model_by_id(form_data.model), BYPASS_MODEL_ACCESS_CONTROL)
if url_idx is None:
await get_all_models(request, user=user)
models = request.app.state.OLLAMA_MODELS
model = form_data.model
+
if model in models:
url_idx = random.choice(models[model]['urls'])
else:
@@ -1001,6 +1020,8 @@ class ChatMessage(BaseModel):
tool_calls: Optional[list[dict]] = None
images: Optional[list[str]] = None
+ model_config = ConfigDict(extra='allow')
+
@validator('content', pre=True)
@classmethod
def check_at_least_one_field(cls, field_value, values, **kwargs):
@@ -1048,9 +1069,9 @@ async def generate_chat_completion(
bypass_system_prompt: bool = False,
):
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (get_model_by_id, AccessGrants.has_access) manage their own short-lived sessions.
# This prevents holding a connection during the entire LLM call (30-60+ seconds),
# which would exhaust the connection pool under concurrent load.
@@ -1079,7 +1100,7 @@ async def generate_chat_completion(
del payload['metadata']
model_id = payload['model']
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id:
@@ -1097,29 +1118,9 @@ async def generate_chat_completion(
if not bypass_system_prompt:
payload = apply_system_prompt_to_body(system, payload, metadata, user)
- # Check if user has access to the model
- if not bypass_filter and user.role == 'user':
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not (
- user.id == model_info.user_id
- or AccessGrants.has_access(
- user_id=user.id,
- resource_type='model',
- resource_id=model_info.id,
- permission='read',
- user_group_ids=user_group_ids,
- )
- ):
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
- elif not bypass_filter:
- if user.role != 'admin':
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, model_info, bypass_filter)
+ else:
+ await check_model_access(user, None, bypass_filter)
url, url_idx = await get_ollama_url(request, payload['model'], url_idx)
api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
@@ -1177,7 +1178,7 @@ async def generate_openai_completion(
url_idx: Optional[int] = None,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (get_model_by_id, AccessGrants.has_access) manage their own short-lived sessions.
# This prevents holding a connection during the entire LLM call (30-60+ seconds),
# which would exhaust the connection pool under concurrent load.
@@ -1197,7 +1198,7 @@ async def generate_openai_completion(
del payload['metadata']
model_id = form_data.model
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id:
payload['model'] = model_info.base_model_id
@@ -1206,29 +1207,9 @@ async def generate_openai_completion(
if params:
payload = apply_model_params_to_body_openai(params, payload)
- # Check if user has access to the model
- if user.role == 'user':
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not (
- user.id == model_info.user_id
- or AccessGrants.has_access(
- user_id=user.id,
- resource_type='model',
- resource_id=model_info.id,
- permission='read',
- user_group_ids=user_group_ids,
- )
- ):
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, model_info)
else:
- if user.role != 'admin':
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, None)
url, url_idx = await get_ollama_url(request, payload['model'], url_idx)
api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
@@ -1259,7 +1240,7 @@ async def generate_openai_chat_completion(
url_idx: Optional[int] = None,
user=Depends(get_verified_user),
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (get_model_by_id, AccessGrants.has_access) manage their own short-lived sessions.
# This prevents holding a connection during the entire LLM call (30-60+ seconds),
# which would exhaust the connection pool under concurrent load.
@@ -1279,7 +1260,7 @@ async def generate_openai_chat_completion(
del payload['metadata']
model_id = completion_form.model
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id:
payload['model'] = model_info.base_model_id
@@ -1292,29 +1273,9 @@ async def generate_openai_chat_completion(
payload = apply_model_params_to_body_openai(params, payload)
payload = apply_system_prompt_to_body(system, payload, metadata, user)
- # Check if user has access to the model
- if user.role == 'user':
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not (
- user.id == model_info.user_id
- or AccessGrants.has_access(
- user_id=user.id,
- resource_type='model',
- resource_id=model_info.id,
- permission='read',
- user_group_ids=user_group_ids,
- )
- ):
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, model_info)
else:
- if user.role != 'admin':
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, None)
url, url_idx = await get_ollama_url(request, payload['model'], url_idx)
api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
@@ -1354,39 +1315,19 @@ async def generate_anthropic_messages(
See https://docs.ollama.com/api/anthropic-compatibility
"""
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
payload = {**form_data}
model_id = payload.get('model', '')
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id:
payload['model'] = model_info.base_model_id
- # Check if user has access to the model
- if user.role == 'user':
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not (
- user.id == model_info.user_id
- or AccessGrants.has_access(
- user_id=user.id,
- resource_type='model',
- resource_id=model_info.id,
- permission='read',
- user_group_ids=user_group_ids,
- )
- ):
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, model_info)
else:
- if user.role != 'admin':
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, None)
url, url_idx = await get_ollama_url(request, payload['model'], url_idx)
api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
@@ -1432,22 +1373,22 @@ async def generate_responses(
See https://ollama.com/blog/responses-api
"""
if not request.app.state.config.ENABLE_OLLAMA_API:
- raise HTTPException(status_code=503, detail='Ollama API is disabled')
+ raise HTTPException(status_code=503, detail=ERROR_MESSAGES.OLLAMA_API_DISABLED)
payload = form_data.model_dump()
model_id = form_data.model
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id:
payload['model'] = model_info.base_model_id
# Check if user has access to the model
if user.role == 'user':
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
if not (
user.id == model_info.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model_info.id,
@@ -1457,13 +1398,13 @@ async def generate_responses(
):
raise HTTPException(
status_code=403,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
else:
if user.role != 'admin':
raise HTTPException(
status_code=403,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
url, url_idx = await get_ollama_url(request, payload['model'], url_idx)
@@ -1492,7 +1433,7 @@ async def get_openai_models(
request: Request,
url_idx: Optional[int] = None,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
models = []
if url_idx is None:
@@ -1524,11 +1465,11 @@ async def get_openai_models(
if user.role == 'user' and not BYPASS_MODEL_ACCESS_CONTROL:
# Filter models based on user access control
model_ids = [model['id'] for model in models]
- model_infos = {model_info.id: model_info for model_info in Models.get_models_by_ids(model_ids, db=db)}
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
+ model_infos = {model_info.id: model_info for model_info in await Models.get_models_by_ids(model_ids, db=db)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
# Batch-fetch accessible resource IDs in a single query instead of N has_access calls
- accessible_model_ids = AccessGrants.get_accessible_resource_ids(
+ accessible_model_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='model',
resource_ids=list(model_infos.keys()),
@@ -1651,7 +1592,7 @@ async def download_model(
file_name = parse_huggingface_url(form_data.url)
if file_name:
- file_path = f'{UPLOAD_DIR}/{file_name}'
+ file_path = os.path.join(UPLOAD_DIR, file_name)
return StreamingResponse(
download_file_stream(url, form_data.url, file_path, file_name),
@@ -1715,9 +1656,7 @@ async def upload_model(
url = f'{ollama_url}/api/blobs/sha256:{file_hash}'
upload_timeout = aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT)
async with aiohttp.ClientSession(timeout=upload_timeout, trust_env=True) as upload_session:
- async with upload_session.post(
- url, data=blob_data, ssl=AIOHTTP_CLIENT_SESSION_SSL
- ) as response:
+ async with upload_session.post(url, data=blob_data, ssl=AIOHTTP_CLIENT_SESSION_SSL) as response:
if not response.ok:
raise Exception('Ollama: Could not create blob, Please try again.')
diff --git a/backend/open_webui/routers/openai.py b/backend/open_webui/routers/openai.py
index 836517df9d..82c844afba 100644
--- a/backend/open_webui/routers/openai.py
+++ b/backend/open_webui/routers/openai.py
@@ -4,15 +4,15 @@ import json
import logging
import re
from typing import Optional
-from urllib.parse import urlparse
+from urllib.parse import quote, urlparse
import aiohttp
from aiocache import cached
-import requests
+
from azure.identity import DefaultAzureCredential, get_bearer_token_provider
-from fastapi import Depends, HTTPException, Request, APIRouter
+from fastapi import Depends, HTTPException, Request, APIRouter, status
from fastapi.responses import (
FileResponse,
StreamingResponse,
@@ -21,13 +21,14 @@ from fastapi.responses import (
)
from pydantic import BaseModel, ConfigDict
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
-from open_webui.internal.db import get_session
+from open_webui.internal.db import get_async_session
from open_webui.models.models import Models
from open_webui.models.access_grants import AccessGrants
from open_webui.models.groups import Groups
+from open_webui.utils.access_control import has_connection_access, check_model_access
from open_webui.config import (
CACHE_DIR,
)
@@ -39,6 +40,7 @@ from open_webui.env import (
ENABLE_FORWARD_USER_INFO_HEADERS,
FORWARD_SESSION_INFO_HEADER_CHAT_ID,
BYPASS_MODEL_ACCESS_CONTROL,
+ ENABLE_OPENAI_API_PASSTHROUGH,
)
from open_webui.models.users import UserModel
@@ -50,9 +52,12 @@ from open_webui.utils.payload import (
apply_system_prompt_to_body,
)
from open_webui.utils.misc import (
- cleanup_response,
convert_logit_bias_input_to_json,
stream_chunks_handler,
+)
+from open_webui.utils.session_pool import (
+ cleanup_response,
+ get_session,
stream_wrapper,
)
@@ -307,19 +312,20 @@ async def speech(request: Request, user=Depends(get_verified_user)):
r = None
try:
- r = requests.post(
+ session = await get_session()
+ r = await session.post(
url=f'{url}/audio/speech',
data=body,
headers=headers,
cookies=cookies,
- stream=True,
+ ssl=AIOHTTP_CLIENT_SESSION_SSL,
)
r.raise_for_status()
# Save the streaming content to a file
with open(file_path, 'wb') as f:
- for chunk in r.iter_content(chunk_size=8192):
+ async for chunk in r.content.iter_chunked(8192):
f.write(chunk)
with open(file_body_path, 'w') as f:
@@ -334,14 +340,14 @@ async def speech(request: Request, user=Depends(get_verified_user)):
detail = None
if r is not None:
try:
- res = r.json()
+ res = await r.json()
if 'error' in res:
detail = f'External: {res["error"]}'
except Exception:
detail = f'External: {e}'
raise HTTPException(
- status_code=r.status_code if r else 500,
+ status_code=r.status if r else 500,
detail=detail if detail else 'Open WebUI: Server Connection Error',
)
@@ -449,11 +455,11 @@ async def get_all_models_responses(request: Request, user: UserModel) -> list:
async def get_filtered_models(models, user, db=None):
# Filter models based on user access control
model_ids = [model['id'] for model in models.get('data', [])]
- model_infos = {model_info.id: model_info for model_info in Models.get_models_by_ids(model_ids, db=db)}
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
+ model_infos = {model_info.id: model_info for model_info in await Models.get_models_by_ids(model_ids, db=db)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
# Batch-fetch accessible resource IDs in a single query instead of N has_access calls
- accessible_model_ids = AccessGrants.get_accessible_resource_ids(
+ accessible_model_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='model',
resource_ids=list(model_infos.keys()),
@@ -686,7 +692,7 @@ async def verify_connection(
elif is_anthropic_url(url):
result = await get_anthropic_models(url, key)
if result is None:
- raise HTTPException(status_code=500, detail='Failed to connect to Anthropic API')
+ raise HTTPException(status_code=500, detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR)
if 'error' in result:
raise HTTPException(status_code=500, detail=result['error'])
return result
@@ -713,10 +719,10 @@ async def verify_connection(
except aiohttp.ClientError as e:
# ClientError covers all aiohttp requests issues
log.exception(f'Client error: {str(e)}')
- raise HTTPException(status_code=500, detail='Open WebUI: Server Connection Error')
+ raise HTTPException(status_code=500, detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR)
except Exception as e:
log.exception(f'Unexpected error: {e}')
- raise HTTPException(status_code=500, detail='Open WebUI: Server Connection Error')
+ raise HTTPException(status_code=500, detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR)
def get_azure_allowed_params(api_version: str) -> set[str]:
@@ -771,6 +777,21 @@ def is_openai_new_model(model: str) -> bool:
return False
+def _sanitize_model_for_url(model: str) -> str:
+ """Sanitize a model name before interpolating it into a URL path.
+
+ Rejects path traversal attempts (../, /, \\) and percent-encodes
+ the name so it is safe to use as a single URL path segment
+ (e.g. Azure deployment name).
+ """
+ if not model or '..' in model or '/' in model or '\\' in model:
+ raise HTTPException(
+ status_code=400,
+ detail='Invalid model name: must not be empty or contain path separators or traversal sequences',
+ )
+ return quote(model, safe='')
+
+
def convert_to_azure_payload(url, payload: dict, api_version: str):
model = payload.get('model', '')
@@ -794,6 +815,9 @@ def convert_to_azure_payload(url, payload: dict, api_version: str):
# Filter out unsupported parameters
payload = {k: v for k, v in payload.items() if k in allowed_params}
+ # Sanitize model name to prevent path traversal in the deployment URL
+ model = _sanitize_model_for_url(model)
+
url = f'{url}/openai/deployments/{model}'
return url, payload
@@ -1006,7 +1030,7 @@ async def generate_chat_completion(
user=Depends(get_verified_user),
bypass_system_prompt: bool = False,
):
- # NOTE: We intentionally do NOT use Depends(get_session) here.
+ # NOTE: We intentionally do NOT use Depends(get_async_session) here.
# Database operations (get_model_by_id, AccessGrants.has_access) manage their own short-lived sessions.
# This prevents holding a connection during the entire LLM call (30-60+ seconds),
# which would exhaust the connection pool under concurrent load.
@@ -1024,7 +1048,7 @@ async def generate_chat_completion(
metadata = payload.pop('metadata', None)
model_id = form_data.get('model')
- model_info = Models.get_model_by_id(model_id)
+ model_info = await Models.get_model_by_id(model_id)
# Check model info and override the payload
if model_info:
@@ -1044,29 +1068,9 @@ async def generate_chat_completion(
if not bypass_system_prompt:
payload = apply_system_prompt_to_body(system, payload, metadata, user)
- # Check if user has access to the model
- if not bypass_filter and user.role == 'user':
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not (
- user.id == model_info.user_id
- or AccessGrants.has_access(
- user_id=user.id,
- resource_type='model',
- resource_id=model_info.id,
- permission='read',
- user_group_ids=user_group_ids,
- )
- ):
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
- elif not bypass_filter:
- if user.role != 'admin':
- raise HTTPException(
- status_code=403,
- detail='Model not found',
- )
+ await check_model_access(user, model_info, bypass_filter)
+ else:
+ await check_model_access(user, None, bypass_filter)
# Check if model is already in app state cache to avoid expensive get_all_models() call
models = request.app.state.OPENAI_MODELS
@@ -1080,7 +1084,7 @@ async def generate_chat_completion(
else:
raise HTTPException(
status_code=404,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Get the API config for the model
@@ -1174,12 +1178,11 @@ async def generate_chat_completion(
payload = json.dumps(payload)
r = None
- session = None
streaming = False
response = None
try:
- session = aiohttp.ClientSession(trust_env=True, timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT))
+ session = await get_session()
r = await session.request(
method='POST',
@@ -1188,13 +1191,33 @@ async def generate_chat_completion(
headers=headers,
cookies=cookies,
ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
)
# Check if response is SSE
if 'text/event-stream' in r.headers.get('Content-Type', ''):
+ # If the provider returned an error status with SSE content-type,
+ # read the body and return a proper error response instead of
+ # streaming the error back (which hides the error from logs).
+ if r.status >= 400:
+ error_body = await r.text()
+ log.error(
+ 'Provider returned HTTP %d with SSE content-type: %s',
+ r.status,
+ error_body[:1000],
+ )
+ try:
+ error_json = json.loads(error_body)
+ return JSONResponse(status_code=r.status, content=error_json)
+ except json.JSONDecodeError:
+ return JSONResponse(
+ status_code=r.status,
+ content={'error': {'message': error_body, 'code': r.status}},
+ )
+
streaming = True
return StreamingResponse(
- stream_wrapper(r, session, stream_chunks_handler),
+ stream_wrapper(r, content_handler=stream_chunks_handler),
status_code=r.status,
headers=dict(r.headers),
)
@@ -1221,11 +1244,11 @@ async def generate_chat_completion(
raise HTTPException(
status_code=r.status if r else 500,
- detail='Open WebUI: Server Connection Error',
+ detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR,
)
finally:
if not streaming:
- await cleanup_response(r, session)
+ await cleanup_response(r)
async def embeddings(request: Request, form_data: dict, user):
@@ -1261,27 +1284,24 @@ async def embeddings(request: Request, form_data: dict, user):
)
r = None
- session = None
streaming = False
headers, cookies = await get_headers_and_cookies(request, url, key, api_config, user=user)
try:
- session = aiohttp.ClientSession(
- trust_env=True,
- timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
- )
+ session = await get_session()
r = await session.request(
method='POST',
url=f'{url}/embeddings',
data=body,
headers=headers,
cookies=cookies,
+ timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
)
if 'text/event-stream' in r.headers.get('Content-Type', ''):
streaming = True
return StreamingResponse(
- stream_wrapper(r, session),
+ stream_wrapper(r),
status_code=r.status,
headers=dict(r.headers),
)
@@ -1302,11 +1322,11 @@ async def embeddings(request: Request, form_data: dict, user):
log.exception(e)
raise HTTPException(
status_code=r.status if r else 500,
- detail='Open WebUI: Server Connection Error',
+ detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR,
)
finally:
if not streaming:
- await cleanup_response(r, session)
+ await cleanup_response(r)
class ResponsesForm(BaseModel):
@@ -1340,10 +1360,15 @@ async def responses(
Routes to the correct upstream backend based on the model field.
"""
payload = form_data.model_dump(exclude_none=True)
- body = json.dumps(payload)
idx = 0
model_id = form_data.model
+
+ # Enforce per-model access control
+ await check_model_access(user, await Models.get_model_by_id(model_id), BYPASS_MODEL_ACCESS_CONTROL)
+
+ body = json.dumps(payload)
+
if model_id:
models = request.app.state.OPENAI_MODELS
if not models or model_id not in models:
@@ -1360,7 +1385,6 @@ async def responses(
)
r = None
- session = None
streaming = False
try:
@@ -1378,15 +1402,12 @@ async def responses(
else:
api_version = api_config.get('api_version', '2023-03-15-preview')
headers['api-version'] = api_version
- model = payload.get('model', '')
+ model = _sanitize_model_for_url(payload.get('model', ''))
request_url = f'{url}/openai/deployments/{model}/responses?api-version={api_version}'
else:
request_url = f'{url}/responses'
- session = aiohttp.ClientSession(
- trust_env=True,
- timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
- )
+ session = await get_session()
r = await session.request(
method='POST',
url=request_url,
@@ -1394,13 +1415,14 @@ async def responses(
headers=headers,
cookies=cookies,
ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
)
# Check if response is SSE
if 'text/event-stream' in r.headers.get('Content-Type', ''):
streaming = True
return StreamingResponse(
- stream_wrapper(r, session),
+ stream_wrapper(r),
status_code=r.status,
headers=dict(r.headers),
)
@@ -1418,23 +1440,32 @@ async def responses(
return response_data
+ except HTTPException:
+ raise
except Exception as e:
log.exception(e)
raise HTTPException(
status_code=r.status if r else 500,
- detail='Open WebUI: Server Connection Error',
+ detail=ERROR_MESSAGES.SERVER_CONNECTION_ERROR,
)
finally:
if not streaming:
- await cleanup_response(r, session)
+ await cleanup_response(r)
@router.api_route('/{path:path}', methods=['GET', 'POST', 'PUT', 'DELETE'])
async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
"""
- Deprecated: proxy all requests to OpenAI API
+ Deprecated: proxy all requests to OpenAI API.
+ Disabled by default. Set ENABLE_OPENAI_API_PASSTHROUGH=True to enable.
"""
+ if not ENABLE_OPENAI_API_PASSTHROUGH:
+ raise HTTPException(
+ status_code=status.HTTP_403_FORBIDDEN,
+ detail='Direct API passthrough is disabled. Set ENABLE_OPENAI_API_PASSTHROUGH=True to enable.',
+ )
+
body = await request.body()
# Parse JSON body to resolve model-based routing
@@ -1465,7 +1496,6 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
)
r = None
- session = None
streaming = False
try:
@@ -1494,10 +1524,7 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
else:
request_url = f'{url}/{path}'
- session = aiohttp.ClientSession(
- trust_env=True,
- timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
- )
+ session = await get_session()
r = await session.request(
method=request.method,
url=request_url,
@@ -1505,13 +1532,14 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
headers=headers,
cookies=cookies,
ssl=AIOHTTP_CLIENT_SESSION_SSL,
+ timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT),
)
# Check if response is SSE
if 'text/event-stream' in r.headers.get('Content-Type', ''):
streaming = True
return StreamingResponse(
- stream_wrapper(r, session),
+ stream_wrapper(r),
status_code=r.status,
headers=dict(r.headers),
)
@@ -1529,6 +1557,8 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
return response_data
+ except HTTPException:
+ raise
except Exception as e:
log.exception(e)
raise HTTPException(
@@ -1537,4 +1567,4 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
)
finally:
if not streaming:
- await cleanup_response(r, session)
+ await cleanup_response(r)
diff --git a/backend/open_webui/routers/prompts.py b/backend/open_webui/routers/prompts.py
index 3b579c2892..11901fc5a7 100644
--- a/backend/open_webui/routers/prompts.py
+++ b/backend/open_webui/routers/prompts.py
@@ -20,8 +20,8 @@ from open_webui.constants import ERROR_MESSAGES
from open_webui.utils.auth import get_admin_user, get_verified_user
from open_webui.utils.access_control import has_permission, filter_allowed_access_grants
from open_webui.config import BYPASS_ADMIN_ACCESS_CONTROL
-from open_webui.internal.db import get_session
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from pydantic import BaseModel
@@ -48,21 +48,21 @@ PAGE_ITEM_COUNT = 30
@router.get('/', response_model=list[PromptModel])
-async def get_prompts(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_prompts(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- prompts = Prompts.get_prompts(db=db)
+ prompts = await Prompts.get_prompts(db=db)
else:
- prompts = Prompts.get_prompts_by_user_id(user.id, 'read', db=db)
+ prompts = await Prompts.get_prompts_by_user_id(user.id, 'read', db=db)
return prompts
@router.get('/tags', response_model=list[str])
-async def get_prompt_tags(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_prompt_tags(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- return Prompts.get_tags(db=db)
+ return await Prompts.get_tags(db=db)
else:
- prompts = Prompts.get_prompts_by_user_id(user.id, 'read', db=db)
+ prompts = await Prompts.get_prompts_by_user_id(user.id, 'read', db=db)
tags = set()
for prompt in prompts:
if prompt.tags:
@@ -79,7 +79,7 @@ async def get_prompt_list(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
limit = PAGE_ITEM_COUNT
@@ -99,7 +99,7 @@ async def get_prompt_list(
filter['direction'] = direction
# Pre-fetch user group IDs once - used for both filter and write_access check
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
user_group_ids = {group.id for group in groups}
if not (user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL):
@@ -108,11 +108,11 @@ async def get_prompt_list(
filter['user_id'] = user.id
- result = Prompts.search_prompts(user.id, filter=filter, skip=skip, limit=limit, db=db)
+ result = await Prompts.search_prompts(user.id, filter=filter, skip=skip, limit=limit, db=db)
# Batch-fetch writable prompt IDs in a single query instead of N has_access calls
prompt_ids = [prompt.id for prompt in result.items]
- writable_prompt_ids = AccessGrants.get_accessible_resource_ids(
+ writable_prompt_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='prompt',
resource_ids=prompt_ids,
@@ -147,16 +147,16 @@ async def create_new_prompt(
request: Request,
form_data: PromptForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if user.role != 'admin' and not (
- has_permission(
+ await has_permission(
user.id,
'workspace.prompts',
request.app.state.config.USER_PERMISSIONS,
db=db,
)
- or has_permission(
+ or await has_permission(
user.id,
'workspace.prompts_import',
request.app.state.config.USER_PERMISSIONS,
@@ -168,7 +168,7 @@ async def create_new_prompt(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -176,9 +176,9 @@ async def create_new_prompt(
'sharing.public_prompts',
)
- prompt = Prompts.get_prompt_by_command(form_data.command, db=db)
+ prompt = await Prompts.get_prompt_by_command(form_data.command, db=db)
if prompt is None:
- prompt = Prompts.insert_new_prompt(user.id, form_data, db=db)
+ prompt = await Prompts.insert_new_prompt(user.id, form_data, db=db)
if prompt:
return prompt
@@ -198,14 +198,16 @@ async def create_new_prompt(
@router.get('/command/{command}', response_model=Optional[PromptAccessResponse])
-async def get_prompt_by_command(command: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- prompt = Prompts.get_prompt_by_command(command, db=db)
+async def get_prompt_by_command(
+ command: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ prompt = await Prompts.get_prompt_by_command(command, db=db)
if prompt:
if (
user.role == 'admin'
or prompt.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -218,7 +220,7 @@ async def get_prompt_by_command(command: str, user=Depends(get_verified_user), d
write_access=(
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == prompt.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -240,14 +242,16 @@ async def get_prompt_by_command(command: str, user=Depends(get_verified_user), d
@router.get('/id/{prompt_id}', response_model=Optional[PromptAccessResponse])
-async def get_prompt_by_id(prompt_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+async def get_prompt_by_id(
+ prompt_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if prompt:
if (
user.role == 'admin'
or prompt.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -260,7 +264,7 @@ async def get_prompt_by_id(prompt_id: str, user=Depends(get_verified_user), db:
write_access=(
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == prompt.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -287,9 +291,9 @@ async def update_prompt_by_id(
prompt_id: str,
form_data: PromptForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -300,7 +304,7 @@ async def update_prompt_by_id(
# Is the user the original creator, in a group with write access, or an admin
if (
prompt.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -316,14 +320,14 @@ async def update_prompt_by_id(
# Check for command collision if command is being changed
if form_data.command != prompt.command:
- existing_prompt = Prompts.get_prompt_by_command(form_data.command, db=db)
+ existing_prompt = await Prompts.get_prompt_by_command(form_data.command, db=db)
if existing_prompt and existing_prompt.id != prompt.id:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail=f"Command '/{form_data.command}' is already in use by another prompt",
+ detail=ERROR_MESSAGES.COMMAND_TAKEN,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -332,7 +336,7 @@ async def update_prompt_by_id(
)
# Use the ID from the found prompt
- updated_prompt = Prompts.update_prompt_by_id(prompt.id, form_data, user.id, db=db)
+ updated_prompt = await Prompts.update_prompt_by_id(prompt.id, form_data, user.id, db=db)
if updated_prompt:
return updated_prompt
else:
@@ -352,10 +356,10 @@ async def update_prompt_metadata(
prompt_id: str,
form_data: PromptMetadataForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Update prompt name and command only (no history created)."""
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -365,7 +369,7 @@ async def update_prompt_metadata(
if (
prompt.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -381,14 +385,16 @@ async def update_prompt_metadata(
# Check for command collision if command is being changed
if form_data.command != prompt.command:
- existing_prompt = Prompts.get_prompt_by_command(form_data.command, db=db)
+ existing_prompt = await Prompts.get_prompt_by_command(form_data.command, db=db)
if existing_prompt and existing_prompt.id != prompt.id:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail=f"Command '/{form_data.command}' is already in use",
+ detail=ERROR_MESSAGES.COMMAND_TAKEN,
)
- updated_prompt = Prompts.update_prompt_metadata(prompt.id, form_data.name, form_data.command, form_data.tags, db=db)
+ updated_prompt = await Prompts.update_prompt_metadata(
+ prompt.id, form_data.name, form_data.command, form_data.tags, db=db
+ )
if updated_prompt:
return updated_prompt
else:
@@ -403,9 +409,9 @@ async def set_prompt_version(
prompt_id: str,
form_data: PromptVersionUpdateForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -414,7 +420,7 @@ async def set_prompt_version(
if (
prompt.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -428,7 +434,7 @@ async def set_prompt_version(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- updated_prompt = Prompts.update_prompt_version(prompt.id, form_data.version_id, db=db)
+ updated_prompt = await Prompts.update_prompt_version(prompt.id, form_data.version_id, db=db)
if updated_prompt:
return updated_prompt
else:
@@ -453,9 +459,9 @@ async def update_prompt_access_by_id(
prompt_id: str,
form_data: PromptAccessGrantsForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -464,7 +470,7 @@ async def update_prompt_access_by_id(
if (
prompt.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -478,7 +484,7 @@ async def update_prompt_access_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -486,9 +492,9 @@ async def update_prompt_access_by_id(
'sharing.public_prompts',
)
- AccessGrants.set_access_grants('prompt', prompt_id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('prompt', prompt_id, form_data.access_grants, db=db)
- return Prompts.get_prompt_by_id(prompt_id, db=db)
+ return await Prompts.get_prompt_by_id(prompt_id, db=db)
############################
@@ -497,8 +503,10 @@ async def update_prompt_access_by_id(
@router.post('/id/{prompt_id}/toggle', response_model=Optional[PromptModel])
-async def toggle_prompt_active(prompt_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+async def toggle_prompt_active(
+ prompt_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -508,7 +516,7 @@ async def toggle_prompt_active(prompt_id: str, user=Depends(get_verified_user),
if (
prompt.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -522,7 +530,7 @@ async def toggle_prompt_active(prompt_id: str, user=Depends(get_verified_user),
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- result = Prompts.toggle_prompt_active(prompt.id, db=db)
+ result = await Prompts.toggle_prompt_active(prompt.id, db=db)
if result:
return result
raise HTTPException(
@@ -537,8 +545,10 @@ async def toggle_prompt_active(prompt_id: str, user=Depends(get_verified_user),
@router.delete('/id/{prompt_id}/delete', response_model=bool)
-async def delete_prompt_by_id(prompt_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+async def delete_prompt_by_id(
+ prompt_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -548,7 +558,7 @@ async def delete_prompt_by_id(prompt_id: str, user=Depends(get_verified_user), d
if (
prompt.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -562,7 +572,7 @@ async def delete_prompt_by_id(prompt_id: str, user=Depends(get_verified_user), d
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- result = Prompts.delete_prompt_by_id(prompt.id, db=db)
+ result = await Prompts.delete_prompt_by_id(prompt.id, db=db)
return result
@@ -576,12 +586,12 @@ async def get_prompt_history(
prompt_id: str,
page: int = 0,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get version history for a prompt."""
PAGE_SIZE = 20
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -593,7 +603,7 @@ async def get_prompt_history(
if not (
user.role == 'admin'
or prompt.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -606,7 +616,7 @@ async def get_prompt_history(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- history = PromptHistories.get_history_by_prompt_id(prompt.id, limit=PAGE_SIZE, offset=page * PAGE_SIZE, db=db)
+ history = await PromptHistories.get_history_by_prompt_id(prompt.id, limit=PAGE_SIZE, offset=page * PAGE_SIZE, db=db)
return history
@@ -615,10 +625,10 @@ async def get_prompt_history_entry(
prompt_id: str,
history_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get a specific version from history."""
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -630,7 +640,7 @@ async def get_prompt_history_entry(
if not (
user.role == 'admin'
or prompt.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -643,7 +653,7 @@ async def get_prompt_history_entry(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- history_entry = PromptHistories.get_history_entry_by_id(history_id, db=db)
+ history_entry = await PromptHistories.get_history_entry_by_id(history_id, db=db)
if not history_entry or history_entry.prompt_id != prompt.id:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -658,10 +668,10 @@ async def delete_prompt_history_entry(
prompt_id: str,
history_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Delete a history entry. Cannot delete the active production version."""
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -673,7 +683,7 @@ async def delete_prompt_history_entry(
if not (
user.role == 'admin'
or prompt.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -693,7 +703,7 @@ async def delete_prompt_history_entry(
detail='Cannot delete the active production version',
)
- success = PromptHistories.delete_history_entry(history_id, db=db)
+ success = await PromptHistories.delete_history_entry(history_id, db=db)
if not success:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -709,10 +719,10 @@ async def get_prompt_diff(
from_id: str,
to_id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get diff between two versions."""
- prompt = Prompts.get_prompt_by_id(prompt_id, db=db)
+ prompt = await Prompts.get_prompt_by_id(prompt_id, db=db)
if not prompt:
raise HTTPException(
@@ -724,7 +734,7 @@ async def get_prompt_diff(
if not (
user.role == 'admin'
or prompt.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='prompt',
resource_id=prompt.id,
@@ -737,11 +747,11 @@ async def get_prompt_diff(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- diff = PromptHistories.compute_diff(from_id, to_id, db=db)
+ diff = await PromptHistories.compute_diff(from_id, to_id, db=db)
if not diff:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='One or both history entries not found',
+ detail=ERROR_MESSAGES.NOT_FOUND,
)
return diff
diff --git a/backend/open_webui/routers/retrieval.py b/backend/open_webui/routers/retrieval.py
index 3921039208..89ef452acf 100644
--- a/backend/open_webui/routers/retrieval.py
+++ b/backend/open_webui/routers/retrieval.py
@@ -40,11 +40,12 @@ from open_webui.models.files import FileModel, FileUpdateForm, Files
from open_webui.utils.access_control.files import has_access_to_file
from open_webui.models.knowledge import Knowledges
from open_webui.storage.provider import Storage
-from open_webui.internal.db import get_session, get_db
-from sqlalchemy.orm import Session
+from open_webui.internal.db import get_async_db, get_async_session
+from sqlalchemy.ext.asyncio import AsyncSession
from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
# Document loaders
from open_webui.retrieval.loaders.main import Loader
@@ -151,7 +152,7 @@ def get_ef(
model_kwargs=SENTENCE_TRANSFORMERS_MODEL_KWARGS,
)
except Exception as e:
- log.debug(f'Error loading SentenceTransformer: {e}')
+ log.error(f'Error loading SentenceTransformer: {e}')
return ef
@@ -1542,11 +1543,11 @@ class ProcessFileForm(BaseModel):
@router.post('/process/file')
-def process_file(
+async def process_file(
request: Request,
form_data: ProcessFileForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""
Process a file and save its content to the vector database.
@@ -1555,9 +1556,9 @@ def process_file(
The session is committed before external API calls, and updates use a fresh session.
"""
if user.role == 'admin':
- file = Files.get_file_by_id(form_data.file_id, db=db)
+ file = await Files.get_file_by_id(form_data.file_id, db=db)
else:
- file = Files.get_file_by_id_and_user_id(form_data.file_id, user.id, db=db)
+ file = await Files.get_file_by_id_and_user_id(form_data.file_id, user.id, db=db)
if file:
try:
@@ -1572,7 +1573,7 @@ def process_file(
try:
# /files/{file_id}/data/content/update
- VECTOR_DB_CLIENT.delete_collection(collection_name=f'file-{file.id}')
+ await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=f'file-{file.id}')
except Exception:
# Audio file upload pipeline
pass
@@ -1595,7 +1596,9 @@ def process_file(
# Check if the file has already been processed and save the content
# Usage: /knowledge/{id}/file/add, /knowledge/{id}/file/update
- result = VECTOR_DB_CLIENT.query(collection_name=f'file-{file.id}', filter={'file_id': file.id})
+ result = await ASYNC_VECTOR_DB_CLIENT.query(
+ collection_name=f'file-{file.id}', filter={'file_id': file.id}
+ )
if result is not None and len(result.ids[0]) > 0:
docs = [
@@ -1625,7 +1628,7 @@ def process_file(
# Usage: /files/
file_path = file.path
if file_path:
- file_path = Storage.get_file(file_path)
+ file_path = await asyncio.to_thread(Storage.get_file, file_path)
loader = Loader(
engine=request.app.state.config.CONTENT_EXTRACTION_ENGINE,
user=user,
@@ -1661,7 +1664,7 @@ def process_file(
MINERU_API_TIMEOUT=request.app.state.config.MINERU_API_TIMEOUT,
MINERU_PARAMS=request.app.state.config.MINERU_PARAMS,
)
- docs = loader.load(file.filename, file.meta.get('content_type'), file_path)
+ docs = await loader.aload(file.filename, file.meta.get('content_type'), file_path)
docs = [
Document(
@@ -1692,7 +1695,7 @@ def process_file(
text_content = ' '.join([doc.page_content for doc in docs])
log.debug(f'text_content: {text_content}')
- Files.update_file_data_by_id(
+ await Files.update_file_data_by_id(
file.id,
{'content': text_content},
db=db,
@@ -1700,8 +1703,8 @@ def process_file(
hash = calculate_sha256_string(text_content)
if request.app.state.config.BYPASS_EMBEDDING_AND_RETRIEVAL:
- Files.update_file_data_by_id(file.id, {'status': 'completed'}, db=db)
- Files.update_file_hash_by_id(file.id, hash, db=db)
+ await Files.update_file_data_by_id(file.id, {'status': 'completed'}, db=db)
+ await Files.update_file_hash_by_id(file.id, hash, db=db)
return {
'status': True,
'collection_name': None,
@@ -1712,11 +1715,16 @@ def process_file(
try:
# Commit any pending changes before the slow embedding step.
# Note: file is already a Pydantic model (not ORM), so no expunge needed.
- db.commit()
+ await db.commit()
# External embedding API takes time (5-60s+).
- # Subsequent updates use fresh sessions via get_db().
- result = save_docs_to_vector_db(
+ # Subsequent updates use fresh async sessions.
+ # NOTE: save_docs_to_vector_db is a sync function that
+ # calls asyncio.run_coroutine_threadsafe(..., main_loop).result()
+ # which blocks the calling thread. We MUST run it in a
+ # worker thread to avoid deadlocking the event loop.
+ result = await run_in_threadpool(
+ save_docs_to_vector_db,
request,
docs=docs,
collection_name=collection_name,
@@ -1732,8 +1740,8 @@ def process_file(
if result:
# Fresh session for the final update.
- with get_db() as session:
- Files.update_file_metadata_by_id(
+ async with get_async_db() as session:
+ await Files.update_file_metadata_by_id(
file.id,
{
'collection_name': collection_name,
@@ -1741,12 +1749,12 @@ def process_file(
db=session,
)
- Files.update_file_data_by_id(
+ await Files.update_file_data_by_id(
file.id,
{'status': 'completed'},
db=session,
)
- Files.update_file_hash_by_id(file.id, hash, db=session)
+ await Files.update_file_hash_by_id(file.id, hash, db=session)
return {
'status': True,
@@ -1762,14 +1770,14 @@ def process_file(
except Exception as e:
log.exception(e)
# Fresh session for error status update.
- with get_db() as session:
- Files.update_file_data_by_id(
+ async with get_async_db() as session:
+ await Files.update_file_data_by_id(
file.id,
{'status': 'failed'},
db=session,
)
# Clear the hash so the file can be re-uploaded after fixing the issue
- Files.update_file_hash_by_id(file.id, None, db=session)
+ await Files.update_file_hash_by_id(file.id, None, db=session)
if 'No pandoc was found' in str(e):
raise HTTPException(
@@ -2193,7 +2201,7 @@ async def process_web_search(request: Request, form_data: SearchForm, user=Depen
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'features.web_search', request.app.state.config.USER_PERMISSIONS
):
raise HTTPException(
@@ -2345,7 +2353,7 @@ async def process_web_search(request: Request, form_data: SearchForm, user=Depen
)
-def _validate_collection_access(collection_names: list[str], user) -> None:
+async def _validate_collection_access(collection_names: list[str], user) -> None:
"""
Prevent users from querying collections they don't own.
Enforces ownership on user-memory-* and file-* collections.
@@ -2362,7 +2370,7 @@ def _validate_collection_access(collection_names: list[str], user) -> None:
)
elif name.startswith('file-'):
file_id = name[len('file-') :]
- if not has_access_to_file(
+ if not await has_access_to_file(
file_id=file_id,
access_type='read',
user=user,
@@ -2388,12 +2396,12 @@ async def query_doc_handler(
form_data: QueryDocForm,
user=Depends(get_verified_user),
):
- _validate_collection_access([form_data.collection_name], user)
+ await _validate_collection_access([form_data.collection_name], user)
try:
if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH and (form_data.hybrid is None or form_data.hybrid):
collection_results = {}
- collection_results[form_data.collection_name] = VECTOR_DB_CLIENT.get(
+ collection_results[form_data.collection_name] = await ASYNC_VECTOR_DB_CLIENT.get(
collection_name=form_data.collection_name
)
return await query_doc_with_hybrid_search(
@@ -2422,7 +2430,10 @@ async def query_doc_handler(
query_embedding = await request.app.state.EMBEDDING_FUNCTION(
form_data.query, prefix=RAG_EMBEDDING_QUERY_PREFIX, user=user
)
- return query_doc(
+ # query_doc wraps a blocking VECTOR_DB_CLIENT.search call;
+ # offload so the request's event loop stays responsive.
+ return await asyncio.to_thread(
+ query_doc,
collection_name=form_data.collection_name,
query_embedding=query_embedding,
k=form_data.k if form_data.k else request.app.state.config.TOP_K,
@@ -2453,7 +2464,7 @@ async def query_collection_handler(
form_data: QueryCollectionsForm,
user=Depends(get_verified_user),
):
- _validate_collection_access(form_data.collection_names, user)
+ await _validate_collection_access(form_data.collection_names, user)
try:
if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH and (form_data.hybrid is None or form_data.hybrid):
@@ -2514,14 +2525,14 @@ class DeleteForm(BaseModel):
@router.post('/delete')
-def delete_entries_from_collection(
+async def delete_entries_from_collection(
form_data: DeleteForm,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
try:
- if VECTOR_DB_CLIENT.has_collection(collection_name=form_data.collection_name):
- file = Files.get_file_by_id(form_data.file_id, db=db)
+ if await ASYNC_VECTOR_DB_CLIENT.has_collection(collection_name=form_data.collection_name):
+ file = await Files.get_file_by_id(form_data.file_id, db=db)
if not file:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -2529,26 +2540,50 @@ def delete_entries_from_collection(
)
hash = file.hash
- VECTOR_DB_CLIENT.delete(
+ # Refuse to issue a `filter={'hash': None}` query — the
+ # match semantics of a null filter value are
+ # backend-dependent (some backends ignore the key, some
+ # match every row whose metadata lacks `hash`) and risk
+ # deleting unrelated entries. Files without a hash are
+ # typically unprocessed / failed / legacy records that
+ # can't be targeted by hash anyway.
+ if hash is None:
+ raise HTTPException(
+ status_code=status.HTTP_400_BAD_REQUEST,
+ detail=ERROR_MESSAGES.DEFAULT('File has no hash; cannot delete vector entries by hash.'),
+ )
+
+ # Pre-existing bug: this used `metadata=` which is not a
+ # parameter on `VectorDBBase.delete` nor on any backend
+ # implementation, so the call always raised TypeError that
+ # was silently swallowed by the surrounding `except
+ # Exception` and the endpoint reported `{'status': False}`
+ # for every request. Use `filter` to actually do what the
+ # endpoint name promises.
+ await ASYNC_VECTOR_DB_CLIENT.delete(
collection_name=form_data.collection_name,
- metadata={'hash': hash},
+ filter={'hash': hash},
)
return {'status': True}
else:
return {'status': False}
+ except HTTPException:
+ # Caller-meaningful errors (404/400 above) must not be
+ # swallowed and re-shaped as `{'status': False}`.
+ raise
except Exception as e:
log.exception(e)
return {'status': False}
@router.post('/reset/db')
-def reset_vector_db(user=Depends(get_admin_user), db: Session = Depends(get_session)):
- VECTOR_DB_CLIENT.reset()
- Knowledges.delete_all_knowledge(db=db)
+async def reset_vector_db(user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
+ await ASYNC_VECTOR_DB_CLIENT.reset()
+ await Knowledges.delete_all_knowledge(db=db)
@router.post('/reset/uploads')
-def reset_upload_dir(user=Depends(get_admin_user)) -> bool:
+async def reset_upload_dir(user=Depends(get_admin_user)) -> bool:
folder = f'{UPLOAD_DIR}'
try:
# Check if the directory exists
@@ -2603,7 +2638,7 @@ async def process_files_batch(
"""
Process a batch of files and save them to the vector database.
- NOTE: We intentionally do NOT use Depends(get_session) here.
+ NOTE: We intentionally do NOT use Depends(get_async_session) here.
The save_docs_to_vector_db() call makes external embedding API calls which
can take 5-60+ seconds for batch operations. Database operations after
embedding (Files.update_file_by_id) manage their own short-lived sessions.
@@ -2621,7 +2656,7 @@ async def process_files_batch(
for file in form_data.files:
try:
# Ownership check: verify the requesting user owns the file or is an admin
- db_file = Files.get_file_by_id(file.id, db=db)
+ db_file = await Files.get_file_by_id(file.id, db=db)
if not db_file:
file_errors.append(
BatchProcessFilesResult(
@@ -2683,7 +2718,7 @@ async def process_files_batch(
# Update all files with collection name
for file_update, file_result in zip(file_updates, file_results):
- Files.update_file_by_id(id=file_result.file_id, form_data=file_update, db=db)
+ await Files.update_file_by_id(id=file_result.file_id, form_data=file_update, db=db)
file_result.status = 'completed'
except Exception as e:
diff --git a/backend/open_webui/routers/scim.py b/backend/open_webui/routers/scim.py
index 56923bc447..75f45bcaf9 100644
--- a/backend/open_webui/routers/scim.py
+++ b/backend/open_webui/routers/scim.py
@@ -5,6 +5,7 @@ Provides System for Cross-domain Identity Management endpoints for users and gro
NOTE: This is an experimental implementation and may not fully comply with SCIM 2.0 standards, and is subject to change.
"""
+import hmac
import logging
import uuid
import time
@@ -29,8 +30,8 @@ from open_webui.config import OAUTH_PROVIDERS
from open_webui.env import SCIM_AUTH_PROVIDER
-from sqlalchemy.orm import Session
-from open_webui.internal.db import get_session
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import get_async_session
log = logging.getLogger(__name__)
@@ -278,7 +279,7 @@ def get_scim_auth(request: Request, authorization: Optional[str] = Header(None))
if hasattr(scim_token, 'value'):
scim_token = scim_token.value
log.debug(f'SCIM token configured: {bool(scim_token)}')
- if not scim_token or token != scim_token:
+ if not scim_token or not hmac.compare_digest(token, scim_token):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail='Invalid SCIM token',
@@ -325,18 +326,18 @@ def get_scim_provider() -> str:
return SCIM_AUTH_PROVIDER
-def find_user_by_external_id(external_id: str, db=None) -> Optional[UserModel]:
+async def find_user_by_external_id(external_id: str, db=None) -> Optional[UserModel]:
"""Find a user by SCIM externalId, falling back to OAuth sub match."""
provider = get_scim_provider()
- user = Users.get_user_by_scim_external_id(provider, external_id, db=db)
+ user = await Users.get_user_by_scim_external_id(provider, external_id, db=db)
if user:
return user
# Fallback: check if externalId matches an existing OAuth sub (account linking)
- return Users.get_user_by_oauth_sub(provider, external_id, db=db)
+ return await Users.get_user_by_oauth_sub(provider, external_id, db=db)
-def user_to_scim(user: UserModel, request: Request, db=None) -> SCIMUser:
+async def user_to_scim(user: UserModel, request: Request, db=None) -> SCIMUser:
"""Convert internal User model to SCIM User"""
# Parse display name into name components
name_parts = user.name.split(' ', 1) if user.name else ['', '']
@@ -344,7 +345,7 @@ def user_to_scim(user: UserModel, request: Request, db=None) -> SCIMUser:
family_name = name_parts[1] if len(name_parts) > 1 else ''
# Get user's groups
- user_groups = Groups.get_groups_by_member_id(user.id, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user.id, db=db)
groups = [
{
'value': group.id,
@@ -378,12 +379,12 @@ def user_to_scim(user: UserModel, request: Request, db=None) -> SCIMUser:
)
-def group_to_scim(group: GroupModel, request: Request, db=None) -> SCIMGroup:
+async def group_to_scim(group: GroupModel, request: Request, db=None) -> SCIMGroup:
"""Convert internal Group model to SCIM Group"""
- member_ids = Groups.get_group_user_ids_by_id(group.id, db) or []
+ member_ids = await Groups.get_group_user_ids_by_id(group.id, db) or []
# Batch-fetch all users to avoid N+1 queries
- users = Users.get_users_by_user_ids(member_ids, db=db) if member_ids else []
+ users = await Users.get_users_by_user_ids(member_ids, db=db) if member_ids else []
members = [
SCIMGroupMember(
value=user.id,
@@ -511,7 +512,7 @@ async def get_users(
count: int = Query(20),
filter: Optional[str] = None,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""List SCIM Users"""
# Clamp per SCIM 2.0 spec (RFC 7644 §3.4.2.4):
@@ -526,25 +527,25 @@ async def get_users(
# Simple filter parsing - supports userName eq, externalId eq
if 'userName eq' in filter:
email = filter.split('"')[1]
- user = Users.get_user_by_email(email, db=db)
+ user = await Users.get_user_by_email(email, db=db)
users_list = [user] if user else []
total = 1 if user else 0
elif 'externalId eq' in filter:
external_id = filter.split('"')[1]
- user = find_user_by_external_id(external_id, db=db)
+ user = await find_user_by_external_id(external_id, db=db)
users_list = [user] if user else []
total = 1 if user else 0
else:
- response = Users.get_users(skip=skip, limit=limit, db=db)
+ response = await Users.get_users(skip=skip, limit=limit, db=db)
users_list = response['users']
total = response['total']
else:
- response = Users.get_users(skip=skip, limit=limit, db=db)
+ response = await Users.get_users(skip=skip, limit=limit, db=db)
users_list = response['users']
total = response['total']
# Convert to SCIM format
- scim_users = [user_to_scim(user, request, db=db) for user in users_list]
+ scim_users = [await user_to_scim(user, request, db=db) for user in users_list]
return SCIMListResponse(
totalResults=total,
@@ -559,14 +560,14 @@ async def get_user(
user_id: str,
request: Request,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get SCIM User by ID"""
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
if not user:
return scim_error(status_code=status.HTTP_404_NOT_FOUND, detail=f'User {user_id} not found')
- return user_to_scim(user, request, db=db)
+ return await user_to_scim(user, request, db=db)
@router.post('/Users', response_model=SCIMUser, status_code=status.HTTP_201_CREATED)
@@ -574,12 +575,12 @@ async def create_user(
request: Request,
user_data: SCIMUserCreateRequest,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Create SCIM User"""
# Check for duplicate by externalId
if user_data.externalId:
- existing_user = find_user_by_external_id(user_data.externalId, db=db)
+ existing_user = await find_user_by_external_id(user_data.externalId, db=db)
if existing_user:
raise HTTPException(
status_code=status.HTTP_409_CONFLICT,
@@ -595,7 +596,7 @@ async def create_user(
email = email.lower()
# Check for duplicate by email
- existing_user = Users.get_user_by_email(email, db=db)
+ existing_user = await Users.get_user_by_email(email, db=db)
if existing_user:
raise HTTPException(
status_code=status.HTTP_409_CONFLICT,
@@ -618,7 +619,7 @@ async def create_user(
if user_data.photos and len(user_data.photos) > 0:
profile_image = user_data.photos[0].value
- new_user = Users.insert_new_user(
+ new_user = await Users.insert_new_user(
id=user_id,
name=name,
email=email,
@@ -636,10 +637,10 @@ async def create_user(
# Store externalId in the scim field
if user_data.externalId:
provider = get_scim_provider()
- Users.update_user_scim_by_id(user_id, provider, user_data.externalId, db=db)
- new_user = Users.get_user_by_id(user_id, db=db)
+ await Users.update_user_scim_by_id(user_id, provider, user_data.externalId, db=db)
+ new_user = await Users.get_user_by_id(user_id, db=db)
- return user_to_scim(new_user, request, db=db)
+ return await user_to_scim(new_user, request, db=db)
@router.put('/Users/{user_id}', response_model=SCIMUser)
@@ -648,10 +649,10 @@ async def update_user(
request: Request,
user_data: SCIMUserUpdateRequest,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Update SCIM User (full update)"""
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
if not user:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -681,7 +682,7 @@ async def update_user(
if user_data.photos and len(user_data.photos) > 0:
update_data['profile_image_url'] = user_data.photos[0].value
- updated_user = Users.update_user_by_id(user_id, update_data, db=db)
+ updated_user = await Users.update_user_by_id(user_id, update_data, db=db)
if not updated_user:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -691,10 +692,10 @@ async def update_user(
# Update externalId in the scim field
if user_data.externalId:
provider = get_scim_provider()
- Users.update_user_scim_by_id(user_id, provider, user_data.externalId, db=db)
- updated_user = Users.get_user_by_id(user_id, db=db)
+ await Users.update_user_scim_by_id(user_id, provider, user_data.externalId, db=db)
+ updated_user = await Users.get_user_by_id(user_id, db=db)
- return user_to_scim(updated_user, request, db=db)
+ return await user_to_scim(updated_user, request, db=db)
@router.patch('/Users/{user_id}', response_model=SCIMUser)
@@ -703,10 +704,10 @@ async def patch_user(
request: Request,
patch_data: SCIMPatchRequest,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Update SCIM User (partial update)"""
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
if not user:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -733,11 +734,11 @@ async def patch_user(
update_data['name'] = value
elif path == 'externalId':
provider = get_scim_provider()
- Users.update_user_scim_by_id(user_id, provider, value, db=db)
+ await Users.update_user_scim_by_id(user_id, provider, value, db=db)
# Update user
if update_data:
- updated_user = Users.update_user_by_id(user_id, update_data, db=db)
+ updated_user = await Users.update_user_by_id(user_id, update_data, db=db)
if not updated_user:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -746,7 +747,7 @@ async def patch_user(
else:
updated_user = user
- return user_to_scim(updated_user, request, db=db)
+ return await user_to_scim(updated_user, request, db=db)
@router.delete('/Users/{user_id}', status_code=status.HTTP_204_NO_CONTENT)
@@ -754,17 +755,17 @@ async def delete_user(
user_id: str,
request: Request,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Delete SCIM User"""
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
if not user:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=f'User {user_id} not found',
)
- success = Users.delete_user_by_id(user_id, db=db)
+ success = await Users.delete_user_by_id(user_id, db=db)
if not success:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -782,7 +783,7 @@ async def get_groups(
count: int = Query(20),
filter: Optional[str] = None,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""List SCIM Groups"""
# Clamp per SCIM 2.0 spec (RFC 7644 §3.4.2.4):
@@ -794,13 +795,13 @@ async def get_groups(
if filter:
if 'displayName eq' in filter:
display_name = filter.split('"')[1]
- group = Groups.get_group_by_name(display_name, db=db)
+ group = await Groups.get_group_by_name(display_name, db=db)
groups_list = [group] if group else []
else:
# Unrecognized filter — fall back to all groups
- groups_list = Groups.get_all_groups(db=db)
+ groups_list = await Groups.get_all_groups(db=db)
else:
- groups_list = Groups.get_all_groups(db=db)
+ groups_list = await Groups.get_all_groups(db=db)
# Apply pagination
total = len(groups_list)
@@ -809,7 +810,7 @@ async def get_groups(
paginated_groups = groups_list[start:end]
# Convert to SCIM format
- scim_groups = [group_to_scim(group, request, db=db) for group in paginated_groups]
+ scim_groups = [await group_to_scim(group, request, db=db) for group in paginated_groups]
return SCIMListResponse(
totalResults=total,
@@ -824,17 +825,17 @@ async def get_group(
group_id: str,
request: Request,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Get SCIM Group by ID"""
- group = Groups.get_group_by_id(group_id, db=db)
+ group = await Groups.get_group_by_id(group_id, db=db)
if not group:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=f'Group {group_id} not found',
)
- return group_to_scim(group, request, db=db)
+ return await group_to_scim(group, request, db=db)
@router.post('/Groups', response_model=SCIMGroup, status_code=status.HTTP_201_CREATED)
@@ -842,7 +843,7 @@ async def create_group(
request: Request,
group_data: SCIMGroupCreateRequest,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Create SCIM Group"""
# Extract member IDs
@@ -860,14 +861,14 @@ async def create_group(
)
# Need to get the creating user's ID - we'll use the first admin
- admin_user = Users.get_super_admin_user(db=db)
+ admin_user = await Users.get_super_admin_user(db=db)
if not admin_user:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail='No admin user found',
)
- new_group = Groups.insert_new_group(admin_user.id, form, db=db)
+ new_group = await Groups.insert_new_group(admin_user.id, form, db=db)
if not new_group:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -883,12 +884,12 @@ async def create_group(
description=new_group.description,
)
- Groups.update_group_by_id(new_group.id, update_form, db=db)
- Groups.set_group_user_ids_by_id(new_group.id, member_ids, db=db)
+ await Groups.update_group_by_id(new_group.id, update_form, db=db)
+ await Groups.set_group_user_ids_by_id(new_group.id, member_ids, db=db)
- new_group = Groups.get_group_by_id(new_group.id, db=db)
+ new_group = await Groups.get_group_by_id(new_group.id, db=db)
- return group_to_scim(new_group, request, db=db)
+ return await group_to_scim(new_group, request, db=db)
@router.put('/Groups/{group_id}', response_model=SCIMGroup)
@@ -897,10 +898,10 @@ async def update_group(
request: Request,
group_data: SCIMGroupUpdateRequest,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Update SCIM Group (full update)"""
- group = Groups.get_group_by_id(group_id, db=db)
+ group = await Groups.get_group_by_id(group_id, db=db)
if not group:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -918,17 +919,17 @@ async def update_group(
# Handle members if provided
if group_data.members is not None:
member_ids = [member.value for member in group_data.members]
- Groups.set_group_user_ids_by_id(group_id, member_ids, db=db)
+ await Groups.set_group_user_ids_by_id(group_id, member_ids, db=db)
# Update group
- updated_group = Groups.update_group_by_id(group_id, update_form, db=db)
+ updated_group = await Groups.update_group_by_id(group_id, update_form, db=db)
if not updated_group:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail='Failed to update group',
)
- return group_to_scim(updated_group, request, db=db)
+ return await group_to_scim(updated_group, request, db=db)
@router.patch('/Groups/{group_id}', response_model=SCIMGroup)
@@ -937,10 +938,10 @@ async def patch_group(
request: Request,
patch_data: SCIMPatchRequest,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Update SCIM Group (partial update)"""
- group = Groups.get_group_by_id(group_id, db=db)
+ group = await Groups.get_group_by_id(group_id, db=db)
if not group:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -964,7 +965,7 @@ async def patch_group(
update_form.name = value
elif path == 'members':
# Replace all members
- Groups.set_group_user_ids_by_id(group_id, [member['value'] for member in value], db=db)
+ await Groups.set_group_user_ids_by_id(group_id, [member['value'] for member in value], db=db)
elif op == 'add':
if path == 'members':
@@ -972,22 +973,22 @@ async def patch_group(
if isinstance(value, list):
for member in value:
if isinstance(member, dict) and 'value' in member:
- Groups.add_users_to_group(group_id, [member['value']], db=db)
+ await Groups.add_users_to_group(group_id, [member['value']], db=db)
elif op == 'remove':
if path and path.startswith('members[value eq'):
# Remove specific member
member_id = path.split('"')[1]
- Groups.remove_users_from_group(group_id, [member_id], db=db)
+ await Groups.remove_users_from_group(group_id, [member_id], db=db)
# Update group
- updated_group = Groups.update_group_by_id(group_id, update_form, db=db)
+ updated_group = await Groups.update_group_by_id(group_id, update_form, db=db)
if not updated_group:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail='Failed to update group',
)
- return group_to_scim(updated_group, request, db=db)
+ return await group_to_scim(updated_group, request, db=db)
@router.delete('/Groups/{group_id}', status_code=status.HTTP_204_NO_CONTENT)
@@ -995,17 +996,17 @@ async def delete_group(
group_id: str,
request: Request,
_: bool = Depends(get_scim_auth),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
"""Delete SCIM Group"""
- group = Groups.get_group_by_id(group_id, db=db)
+ group = await Groups.get_group_by_id(group_id, db=db)
if not group:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=f'Group {group_id} not found',
)
- success = Groups.delete_group_by_id(group_id, db=db)
+ success = await Groups.delete_group_by_id(group_id, db=db)
if not success:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
diff --git a/backend/open_webui/routers/skills.py b/backend/open_webui/routers/skills.py
index 1838914e4a..490d1706d5 100644
--- a/backend/open_webui/routers/skills.py
+++ b/backend/open_webui/routers/skills.py
@@ -5,9 +5,9 @@ from open_webui.models.groups import Groups
from pydantic import BaseModel
from fastapi import APIRouter, Depends, HTTPException, Request, status
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
-from open_webui.internal.db import get_session
+from open_webui.internal.db import get_async_session
from open_webui.models.skills import (
SkillForm,
SkillModel,
@@ -40,18 +40,18 @@ router = APIRouter()
async def get_skills(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- skills = Skills.get_skills(db=db)
+ skills = await Skills.get_skills(db=db)
else:
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
- all_skills = Skills.get_skills(db=db)
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
+ all_skills = await Skills.get_skills(db=db)
skills = [
skill
for skill in all_skills
if skill.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -75,7 +75,7 @@ async def get_skill_list(
view_option: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
limit = PAGE_ITEM_COUNT
@@ -89,13 +89,13 @@ async def get_skill_list(
filter['view_option'] = view_option
if not (user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL):
- groups = Groups.get_groups_by_member_id(user.id, db=db)
+ groups = await Groups.get_groups_by_member_id(user.id, db=db)
if groups:
filter['group_ids'] = [group.id for group in groups]
filter['user_id'] = user.id
- result = Skills.search_skills(user.id, filter=filter, skip=skip, limit=limit, db=db)
+ result = await Skills.search_skills(user.id, filter=filter, skip=skip, limit=limit, db=db)
return SkillAccessListResponse(
items=[
@@ -104,7 +104,7 @@ async def get_skill_list(
write_access=(
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == skill.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -128,9 +128,9 @@ async def get_skill_list(
async def export_skills(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id,
'workspace.skills',
request.app.state.config.USER_PERMISSIONS,
@@ -142,9 +142,9 @@ async def export_skills(
)
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- return Skills.get_skills(db=db)
+ return await Skills.get_skills(db=db)
else:
- return Skills.get_skills_by_user_id(user.id, 'read', db=db)
+ return await Skills.get_skills_by_user_id(user.id, 'read', db=db)
############################
@@ -157,9 +157,9 @@ async def create_new_skill(
request: Request,
form_data: SkillForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id, 'workspace.skills', request.app.state.config.USER_PERMISSIONS, db=db
):
raise HTTPException(
@@ -169,7 +169,7 @@ async def create_new_skill(
form_data.id = form_data.id.lower().replace(' ', '-')
- existing = Skills.get_skill_by_id(form_data.id, db=db)
+ existing = await Skills.get_skill_by_id(form_data.id, db=db)
if existing is not None:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -177,7 +177,7 @@ async def create_new_skill(
)
try:
- skill = Skills.insert_new_skill(user.id, form_data, db=db)
+ skill = await Skills.insert_new_skill(user.id, form_data, db=db)
if skill:
return skill
else:
@@ -199,14 +199,14 @@ async def create_new_skill(
@router.get('/id/{id}', response_model=Optional[SkillAccessResponse])
-async def get_skill_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- skill = Skills.get_skill_by_id(id, db=db)
+async def get_skill_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ skill = await Skills.get_skill_by_id(id, db=db)
if skill:
if (
user.role == 'admin'
or skill.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -219,7 +219,7 @@ async def get_skill_by_id(id: str, user=Depends(get_verified_user), db: Session
write_access=(
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == skill.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -251,9 +251,9 @@ async def update_skill_by_id(
id: str,
form_data: SkillForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- skill = Skills.get_skill_by_id(id, db=db)
+ skill = await Skills.get_skill_by_id(id, db=db)
if not skill:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -262,7 +262,7 @@ async def update_skill_by_id(
if (
skill.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -281,7 +281,7 @@ async def update_skill_by_id(
**form_data.model_dump(exclude={'id'}),
}
- skill = Skills.update_skill_by_id(id, updated, db=db)
+ skill = await Skills.update_skill_by_id(id, updated, db=db)
if skill:
return skill
@@ -312,9 +312,9 @@ async def update_skill_access_by_id(
id: str,
form_data: SkillAccessGrantsForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- skill = Skills.get_skill_by_id(id, db=db)
+ skill = await Skills.get_skill_by_id(id, db=db)
if not skill:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -323,7 +323,7 @@ async def update_skill_access_by_id(
if (
skill.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -337,7 +337,7 @@ async def update_skill_access_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -345,9 +345,9 @@ async def update_skill_access_by_id(
'sharing.public_skills',
)
- AccessGrants.set_access_grants('skill', id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('skill', id, form_data.access_grants, db=db)
- return Skills.get_skill_by_id(id, db=db)
+ return await Skills.get_skill_by_id(id, db=db)
############################
@@ -356,13 +356,13 @@ async def update_skill_access_by_id(
@router.post('/id/{id}/toggle', response_model=Optional[SkillModel])
-async def toggle_skill_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- skill = Skills.get_skill_by_id(id, db=db)
+async def toggle_skill_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ skill = await Skills.get_skill_by_id(id, db=db)
if skill:
if (
user.role == 'admin'
or skill.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -370,7 +370,7 @@ async def toggle_skill_by_id(id: str, user=Depends(get_verified_user), db: Sessi
db=db,
)
):
- skill = Skills.toggle_skill_by_id(id, db=db)
+ skill = await Skills.toggle_skill_by_id(id, db=db)
if skill:
return skill
@@ -401,9 +401,9 @@ async def delete_skill_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- skill = Skills.get_skill_by_id(id, db=db)
+ skill = await Skills.get_skill_by_id(id, db=db)
if not skill:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -412,7 +412,7 @@ async def delete_skill_by_id(
if (
skill.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='skill',
resource_id=skill.id,
@@ -426,5 +426,5 @@ async def delete_skill_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- result = Skills.delete_skill_by_id(id, db=db)
+ result = await Skills.delete_skill_by_id(id, db=db)
return result
diff --git a/backend/open_webui/routers/tasks.py b/backend/open_webui/routers/tasks.py
index 0bb5813e6f..b921f7b3e6 100644
--- a/backend/open_webui/routers/tasks.py
+++ b/backend/open_webui/routers/tasks.py
@@ -18,7 +18,7 @@ from open_webui.utils.task import (
moa_response_generation_template,
)
from open_webui.utils.auth import get_admin_user, get_verified_user
-from open_webui.constants import TASKS
+from open_webui.constants import ERROR_MESSAGES, TASKS
from open_webui.routers.pipelines import process_pipeline_inlet_filter
@@ -168,7 +168,7 @@ async def generate_title(request: Request, form_data: dict, user=Depends(get_ver
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -245,7 +245,7 @@ async def generate_follow_ups(request: Request, form_data: dict, user=Depends(ge
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -313,7 +313,7 @@ async def generate_chat_tags(request: Request, form_data: dict, user=Depends(get
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -375,7 +375,7 @@ async def generate_image_prompt(request: Request, form_data: dict, user=Depends(
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -431,13 +431,13 @@ async def generate_queries(request: Request, form_data: dict, user=Depends(get_v
if not request.app.state.config.ENABLE_SEARCH_QUERY_GENERATION:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail=f'Search query generation is disabled',
+ detail=ERROR_MESSAGES.FEATURE_DISABLED('Search query generation'),
)
elif type == 'retrieval':
if not request.app.state.config.ENABLE_RETRIEVAL_QUERY_GENERATION:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail=f'Query generation is disabled',
+ detail=ERROR_MESSAGES.FEATURE_DISABLED('Query generation'),
)
if getattr(request.state, 'cached_queries', None):
@@ -455,7 +455,7 @@ async def generate_queries(request: Request, form_data: dict, user=Depends(get_v
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -508,7 +508,7 @@ async def generate_autocompletion(request: Request, form_data: dict, user=Depend
if not request.app.state.config.ENABLE_AUTOCOMPLETE_GENERATION:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail=f'Autocompletion generation is disabled',
+ detail=ERROR_MESSAGES.FEATURE_DISABLED('Autocompletion generation'),
)
type = form_data.get('type')
@@ -519,7 +519,7 @@ async def generate_autocompletion(request: Request, form_data: dict, user=Depend
if len(prompt) > request.app.state.config.AUTOCOMPLETE_GENERATION_INPUT_MAX_LENGTH:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
- detail=f'Input prompt exceeds maximum length of {request.app.state.config.AUTOCOMPLETE_GENERATION_INPUT_MAX_LENGTH}',
+ detail=ERROR_MESSAGES.INPUT_TOO_LONG(request.app.state.config.AUTOCOMPLETE_GENERATION_INPUT_MAX_LENGTH),
)
if getattr(request.state, 'direct', False) and hasattr(request.state, 'model'):
@@ -533,7 +533,7 @@ async def generate_autocompletion(request: Request, form_data: dict, user=Depend
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -595,7 +595,7 @@ async def generate_emoji(request: Request, form_data: dict, user=Depends(get_ver
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
# Check if the user has a custom task model
@@ -661,7 +661,7 @@ async def generate_moa_response(request: Request, form_data: dict, user=Depends(
if model_id not in models:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
- detail='Model not found',
+ detail=ERROR_MESSAGES.MODEL_NOT_FOUND(),
)
template = DEFAULT_MOA_GENERATION_PROMPT_TEMPLATE
diff --git a/backend/open_webui/routers/terminals.py b/backend/open_webui/routers/terminals.py
index 34d5eb96d6..0d607d1f78 100644
--- a/backend/open_webui/routers/terminals.py
+++ b/backend/open_webui/routers/terminals.py
@@ -52,7 +52,7 @@ def _sanitize_proxy_path(path: str) -> str | None:
async def list_terminal_servers(request: Request, user=Depends(get_verified_user)):
"""Return terminal servers the authenticated user has access to."""
connections = request.app.state.config.TERMINAL_SERVER_CONNECTIONS or []
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
return [
{
@@ -61,7 +61,7 @@ async def list_terminal_servers(request: Request, user=Depends(get_verified_user
'name': connection.get('name', ''),
}
for connection in connections
- if connection.get('enabled', True) and has_connection_access(user, connection, user_group_ids)
+ if connection.get('enabled', True) and await has_connection_access(user, connection, user_group_ids)
]
@@ -82,8 +82,8 @@ async def proxy_terminal(
if connection is None:
return JSONResponse({'error': f"Terminal server '{server_id}' not found"}, status_code=404)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not has_connection_access(user, connection, user_group_ids):
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
+ if not await has_connection_access(user, connection, user_group_ids):
return JSONResponse({'error': 'Access denied'}, status_code=403)
base_url = (connection.get('url') or '').rstrip('/')
@@ -208,7 +208,7 @@ async def _resolve_authenticated_connection(ws: WebSocket, server_id: str):
if data is None or 'id' not in data:
await ws.close(code=4001, reason='Invalid token')
return None
- user = Users.get_user_by_id(data['id'])
+ user = await Users.get_user_by_id(data['id'])
if user is None:
await ws.close(code=4001, reason='User not found')
return None
@@ -227,8 +227,8 @@ async def _resolve_authenticated_connection(ws: WebSocket, server_id: str):
await ws.close(code=4004, reason='Terminal server not found')
return None
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not has_connection_access(user, connection, user_group_ids):
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
+ if not await has_connection_access(user, connection, user_group_ids):
await ws.close(code=4003, reason='Access denied')
return None
diff --git a/backend/open_webui/routers/tools.py b/backend/open_webui/routers/tools.py
index 195a4eec3e..d70b4038fe 100644
--- a/backend/open_webui/routers/tools.py
+++ b/backend/open_webui/routers/tools.py
@@ -8,8 +8,8 @@ from open_webui.env import AIOHTTP_CLIENT_TIMEOUT
from open_webui.models.groups import Groups
from pydantic import BaseModel, HttpUrl
from fastapi import APIRouter, Depends, HTTPException, Request, status
-from sqlalchemy.orm import Session
-from open_webui.internal.db import get_session
+from sqlalchemy.ext.asyncio import AsyncSession
+from open_webui.internal.db import get_async_session
from open_webui.models.oauth_sessions import OAuthSessions
@@ -46,11 +46,11 @@ log = logging.getLogger(__name__)
router = APIRouter()
-def get_tool_module(request, tool_id, load_from_db=True):
+async def get_tool_module(request, tool_id, load_from_db=True):
"""
Get the tool module by its ID.
"""
- tool_module, _ = get_tool_module_from_cache(request, tool_id, load_from_db)
+ tool_module, _ = await get_tool_module_from_cache(request, tool_id, load_from_db)
return tool_module
@@ -65,12 +65,12 @@ def get_tool_module(request, tool_id, load_from_db=True):
async def get_tools(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
tools = []
# Local Tools
- for tool in Tools.get_tools(defer_content=True, db=db):
+ for tool in await Tools.get_tools(defer_content=True, db=db):
tool_module = request.app.state.TOOLS.get(tool.id) if hasattr(request.app.state, 'TOOLS') else None
tools.append(
ToolUserResponse(
@@ -159,31 +159,30 @@ async def get_tools(
# Admin can see all tools
return tools
else:
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
- tools = [
- tool
- for tool in tools
- if tool.user_id == user.id
- or (
- has_access(
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
+ filtered_tools = []
+ for tool in tools:
+ if tool.user_id == user.id:
+ filtered_tools.append(tool)
+ elif str(tool.id).startswith('server:'):
+ if await has_access(
user.id,
'read',
server_access_grants.get(str(tool.id), []),
user_group_ids,
db=db,
- )
- if str(tool.id).startswith('server:')
- else AccessGrants.has_access(
- user_id=user.id,
- resource_type='tool',
- resource_id=tool.id,
- permission='read',
- user_group_ids=user_group_ids,
- db=db,
- )
- )
- ]
- return tools
+ ):
+ filtered_tools.append(tool)
+ elif await AccessGrants.has_access(
+ user_id=user.id,
+ resource_type='tool',
+ resource_id=tool.id,
+ permission='read',
+ user_group_ids=user_group_ids,
+ db=db,
+ ):
+ filtered_tools.append(tool)
+ return filtered_tools
############################
@@ -192,13 +191,13 @@ async def get_tools(
@router.get('/list', response_model=list[ToolAccessResponse])
-async def get_tool_list(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_tool_list(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- tools = Tools.get_tools(defer_content=True, db=db)
+ tools = await Tools.get_tools(defer_content=True, db=db)
else:
- tools = Tools.get_tools_by_user_id(user.id, 'read', defer_content=True, db=db)
+ tools = await Tools.get_tools_by_user_id(user.id, 'read', defer_content=True, db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
result = []
for tool in tools:
@@ -286,7 +285,7 @@ async def load_tool_from_url(request: Request, form_data: LoadUrlForm, user=Depe
'content': data,
}
except Exception as e:
- raise HTTPException(status_code=500, detail=f'Error importing tool: {e}')
+ raise HTTPException(status_code=500, detail=ERROR_MESSAGES.DEFAULT(e))
############################
@@ -298,9 +297,9 @@ async def load_tool_from_url(request: Request, form_data: LoadUrlForm, user=Depe
async def export_tools(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- if user.role != 'admin' and not has_permission(
+ if user.role != 'admin' and not await has_permission(
user.id,
'workspace.tools_export',
request.app.state.config.USER_PERMISSIONS,
@@ -312,9 +311,9 @@ async def export_tools(
)
if user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL:
- return Tools.get_tools(db=db)
+ return await Tools.get_tools(db=db)
else:
- return Tools.get_tools_by_user_id(user.id, 'read', db=db)
+ return await Tools.get_tools_by_user_id(user.id, 'read', db=db)
############################
@@ -327,11 +326,11 @@ async def create_new_tools(
request: Request,
form_data: ToolForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if user.role != 'admin' and not (
- has_permission(user.id, 'workspace.tools', request.app.state.config.USER_PERMISSIONS, db=db)
- or has_permission(
+ await has_permission(user.id, 'workspace.tools', request.app.state.config.USER_PERMISSIONS, db=db)
+ or await has_permission(
user.id,
'workspace.tools_import',
request.app.state.config.USER_PERMISSIONS,
@@ -351,10 +350,10 @@ async def create_new_tools(
form_data.id = form_data.id.lower()
- tools = Tools.get_tool_by_id(form_data.id, db=db)
+ tools = await Tools.get_tool_by_id(form_data.id, db=db)
if tools is None:
try:
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -363,14 +362,14 @@ async def create_new_tools(
)
form_data.content = replace_imports(form_data.content)
- tool_module, frontmatter = load_tool_module_by_id(form_data.id, content=form_data.content)
+ tool_module, frontmatter = await load_tool_module_by_id(form_data.id, content=form_data.content)
form_data.meta.manifest = frontmatter
TOOLS = request.app.state.TOOLS
TOOLS[form_data.id] = tool_module
specs = get_tool_specs(TOOLS[form_data.id])
- tools = Tools.insert_new_tool(user.id, form_data, specs, db=db)
+ tools = await Tools.insert_new_tool(user.id, form_data, specs, db=db)
tool_cache_dir = CACHE_DIR / 'tools' / form_data.id
tool_cache_dir.mkdir(parents=True, exist_ok=True)
@@ -401,14 +400,14 @@ async def create_new_tools(
@router.get('/id/{id}', response_model=Optional[ToolAccessResponse])
-async def get_tools_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- tools = Tools.get_tool_by_id(id, db=db)
+async def get_tools_by_id(id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ tools = await Tools.get_tool_by_id(id, db=db)
if tools:
if (
user.role == 'admin'
or tools.user_id == user.id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -421,7 +420,7 @@ async def get_tools_by_id(id: str, user=Depends(get_verified_user), db: Session
write_access=(
(user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == tools.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -453,9 +452,9 @@ async def update_tools_by_id(
id: str,
form_data: ToolForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -465,7 +464,7 @@ async def update_tools_by_id(
# Is the user the original creator, in a group with write access, or an admin
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -481,7 +480,7 @@ async def update_tools_by_id(
try:
form_data.content = replace_imports(form_data.content)
- tool_module, frontmatter = load_tool_module_by_id(id, content=form_data.content)
+ tool_module, frontmatter = await load_tool_module_by_id(id, content=form_data.content)
form_data.meta.manifest = frontmatter
TOOLS = request.app.state.TOOLS
@@ -489,7 +488,7 @@ async def update_tools_by_id(
specs = get_tool_specs(TOOLS[id])
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -503,7 +502,7 @@ async def update_tools_by_id(
}
log.debug(updated)
- tools = Tools.update_tool_by_id(id, updated, db=db)
+ tools = await Tools.update_tool_by_id(id, updated, db=db)
if tools:
return tools
@@ -535,9 +534,9 @@ async def update_tool_access_by_id(
id: str,
form_data: ToolAccessGrantsForm,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -546,7 +545,7 @@ async def update_tool_access_by_id(
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -560,7 +559,7 @@ async def update_tool_access_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- form_data.access_grants = filter_allowed_access_grants(
+ form_data.access_grants = await filter_allowed_access_grants(
request.app.state.config.USER_PERMISSIONS,
user.id,
user.role,
@@ -568,9 +567,9 @@ async def update_tool_access_by_id(
'sharing.public_tools',
)
- AccessGrants.set_access_grants('tool', id, form_data.access_grants, db=db)
+ await AccessGrants.set_access_grants('tool', id, form_data.access_grants, db=db)
- return Tools.get_tool_by_id(id, db=db)
+ return await Tools.get_tool_by_id(id, db=db)
############################
@@ -583,9 +582,9 @@ async def delete_tools_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -594,7 +593,7 @@ async def delete_tools_by_id(
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -608,7 +607,7 @@ async def delete_tools_by_id(
detail=ERROR_MESSAGES.UNAUTHORIZED,
)
- result = Tools.delete_tool_by_id(id, db=db)
+ result = await Tools.delete_tool_by_id(id, db=db)
if result:
TOOLS = request.app.state.TOOLS
if id in TOOLS:
@@ -623,8 +622,10 @@ async def delete_tools_by_id(
@router.get('/id/{id}/valves', response_model=Optional[dict])
-async def get_tools_valves_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- tools = Tools.get_tool_by_id(id, db=db)
+async def get_tools_valves_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -633,7 +634,7 @@ async def get_tools_valves_by_id(id: str, user=Depends(get_verified_user), db: S
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -648,7 +649,7 @@ async def get_tools_valves_by_id(id: str, user=Depends(get_verified_user), db: S
)
try:
- valves = Tools.get_tool_valves_by_id(id, db=db)
+ valves = await Tools.get_tool_valves_by_id(id, db=db)
return valves
except Exception as e:
raise HTTPException(
@@ -667,9 +668,9 @@ async def get_tools_valves_spec_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -678,7 +679,7 @@ async def get_tools_valves_spec_by_id(
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -695,7 +696,7 @@ async def get_tools_valves_spec_by_id(
if id in request.app.state.TOOLS:
tools_module = request.app.state.TOOLS[id]
else:
- tools_module, _ = load_tool_module_by_id(id)
+ tools_module, _ = await load_tool_module_by_id(id)
request.app.state.TOOLS[id] = tools_module
if hasattr(tools_module, 'Valves'):
@@ -718,9 +719,9 @@ async def update_tools_valves_by_id(
id: str,
form_data: dict,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -729,7 +730,7 @@ async def update_tools_valves_by_id(
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -746,7 +747,7 @@ async def update_tools_valves_by_id(
if id in request.app.state.TOOLS:
tools_module = request.app.state.TOOLS[id]
else:
- tools_module, _ = load_tool_module_by_id(id)
+ tools_module, _ = await load_tool_module_by_id(id)
request.app.state.TOOLS[id] = tools_module
if not hasattr(tools_module, 'Valves'):
@@ -760,7 +761,7 @@ async def update_tools_valves_by_id(
form_data = {k: v for k, v in form_data.items() if v is not None}
valves = Valves(**form_data)
valves_dict = valves.model_dump(exclude_unset=True)
- Tools.update_tool_valves_by_id(id, valves_dict, db=db)
+ await Tools.update_tool_valves_by_id(id, valves_dict, db=db)
return valves_dict
except Exception as e:
log.exception(f'Failed to update tool valves by id {id}: {e}')
@@ -776,8 +777,10 @@ async def update_tools_valves_by_id(
@router.get('/id/{id}/valves/user', response_model=Optional[dict])
-async def get_tools_user_valves_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- tools = Tools.get_tool_by_id(id, db=db)
+async def get_tools_user_valves_by_id(
+ id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -786,7 +789,7 @@ async def get_tools_user_valves_by_id(id: str, user=Depends(get_verified_user),
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -801,7 +804,7 @@ async def get_tools_user_valves_by_id(id: str, user=Depends(get_verified_user),
)
try:
- user_valves = Tools.get_user_valves_by_id_and_user_id(id, user.id, db=db)
+ user_valves = await Tools.get_user_valves_by_id_and_user_id(id, user.id, db=db)
return user_valves
except Exception as e:
raise HTTPException(
@@ -815,9 +818,9 @@ async def get_tools_user_valves_spec_by_id(
request: Request,
id: str,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -826,7 +829,7 @@ async def get_tools_user_valves_spec_by_id(
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -843,7 +846,7 @@ async def get_tools_user_valves_spec_by_id(
if id in request.app.state.TOOLS:
tools_module = request.app.state.TOOLS[id]
else:
- tools_module, _ = load_tool_module_by_id(id)
+ tools_module, _ = await load_tool_module_by_id(id)
request.app.state.TOOLS[id] = tools_module
if hasattr(tools_module, 'UserValves'):
@@ -861,9 +864,9 @@ async def update_tools_user_valves_by_id(
id: str,
form_data: dict,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- tools = Tools.get_tool_by_id(id, db=db)
+ tools = await Tools.get_tool_by_id(id, db=db)
if not tools:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
@@ -872,7 +875,7 @@ async def update_tools_user_valves_by_id(
if (
tools.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tools.id,
@@ -889,7 +892,7 @@ async def update_tools_user_valves_by_id(
if id in request.app.state.TOOLS:
tools_module = request.app.state.TOOLS[id]
else:
- tools_module, _ = load_tool_module_by_id(id)
+ tools_module, _ = await load_tool_module_by_id(id)
request.app.state.TOOLS[id] = tools_module
if hasattr(tools_module, 'UserValves'):
@@ -899,7 +902,7 @@ async def update_tools_user_valves_by_id(
form_data = {k: v for k, v in form_data.items() if v is not None}
user_valves = UserValves(**form_data)
user_valves_dict = user_valves.model_dump(exclude_unset=True)
- Tools.update_user_valves_by_id_and_user_id(id, user.id, user_valves_dict, db=db)
+ await Tools.update_user_valves_by_id_and_user_id(id, user.id, user_valves_dict, db=db)
return user_valves_dict
except Exception as e:
log.exception(f'Failed to update user valves by id {id}: {e}')
diff --git a/backend/open_webui/routers/users.py b/backend/open_webui/routers/users.py
index 0ccc20185e..9fd2479ada 100644
--- a/backend/open_webui/routers/users.py
+++ b/backend/open_webui/routers/users.py
@@ -1,6 +1,6 @@
import logging
from typing import Optional
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
import base64
import io
@@ -30,7 +30,7 @@ from open_webui.models.users import (
from open_webui.constants import ERROR_MESSAGES
from open_webui.env import STATIC_DIR
-from open_webui.internal.db import get_session
+from open_webui.internal.db import get_async_session
from open_webui.utils.auth import (
@@ -40,6 +40,7 @@ from open_webui.utils.auth import (
validate_password,
)
from open_webui.utils.access_control import get_permissions, has_permission
+from open_webui.socket.main import disconnect_user_sessions
log = logging.getLogger(__name__)
@@ -63,7 +64,7 @@ async def get_users(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
limit = PAGE_ITEM_COUNT
@@ -80,14 +81,14 @@ async def get_users(
filter['direction'] = direction
- result = Users.get_users(filter=filter, skip=skip, limit=limit, db=db)
+ result = await Users.get_users(filter=filter, skip=skip, limit=limit, db=db)
users = result['users']
total = result['total']
# Fetch groups for all users in a single query to avoid N+1
user_ids = [user.id for user in users]
- user_groups = Groups.get_groups_by_member_ids(user_ids, db=db)
+ user_groups = await Groups.get_groups_by_member_ids(user_ids, db=db)
return {
'users': [
@@ -106,9 +107,9 @@ async def get_users(
@router.get('/all', response_model=UserInfoListResponse)
async def get_all_users(
user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- return Users.get_users(db=db)
+ return await Users.get_users(db=db)
@router.get('/search', response_model=UserInfoListResponse)
@@ -118,7 +119,7 @@ async def search_users(
direction: Optional[str] = None,
page: Optional[int] = 1,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
limit = PAGE_ITEM_COUNT
@@ -133,7 +134,7 @@ async def search_users(
if direction:
filter['direction'] = direction
- return Users.get_users(filter=filter, skip=skip, limit=limit, db=db)
+ return await Users.get_users(filter=filter, skip=skip, limit=limit, db=db)
############################
@@ -142,8 +143,8 @@ async def search_users(
@router.get('/groups')
-async def get_user_groups(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- return Groups.get_groups_by_member_id(user.id, db=db)
+async def get_user_groups(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ return await Groups.get_groups_by_member_id(user.id, db=db)
############################
@@ -155,9 +156,9 @@ async def get_user_groups(user=Depends(get_verified_user), db: Session = Depends
async def get_user_permissisions(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
- user_permissions = get_permissions(user.id, request.app.state.config.USER_PERMISSIONS, db=db)
+ user_permissions = await get_permissions(user.id, request.app.state.config.USER_PERMISSIONS, db=db)
return user_permissions
@@ -272,8 +273,10 @@ async def update_default_user_permissions(request: Request, form_data: UserPermi
@router.get('/user/settings', response_model=Optional[UserSettings])
-async def get_user_settings_by_session_user(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- user = Users.get_user_by_id(user.id, db=db)
+async def get_user_settings_by_session_user(
+ user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ user = await Users.get_user_by_id(user.id, db=db)
if user:
return user.settings
else:
@@ -293,7 +296,7 @@ async def update_user_settings_by_session_user(
request: Request,
form_data: UserSettings,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
updated_user_settings = form_data.model_dump()
ui_settings = updated_user_settings.get('ui')
@@ -301,7 +304,7 @@ async def update_user_settings_by_session_user(
user.role != 'admin'
and ui_settings is not None
and 'toolServers' in ui_settings.keys()
- and not has_permission(
+ and not await has_permission(
user.id,
'features.direct_tool_servers',
request.app.state.config.USER_PERMISSIONS,
@@ -310,7 +313,7 @@ async def update_user_settings_by_session_user(
# If the user is not an admin and does not have permission to use tool servers, remove the key
updated_user_settings['ui'].pop('toolServers', None)
- user = Users.update_user_settings_by_id(user.id, updated_user_settings, db=db)
+ user = await Users.update_user_settings_by_id(user.id, updated_user_settings, db=db)
if user:
return user.settings
else:
@@ -329,14 +332,14 @@ async def update_user_settings_by_session_user(
async def get_user_status_by_session_user(
request: Request,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not request.app.state.config.ENABLE_USER_STATUS:
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACTION_PROHIBITED,
)
- user = Users.get_user_by_id(user.id, db=db)
+ user = await Users.get_user_by_id(user.id, db=db)
if user:
return user
else:
@@ -356,16 +359,16 @@ async def update_user_status_by_session_user(
request: Request,
form_data: UserStatus,
user=Depends(get_verified_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
if not request.app.state.config.ENABLE_USER_STATUS:
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACTION_PROHIBITED,
)
- user = Users.get_user_by_id(user.id, db=db)
+ user = await Users.get_user_by_id(user.id, db=db)
if user:
- user = Users.update_user_status_by_id(user.id, form_data, db=db)
+ user = await Users.update_user_status_by_id(user.id, form_data, db=db)
return user
else:
raise HTTPException(
@@ -380,8 +383,8 @@ async def update_user_status_by_session_user(
@router.get('/user/info', response_model=Optional[dict])
-async def get_user_info_by_session_user(user=Depends(get_verified_user), db: Session = Depends(get_session)):
- user = Users.get_user_by_id(user.id, db=db)
+async def get_user_info_by_session_user(user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)):
+ user = await Users.get_user_by_id(user.id, db=db)
if user:
return user.info
else:
@@ -398,14 +401,14 @@ async def get_user_info_by_session_user(user=Depends(get_verified_user), db: Ses
@router.post('/user/info/update', response_model=Optional[dict])
async def update_user_info_by_session_user(
- form_data: dict, user=Depends(get_verified_user), db: Session = Depends(get_session)
+ form_data: dict, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
):
- user = Users.get_user_by_id(user.id, db=db)
+ user = await Users.get_user_by_id(user.id, db=db)
if user:
if user.info is None:
user.info = {}
- user = Users.update_user_by_id(user.id, {'info': {**user.info, **form_data}}, db=db)
+ user = await Users.update_user_by_id(user.id, {'info': {**user.info, **form_data}}, db=db)
if user:
return user.info
else:
@@ -435,12 +438,12 @@ class UserActiveResponse(UserStatus):
@router.get('/{user_id}', response_model=UserActiveResponse)
-async def get_user_by_id(user_id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
+async def get_user_by_id(user_id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
# Check if user_id is a shared chat
# If it is, get the user_id from the chat
if user_id.startswith('shared-'):
chat_id = user_id.replace('shared-', '')
- chat = Chats.get_chat_by_id(chat_id)
+ chat = await Chats.get_chat_by_id(chat_id)
if chat:
user_id = chat.user_id
else:
@@ -449,14 +452,14 @@ async def get_user_by_id(user_id: str, user=Depends(get_admin_user), db: Session
detail=ERROR_MESSAGES.USER_NOT_FOUND,
)
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
if user:
- groups = Groups.get_groups_by_member_id(user_id, db=db)
+ groups = await Groups.get_groups_by_member_id(user_id, db=db)
return UserActiveResponse(
**{
**user.model_dump(),
'groups': [{'id': group.id, 'name': group.name} for group in groups],
- 'is_active': Users.is_user_active(user_id, db=db),
+ 'is_active': await Users.is_user_active(user_id, db=db),
}
)
else:
@@ -467,15 +470,17 @@ async def get_user_by_id(user_id: str, user=Depends(get_admin_user), db: Session
@router.get('/{user_id}/info', response_model=UserInfoResponse)
-async def get_user_info_by_id(user_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
- user = Users.get_user_by_id(user_id, db=db)
+async def get_user_info_by_id(
+ user_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
+):
+ user = await Users.get_user_by_id(user_id, db=db)
if user:
- groups = Groups.get_groups_by_member_id(user_id, db=db)
+ groups = await Groups.get_groups_by_member_id(user_id, db=db)
return UserInfoResponse(
**{
**user.model_dump(),
'groups': [{'id': group.id, 'name': group.name} for group in groups],
- 'is_active': Users.is_user_active(user_id, db=db),
+ 'is_active': await Users.is_user_active(user_id, db=db),
}
)
else:
@@ -486,8 +491,10 @@ async def get_user_info_by_id(user_id: str, user=Depends(get_verified_user), db:
@router.get('/{user_id}/oauth/sessions')
-async def get_user_oauth_sessions_by_id(user_id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- sessions = OAuthSessions.get_sessions_by_user_id(user_id, db=db)
+async def get_user_oauth_sessions_by_id(
+ user_id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)
+):
+ sessions = await OAuthSessions.get_sessions_by_user_id(user_id, db=db)
if sessions and len(sessions) > 0:
return sessions
else:
@@ -503,8 +510,8 @@ async def get_user_oauth_sessions_by_id(user_id: str, user=Depends(get_admin_use
@router.get('/{user_id}/profile/image')
-def get_user_profile_image_by_id(user_id: str, user=Depends(get_verified_user)):
- user = Users.get_user_by_id(user_id)
+async def get_user_profile_image_by_id(user_id: str, user=Depends(get_verified_user)):
+ user = await Users.get_user_by_id(user_id)
if user:
if user.profile_image_url:
# check if it's url or base64
@@ -542,10 +549,10 @@ def get_user_profile_image_by_id(user_id: str, user=Depends(get_verified_user)):
@router.get('/{user_id}/active', response_model=dict)
async def get_user_active_status_by_id(
- user_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)
+ user_id: str, user=Depends(get_verified_user), db: AsyncSession = Depends(get_async_session)
):
return {
- 'active': Users.is_user_active(user_id, db=db),
+ 'active': await Users.is_user_active(user_id, db=db),
}
@@ -559,11 +566,11 @@ async def update_user_by_id(
user_id: str,
form_data: UserUpdateForm,
session_user=Depends(get_admin_user),
- db: Session = Depends(get_session),
+ db: AsyncSession = Depends(get_async_session),
):
# Prevent modification of the primary admin user by other admins
try:
- first_user = Users.get_first_user(db=db)
+ first_user = await Users.get_first_user(db=db)
if first_user:
if user_id == first_user.id:
if session_user.id != user_id:
@@ -573,7 +580,7 @@ async def update_user_by_id(
detail=ERROR_MESSAGES.ACTION_PROHIBITED,
)
- if form_data.role != 'admin':
+ if form_data.role is not None and form_data.role != 'admin':
# If the primary admin is trying to change their own role, prevent it
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
@@ -587,11 +594,11 @@ async def update_user_by_id(
detail='Could not verify primary admin status.',
)
- user = Users.get_user_by_id(user_id, db=db)
+ user = await Users.get_user_by_id(user_id, db=db)
if user:
- if form_data.email.lower() != user.email:
- email_user = Users.get_user_by_email(form_data.email.lower(), db=db)
+ if form_data.email is not None and form_data.email.lower() != user.email:
+ email_user = await Users.get_user_by_email(form_data.email.lower(), db=db)
if email_user:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -605,21 +612,34 @@ async def update_user_by_id(
raise HTTPException(400, detail=str(e))
hashed = get_password_hash(form_data.password)
- Auths.update_user_password_by_id(user_id, hashed, db=db)
+ await Auths.update_user_password_by_id(user_id, hashed, db=db)
- Auths.update_email_by_id(user_id, form_data.email.lower(), db=db)
- updated_user = Users.update_user_by_id(
- user_id,
- {
- 'role': form_data.role,
- 'name': form_data.name,
- 'email': form_data.email.lower(),
- 'profile_image_url': form_data.profile_image_url,
- },
- db=db,
- )
+ # Build update dict from only the provided fields
+ update_data = {}
+ if form_data.role is not None:
+ update_data['role'] = form_data.role
+ if form_data.name is not None:
+ update_data['name'] = form_data.name
+ if form_data.email is not None:
+ update_data['email'] = form_data.email.lower()
+ await Auths.update_email_by_id(user_id, form_data.email.lower(), db=db)
+ if form_data.profile_image_url is not None:
+ update_data['profile_image_url'] = form_data.profile_image_url
+
+ if update_data:
+ updated_user = await Users.update_user_by_id(
+ user_id,
+ update_data,
+ db=db,
+ )
+ else:
+ updated_user = user
if updated_user:
+ # If the role changed, disconnect all socket sessions so stale
+ # privileges cached in SESSION_POOL are invalidated.
+ if updated_user.role != user.role:
+ await disconnect_user_sessions(user_id)
return updated_user
raise HTTPException(
@@ -639,10 +659,10 @@ async def update_user_by_id(
@router.delete('/{user_id}', response_model=bool)
-async def delete_user_by_id(user_id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
+async def delete_user_by_id(user_id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)):
# Prevent deletion of the primary admin user
try:
- first_user = Users.get_first_user(db=db)
+ first_user = await Users.get_first_user(db=db)
if first_user and user_id == first_user.id:
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
@@ -656,9 +676,10 @@ async def delete_user_by_id(user_id: str, user=Depends(get_admin_user), db: Sess
)
if user.id != user_id:
- result = Auths.delete_auth_by_id(user_id, db=db)
+ result = await Auths.delete_auth_by_id(user_id, db=db)
if result:
+ await disconnect_user_sessions(user_id)
return True
raise HTTPException(
@@ -679,5 +700,7 @@ async def delete_user_by_id(user_id: str, user=Depends(get_admin_user), db: Sess
@router.get('/{user_id}/groups')
-async def get_user_groups_by_id(user_id: str, user=Depends(get_admin_user), db: Session = Depends(get_session)):
- return Groups.get_groups_by_member_id(user_id, db=db)
+async def get_user_groups_by_id(
+ user_id: str, user=Depends(get_admin_user), db: AsyncSession = Depends(get_async_session)
+):
+ return await Groups.get_groups_by_member_id(user_id, db=db)
diff --git a/backend/open_webui/routers/utils.py b/backend/open_webui/routers/utils.py
index c79d8fe5d8..20705c2c44 100644
--- a/backend/open_webui/routers/utils.py
+++ b/backend/open_webui/routers/utils.py
@@ -45,7 +45,7 @@ async def execute_code(request: Request, form_data: CodeForm, user=Depends(get_v
if not request.app.state.config.ENABLE_CODE_EXECUTION:
raise HTTPException(
status_code=403,
- detail='Code execution is disabled',
+ detail=ERROR_MESSAGES.FEATURE_DISABLED('Code execution'),
)
if request.app.state.config.CODE_EXECUTION_ENGINE == 'jupyter':
@@ -69,7 +69,7 @@ async def execute_code(request: Request, form_data: CodeForm, user=Depends(get_v
else:
raise HTTPException(
status_code=400,
- detail='Code execution engine not supported',
+ detail=ERROR_MESSAGES.DEFAULT('Code execution engine not supported'),
)
diff --git a/backend/open_webui/socket/main.py b/backend/open_webui/socket/main.py
index 80e8b5be1c..2c44eb25c5 100644
--- a/backend/open_webui/socket/main.py
+++ b/backend/open_webui/socket/main.py
@@ -312,6 +312,24 @@ async def enter_room_for_users(room: str, user_ids: list[str]):
log.debug(f'Failed to make users {user_ids} join room {room}: {e}')
+async def disconnect_user_sessions(user_id: str):
+ """Disconnect all Socket.IO sessions belonging to a user.
+
+ Call this when a user's role is changed or the user is deleted so that
+ stale role/permission data cached in SESSION_POOL is invalidated.
+ The client will automatically reconnect and re-authenticate with
+ fresh data from the database.
+ """
+ try:
+ session_ids = get_session_ids_from_room(f'user:{user_id}')
+ for sid in session_ids:
+ await sio.disconnect(sid)
+ if session_ids:
+ log.info(f'Disconnected {len(session_ids)} session(s) for user {user_id}')
+ except Exception as e:
+ log.warning(f'Failed to disconnect sessions for user {user_id}: {e}')
+
+
@sio.on('usage')
async def usage(sid, data):
if sid in SESSION_POOL:
@@ -333,7 +351,7 @@ async def connect(sid, environ, auth):
data = decode_token(auth['token'])
if data is not None and 'id' in data:
- user = Users.get_user_by_id(data['id'])
+ user = await Users.get_user_by_id(data['id'])
if user:
SESSION_POOL[sid] = {
@@ -361,7 +379,7 @@ async def user_join(sid, data):
if data is None or 'id' not in data:
return
- user = Users.get_user_by_id(data['id'])
+ user = await Users.get_user_by_id(data['id'])
if not user:
return
@@ -381,8 +399,8 @@ async def user_join(sid, data):
await sio.enter_room(sid, f'user:{user.id}')
# Join all the channels only if user has channels permission
- if user.role == 'admin' or has_permission(user.id, 'features.channels'):
- channels = Channels.get_channels_by_user_id(user.id)
+ if user.role == 'admin' or await has_permission(user.id, 'features.channels'):
+ channels = await Channels.get_channels_by_user_id(user.id)
log.debug(f'{channels=}')
for channel in channels:
await sio.enter_room(sid, f'channel:{channel.id}')
@@ -395,7 +413,7 @@ async def heartbeat(sid, data):
user = SESSION_POOL.get(sid)
if user:
SESSION_POOL[sid] = {**user, 'last_seen_at': int(time.time())}
- await asyncio.to_thread(Users.update_last_active_by_id, user['id'])
+ await Users.update_last_active_by_id(user['id'])
@sio.on('join-channels')
@@ -408,13 +426,13 @@ async def join_channel(sid, data):
if data is None or 'id' not in data:
return
- user = Users.get_user_by_id(data['id'])
+ user = await Users.get_user_by_id(data['id'])
if not user:
return
# Join all the channels only if user has channels permission
- if user.role == 'admin' or has_permission(user.id, 'features.channels'):
- channels = Channels.get_channels_by_user_id(user.id)
+ if user.role == 'admin' or await has_permission(user.id, 'features.channels'):
+ channels = await Channels.get_channels_by_user_id(user.id)
log.debug(f'{channels=}')
for channel in channels:
await sio.enter_room(sid, f'channel:{channel.id}')
@@ -430,11 +448,11 @@ async def join_note(sid, data):
if token_data is None or 'id' not in token_data:
return
- user = Users.get_user_by_id(token_data['id'])
+ user = await Users.get_user_by_id(token_data['id'])
if not user:
return
- note = Notes.get_note_by_id(data['note_id'])
+ note = await Notes.get_note_by_id(data['note_id'])
if not note:
log.error(f'Note {data["note_id"]} not found for user {user.id}')
return
@@ -442,7 +460,7 @@ async def join_note(sid, data):
if (
user.role != 'admin'
and user.id != note.user_id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='note',
resource_id=note.id,
@@ -488,7 +506,7 @@ async def channel_events(sid, data):
room=room,
)
elif event_type == 'last_read_at':
- Channels.update_member_last_read_at(data['channel_id'], user['id'])
+ await Channels.update_member_last_read_at(data['channel_id'], user['id'])
@sio.on('events:chat')
@@ -501,7 +519,7 @@ async def chat_events(sid, data):
event_type = event_data.get('type')
if event_type == 'last_read_at':
- await asyncio.to_thread(Chats.update_chat_last_read_at_by_id, data['chat_id'], user['id'])
+ await Chats.update_chat_last_read_at_by_id(data['chat_id'], user['id'])
def normalize_document_id(document_id: str) -> str:
@@ -529,7 +547,7 @@ async def ydoc_document_join(sid, data):
if document_id.startswith('note:'):
note_id = document_id.split(':')[1]
- note = Notes.get_note_by_id(note_id)
+ note = await Notes.get_note_by_id(note_id)
if not note:
log.error(f'Note {note_id} not found')
return
@@ -537,7 +555,7 @@ async def ydoc_document_join(sid, data):
if (
user.get('role') != 'admin'
and user.get('id') != note.user_id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.get('id'),
resource_type='note',
resource_id=note.id,
@@ -602,7 +620,7 @@ async def document_save_handler(document_id, data, user):
if document_id.startswith('note:'):
note_id = document_id.split(':')[1]
- note = Notes.get_note_by_id(note_id)
+ note = await Notes.get_note_by_id(note_id)
if not note:
log.error(f'Note {note_id} not found')
return
@@ -610,7 +628,7 @@ async def document_save_handler(document_id, data, user):
if (
user.get('role') != 'admin'
and user.get('id') != note.user_id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.get('id'),
resource_type='note',
resource_id=note.id,
@@ -620,7 +638,7 @@ async def document_save_handler(document_id, data, user):
log.error(f'User {user.get("id")} does not have write access to note {note_id}')
return
- Notes.update_note_by_id(note_id, NoteUpdateForm(data=data))
+ await Notes.update_note_by_id(note_id, NoteUpdateForm(data=data))
@sio.on('ydoc:document:state')
@@ -793,7 +811,7 @@ async def disconnect(sid):
# print(f"Unknown session ID {sid} disconnected")
-def get_event_emitter(request_info, update_db=True):
+async def get_event_emitter(request_info, update_db=True):
async def __event_emitter__(event_data):
user_id = request_info['user_id']
chat_id = request_info['chat_id']
@@ -813,16 +831,14 @@ def get_event_emitter(request_info, update_db=True):
event_type = event_data.get('type')
if event_type == 'status':
- await asyncio.to_thread(
- Chats.add_message_status_to_chat_by_id_and_message_id,
+ await Chats.add_message_status_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
event_data.get('data', {}),
)
elif event_type == 'message':
- message = await asyncio.to_thread(
- Chats.get_message_by_id_and_message_id,
+ message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
@@ -831,8 +847,7 @@ def get_event_emitter(request_info, update_db=True):
content = message.get('content', '')
content += event_data.get('data', {}).get('content', '')
- await asyncio.to_thread(
- Chats.upsert_message_to_chat_by_id_and_message_id,
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
{
@@ -843,8 +858,7 @@ def get_event_emitter(request_info, update_db=True):
elif event_type == 'replace':
content = event_data.get('data', {}).get('content', '')
- await asyncio.to_thread(
- Chats.upsert_message_to_chat_by_id_and_message_id,
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
{
@@ -853,8 +867,7 @@ def get_event_emitter(request_info, update_db=True):
)
elif event_type == 'embeds':
- message = await asyncio.to_thread(
- Chats.get_message_by_id_and_message_id,
+ message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
@@ -862,8 +875,7 @@ def get_event_emitter(request_info, update_db=True):
embeds = event_data.get('data', {}).get('embeds', [])
embeds.extend(message.get('embeds', []))
- await asyncio.to_thread(
- Chats.upsert_message_to_chat_by_id_and_message_id,
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
{
@@ -872,8 +884,7 @@ def get_event_emitter(request_info, update_db=True):
)
elif event_type == 'files':
- message = await asyncio.to_thread(
- Chats.get_message_by_id_and_message_id,
+ message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
@@ -881,8 +892,7 @@ def get_event_emitter(request_info, update_db=True):
files = event_data.get('data', {}).get('files', [])
files.extend(message.get('files', []))
- await asyncio.to_thread(
- Chats.upsert_message_to_chat_by_id_and_message_id,
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
{
@@ -893,8 +903,7 @@ def get_event_emitter(request_info, update_db=True):
elif event_type in ('source', 'citation'):
data = event_data.get('data', {})
if data.get('type') is None:
- message = await asyncio.to_thread(
- Chats.get_message_by_id_and_message_id,
+ message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
@@ -902,8 +911,7 @@ def get_event_emitter(request_info, update_db=True):
sources = message.get('sources', [])
sources.append(data)
- await asyncio.to_thread(
- Chats.upsert_message_to_chat_by_id_and_message_id,
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
{
@@ -917,7 +925,7 @@ def get_event_emitter(request_info, update_db=True):
return None
-def get_event_call(request_info):
+async def get_event_call(request_info):
async def __event_caller__(event_data):
response = await sio.call(
'events',
diff --git a/backend/open_webui/storage/provider.py b/backend/open_webui/storage/provider.py
index 3c29462349..f70f3e862b 100644
--- a/backend/open_webui/storage/provider.py
+++ b/backend/open_webui/storage/provider.py
@@ -61,7 +61,7 @@ class LocalStorageProvider(StorageProvider):
contents = file.read()
if not contents:
raise ValueError(ERROR_MESSAGES.EMPTY_CONTENT)
- file_path = f'{UPLOAD_DIR}/{filename}'
+ file_path = os.path.join(UPLOAD_DIR, filename)
with open(file_path, 'wb') as f:
f.write(contents)
return contents, file_path
@@ -74,8 +74,8 @@ class LocalStorageProvider(StorageProvider):
@staticmethod
def delete_file(file_path: str) -> None:
"""Handles deletion of the file from local storage."""
- filename = file_path.split('/')[-1]
- file_path = f'{UPLOAD_DIR}/{filename}'
+ filename = os.path.basename(file_path)
+ file_path = os.path.join(UPLOAD_DIR, filename)
if os.path.isfile(file_path):
os.remove(file_path)
else:
@@ -140,7 +140,7 @@ class S3StorageProvider(StorageProvider):
def upload_file(self, file: BinaryIO, filename: str, tags: Dict[str, str]) -> Tuple[bytes, str]:
"""Handles uploading of the file to S3 storage."""
- _, file_path = LocalStorageProvider.upload_file(file, filename, tags)
+ contents, file_path = LocalStorageProvider.upload_file(file, filename, tags)
s3_key = os.path.join(self.key_prefix, filename)
try:
self.s3_client.upload_file(file_path, self.bucket_name, s3_key)
@@ -153,7 +153,7 @@ class S3StorageProvider(StorageProvider):
Tagging=tagging,
)
return (
- open(file_path, 'rb').read(),
+ contents,
f's3://{self.bucket_name}/{s3_key}',
)
except ClientError as e:
@@ -202,7 +202,7 @@ class S3StorageProvider(StorageProvider):
return '/'.join(full_file_path.split('//')[1].split('/')[1:])
def _get_local_file_path(self, s3_key: str) -> str:
- return f'{UPLOAD_DIR}/{s3_key.split("/")[-1]}'
+ return os.path.join(UPLOAD_DIR, s3_key.split('/')[-1])
class GCSStorageProvider(StorageProvider):
@@ -234,7 +234,7 @@ class GCSStorageProvider(StorageProvider):
"""Handles downloading of the file from GCS storage."""
try:
filename = file_path.removeprefix('gs://').split('/')[1]
- local_file_path = f'{UPLOAD_DIR}/{filename}'
+ local_file_path = os.path.join(UPLOAD_DIR, filename)
blob = self.bucket.get_blob(filename)
blob.download_to_filename(local_file_path)
@@ -298,7 +298,7 @@ class AzureStorageProvider(StorageProvider):
"""Handles downloading of the file from Azure Blob Storage."""
try:
filename = file_path.split('/')[-1]
- local_file_path = f'{UPLOAD_DIR}/{filename}'
+ local_file_path = os.path.join(UPLOAD_DIR, filename)
blob_client = self.container_client.get_blob_client(filename)
with open(local_file_path, 'wb') as download_file:
download_file.write(blob_client.download_blob().readall())
diff --git a/backend/open_webui/tools/builtin.py b/backend/open_webui/tools/builtin.py
index 1823e71df2..8aa27520f3 100644
--- a/backend/open_webui/tools/builtin.py
+++ b/backend/open_webui/tools/builtin.py
@@ -37,7 +37,7 @@ from open_webui.models.channels import Channels, ChannelMember, Channel
from open_webui.models.messages import Messages, Message
from open_webui.models.groups import Groups
from open_webui.models.memories import Memories
-from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
from open_webui.utils.sanitize import sanitize_code
log = logging.getLogger(__name__)
@@ -250,7 +250,7 @@ async def generate_image(
# Persist files to DB if chat context is available
if __chat_id__ and __message_id__ and images:
- db_files = Chats.add_message_files_by_id_and_message_id(
+ db_files = await Chats.add_message_files_by_id_and_message_id(
__chat_id__,
__message_id__,
image_files,
@@ -317,7 +317,7 @@ async def edit_image(
# Persist files to DB if chat context is available
if __chat_id__ and __message_id__ and images:
- db_files = Chats.add_message_files_by_id_and_message_id(
+ db_files = await Chats.add_message_files_by_id_and_message_id(
__chat_id__,
__message_id__,
image_files,
@@ -473,14 +473,14 @@ async def execute_code(
from open_webui.models.users import Users
from open_webui.utils.files import get_image_url_from_base64
- user = Users.get_user_by_id(__user__['id'])
+ user = await Users.get_user_by_id(__user__['id'])
# Extract and upload images from stdout
if stdout and isinstance(stdout, str):
stdout_lines = stdout.split('\n')
for idx, line in enumerate(stdout_lines):
if 'data:image/png;base64' in line:
- image_url = get_image_url_from_base64(
+ image_url = await get_image_url_from_base64(
__request__,
line,
__metadata__ or {},
@@ -495,7 +495,7 @@ async def execute_code(
result_lines = result.split('\n')
for idx, line in enumerate(result_lines):
if 'data:image/png;base64' in line:
- image_url = get_image_url_from_base64(
+ image_url = await get_image_url_from_base64(
__request__,
line,
__metadata__ or {},
@@ -650,10 +650,10 @@ async def delete_memory(
try:
user = UserModel(**__user__) if __user__ else None
- result = Memories.delete_memory_by_id_and_user_id(memory_id, user.id)
+ result = await Memories.delete_memory_by_id_and_user_id(memory_id, user.id)
if result:
- VECTOR_DB_CLIENT.delete(collection_name=f'user-memory-{user.id}', ids=[memory_id])
+ await ASYNC_VECTOR_DB_CLIENT.delete(collection_name=f'user-memory-{user.id}', ids=[memory_id])
return json.dumps(
{'status': 'success', 'message': f'Memory {memory_id} deleted'},
ensure_ascii=False,
@@ -680,7 +680,7 @@ async def list_memories(
try:
user = UserModel(**__user__) if __user__ else None
- memories = Memories.get_memories_by_user_id(user.id)
+ memories = await Memories.get_memories_by_user_id(user.id)
if memories:
result = [
@@ -730,9 +730,9 @@ async def search_notes(
try:
user_id = __user__.get('id')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
- result = Notes.search_notes(
+ result = await Notes.search_notes(
user_id=user_id,
filter={
'query': query,
@@ -760,14 +760,26 @@ async def search_notes(
content_snippet = ''
if note.data and note.data.get('content', {}).get('md'):
md_content = note.data['content']['md']
- lower_content = md_content.lower()
- lower_query = query.lower()
- idx = lower_content.find(lower_query)
- if idx != -1:
- start = max(0, idx - 50)
- end = min(len(md_content), idx + len(query) + 100)
+ content_lower = md_content.lower()
+
+ # Find the first matching word to center the snippet around.
+ search_words = query.lower().split()
+ match_pos = -1
+ match_len = len(query)
+ for word in search_words:
+ found_pos = content_lower.find(word)
+ if found_pos != -1:
+ match_pos = found_pos
+ match_len = len(word)
+ break
+
+ if match_pos != -1:
+ snippet_start = max(0, match_pos - 50)
+ snippet_end = min(len(md_content), match_pos + match_len + 100)
content_snippet = (
- ('...' if start > 0 else '') + md_content[start:end] + ('...' if end < len(md_content) else '')
+ ('...' if snippet_start > 0 else '')
+ + md_content[snippet_start:snippet_end]
+ + ('...' if snippet_end < len(md_content) else '')
)
else:
content_snippet = md_content[:150] + ('...' if len(md_content) > 150 else '')
@@ -808,18 +820,18 @@ async def view_note(
return json.dumps({'error': 'User context not available'})
try:
- note = Notes.get_note_by_id(note_id)
+ note = await Notes.get_note_by_id(note_id)
if not note:
return json.dumps({'error': 'Note not found'})
# Check access permission
user_id = __user__.get('id')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
from open_webui.models.access_grants import AccessGrants
- if note.user_id != user_id and not AccessGrants.has_access(
+ if note.user_id != user_id and not await AccessGrants.has_access(
user_id=user_id,
resource_type='note',
resource_id=note.id,
@@ -878,7 +890,7 @@ async def write_note(
access_grants=[], # Private by default - only owner can access
)
- new_note = Notes.insert_new_note(user_id, form)
+ new_note = await Notes.insert_new_note(user_id, form)
if not new_note:
return json.dumps({'error': 'Failed to create note'})
@@ -921,18 +933,18 @@ async def replace_note_content(
try:
from open_webui.models.notes import NoteUpdateForm
- note = Notes.get_note_by_id(note_id)
+ note = await Notes.get_note_by_id(note_id)
if not note:
return json.dumps({'error': 'Note not found'})
# Check write permission
user_id = __user__.get('id')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
from open_webui.models.access_grants import AccessGrants
- if note.user_id != user_id and not AccessGrants.has_access(
+ if note.user_id != user_id and not await AccessGrants.has_access(
user_id=user_id,
resource_type='note',
resource_id=note.id,
@@ -947,7 +959,7 @@ async def replace_note_content(
update_data['title'] = title
form = NoteUpdateForm(**update_data)
- updated_note = Notes.update_note_by_id(note_id, form)
+ updated_note = await Notes.update_note_by_id(note_id, form)
if not updated_note:
return json.dumps({'error': 'Failed to update note'})
@@ -998,7 +1010,7 @@ async def search_chats(
try:
user_id = __user__.get('id')
- chats = Chats.get_chats_by_user_id_and_search_text(
+ chats = await Chats.get_chats_by_user_id_and_search_text(
user_id=user_id,
search_text=query,
include_archived=False,
@@ -1073,7 +1085,7 @@ async def view_chat(
try:
user_id = __user__.get('id')
- chat = Chats.get_chat_by_id_and_user_id(chat_id, user_id)
+ chat = await Chats.get_chat_by_id_and_user_id(chat_id, user_id)
if not chat:
return json.dumps({'error': 'Chat not found or access denied'})
@@ -1145,7 +1157,7 @@ async def search_channels(
user_id = __user__.get('id')
# Get all channels the user has access to
- all_channels = Channels.get_channels_by_user_id(user_id)
+ all_channels = await Channels.get_channels_by_user_id(user_id)
# Filter by query
lower_query = query.lower()
@@ -1201,7 +1213,7 @@ async def search_channel_messages(
user_id = __user__.get('id')
# Get all channels the user has access to
- user_channels = Channels.get_channels_by_user_id(user_id)
+ user_channels = await Channels.get_channels_by_user_id(user_id)
channel_ids = [c.id for c in user_channels]
channel_map = {c.id: c for c in user_channels}
@@ -1213,7 +1225,7 @@ async def search_channel_messages(
end_ts = end_timestamp * 1_000_000_000 if end_timestamp else None
# Search messages using the model method
- matching_messages = Messages.search_messages_by_channel_ids(
+ matching_messages = await Messages.search_messages_by_channel_ids(
channel_ids=channel_ids,
query=query,
start_timestamp=start_ts,
@@ -1274,18 +1286,18 @@ async def view_channel_message(
try:
user_id = __user__.get('id')
- message = Messages.get_message_by_id(message_id)
+ message = await Messages.get_message_by_id(message_id)
if not message:
return json.dumps({'error': 'Message not found'})
# Verify user has access to the channel
- channel = Channels.get_channel_by_id(message.channel_id)
+ channel = await Channels.get_channel_by_id(message.channel_id)
if not channel:
return json.dumps({'error': 'Channel not found'})
# Check if user has access to the channel
- user_channels = Channels.get_channels_by_user_id(user_id)
+ user_channels = await Channels.get_channels_by_user_id(user_id)
channel_ids = [c.id for c in user_channels]
if message.channel_id not in channel_ids:
@@ -1336,24 +1348,24 @@ async def view_channel_thread(
user_id = __user__.get('id')
# Get the parent message
- parent_message = Messages.get_message_by_id(parent_message_id)
+ parent_message = await Messages.get_message_by_id(parent_message_id)
if not parent_message:
return json.dumps({'error': 'Message not found'})
# Verify user has access to the channel
- channel = Channels.get_channel_by_id(parent_message.channel_id)
+ channel = await Channels.get_channel_by_id(parent_message.channel_id)
if not channel:
return json.dumps({'error': 'Channel not found'})
- user_channels = Channels.get_channels_by_user_id(user_id)
+ user_channels = await Channels.get_channels_by_user_id(user_id)
channel_ids = [c.id for c in user_channels]
if parent_message.channel_id not in channel_ids:
return json.dumps({'error': 'Access denied'})
# Get all thread replies
- thread_replies = Messages.get_thread_replies_by_message_id(parent_message_id)
+ thread_replies = await Messages.get_thread_replies_by_message_id(parent_message_id)
# Build the response
messages = []
@@ -1427,9 +1439,9 @@ async def list_knowledge_bases(
from open_webui.models.knowledge import Knowledges
user_id = __user__.get('id')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
- result = Knowledges.search_knowledge_bases(
+ result = await Knowledges.search_knowledge_bases(
user_id,
filter={
'query': '',
@@ -1442,7 +1454,7 @@ async def list_knowledge_bases(
knowledge_bases = []
for knowledge_base in result.items:
- files = Knowledges.get_files_by_id(knowledge_base.id)
+ files = await Knowledges.get_files_by_id(knowledge_base.id)
file_count = len(files) if files else 0
knowledge_bases.append(
@@ -1486,9 +1498,9 @@ async def search_knowledge_bases(
from open_webui.models.knowledge import Knowledges
user_id = __user__.get('id')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
- result = Knowledges.search_knowledge_bases(
+ result = await Knowledges.search_knowledge_bases(
user_id,
filter={
'query': query,
@@ -1501,7 +1513,7 @@ async def search_knowledge_bases(
knowledge_bases = []
for knowledge_base in result.items:
- files = Knowledges.get_files_by_id(knowledge_base.id)
+ files = await Knowledges.get_files_by_id(knowledge_base.id)
file_count = len(files) if files else 0
knowledge_bases.append(
@@ -1552,7 +1564,7 @@ async def search_knowledge_files(
user_id = __user__.get('id')
user_role = __user__.get('role', 'user')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
# When model has attached knowledge, scope to attached KBs/files only
if __model_knowledge__:
@@ -1577,14 +1589,14 @@ async def search_knowledge_files(
# Search within attached KBs
for kb_id in attached_kb_ids:
- knowledge = Knowledges.get_knowledge_by_id(kb_id)
+ knowledge = await Knowledges.get_knowledge_by_id(kb_id)
if not knowledge:
continue
if not (
user_role == 'admin'
or knowledge.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -1594,7 +1606,7 @@ async def search_knowledge_files(
):
continue
- result = Knowledges.search_files_by_id(
+ result = await Knowledges.search_files_by_id(
knowledge_id=kb_id,
user_id=user_id,
filter={'query': query},
@@ -1617,7 +1629,7 @@ async def search_knowledge_files(
if not knowledge_id and attached_file_ids:
query_lower = query.lower() if query else ''
for file_id in attached_file_ids:
- file = Files.get_file_by_id(file_id)
+ file = await Files.get_file_by_id(file_id)
if file and (not query_lower or query_lower in file.filename.lower()):
all_files.append(
{
@@ -1633,7 +1645,7 @@ async def search_knowledge_files(
# No attached knowledge - search all accessible KBs
if knowledge_id:
- result = Knowledges.search_files_by_id(
+ result = await Knowledges.search_files_by_id(
knowledge_id=knowledge_id,
user_id=user_id,
filter={'query': query},
@@ -1641,7 +1653,7 @@ async def search_knowledge_files(
limit=count,
)
else:
- result = Knowledges.search_knowledge_files(
+ result = await Knowledges.search_knowledge_files(
filter={
'query': query,
'user_id': user_id,
@@ -1719,7 +1731,7 @@ async def view_file(
user_id = __user__.get('id')
user_role = __user__.get('role', 'user')
- file = Files.get_file_by_id(file_id)
+ file = await Files.get_file_by_id(file_id)
if not file:
return json.dumps({'error': 'File not found'})
@@ -1729,7 +1741,7 @@ async def view_file(
and not any(
item.get('type') == 'file' and item.get('id') == file_id for item in (__model_knowledge__ or [])
)
- and not has_access_to_file(
+ and not await has_access_to_file(
file_id=file_id,
access_type='read',
user=UserModel(**__user__),
@@ -1811,14 +1823,14 @@ async def view_knowledge_file(
user_id = __user__.get('id')
user_role = __user__.get('role', 'user')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
- file = Files.get_file_by_id(file_id)
+ file = await Files.get_file_by_id(file_id)
if not file:
return json.dumps({'error': 'File not found'})
# Check access via any KB containing this file
- knowledges = Knowledges.get_knowledges_by_file_id(file_id)
+ knowledges = await Knowledges.get_knowledges_by_file_id(file_id)
has_knowledge_access = False
knowledge_info = None
@@ -1826,7 +1838,7 @@ async def view_knowledge_file(
if (
user_role == 'admin'
or knowledge_base.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge_base.id,
@@ -1903,7 +1915,7 @@ async def list_knowledge(
user_id = __user__.get('id')
user_role = __user__.get('role', 'user')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
knowledge_bases = []
files = []
@@ -1914,11 +1926,11 @@ async def list_knowledge(
item_id = item.get('id')
if item_type == 'collection':
- knowledge = Knowledges.get_knowledge_by_id(item_id)
+ knowledge = await Knowledges.get_knowledge_by_id(item_id)
if knowledge and (
user_role == 'admin'
or knowledge.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -1926,7 +1938,7 @@ async def list_knowledge(
user_group_ids=set(user_group_ids),
)
):
- kb_files = Knowledges.get_files_by_id(knowledge.id)
+ kb_files = await Knowledges.get_files_by_id(knowledge.id)
file_count = len(kb_files) if kb_files else 0
kb_entry = {
@@ -1943,7 +1955,7 @@ async def list_knowledge(
knowledge_bases.append(kb_entry)
elif item_type == 'file':
- file = Files.get_file_by_id(item_id)
+ file = await Files.get_file_by_id(item_id)
if file:
files.append(
{
@@ -1954,11 +1966,11 @@ async def list_knowledge(
)
elif item_type == 'note':
- note = Notes.get_note_by_id(item_id)
+ note = await Notes.get_note_by_id(item_id)
if note and (
user_role == 'admin'
or note.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='note',
resource_id=note.id,
@@ -2036,7 +2048,7 @@ async def query_knowledge_files(
user_id = __user__.get('id')
user_role = __user__.get('role', 'user')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
embedding_function = __request__.app.state.EMBEDDING_FUNCTION
if not embedding_function:
@@ -2053,11 +2065,11 @@ async def query_knowledge_files(
if item_type == 'collection':
# Knowledge base - use KB ID as collection name
- knowledge = Knowledges.get_knowledge_by_id(item_id)
+ knowledge = await Knowledges.get_knowledge_by_id(item_id)
if knowledge and (
user_role == 'admin'
or knowledge.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -2069,17 +2081,17 @@ async def query_knowledge_files(
elif item_type == 'file':
# Individual file - use file-{id} as collection name
- file = Files.get_file_by_id(item_id)
+ file = await Files.get_file_by_id(item_id)
if file:
collection_names.append(f'file-{item_id}')
elif item_type == 'note':
# Note - always return full content as context
- note = Notes.get_note_by_id(item_id)
+ note = await Notes.get_note_by_id(item_id)
if note and (
user_role == 'admin'
or note.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='note',
resource_id=note.id,
@@ -2099,11 +2111,11 @@ async def query_knowledge_files(
elif knowledge_ids:
# User specified specific KBs
for knowledge_id in knowledge_ids:
- knowledge = Knowledges.get_knowledge_by_id(knowledge_id)
+ knowledge = await Knowledges.get_knowledge_by_id(knowledge_id)
if knowledge and (
user_role == 'admin'
or knowledge.user_id == user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user_id,
resource_type='knowledge',
resource_id=knowledge.id,
@@ -2114,7 +2126,7 @@ async def query_knowledge_files(
collection_names.append(knowledge_id)
else:
# No model knowledge and no specific IDs - search all accessible KBs
- result = Knowledges.search_knowledge_bases(
+ result = await Knowledges.search_knowledge_bases(
user_id,
filter={
'query': '',
@@ -2190,10 +2202,10 @@ async def query_knowledge_bases(
import heapq
from open_webui.models.knowledge import Knowledges
from open_webui.routers.knowledge import KNOWLEDGE_BASES_COLLECTION
- from open_webui.retrieval.vector.factory import VECTOR_DB_CLIENT
+ from open_webui.retrieval.vector.async_client import ASYNC_VECTOR_DB_CLIENT
user_id = __user__.get('id')
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
query_embedding = await __request__.app.state.EMBEDDING_FUNCTION(query)
# Min-heap of (distance, knowledge_base_id) - only holds top `count` results
@@ -2203,7 +2215,7 @@ async def query_knowledge_bases(
page_size = 100
while True:
- accessible_knowledge_bases = Knowledges.search_knowledge_bases(
+ accessible_knowledge_bases = await Knowledges.search_knowledge_bases(
user_id,
filter={'user_id': user_id, 'group_ids': user_group_ids},
skip=page_offset,
@@ -2215,7 +2227,7 @@ async def query_knowledge_bases(
accessible_ids = [kb.id for kb in accessible_knowledge_bases.items]
- search_results = VECTOR_DB_CLIENT.search(
+ search_results = await ASYNC_VECTOR_DB_CLIENT.search(
collection_name=KNOWLEDGE_BASES_COLLECTION,
vectors=[query_embedding],
filter={'knowledge_base_id': {'$in': accessible_ids}},
@@ -2247,7 +2259,7 @@ async def query_knowledge_bases(
matching_knowledge_bases = []
for distance, knowledge_base_id in sorted_results:
- knowledge_base = Knowledges.get_knowledge_by_id(knowledge_base_id)
+ knowledge_base = await Knowledges.get_knowledge_by_id(knowledge_base_id)
if knowledge_base:
matching_knowledge_bases.append(
{
@@ -2295,7 +2307,7 @@ async def view_skill(
user_id = __user__.get('id')
# Direct DB lookup by id (case-insensitive since IDs are stored lowercase)
- skill = Skills.get_skill_by_id(id.lower())
+ skill = await Skills.get_skill_by_id(id.lower())
if not skill or not skill.is_active:
return json.dumps({'error': f"Skill '{id}' not found"})
@@ -2303,8 +2315,8 @@ async def view_skill(
# Check user access
user_role = __user__.get('role', 'user')
if user_role != 'admin' and skill.user_id != user_id:
- user_group_ids = [group.id for group in Groups.get_groups_by_member_id(user_id)]
- if not AccessGrants.has_access(
+ user_group_ids = [group.id for group in await Groups.get_groups_by_member_id(user_id)]
+ if not await AccessGrants.has_access(
user_id=user_id,
resource_type='skill',
resource_id=skill.id,
@@ -2337,13 +2349,40 @@ VALID_TASK_STATUSES = {'pending', 'in_progress', 'completed', 'cancelled'}
class TaskItem(BaseModel):
id: Optional[str] = Field(None, description='Unique identifier for the task. Auto-generated if omitted.')
- content: Optional[str] = Field(None, description='Task description. Aliases: title, name, description.')
+ content: str = Field(..., description='Task description.')
status: Literal['pending', 'in_progress', 'completed', 'cancelled'] = Field('pending', description='Task status.')
-async def tasks(
- tasks: Optional[list[TaskItem]] = None,
- overwrite: bool = True,
+def _task_summary(all_tasks: list[dict]) -> dict:
+ """Build summary counts for a task list."""
+ pending = sum(1 for t in all_tasks if t['status'] == 'pending')
+ in_progress = sum(1 for t in all_tasks if t['status'] == 'in_progress')
+ completed = sum(1 for t in all_tasks if t['status'] == 'completed')
+ cancelled = sum(1 for t in all_tasks if t['status'] == 'cancelled')
+ return {
+ 'total': len(all_tasks),
+ 'pending': pending,
+ 'in_progress': in_progress,
+ 'completed': completed,
+ 'cancelled': cancelled,
+ }
+
+
+async def _emit_tasks(event_emitter, all_tasks: list[dict]):
+ """Persist task state to the UI."""
+ if event_emitter:
+ await event_emitter(
+ {
+ 'type': 'chat:message:tasks',
+ 'data': {
+ 'tasks': all_tasks,
+ },
+ }
+ )
+
+
+async def create_tasks(
+ tasks: list[TaskItem],
__chat_id__: str = None,
__message_id__: str = None,
__event_emitter__: callable = None,
@@ -2351,146 +2390,425 @@ async def tasks(
__user__: dict = None,
) -> str:
"""
- Track progress on multi-step work by maintaining a task checklist.
- Use this whenever a request involves multiple steps or could take
- significant effort. Call to set the full list, then call again
- with overwrite=false after completing each task to mark it
- completed. Do not leave tasks in_progress when the work is done.
- Each task has an id, content, and status (pending, in_progress,
- completed, cancelled).
+ Create a task checklist to track progress on multi-step work.
+ Call this once at the start to define all steps, then use
+ update_task to mark each task as you complete it.
- :param tasks: Optional list of task items. Each item: id (string), content (string, required for new tasks), status (pending|in_progress|completed|cancelled). Leave empty to fetch without modifying.
- :param overwrite: If true (default), replaces the entire task list. If false, updates/adds tasks by id while keeping existing ones.
+ :param tasks: List of task items. Each item: content (string, required), status (pending|in_progress|completed|cancelled, default pending), id (optional, auto-generated).
:return: JSON with the full task list and summary counts
"""
if __chat_id__ is None:
return json.dumps({'error': 'Chat context not available'})
try:
-
- def _to_dict(task) -> dict:
- """Convert TaskItem or dict to plain dict."""
+ all_tasks = []
+ for idx, task in enumerate(tasks):
if hasattr(task, 'model_dump'):
d = task.model_dump(exclude_none=True)
- # Include any extra fields the model sent
- if hasattr(task, 'model_extra') and task.model_extra:
- d.update(task.model_extra)
- return d
- return dict(task) if not isinstance(task, dict) else task
+ elif isinstance(task, dict):
+ d = task
+ else:
+ d = dict(task)
- def _resolve_content(d: dict) -> str:
- """Accept content, title, name, or description as the task text."""
- for key in ('content', 'title', 'name', 'description'):
- val = str(d.get(key, '')).strip()
- if val:
- return val
- return ''
+ content = str(d.get('content', '')).strip()
+ if not content:
+ continue
- def _resolve_id(d: dict, idx: int) -> str:
- """Use provided id, or auto-generate from index."""
- item_id = str(d.get('id', '') or '').strip()
- return item_id if item_id else str(idx + 1)
+ item_id = str(d.get('id', '') or '').strip() or str(idx + 1)
+ status = str(d.get('status', 'pending')).strip().lower()
+ if status not in VALID_TASK_STATUSES:
+ status = 'pending'
- if tasks is None:
- # Read-only - return current list
- all_tasks = Chats.get_chat_tasks_by_id(__chat_id__)
- elif overwrite:
- # Full replacement - validate and write
- all_tasks = []
- for idx, task in enumerate(tasks):
- d = _to_dict(task)
- item_id = _resolve_id(d, idx)
- content = _resolve_content(d)
- if not content:
- continue
+ all_tasks.append({'id': item_id, 'content': content, 'status': status})
- status = str(d.get('status', 'pending')).strip().lower()
- if status not in VALID_TASK_STATUSES:
- status = 'pending'
-
- all_tasks.append(
- {
- 'id': item_id,
- 'content': content,
- 'status': status,
- }
- )
- else:
- # Partial update - merge by id
- existing_tasks = Chats.get_chat_tasks_by_id(__chat_id__)
- existing_by_id = {t['id']: t for t in existing_tasks}
-
- seen_ids = set()
- for idx, task in enumerate(tasks):
- d = _to_dict(task)
- item_id = _resolve_id(d, len(existing_tasks) + idx)
-
- seen_ids.add(item_id)
-
- if item_id in existing_by_id:
- resolved = _resolve_content(d)
- if resolved:
- existing_by_id[item_id]['content'] = resolved
- status = str(d.get('status', '')).strip().lower()
- if status and status in VALID_TASK_STATUSES:
- existing_by_id[item_id]['status'] = status
- else:
- content = _resolve_content(d)
- if not content:
- continue
-
- status = str(d.get('status', 'pending')).strip().lower()
- if status not in VALID_TASK_STATUSES:
- status = 'pending'
-
- existing_by_id[item_id] = {
- 'id': item_id,
- 'content': content,
- 'status': status,
- }
-
- # Preserve order of existing, append new
- all_tasks = []
- for t in existing_tasks:
- if t['id'] in existing_by_id:
- all_tasks.append(existing_by_id[t['id']])
- for item_id in seen_ids:
- if not any(t['id'] == item_id for t in existing_tasks):
- all_tasks.append(existing_by_id[item_id])
-
- # Persist to DB and emit (skip for read-only)
- if tasks is not None:
- Chats.update_chat_tasks_by_id(__chat_id__, all_tasks)
-
- if __event_emitter__:
- await __event_emitter__(
- {
- 'type': 'chat:message:tasks',
- 'data': {
- 'tasks': all_tasks,
- },
- }
- )
-
- # Build summary counts
- pending = sum(1 for t in all_tasks if t['status'] == 'pending')
- in_progress = sum(1 for t in all_tasks if t['status'] == 'in_progress')
- completed = sum(1 for t in all_tasks if t['status'] == 'completed')
- cancelled = sum(1 for t in all_tasks if t['status'] == 'cancelled')
+ await Chats.update_chat_tasks_by_id(__chat_id__, all_tasks)
+ await _emit_tasks(__event_emitter__, all_tasks)
return json.dumps(
- {
- 'tasks': all_tasks,
- 'summary': {
- 'total': len(all_tasks),
- 'pending': pending,
- 'in_progress': in_progress,
- 'completed': completed,
- 'cancelled': cancelled,
- },
- },
+ {'tasks': all_tasks, 'summary': _task_summary(all_tasks)},
ensure_ascii=False,
)
except Exception as e:
log.exception(f'tasks error: {e}')
return json.dumps({'error': str(e)})
+
+
+async def update_task(
+ id: str,
+ status: str = 'completed',
+ __chat_id__: str = None,
+ __message_id__: str = None,
+ __event_emitter__: callable = None,
+ __request__: Request = None,
+ __user__: dict = None,
+) -> str:
+ """
+ Mark a single task as completed, in_progress, pending, or cancelled.
+ Call this after finishing each step. You MUST call this for every
+ task, including the very last one.
+
+ :param id: The task ID to update
+ :param status: New status: completed, in_progress, pending, or cancelled (default: completed)
+ :return: JSON with the updated task list and summary counts
+ """
+ if __chat_id__ is None:
+ return json.dumps({'error': 'Chat context not available'})
+
+ try:
+ status = status.strip().lower()
+ if status not in VALID_TASK_STATUSES:
+ return json.dumps(
+ {'error': f'Invalid status: {status}. Must be one of: {", ".join(sorted(VALID_TASK_STATUSES))}'}
+ )
+
+ all_tasks = await Chats.get_chat_tasks_by_id(__chat_id__)
+
+ found = False
+ for task in all_tasks:
+ if task['id'] == id:
+ task['status'] = status
+ found = True
+ break
+
+ if not found:
+ return json.dumps({'error': f'Task with id "{id}" not found'})
+
+ await Chats.update_chat_tasks_by_id(__chat_id__, all_tasks)
+ await _emit_tasks(__event_emitter__, all_tasks)
+
+ return json.dumps(
+ {'tasks': all_tasks, 'summary': _task_summary(all_tasks)},
+ ensure_ascii=False,
+ )
+ except Exception as e:
+ log.exception(f'update_task_status error: {e}')
+ return json.dumps({'error': str(e)})
+
+
+# =============================================================================
+# AUTOMATION TOOLS
+# =============================================================================
+
+
+async def create_automation(
+ name: str,
+ prompt: str,
+ rrule: str,
+ model_id: Optional[str] = None,
+ __request__: Request = None,
+ __user__: dict = None,
+ __metadata__: dict = None,
+) -> str:
+ """
+ Create a scheduled automation that runs a prompt on a recurring or one-time schedule.
+ Use this when the user wants to schedule a task to run automatically.
+
+ The rrule parameter must be a valid iCalendar RRULE string. Common examples:
+ - Every day at 9am: "DTSTART:20250101T090000\\nRRULE:FREQ=DAILY"
+ - Every Monday at 8am: "DTSTART:20250106T080000\\nRRULE:FREQ=WEEKLY;BYDAY=MO"
+ - Every hour: "RRULE:FREQ=HOURLY;INTERVAL=1"
+ - Every 30 minutes: "RRULE:FREQ=MINUTELY;INTERVAL=30"
+ - Once at a specific time: "DTSTART:20250415T140000\\nRRULE:FREQ=DAILY;COUNT=1"
+ - First day of every month: "DTSTART:20250101T090000\\nRRULE:FREQ=MONTHLY;BYMONTHDAY=1"
+
+ The DTSTART time should reflect the desired execution time. Use COUNT=1 for one-time automations.
+
+ :param name: A short descriptive name for the automation
+ :param prompt: The prompt/instructions to execute on each run
+ :param rrule: An iCalendar RRULE string defining the schedule
+ :param model_id: Optional model ID to use. Defaults to the current chat model if omitted.
+ :return: JSON with the created automation details including id, next scheduled runs
+ """
+ if __request__ is None:
+ return json.dumps({'error': 'Request context not available'})
+
+ if not __user__:
+ return json.dumps({'error': 'User context not available'})
+
+ try:
+ from open_webui.models.automations import Automations, AutomationForm, AutomationData
+ from open_webui.models.users import Users
+ from open_webui.utils.automations import validate_rrule, next_run_ns, next_n_runs_ns
+
+ user_id = __user__.get('id')
+ user = await Users.get_user_by_id(user_id)
+ if not user:
+ return json.dumps({'error': 'User not found'})
+
+ # Default to current chat's model if not specified
+ if not model_id:
+ model_id = (__metadata__ or {}).get('model_id') or (__metadata__ or {}).get('model')
+ if not model_id:
+ return json.dumps({'error': 'model_id is required (could not detect current model)'})
+
+ # Validate the RRULE
+ try:
+ validate_rrule(rrule)
+ except ValueError as e:
+ return json.dumps({'error': f'Invalid schedule: {e}'})
+
+ tz = user.timezone
+ form = AutomationForm(
+ name=name,
+ data=AutomationData(
+ prompt=prompt,
+ model_id=model_id,
+ rrule=rrule,
+ ),
+ is_active=True,
+ )
+
+ automation = await Automations.insert(user_id, form, next_run_ns(rrule, tz=tz))
+
+ return json.dumps(
+ {
+ 'status': 'success',
+ 'id': automation.id,
+ 'name': automation.name,
+ 'model_id': model_id,
+ 'is_active': automation.is_active,
+ 'next_runs': next_n_runs_ns(rrule, tz=tz),
+ },
+ ensure_ascii=False,
+ )
+ except Exception as e:
+ log.exception(f'create_automation error: {e}')
+ return json.dumps({'error': str(e)})
+
+
+async def update_automation(
+ automation_id: str,
+ name: Optional[str] = None,
+ prompt: Optional[str] = None,
+ rrule: Optional[str] = None,
+ model_id: Optional[str] = None,
+ __request__: Request = None,
+ __user__: dict = None,
+) -> str:
+ """
+ Update an existing automation. Only the provided fields are changed; omitted fields stay the same.
+
+ :param automation_id: The ID of the automation to update
+ :param name: New name for the automation (optional)
+ :param prompt: New prompt/instructions (optional)
+ :param rrule: New iCalendar RRULE schedule string (optional). See create_automation for format examples.
+ :param model_id: New model ID to use (optional)
+ :return: JSON with the updated automation details
+ """
+ if __request__ is None:
+ return json.dumps({'error': 'Request context not available'})
+
+ if not __user__:
+ return json.dumps({'error': 'User context not available'})
+
+ try:
+ from open_webui.models.automations import Automations, AutomationForm, AutomationData
+ from open_webui.models.users import Users
+ from open_webui.utils.automations import validate_rrule, next_run_ns, next_n_runs_ns
+
+ user_id = __user__.get('id')
+ user = await Users.get_user_by_id(user_id)
+
+ automation = await Automations.get_by_id(automation_id)
+ if not automation:
+ return json.dumps({'error': 'Automation not found'})
+ if automation.user_id != user_id:
+ return json.dumps({'error': 'Access denied'})
+
+ # Merge provided fields with existing values
+ new_name = name if name is not None else automation.name
+ new_prompt = prompt if prompt is not None else automation.data.get('prompt', '')
+ new_model_id = model_id if model_id is not None else automation.data.get('model_id', '')
+ new_rrule = rrule if rrule is not None else automation.data.get('rrule', '')
+
+ # Validate RRULE if changed
+ if rrule is not None:
+ try:
+ validate_rrule(new_rrule)
+ except ValueError as e:
+ return json.dumps({'error': f'Invalid schedule: {e}'})
+
+ tz = user.timezone if user else None
+ form = AutomationForm(
+ name=new_name,
+ data=AutomationData(
+ prompt=new_prompt,
+ model_id=new_model_id,
+ rrule=new_rrule,
+ ),
+ is_active=automation.is_active,
+ )
+
+ updated = await Automations.update(automation_id, form, next_run_ns(new_rrule, tz=tz))
+
+ return json.dumps(
+ {
+ 'status': 'success',
+ 'id': updated.id,
+ 'name': updated.name,
+ 'model_id': new_model_id,
+ 'is_active': updated.is_active,
+ 'next_runs': next_n_runs_ns(new_rrule, tz=tz),
+ },
+ ensure_ascii=False,
+ )
+ except Exception as e:
+ log.exception(f'update_automation error: {e}')
+ return json.dumps({'error': str(e)})
+
+
+async def list_automations(
+ status: Optional[str] = None,
+ count: int = 10,
+ __request__: Request = None,
+ __user__: dict = None,
+) -> str:
+ """
+ List the user's scheduled automations.
+
+ :param status: Filter by status: "active", "paused", or omit for all
+ :param count: Maximum number of automations to return (default: 10)
+ :return: JSON list of automations with id, name, prompt snippet, schedule, status, and next runs
+ """
+ if __request__ is None:
+ return json.dumps({'error': 'Request context not available'})
+
+ if not __user__:
+ return json.dumps({'error': 'User context not available'})
+
+ try:
+ from open_webui.models.automations import Automations
+ from open_webui.models.users import Users
+ from open_webui.utils.automations import next_n_runs_ns
+
+ user_id = __user__.get('id')
+ user = await Users.get_user_by_id(user_id)
+
+ result = await Automations.search_automations(
+ user_id=user_id,
+ status=status,
+ skip=0,
+ limit=count,
+ )
+
+ automations = []
+ for item in result.items:
+ rrule = item.data.get('rrule', '')
+ prompt_text = item.data.get('prompt', '')
+ snippet = prompt_text[:100] + ('...' if len(prompt_text) > 100 else '')
+
+ automations.append(
+ {
+ 'id': item.id,
+ 'name': item.name,
+ 'prompt_snippet': snippet,
+ 'model_id': item.data.get('model_id', ''),
+ 'rrule': rrule,
+ 'is_active': item.is_active,
+ 'last_run_at': item.last_run_at,
+ 'next_runs': next_n_runs_ns(rrule, tz=user.timezone if user else None),
+ }
+ )
+
+ return json.dumps(
+ {'automations': automations, 'total': result.total},
+ ensure_ascii=False,
+ )
+ except Exception as e:
+ log.exception(f'list_automations error: {e}')
+ return json.dumps({'error': str(e)})
+
+
+async def toggle_automation(
+ automation_id: str,
+ __request__: Request = None,
+ __user__: dict = None,
+) -> str:
+ """
+ Pause or resume a scheduled automation. If active, it will be paused. If paused, it will be resumed.
+
+ :param automation_id: The ID of the automation to toggle
+ :return: JSON with the updated automation status
+ """
+ if __request__ is None:
+ return json.dumps({'error': 'Request context not available'})
+
+ if not __user__:
+ return json.dumps({'error': 'User context not available'})
+
+ try:
+ from open_webui.models.automations import Automations
+ from open_webui.models.users import Users
+ from open_webui.utils.automations import next_run_ns
+
+ user_id = __user__.get('id')
+ user = await Users.get_user_by_id(user_id)
+
+ automation = await Automations.get_by_id(automation_id)
+ if not automation:
+ return json.dumps({'error': 'Automation not found'})
+ if automation.user_id != user_id:
+ return json.dumps({'error': 'Access denied'})
+
+ rrule = automation.data.get('rrule', '')
+ toggled = await Automations.toggle(
+ automation_id,
+ next_run_ns(rrule, tz=user.timezone if user else None),
+ )
+
+ return json.dumps(
+ {
+ 'status': 'success',
+ 'id': toggled.id,
+ 'name': toggled.name,
+ 'is_active': toggled.is_active,
+ },
+ ensure_ascii=False,
+ )
+ except Exception as e:
+ log.exception(f'toggle_automation error: {e}')
+ return json.dumps({'error': str(e)})
+
+
+async def delete_automation(
+ automation_id: str,
+ __request__: Request = None,
+ __user__: dict = None,
+) -> str:
+ """
+ Delete a scheduled automation and all its run history.
+
+ :param automation_id: The ID of the automation to delete
+ :return: JSON confirming the automation was deleted
+ """
+ if __request__ is None:
+ return json.dumps({'error': 'Request context not available'})
+
+ if not __user__:
+ return json.dumps({'error': 'User context not available'})
+
+ try:
+ from open_webui.models.automations import Automations, AutomationRuns
+
+ user_id = __user__.get('id')
+
+ automation = await Automations.get_by_id(automation_id)
+ if not automation:
+ return json.dumps({'error': 'Automation not found'})
+ if automation.user_id != user_id:
+ return json.dumps({'error': 'Access denied'})
+
+ name = automation.name
+ await AutomationRuns.delete_by_automation(automation_id)
+ await Automations.delete(automation_id)
+
+ return json.dumps(
+ {
+ 'status': 'success',
+ 'message': f'Automation "{name}" deleted',
+ },
+ ensure_ascii=False,
+ )
+ except Exception as e:
+ log.exception(f'delete_automation error: {e}')
+ return json.dumps({'error': str(e)})
diff --git a/backend/open_webui/utils/access_control/__init__.py b/backend/open_webui/utils/access_control/__init__.py
index f31c59e158..41c888d441 100644
--- a/backend/open_webui/utils/access_control/__init__.py
+++ b/backend/open_webui/utils/access_control/__init__.py
@@ -11,7 +11,7 @@ from open_webui.models.access_grants import (
)
from open_webui.config import DEFAULT_USER_PERMISSIONS
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
def fill_missing_permissions(permissions: dict[str, Any], default_permissions: dict[str, Any]) -> dict[str, Any]:
@@ -28,10 +28,10 @@ def fill_missing_permissions(permissions: dict[str, Any], default_permissions: d
return permissions
-def get_permissions(
+async def get_permissions(
user_id: str,
default_permissions: dict[str, Any],
- db: Session | None = None,
+ db: AsyncSession | None = None,
) -> dict[str, Any]:
"""
Get all permissions for a user by combining the permissions of all groups the user is a member of.
@@ -53,7 +53,7 @@ def get_permissions(
permissions[key] = permissions[key] or value # Use the most permissive value (True > False)
return permissions
- user_groups = Groups.get_groups_by_member_id(user_id, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
# Deep copy default permissions to avoid modifying the original dict
permissions = json.loads(json.dumps(default_permissions))
@@ -68,11 +68,11 @@ def get_permissions(
return permissions
-def has_permission(
+async def has_permission(
user_id: str,
permission_key: str,
default_permissions: dict[str, Any] = {},
- db: Session | None = None,
+ db: AsyncSession | None = None,
) -> bool:
"""
Check if a user has a specific permission by checking the group permissions
@@ -93,7 +93,7 @@ def has_permission(
permission_hierarchy = permission_key.split('.')
# Retrieve user group permissions
- user_groups = Groups.get_groups_by_member_id(user_id, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
for group in user_groups:
if get_permission(group.permissions or {}, permission_hierarchy):
@@ -104,12 +104,12 @@ def has_permission(
return get_permission(default_permissions, permission_hierarchy)
-def has_access(
+async def has_access(
user_id: str,
permission: str = 'read',
access_grants: list | None = None,
user_group_ids: set[str] | None = None,
- db: Session | None = None,
+ db: AsyncSession | None = None,
) -> bool:
"""
Check if a user has the specified permission using an in-memory access_grants list.
@@ -126,7 +126,7 @@ def has_access(
return False
if user_group_ids is None:
- user_groups = Groups.get_groups_by_member_id(user_id, db=db)
+ user_groups = await Groups.get_groups_by_member_id(user_id, db=db)
user_group_ids = {group.id for group in user_groups}
for grant in access_grants:
@@ -144,7 +144,7 @@ def has_access(
return False
-def has_connection_access(
+async def has_connection_access(
user: UserModel,
connection: dict,
user_group_ids: set[str] | None = None,
@@ -163,10 +163,10 @@ def has_connection_access(
return True
if user_group_ids is None:
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
access_grants = (connection.get('config') or {}).get('access_grants', [])
- return has_access(user.id, 'read', access_grants, user_group_ids)
+ return await has_access(user.id, 'read', access_grants, user_group_ids)
def migrate_access_control(data: dict, ac_key: str = 'access_control', grants_key: str = 'access_grants') -> None:
@@ -210,13 +210,13 @@ def migrate_access_control(data: dict, ac_key: str = 'access_control', grants_ke
data.pop(ac_key, None)
-def filter_allowed_access_grants(
+async def filter_allowed_access_grants(
default_permissions: dict[str, Any],
user_id: str,
user_role: str,
access_grants: list,
public_permission_key: str,
- db: Session | None = None,
+ db: AsyncSession | None = None,
) -> list:
"""
Checks if the user has the required permissions to grant access to a resource.
@@ -228,7 +228,7 @@ def filter_allowed_access_grants(
# Check if user can share publicly
if (
has_public_read_access_grant(access_grants) or has_public_write_access_grant(access_grants)
- ) and not has_permission(
+ ) and not await has_permission(
user_id,
public_permission_key,
default_permissions,
@@ -246,7 +246,7 @@ def filter_allowed_access_grants(
]
# Strip individual user sharing if user lacks permission
- if has_user_access_grant(access_grants) and not has_permission(
+ if has_user_access_grant(access_grants) and not await has_permission(
user_id,
'access_grants.allow_users',
default_permissions,
@@ -255,3 +255,47 @@ def filter_allowed_access_grants(
access_grants = strip_user_access_grants(access_grants)
return access_grants
+
+
+async def check_model_access(
+ user: UserModel,
+ model_info,
+ bypass_filter: bool = False,
+) -> None:
+ """
+ Enforce per-model read access for the given user.
+
+ Raises HTTPException(403) if the user is not authorized.
+ Does nothing if bypass_filter is True.
+
+ Args:
+ user: The authenticated user.
+ model_info: The model record from await Models.get_model_by_id(),
+ or None if the model is not registered.
+ bypass_filter: If True, skip all access checks (used by
+ internal callers and BYPASS_MODEL_ACCESS_CONTROL).
+ """
+ from fastapi import HTTPException
+
+ if bypass_filter:
+ return
+
+ if model_info:
+ if user.role == 'user':
+ from open_webui.models.access_grants import AccessGrants
+
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
+ if not (
+ user.id == model_info.user_id
+ or await AccessGrants.has_access(
+ user_id=user.id,
+ resource_type='model',
+ resource_id=model_info.id,
+ permission='read',
+ user_group_ids=user_group_ids,
+ )
+ ):
+ raise HTTPException(status_code=403, detail='Model not found')
+ else:
+ if user.role != 'admin':
+ raise HTTPException(status_code=403, detail='Model not found')
diff --git a/backend/open_webui/utils/access_control/files.py b/backend/open_webui/utils/access_control/files.py
index a7e35fd506..5e7efb5b26 100644
--- a/backend/open_webui/utils/access_control/files.py
+++ b/backend/open_webui/utils/access_control/files.py
@@ -9,16 +9,16 @@ from open_webui.models.groups import Groups
from open_webui.models.models import Models
from open_webui.models.access_grants import AccessGrants
-from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
log = logging.getLogger(__name__)
-def has_access_to_file(
+async def has_access_to_file(
file_id: str | None,
access_type: str,
user: UserModel,
- db: Session | None = None,
+ db: AsyncSession | None = None,
) -> bool:
"""
Check if a user has the specified access to a file through any of:
@@ -30,7 +30,7 @@ def has_access_to_file(
NOTE: This does NOT check direct file ownership — callers should check
file.user_id == user.id separately before calling this.
"""
- file = Files.get_file_by_id(file_id, db=db)
+ file = await Files.get_file_by_id(file_id, db=db)
log.debug(f'Checking if user has {access_type} access to file')
if not file:
return False
@@ -40,10 +40,10 @@ def has_access_to_file(
return True
# Check if the file is associated with any knowledge bases the user has access to
- knowledge_bases = Knowledges.get_knowledges_by_file_id(file_id, db=db)
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
+ knowledge_bases = await Knowledges.get_knowledges_by_file_id(file_id, db=db)
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
for knowledge_base in knowledge_bases:
- if knowledge_base.user_id == user.id or AccessGrants.has_access(
+ if knowledge_base.user_id == user.id or await AccessGrants.has_access(
user_id=user.id,
resource_type='knowledge',
resource_id=knowledge_base.id,
@@ -55,24 +55,24 @@ def has_access_to_file(
knowledge_base_id = file.meta.get('collection_name') if file.meta else None
if knowledge_base_id:
- knowledge_bases = Knowledges.get_knowledge_bases_by_user_id(user.id, access_type, db=db)
+ knowledge_bases = await Knowledges.get_knowledge_bases_by_user_id(user.id, access_type, db=db)
for knowledge_base in knowledge_bases:
if knowledge_base.id == knowledge_base_id:
return True
# Check if the file is associated with any channels the user has access to
- channels = Channels.get_channels_by_file_id_and_user_id(file_id, user.id, db=db)
+ channels = await Channels.get_channels_by_file_id_and_user_id(file_id, user.id, db=db)
if access_type == 'read' and channels:
return True
# Check if the file is associated with any chats the user has access to
# TODO: Granular access control for chats
- chats = Chats.get_shared_chats_by_file_id(file_id, db=db)
+ chats = await Chats.get_shared_chats_by_file_id(file_id, db=db)
if chats:
return True
# Check if the file is directly attached to a shared workspace model
- for model in Models.get_models_by_user_id(user.id, permission=access_type, db=db):
+ for model in await Models.get_models_by_user_id(user.id, permission=access_type, db=db):
knowledge_items = getattr(model.meta, 'knowledge', None) or []
for item in knowledge_items:
if isinstance(item, dict) and item.get('type') == 'file' and item.get('id') == file.id:
diff --git a/backend/open_webui/utils/actions.py b/backend/open_webui/utils/actions.py
index 5c5712fa0f..7b1789580b 100644
--- a/backend/open_webui/utils/actions.py
+++ b/backend/open_webui/utils/actions.py
@@ -26,7 +26,7 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
else:
sub_action_id = None
- action = Functions.get_function_by_id(action_id)
+ action = await Functions.get_function_by_id(action_id)
if not action:
raise Exception(f'Action not found: {action_id}')
@@ -47,7 +47,7 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
raise Exception('Model not found')
model = models[model_id]
- __event_emitter__ = get_event_emitter(
+ __event_emitter__ = await get_event_emitter(
{
'chat_id': data['chat_id'],
'message_id': data['id'],
@@ -55,7 +55,7 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
'user_id': user.id,
}
)
- __event_call__ = get_event_call(
+ __event_call__ = await get_event_call(
{
'chat_id': data['chat_id'],
'message_id': data['id'],
@@ -64,10 +64,10 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
}
)
- function_module, _, _ = get_function_module_from_cache(request, action_id)
+ function_module, _, _ = await get_function_module_from_cache(request, action_id)
if hasattr(function_module, 'valves') and hasattr(function_module, 'Valves'):
- valves = Functions.get_function_valves_by_id(action_id)
+ valves = await Functions.get_function_valves_by_id(action_id)
function_module.valves = function_module.Valves(**(valves if valves else {}))
if hasattr(function_module, 'action'):
@@ -98,7 +98,7 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
try:
if hasattr(function_module, 'UserValves'):
__user__['valves'] = function_module.UserValves(
- **Functions.get_user_valves_by_id_and_user_id(action_id, user.id)
+ **await Functions.get_user_valves_by_id_and_user_id(action_id, user.id)
)
except Exception as e:
log.exception(f'Failed to get user values: {e}')
@@ -111,7 +111,7 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
data = action(**params)
# Process action result for Rich UI embeds (HTMLResponse, tuple with headers)
- processed_result, _, action_embeds = process_tool_result(
+ processed_result, _, action_embeds = await process_tool_result(
request,
action_id,
data,
diff --git a/backend/open_webui/utils/anthropic.py b/backend/open_webui/utils/anthropic.py
index 5ba4099fb4..a01184143f 100644
--- a/backend/open_webui/utils/anthropic.py
+++ b/backend/open_webui/utils/anthropic.py
@@ -181,17 +181,99 @@ def convert_anthropic_to_openai_payload(anthropic_payload: dict) -> dict:
)
elif block_type == 'tool_result':
# Tool results become separate tool messages in OpenAI format
- tool_content = block.get('content', '')
- if isinstance(tool_content, list):
- tool_text_parts = []
- for tc in tool_content:
- if isinstance(tc, dict) and tc.get('type') == 'text':
- tool_text_parts.append(tc.get('text', ''))
- tool_content = '\n'.join(tool_text_parts)
+ tool_result_content = block.get('content', '')
+ tool_content: str | list = ''
+
+ if isinstance(tool_result_content, str):
+ tool_content = tool_result_content
+ elif isinstance(tool_result_content, list):
+ # Build a multimodal content array to preserve
+ # images and other non-text content types.
+ converted_parts = []
+ for content_block in tool_result_content:
+ if not isinstance(content_block, dict):
+ continue
+ content_type = content_block.get('type', 'text')
+
+ if content_type == 'text':
+ converted_parts.append(
+ {
+ 'type': 'text',
+ 'text': content_block.get('text', ''),
+ }
+ )
+ elif content_type == 'image':
+ source = content_block.get('source', {})
+ if source.get('type') == 'base64':
+ media_type = source.get('media_type', 'image/png')
+ data = source.get('data', '')
+ converted_parts.append(
+ {
+ 'type': 'image_url',
+ 'image_url': {
+ 'url': f'data:{media_type};base64,{data}',
+ },
+ }
+ )
+ elif source.get('type') == 'url':
+ converted_parts.append(
+ {
+ 'type': 'image_url',
+ 'image_url': {
+ 'url': source.get('url', ''),
+ },
+ }
+ )
+ elif content_type == 'document':
+ # Documents have no direct OpenAI equivalent;
+ # convert to a text representation.
+ document_source = content_block.get('source', {})
+ document_title = content_block.get('title', 'Document')
+ document_context = content_block.get('context', '')
+ document_text = f'[Document: {document_title}]'
+ if document_context:
+ document_text += f'\n{document_context}'
+ if document_source.get('type') == 'text' and document_source.get('data'):
+ document_text += f'\n{document_source["data"]}'
+ converted_parts.append({'type': 'text', 'text': document_text})
+ elif content_type == 'search_result':
+ # Convert search results to a text
+ # representation with source attribution.
+ search_title = content_block.get('title', '')
+ search_url = content_block.get('source', '')
+ search_content_blocks = content_block.get('content', [])
+ search_texts = []
+ for search_block in search_content_blocks:
+ if isinstance(search_block, dict) and search_block.get('type') == 'text':
+ search_texts.append(search_block.get('text', ''))
+ search_body = '\n'.join(search_texts)
+ search_text = f'[Search Result: {search_title}]'
+ if search_url:
+ search_text += f'\nSource: {search_url}'
+ if search_body:
+ search_text += f'\n{search_body}'
+ converted_parts.append({'type': 'text', 'text': search_text})
+
+ # Flatten to string when only text parts are present
+ if all(part.get('type') == 'text' for part in converted_parts):
+ tool_content = '\n'.join(part.get('text', '') for part in converted_parts)
+ elif converted_parts:
+ tool_content = converted_parts
+ else:
+ tool_content = ''
# Propagate error status if present
if block.get('is_error'):
- tool_content = f'Error: {tool_content}'
+ if isinstance(tool_content, str):
+ tool_content = f'Error: {tool_content}'
+ elif isinstance(tool_content, list):
+ tool_content.insert(
+ 0,
+ {
+ 'type': 'text',
+ 'text': 'Error: ',
+ },
+ )
messages.append(
{
diff --git a/backend/open_webui/utils/asgi_middleware.py b/backend/open_webui/utils/asgi_middleware.py
new file mode 100644
index 0000000000..05389d8f94
--- /dev/null
+++ b/backend/open_webui/utils/asgi_middleware.py
@@ -0,0 +1,268 @@
+"""
+Pure-ASGI replacements for the project's previous
+`@app.middleware('http')` / `BaseHTTPMiddleware` middlewares.
+
+Why this matters
+----------------
+Starlette's `BaseHTTPMiddleware` (which `@app.middleware('http')` is
+sugar for) runs the downstream app inside an `anyio` task group. When
+the wrapper exits — for any reason: response complete, client
+disconnect, an outer middleware bailing out — the task group cancels
+the inner task. That `CancelledError` then propagates into whatever
+the inner task was doing, including in-flight DB queries, embedding
+calls and disk I/O.
+
+In Open WebUI this surfaces as:
+
+* SQLAlchemy logging multi-page `NotImplementedError:
+ terminate_force_close()` tracebacks at ERROR every time a request is
+ cancelled mid-DB-call (the aiosqlite connector cleanup path).
+* Spurious cancellations cascading through the four stacked
+ `@app.middleware('http')` wrappers.
+
+Pure ASGI middleware does not introduce a cancel scope around the
+downstream app, so client disconnects propagate the way ASGI was
+designed to (via `receive()` returning `http.disconnect`) instead of
+being injected as `CancelledError` into arbitrary `await` points.
+
+Reference: https://www.starlette.io/middleware/#limitations
+"""
+
+from __future__ import annotations
+
+import logging
+import re
+import time
+from urllib.parse import parse_qs, urlencode
+
+from fastapi.responses import JSONResponse, RedirectResponse
+from fastapi.security import HTTPAuthorizationCredentials
+from starlette.datastructures import MutableHeaders
+from starlette.requests import Request
+from starlette.types import ASGIApp, Message, Receive, Scope, Send
+
+from open_webui.internal.db import ScopedSession
+from open_webui.utils.auth import get_http_authorization_cred
+
+log = logging.getLogger(__name__)
+
+
+class CommitSessionMiddleware:
+ """Commit and release the thread-local sync `ScopedSession` after each
+ HTTP request.
+
+ Most requests now use the async session; the sync ScopedSession is
+ only touched by startup, healthchecks, and a handful of legacy
+ helpers (notably the pgvector / opengauss vector-DB clients). The
+ middleware exists so that PostgreSQL connections do not accumulate
+ as "idle in transaction" and so that any pending sync work made
+ inside the request is durably persisted.
+
+ Failure semantics
+ -----------------
+ * Downstream raised → roll back any pending sync work, release the
+ connection, and re-raise so the outer exception middleware can
+ turn it into an error response. We never commit work on a
+ request that did not complete successfully.
+ * Downstream returned → commit pending sync work; on commit
+ failure, log loudly, roll back, and re-raise. Note that in pure
+ ASGI the response messages have already been emitted by the
+ time `await self.app(...)` returns, so a commit failure cannot
+ retroactively change what the client sees on the wire — but
+ re-raising still surfaces the error in logs and to ASGI servers
+ that expose it. We deliberately do not buffer the response to
+ gate it on commit success, because that would defeat streaming
+ responses (chat completions, SSE) which are core to the app.
+
+ For request paths where commit-before-send is required, manage the
+ sync session explicitly inside the handler instead of relying on
+ this middleware.
+ """
+
+ def __init__(self, app: ASGIApp) -> None:
+ self.app = app
+
+ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+ if scope['type'] != 'http':
+ await self.app(scope, receive, send)
+ return
+
+ try:
+ await self.app(scope, receive, send)
+ except BaseException:
+ # Downstream did not complete successfully. Roll back any
+ # pending sync writes, release the connection, and let the
+ # exception propagate.
+ try:
+ ScopedSession.rollback()
+ except Exception:
+ log.exception('CommitSessionMiddleware: rollback failed after downstream error')
+ finally:
+ ScopedSession.remove()
+ raise
+
+ # Downstream completed. Commit pending sync work.
+ try:
+ ScopedSession.commit()
+ except Exception:
+ log.exception('CommitSessionMiddleware: post-request commit failed; response was already sent to client')
+ try:
+ ScopedSession.rollback()
+ except Exception:
+ log.exception('CommitSessionMiddleware: rollback failed after commit failure')
+ raise
+ finally:
+ # CRITICAL: remove() returns the connection to the pool.
+ # Without this, connections remain "checked out" and
+ # accumulate as "idle in transaction" in PostgreSQL.
+ ScopedSession.remove()
+
+
+class AuthTokenMiddleware:
+ """Extract the bearer/cookie/x-api-key credential and stash it on
+ `request.state.token`.
+
+ Routes that depend on `get_verified_user` etc. read this state.
+ Also exposes `request.state.enable_api_keys` (snapshotted at request
+ entry from runtime config) and stamps an `X-Process-Time` response
+ header.
+ """
+
+ def __init__(self, app: ASGIApp, *, fastapi_app) -> None:
+ self.app = app
+ self._fastapi_app = fastapi_app
+
+ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+ if scope['type'] != 'http':
+ await self.app(scope, receive, send)
+ return
+
+ start_time = time.monotonic()
+ request = Request(scope)
+
+ token = get_http_authorization_cred(request.headers.get('Authorization'))
+ if token is None:
+ cookie_token = request.cookies.get('token')
+ if cookie_token:
+ token = HTTPAuthorizationCredentials(scheme='Bearer', credentials=cookie_token)
+ if token is None:
+ api_key = request.headers.get('x-api-key')
+ if api_key:
+ token = HTTPAuthorizationCredentials(scheme='Bearer', credentials=api_key)
+
+ request.state.token = token
+ request.state.enable_api_keys = self._fastapi_app.state.config.ENABLE_API_KEYS
+
+ async def send_with_timing(message: Message) -> None:
+ if message['type'] == 'http.response.start':
+ process_time = int(time.monotonic() - start_time)
+ headers = MutableHeaders(scope=message)
+ headers['X-Process-Time'] = str(process_time)
+ await send(message)
+
+ await self.app(scope, receive, send_with_timing)
+
+
+class WebsocketUpgradeGuardMiddleware:
+ """Reject HTTP requests to `/ws/socket.io` that claim
+ `transport=websocket` but lack the proper `Upgrade`/`Connection`
+ headers.
+
+ Works around https://github.com/miguelgrinberg/python-engineio/issues/367
+ where engineio mishandles such requests.
+ """
+
+ def __init__(self, app: ASGIApp) -> None:
+ self.app = app
+
+ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+ if scope['type'] != 'http':
+ await self.app(scope, receive, send)
+ return
+
+ path = scope.get('path', '')
+ if '/ws/socket.io' in path:
+ query_string = scope.get('query_string', b'').decode('latin-1', errors='replace')
+ query_params = parse_qs(query_string)
+ if query_params.get('transport', [''])[0] == 'websocket':
+ headers = _scope_headers(scope)
+ upgrade = headers.get('upgrade', '').lower()
+ connection_tokens = [token.strip() for token in headers.get('connection', '').lower().split(',')]
+ if upgrade != 'websocket' or 'upgrade' not in connection_tokens:
+ response = JSONResponse(
+ status_code=400,
+ content={'detail': 'Invalid WebSocket upgrade request'},
+ )
+ await response(scope, receive, send)
+ return
+
+ await self.app(scope, receive, send)
+
+
+class RedirectMiddleware:
+ """Rewrites a couple of legacy entry-points to the SPA's own routes:
+
+ * ``GET /watch?v=ID`` (YouTube) → ``/?youtube=ID``
+ * ``GET /?shared=…`` (PWA share-target) → ``/?youtube=…`` /
+ ``/?load-url=…`` / ``/?q=…``
+ """
+
+ def __init__(self, app: ASGIApp) -> None:
+ self.app = app
+
+ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+ if scope['type'] != 'http' or scope.get('method', '').upper() != 'GET':
+ await self.app(scope, receive, send)
+ return
+
+ path = scope.get('path', '')
+ query_string = scope.get('query_string', b'').decode('latin-1', errors='replace')
+ query_params = parse_qs(query_string)
+
+ redirect_params: dict[str, str] = {}
+ if path.endswith('/watch') and 'v' in query_params and query_params['v']:
+ redirect_params['youtube'] = query_params['v'][0]
+
+ if 'shared' in query_params and query_params['shared']:
+ text = query_params['shared'][0]
+ if text:
+ url_match = re.match(r'https://\S+', text)
+ if url_match:
+ # Local import: youtube loader pulls heavy deps and is
+ # only needed when a share-target actually contains a
+ # YouTube URL.
+ from open_webui.retrieval.loaders.youtube import _parse_video_id
+
+ youtube_video_id = _parse_video_id(url_match[0])
+ if youtube_video_id:
+ redirect_params['youtube'] = youtube_video_id
+ else:
+ redirect_params['load-url'] = url_match[0]
+ else:
+ redirect_params['q'] = text
+
+ if redirect_params:
+ redirect_url = f'/?{urlencode(redirect_params)}'
+ response = RedirectResponse(url=redirect_url)
+ await response(scope, receive, send)
+ return
+
+ await self.app(scope, receive, send)
+
+
+def _scope_headers(scope: Scope) -> dict[str, str]:
+ """Return ASGI scope headers as a lower-cased str→str dict.
+
+ ASGI delivers headers as a list of (bytes, bytes) pairs. For
+ convenience, fold duplicate keys with comma-joining (matching
+ HTTP/1.1 semantics).
+ """
+ decoded: dict[str, str] = {}
+ for raw_key, raw_value in scope.get('headers', []):
+ key = raw_key.decode('latin-1').lower()
+ value = raw_value.decode('latin-1')
+ if key in decoded:
+ decoded[key] = f'{decoded[key]}, {value}'
+ else:
+ decoded[key] = value
+ return decoded
diff --git a/backend/open_webui/utils/audit.py b/backend/open_webui/utils/audit.py
index 1200d813af..5686c88d5d 100644
--- a/backend/open_webui/utils/audit.py
+++ b/backend/open_webui/utils/audit.py
@@ -24,7 +24,7 @@ from asgiref.typing import (
from loguru import logger
from starlette.requests import Request
-from open_webui.env import AUDIT_LOG_LEVEL, AUDIT_INCLUDED_PATHS, MAX_BODY_LOG_SIZE
+from open_webui.env import AUDIT_LOG_LEVEL, ENABLE_AUDIT_GET_REQUESTS, AUDIT_INCLUDED_PATHS, MAX_BODY_LOG_SIZE
from open_webui.utils.auth import get_current_user, get_http_authorization_cred
from open_webui.models.users import UserModel
@@ -117,7 +117,7 @@ class AuditLoggingMiddleware:
ASGI middleware that intercepts HTTP requests and responses to perform audit logging. It captures request/response bodies (depending on audit level), headers, HTTP methods, and user information, then logs a structured audit entry at the end of the request cycle.
"""
- AUDITED_METHODS = {'PUT', 'PATCH', 'DELETE', 'POST'}
+ DEFAULT_AUDITED_METHODS = {'PUT', 'PATCH', 'DELETE', 'POST'}
def __init__(
self,
@@ -127,12 +127,16 @@ class AuditLoggingMiddleware:
included_paths: Optional[list[str]] = None,
max_body_size: int = MAX_BODY_LOG_SIZE,
audit_level: AuditLevel = AuditLevel.NONE,
+ audit_get_requests: bool = False,
) -> None:
self.app = app
self.audit_logger = AuditLogger(logger)
self.excluded_paths = excluded_paths or []
self.included_paths = included_paths or []
self.max_body_size = max_body_size
+ self.audited_methods = set(self.DEFAULT_AUDITED_METHODS)
+ if audit_get_requests:
+ self.audited_methods.add('GET')
self.audit_level = audit_level
if self.included_paths and self.excluded_paths:
@@ -202,7 +206,10 @@ class AuditLoggingMiddleware:
return None
def _should_skip_auditing(self, request: Request) -> bool:
- if request.method not in {'POST', 'PUT', 'PATCH', 'DELETE'} or AUDIT_LOG_LEVEL == 'NONE':
+ if AUDIT_LOG_LEVEL == 'NONE':
+ return True
+
+ if request.method not in self.audited_methods:
return True
ALWAYS_LOG_ENDPOINTS = {
diff --git a/backend/open_webui/utils/auth.py b/backend/open_webui/utils/auth.py
index 16bc36500b..e0f331a9df 100644
--- a/backend/open_webui/utils/auth.py
+++ b/backend/open_webui/utils/auth.py
@@ -19,8 +19,6 @@ import pytz
from pytz import UTC
from typing import Optional, Union, List, Dict
-from opentelemetry import trace
-
from open_webui.utils.access_control import has_permission
from open_webui.models.users import Users
@@ -30,6 +28,7 @@ from open_webui.models.auths import Auths
from open_webui.constants import ERROR_MESSAGES
from open_webui.env import (
+ ENABLE_OTEL,
ENABLE_PASSWORD_VALIDATION,
OFFLINE_MODE,
LICENSE_BLOB,
@@ -238,9 +237,7 @@ async def is_valid_token(request, decoded) -> bool:
# Per-user revocation (OIDC back-channel logout)
user_id = decoded.get('id')
if user_id:
- revoked_at = await request.app.state.redis.get(
- f'{REDIS_KEY_PREFIX}:auth:user:{user_id}:revoked_at'
- )
+ revoked_at = await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:auth:user:{user_id}:revoked_at')
if revoked_at:
try:
revoked_at_ts = int(revoked_at)
@@ -324,15 +321,18 @@ async def get_current_user(
# auth by api key
if token.startswith('sk-'):
- user = get_current_user_by_api_key(request, token)
+ user = await get_current_user_by_api_key(request, token)
# Add user info to current span
- current_span = trace.get_current_span()
- if current_span:
- current_span.set_attribute('client.user.id', user.id)
- current_span.set_attribute('client.user.email', user.email)
- current_span.set_attribute('client.user.role', user.role)
- current_span.set_attribute('client.auth.type', 'api_key')
+ if ENABLE_OTEL:
+ from opentelemetry import trace
+
+ current_span = trace.get_current_span()
+ if current_span:
+ current_span.set_attribute('client.user.id', user.id)
+ current_span.set_attribute('client.user.email', user.email)
+ current_span.set_attribute('client.user.role', user.role)
+ current_span.set_attribute('client.auth.type', 'api_key')
return user
@@ -353,7 +353,7 @@ async def get_current_user(
detail='Invalid token',
)
- user = Users.get_user_by_id(data['id'])
+ user = await Users.get_user_by_id(data['id'])
if user is None:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -369,17 +369,21 @@ async def get_current_user(
)
# Add user info to current span
- current_span = trace.get_current_span()
- if current_span:
- current_span.set_attribute('client.user.id', user.id)
- current_span.set_attribute('client.user.email', user.email)
- current_span.set_attribute('client.user.role', user.role)
- current_span.set_attribute('client.auth.type', 'jwt')
+ if ENABLE_OTEL:
+ from opentelemetry import trace
- # Refresh the user's last active timestamp asynchronously
- # to prevent blocking the request
- if background_tasks:
- background_tasks.add_task(Users.update_last_active_by_id, user.id)
+ current_span = trace.get_current_span()
+ if current_span:
+ current_span.set_attribute('client.user.id', user.id)
+ current_span.set_attribute('client.user.email', user.email)
+ current_span.set_attribute('client.user.role', user.role)
+ current_span.set_attribute('client.auth.type', 'jwt')
+
+ # Refresh the user's last active timestamp
+ # Fire-and-forget via asyncio.create_task to avoid blocking
+ import asyncio
+
+ asyncio.create_task(Users.update_last_active_by_id(user.id))
return user
else:
raise HTTPException(
@@ -401,9 +405,9 @@ async def get_current_user(
raise e
-def get_current_user_by_api_key(request, api_key: str):
+async def get_current_user_by_api_key(request, api_key: str):
# Each function call manages its own short-lived session internally
- user = Users.get_user_by_api_key(api_key)
+ user = await Users.get_user_by_api_key(api_key)
if user is None:
raise HTTPException(
@@ -413,7 +417,7 @@ def get_current_user_by_api_key(request, api_key: str):
if not request.state.enable_api_keys or (
user.role != 'admin'
- and not has_permission(
+ and not await has_permission(
user.id,
'features.api_keys',
request.app.state.config.USER_PERMISSIONS,
@@ -421,15 +425,33 @@ def get_current_user_by_api_key(request, api_key: str):
):
raise HTTPException(status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.API_KEY_NOT_ALLOWED)
- # Add user info to current span
- current_span = trace.get_current_span()
- if current_span:
- current_span.set_attribute('client.user.id', user.id)
- current_span.set_attribute('client.user.email', user.email)
- current_span.set_attribute('client.user.role', user.role)
- current_span.set_attribute('client.auth.type', 'api_key')
+ # Enforce endpoint restrictions — checked here (not in middleware)
+ # so it applies regardless of how the API key was transported
+ # (Authorization header, cookie, x-api-key header, etc.).
+ if request.app.state.config.ENABLE_API_KEYS_ENDPOINT_RESTRICTIONS:
+ allowed_paths = [
+ path.strip() for path in str(request.app.state.config.API_KEYS_ALLOWED_ENDPOINTS).split(',') if path.strip()
+ ]
+ request_path = request.url.path
+ is_allowed = any(request_path == allowed or request_path.startswith(allowed + '/') for allowed in allowed_paths)
+ if not is_allowed:
+ raise HTTPException(
+ status_code=status.HTTP_403_FORBIDDEN,
+ detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
+ )
- Users.update_last_active_by_id(user.id)
+ # Add user info to current span
+ if ENABLE_OTEL:
+ from opentelemetry import trace
+
+ current_span = trace.get_current_span()
+ if current_span:
+ current_span.set_attribute('client.user.id', user.id)
+ current_span.set_attribute('client.user.email', user.email)
+ current_span.set_attribute('client.user.role', user.role)
+ current_span.set_attribute('client.auth.type', 'api_key')
+
+ await Users.update_last_active_by_id(user.id)
return user
@@ -451,7 +473,7 @@ def get_admin_user(user=Depends(get_current_user)):
return user
-def create_admin_user(email: str, password: str, name: str = 'Admin'):
+async def create_admin_user(email: str, password: str, name: str = 'Admin'):
"""
Create an admin user from environment variables.
Used for headless/automated deployments.
@@ -461,14 +483,14 @@ def create_admin_user(email: str, password: str, name: str = 'Admin'):
if not email or not password:
return None
- if Users.has_users():
+ if await Users.has_users():
log.debug('Users already exist, skipping admin creation')
return None
log.info(f'Creating admin account from environment variables: {email}')
try:
hashed = get_password_hash(password)
- user = Auths.insert_new_auth(
+ user = await Auths.insert_new_auth(
email=email.lower(),
password=hashed,
name=name,
diff --git a/backend/open_webui/utils/automations.py b/backend/open_webui/utils/automations.py
index 4d9eb2fb6c..3866eb865a 100644
--- a/backend/open_webui/utils/automations.py
+++ b/backend/open_webui/utils/automations.py
@@ -22,11 +22,12 @@ from dateutil.rrule import rrulestr
from fastapi import Request
from starlette.datastructures import Headers
+from open_webui.constants import ERROR_MESSAGES
from open_webui.models.automations import Automations, AutomationRuns, AutomationModel
from open_webui.models.chats import ChatForm, Chats
from open_webui.models.users import Users
from open_webui.utils.task import prompt_template
-from open_webui.internal.db import get_db
+from open_webui.internal.db import get_async_db
log = logging.getLogger(__name__)
@@ -59,9 +60,9 @@ def validate_rrule(s: str) -> None:
try:
rule = _parse_rule(s)
except Exception as e:
- raise ValueError(f'Invalid RRULE: {e}')
+ raise ValueError(ERROR_MESSAGES.AUTOMATION_INVALID_RRULE(e))
if rule.after(datetime.now()) is None:
- raise ValueError('RRULE has no future occurrences')
+ raise ValueError(ERROR_MESSAGES.AUTOMATION_NO_FUTURE_RUNS)
def next_run_ns(s: str, tz: str = None) -> Optional[int]:
@@ -92,6 +93,25 @@ def next_n_runs_ns(s: str, n: int = 5, tz: str = None) -> list[int]:
return result
+def rrule_interval_seconds(s: str) -> Optional[int]:
+ """Approximate interval between recurrences in seconds.
+
+ Returns None for one-shot (COUNT=1) schedules or rules
+ with fewer than two future occurrences.
+ """
+ if 'COUNT=1' in s:
+ return None
+ rule = _parse_rule(s)
+ now = datetime.now()
+ first = rule.after(now)
+ if first is None:
+ return None
+ second = rule.after(first)
+ if second is None:
+ return None
+ return int((second - first).total_seconds())
+
+
############################
# Worker Loop
############################
@@ -106,8 +126,8 @@ async def automation_worker_loop(app) -> None:
log.info(f'Automation worker started (poll interval: {AUTOMATION_POLL_INTERVAL}s)')
while True:
try:
- with get_db() as db:
- batch = Automations.claim_due(int(time.time_ns()), limit=10, db=db)
+ async with get_async_db() as db:
+ batch = await Automations.claim_due(int(time.time_ns()), limit=10, db=db)
if batch:
log.info(f'Claimed {len(batch)} due automation(s)')
for automation in batch:
@@ -205,6 +225,16 @@ def _resolve_model_filter_ids(app, model_id: str) -> list[str]:
return list(filter_ids) if filter_ids else []
+def _resolve_model_terminal_id(app, model_id: str) -> Optional[str]:
+ """Read model default terminal_id from model config.
+
+ The frontend does this in Chat.svelte (model.info.meta.terminalId).
+ """
+ models = getattr(app.state, 'MODELS', {})
+ model = models.get(model_id, {})
+ return model.get('info', {}).get('meta', {}).get('terminalId') or None
+
+
async def _set_terminal_cwd(app, server_id: str, user, cwd: str, chat_id: str) -> None:
"""Set the working directory on a terminal server via the proxy.
@@ -264,9 +294,9 @@ async def execute_automation(app, automation: AutomationModel) -> None:
(filters, model params, knowledge/RAG, tools, DB saves, webhooks).
"""
try:
- user = Users.get_user_by_id(automation.user_id)
+ user = await Users.get_user_by_id(automation.user_id)
if not user:
- _record_run(automation.id, 'error', error='User not found')
+ await _record_run(automation.id, 'error', error='User not found')
return
prompt = prompt_template(automation.data['prompt'], user)
@@ -277,8 +307,9 @@ async def execute_automation(app, automation: AutomationModel) -> None:
user_msg_id = str(uuid4())
assistant_msg_id = str(uuid4())
- # Create the chat with user message (same structure as frontend)
- chat = Chats.insert_new_chat(
+ chat_id = str(uuid4())
+ chat = await Chats.insert_new_chat(
+ chat_id,
automation.user_id,
ChatForm(
chat={
@@ -317,7 +348,7 @@ async def execute_automation(app, automation: AutomationModel) -> None:
)
if not chat:
- _record_run(automation.id, 'error', error='Failed to create chat')
+ await _record_run(automation.id, 'error', error='Failed to create chat')
return
# Notify frontend to refresh chat list
@@ -338,13 +369,8 @@ async def execute_automation(app, automation: AutomationModel) -> None:
features = _resolve_model_features(app, model_id)
filter_ids = _resolve_model_filter_ids(app, model_id)
- # If a terminal is linked, set the CWD before building the payload
- terminal_id = None
- if terminal_config and terminal_config.get('server_id'):
- terminal_id = terminal_config['server_id']
- cwd = terminal_config.get('cwd')
- if cwd:
- await _set_terminal_cwd(app, terminal_id, user, cwd, chat.id)
+ # Resolve terminal from model config
+ terminal_id = _resolve_model_terminal_id(app, model_id)
# Build the same payload the frontend sends to /api/chat/completions
form_data = {
@@ -353,7 +379,13 @@ async def execute_automation(app, automation: AutomationModel) -> None:
'stream': True,
'chat_id': chat.id,
'id': assistant_msg_id,
- 'parent_id': user_msg_id,
+ 'parent_id': None, # Root message (chat already created above)
+ 'user_message': {
+ 'id': user_msg_id,
+ 'parentId': None,
+ 'role': 'user',
+ 'content': prompt,
+ },
'session_id': f'automation:{automation.id}',
'background_tasks': {},
}
@@ -385,11 +417,11 @@ async def execute_automation(app, automation: AutomationModel) -> None:
room=f'user:{automation.user_id}',
)
- _record_run(automation.id, 'success', chat_id=chat.id)
+ await _record_run(automation.id, 'success', chat_id=chat.id)
except Exception as e:
log.exception(f'Automation {automation.id} failed')
- _record_run(automation.id, 'error', error=str(e)[:4000])
+ await _record_run(automation.id, 'error', error=str(e)[:4000])
####################
@@ -397,12 +429,12 @@ async def execute_automation(app, automation: AutomationModel) -> None:
####################
-def _record_run(
+async def _record_run(
automation_id: str,
status: str,
chat_id: str = None,
error: str = None,
):
"""Insert a run record into automation_run."""
- with get_db() as db:
- AutomationRuns.insert(automation_id, status, chat_id=chat_id, error=error, db=db)
+ async with get_async_db() as db:
+ await AutomationRuns.insert(automation_id, status, chat_id=chat_id, error=error, db=db)
diff --git a/backend/open_webui/utils/chat.py b/backend/open_webui/utils/chat.py
index 9a9e810331..3539d57c86 100644
--- a/backend/open_webui/utils/chat.py
+++ b/backend/open_webui/utils/chat.py
@@ -72,7 +72,7 @@ async def generate_direct_chat_completion(
session_id = metadata.get('session_id')
request_id = str(uuid.uuid4()) # Generate a unique request ID
- event_caller = get_event_call(metadata)
+ event_caller = await get_event_call(metadata)
channel = f'{user_id}:{session_id}:{request_id}'
logging.info(f'WebSocket channel: {channel}')
@@ -199,7 +199,7 @@ async def generate_chat_completion(
# Check if user has access to the model
if not bypass_filter and user.role == 'user':
try:
- check_model_access(user, model)
+ await check_model_access(user, model)
except Exception as e:
raise e
@@ -343,8 +343,8 @@ async def chat_completed(request: Request, form_data: dict, user: Any):
}
extra_params = {
- '__event_emitter__': get_event_emitter(metadata),
- '__event_call__': get_event_call(metadata),
+ '__event_emitter__': await get_event_emitter(metadata),
+ '__event_call__': await get_event_call(metadata),
'__user__': user.model_dump() if isinstance(user, UserModel) else {},
'__metadata__': metadata,
'__request__': request,
@@ -352,8 +352,8 @@ async def chat_completed(request: Request, form_data: dict, user: Any):
}
try:
- filter_ids = get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
- filter_functions = Functions.get_functions_by_ids(filter_ids)
+ filter_ids = await get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
+ filter_functions = await Functions.get_functions_by_ids(filter_ids)
result, _ = await process_filter_functions(
request=request,
diff --git a/backend/open_webui/utils/embeddings.py b/backend/open_webui/utils/embeddings.py
index 251b5edf7e..1717886326 100644
--- a/backend/open_webui/utils/embeddings.py
+++ b/backend/open_webui/utils/embeddings.py
@@ -68,7 +68,7 @@ async def generate_embeddings(
# Access filtering
if not getattr(request.state, 'direct', False):
if not bypass_filter and user.role == 'user':
- check_model_access(user, model)
+ await check_model_access(user, model)
# Ollama backend — use /api/embed which supports batch input natively
if model.get('owned_by') == 'ollama':
diff --git a/backend/open_webui/utils/files.py b/backend/open_webui/utils/files.py
index 06bec33250..eea3a8b486 100644
--- a/backend/open_webui/utils/files.py
+++ b/backend/open_webui/utils/files.py
@@ -20,36 +20,39 @@ from open_webui.models.files import Files
from open_webui.routers.files import upload_file_handler
from open_webui.retrieval.web.utils import validate_url
+import asyncio
import mimetypes
import base64
import io
import re
-import requests
+from open_webui.env import AIOHTTP_CLIENT_SESSION_SSL
+from open_webui.utils.session_pool import get_session
BASE64_IMAGE_URL_PREFIX = re.compile(r'data:image/\w+;base64,', re.IGNORECASE)
MARKDOWN_IMAGE_URL_PATTERN = re.compile(r'!\[(.*?)\]\((.+?)\)', re.IGNORECASE)
-def get_image_base64_from_url(url: str) -> Optional[str]:
+async def get_image_base64_from_url(url: str) -> Optional[str]:
try:
if url.startswith('http'):
# Validate URL to prevent SSRF attacks against local/private networks
validate_url(url)
# Download the image from the URL
- response = requests.get(url)
- response.raise_for_status()
- image_data = response.content
- encoded_string = base64.b64encode(image_data).decode('utf-8')
- content_type = response.headers.get('Content-Type', 'image/png')
- return f'data:{content_type};base64,{encoded_string}'
+ session = await get_session()
+ async with session.get(url, ssl=AIOHTTP_CLIENT_SESSION_SSL) as response:
+ response.raise_for_status()
+ image_data = await response.read()
+ encoded_string = base64.b64encode(image_data).decode('utf-8')
+ content_type = response.headers.get('Content-Type', 'image/png')
+ return f'data:{content_type};base64,{encoded_string}'
else:
- file = Files.get_file_by_id(url)
+ file = await Files.get_file_by_id(url)
if not file:
return None
- file_path = Storage.get_file(file.path)
+ file_path = await asyncio.to_thread(Storage.get_file, file.path)
file_path = Path(file_path)
if file_path.is_file():
@@ -64,13 +67,13 @@ def get_image_base64_from_url(url: str) -> Optional[str]:
return None
-def get_image_url_from_base64(request, base64_image_string, metadata, user):
+async def get_image_url_from_base64(request, base64_image_string, metadata, user):
if BASE64_IMAGE_URL_PREFIX.match(base64_image_string):
image_url = ''
# Extract base64 image data from the line
image_data, content_type = get_image_data(base64_image_string)
if image_data is not None:
- _, image_url = upload_image(
+ _, image_url = await upload_image(
request,
image_data,
content_type,
@@ -82,17 +85,26 @@ def get_image_url_from_base64(request, base64_image_string, metadata, user):
return None
-def convert_markdown_base64_images(request, content: str, metadata, user):
- def replace(match):
- base64_string = match.group(2)
- MIN_REPLACEMENT_URL_LENGTH = 1024
- if len(base64_string) > MIN_REPLACEMENT_URL_LENGTH:
- url = get_image_url_from_base64(request, base64_string, metadata, user)
- if url:
- return f''
- return match.group(0)
+async def convert_markdown_base64_images(request, content: str, metadata, user):
+ MIN_REPLACEMENT_URL_LENGTH = 1024
+ result_parts = []
+ last_end = 0
- return MARKDOWN_IMAGE_URL_PATTERN.sub(replace, content)
+ for match in MARKDOWN_IMAGE_URL_PATTERN.finditer(content):
+ result_parts.append(content[last_end : match.start()])
+ base64_string = match.group(2)
+ if len(base64_string) > MIN_REPLACEMENT_URL_LENGTH:
+ url = await get_image_url_from_base64(request, base64_string, metadata, user)
+ if url:
+ result_parts.append(f'')
+ else:
+ result_parts.append(match.group(0))
+ else:
+ result_parts.append(match.group(0))
+ last_end = match.end()
+
+ result_parts.append(content[last_end:])
+ return ''.join(result_parts)
def load_b64_audio_data(b64_str):
@@ -110,7 +122,7 @@ def load_b64_audio_data(b64_str):
return None, None
-def upload_audio(request, audio_data, content_type, metadata, user):
+async def upload_audio(request, audio_data, content_type, metadata, user):
audio_format = mimetypes.guess_extension(content_type)
file = UploadFile(
file=io.BytesIO(audio_data),
@@ -119,7 +131,7 @@ def upload_audio(request, audio_data, content_type, metadata, user):
'content-type': content_type,
},
)
- file_item = upload_file_handler(
+ file_item = await upload_file_handler(
request,
file=file,
metadata=metadata,
@@ -130,13 +142,13 @@ def upload_audio(request, audio_data, content_type, metadata, user):
return url
-def get_audio_url_from_base64(request, base64_audio_string, metadata, user):
+async def get_audio_url_from_base64(request, base64_audio_string, metadata, user):
if 'data:audio/wav;base64' in base64_audio_string:
audio_url = ''
# Extract base64 audio data from the line
audio_data, content_type = load_b64_audio_data(base64_audio_string)
if audio_data is not None:
- audio_url = upload_audio(
+ audio_url = await upload_audio(
request,
audio_data,
content_type,
@@ -147,21 +159,21 @@ def get_audio_url_from_base64(request, base64_audio_string, metadata, user):
return None
-def get_file_url_from_base64(request, base64_file_string, metadata, user):
+async def get_file_url_from_base64(request, base64_file_string, metadata, user):
if BASE64_IMAGE_URL_PREFIX.match(base64_file_string):
- return get_image_url_from_base64(request, base64_file_string, metadata, user)
+ return await get_image_url_from_base64(request, base64_file_string, metadata, user)
elif 'data:audio/wav;base64' in base64_file_string:
- return get_audio_url_from_base64(request, base64_file_string, metadata, user)
+ return await get_audio_url_from_base64(request, base64_file_string, metadata, user)
return None
-def get_image_base64_from_file_id(id: str) -> Optional[str]:
- file = Files.get_file_by_id(id)
+async def get_image_base64_from_file_id(id: str) -> Optional[str]:
+ file = await Files.get_file_by_id(id)
if not file:
return None
try:
- file_path = Storage.get_file(file.path)
+ file_path = await asyncio.to_thread(Storage.get_file, file.path)
file_path = Path(file_path)
# Check if the file already exists in the cache
diff --git a/backend/open_webui/utils/filter.py b/backend/open_webui/utils/filter.py
index df07dea4a1..50b1583088 100644
--- a/backend/open_webui/utils/filter.py
+++ b/backend/open_webui/utils/filter.py
@@ -10,44 +10,53 @@ from open_webui.models.functions import Functions
log = logging.getLogger(__name__)
-def get_function_module(request, function_id, load_from_db=True):
+async def get_function_module(request, function_id, load_from_db=True):
"""
Get the function module by its ID.
"""
- function_module, _, _ = get_function_module_from_cache(request, function_id, load_from_db)
+ function_module, _, _ = await get_function_module_from_cache(request, function_id, load_from_db)
return function_module
-def get_sorted_filter_ids(request, model: dict, enabled_filter_ids: list = None):
- def get_priority(function_id):
+async def get_sorted_filter_ids(request, model: dict, enabled_filter_ids: list = None):
+ async def get_priority(function_id):
try:
- function_module = get_function_module(request, function_id)
+ function_module = await get_function_module(request, function_id)
if function_module and hasattr(function_module, 'Valves'):
- valves_db = Functions.get_function_valves_by_id(function_id)
+ valves_db = await Functions.get_function_valves_by_id(function_id)
valves = function_module.Valves(**(valves_db if valves_db else {}))
return getattr(valves, 'priority', 0)
except Exception:
pass
return 0
- filter_ids = [function.id for function in Functions.get_global_filter_functions()]
+ filter_ids = [function.id for function in await Functions.get_global_filter_functions()]
if 'info' in model and 'meta' in model['info']:
filter_ids.extend(model['info']['meta'].get('filterIds', []))
filter_ids = list(set(filter_ids))
- active_filter_ids = {function.id for function in Functions.get_functions_by_type('filter', active_only=True)}
+ active_filter_ids = {function.id for function in await Functions.get_functions_by_type('filter', active_only=True)}
- def get_active_status(filter_id):
- function_module = get_function_module(request, filter_id)
+ async def get_active_status(filter_id):
+ function_module = await get_function_module(request, filter_id)
if getattr(function_module, 'toggle', None):
return filter_id in (enabled_filter_ids or set())
return True
- active_filter_ids = {filter_id for filter_id in active_filter_ids if get_active_status(filter_id)}
+ # Pre-compute active status for each filter (async functions can't be used in set comprehensions)
+ resolved_active = {}
+ for filter_id in active_filter_ids:
+ resolved_active[filter_id] = await get_active_status(filter_id)
+ active_filter_ids = {fid for fid, is_active in resolved_active.items() if is_active}
filter_ids = [fid for fid in filter_ids if fid in active_filter_ids]
- filter_ids.sort(key=lambda fid: (get_priority(fid), fid))
+
+ # Pre-compute priorities (async functions can't be used in sort keys)
+ priorities = {}
+ for fid in filter_ids:
+ priorities[fid] = await get_priority(fid)
+ filter_ids.sort(key=lambda fid: (priorities.get(fid, 0), fid))
return filter_ids
@@ -63,7 +72,7 @@ async def process_filter_functions(request, filter_functions, filter_type, form_
if not filter:
continue
- function_module = get_function_module(request, filter_id, load_from_db=(filter_type != 'stream'))
+ function_module = await get_function_module(request, filter_id, load_from_db=(filter_type != 'stream'))
# Prepare handler function
handler = getattr(function_module, filter_type, None)
if not handler:
@@ -75,7 +84,7 @@ async def process_filter_functions(request, filter_functions, filter_type, form_
# Apply valves to the function
if hasattr(function_module, 'valves') and hasattr(function_module, 'Valves'):
- valves = Functions.get_function_valves_by_id(filter_id)
+ valves = await Functions.get_function_valves_by_id(filter_id)
function_module.valves = function_module.Valves(**(valves if valves else {}))
try:
@@ -100,7 +109,7 @@ async def process_filter_functions(request, filter_functions, filter_type, form_
if hasattr(function_module, 'UserValves'):
try:
params['__user__']['valves'] = function_module.UserValves(
- **Functions.get_user_valves_by_id_and_user_id(filter_id, params['__user__']['id'])
+ **await Functions.get_user_valves_by_id_and_user_id(filter_id, params['__user__']['id'])
)
except Exception as e:
log.exception(f'Failed to get user values: {e}')
diff --git a/backend/open_webui/utils/groups.py b/backend/open_webui/utils/groups.py
index 90c4593cec..50099b2ee7 100644
--- a/backend/open_webui/utils/groups.py
+++ b/backend/open_webui/utils/groups.py
@@ -4,7 +4,7 @@ from open_webui.models.groups import Groups
log = logging.getLogger(__name__)
-def apply_default_group_assignment(
+async def apply_default_group_assignment(
default_group_id: str,
user_id: str,
db=None,
@@ -18,6 +18,6 @@ def apply_default_group_assignment(
"""
if default_group_id:
try:
- Groups.add_users_to_group(default_group_id, [user_id], db=db)
+ await Groups.add_users_to_group(default_group_id, [user_id], db=db)
except Exception as e:
log.error(f'Failed to add user {user_id} to default group {default_group_id}: {e}')
diff --git a/backend/open_webui/utils/logger.py b/backend/open_webui/utils/logger.py
index 49b7973c57..fa4e77f53d 100644
--- a/backend/open_webui/utils/logger.py
+++ b/backend/open_webui/utils/logger.py
@@ -4,7 +4,7 @@ import sys
from typing import TYPE_CHECKING
from loguru import logger
-from opentelemetry import trace
+
from open_webui.env import (
ENABLE_AUDIT_STDOUT,
ENABLE_AUDIT_LOGS_FILE,
@@ -100,6 +100,8 @@ class InterceptHandler(logging.Handler):
if not ENABLE_OTEL:
return {}
+ from opentelemetry import trace
+
extras = {}
context = trace.get_current_span().get_span_context()
if context.is_valid:
diff --git a/backend/open_webui/utils/mcp/client.py b/backend/open_webui/utils/mcp/client.py
index beb2f15079..effe4b1637 100644
--- a/backend/open_webui/utils/mcp/client.py
+++ b/backend/open_webui/utils/mcp/client.py
@@ -44,7 +44,7 @@ def create_httpx_client(headers=None, timeout=None, auth=None):
return _build_httpx_client(headers=headers, timeout=timeout, auth=auth, verify=True)
-def create_insecure_httpx_client(headers=None, timeout=None, auth=None):
+async def create_insecure_httpx_client(headers=None, timeout=None, auth=None):
return _build_httpx_client(headers=headers, timeout=timeout, auth=auth, verify=False)
diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index 0dedd7f2f6..0d330ff31f 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -444,7 +444,7 @@ def serialize_output(output: list) -> str:
files = result_item.get('files')
embeds = result_item.get('embeds', '')
- content += f'\nTool Executed \n \n'
+ content += f'\nTool Executed \n{html.escape(json.dumps(result_text, ensure_ascii=False))}\n \n'
else:
content += f'\nExecuting... \n \n'
@@ -889,10 +889,14 @@ def get_source_context(sources: list, source_ids: dict = None, include_content:
if source_id not in source_ids:
source_ids[source_id] = len(source_ids) + 1
src_name = source.get('source', {}).get('name')
+ src_type = source.get('source', {}).get('type')
+ src_rid = source.get('source', {}).get('id')
body = doc if include_content else ''
context_string += (
f'{body} \n'
)
return context_string
@@ -936,7 +940,7 @@ def apply_source_context_to_messages(
)
-def process_tool_result(
+async def process_tool_result(
request,
tool_function_name,
tool_result,
@@ -1075,7 +1079,7 @@ def process_tool_result(
pass
tool_response.append(text)
elif item.get('type') in ['image', 'audio']:
- file_url = get_file_url_from_base64(
+ file_url = await get_file_url_from_base64(
request,
f'data:{item.get("mimeType")};base64,{item.get("data", item.get("blob", ""))}',
{
@@ -1304,7 +1308,7 @@ async def chat_completion_tools_handler(
except Exception as e:
tool_result = str(e)
- tool_result, tool_result_files, tool_result_embeds = process_tool_result(
+ tool_result, tool_result_files, tool_result_embeds = await process_tool_result(
request,
tool_function_name,
tool_result,
@@ -1602,7 +1606,7 @@ def get_images_from_messages(message_list):
return images
-def get_image_urls(delta_images, request, metadata, user) -> list[str]:
+async def get_image_urls(delta_images, request, metadata, user) -> list[str]:
if not isinstance(delta_images, list):
return []
@@ -1616,21 +1620,21 @@ def get_image_urls(delta_images, request, metadata, user) -> list[str]:
continue
if url.startswith('data:image/png;base64'):
- url = get_image_url_from_base64(request, url, metadata, user)
+ url = await get_image_url_from_base64(request, url, metadata, user)
image_urls.append(url)
return image_urls
-def add_file_context(messages: list, chat_id: str, user) -> list:
+async def add_file_context(messages: list, chat_id: str, user) -> list:
"""
Add file URLs to messages for native function calling.
"""
if not chat_id or chat_id.startswith('local:'):
return messages
- chat = Chats.get_chat_by_id_and_user_id(chat_id, user.id)
+ chat = await Chats.get_chat_by_id_and_user_id(chat_id, user.id)
if not chat:
return messages
@@ -1686,7 +1690,7 @@ async def chat_image_generation_handler(request: Request, form_data: dict, extra
if chat_id.startswith('local:'):
message_list = form_data.get('messages', [])
else:
- chat = Chats.get_chat_by_id_and_user_id(chat_id, user.id)
+ chat = await Chats.get_chat_by_id_and_user_id(chat_id, user.id)
await __event_emitter__(
{
'type': 'status',
@@ -2050,13 +2054,16 @@ async def convert_url_images_to_base64(form_data):
continue
try:
- base64_data = await asyncio.to_thread(get_image_base64_from_url, image_url)
- new_content.append(
- {
- 'type': 'image_url',
- 'image_url': {'url': base64_data},
- }
- )
+ base64_data = await get_image_base64_from_url(image_url)
+ if base64_data:
+ new_content.append(
+ {
+ 'type': 'image_url',
+ 'image_url': {'url': base64_data},
+ }
+ )
+ else:
+ new_content.append(item)
except Exception as e:
log.debug(f'Error converting image URL to base64: {e}')
new_content.append(item)
@@ -2066,12 +2073,12 @@ async def convert_url_images_to_base64(form_data):
return form_data
-def load_messages_from_db(chat_id: str, message_id: str) -> Optional[list[dict]]:
+async def load_messages_from_db(chat_id: str, message_id: str) -> Optional[list[dict]]:
"""
Load the message chain from DB up to message_id,
keeping only LLM-relevant fields (role, content, output).
"""
- messages_map = Chats.get_messages_map_by_chat_id(chat_id)
+ messages_map = await Chats.get_messages_map_by_chat_id(chat_id)
if not messages_map:
return None
@@ -2146,10 +2153,10 @@ async def process_chat_payload(request, form_data, user, metadata, model):
# Load messages from DB when available — DB preserves structured 'output' items
# which the frontend strips, causing tool calls to be merged into content.
chat_id = metadata.get('chat_id')
- parent_message_id = metadata.get('parent_message_id')
+ user_message_id = metadata.get('user_message_id')
- if chat_id and parent_message_id and not chat_id.startswith('local:'):
- db_messages = load_messages_from_db(chat_id, parent_message_id)
+ if chat_id and user_message_id and not chat_id.startswith('local:'):
+ db_messages = await load_messages_from_db(chat_id, user_message_id)
if db_messages:
system_message = get_system_message(form_data.get('messages', []))
form_data['messages'] = [system_message, *db_messages] if system_message else db_messages
@@ -2192,8 +2199,8 @@ async def process_chat_payload(request, form_data, user, metadata, model):
form_data = await convert_url_images_to_base64(form_data)
- event_emitter = get_event_emitter(metadata)
- event_caller = get_event_call(metadata)
+ event_emitter = await get_event_emitter(metadata)
+ event_caller = await get_event_call(metadata)
extra_params = {
'__event_emitter__': event_emitter,
@@ -2231,14 +2238,14 @@ async def process_chat_payload(request, form_data, user, metadata, model):
chat_id = metadata.get('chat_id', None)
folder_id = None
if chat_id and user:
- folder_id = Chats.get_chat_folder_id(chat_id, user.id)
+ folder_id = await Chats.get_chat_folder_id(chat_id, user.id)
# Fallback: use folder_id from metadata (temporary chats have no DB record)
if not folder_id:
folder_id = metadata.get('folder_id', None)
if folder_id and user:
- folder = Folders.get_folder_by_id_and_user_id(folder_id, user.id)
+ folder = await Folders.get_folder_by_id_and_user_id(folder_id, user.id)
if folder and folder.data:
if 'system_prompt' in folder.data:
@@ -2305,8 +2312,8 @@ async def process_chat_payload(request, form_data, user, metadata, model):
raise e
try:
- filter_ids = get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
- filter_functions = Functions.get_functions_by_ids(filter_ids)
+ filter_ids = await get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
+ filter_functions = await Functions.get_functions_by_ids(filter_ids)
form_data, flags = await process_filter_functions(
request=request,
@@ -2399,12 +2406,13 @@ async def process_chat_payload(request, form_data, user, metadata, model):
if all_skill_ids:
from open_webui.models.skills import Skills as SkillsModel
- accessible_skill_ids = {s.id for s in SkillsModel.get_skills_by_user_id(user.id, 'read')}
- available_skills = [
- s
- for sid in all_skill_ids
- if sid in accessible_skill_ids and (s := SkillsModel.get_skill_by_id(sid)) and s.is_active
- ]
+ accessible_skill_ids = {s.id for s in await SkillsModel.get_skills_by_user_id(user.id, 'read')}
+ available_skills = []
+ for sid in all_skill_ids:
+ if sid in accessible_skill_ids:
+ s = await SkillsModel.get_skill_by_id(sid)
+ if s and s.is_active:
+ available_skills.append(s)
skill_descriptions = ''
for skill in available_skills:
@@ -2441,7 +2449,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
# Get folder files
folder_id = file_item.get('id', None)
if folder_id:
- folder = Folders.get_folder_by_id_and_user_id(folder_id, user.id)
+ folder = await Folders.get_folder_by_id_and_user_id(folder_id, user.id)
if folder and folder.data and 'files' in folder.data:
files = [f for f in files if f.get('id', None) != folder_id]
files = [*files, *folder.data['files']]
@@ -2495,7 +2503,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
continue
# Check access control for MCP server
- if not has_connection_access(user, mcp_server_connection):
+ if not await has_connection_access(user, mcp_server_connection):
log.warning(f'Access denied to MCP server {server_id} for user {user.id}')
continue
@@ -2556,7 +2564,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
tool_specs = await mcp_clients[server_id].list_tool_specs()
for tool_spec in tool_specs:
- def make_tool_function(client, function_name):
+ async def make_tool_function(client, function_name):
async def tool_function(**kwargs):
return await client.call_tool(
function_name,
@@ -2570,7 +2578,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
# Skip this function
continue
- tool_function = make_tool_function(mcp_clients[server_id], tool_spec['name'])
+ tool_function = await make_tool_function(mcp_clients[server_id], tool_spec['name'])
mcp_tools_dict[f'{server_id}_{tool_spec["name"]}'] = {
'spec': {
@@ -2664,8 +2672,8 @@ async def process_chat_payload(request, form_data, user, metadata, model):
if metadata.get('params', {}).get('function_calling') == 'native' and builtin_tools_enabled:
# Add file context to user messages
chat_id = metadata.get('chat_id')
- form_data['messages'] = add_file_context(form_data.get('messages', []), chat_id, user)
- builtin_tools = get_builtin_tools(
+ form_data['messages'] = await add_file_context(form_data.get('messages', []), chat_id, user)
+ builtin_tools = await get_builtin_tools(
request,
{
**extra_params,
@@ -2755,7 +2763,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
return form_data, metadata, events
-def get_event_emitter_and_caller(metadata):
+async def get_event_emitter_and_caller(metadata):
event_emitter = None
event_caller = None
@@ -2763,18 +2771,18 @@ def get_event_emitter_and_caller(metadata):
# It broadcasts to user:{user_id} room AND persists to DB,
# so it works for backend-initiated calls (automations, API).
if metadata.get('chat_id') and metadata.get('message_id'):
- event_emitter = get_event_emitter(metadata)
+ event_emitter = await get_event_emitter(metadata)
# event_caller needs session_id — it calls back to a specific
# websocket session (used by direct tools, pyodide code interpreter).
if metadata.get('session_id') and metadata.get('chat_id') and metadata.get('message_id'):
- event_caller = get_event_call(metadata)
+ event_caller = await get_event_call(metadata)
return event_emitter, event_caller
-def build_chat_response_context(request, form_data, user, model, metadata, tasks, events):
- event_emitter, event_caller = get_event_emitter_and_caller(metadata)
+async def build_chat_response_context(request, form_data, user, model, metadata, tasks, events):
+ event_emitter, event_caller = await get_event_emitter_and_caller(metadata)
return {
'request': request,
'form_data': form_data,
@@ -2862,7 +2870,7 @@ async def background_tasks_handler(ctx):
messages = []
if 'chat_id' in metadata and not metadata['chat_id'].startswith('local:'):
- messages_map = Chats.get_messages_map_by_chat_id(metadata['chat_id'])
+ messages_map = await Chats.get_messages_map_by_chat_id(metadata['chat_id'])
message = messages_map.get(metadata['message_id']) if messages_map else None
message_list = get_message_list(messages_map, metadata['message_id'])
@@ -2942,7 +2950,7 @@ async def background_tasks_handler(ctx):
)
if not metadata.get('chat_id', '').startswith('local:'):
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -2995,7 +3003,7 @@ async def background_tasks_handler(ctx):
if not title:
title = messages[0].get('content', user_message)
- Chats.update_chat_title_by_id(metadata['chat_id'], title)
+ await Chats.update_chat_title_by_id(metadata['chat_id'], title)
await event_emitter(
{
@@ -3007,7 +3015,7 @@ async def background_tasks_handler(ctx):
if title == None and len(messages) == 2 and (not messages_map or len(messages_map) <= 2):
title = messages[0].get('content', user_message)
- Chats.update_chat_title_by_id(metadata['chat_id'], title)
+ await Chats.update_chat_title_by_id(metadata['chat_id'], title)
await event_emitter(
{
@@ -3041,7 +3049,7 @@ async def background_tasks_handler(ctx):
try:
tags = json.loads(tags_string).get('tags', [])
- Chats.update_chat_tags_by_id(metadata['chat_id'], tags, user)
+ await Chats.update_chat_tags_by_id(metadata['chat_id'], tags, user)
await event_emitter(
{
@@ -3053,6 +3061,112 @@ async def background_tasks_handler(ctx):
pass
+async def outlet_filter_handler(ctx):
+ """Run outlet filters inline after chat completion.
+
+ Replaces the separate POST /api/chat/completed round-trip.
+ Persists outlet-modified content to DB and emits a chat:outlet event
+ so the frontend can sync its in-memory state.
+ """
+ request = ctx['request']
+ user = ctx['user']
+ model = ctx['model']
+ metadata = ctx['metadata']
+ event_emitter = ctx.get('event_emitter')
+ event_caller = ctx.get('event_caller')
+
+ chat_id = metadata.get('chat_id', '')
+ message_id = metadata.get('message_id')
+
+ if not chat_id or chat_id.startswith('local:') or not message_id:
+ return
+
+ try:
+ messages_map = await Chats.get_messages_map_by_chat_id(chat_id)
+ if not messages_map:
+ return
+
+ message_list = get_message_list(messages_map, message_id)
+ if not message_list:
+ return
+
+ model_id = model.get('id') if isinstance(model, dict) else model
+
+ outlet_data = {
+ 'model': model_id,
+ 'messages': [
+ {
+ 'id': m.get('id'),
+ 'role': m.get('role'),
+ 'content': m.get('content', ''),
+ 'info': m.get('info'),
+ 'timestamp': m.get('timestamp'),
+ **({'usage': m['usage']} if m.get('usage') else {}),
+ **({'sources': m['sources']} if m.get('sources') else {}),
+ }
+ for m in message_list
+ ],
+ 'filter_ids': metadata.get('filter_ids', []),
+ 'chat_id': chat_id,
+ 'session_id': metadata.get('session_id'),
+ 'id': message_id,
+ }
+
+ # Pipeline outlet filters
+ models = request.app.state.MODELS
+ try:
+ outlet_data = await process_pipeline_outlet_filter(request, outlet_data, user, models)
+ except Exception as e:
+ log.debug(f'Pipeline outlet filter error: {e}')
+
+ # Function outlet filters
+ extra_params = {
+ '__event_emitter__': event_emitter,
+ '__event_call__': event_caller,
+ '__user__': user.model_dump() if isinstance(user, UserModel) else {},
+ '__metadata__': metadata,
+ '__request__': request,
+ '__model__': model,
+ }
+
+ filter_ids = await get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
+ filter_functions = await Functions.get_functions_by_ids(filter_ids)
+
+ outlet_result, _ = await process_filter_functions(
+ request=request,
+ filter_functions=filter_functions,
+ filter_type='outlet',
+ form_data=outlet_data,
+ extra_params=extra_params,
+ )
+
+ # Persist outlet-modified content and notify frontend
+ if outlet_result and outlet_result.get('messages'):
+ for msg in outlet_result['messages']:
+ msg_id = msg.get('id')
+ if msg_id and msg_id in messages_map:
+ original = messages_map[msg_id]
+ if original.get('content') != msg.get('content'):
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
+ chat_id,
+ msg_id,
+ {
+ 'content': msg['content'],
+ 'originalContent': original.get('content'),
+ },
+ )
+
+ if event_emitter:
+ await event_emitter(
+ {
+ 'type': 'chat:outlet',
+ 'data': {'messages': outlet_result['messages']},
+ }
+ )
+ except Exception as e:
+ log.debug(f'Error running outlet filters: {e}')
+
+
async def non_streaming_chat_response_handler(response, ctx):
request = ctx['request']
@@ -3076,7 +3190,9 @@ async def non_streaming_chat_response_handler(response, ctx):
else:
error = str(error)
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ log.error('Provider returned error (non-streaming): %s', error)
+
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -3092,7 +3208,7 @@ async def non_streaming_chat_response_handler(response, ctx):
)
if 'selected_model_id' in response_data:
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -3112,7 +3228,7 @@ async def non_streaming_chat_response_handler(response, ctx):
}
)
- title = Chats.get_chat_title_by_id(metadata['chat_id'])
+ title = await Chats.get_chat_title_by_id(metadata['chat_id'])
# Use output from backend if provided (OR-compliant backends),
# otherwise generate from response content
@@ -3143,7 +3259,7 @@ async def non_streaming_chat_response_handler(response, ctx):
# Save message in the database
usage = normalize_usage(response_data.get('usage', {}) or {})
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -3156,8 +3272,8 @@ async def non_streaming_chat_response_handler(response, ctx):
)
# Send a webhook notification if the user is not active
- if request.app.state.config.ENABLE_USER_WEBHOOKS and not Users.is_user_active(user.id):
- webhook_url = Users.get_user_webhook_url_by_id(user.id)
+ if request.app.state.config.ENABLE_USER_WEBHOOKS and not await Users.is_user_active(user.id):
+ webhook_url = await Users.get_user_webhook_url_by_id(user.id)
if webhook_url:
await post_webhook(
request.app.state.WEBUI_NAME,
@@ -3172,6 +3288,7 @@ async def non_streaming_chat_response_handler(response, ctx):
)
await background_tasks_handler(ctx)
+ await outlet_filter_handler(ctx)
response = build_response_object(response, merge_events_into_response(response_data, events))
except Exception as e:
@@ -3211,8 +3328,8 @@ async def streaming_chat_response_handler(response, ctx):
}
filter_functions = [
- Functions.get_function_by_id(filter_id)
- for filter_id in get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
+ await Functions.get_function_by_id(filter_id)
+ for filter_id in await get_sorted_filter_ids(request, model, metadata.get('filter_ids', []))
]
# Standard streaming response handler
@@ -3447,7 +3564,7 @@ async def streaming_chat_response_handler(response, ctx):
return output, end_flag
- message = Chats.get_message_by_id_and_message_id(metadata['chat_id'], metadata['message_id'])
+ message = await Chats.get_message_by_id_and_message_id(metadata['chat_id'], metadata['message_id'])
tool_calls = []
@@ -3509,7 +3626,7 @@ async def streaming_chat_response_handler(response, ctx):
)
# Save message in the database
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -3579,7 +3696,7 @@ async def streaming_chat_response_handler(response, ctx):
if 'selected_model_id' in data:
model_id = data['selected_model_id']
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -3644,8 +3761,9 @@ async def streaming_chat_response_handler(response, ctx):
if not choices:
error = data.get('error', {})
if error:
+ log.error('Provider returned error (streaming): %s', error)
try:
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -3762,10 +3880,10 @@ async def streaming_chat_response_handler(response, ctx):
}
)
- image_urls = get_image_urls(delta.get('images', []), request, metadata, user)
+ image_urls = await get_image_urls(delta.get('images', []), request, metadata, user)
if image_urls:
image_file_list = [{'type': 'image', 'url': url} for url in image_urls]
- message_files = Chats.add_message_files_by_id_and_message_id(
+ message_files = await Chats.add_message_files_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
image_file_list,
@@ -3847,7 +3965,7 @@ async def streaming_chat_response_handler(response, ctx):
)
if ENABLE_CHAT_RESPONSE_BASE64_IMAGE_URL_CONVERSION:
- value = convert_markdown_base64_images(
+ value = await convert_markdown_base64_images(
request,
value,
{
@@ -3963,7 +4081,7 @@ async def streaming_chat_response_handler(response, ctx):
if ENABLE_REALTIME_CHAT_SAVE:
# Save message in the database
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -4059,11 +4177,12 @@ async def streaming_chat_response_handler(response, ctx):
if responses_api_tool_calls:
tool_calls.append(_split_tool_calls(responses_api_tool_calls))
+ try:
+ await stream_body_handler(response, form_data)
+ finally:
if response.background:
await response.background()
- await stream_body_handler(response, form_data)
-
tool_call_retries = 0
tool_call_sources = [] # Track citation sources from tool results
all_tool_call_sources = [] # Accumulated sources across all iterations
@@ -4184,7 +4303,7 @@ async def streaming_chat_response_handler(response, ctx):
)
else:
- tool_function = get_updated_tool_function(
+ tool_function = await get_updated_tool_function(
function=tool['callable'],
extra_params={
'__messages__': form_data.get('messages', []),
@@ -4197,7 +4316,7 @@ async def streaming_chat_response_handler(response, ctx):
except Exception as e:
tool_result = str(e)
- tool_result, tool_result_files, tool_result_embeds = process_tool_result(
+ tool_result, tool_result_files, tool_result_embeds = await process_tool_result(
request,
tool_function_name,
tool_result,
@@ -4487,7 +4606,7 @@ async def streaming_chat_response_handler(response, ctx):
BLOCKED_MODULES = {CODE_INTERPRETER_BLOCKED_MODULES}
_real_import = builtins.__import__
- def restricted_import(name, globals=None, locals=None, fromlist=(), level=0):
+ async def restricted_import(name, globals=None, locals=None, fromlist=(), level=0):
if name.split('.')[0] in BLOCKED_MODULES:
importer_name = globals.get('__name__') if globals else None
if importer_name == '__main__':
@@ -4541,7 +4660,7 @@ async def streaming_chat_response_handler(response, ctx):
stdoutLines = stdout.split('\n')
for idx, line in enumerate(stdoutLines):
if re.match(r'data:image/\w+;base64', line):
- image_url = get_image_url_from_base64(
+ image_url = await get_image_url_from_base64(
request,
line,
metadata,
@@ -4558,7 +4677,7 @@ async def streaming_chat_response_handler(response, ctx):
resultLines = result.split('\n')
for idx, line in enumerate(resultLines):
if re.match(r'data:image/\w+;base64', line):
- image_url = get_image_url_from_base64(
+ image_url = await get_image_url_from_base64(
request,
line,
metadata,
@@ -4623,7 +4742,7 @@ async def streaming_chat_response_handler(response, ctx):
if item.get('status') == 'in_progress':
item['status'] = 'completed'
- title = Chats.get_chat_title_by_id(metadata['chat_id'])
+ title = await Chats.get_chat_title_by_id(metadata['chat_id'])
data = {
'done': True,
'content': serialize_output(output),
@@ -4634,7 +4753,7 @@ async def streaming_chat_response_handler(response, ctx):
if not ENABLE_REALTIME_CHAT_SAVE:
# Save message in the database
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{
@@ -4645,21 +4764,21 @@ async def streaming_chat_response_handler(response, ctx):
},
)
elif usage:
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{'done': True, 'usage': usage},
)
else:
- Chats.upsert_message_to_chat_by_id_and_message_id(
+ await Chats.upsert_message_to_chat_by_id_and_message_id(
metadata['chat_id'],
metadata['message_id'],
{'done': True},
)
# Send a webhook notification if the user is not active
- if request.app.state.config.ENABLE_USER_WEBHOOKS and not Users.is_user_active(user.id):
- webhook_url = Users.get_user_webhook_url_by_id(user.id)
+ if request.app.state.config.ENABLE_USER_WEBHOOKS and not await Users.is_user_active(user.id):
+ webhook_url = await Users.get_user_webhook_url_by_id(user.id)
if webhook_url:
await post_webhook(
request.app.state.WEBUI_NAME,
@@ -4681,27 +4800,36 @@ async def streaming_chat_response_handler(response, ctx):
)
await background_tasks_handler(ctx)
+ await outlet_filter_handler(ctx)
except asyncio.CancelledError:
log.warning('Task was cancelled!')
- await event_emitter({'type': 'chat:tasks:cancel'})
+ try:
+ await asyncio.shield(event_emitter({'type': 'chat:tasks:cancel'}))
- if not ENABLE_REALTIME_CHAT_SAVE:
- # Save message in the database
- Chats.upsert_message_to_chat_by_id_and_message_id(
- metadata['chat_id'],
- metadata['message_id'],
- {
- 'done': True,
- 'content': serialize_output(output),
- 'output': output,
- },
- )
- else:
- Chats.upsert_message_to_chat_by_id_and_message_id(
- metadata['chat_id'],
- metadata['message_id'],
- {'done': True},
- )
+ if not ENABLE_REALTIME_CHAT_SAVE:
+ # Save message in the database
+ await asyncio.shield(
+ Chats.upsert_message_to_chat_by_id_and_message_id(
+ metadata['chat_id'],
+ metadata['message_id'],
+ {
+ 'done': True,
+ 'content': serialize_output(output),
+ 'output': output,
+ },
+ )
+ )
+ else:
+ await asyncio.shield(
+ Chats.upsert_message_to_chat_by_id_and_message_id(
+ metadata['chat_id'],
+ metadata['message_id'],
+ {'done': True},
+ )
+ )
+ except Exception:
+ pass
+ raise # re-raise CancelledError for proper propagation
if response.background is not None:
await response.background()
diff --git a/backend/open_webui/utils/misc.py b/backend/open_webui/utils/misc.py
index 787ef4d6e8..345165db28 100644
--- a/backend/open_webui/utils/misc.py
+++ b/backend/open_webui/utils/misc.py
@@ -905,9 +905,17 @@ async def cleanup_response(
session: Optional[aiohttp.ClientSession],
):
if response:
- response.close()
+ if not response.closed:
+ # aiohttp 3.9+ made ClientResponse.close() synchronous (returns None).
+ # Older versions returned a coroutine. Handle both gracefully.
+ result = response.close()
+ if result is not None:
+ await result
if session:
- await session.close()
+ if not session.closed:
+ result = session.close()
+ if result is not None:
+ await result
async def stream_wrapper(response, session, content_handler=None):
diff --git a/backend/open_webui/utils/models.py b/backend/open_webui/utils/models.py
index b57c74744c..c8ebc190e5 100644
--- a/backend/open_webui/utils/models.py
+++ b/backend/open_webui/utils/models.py
@@ -130,13 +130,13 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
]
models = models + arena_models
- global_action_ids = {function.id for function in Functions.get_global_action_functions()}
- enabled_action_ids = {function.id for function in Functions.get_functions_by_type('action', active_only=True)}
+ global_action_ids = {function.id for function in await Functions.get_global_action_functions()}
+ enabled_action_ids = {function.id for function in await Functions.get_functions_by_type('action', active_only=True)}
- global_filter_ids = {function.id for function in Functions.get_global_filter_functions()}
- enabled_filter_ids = {function.id for function in Functions.get_functions_by_type('filter', active_only=True)}
+ global_filter_ids = {function.id for function in await Functions.get_global_filter_functions()}
+ enabled_filter_ids = {function.id for function in await Functions.get_functions_by_type('filter', active_only=True)}
- custom_models = Models.get_all_models()
+ custom_models = await Models.get_all_models()
# Single O(1) lookup: Ollama base names first, then exact IDs (exact wins).
base_model_lookup = {}
@@ -278,14 +278,14 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
all_function_ids.update(global_action_ids)
all_function_ids.update(global_filter_ids)
- functions_by_id = {f.id: f for f in Functions.get_functions_by_ids(list(all_function_ids))}
+ functions_by_id = {f.id: f for f in await Functions.get_functions_by_ids(list(all_function_ids))}
# Pre-warm the function module cache once per unique function ID.
# This ensures each function's DB freshness check runs exactly once,
# not once per (model × function) pair.
for function_id in all_function_ids:
try:
- get_function_module_from_cache(request, function_id)
+ await get_function_module_from_cache(request, function_id)
except Exception as e:
log.info(f'Failed to load function module for {function_id}: {e}')
@@ -312,7 +312,7 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
# Batch-fetch all function valves in one query to avoid N+1 DB hits
# inside get_action_priority (previously called per action × per model).
- all_function_valves = Functions.get_function_valves_by_ids(list(all_function_ids))
+ all_function_valves = await Functions.get_function_valves_by_ids(list(all_function_ids))
def get_action_priority(action_id):
try:
@@ -377,11 +377,11 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
return models
-def check_model_access(user, model, db=None):
+async def check_model_access(user, model, db=None):
if model.get('arena'):
meta = model.get('info', {}).get('meta', {})
access_grants = meta.get('access_grants', [])
- if not has_access(
+ if not await has_access(
user.id,
permission='read',
access_grants=access_grants,
@@ -389,12 +389,12 @@ def check_model_access(user, model, db=None):
):
raise Exception('Model not found')
else:
- model_info = Models.get_model_by_id(model.get('id'), db=db)
+ model_info = await Models.get_model_by_id(model.get('id'), db=db)
if not model_info:
raise Exception('Model not found')
elif not (
user.id == model_info.user_id
- or AccessGrants.has_access(
+ or await AccessGrants.has_access(
user_id=user.id,
resource_type='model',
resource_id=model_info.id,
@@ -405,7 +405,7 @@ def check_model_access(user, model, db=None):
raise Exception('Model not found')
-def get_filtered_models(models, user, db=None):
+async def get_filtered_models(models, user, db=None):
# Filter out models that the user does not have access to
if (
user.role == 'user' or (user.role == 'admin' and not BYPASS_ADMIN_ACCESS_CONTROL)
@@ -418,10 +418,10 @@ def get_filtered_models(models, user, db=None):
if info:
model_infos[model['id']] = info
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id, db=db)}
# Batch-fetch accessible resource IDs in a single query instead of N has_access calls
- accessible_model_ids = AccessGrants.get_accessible_resource_ids(
+ accessible_model_ids = await AccessGrants.get_accessible_resource_ids(
user_id=user.id,
resource_type='model',
resource_ids=list(model_infos.keys()),
@@ -435,7 +435,7 @@ def get_filtered_models(models, user, db=None):
if model.get('arena'):
meta = model.get('info', {}).get('meta', {})
access_grants = meta.get('access_grants', [])
- if has_access(
+ if await has_access(
user.id,
permission='read',
access_grants=access_grants,
diff --git a/backend/open_webui/utils/oauth.py b/backend/open_webui/utils/oauth.py
index 8aaadc271a..e8527fce4b 100644
--- a/backend/open_webui/utils/oauth.py
+++ b/backend/open_webui/utils/oauth.py
@@ -18,6 +18,7 @@ from typing import Literal
import aiohttp
from authlib.integrations.starlette_client import OAuth
+from authlib.jose.errors import BadSignatureError
from authlib.oidc.core import UserInfo
from fastapi import (
HTTPException,
@@ -139,6 +140,41 @@ auth_manager_config.OAUTH_UPDATE_EMAIL_ON_LOGIN = OAUTH_UPDATE_EMAIL_ON_LOGIN
auth_manager_config.OAUTH_AUDIENCE = OAUTH_AUDIENCE
+# Conservative default when the provider omits both expires_in and expires_at.
+# Matches the value recommended by Authlib's compliance_fix documentation.
+DEFAULT_TOKEN_EXPIRY_SECONDS = 3600
+
+
+def _normalize_token_expiry(token: dict) -> dict:
+ """Ensure a token dict always has a numeric ``expires_at``.
+
+ Resolution order:
+ 1. If *expires_at* is already present and non-None, trust it.
+ 2. Else if *expires_in* is present and non-None, compute *expires_at*.
+ 3. Otherwise fall back to ``DEFAULT_TOKEN_EXPIRY_SECONDS`` and log a
+ warning so operators can identify providers that omit expiration.
+
+ Also stamps *issued_at* for auditing.
+ """
+ token['issued_at'] = datetime.now().timestamp()
+
+ if token.get('expires_at') is not None:
+ token['expires_at'] = int(token['expires_at'])
+ return token
+
+ if token.get('expires_in') is not None:
+ token['expires_at'] = int(datetime.now().timestamp() + token['expires_in'])
+ return token
+
+ # Neither field present — conservative fallback
+ log.warning(
+ "OAuth token response missing both 'expires_in' and 'expires_at'; "
+ f'defaulting to {DEFAULT_TOKEN_EXPIRY_SECONDS}s from now'
+ )
+ token['expires_at'] = int(datetime.now().timestamp() + DEFAULT_TOKEN_EXPIRY_SECONDS)
+ return token
+
+
FERNET = None
if len(OAUTH_CLIENT_INFO_ENCRYPTION_KEY) != 44:
@@ -512,6 +548,24 @@ async def get_oauth_client_info_with_static_credentials(
raise e
+def resolve_oauth_client_info(connection: dict) -> dict:
+ """
+ Decrypt OAuth client info from a tool server connection config.
+
+ For oauth_2.1_static, overlays admin-provided credentials from
+ info.oauth_client_id and info.oauth_client_secret onto the blob.
+ """
+ info = connection.get('info', {})
+ data = decrypt_data(info.get('oauth_client_info', ''))
+
+ if connection.get('auth_type') == 'oauth_2.1_static':
+ if info.get('oauth_client_id') and info.get('oauth_client_secret'):
+ data['client_id'] = info['oauth_client_id']
+ data['client_secret'] = info['oauth_client_secret']
+
+ return data
+
+
class OAuthClientManager:
def __init__(self, app):
self.oauth = OAuth()
@@ -524,6 +578,7 @@ class OAuthClientManager:
'client_id': oauth_client_info.client_id,
'client_secret': oauth_client_info.client_secret,
'client_kwargs': {
+ 'follow_redirects': True,
**({'scope': oauth_client_info.scope} if oauth_client_info.scope else {}),
**(
{'token_endpoint_auth_method': oauth_client_info.token_endpoint_auth_method}
@@ -534,15 +589,20 @@ class OAuthClientManager:
'server_metadata_url': (oauth_client_info.issuer if oauth_client_info.issuer else None),
}
- if oauth_client_info.server_metadata and oauth_client_info.server_metadata.code_challenge_methods_supported:
- if (
- isinstance(
- oauth_client_info.server_metadata.code_challenge_methods_supported,
- list,
- )
- and 'S256' in oauth_client_info.server_metadata.code_challenge_methods_supported
- ):
- kwargs['code_challenge_method'] = 'S256'
+ # Default to S256 for OAuth 2.1 (PKCE is mandatory per RFC 9700)
+ kwargs['code_challenge_method'] = 'S256'
+
+ # Only remove PKCE if metadata explicitly excludes S256
+ if (
+ oauth_client_info.server_metadata
+ and oauth_client_info.server_metadata.code_challenge_methods_supported
+ and isinstance(
+ oauth_client_info.server_metadata.code_challenge_methods_supported,
+ list,
+ )
+ and 'S256' not in oauth_client_info.server_metadata.code_challenge_methods_supported
+ ):
+ del kwargs['code_challenge_method']
self.clients[client_id] = {
'client': self.oauth.register(**kwargs),
@@ -582,7 +642,7 @@ class OAuthClientManager:
continue
try:
- oauth_client_info = decrypt_data(oauth_client_info)
+ oauth_client_info = resolve_oauth_client_info(connection)
return self.add_client(expected_client_id, OAuthClientInformationFull(**oauth_client_info))['client']
except Exception as e:
log.error(f'Failed to lazily add OAuth client {expected_client_id} from config: {e}')
@@ -700,12 +760,16 @@ class OAuthClientManager:
"""
try:
# Get the OAuth session
- session = OAuthSessions.get_session_by_provider_and_user_id(client_id, user_id)
+ session = await OAuthSessions.get_session_by_provider_and_user_id(client_id, user_id)
if not session:
log.warning(f'No OAuth session found for user {user_id}, client_id {client_id}')
return None
- if force_refresh or datetime.now() + timedelta(minutes=5) >= datetime.fromtimestamp(session.expires_at):
+ if (
+ force_refresh
+ or session.expires_at is None
+ or datetime.now() + timedelta(minutes=5) >= datetime.fromtimestamp(session.expires_at)
+ ):
log.debug(f'Token refresh needed for user {user_id}, client_id {session.provider}')
refreshed_token = await self._refresh_token(session)
if refreshed_token:
@@ -714,7 +778,7 @@ class OAuthClientManager:
log.warning(
f'Token refresh failed for user {user_id}, client_id {session.provider}, deleting session {session.id}'
)
- OAuthSessions.delete_session_by_id(session.id)
+ await OAuthSessions.delete_session_by_id(session.id)
return None
return session.token
@@ -738,7 +802,7 @@ class OAuthClientManager:
if refreshed_token:
# Update the session with new token data
- session = OAuthSessions.update_session_by_id(session.id, refreshed_token)
+ session = await OAuthSessions.update_session_by_id(session.id, refreshed_token)
log.info(f'Successfully refreshed token for session {session.id}')
return session.token
else:
@@ -816,14 +880,7 @@ class OAuthClientManager:
if 'refresh_token' not in new_token_data:
new_token_data['refresh_token'] = token_data['refresh_token']
- # Add timestamp for tracking
- new_token_data['issued_at'] = datetime.now().timestamp()
-
- # Calculate expires_at if we have expires_in
- if 'expires_in' in new_token_data and 'expires_at' not in new_token_data:
- new_token_data['expires_at'] = int(
- datetime.now().timestamp() + new_token_data['expires_in']
- )
+ _normalize_token_expiry(new_token_data)
log.debug(f'Token refresh successful for client_id {client_id}')
return new_token_data
@@ -876,20 +933,15 @@ class OAuthClientManager:
if token:
try:
- # Add timestamp for tracking
- token['issued_at'] = datetime.now().timestamp()
-
- # Calculate expires_at if we have expires_in
- if 'expires_in' in token and 'expires_at' not in token:
- token['expires_at'] = datetime.now().timestamp() + token['expires_in']
+ _normalize_token_expiry(token)
# Clean up any existing sessions for this user/client_id first
- sessions = OAuthSessions.get_sessions_by_user_id(user_id)
+ sessions = await OAuthSessions.get_sessions_by_user_id(user_id)
for session in sessions:
if session.provider == client_id:
- OAuthSessions.delete_session_by_id(session.id)
+ await OAuthSessions.delete_session_by_id(session.id)
- session = OAuthSessions.create_session(
+ session = await OAuthSessions.create_session(
user_id=user_id,
provider=client_id,
token=token,
@@ -963,12 +1015,16 @@ class OAuthManager:
"""
try:
# Get the OAuth session
- session = OAuthSessions.get_session_by_id_and_user_id(session_id, user_id)
+ session = await OAuthSessions.get_session_by_id_and_user_id(session_id, user_id)
if not session:
log.warning(f'No OAuth session found for user {user_id}, session {session_id}')
return None
- if force_refresh or datetime.now() + timedelta(minutes=5) >= datetime.fromtimestamp(session.expires_at):
+ if (
+ force_refresh
+ or session.expires_at is None
+ or datetime.now() + timedelta(minutes=5) >= datetime.fromtimestamp(session.expires_at)
+ ):
log.debug(f'Token refresh needed for user {user_id}, provider {session.provider}')
refreshed_token = await self._refresh_token(session)
if refreshed_token:
@@ -977,7 +1033,7 @@ class OAuthManager:
log.warning(
f'Token refresh failed for user {user_id}, provider {session.provider}, deleting session {session.id}'
)
- OAuthSessions.delete_session_by_id(session.id)
+ await OAuthSessions.delete_session_by_id(session.id)
return None
return session.token
@@ -1002,7 +1058,7 @@ class OAuthManager:
if refreshed_token:
# Update the session with new token data
- session = OAuthSessions.update_session_by_id(session.id, refreshed_token)
+ session = await OAuthSessions.update_session_by_id(session.id, refreshed_token)
log.info(f'Successfully refreshed token for session {session.id}')
return session.token
else:
@@ -1082,14 +1138,7 @@ class OAuthManager:
if 'refresh_token' not in new_token_data:
new_token_data['refresh_token'] = token_data['refresh_token']
- # Add timestamp for tracking
- new_token_data['issued_at'] = datetime.now().timestamp()
-
- # Calculate expires_at if we have expires_in
- if 'expires_in' in new_token_data and 'expires_at' not in new_token_data:
- new_token_data['expires_at'] = int(
- datetime.now().timestamp() + new_token_data['expires_in']
- )
+ _normalize_token_expiry(new_token_data)
log.debug(f'Token refresh successful for provider {provider}')
return new_token_data
@@ -1102,16 +1151,19 @@ class OAuthManager:
log.error(f'Exception during token refresh for provider {provider}: {e}')
return None
- def get_user_role(self, user, user_data):
- user_count = Users.get_num_users()
+ async def get_user_role(self, user, user_data):
+ user_count = await Users.get_num_users()
if user and user_count == 1:
# If the user is the only user, assign the role "admin" - actually repairs role for single user on login
log.debug('Assigning the only user the admin role')
return 'admin'
if not user and user_count == 0:
- # If there are no users, assign the role "admin", as the first user will be an admin
- log.debug('Assigning the first user the admin role')
- return 'admin'
+ # First-user bootstrap: skip role management gating so the
+ # instance can be initialized. We intentionally return the
+ # default role here (not 'admin') — admin promotion happens
+ # race-safely *after* insert via get_num_users() == 1.
+ log.debug('First user bootstrap: using default role (admin promotion deferred to post-insert)')
+ return auth_manager_config.DEFAULT_USER_ROLE
if auth_manager_config.ENABLE_OAUTH_ROLE_MANAGEMENT:
log.debug('Running OAUTH Role management')
@@ -1185,7 +1237,7 @@ class OAuthManager:
return role
- def update_user_groups(self, user, user_data, default_permissions, db=None):
+ async def update_user_groups(self, user, user_data, default_permissions, db=None):
log.debug('Running OAUTH Group management')
oauth_claim = auth_manager_config.OAUTH_GROUPS_CLAIM
@@ -1214,8 +1266,8 @@ class OAuthManager:
else:
user_oauth_groups = []
- user_current_groups: list[GroupModel] = Groups.get_groups_by_member_id(user.id, db=db)
- all_available_groups: list[GroupModel] = Groups.get_all_groups(db=db)
+ user_current_groups: list[GroupModel] = await Groups.get_groups_by_member_id(user.id, db=db)
+ all_available_groups: list[GroupModel] = await Groups.get_all_groups(db=db)
# Create groups if they don't exist and creation is enabled
if auth_manager_config.ENABLE_OAUTH_GROUP_CREATION:
@@ -1223,7 +1275,7 @@ class OAuthManager:
all_group_names = {g.name for g in all_available_groups}
groups_created = False
# Determine creator ID: Prefer admin, fallback to current user if no admin exists
- admin_user = Users.get_super_admin_user()
+ admin_user = await Users.get_super_admin_user()
creator_id = admin_user.id if admin_user else user.id
log.debug(f'Using creator ID {creator_id} for potential group creation.')
@@ -1238,7 +1290,7 @@ class OAuthManager:
data={'config': {'share': auth_manager_config.OAUTH_GROUP_DEFAULT_SHARE}},
)
# Use determined creator ID (admin or fallback to current user)
- created_group = Groups.insert_new_group(creator_id, new_group_form, db=db)
+ created_group = await Groups.insert_new_group(creator_id, new_group_form, db=db)
if created_group:
log.info(
f"Successfully created group '{group_name}' with ID {created_group.id} using creator ID {creator_id}"
@@ -1253,7 +1305,7 @@ class OAuthManager:
# Refresh the list of all available groups if any were created
if groups_created:
- all_available_groups = Groups.get_all_groups(db=db)
+ all_available_groups = await Groups.get_all_groups(db=db)
log.debug('Refreshed list of all available groups after creation.')
log.debug(f'Oauth Groups claim: {oauth_claim}')
@@ -1270,14 +1322,14 @@ class OAuthManager:
):
# Remove group from user
log.debug(f'Removing user from group {group_model.name} as it is no longer in their oauth groups')
- Groups.remove_users_from_group(group_model.id, [user.id], db=db)
+ await Groups.remove_users_from_group(group_model.id, [user.id], db=db)
# In case a group is created, but perms are never assigned to the group by hitting "save"
group_permissions = group_model.permissions
if not group_permissions:
group_permissions = default_permissions
- Groups.update_group_by_id(
+ await Groups.update_group_by_id(
id=group_model.id,
form_data=GroupUpdateForm(
name=group_model.name,
@@ -1299,14 +1351,14 @@ class OAuthManager:
# Add user to group
log.debug(f'Adding user to group {group_model.name} as it was found in their oauth groups')
- Groups.add_users_to_group(group_model.id, [user.id], db=db)
+ await Groups.add_users_to_group(group_model.id, [user.id], db=db)
# In case a group is created, but perms are never assigned to the group by hitting "save"
group_permissions = group_model.permissions
if not group_permissions:
group_permissions = default_permissions
- Groups.update_group_by_id(
+ await Groups.update_group_by_id(
id=group_model.id,
form_data=GroupUpdateForm(
name=group_model.name,
@@ -1389,6 +1441,27 @@ class OAuthManager:
try:
token = await client.authorize_access_token(request, **auth_params)
+ except BadSignatureError:
+ # The IdP likely rotated its signing keys and the cached JWKS
+ # is stale. Evict the cached key set so the next attempt
+ # fetches fresh keys from the jwks_uri.
+ log.warning(
+ 'OIDC bad_signature for provider %s — evicting cached JWKS and retrying',
+ provider,
+ )
+ if hasattr(client, 'server_metadata') and isinstance(client.server_metadata, dict):
+ client.server_metadata.pop('jwks', None)
+ try:
+ token = await client.authorize_access_token(request, **auth_params)
+ except Exception as retry_exc:
+ detailed_error = _build_oauth_callback_error_message(retry_exc)
+ log.warning(
+ 'OAuth callback error during authorize_access_token retry for provider %s: %s',
+ provider,
+ detailed_error,
+ exc_info=True,
+ )
+ raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)
except Exception as e:
detailed_error = _build_oauth_callback_error_message(e)
log.warning(
@@ -1487,20 +1560,20 @@ class OAuthManager:
raise HTTPException(400, detail=ERROR_MESSAGES.INVALID_CRED)
# Check if the user exists
- user = Users.get_user_by_oauth_sub(provider, sub, db=db)
+ user = await Users.get_user_by_oauth_sub(provider, sub, db=db)
if not user:
# If the user does not exist, check if merging is enabled
if auth_manager_config.OAUTH_MERGE_ACCOUNTS_BY_EMAIL:
# Check if the user exists by email
- user = Users.get_user_by_email(email, db=db)
+ user = await Users.get_user_by_email(email, db=db)
if user:
# Update the user with the new oauth sub
- Users.update_user_oauth_by_id(user.id, provider, sub, db=db)
+ await Users.update_user_oauth_by_id(user.id, provider, sub, db=db)
if user:
- determined_role = self.get_user_role(user, user_data)
+ determined_role = await self.get_user_role(user, user_data)
if user.role != determined_role:
- Users.update_user_role_by_id(user.id, determined_role, db=db)
+ await Users.update_user_role_by_id(user.id, determined_role, db=db)
# Update the user object in memory as well,
# to avoid problems with the ENABLE_OAUTH_GROUP_MANAGEMENT check below
user.role = determined_role
@@ -1510,7 +1583,7 @@ class OAuthManager:
if username_claim:
new_name = user_data.get(username_claim)
if new_name and new_name != user.name:
- Users.update_user_by_id(user.id, {'name': new_name}, db=db)
+ await Users.update_user_by_id(user.id, {'name': new_name}, db=db)
user.name = new_name
log.debug(f'Updated name for user {user.email}')
@@ -1519,13 +1592,13 @@ class OAuthManager:
if email_claim:
new_email = user_data.get(email_claim)
if new_email and new_email.lower() != user.email.lower():
- existing_user = Users.get_user_by_email(new_email, db=db)
+ existing_user = await Users.get_user_by_email(new_email, db=db)
if existing_user:
log.error(
f'Cannot update email to {new_email} for user {user.id} because it is already taken.'
)
else:
- Auths.update_email_by_id(user.id, new_email.lower(), db=db)
+ await Auths.update_email_by_id(user.id, new_email.lower(), db=db)
user.email = new_email.lower()
log.debug(f'Updated email for user {user.id}')
@@ -1541,13 +1614,13 @@ class OAuthManager:
new_picture_url, token.get('access_token')
)
if processed_picture_url != user.profile_image_url:
- Users.update_user_profile_image_url_by_id(user.id, processed_picture_url, db=db)
+ await Users.update_user_profile_image_url_by_id(user.id, processed_picture_url, db=db)
log.debug(f'Updated profile picture for user {user.email}')
else:
# If the user does not exist, check if signups are enabled
if auth_manager_config.ENABLE_OAUTH_SIGNUP:
# Check if an existing user with the same email already exists
- existing_user = Users.get_user_by_email(email, db=db)
+ existing_user = await Users.get_user_by_email(email, db=db)
if existing_user:
raise HTTPException(400, detail=ERROR_MESSAGES.EMAIL_TAKEN)
@@ -1567,16 +1640,26 @@ class OAuthManager:
log.warning('Username claim is missing, using email as name')
name = email
- user = Auths.insert_new_auth(
+ user = await Auths.insert_new_auth(
email=email,
password=get_password_hash(str(uuid.uuid4())), # Random password, not used
name=name,
profile_image_url=picture_url,
- role=self.get_user_role(None, user_data),
+ role=await self.get_user_role(None, user_data),
oauth=oauth_data,
db=db,
)
+ if not user:
+ raise HTTPException(500, detail=ERROR_MESSAGES.CREATE_USER_ERROR)
+
+ # Atomically check if this is the only user *after* the
+ # insert to avoid TOCTOU race on first-user registration.
+ # Matches signup_handler pattern.
+ if await Users.get_num_users(db=db) == 1:
+ await Users.update_user_role_by_id(user.id, 'admin', db=db)
+ user = await Users.get_user_by_id(user.id, db=db)
+
if auth_manager_config.WEBHOOK_URL:
await post_webhook(
WEBUI_NAME,
@@ -1589,7 +1672,7 @@ class OAuthManager:
},
)
- apply_default_group_assignment(request.app.state.config.DEFAULT_GROUP_ID, user.id, db=db)
+ await apply_default_group_assignment(request.app.state.config.DEFAULT_GROUP_ID, user.id, db=db)
else:
raise HTTPException(
@@ -1602,7 +1685,7 @@ class OAuthManager:
expires_delta=parse_duration(auth_manager_config.JWT_EXPIRES_IN),
)
if auth_manager_config.ENABLE_OAUTH_GROUP_MANAGEMENT:
- self.update_user_groups(
+ await self.update_user_groups(
user=user,
user_data=user_data,
default_permissions=request.app.state.config.USER_PERMISSIONS,
@@ -1653,16 +1736,11 @@ class OAuthManager:
)
try:
- # Add timestamp for tracking
- token['issued_at'] = datetime.now().timestamp()
-
- # Calculate expires_at if we have expires_in
- if 'expires_in' in token and 'expires_at' not in token:
- token['expires_at'] = datetime.now().timestamp() + token['expires_in']
+ _normalize_token_expiry(token)
# Enforce max concurrent sessions per user/provider to prevent
# unbounded growth while allowing multi-device usage
- sessions = OAuthSessions.get_sessions_by_user_id(user.id, db=db)
+ sessions = await OAuthSessions.get_sessions_by_user_id(user.id, db=db)
provider_sessions = sorted(
[session for session in sessions if session.provider == provider],
key=lambda session: session.created_at,
@@ -1671,9 +1749,9 @@ class OAuthManager:
# Keep the newest sessions up to the limit, prune the rest
if len(provider_sessions) >= OAUTH_MAX_SESSIONS_PER_USER:
for old_session in provider_sessions[OAUTH_MAX_SESSIONS_PER_USER - 1 :]:
- OAuthSessions.delete_session_by_id(old_session.id, db=db)
+ await OAuthSessions.delete_session_by_id(old_session.id, db=db)
- session = OAuthSessions.create_session(
+ session = await OAuthSessions.create_session(
user_id=user.id,
provider=provider,
token=token,
@@ -1772,7 +1850,10 @@ class OAuthManager:
log.warning(f'Back-channel logout: no configured provider matches issuer {token_issuer}')
return JSONResponse(
status_code=400,
- content={'error': 'invalid_request', 'error_description': 'No configured provider matches token issuer'},
+ content={
+ 'error': 'invalid_request',
+ 'error_description': 'No configured provider matches token issuer',
+ },
)
# 4. Validate the logout_token signature and claims
@@ -1834,7 +1915,7 @@ class OAuthManager:
# 8. Identify users to log out
users_to_logout = []
if sub:
- user = Users.get_user_by_oauth_sub(matched_provider, sub, db=db)
+ user = await Users.get_user_by_oauth_sub(matched_provider, sub, db=db)
if user:
users_to_logout.append(user)
@@ -1855,9 +1936,9 @@ class OAuthManager:
revoked_count = 0
for user in users_to_logout:
- sessions = OAuthSessions.get_sessions_by_user_id(user.id, db=db)
+ sessions = await OAuthSessions.get_sessions_by_user_id(user.id, db=db)
for oauth_session in sessions:
- OAuthSessions.delete_session_by_id(oauth_session.id, db=db)
+ await OAuthSessions.delete_session_by_id(oauth_session.id, db=db)
if redis:
revocation_key = f'{REDIS_KEY_PREFIX}:auth:user:{user.id}:revoked_at'
@@ -1873,5 +1954,7 @@ class OAuthManager:
f'(email={user.email}, provider={matched_provider}, sessions_deleted={len(sessions)})'
)
- log.info(f'Back-channel logout: completed for {len(users_to_logout)} user(s), {revoked_count} revocation(s) set')
+ log.info(
+ f'Back-channel logout: completed for {len(users_to_logout)} user(s), {revoked_count} revocation(s) set'
+ )
return JSONResponse(status_code=200, content={})
diff --git a/backend/open_webui/utils/payload.py b/backend/open_webui/utils/payload.py
index 440927caf1..7cf4fe4de3 100644
--- a/backend/open_webui/utils/payload.py
+++ b/backend/open_webui/utils/payload.py
@@ -204,6 +204,11 @@ def convert_messages_openai_to_ollama(messages: list[dict]) -> list[dict]:
# Initialize the new message structure with the role
new_message = {'role': message['role']}
+ # Preserve Ollama-native 'thinking' field (used by reasoning models,
+ # may be injected by filter inlet functions).
+ if 'thinking' in message:
+ new_message['thinking'] = message['thinking']
+
content = message.get('content', [])
tool_calls = message.get('tool_calls', None)
tool_call_id = message.get('tool_call_id', None)
diff --git a/backend/open_webui/utils/plugin.py b/backend/open_webui/utils/plugin.py
index 46622e21ae..84671bbd3b 100644
--- a/backend/open_webui/utils/plugin.py
+++ b/backend/open_webui/utils/plugin.py
@@ -199,16 +199,16 @@ def replace_imports(content):
# May the intent of the one who wrote it survive every
# import and transformation, as a deed survives the generations.
-def load_tool_module_by_id(tool_id, content=None):
+async def load_tool_module_by_id(tool_id, content=None):
if content is None:
- tool = Tools.get_tool_by_id(tool_id)
+ tool = await Tools.get_tool_by_id(tool_id)
if not tool:
raise Exception(f'Toolkit not found: {tool_id}')
content = tool.content
content = replace_imports(content)
- Tools.update_tool_by_id(tool_id, {'content': content})
+ await Tools.update_tool_by_id(tool_id, {'content': content})
else:
frontmatter = extract_frontmatter(content)
# Install required packages found within the frontmatter
@@ -245,15 +245,15 @@ def load_tool_module_by_id(tool_id, content=None):
os.unlink(temp_file.name)
-def load_function_module_by_id(function_id: str, content: str | None = None):
+async def load_function_module_by_id(function_id: str, content: str | None = None):
if content is None:
- function = Functions.get_function_by_id(function_id)
+ function = await Functions.get_function_by_id(function_id)
if not function:
raise Exception(f'Function not found: {function_id}')
content = function.content
content = replace_imports(content)
- Functions.update_function_by_id(function_id, {'content': content})
+ await Functions.update_function_by_id(function_id, {'content': content})
else:
frontmatter = extract_frontmatter(content)
install_frontmatter_requirements(frontmatter.get('requirements', ''))
@@ -290,16 +290,16 @@ def load_function_module_by_id(function_id: str, content: str | None = None):
# Cleanup by removing the module in case of error
del sys.modules[module_name]
- Functions.update_function_by_id(function_id, {'is_active': False})
+ await Functions.update_function_by_id(function_id, {'is_active': False})
raise e
finally:
os.unlink(temp_file.name)
-def get_tool_module_from_cache(request, tool_id, load_from_db=True):
+async def get_tool_module_from_cache(request, tool_id, load_from_db=True):
if load_from_db:
# Always load from the database by default
- tool = Tools.get_tool_by_id(tool_id)
+ tool = await Tools.get_tool_by_id(tool_id)
if not tool:
raise Exception(f'Tool not found: {tool_id}')
content = tool.content
@@ -308,7 +308,7 @@ def get_tool_module_from_cache(request, tool_id, load_from_db=True):
if new_content != content:
content = new_content
# Update the tool content in the database
- Tools.update_tool_by_id(tool_id, {'content': content})
+ await Tools.update_tool_by_id(tool_id, {'content': content})
if (hasattr(request.app.state, 'TOOL_CONTENTS') and tool_id in request.app.state.TOOL_CONTENTS) and (
hasattr(request.app.state, 'TOOLS') and tool_id in request.app.state.TOOLS
@@ -316,12 +316,12 @@ def get_tool_module_from_cache(request, tool_id, load_from_db=True):
if request.app.state.TOOL_CONTENTS[tool_id] == content:
return request.app.state.TOOLS[tool_id], None
- tool_module, frontmatter = load_tool_module_by_id(tool_id, content)
+ tool_module, frontmatter = await load_tool_module_by_id(tool_id, content)
else:
if hasattr(request.app.state, 'TOOLS') and tool_id in request.app.state.TOOLS:
return request.app.state.TOOLS[tool_id], None
- tool_module, frontmatter = load_tool_module_by_id(tool_id)
+ tool_module, frontmatter = await load_tool_module_by_id(tool_id)
if not hasattr(request.app.state, 'TOOLS'):
request.app.state.TOOLS = {}
@@ -335,13 +335,13 @@ def get_tool_module_from_cache(request, tool_id, load_from_db=True):
return tool_module, frontmatter
-def get_function_module_from_cache(request, function_id, load_from_db=True):
+async def get_function_module_from_cache(request, function_id, load_from_db=True):
if load_from_db:
# Always load from the database by default
# This is useful for hooks like "inlet" or "outlet" where the content might change
# and we want to ensure the latest content is used.
- function = Functions.get_function_by_id(function_id)
+ function = await Functions.get_function_by_id(function_id)
if not function:
raise Exception(f'Function not found: {function_id}')
content = function.content
@@ -350,7 +350,7 @@ def get_function_module_from_cache(request, function_id, load_from_db=True):
if new_content != content:
content = new_content
# Update the function content in the database
- Functions.update_function_by_id(function_id, {'content': content})
+ await Functions.update_function_by_id(function_id, {'content': content})
if (
hasattr(request.app.state, 'FUNCTION_CONTENTS') and function_id in request.app.state.FUNCTION_CONTENTS
@@ -358,7 +358,7 @@ def get_function_module_from_cache(request, function_id, load_from_db=True):
if request.app.state.FUNCTION_CONTENTS[function_id] == content:
return request.app.state.FUNCTIONS[function_id], None, None
- function_module, function_type, frontmatter = load_function_module_by_id(function_id, content)
+ function_module, function_type, frontmatter = await load_function_module_by_id(function_id, content)
else:
# Load from cache (e.g. "stream" hook)
# This is useful for performance reasons
@@ -366,7 +366,7 @@ def get_function_module_from_cache(request, function_id, load_from_db=True):
if hasattr(request.app.state, 'FUNCTIONS') and function_id in request.app.state.FUNCTIONS:
return request.app.state.FUNCTIONS[function_id], None, None
- function_module, function_type, frontmatter = load_function_module_by_id(function_id)
+ function_module, function_type, frontmatter = await load_function_module_by_id(function_id)
if not hasattr(request.app.state, 'FUNCTIONS'):
request.app.state.FUNCTIONS = {}
@@ -404,7 +404,7 @@ def install_frontmatter_requirements(requirements: str):
log.info('No requirements found in frontmatter.')
-def install_tool_and_function_dependencies():
+async def install_tool_and_function_dependencies():
"""
Install all dependencies for all admin tools and active functions.
@@ -412,8 +412,8 @@ def install_tool_and_function_dependencies():
and then installing them using pip. Duplicates or similar version specifications are
handled by pip as much as possible.
"""
- function_list = Functions.get_functions(active_only=True)
- tool_list = Tools.get_tools()
+ function_list = await Functions.get_functions(active_only=True)
+ tool_list = await Tools.get_tools()
all_dependencies = ''
try:
diff --git a/backend/open_webui/utils/redis.py b/backend/open_webui/utils/redis.py
index 55d08147a9..cb570cb45a 100644
--- a/backend/open_webui/utils/redis.py
+++ b/backend/open_webui/utils/redis.py
@@ -9,7 +9,9 @@ import redis
from open_webui.env import (
REDIS_CLUSTER,
+ REDIS_HEALTH_CHECK_INTERVAL,
REDIS_SOCKET_CONNECT_TIMEOUT,
+ REDIS_SOCKET_KEEPALIVE,
REDIS_SENTINEL_HOSTS,
REDIS_SENTINEL_MAX_RETRY_COUNT,
REDIS_SENTINEL_PORT,
@@ -36,7 +38,7 @@ class SentinelRedisProxy:
def _master(self):
return self._sentinel.master_for(self._service, **self._kw)
- def __getattr__(self, item):
+ async def __getattr__(self, item):
master = self._master()
orig_attr = getattr(master, item)
@@ -191,6 +193,14 @@ def get_redis_connection(
connection = None
+ connect_timeout_kwargs = (
+ {'socket_connect_timeout': REDIS_SOCKET_CONNECT_TIMEOUT} if REDIS_SOCKET_CONNECT_TIMEOUT is not None else {}
+ )
+
+ keepalive_kwargs = {'socket_keepalive': True} if REDIS_SOCKET_KEEPALIVE else {}
+
+ health_check_kwargs = {'health_check_interval': REDIS_HEALTH_CHECK_INTERVAL} if REDIS_HEALTH_CHECK_INTERVAL else {}
+
if async_mode:
import redis.asyncio as redis
@@ -205,6 +215,8 @@ def get_redis_connection(
password=redis_config['password'],
decode_responses=decode_responses,
socket_connect_timeout=REDIS_SOCKET_CONNECT_TIMEOUT,
+ **keepalive_kwargs,
+ **health_check_kwargs,
)
connection = SentinelRedisProxy(
sentinel,
@@ -214,9 +226,21 @@ def get_redis_connection(
elif redis_cluster:
if not redis_url:
raise ValueError('Redis URL must be provided for cluster mode.')
- return redis.cluster.RedisCluster.from_url(redis_url, decode_responses=decode_responses)
+ return redis.cluster.RedisCluster.from_url(
+ redis_url,
+ decode_responses=decode_responses,
+ **connect_timeout_kwargs,
+ **keepalive_kwargs,
+ **health_check_kwargs,
+ )
elif redis_url:
- connection = redis.from_url(redis_url, decode_responses=decode_responses)
+ connection = redis.from_url(
+ redis_url,
+ decode_responses=decode_responses,
+ **connect_timeout_kwargs,
+ **keepalive_kwargs,
+ **health_check_kwargs,
+ )
else:
import redis
@@ -230,6 +254,8 @@ def get_redis_connection(
password=redis_config['password'],
decode_responses=decode_responses,
socket_connect_timeout=REDIS_SOCKET_CONNECT_TIMEOUT,
+ **keepalive_kwargs,
+ **health_check_kwargs,
)
connection = SentinelRedisProxy(
sentinel,
@@ -239,9 +265,21 @@ def get_redis_connection(
elif redis_cluster:
if not redis_url:
raise ValueError('Redis URL must be provided for cluster mode.')
- return redis.cluster.RedisCluster.from_url(redis_url, decode_responses=decode_responses)
+ return redis.cluster.RedisCluster.from_url(
+ redis_url,
+ decode_responses=decode_responses,
+ **connect_timeout_kwargs,
+ **keepalive_kwargs,
+ **health_check_kwargs,
+ )
elif redis_url:
- connection = redis.Redis.from_url(redis_url, decode_responses=decode_responses)
+ connection = redis.Redis.from_url(
+ redis_url,
+ decode_responses=decode_responses,
+ **connect_timeout_kwargs,
+ **keepalive_kwargs,
+ **health_check_kwargs,
+ )
_CONNECTION_CACHE[cache_key] = connection
return connection
diff --git a/backend/open_webui/utils/session_pool.py b/backend/open_webui/utils/session_pool.py
new file mode 100644
index 0000000000..d74eae4f04
--- /dev/null
+++ b/backend/open_webui/utils/session_pool.py
@@ -0,0 +1,119 @@
+"""Shared aiohttp ClientSession pool.
+
+Instead of creating a new ClientSession (and TCPConnector) per request,
+callers acquire a long-lived session from this module. The pool manages
+a single TCPConnector with configurable limits, enabling TCP/SSL connection
+reuse, shared DNS cache, and bounded concurrency.
+
+All pool parameters are configurable via environment variables:
+ - AIOHTTP_POOL_CONNECTIONS (default 100) — max total connections
+ - AIOHTTP_POOL_CONNECTIONS_PER_HOST (default 30) — per-host limit
+ - AIOHTTP_POOL_DNS_TTL (default 300) — DNS cache TTL in seconds
+
+Usage:
+ from open_webui.utils.session_pool import get_session, cleanup_response
+
+ session = await get_session()
+ r = await session.request(...)
+ # When done with the *response* (not the session):
+ await cleanup_response(r)
+
+IMPORTANT: Callers must NOT close the shared session. Only the response
+needs cleanup. The session is closed once during application shutdown
+via ``close_session()``.
+"""
+
+import logging
+from typing import Optional
+
+import aiohttp
+
+from open_webui.env import (
+ AIOHTTP_CLIENT_TIMEOUT,
+ AIOHTTP_POOL_CONNECTIONS,
+ AIOHTTP_POOL_CONNECTIONS_PER_HOST,
+ AIOHTTP_POOL_DNS_TTL,
+)
+
+log = logging.getLogger(__name__)
+
+_session: Optional[aiohttp.ClientSession] = None
+
+
+async def get_session() -> aiohttp.ClientSession:
+ """Return the shared aiohttp ClientSession, creating it lazily."""
+ global _session
+ if _session is None or _session.closed:
+ connector_kwargs = {
+ 'ttl_dns_cache': AIOHTTP_POOL_DNS_TTL,
+ 'enable_cleanup_closed': True,
+ }
+ if AIOHTTP_POOL_CONNECTIONS is not None:
+ connector_kwargs['limit'] = AIOHTTP_POOL_CONNECTIONS
+ else:
+ connector_kwargs['limit'] = 0 # aiohttp: 0 = unlimited
+ if AIOHTTP_POOL_CONNECTIONS_PER_HOST is not None:
+ connector_kwargs['limit_per_host'] = AIOHTTP_POOL_CONNECTIONS_PER_HOST
+ else:
+ connector_kwargs['limit_per_host'] = 0 # aiohttp: 0 = unlimited
+ connector = aiohttp.TCPConnector(**connector_kwargs)
+ timeout = aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT)
+ _session = aiohttp.ClientSession(
+ connector=connector,
+ timeout=timeout,
+ trust_env=True,
+ )
+ log.info(
+ 'Created shared aiohttp session pool (limit=%s, per_host=%s, dns_ttl=%d)',
+ AIOHTTP_POOL_CONNECTIONS or 'unlimited',
+ AIOHTTP_POOL_CONNECTIONS_PER_HOST or 'unlimited',
+ AIOHTTP_POOL_DNS_TTL,
+ )
+ return _session
+
+
+async def close_session():
+ """Close the shared session. Called during application shutdown."""
+ global _session
+ if _session and not _session.closed:
+ await _session.close()
+ log.info('Closed shared aiohttp session pool')
+ _session = None
+
+
+async def cleanup_response(
+ response: Optional[aiohttp.ClientResponse],
+ session: Optional[aiohttp.ClientSession] = None,
+):
+ """Release and close an aiohttp response, optionally closing the session.
+
+ When using the shared pool, ``session`` should be ``None`` (the pool
+ session is never closed per-request). When a caller creates its own
+ one-off session, pass it here to close it after the response.
+ """
+ if response:
+ if not response.closed:
+ # aiohttp 3.9+ made ClientResponse.close() synchronous (returns None).
+ # Older versions returned a coroutine. Handle both gracefully.
+ result = response.close()
+ if result is not None:
+ await result
+ if session:
+ if not session.closed:
+ result = session.close()
+ if result is not None:
+ await result
+
+
+async def stream_wrapper(response, session=None, content_handler=None):
+ """Wrap a stream to ensure cleanup happens even if streaming is interrupted.
+
+ This is more reliable than BackgroundTask which may not run if the client
+ disconnects. When using the shared pool, ``session`` should be ``None``.
+ """
+ try:
+ stream = content_handler(response.content) if content_handler else response.content
+ async for chunk in stream:
+ yield chunk
+ finally:
+ await cleanup_response(response, session)
diff --git a/backend/open_webui/utils/telemetry/instrumentors.py b/backend/open_webui/utils/telemetry/instrumentors.py
index 394e7178d6..fe8e9ba799 100644
--- a/backend/open_webui/utils/telemetry/instrumentors.py
+++ b/backend/open_webui/utils/telemetry/instrumentors.py
@@ -7,8 +7,8 @@ from aiohttp import (
TraceRequestEndParams,
TraceRequestExceptionParams,
)
-from chromadb.telemetry.opentelemetry.fastapi import instrument_fastapi
from fastapi import FastAPI
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
from opentelemetry.instrumentation.httpx import (
HTTPXClientInstrumentor,
RequestInfo,
@@ -176,7 +176,7 @@ class Instrumentor(BaseInstrumentor):
return []
def _instrument(self, **kwargs):
- instrument_fastapi(app=self.app)
+ FastAPIInstrumentor.instrument_app(app=self.app)
SQLAlchemyInstrumentor().instrument(engine=self.db_engine)
RedisInstrumentor().instrument(request_hook=redis_request_hook)
RequestsInstrumentor().instrument(request_hook=requests_hook, response_hook=response_hook)
diff --git a/backend/open_webui/utils/telemetry/metrics.py b/backend/open_webui/utils/telemetry/metrics.py
index 4c43de3342..26216b6ca4 100644
--- a/backend/open_webui/utils/telemetry/metrics.py
+++ b/backend/open_webui/utils/telemetry/metrics.py
@@ -124,16 +124,16 @@ def setup_metrics(app: FastAPI, resource: Resource) -> None:
unit='ms',
)
- def observe_active_users(
+ async def observe_active_users(
options: metrics.CallbackOptions,
) -> Sequence[metrics.Observation]:
return [
metrics.Observation(
- value=Users.get_active_user_count(),
+ value=await Users.get_active_user_count(),
)
]
- def observe_total_registered_users(
+ async def observe_total_registered_users(
options: metrics.CallbackOptions,
) -> Sequence[metrics.Observation]:
# IMPORTANT: Use get_num_users() for efficient COUNT(*) query.
@@ -141,7 +141,7 @@ def setup_metrics(app: FastAPI, resource: Resource) -> None:
# causing connection pool exhaustion on high-latency databases (e.g., Aurora).
return [
metrics.Observation(
- value=Users.get_num_users() or 0,
+ value=await Users.get_num_users() or 0,
)
]
@@ -159,10 +159,10 @@ def setup_metrics(app: FastAPI, resource: Resource) -> None:
callbacks=[observe_active_users],
)
- def observe_users_active_today(
+ async def observe_users_active_today(
options: metrics.CallbackOptions,
) -> Sequence[metrics.Observation]:
- return [metrics.Observation(value=Users.get_num_users_active_today())]
+ return [metrics.Observation(value=await Users.get_num_users_active_today())]
meter.create_observable_gauge(
name='webui.users.active.today',
diff --git a/backend/open_webui/utils/tools.py b/backend/open_webui/utils/tools.py
index 377a81d749..da817c4741 100644
--- a/backend/open_webui/utils/tools.py
+++ b/backend/open_webui/utils/tools.py
@@ -51,6 +51,7 @@ from open_webui.env import (
ENABLE_FORWARD_USER_INFO_HEADERS,
FORWARD_SESSION_INFO_HEADER_CHAT_ID,
FORWARD_SESSION_INFO_HEADER_MESSAGE_ID,
+ REDIS_KEY_PREFIX,
)
from open_webui.utils.headers import include_user_info_headers
from open_webui.tools.builtin import (
@@ -85,17 +86,26 @@ from open_webui.tools.builtin import (
view_file,
view_knowledge_file,
view_skill,
- tasks,
+ create_tasks,
+ update_task,
+ create_automation,
+ update_automation,
+ list_automations,
+ toggle_automation,
+ delete_automation,
)
import copy
+from open_webui.utils.access_control import has_permission
log = logging.getLogger(__name__)
# Let no function be called without need, and let what
# it yields justify the cost of running it.
-def get_async_tool_function_and_apply_extra_params(function: Callable, extra_params: dict) -> Callable[..., Awaitable]:
+async def get_async_tool_function_and_apply_extra_params(
+ function: Callable, extra_params: dict
+) -> Callable[..., Awaitable]:
sig = inspect.signature(function)
extra_params = {k: v for k, v in extra_params.items() if k in sig.parameters}
partial_func = partial(function, **extra_params)
@@ -132,13 +142,13 @@ def get_async_tool_function_and_apply_extra_params(function: Callable, extra_par
return new_function
-def get_updated_tool_function(function: Callable, extra_params: dict):
+async def get_updated_tool_function(function: Callable, extra_params: dict):
# Get the original function and merge updated params
__function__ = getattr(function, '__function__', None)
__extra_params__ = getattr(function, '__extra_params__', None)
if __function__ is not None and __extra_params__ is not None:
- return get_async_tool_function_and_apply_extra_params(
+ return await get_async_tool_function_and_apply_extra_params(
__function__,
{**__extra_params__, **extra_params},
)
@@ -154,16 +164,16 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
tools_dict = {}
# Get user's group memberships for access control checks
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
for tool_id in tool_ids:
- tool = Tools.get_tool_by_id(tool_id)
+ tool = await Tools.get_tool_by_id(tool_id)
if tool:
# Check access control for local tools
if (
not (user.role == 'admin' and BYPASS_ADMIN_ACCESS_CONTROL)
and tool.user_id != user.id
- and not AccessGrants.has_access(
+ and not await AccessGrants.has_access(
user_id=user.id,
resource_type='tool',
resource_id=tool.id,
@@ -176,7 +186,7 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
module = request.app.state.TOOLS.get(tool_id, None)
if module is None:
- module, _ = load_tool_module_by_id(tool_id)
+ module, _ = await load_tool_module_by_id(tool_id)
request.app.state.TOOLS[tool_id] = module
__user__ = {
@@ -185,11 +195,11 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
# Set valves for the tool
if hasattr(module, 'valves') and hasattr(module, 'Valves'):
- valves = Tools.get_tool_valves_by_id(tool_id) or {}
+ valves = await Tools.get_tool_valves_by_id(tool_id) or {}
module.valves = module.Valves(**valves)
if hasattr(module, 'UserValves'):
__user__['valves'] = module.UserValves( # type: ignore
- **Tools.get_user_valves_by_id_and_user_id(tool_id, user.id)
+ **await Tools.get_user_valves_by_id_and_user_id(tool_id, user.id)
)
for spec in tool.specs:
@@ -207,7 +217,7 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
# convert to function that takes only model params and inserts custom params
function_name = spec['name']
tool_function = getattr(module, function_name)
- callable = get_async_tool_function_and_apply_extra_params(
+ callable = await get_async_tool_function_and_apply_extra_params(
tool_function,
{
**extra_params,
@@ -279,7 +289,7 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
tool_server_connection = connections[tool_server_idx]
# Check access control for tool server
- if not has_connection_access(user, tool_server_connection, user_group_ids):
+ if not await has_connection_access(user, tool_server_connection, user_group_ids):
log.warning(f'Access denied to tool server {server_id} for user {user.id}')
continue
@@ -333,7 +343,7 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
if metadata and metadata.get('message_id'):
headers[FORWARD_SESSION_INFO_HEADER_MESSAGE_ID] = metadata.get('message_id')
- def make_tool_function(function_name, tool_server_data, headers):
+ async def make_tool_function(function_name, tool_server_data, headers):
async def tool_function(**kwargs):
return await execute_tool_server(
url=tool_server_data['url'],
@@ -346,9 +356,9 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
return tool_function
- tool_function = make_tool_function(function_name, tool_server_data, headers)
+ tool_function = await make_tool_function(function_name, tool_server_data, headers)
- callable = get_async_tool_function_and_apply_extra_params(
+ callable = await get_async_tool_function_and_apply_extra_params(
tool_function,
{},
)
@@ -375,7 +385,7 @@ async def get_tools(request: Request, tool_ids: list[str], user: UserModel, extr
return tools_dict
-def get_builtin_tools(
+async def get_builtin_tools(
request: Request, extra_params: dict, features: dict = None, model: dict = None
) -> dict[str, dict]:
"""
@@ -397,6 +407,18 @@ def get_builtin_tools(
builtin_tools = model.get('info', {}).get('meta', {}).get('builtinTools', {})
return builtin_tools.get(category, True)
+ # Helper to check user-level feature permission (admins always pass)
+ user = extra_params.get('__user__', {})
+
+ async def has_user_permission(feature_key: str) -> bool:
+ if user.get('role') == 'admin':
+ return True
+ return await has_permission(
+ user.get('id', ''),
+ f'features.{feature_key}',
+ request.app.state.config.USER_PERMISSIONS,
+ )
+
# Time utilities - available for date calculations
if is_builtin_tool_enabled('time'):
builtin_functions.extend([get_current_timestamp, calculate_timestamp])
@@ -440,7 +462,11 @@ def get_builtin_tools(
builtin_functions.extend([search_chats, view_chat])
# Add memory tools if builtin category enabled AND enabled for this chat
- if is_builtin_tool_enabled('memory') and (features.get('memory') or get_model_capability('memory', False)):
+ if (
+ is_builtin_tool_enabled('memory')
+ and (features.get('memory') or get_model_capability('memory', False))
+ and await has_user_permission('memories')
+ ):
builtin_functions.extend(
[
search_memories,
@@ -457,6 +483,7 @@ def get_builtin_tools(
and getattr(request.app.state.config, 'ENABLE_WEB_SEARCH', False)
and get_model_capability('web_search')
and features.get('web_search')
+ and await has_user_permission('web_search')
):
builtin_functions.extend([search_web, fetch_url])
@@ -466,6 +493,7 @@ def get_builtin_tools(
and getattr(request.app.state.config, 'ENABLE_IMAGE_GENERATION', False)
and get_model_capability('image_generation')
and features.get('image_generation')
+ and await has_user_permission('image_generation')
):
builtin_functions.append(generate_image)
if (
@@ -473,6 +501,7 @@ def get_builtin_tools(
and getattr(request.app.state.config, 'ENABLE_IMAGE_EDIT', False)
and get_model_capability('image_generation')
and features.get('image_generation')
+ and await has_user_permission('image_generation')
):
builtin_functions.append(edit_image)
@@ -482,15 +511,24 @@ def get_builtin_tools(
and getattr(request.app.state.config, 'ENABLE_CODE_INTERPRETER', True)
and get_model_capability('code_interpreter')
and features.get('code_interpreter')
+ and await has_user_permission('code_interpreter')
):
builtin_functions.append(execute_code)
- # Notes tools - search, view, create, and update user's notes (if builtin category enabled AND notes enabled globally)
- if is_builtin_tool_enabled('notes') and getattr(request.app.state.config, 'ENABLE_NOTES', False):
+ # Notes tools - search, view, create, and update user's notes
+ if (
+ is_builtin_tool_enabled('notes')
+ and getattr(request.app.state.config, 'ENABLE_NOTES', False)
+ and await has_user_permission('notes')
+ ):
builtin_functions.extend([search_notes, view_note, write_note, replace_note_content])
- # Channels tools - search channels and messages (if builtin category enabled AND channels enabled globally)
- if is_builtin_tool_enabled('channels') and getattr(request.app.state.config, 'ENABLE_CHANNELS', False):
+ # Channels tools - search channels and messages
+ if (
+ is_builtin_tool_enabled('channels')
+ and getattr(request.app.state.config, 'ENABLE_CHANNELS', False)
+ and await has_user_permission('channels')
+ ):
builtin_functions.extend(
[
search_channels,
@@ -506,10 +544,16 @@ def get_builtin_tools(
# Task management - break down complex work into trackable steps
if is_builtin_tool_enabled('tasks'):
- builtin_functions.append(tasks)
+ builtin_functions.extend([create_tasks, update_task])
+
+ # Automation tools - create and manage scheduled automations from chat
+ if is_builtin_tool_enabled('automations') and await has_user_permission('automations'):
+ builtin_functions.extend(
+ [create_automation, update_automation, list_automations, toggle_automation, delete_automation]
+ )
for func in builtin_functions:
- callable = get_async_tool_function_and_apply_extra_params(
+ callable = await get_async_tool_function_and_apply_extra_params(
func,
{
'__request__': request,
@@ -696,20 +740,31 @@ def get_tool_specs(tool_module: object) -> list[dict]:
return specs
-def resolve_schema(schema, components):
+def resolve_schema(schema, components, resolved_schemas=None):
"""
Recursively resolves a JSON schema using OpenAPI components.
"""
if not schema:
return {}
+ if resolved_schemas is None:
+ resolved_schemas = set()
+
if '$ref' in schema:
ref_path = schema['$ref']
+ schema_name = ref_path.split('/')[-1]
+
+ if schema_name in resolved_schemas:
+ # Avoid infinite recursion on circular references
+ return {}
+
+ resolved_schemas.add(schema_name)
+
ref_parts = ref_path.strip('#/').split('/')
resolved = components
for part in ref_parts[1:]: # Skip the initial 'components'
resolved = resolved.get(part, {})
- return resolve_schema(resolved, components)
+ return resolve_schema(resolved, components, resolved_schemas)
resolved_schema = copy.deepcopy(schema)
@@ -757,7 +812,7 @@ def convert_openapi_to_tool_payload(openapi_spec):
if not description:
description = param.get('description') or ''
if param_schema.get('enum') and isinstance(param_schema.get('enum'), list):
- description += f'. Possible values: {", ".join(param_schema.get("enum"))}'
+ description += f'. Possible values: {", ".join(str(v) for v in param_schema.get("enum"))}'
param_property = {
'type': param_schema.get('type') or 'string',
'description': description,
@@ -800,7 +855,9 @@ async def set_tool_servers(request: Request):
request.app.state.TOOL_SERVERS = await get_tool_servers_data(request.app.state.config.TOOL_SERVER_CONNECTIONS)
if request.app.state.redis is not None:
- await request.app.state.redis.set('tool_servers', json.dumps(request.app.state.TOOL_SERVERS))
+ await request.app.state.redis.set(
+ f'{REDIS_KEY_PREFIX}:tool_servers', json.dumps(request.app.state.TOOL_SERVERS)
+ )
return request.app.state.TOOL_SERVERS
@@ -809,7 +866,7 @@ async def get_tool_servers(request: Request):
tool_servers = []
if request.app.state.redis is not None:
try:
- tool_servers = json.loads(await request.app.state.redis.get('tool_servers'))
+ tool_servers = json.loads(await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:tool_servers'))
request.app.state.TOOL_SERVERS = tool_servers
except Exception as e:
log.error(f'Error fetching tool_servers from Redis: {e}')
@@ -935,7 +992,9 @@ async def set_terminal_servers(request: Request):
)
if request.app.state.redis is not None:
- await request.app.state.redis.set('terminal_servers', json.dumps(request.app.state.TERMINAL_SERVERS))
+ await request.app.state.redis.set(
+ f'{REDIS_KEY_PREFIX}:terminal_servers', json.dumps(request.app.state.TERMINAL_SERVERS)
+ )
return request.app.state.TERMINAL_SERVERS
@@ -945,7 +1004,7 @@ async def get_terminal_servers(request: Request):
terminal_servers = []
if request.app.state.redis is not None:
try:
- terminal_servers = json.loads(await request.app.state.redis.get('terminal_servers'))
+ terminal_servers = json.loads(await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:terminal_servers'))
request.app.state.TERMINAL_SERVERS = terminal_servers
except Exception as e:
log.error(f'Error fetching terminal_servers from Redis: {e}')
@@ -975,8 +1034,8 @@ async def get_terminal_tools(
log.warning(f'Terminal server not found: {terminal_id}')
return {}
- user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
- if not has_connection_access(user, connection, user_group_ids):
+ user_group_ids = {group.id for group in await Groups.get_groups_by_member_id(user.id)}
+ if not await has_connection_access(user, connection, user_group_ids):
log.warning(f'Access denied to terminal {terminal_id} for user {user.id}')
return {}
@@ -1028,7 +1087,7 @@ async def get_terminal_tools(
tool_spec.get('description', '') + f'\n\nThe current working directory is: {terminal_cwd}'
)
- def make_tool_function(fn_name, srv_data, hdrs, cks):
+ async def make_tool_function(fn_name, srv_data, hdrs, cks):
async def tool_function(**kwargs):
return await execute_tool_server(
url=srv_data['url'],
@@ -1041,8 +1100,8 @@ async def get_terminal_tools(
return tool_function
- tool_function = make_tool_function(function_name, server_data, headers, cookies)
- callable = get_async_tool_function_and_apply_extra_params(tool_function, {})
+ tool_function = await make_tool_function(function_name, server_data, headers, cookies)
+ callable = await get_async_tool_function_and_apply_extra_params(tool_function, {})
tools_dict[function_name] = {
'tool_id': f'terminal:{terminal_id}',
diff --git a/backend/open_webui/utils/validate.py b/backend/open_webui/utils/validate.py
index eec53317f5..1e98b41105 100644
--- a/backend/open_webui/utils/validate.py
+++ b/backend/open_webui/utils/validate.py
@@ -1,13 +1,31 @@
"""Validation utilities for user-supplied input."""
-# Known static asset paths used as default profile images
-_ALLOWED_STATIC_PATHS = (
- '/user.png',
- '/static/favicon.png',
-)
+import re
+from urllib.parse import urlparse
-# External URL prefixes that are explicitly trusted for profile images
-_ALLOWED_URL_PREFIXES = ('https://www.gravatar.com/avatar/',)
+# Matches the OWUI-generated profile image route. ``[^/?#]+`` accepts
+# any user-ID without allowing path-traversal or query/fragment injection,
+# and the ``$`` anchor rejects trailing path components.
+_USER_PROFILE_IMAGE_RE = re.compile(r'^/api/v1/users/[^/?#]+/profile/image$')
+
+# Validates MIME type and structure of base64 data URIs. Only the prefix
+# is checked — validating the full base64 payload would mean running a
+# regex across megabytes of data on every Pydantic instantiation for zero
+# security benefit (corrupt base64 simply renders a broken image, same as
+# a 404 URL). SVG is intentionally excluded: it can carry embedded scripts.
+_SAFE_DATA_URI_RE = re.compile(r'^data:image/(png|jpeg|gif|webp);base64,', re.IGNORECASE)
+
+# Exact relative paths accepted as profile images. These are the only
+# static-asset paths OWUI itself assigns; no prefix/wildcard matching is
+# used so that arbitrary relative paths cannot trigger authenticated GETs
+# against internal endpoints when rendered as ``
diff --git a/src/lib/components/automations/AutomationEditor.svelte b/src/lib/components/automations/AutomationEditor.svelte
index fbdf448179..463cb0fa70 100644
--- a/src/lib/components/automations/AutomationEditor.svelte
+++ b/src/lib/components/automations/AutomationEditor.svelte
@@ -19,7 +19,6 @@
type AutomationResponse,
type AutomationRunModel
} from '$lib/apis/automations';
- import { getTerminalServers, type TerminalServer } from '$lib/apis/terminal/index';
import Spinner from '$lib/components/common/Spinner.svelte';
import Tooltip from '$lib/components/common/Tooltip.svelte';
@@ -29,7 +28,6 @@
import ScheduleDropdown from '$lib/components/automations/ScheduleDropdown.svelte';
import ModelDropdown from '$lib/components/automations/ModelDropdown.svelte';
- import TerminalDropdown from '$lib/components/automations/TerminalDropdown.svelte';
dayjs.extend(relativeTime);
dayjs.extend(localizedFormat);
@@ -43,10 +41,6 @@
let model_id = '';
let is_active = true;
- let terminalServers: TerminalServer[] = [];
- let terminalServerId = '';
- let terminalCwd = '';
-
let loading = false;
let saving = false;
let showDeleteConfirm = false;
@@ -97,15 +91,7 @@
data: {
prompt: prompt.trim(),
model_id: model_id.trim(),
- rrule: scheduleDropdown.buildRrule(),
- ...(terminalServerId
- ? {
- terminal: {
- server_id: terminalServerId,
- ...(terminalCwd.trim() ? { cwd: terminalCwd.trim() } : {})
- }
- }
- : {})
+ rrule: scheduleDropdown.buildRrule()
},
is_active
};
@@ -204,18 +190,11 @@
prompt = automation.data.prompt;
model_id = automation.data.model_id;
is_active = automation.is_active;
- terminalServerId = automation.data.terminal?.server_id || '';
- terminalCwd = automation.data.terminal?.cwd || '';
if (scheduleDropdown) {
scheduleDropdown.parseRrule(automation.data.rrule);
}
- try {
- terminalServers = await getTerminalServers(localStorage.token);
- } catch {
- terminalServers = [];
- }
await loadRuns();
});
@@ -354,21 +333,6 @@
{$i18n.t('Model')}
-
-
- {#if terminalServers.length > 0}
-
- {$i18n.t('Terminal')}
-
- {/if}
diff --git a/src/lib/components/automations/ScheduleDropdown.svelte b/src/lib/components/automations/ScheduleDropdown.svelte
index 67f4fbce69..4f7b63a541 100644
--- a/src/lib/components/automations/ScheduleDropdown.svelte
+++ b/src/lib/components/automations/ScheduleDropdown.svelte
@@ -1,9 +1,10 @@
+
+{#if terminals.length > 0}
+
+
{$i18n.t('Terminal')}
+
+ {$i18n.t('None')}
+ {#each terminals as terminal (terminal.id)}
+ {terminal.name || terminal.id}
+ {/each}
+
+{/if}
diff --git a/src/lib/components/workspace/Prompts/PromptEditor.svelte b/src/lib/components/workspace/Prompts/PromptEditor.svelte
index 66db6df208..5a2cf1389b 100644
--- a/src/lib/components/workspace/Prompts/PromptEditor.svelte
+++ b/src/lib/components/workspace/Prompts/PromptEditor.svelte
@@ -82,23 +82,27 @@
loading = true;
if (validateCommandString(command)) {
- await onSubmit({
- id: prompt?.id,
- name,
- command,
- content,
- tags: tags.map((tag) => tag.name),
- access_grants: accessGrants,
- commit_message: commitMessage || undefined,
- is_production: isProduction
- });
- showEditModal = false;
- commitMessage = '';
- isProduction = true;
- await loadHistory(true); // Reset and reload
- // Select the newest version after saving
- if (history.length > 0) {
- selectedHistoryEntry = history[0];
+ try {
+ await onSubmit({
+ id: prompt?.id,
+ name,
+ command,
+ content,
+ tags: tags.map((tag) => tag.name),
+ access_grants: accessGrants,
+ commit_message: commitMessage || undefined,
+ is_production: isProduction
+ });
+ showEditModal = false;
+ commitMessage = '';
+ isProduction = true;
+ await loadHistory(true); // Reset and reload
+ // Select the newest version after saving
+ if (history.length > 0) {
+ selectedHistoryEntry = history[0];
+ }
+ } catch (error) {
+ toast.error(`${error}`);
}
} else {
toast.error(
diff --git a/src/lib/i18n/locales/ar-BH/translation.json b/src/lib/i18n/locales/ar-BH/translation.json
index 34647cd1f6..c390eff0fd 100644
--- a/src/lib/i18n/locales/ar-BH/translation.json
+++ b/src/lib/i18n/locales/ar-BH/translation.json
@@ -458,6 +458,7 @@
"Create a new note": "",
"Create Account": "إنشاء حساب",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -483,6 +484,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "مظلم",
@@ -973,6 +975,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1047,6 +1050,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "كيف استطيع مساعدتك اليوم؟",
"How would you rate this response?": "",
@@ -1269,10 +1273,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "تم تحميل النموذج '{{modelName}}' بنجاح",
"Model '{{modelTag}}' is already in queue for downloading.": "النموذج '{{modelTag}}' موجود بالفعل في قائمة الانتظار للتحميل",
- "Model {{modelId}} not found": "لم يتم العثور على النموذج {{modelId}}.",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "نموذج {{modelName}} غير قادر على الرؤية",
"Model {{name}} is now {{status}}": "نموذج {{name}} هو الآن {{status}}",
@@ -1313,6 +1317,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "المزيد",
"More Concise": "",
"More options": "",
@@ -1433,6 +1438,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama الاصدار",
"On": "تشغيل",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1517,6 +1523,7 @@
"Persistent": "",
"Personalization": "التخصيص",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1681,6 +1688,7 @@
"Running": "",
"Running...": "جارٍ التنفيذ...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "حفظ",
"Save & Create": "حفظ وإنشاء",
"Save & Update": "حفظ وتحديث",
@@ -1897,6 +1905,7 @@
"STT Model": "",
"STT Settings": "STT اعدادات",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1942,6 +1951,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "محرك تحويل النص إلى كلام",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "شكرا لملاحظاتك!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2055,6 +2065,7 @@
"TTS Model": "",
"TTS Settings": "TTS اعدادات",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "نوع",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "اكتب عنوان URL لحل مشكلة الوجه (تنزيل).",
@@ -2157,6 +2168,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "",
"Web Loader Engine": "",
@@ -2173,6 +2185,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ar/translation.json b/src/lib/i18n/locales/ar/translation.json
index 9e64b36427..50efd0c748 100644
--- a/src/lib/i18n/locales/ar/translation.json
+++ b/src/lib/i18n/locales/ar/translation.json
@@ -458,6 +458,7 @@
"Create a new note": "",
"Create Account": "إنشاء حساب",
"Create Admin Account": "إنشاء حساب مسؤول",
+ "Create and manage scheduled automations": "",
"Create Channel": "إنشاء قناة",
"Create Folder": "",
"Create Image": "",
@@ -483,6 +484,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "منطقة الخطر",
"Dark": "داكن",
@@ -973,6 +975,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "وضع السياق الكامل",
"Function": "وظيفة",
"Function Calling": "استدعاء الوظائف",
@@ -1047,6 +1050,7 @@
"History": "",
"Home": "الصفحة الرئيسية",
"Host": "المضيف",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "كيف استطيع مساعدتك اليوم؟",
"How would you rate this response?": "كيف تقيّم هذا الرد؟",
@@ -1269,10 +1273,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "النموذج",
"Model '{{modelName}}' has been successfully downloaded.": "تم تحميل النموذج '{{modelName}}' بنجاح",
"Model '{{modelTag}}' is already in queue for downloading.": "النموذج '{{modelTag}}' موجود بالفعل في قائمة الانتظار للتحميل",
- "Model {{modelId}} not found": "لم يتم العثور على النموذج {{modelId}}.",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "نموذج {{modelName}} غير قادر على الرؤية",
"Model {{name}} is now {{status}}": "نموذج {{name}} هو الآن {{status}}",
@@ -1313,6 +1317,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "مفتاح API لـ Mojeek Search",
+ "Monthly": "",
"More": "المزيد",
"More Concise": "",
"More options": "",
@@ -1433,6 +1438,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama الاصدار",
"On": "تشغيل",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1517,6 +1523,7 @@
"Persistent": "",
"Personalization": "التخصيص",
"Pin": "تثبيت",
+ "Pin to Sidebar": "",
"Pinned": "مثبت",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1681,6 +1688,7 @@
"Running": "جارٍ التنفيذ",
"Running...": "جارٍ التنفيذ...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "حفظ",
"Save & Create": "حفظ وإنشاء",
"Save & Update": "حفظ وتحديث",
@@ -1897,6 +1905,7 @@
"STT Model": "نموذج تحويل الصوت إلى نص (STT)",
"STT Settings": "STT اعدادات",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1942,6 +1951,7 @@
"Text Splitter": "تقسيم النص",
"Text-to-Speech": "",
"Text-to-Speech Engine": "محرك تحويل النص إلى كلام",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "شكرا لملاحظاتك!",
"The Application Account DN you bind with for search": "DN لحساب التطبيق الذي تستخدمه للبحث",
"The base to search for users": "الأساس الذي يُستخدم للبحث عن المستخدمين",
@@ -2055,6 +2065,7 @@
"TTS Model": "نموذج تحويل النص إلى كلام (TTS)",
"TTS Settings": "TTS اعدادات",
"TTS Voice": "صوت TTS",
+ "Tu_day_of_week": "",
"Type": "نوع",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "اكتب عنوان URL لحل مشكلة الوجه (تنزيل).",
@@ -2157,6 +2168,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "تحذير: تفعيل هذا الخيار سيسمح للمستخدمين برفع كود عشوائي على الخادم.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "تحذير: تنفيذ كود Jupyter يتيح تنفيذ كود عشوائي مما يشكل مخاطر أمنية جسيمة—تابع بحذر شديد.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "واجهة برمجة التطبيقات (API)",
"Web Loader Engine": "",
@@ -2173,6 +2185,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "ستقوم WebUI بإرسال الطلبات إلى \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "ستقوم WebUI بإرسال الطلبات إلى \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "ما الذي تحاول تحقيقه؟",
"What are you working on?": "على ماذا تعمل؟",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/az-AZ/translation.json b/src/lib/i18n/locales/az-AZ/translation.json
index e9c75e8671..5aa624f6d3 100644
--- a/src/lib/i18n/locales/az-AZ/translation.json
+++ b/src/lib/i18n/locales/az-AZ/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Yeni qeyd yarat",
"Create Account": "Hesab yarat",
"Create Admin Account": "Admin hesabı yarat",
+ "Create and manage scheduled automations": "",
"Create Channel": "Kanal yarat",
"Create Folder": "Qovluq yarat",
"Create Image": "Şəkil yarat",
@@ -479,6 +480,7 @@
"Custom Gender": "Fərdi cins",
"Custom Parameter Name": "Fərdi parametr adı",
"Custom Parameter Value": "Fərdi parametr dəyəri",
+ "Daily": "",
"Daily Messages": "Gündəlik mesajlar",
"Danger Zone": "Təhlükəli zona",
"Dark": "Tünd",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Autentifikasiya üçün sistem istifadəçisinin OAuth giriş tokenini yönləndirir",
"Forwards system user session credentials to authenticate": "Autentifikasiya üçün sistem istifadəçisinin sessiya məlumatlarını yönləndirir",
+ "Fr_day_of_week": "",
"Full Context Mode": "Tam kontekst rejimi",
"Function": "Funksiya",
"Function Calling": "Funksiya çağırışı",
@@ -1043,6 +1046,7 @@
"History": "Tarixçə",
"Home": "Ana səhifə",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "Saatlıq mesajlar",
"How can I help you today?": "Bu gün sizə necə kömək edə bilərəm?",
"How would you rate this response?": "Bu cavabı necə qiymətləndirərdiniz?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API açarı tələb olunur.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "'{{modelName}}' modeli uğurla yükləndi.",
"Model '{{modelTag}}' is already in queue for downloading.": "'{{modelTag}}' modeli artıq yükləmə növbəsindədir.",
- "Model {{modelId}} not found": "{{modelId}} modeli tapılmadı",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "{{modelName}} modeli görüntünü tanıma (vision) qabiliyyətinə malik deyil",
"Model {{name}} is now {{status}}": "{{name}} modeli indi {{status}} statusundadır",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Modellərin paylaşılması",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Mojeek Axtarış API Açarı",
+ "Monthly": "",
"More": "Daha çox",
"More Concise": "Daha yığcam",
"More options": "Daha çox seçim",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama Cloud API Açarı",
"Ollama Version": "Ollama Versiyası",
"On": "Açıq",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Yalnız \"Böyük mətni fayl kimi yapışdır\" ayarı aktiv olduqda işləyir.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Yalnız çat girişi fokusda olduqda və LLM cavab yaratdıqda aktiv olur.",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Fərdiləşdirmə",
"Pin": "Bərkit",
+ "Pin to Sidebar": "",
"Pinned": "Bərkidilib",
"Pinned Messages": "Bərkidilmiş mesajlar",
"Pinned Models": "Bərkidilmiş modellər",
@@ -1673,6 +1680,7 @@
"Running": "İcra edilir",
"Running...": "İcra edilir...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Emalı sürətləndirmək üçün yerləşdirmə (embedding) tapşırıqlarını eyni vaxtda icra edir. Əgər sorğu limiti problemi yaranarsa, bunu söndürün.",
+ "Sa_day_of_week": "",
"Save": "Yadda saxla",
"Save & Create": "Saxla və Yarat",
"Save & Update": "Saxla və Yenilə",
@@ -1889,6 +1897,7 @@
"STT Model": "STT (Səsdən mətnə) Modeli",
"STT Settings": "STT Ayarları",
"Stylized PDF Export": "Stil verilmiş PDF ixracı",
+ "Su_day_of_week": "",
"Submit question": "Sualı göndər",
"Submit suggestion": "Təklifi göndər",
"Subtitle": "Altyazı",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Mətn bölücü",
"Text-to-Speech": "Mətndən səsə (TTS)",
"Text-to-Speech Engine": "Mətndən səsə mühərriki",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Rəyiniz üçün təşəkkürlər!",
"The Application Account DN you bind with for search": "Axtarış üçün bağladığınız Tətbiq Hesabı DN (Application Account DN)",
"The base to search for users": "İstifadəçilərin axtarışı üçün baza",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS Modeli",
"TTS Settings": "TTS Ayarları",
"TTS Voice": "TTS Səsi",
+ "Tu_day_of_week": "",
"Type": "Növ",
"Type here...": "Bura yazın...",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (Yükləmə) URL-ni yazın",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Xəbərdarlıq: Bunun aktiv edilməsi istifadəçilərə serverə ixtiyari kod yükləməyə icazə verəcək.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Xəbərdarlıq: Jupyter icrası ixtiyari kodun işlədilməsinə imkan verir və ciddi təhlükəsizlik riskləri yaradır — son dərəcə ehtiyatlı olun.",
+ "We_day_of_week": "",
"Web": "Veb",
"Web API": "Veb API",
"Web Loader Engine": "Veb yükləyici mühərrik",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI \"{{url}}\" ünvanına sorğular göndərəcək",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI \"{{url}}/api/chat\" ünvanına sorğular göndərəcək",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI \"{{url}}/chat/completions\" ünvanına sorğular göndərəcək",
+ "Weekly": "",
"What are you trying to achieve?": "Nəyə nail olmaq istəyirsiniz?",
"What are you working on?": "Nəyin üzərində işləyirsiniz?",
"What is NOT shared:": "Nələr paylaşılmır:",
diff --git a/src/lib/i18n/locales/bg-BG/translation.json b/src/lib/i18n/locales/bg-BG/translation.json
index e425f2063d..07ca792153 100644
--- a/src/lib/i18n/locales/bg-BG/translation.json
+++ b/src/lib/i18n/locales/bg-BG/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Създаване на Акаунт",
"Create Admin Account": "Създаване на администраторски акаунт",
+ "Create and manage scheduled automations": "",
"Create Channel": "Създаване на канал",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Тъмен",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "Режим на пълен контекст",
"Function": "Функция",
"Function Calling": "Извикване на функция",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Начало",
"Host": "Хост",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Как мога да ви помогна днес?",
"How would you rate this response?": "Как бихте оценили този отговор?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Модел",
"Model '{{modelName}}' has been successfully downloaded.": "Моделът '{{modelName}}' беше успешно свален.",
"Model '{{modelTag}}' is already in queue for downloading.": "Моделът '{{modelTag}}' е вече в очакване за сваляне.",
- "Model {{modelId}} not found": "Моделът {{modelId}} не е намерен",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Моделът {{modelName}} не поддържа визуални възможности",
"Model {{name}} is now {{status}}": "Моделът {{name}} сега е {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "API ключ за Mojeek Search",
+ "Monthly": "",
"More": "Повече",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama Версия",
"On": "Вкл.",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Персонализация",
"Pin": "Закачи",
+ "Pin to Sidebar": "",
"Pinned": "Закачено",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Изпълнява се",
"Running...": "Изпълнява се...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Запис",
"Save & Create": "Запис & Създаване",
"Save & Update": "Запис & Актуализиране",
@@ -1889,6 +1897,7 @@
"STT Model": "STT Модел",
"STT Settings": "STT Настройки",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Разделител на текст",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Двигател за преобразуване на текст в реч",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Благодарим ви за вашия отзив!",
"The Application Account DN you bind with for search": "DN на акаунта на приложението, с който се свързвате за търсене",
"The base to search for users": "Базата за търсене на потребители",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS Модел",
"TTS Settings": "TTS Настройки",
"TTS Voice": "TTS Глас",
+ "Tu_day_of_week": "",
"Type": "Вид",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Въведете Hugging Face Resolve (Изтегляне) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Предупреждение: Активирането на това ще позволи на потребителите да качват произволен код на сървъра.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Предупреждение: Изпълнението на Jupyter позволява произволно изпълнение на код, което представлява сериозни рискове за сигурността-продължете с изключително внимание.",
+ "We_day_of_week": "",
"Web": "Уеб",
"Web API": "Уеб API",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI ще прави заявки към \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI ще прави заявки към \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Какво се опитвате да постигнете?",
"What are you working on?": "Върху какво работите?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/bn-BD/translation.json b/src/lib/i18n/locales/bn-BD/translation.json
index fecb6f5ffe..a20abb0d9a 100644
--- a/src/lib/i18n/locales/bn-BD/translation.json
+++ b/src/lib/i18n/locales/bn-BD/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "একাউন্ট তৈরি করুন",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "ডার্ক",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "আপনাকে আজ কিভাবে সাহায্য করতে পারি?",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "'{{modelName}}' মডেল সফলভাবে ডাউনলোড হয়েছে।",
"Model '{{modelTag}}' is already in queue for downloading.": "{{modelTag}} ডাউনলোডের জন্য আগে থেকেই অপেক্ষমান আছে।",
- "Model {{modelId}} not found": "{{modelId}} মডেল পাওয়া যায়নি",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "মডেল {{modelName}} দৃষ্টি সক্ষম নয়",
"Model {{name}} is now {{status}}": "মডেল {{name}} এখন {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "আরো",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama ভার্সন",
"On": "চালু",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "ডিজিটাল বাংলা",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "চলমান...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "সংরক্ষণ",
"Save & Create": "সংরক্ষণ এবং তৈরি করুন",
"Save & Update": "সংরক্ষণ এবং আপডেট করুন",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "STT সেটিংস",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "টেক্সট-টু-স্পিচ ইঞ্জিন",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "আপনার মতামত ধন্যবাদ!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "TTS সেটিংসমূহ",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "টাইপ",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face থেকে ডাউনলোড করার ইউআরএল টাইপ করুন",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "ওয়েব",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/bo-TB/translation.json b/src/lib/i18n/locales/bo-TB/translation.json
index 30ca5f031a..fef2cb1847 100644
--- a/src/lib/i18n/locales/bo-TB/translation.json
+++ b/src/lib/i18n/locales/bo-TB/translation.json
@@ -453,6 +453,7 @@
"Create a new note": "",
"Create Account": "རྩིས་ཁྲ་གསར་བཟོ།",
"Create Admin Account": "དོ་དམ་པའི་རྩིས་ཁྲ་གསར་བཟོ།",
+ "Create and manage scheduled automations": "",
"Create Channel": "བགྲོ་གླེང་གསར་བཟོ།",
"Create Folder": "",
"Create Image": "",
@@ -478,6 +479,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "ཉེན་ཁའི་ས་ཁུལ།",
"Dark": "ནག་པོ།",
@@ -968,6 +970,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "ནང་དོན་ཆ་ཚང་མ་དཔེ།",
"Function": "ལས་འགན།",
"Function Calling": "ལས་འགན་འབོད་པ།",
@@ -1042,6 +1045,7 @@
"History": "",
"Home": "གཙོ་ངོས།",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "དེ་རིང་ངས་ཁྱེད་ལ་རོགས་པ་ཅི་ཞིག་བྱེད་ཐུབ་བམ།",
"How would you rate this response?": "ལན་འདི་ལ་ཁྱེད་ཀྱིས་སྐར་མ་ག་ཚོད་སྤྲོད་འདོད་དམ།",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "དཔེ་དབྱིབས།",
"Model '{{modelName}}' has been successfully downloaded.": "དཔེ་དབྱིབས། '{{modelName}}' ལེགས་པར་ཕབ་ལེན་བྱས་ཟིན།",
"Model '{{modelTag}}' is already in queue for downloading.": "དཔེ་དབྱིབས། '{{modelTag}}' ཕབ་ལེན་གྱི་སྒུག་ཐོ་ནང་ཡོད་ཟིན།",
- "Model {{modelId}} not found": "དཔེ་དབྱིབས། {{modelId}} མ་རྙེད།",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "དཔེ་དབྱིབས། {{modelName}} ལ་མཐོང་ནུས་མེད།",
"Model {{name}} is now {{status}}": "དཔེ་དབྱིབས། {{name}} ད་ལྟ་ {{status}} ཡིན།",
@@ -1308,6 +1312,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API ལྡེ་མིག",
+ "Monthly": "",
"More": "མང་བ།",
"More Concise": "",
"More options": "",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama པར་གཞི།",
"On": "ཁ་ཕྱེ་བ།",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "སྒེར་སྤྱོད་ཅན།",
"Pin": "གདབ་པ།",
+ "Pin to Sidebar": "",
"Pinned": "གདབ་ཟིན།",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1671,6 +1678,7 @@
"Running": "ལག་བསྟར་བྱེད་བཞིན་པ།",
"Running...": "ལག་བསྟར་བྱེད་བཞིན་པ།...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "ཉར་ཚགས།",
"Save & Create": "ཉར་ཚགས་ & གསར་བཟོ།",
"Save & Update": "ཉར་ཚགས་ & གསར་སྒྱུར།",
@@ -1887,6 +1895,7 @@
"STT Model": "STT དཔེ་དབྱིབས།",
"STT Settings": "STT སྒྲིག་འགོད།",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1932,6 +1941,7 @@
"Text Splitter": "ཡིག་རྐྱང་བགོ་བྱེད།",
"Text-to-Speech": "",
"Text-to-Speech Engine": "ཡིག་རྐྱང་ནས་གཏམ་བཤད་ཀྱི་འཕྲུལ་འཁོར།",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "ཁྱེད་ཀྱི་བསམ་འཆར་ལ་ཐུགས་རྗེ་ཆེ།",
"The Application Account DN you bind with for search": "ཁྱེད་ཀྱིས་འཚོལ་བཤེར་གྱི་ཆེད་དུ་སྦྲེལ་བའི་ Application Account DN",
"The base to search for users": "བེད་སྤྱོད་མཁན་འཚོལ་བའི་གཞི་རྩ།",
@@ -2045,6 +2055,7 @@
"TTS Model": "TTS དཔེ་དབྱིབས།",
"TTS Settings": "TTS སྒྲིག་འགོད།",
"TTS Voice": "TTS སྐད།",
+ "Tu_day_of_week": "",
"Type": "རིགས།",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (ཕབ་ལེན།) URL མནན་པ།",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "ཉེན་བརྡ།: འདི་སྒུལ་བསྐྱོད་བྱས་ན་བེད་སྤྱོད་མཁན་ཚོས་སར་བར་སྟེང་གང་འདོད་ཀྱི་ཀོཌ་སྤར་བར་གནང་བ་སྤྲོད་ངེས།",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "ཉེན་བརྡ།: Jupyter ལག་བསྟར་གྱིས་གང་འདོད་ཀྱི་ཀོཌ་ལག་བསྟར་སྒུལ་བསྐྱོད་བྱས་ནས། བདེ་འཇགས་ཀྱི་ཉེན་ཁ་ཚབས་ཆེན་བཟོ་གི་ཡོད།—ཧ་ཅང་གཟབ་ནན་གྱིས་སྔོན་སྐྱོད་བྱེད་རོགས།",
+ "We_day_of_week": "",
"Web": "དྲ་བ།",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI ཡིས་ \"{{url}}/api/chat\" ལ་རེ་ཞུ་གཏོང་ངེས།",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI ཡིས་ \"{{url}}/chat/completions\" ལ་རེ་ཞུ་གཏོང་ངེས།",
+ "Weekly": "",
"What are you trying to achieve?": "ཁྱེད་ཀྱིས་ཅི་ཞིག་འགྲུབ་ཐབས་བྱེད་བཞིན་ཡོད།",
"What are you working on?": "ཁྱེད་ཀྱིས་ཅི་ཞིག་ལས་ཀ་བྱེད་བཞིན་ཡོད།",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/bs-BA/translation.json b/src/lib/i18n/locales/bs-BA/translation.json
index e539d89c1c..223a0ff884 100644
--- a/src/lib/i18n/locales/bs-BA/translation.json
+++ b/src/lib/i18n/locales/bs-BA/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "Stvori račun",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Tamno",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Kako vam mogu pomoći danas?",
"How would you rate this response?": "",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' je uspješno preuzet.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' je već u redu za preuzimanje.",
- "Model {{modelId}} not found": "Model {{modelId}} nije pronađen",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} ne čita vizualne impute",
"Model {{name}} is now {{status}}": "Model {{name}} sada je {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Više",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama verzija",
"On": "Uključeno",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Prilagodba",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "Pokrenuto",
"Running...": "Pokrenuto...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Spremi",
"Save & Create": "Spremi i stvori",
"Save & Update": "Spremi i ažuriraj",
@@ -1891,6 +1899,7 @@
"STT Model": "STT model",
"STT Settings": "STT postavke",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Stroj za pretvorbu teksta u govor",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Hvala na povratnim informacijama!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2049,6 +2059,7 @@
"TTS Model": "TTS model",
"TTS Settings": "TTS postavke",
"TTS Voice": "TTS glas",
+ "Tu_day_of_week": "",
"Type": "Tip",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Upišite Hugging Face Resolve (Download) URL",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ca-ES/translation.json b/src/lib/i18n/locales/ca-ES/translation.json
index fad81172aa..fd49c6ae07 100644
--- a/src/lib/i18n/locales/ca-ES/translation.json
+++ b/src/lib/i18n/locales/ca-ES/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "Crear una nova nota",
"Create Account": "Crear un compte",
"Create Admin Account": "Crear un compte d'Administrador",
+ "Create and manage scheduled automations": "",
"Create Channel": "Crear un canal",
"Create Folder": "Crear carpeta",
"Create Image": "Crear imatge",
@@ -480,6 +481,7 @@
"Custom Gender": "Gènere personalitzat",
"Custom Parameter Name": "Nom del paràmetre personalitzat",
"Custom Parameter Value": "Valor del paràmetre personalitzat",
+ "Daily": "",
"Daily Messages": "Missatges diaris",
"Danger Zone": "Zona de perill",
"Dark": "Fosc",
@@ -970,6 +972,7 @@
"Forward": "Endavant",
"Forwards system user OAuth access token to authenticate": "Reenvia el testimoni d'accés OAuth de l'usuari del sistema per autenticar-se.",
"Forwards system user session credentials to authenticate": "Envia les credencials de l'usuari del sistema per autenticar",
+ "Fr_day_of_week": "",
"Full Context Mode": "Mode de context complert",
"Function": "Funció",
"Function Calling": "Crida a funcions",
@@ -1044,6 +1047,7 @@
"History": "Historial",
"Home": "Inici",
"Host": "Servidor",
+ "Hourly": "",
"Hourly Messages": "Missatges horaris",
"How can I help you today?": "Com et puc ajudar avui?",
"How would you rate this response?": "Com avaluaries aquesta resposta?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "És necessària la clau API de Mistral OCR",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "El model '{{modelName}}' s'ha descarregat correctament.",
"Model '{{modelTag}}' is already in queue for downloading.": "El model '{{modelTag}}' ja està en cua per ser descarregat.",
- "Model {{modelId}} not found": "No s'ha trobat el model {{modelId}}",
"Model {{modelName}} deleted successfully": "El model {{modelName}} s'ha eliminat correctament",
"Model {{modelName}} is not vision capable": "El model {{modelName}} no és capaç de visió",
"Model {{name}} is now {{status}}": "El model {{name}} ara és {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "Compartir els models",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Clau API de Mojeek Search",
+ "Monthly": "",
"More": "Més",
"More Concise": "Més precís",
"More options": "Més opcions",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "Clau API d'Ollama Cloud",
"Ollama Version": "Versió d'Ollama",
"On": "Activat",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Només està actiu quan l'opció \"Enganxa text gran com a fitxer\" està activada.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Només s'activa quan l'entrada del xat està en focus i un LLM està generant una resposta.",
@@ -1514,6 +1520,7 @@
"Persistent": "Persistent",
"Personalization": "Personalització",
"Pin": "Fixar",
+ "Pin to Sidebar": "",
"Pinned": "Fixat",
"Pinned Messages": "Missatges fixats",
"Pinned Models": "Models fixats",
@@ -1675,6 +1682,7 @@
"Running": "S'està executant",
"Running...": "S'està executant...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Executa tasques d'incrustació simultàniament per accelerar el processament. Desactiva-ho si els límits de velocitat es converteixen en un problema.",
+ "Sa_day_of_week": "",
"Save": "Desar",
"Save & Create": "Desar i crear",
"Save & Update": "Desar i actualitzar",
@@ -1891,6 +1899,7 @@
"STT Model": "Model SST",
"STT Settings": "Preferències de STT",
"Stylized PDF Export": "Exportació en PDF estilitzat",
+ "Su_day_of_week": "",
"Submit question": "Enviar la pregunta",
"Submit suggestion": "Enviar un suggeriment",
"Subtitle": "Subtítol",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Separador de text",
"Text-to-Speech": "Text-a-veu",
"Text-to-Speech Engine": "Motor de text a veu",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Gràcies pel teu comentari!",
"The Application Account DN you bind with for search": "El DN del compte d'aplicació per realitzar la cerca",
"The base to search for users": "La base per cercar usuaris",
@@ -2049,6 +2059,7 @@
"TTS Model": "Model TTS",
"TTS Settings": "Preferències de TTS",
"TTS Voice": "Veu TTS",
+ "Tu_day_of_week": "",
"Type": "Tipus",
"Type here...": "Escriu aquí...",
"Type Hugging Face Resolve (Download) URL": "Escriu la URL de Resolució (Descàrrega) de Hugging Face",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "Avís: Si actives aquesta opció, els usuaris podran executar sol·licituds programades automàticament.",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Avís: Habilitar això permetrà als usuaris penjar codi arbitrari al servidor.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Avís: l'execució de Jupyter permet l'execució de codi arbitrari, la qual cosa comporta greus riscos de seguretat; procediu amb extrema precaució.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "Motor de càrrega Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI farà peticions a \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI farà peticions a \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI farà peticions a \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Què intentes aconseguir?",
"What are you working on?": "En què estàs treballant?",
"What is NOT shared:": "Què no es comparteix",
diff --git a/src/lib/i18n/locales/ceb-PH/translation.json b/src/lib/i18n/locales/ceb-PH/translation.json
index 138fa01b1e..34a49f5a86 100644
--- a/src/lib/i18n/locales/ceb-PH/translation.json
+++ b/src/lib/i18n/locales/ceb-PH/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Paghimo og account",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Ngitngit",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Unsaon nako pagtabang kanimo karon?",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "Ang modelo'{{modelName}}' malampuson nga na-download.",
"Model '{{modelTag}}' is already in queue for downloading.": "Ang modelo'{{modelTag}}' naa na sa pila para ma-download.",
- "Model {{modelId}} not found": "Modelo {{modelId}} wala makit-an",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "",
"Model {{name}} is now {{status}}": "",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama nga bersyon",
"On": "Gipaandar",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "Nagdagan...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Tipigi",
"Save & Create": "I-save ug Paghimo",
"Save & Update": "I-save ug I-update",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "Mga setting sa STT",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Text-to-speech nga makina",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "Mga Setting sa TTS",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Pagsulod sa resolusyon (pag-download) URL Hugging Face",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/cs-CZ/translation.json b/src/lib/i18n/locales/cs-CZ/translation.json
index 6bb5fca5b3..227d6b8a61 100644
--- a/src/lib/i18n/locales/cs-CZ/translation.json
+++ b/src/lib/i18n/locales/cs-CZ/translation.json
@@ -456,6 +456,7 @@
"Create a new note": "",
"Create Account": "Vytvořit účet",
"Create Admin Account": "Vytvořit účet administrátora",
+ "Create and manage scheduled automations": "",
"Create Channel": "Vytvořit kanál",
"Create Folder": "Vytvořit složku",
"Create Image": "",
@@ -481,6 +482,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Název vlastního parametru",
"Custom Parameter Value": "Hodnota vlastního parametru",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Nebezpečná zóna",
"Dark": "Tmavý",
@@ -971,6 +973,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Přeposílá přihlašovací údaje relace systémového uživatele pro ověření",
+ "Fr_day_of_week": "",
"Full Context Mode": "Režim plného kontextu",
"Function": "Funkce",
"Function Calling": "function calling",
@@ -1045,6 +1048,7 @@
"History": "",
"Home": "Domů",
"Host": "Hostitel",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Jak vám dnes mohu pomoci?",
"How would you rate this response?": "Jak byste ohodnotili tuto odpověď?",
@@ -1267,10 +1271,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Je vyžadován API klíč pro Mistral OCR.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' byl úspěšně stažen.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' je již ve frontě na stažení.",
- "Model {{modelId}} not found": "Model {{modelId}} nenalezen",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} nemá schopnost zpracování obrazu.",
"Model {{name}} is now {{status}}": "Model {{name}} je nyní {{status}}.",
@@ -1311,6 +1315,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "API klíč pro Mojeek Search",
+ "Monthly": "",
"More": "Více",
"More Concise": "Stručnější",
"More options": "",
@@ -1431,6 +1436,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Verze Ollama",
"On": "Zapnuto",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1515,6 +1521,7 @@
"Persistent": "",
"Personalization": "Personalizace",
"Pin": "Připnout",
+ "Pin to Sidebar": "",
"Pinned": "Připnuto",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1677,6 +1684,7 @@
"Running": "Běží",
"Running...": "Běží...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Uložit",
"Save & Create": "Uložit a vytvořit",
"Save & Update": "Uložit a aktualizovat",
@@ -1893,6 +1901,7 @@
"STT Model": "Model STT",
"STT Settings": "Nastavení STT",
"Stylized PDF Export": "Stylizovaný export do PDF",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1938,6 +1947,7 @@
"Text Splitter": "Rozdělovač textu",
"Text-to-Speech": "Převod textu na řeč",
"Text-to-Speech Engine": "Jádro pro převod textu na řeč",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Děkujeme za vaši zpětnou vazbu!",
"The Application Account DN you bind with for search": "DN aplikačního účtu, se kterým se vážete pro vyhledávání",
"The base to search for users": "Základ pro vyhledávání uživatelů",
@@ -2051,6 +2061,7 @@
"TTS Model": "Model TTS",
"TTS Settings": "Nastavení TTS",
"TTS Voice": "Hlas TTS",
+ "Tu_day_of_week": "",
"Type": "Typ",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Zadejte URL pro stažení z Hugging Face.",
@@ -2153,6 +2164,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Varování: Povolení této volby umožní uživatelům nahrávat na server libovolný kód.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Varování: Spouštění Jupyteru umožňuje provádění libovolného kódu, což představuje vážná bezpečnostní rizika – postupujte s maximální opatrností.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Webové API",
"Web Loader Engine": "Jádro webového zavaděče",
@@ -2169,6 +2181,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI bude odesílat požadavky na \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI bude odesílat požadavky na \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI bude odesílat požadavky na \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Čeho se snažíte dosáhnout?",
"What are you working on?": "Na čem pracujete?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/da-DK/translation.json b/src/lib/i18n/locales/da-DK/translation.json
index 58c6d6adb1..5e0e4af930 100644
--- a/src/lib/i18n/locales/da-DK/translation.json
+++ b/src/lib/i18n/locales/da-DK/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Opret ny note",
"Create Account": "Opret profil",
"Create Admin Account": "Opret administrator profil",
+ "Create and manage scheduled automations": "",
"Create Channel": "Opret kanal",
"Create Folder": "Opret mappe",
"Create Image": "Opret billede",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Brugerdefineret parameternavn",
"Custom Parameter Value": "Brugerdefineret parameterværdi",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Danger Zone",
"Dark": "Mørk",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Videresender system bruger OAuth access token til autentificering",
"Forwards system user session credentials to authenticate": "Videresender system bruger session credentials til autentificering",
+ "Fr_day_of_week": "",
"Full Context Mode": "Fuld kontekst tilstand",
"Function": "Funktion",
"Function Calling": "Funktionskald",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Hjem",
"Host": "Vært",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Hvordan kan jeg hjælpe dig i dag?",
"How would you rate this response?": "Hvordan vurderer du dette svar?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API nøgle påkrævet.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' er blevet downloadet.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' er allerede i kø til download.",
- "Model {{modelId}} not found": "Model {{modelId}} ikke fundet",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} understøtter ikke billeder",
"Model {{name}} is now {{status}}": "Model {{name}} er nu {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Modeldeling",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API nøgle",
+ "Monthly": "",
"More": "Mere",
"More Concise": "Mere kortfattet",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama Cloud API nøgle",
"Ollama Version": "Ollama-version",
"On": "Til",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Kun aktiv når \"Indsæt store tekster som fil\" indstillingen er slået til.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Kun aktiv når chat-input er fokuseret og en LLM er ved at generere et svar.",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Personalisering",
"Pin": "Fastgør",
+ "Pin to Sidebar": "",
"Pinned": "Fastgjort",
"Pinned Messages": "Fastgjorte beskeder",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Kører",
"Running...": "Kører...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Kører embedding-opgaver sideløbende for at fremskynde behandlingen. Slå fra hvis hastighedsbegrænsninger bliver et problem.",
+ "Sa_day_of_week": "",
"Save": "Gem",
"Save & Create": "Gem og opret",
"Save & Update": "Gem og opdater",
@@ -1889,6 +1897,7 @@
"STT Model": "STT-model",
"STT Settings": "STT-indstillinger",
"Stylized PDF Export": "Stiliseret PDF eksport",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "Undertekst",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Tekstopdeler",
"Text-to-Speech": "Tekst-til-tale",
"Text-to-Speech Engine": "Tekst-til-tale-engine",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Tak for din feedback!",
"The Application Account DN you bind with for search": "Application Account DN du binder med til søgning",
"The base to search for users": "Basen til at søge efter brugere",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS-model",
"TTS Settings": "TTS-indstillinger",
"TTS Voice": "TTS-stemme",
+ "Tu_day_of_week": "",
"Type": "Type",
"Type here...": "Skriv her...",
"Type Hugging Face Resolve (Download) URL": "Indtast Hugging Face Resolve (Download) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Advarsel: Hvis du aktiverer dette, vil brugerne kunne uploade vilkårlig kode på serveren.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Advarsel: Jupyter-udførelse gør det muligt at udføre vilkårlig kode, hvilket udfordrer alvorlige sikkerhedsrisici - fortsæt med ekstremt omhyggelighed.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "Web indlæser motor",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI vil lave forespørgsler til \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI vil lave forespørgsler til \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI vil lave forespørgsler til \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Hvad prøver du at opnå?",
"What are you working on?": "Hvad arbejder du på?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/de-DE/translation.json b/src/lib/i18n/locales/de-DE/translation.json
index 4fe9f1a2cb..58171851e9 100644
--- a/src/lib/i18n/locales/de-DE/translation.json
+++ b/src/lib/i18n/locales/de-DE/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Neue Notiz erstellen",
"Create Account": "Konto erstellen",
"Create Admin Account": "Admin-Konto erstellen",
+ "Create and manage scheduled automations": "",
"Create Channel": "Kanal erstellen",
"Create Folder": "Ordner erstellen",
"Create Image": "Bild erstellen",
@@ -479,6 +480,7 @@
"Custom Gender": "Benutzerdefiniertes Geschlecht",
"Custom Parameter Name": "Name des benutzerdef. Parameters",
"Custom Parameter Value": "Wert des benutzerdef. Parameters",
+ "Daily": "",
"Daily Messages": "Tägliche Nachrichten",
"Danger Zone": "Gefahrenzone",
"Dark": "Dunkel",
@@ -969,6 +971,7 @@
"Forward": "Weiterleiten",
"Forwards system user OAuth access token to authenticate": "Leitet OAuth-Zugriffstoken des Systembenutzers zur Authentifizierung weiter",
"Forwards system user session credentials to authenticate": "Leitet Sitzungsdaten des Systembenutzers zur Authentifizierung weiter",
+ "Fr_day_of_week": "",
"Full Context Mode": "Vollkontext-Modus",
"Function": "Funktion",
"Function Calling": "Funktionsaufruf",
@@ -1043,6 +1046,7 @@
"History": "History",
"Home": "Startseite",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "Stündliche Nachrichten",
"How can I help you today?": "Wie kann ich Ihnen heute helfen?",
"How would you rate this response?": "Wie bewerten Sie diese Antwort?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral-OCR-API-Schlüssel erforderlich.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Modell",
"Model '{{modelName}}' has been successfully downloaded.": "Modell '{{modelName}}' wurde erfolgreich heruntergeladen.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modell '{{modelTag}}' befindet sich bereits in der Warteschlange.",
- "Model {{modelId}} not found": "Modell {{modelId}} nicht gefunden",
"Model {{modelName}} deleted successfully": "Modell {{modelName}} erfolgreich gelöscht",
"Model {{modelName}} is not vision capable": "Das Modell {{modelName}} unterstützt keine Bilderkennung",
"Model {{name}} is now {{status}}": "Modell {{name}} ist jetzt {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Modelle teilen",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Mojeek Search API-Schlüssel",
+ "Monthly": "",
"More": "Mehr",
"More Concise": "Kürzer",
"More options": "Mehr Optionen",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama Cloud API-Schlüssel",
"Ollama Version": "Ollama-Version",
"On": "Ein",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Nur aktiv, wenn die Einstellung „Großen Text als Datei einfügen“ aktiviert ist.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Nur aktiv, wenn das Chat-Eingabefeld fokussiert ist und ein LLM eine Antwort generiert.",
@@ -1513,6 +1519,7 @@
"Persistent": "Persistent",
"Personalization": "Personalisierung",
"Pin": "Anheften",
+ "Pin to Sidebar": "",
"Pinned": "Angeheftet",
"Pinned Messages": "Angeheftete Nachrichten",
"Pinned Models": "Angepinnte Modelle",
@@ -1673,6 +1680,7 @@
"Running": "Läuft",
"Running...": "Läuft...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Führt Embedding-Aufgaben parallel aus, um die Verarbeitung zu beschleunigen. Deaktivieren Sie dies, falls Rate-Limits oder Ressourcenprobleme auftreten.",
+ "Sa_day_of_week": "",
"Save": "Speichern",
"Save & Create": "Speichern & Erstellen",
"Save & Update": "Speichern & Aktualisieren",
@@ -1889,6 +1897,7 @@
"STT Model": "STT-Modell",
"STT Settings": "STT-Einstellungen",
"Stylized PDF Export": "Stilisierter PDF-Export",
+ "Su_day_of_week": "",
"Submit question": "Frage absenden",
"Submit suggestion": "Vorschlag absenden",
"Subtitle": "Untertitel",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Text-Splitter",
"Text-to-Speech": "Text-zu-Sprache",
"Text-to-Speech Engine": "Text-zu-Sprache-Engine",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Danke für Ihr Feedback!",
"The Application Account DN you bind with for search": "Der Anwendungs-Konto-DN für die Suche",
"The base to search for users": "Die Basis, in der nach Benutzern gesucht wird",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS-Modell",
"TTS Settings": "TTS-Einstellungen",
"TTS Voice": "TTS-Stimme",
+ "Tu_day_of_week": "",
"Type": "Typ",
"Type here...": "Hier eingeben...",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (Download) URL eingeben",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "Warnung: Wenn Sie dies aktivieren, können Nutzer geplante Prompts automatisch ausführen.",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Warnung: Wenn Sie dies aktivieren, können Benutzer beliebigen Code auf den Server hochladen.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Warnung: Die Jupyter-Ausführung ermöglicht beliebige Codeausführung und birgt erhebliche Sicherheitsrisiken – gehen Sie mit äußerster Vorsicht vor.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web-API",
"Web Loader Engine": "Web-Loader-Engine",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI wird Anfragen an \"{{url}}\" senden",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI wird Anfragen an \"{{url}}/api/chat\" senden",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI wird Anfragen an \"{{url}}/chat/completions\" senden",
+ "Weekly": "",
"What are you trying to achieve?": "Was möchten Sie erreichen?",
"What are you working on?": "Woran arbeiten Sie?",
"What is NOT shared:": "Was NICHT geteilt wird:",
diff --git a/src/lib/i18n/locales/dg-DG/translation.json b/src/lib/i18n/locales/dg-DG/translation.json
index c4a87b0687..9f2ad4240f 100644
--- a/src/lib/i18n/locales/dg-DG/translation.json
+++ b/src/lib/i18n/locales/dg-DG/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Create Account",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Dark",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "How can I halp u today?",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' has been successfully downloaded.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' is already in queue for downloading.",
- "Model {{modelId}} not found": "Model {{modelId}} not found",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "",
"Model {{name}} is now {{status}}": "",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama Version",
"On": "On",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Personalization",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "Running... wow",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Save much wow",
"Save & Create": "Save & Create much create",
"Save & Update": "Save & Update much update",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "STT Settings very settings",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Text-to-Speech Engine much speak",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "TTS Settings much settings",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Type Hugging Face Resolve (Download) URL much download",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web very web",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/el-GR/translation.json b/src/lib/i18n/locales/el-GR/translation.json
index fe4da3a8d4..6e347a9e94 100644
--- a/src/lib/i18n/locales/el-GR/translation.json
+++ b/src/lib/i18n/locales/el-GR/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Δημιουργία Λογαριασμού",
"Create Admin Account": "Δημιουργία Λογαριασμού Διαχειριστή",
+ "Create and manage scheduled automations": "",
"Create Channel": "Δημιουργία Καναλιού",
"Create Folder": "Δημιουργία Φακέλου",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Περιοχή Κινδύνου",
"Dark": "Σκούρο",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "Λειτουργία χρήσης όλων των συμφραζομένων",
"Function": "Λειτουργία",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Αρχική",
"Host": "Διακομιστής",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Πώς μπορώ να σας βοηθήσω σήμερα;",
"How would you rate this response?": "Πώς θα βαθμολογούσατε αυτή την απάντηση;",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "Απαιτείται το API κλειδί του Mistral OCR.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Μοντέλο",
"Model '{{modelName}}' has been successfully downloaded.": "Το μοντέλο '{{modelName}}' κατεβάστηκε με επιτυχία.",
"Model '{{modelTag}}' is already in queue for downloading.": "Το μοντέλο '{{modelTag}}' βρίσκεται ήδη στην ουρά για λήψη.",
- "Model {{modelId}} not found": "Το μοντέλο {{modelId}} δεν βρέθηκε",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Το μοντέλο {{modelName}} δεν έχει δυνατότητα όρασης",
"Model {{name}} is now {{status}}": "Το μοντέλο {{name}} είναι τώρα {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Κλειδί API Mojeek Search",
+ "Monthly": "",
"More": "Περισσότερα",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Έκδοση Ollama",
"On": "Ενεργό",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Προσωποποίηση",
"Pin": "Καρφίτσωμα",
+ "Pin to Sidebar": "",
"Pinned": "Καρφιτσωμένο",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Εκτέλεση",
"Running...": "Εκτέλεση...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Αποθήκευση",
"Save & Create": "Αποθήκευση & Δημιουργία",
"Save & Update": "Αποθήκευση & Ενημέρωση",
@@ -1889,6 +1897,7 @@
"STT Model": "Μοντέλο Μετατροπής Ομιλίας σε Κείμενο",
"STT Settings": "Ρυθμίσεις Μετατροπής Ομιλίας σε Κείμενο",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Διαχωριστής Κειμένου",
"Text-to-Speech": "Κείμενο σε Ομιλία",
"Text-to-Speech Engine": "Μηχανή Ομιλίας σε Κείμενο",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Ευχαριστούμε για τα σχόλιά σας!",
"The Application Account DN you bind with for search": "Το DN του Λογαριασμού Εφαρμογής που συνδέετε για αναζήτηση",
"The base to search for users": "Η βάση για αναζήτηση χρηστών",
@@ -2047,6 +2057,7 @@
"TTS Model": "Μοντέλο μετατροπής Κειμένου σε Ομιλία",
"TTS Settings": "Ρυθμίσεις μετατροπής Κειμένου σε Ομιλία",
"TTS Voice": "Φωνή Κειμένου σε Ομιλία",
+ "Tu_day_of_week": "",
"Type": "Τύπος",
"Type here...": "Πληκτρολογήστε εδώ...",
"Type Hugging Face Resolve (Download) URL": "Τύπος URL Ανάλυσης Hugging Face Resolve (Λήψη)",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Προειδοποίηση: Η ενεργοποίηση αυτού θα επιτρέψει στους χρήστες να ανεβάσουν αυθαίρετο κώδικα στον διακομιστή.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Προειδοποίηση: Η εκτέλεση του Jupyter επιτρέπει την εκτέλεση αυθαίρετου κώδικα, γεγονός που θέτει σοβαρούς κινδύνους ασφαλεία - προχωρήστε με εξαιρετική προσοχή.",
+ "We_day_of_week": "",
"Web": "Διαδίκτυο",
"Web API": "",
"Web Loader Engine": "Μηχανή Φόρτωσης Διαδικτύου",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "Το WebUI θα κάνει αιτήματα στο \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "Το WebUI θα κάνει αιτήματα στο \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "Το WebUI θα κάνει αιτήματα στο \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Τι προσπαθείτε να πετύχετε?",
"What are you working on?": "Τι εργάζεστε;",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/en-GB/translation.json b/src/lib/i18n/locales/en-GB/translation.json
index e688ca17b1..28a735ae18 100644
--- a/src/lib/i18n/locales/en-GB/translation.json
+++ b/src/lib/i18n/locales/en-GB/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "",
"Model '{{modelTag}}' is already in queue for downloading.": "",
- "Model {{modelId}} not found": "",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "",
"Model {{name}} is now {{status}}": "",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "",
"On": "",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Personalisation",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "",
"Save & Create": "",
"Save & Update": "",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "",
"Stylized PDF Export": "Stylised PDF Export",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/en-US/translation.json b/src/lib/i18n/locales/en-US/translation.json
index 7ac7cb3e5b..b9f462c6a9 100644
--- a/src/lib/i18n/locales/en-US/translation.json
+++ b/src/lib/i18n/locales/en-US/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "",
"Model '{{modelTag}}' is already in queue for downloading.": "",
- "Model {{modelId}} not found": "",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "",
"Model {{name}} is now {{status}}": "",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "",
"On": "",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "",
"Save & Create": "",
"Save & Update": "",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/es-ES/translation.json b/src/lib/i18n/locales/es-ES/translation.json
index b1339c0456..63e98ee49d 100644
--- a/src/lib/i18n/locales/es-ES/translation.json
+++ b/src/lib/i18n/locales/es-ES/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "Crea una nueva nota",
"Create Account": "Crear Cuenta",
"Create Admin Account": "Crear Cuenta Administrativa",
+ "Create and manage scheduled automations": "",
"Create Channel": "Crear Canal",
"Create Folder": "Crear Carpeta",
"Create Image": "Crear Imagen",
@@ -480,6 +481,7 @@
"Custom Gender": "Género Personalizado",
"Custom Parameter Name": "Nombre del Parámetro Personalizado",
"Custom Parameter Value": "Valor del Parámetro Personalizado",
+ "Daily": "",
"Daily Messages": "Mensajes Diarios",
"Danger Zone": "Zona Peligrosa",
"Dark": "Oscuro",
@@ -970,6 +972,7 @@
"Forward": "Reenviar",
"Forwards system user OAuth access token to authenticate": "Reenvía el token de acceso OAuth del usuario del sistema para autenticarse",
"Forwards system user session credentials to authenticate": "Reenvío de las credenciales de la sesión del usuario del sistema para autenticación",
+ "Fr_day_of_week": "",
"Full Context Mode": "Modo Contexto Completo",
"Function": "Función",
"Function Calling": "Modo de Llamada a Funciones (Herramientas)",
@@ -1044,6 +1047,7 @@
"History": "Historial",
"Home": "Inicio",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "Mensajes por Hora",
"How can I help you today?": "¿Cómo puedo ayudarte hoy?",
"How would you rate this response?": "¿Cómo calificarías esta respuesta?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "OCR Mistral",
"Mistral OCR API Key required.": "Clave API de Mistral OCR requerida",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Modelo",
"Model '{{modelName}}' has been successfully downloaded.": "Modelo '{{modelName}}' se ha descargado correctamente.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modelo '{{modelTag}}' ya está en cola para descargar.",
- "Model {{modelId}} not found": "Modelo {{modelId}} no encontrado",
"Model {{modelName}} deleted successfully": "Modelo {{modelName}} borrado correctamente",
"Model {{modelName}} is not vision capable": "Modelo {{modelName}} no esta capacitado para visión",
"Model {{name}} is now {{status}}": "Modelo {{name}} está ahora {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "Compartir Modelos",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Clave API de Mojeek Search",
+ "Monthly": "",
"More": "Más",
"More Concise": "Más Conciso",
"More options": "Más Opciones",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "Clave API de Ollama Cloud",
"Ollama Version": "Versión de Ollama",
"On": "Activado",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Solo activo cuando \"Pegar el Texto Largo como Archivo\" está activado",
"Only active when the chat input is in focus and an LLM is generating a response.": "Solo activo con el foco en la entrada del chat y se está generando una respuesta",
@@ -1514,6 +1520,7 @@
"Persistent": "Persistente",
"Personalization": "Personalización",
"Pin": "Fijar",
+ "Pin to Sidebar": "",
"Pinned": "Fijado",
"Pinned Messages": "Mensajes Fijados",
"Pinned Models": "Modelos Fijados",
@@ -1675,6 +1682,7 @@
"Running": "Ejecutando",
"Running...": "Ejecutando...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Ejecuta tareas de incrustración concurrentes para acelerar el procesado. Desactivar si se generan problemas (por limitaciones de los motores de incrustracción en uso)",
+ "Sa_day_of_week": "",
"Save": "Guardar",
"Save & Create": "Guardar y Crear",
"Save & Update": "Guardar y Actualizar",
@@ -1891,6 +1899,7 @@
"STT Model": "Modelo STT",
"STT Settings": "Ajustes Voz a Texto (STT)",
"Stylized PDF Export": "Exportar PDF Estilizado",
+ "Su_day_of_week": "",
"Submit question": "Enviar pregunta",
"Submit suggestion": "Enviar sugerencia",
"Subtitle": "Subtítulo",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Divisor de Texto",
"Text-to-Speech": "Texto a Voz",
"Text-to-Speech Engine": "Motor Texto a Voz(TTS)",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "¡Gracias por tu comentario!",
"The Application Account DN you bind with for search": "Cuenta DN de la aplicación vinculada para búsqueda",
"The base to search for users": "Base para buscar usuarios",
@@ -2049,6 +2059,7 @@
"TTS Model": "Modelo TTS",
"TTS Settings": "Ajustes Texto a Voz (TTS)",
"TTS Voice": "Voz TTS",
+ "Tu_day_of_week": "",
"Type": "Tipo",
"Type here...": "Teclea aquí...",
"Type Hugging Face Resolve (Download) URL": "Escribir la URL de Hugging Face Resolve (Descarga)",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "Aviso: Habilitar esto permitirá a los usuarios ejecutar automáticamente indicadores programados.",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Aviso: Habilitar esto permitirá a los usuarios subir código arbitrario al servidor.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Aviso: La ejecución Jupyter habilita la ejecución de código arbitrario, planteando graves riesgos de seguridad; Proceder con extrema precaución.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Motor Cargador Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI hará solicitudes a \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI hará solicitudes a \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI hará solicitudes a \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "¿Qué estás tratando de conseguir?",
"What are you working on?": "¿En qué estás trabajando?",
"What is NOT shared:": "Que NO es compartido:",
diff --git a/src/lib/i18n/locales/et-EE/translation.json b/src/lib/i18n/locales/et-EE/translation.json
index 4f7bc75de7..1d061c333b 100644
--- a/src/lib/i18n/locales/et-EE/translation.json
+++ b/src/lib/i18n/locales/et-EE/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Loo uus märge",
"Create Account": "Loo konto",
"Create Admin Account": "Loo administraatori konto",
+ "Create and manage scheduled automations": "",
"Create Channel": "Loo kanal",
"Create Folder": "Loo kaust",
"Create Image": "Loo pilt",
@@ -479,6 +480,7 @@
"Custom Gender": "Kohandatud sugu",
"Custom Parameter Name": "Kohandatud parameetri nimi",
"Custom Parameter Value": "Kohandatud parameetri väärtus",
+ "Daily": "",
"Daily Messages": "Päevased sõnumid",
"Danger Zone": "Ohutsoon",
"Dark": "Tume",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Edastab süsteemikasutaja OAuthi juurdepääsumärgi autentimiseks",
"Forwards system user session credentials to authenticate": "Edastab süsteemikasutaja seansi mandaadid autentimiseks",
+ "Fr_day_of_week": "",
"Full Context Mode": "Täiskonteksti režiim",
"Function": "Funktsioon",
"Function Calling": "Funktsiooni kutsumine",
@@ -1043,6 +1046,7 @@
"History": "Ajalugu",
"Home": "Avaleht",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "Tunni sõnumid",
"How can I help you today?": "Kuidas saan teid täna aidata?",
"How would you rate this response?": "Kuidas hindaksite seda vastust?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API võti on nõutav.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Mudel",
"Model '{{modelName}}' has been successfully downloaded.": "Mudel '{{modelName}}' on edukalt alla laaditud.",
"Model '{{modelTag}}' is already in queue for downloading.": "Mudel '{{modelTag}}' on juba allalaadimise järjekorras.",
- "Model {{modelId}} not found": "Mudelit {{modelId}} ei leitud",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Mudel {{modelName}} ei ole võimeline visuaalseid sisendeid töötlema",
"Model {{name}} is now {{status}}": "Mudel {{name}} on nüüd {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Mudelite jagamine",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Mojeek Search API võti",
+ "Monthly": "",
"More": "Rohkem",
"More Concise": "Kokkuvõtlikum",
"More options": "Rohkem valikuid",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama Cloud API Võti",
"Ollama Version": "Ollama versioon",
"On": "Sees",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Aktiivne ainult siis, kui seade \"Kleebi suur tekst failina\" on sisse lülitatud.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Aktiivne ainult siis, kui vestluse sisend on fookuses ja LLM genereerib vastust.",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Isikupärastamine",
"Pin": "Kinnita",
+ "Pin to Sidebar": "",
"Pinned": "Kinnitatud",
"Pinned Messages": "Kinnitatud sõnumid",
"Pinned Models": "Kinnitatud mudelid",
@@ -1673,6 +1680,7 @@
"Running": "Töötab",
"Running...": "Töötab...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Käitab manustamisülesandeid samaaegselt töötlemise kiirendamiseks. Lülitage välja, kui piirangud muutuvad probleemiks.",
+ "Sa_day_of_week": "",
"Save": "Salvesta",
"Save & Create": "Salvesta ja loo",
"Save & Update": "Salvesta ja uuenda",
@@ -1889,6 +1897,7 @@
"STT Model": "STT mudel",
"STT Settings": "STT seaded",
"Stylized PDF Export": "Stiliseeritud PDF eksport",
+ "Su_day_of_week": "",
"Submit question": "Esita küsimus",
"Submit suggestion": "Esita soovitus",
"Subtitle": "Alampealkiri",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Teksti tükeldaja",
"Text-to-Speech": "Text-to-Speech",
"Text-to-Speech Engine": "Tekst-kõneks mootor",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Täname tagasiside eest!",
"The Application Account DN you bind with for search": "Rakenduse konto DN, millega seote otsingu jaoks",
"The base to search for users": "Baas kasutajate otsimiseks",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS mudel",
"TTS Settings": "TTS seaded",
"TTS Voice": "TTS hääl",
+ "Tu_day_of_week": "",
"Type": "Tüüp",
"Type here...": "Sisestage siia...",
"Type Hugging Face Resolve (Download) URL": "Sisestage Hugging Face Resolve (Allalaadimise) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Hoiatus: Selle lubamine võimaldab kasutajatel üles laadida suvalist koodi serverisse.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Hoiatus: Jupyter täitmine võimaldab suvalise koodi käivitamist, mis kujutab endast tõsist turvariski - jätkake äärmise ettevaatusega.",
+ "We_day_of_week": "",
"Web": "Veeb",
"Web API": "Veebi API",
"Web Loader Engine": "Veebilaadija mootor",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI teeb päringuid aadressile \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI teeb päringuid aadressile \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI teeb päringuid aadressile \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Mida te püüate saavutada?",
"What are you working on?": "Millega te tegelete?",
"What is NOT shared:": "Mida EI jagata:",
diff --git a/src/lib/i18n/locales/eu-ES/translation.json b/src/lib/i18n/locales/eu-ES/translation.json
index ab7ee3117b..af50fb0122 100644
--- a/src/lib/i18n/locales/eu-ES/translation.json
+++ b/src/lib/i18n/locales/eu-ES/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Sortu Kontua",
"Create Admin Account": "Sortu Administratzaile Kontua",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Iluna",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "Funtzioa",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "Ostalaria",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Zertan lagun zaitzaket gaur?",
"How would you rate this response?": "Nola baloratuko zenuke erantzun hau?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modeloa",
"Model '{{modelName}}' has been successfully downloaded.": "'{{modelName}}' modeloa ongi deskargatu da.",
"Model '{{modelTag}}' is already in queue for downloading.": "'{{modelTag}}' modeloa dagoeneko deskarga ilaran dago.",
- "Model {{modelId}} not found": "{{modelId}} modeloa ez da aurkitu",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "{{modelName}} modeloak ez du ikusmen gaitasunik",
"Model {{name}} is now {{status}}": "{{name}} modeloa orain {{status}} dago",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek bilaketa API gakoa",
+ "Monthly": "",
"More": "Gehiago",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama bertsioa",
"On": "Piztuta",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Pertsonalizazioa",
"Pin": "Ainguratu",
+ "Pin to Sidebar": "",
"Pinned": "Ainguratuta",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Exekutatzen",
"Running...": "Exekutatzen...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Gorde",
"Save & Create": "Gorde eta sortu",
"Save & Update": "Gorde eta eguneratu",
@@ -1889,6 +1897,7 @@
"STT Model": "STT modeloa",
"STT Settings": "STT ezarpenak",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Testu banatzailea",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Testutik-ahotsera motorra",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Eskerrik asko zure iritzia emateagatik!",
"The Application Account DN you bind with for search": "Bilaketarako lotzen duzun aplikazio kontuaren DN-a",
"The base to search for users": "Erabiltzaileak bilatzeko oinarria",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS modeloa",
"TTS Settings": "TTS ezarpenak",
"TTS Voice": "TTS ahotsa",
+ "Tu_day_of_week": "",
"Type": "Mota",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Idatzi Hugging Face Resolve (Deskarga) URLa",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Abisua: Hau gaitzeak erabiltzaileei zerbitzarian kode arbitrarioa kargatzea ahalbidetuko die.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Weba",
"Web API": "Web APIa",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI-k eskaerak egingo ditu \"{{url}}/api/chat\"-era",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI-k eskaerak egingo ditu \"{{url}}/chat/completions\"-era",
+ "Weekly": "",
"What are you trying to achieve?": "Zer lortu nahi duzu?",
"What are you working on?": "Zertan ari zara lanean?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/fa-IR/translation.json b/src/lib/i18n/locales/fa-IR/translation.json
index 725f340932..cd2f7839a9 100644
--- a/src/lib/i18n/locales/fa-IR/translation.json
+++ b/src/lib/i18n/locales/fa-IR/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "ایجاد یک یادداشت جدید",
"Create Account": "ساخت حساب کاربری",
"Create Admin Account": "ایجاد حساب مدیر",
+ "Create and manage scheduled automations": "",
"Create Channel": "ایجاد کانال",
"Create Folder": "ایجاد پوشه",
"Create Image": "ایجاد تصویر",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "نام پارامتر سفارشی",
"Custom Parameter Value": "مقدار پارامتر سفارشی",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "منطقه خطر",
"Dark": "تیره",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "ارسال توکن دسترسی OAuth کاربر سیستم برای احراز هویت",
"Forwards system user session credentials to authenticate": "اعتبارنامه\u200cهای نشست کاربر سیستم را برای احراز هویت ارسال می\u200cکند",
+ "Fr_day_of_week": "",
"Full Context Mode": "حالت متن کامل",
"Function": "تابع",
"Function Calling": "فراخوانی تابع",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "خانه",
"Host": "میزبان",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "امروز چطور می توانم کمک تان کنم؟",
"How would you rate this response?": "این پاسخ را چگونه ارزیابی می\u200cکنید؟",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "تشخیص متن میسترال",
"Mistral OCR API Key required.": "کلید API تشخیص متن میسترال مورد نیاز است.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "مدل",
"Model '{{modelName}}' has been successfully downloaded.": "مدل '{{modelName}}' با موفقیت دانلود شد.",
"Model '{{modelTag}}' is already in queue for downloading.": "مدل '{{modelTag}}' در حال حاضر در صف برای دانلود است.",
- "Model {{modelId}} not found": "مدل {{modelId}} یافت نشد",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "مدل {{modelName}} قادر به بینایی نیست",
"Model {{name}} is now {{status}}": "مدل {{name}} در حال حاضر {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "کلید API جستجوی موجیک",
+ "Monthly": "",
"More": "بیشتر",
"More Concise": "خلاصه\u200cتر",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "کلید API ابری اُلاما",
"Ollama Version": "نسخه ollama",
"On": "روشن",
+ "Once": "",
"OneDrive": "وان\u200cدرایو",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "فقط زمانی فعال است که تنظیم «چسباندن متن بزرگ به عنوان فایل» روشن باشد.",
"Only active when the chat input is in focus and an LLM is generating a response.": "فقط زمانی فعال است که ورودی چت در فوکوس باشد و یک LLM در حال تولید پاسخ باشد.",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "شخصی سازی",
"Pin": "پین کردن",
+ "Pin to Sidebar": "",
"Pinned": "پین شده",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "در حال اجرا",
"Running...": "در حال اجرا...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "ذخیره",
"Save & Create": "ذخیره و ایجاد",
"Save & Update": "ذخیره و به\u200cروزرسانی",
@@ -1889,6 +1897,7 @@
"STT Model": "مدل تبدیل صدا به متن",
"STT Settings": "تنظیمات تبدیل صدا به متن",
"Stylized PDF Export": "خروجی گرفتن از PDF با استایل",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "تقسیم\u200cکننده متن",
"Text-to-Speech": "متن به گفتار",
"Text-to-Speech Engine": "موتور تبدیل متن به گفتار",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "با تشکر از بازخورد شما!",
"The Application Account DN you bind with for search": "DN حساب برنامه که برای جستجو به آن متصل می\u200cشوید",
"The base to search for users": "پایه برای جستجوی کاربران",
@@ -2047,6 +2057,7 @@
"TTS Model": "مدل TTS",
"TTS Settings": "تنظیمات TTS",
"TTS Voice": "صدای TTS",
+ "Tu_day_of_week": "",
"Type": "نوع",
"Type here...": "اینجا تایپ کنید...",
"Type Hugging Face Resolve (Download) URL": "مقدار URL دانلود (Resolve) Hugging Face را وارد کنید",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "هشدار: فعال کردن این گزینه به کاربران اجازه می\u200cدهد کد دلخواه را روی سرور آپلود کنند.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "هشدار: اجرای ژوپیتر امکان اجرای کد دلخواه را فراهم می\u200cکند که خطرات امنیتی جدی به همراه دارد - با احتیاط زیاد ادامه دهید.",
+ "We_day_of_week": "",
"Web": "وب",
"Web API": "API وب",
"Web Loader Engine": "موتور بارگذاری وب",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI به \"{{url}}\" درخواست خواهد داد",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI به \"{{url}}/api/chat\" درخواست خواهد داد",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI به \"{{url}}/chat/completions\" درخواست خواهد داد",
+ "Weekly": "",
"What are you trying to achieve?": "به دنبال دستیابی به چه هدفی هستید؟",
"What are you working on?": "روی چه چیزی کار می\u200cکنید؟",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/fi-FI/translation.json b/src/lib/i18n/locales/fi-FI/translation.json
index 94a728a77f..10d63179d3 100644
--- a/src/lib/i18n/locales/fi-FI/translation.json
+++ b/src/lib/i18n/locales/fi-FI/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Luo uusi muistiinpano",
"Create Account": "Luo tili",
"Create Admin Account": "Luo ylläpitäjätili",
+ "Create and manage scheduled automations": "",
"Create Channel": "Luo kanava",
"Create Folder": "Luo kansio",
"Create Image": "Luo kuva",
@@ -479,6 +480,7 @@
"Custom Gender": "Muu sukupuoli",
"Custom Parameter Name": "Mukautetun parametrin nimi",
"Custom Parameter Value": "Mukautetun parametrin arvo",
+ "Daily": "",
"Daily Messages": "Päivittäiset viestit",
"Danger Zone": "Vaara-alue",
"Dark": "Tumma",
@@ -969,6 +971,7 @@
"Forward": "Eteenpäin",
"Forwards system user OAuth access token to authenticate": "Välittää järjestelmä käyttäjän OAuth tunniste todennuksessa",
"Forwards system user session credentials to authenticate": "Välittää järjestelmän käyttäjän istunnon tunnistetiedot todennusta varten",
+ "Fr_day_of_week": "",
"Full Context Mode": "Koko kontekstitila",
"Function": "Toiminto",
"Function Calling": "Toiminto kutsu",
@@ -1043,6 +1046,7 @@
"History": "Historia",
"Home": "Koti",
"Host": "Palvelin",
+ "Hourly": "",
"Hourly Messages": "Tuntikohtaiset viestit",
"How can I help you today?": "Miten voin auttaa sinua tänään?",
"How would you rate this response?": "Kuinka arvioisit tätä vastausta?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR api-avain vaaditaan",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Malli",
"Model '{{modelName}}' has been successfully downloaded.": "Malli '{{modelName}}' ladattiin onnistuneesti.",
"Model '{{modelTag}}' is already in queue for downloading.": "Malli '{{modelTag}}' on jo jonossa ladattavaksi.",
- "Model {{modelId}} not found": "Mallia {{modelId}} ei löytynyt",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Malli {{modelName}} ei kykene näkökykyyn",
"Model {{name}} is now {{status}}": "Malli {{name}} on nyt {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Mallien jako",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API -avain",
+ "Monthly": "",
"More": "Lisää",
"More Concise": "Lyhyemmin",
"More options": "Lisää vaihtoehtoja",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama Cloud API avain",
"Ollama Version": "Ollama-versio",
"On": "Päällä",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Aktiivinen vain, kun \"Liitä suuri teksti tiedostona\" -asetus on käytössä.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Aktiivinen vain, kun tekstikenttä on kohdistettuna ja LLM luo vastausta.",
@@ -1513,6 +1519,7 @@
"Persistent": "Pysyvä",
"Personalization": "Personointi",
"Pin": "Kiinnitä",
+ "Pin to Sidebar": "",
"Pinned": "Kiinnitetty",
"Pinned Messages": "Kiinnitetyt viestit",
"Pinned Models": "Kiinnitetyt mallit",
@@ -1673,6 +1680,7 @@
"Running": "Käynnissä",
"Running...": "Käynnissä...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Suorittaa upotustehtäviä samanaikaisesti käsittelyn nopeuttamiseksi. Poista käytöstä, jos kutsurajoituksesta tulee ongelma.",
+ "Sa_day_of_week": "",
"Save": "Tallenna",
"Save & Create": "Tallenna ja luo",
"Save & Update": "Tallenna ja päivitä",
@@ -1889,6 +1897,7 @@
"STT Model": "Puheentunnistusmalli",
"STT Settings": "Puheentunnistuksen asetukset",
"Stylized PDF Export": "Muotoiltun PDF-vienti",
+ "Su_day_of_week": "",
"Submit question": "Lähetä kysymys",
"Submit suggestion": "Lähetä ehdotus",
"Subtitle": "Alaotsikko",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Tekstin jakaja",
"Text-to-Speech": "Puhesynteesi",
"Text-to-Speech Engine": "Puhesynteesimoottori",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Kiitos palautteestasi!",
"The Application Account DN you bind with for search": "Hakua varten sidottu sovelluksen käyttäjätilin DN",
"The base to search for users": "Käyttäjien haun perusta",
@@ -2047,6 +2057,7 @@
"TTS Model": "Puhesynteesimalli",
"TTS Settings": "Puhesynteesiasetukset",
"TTS Voice": "Puhesynteesiääni",
+ "Tu_day_of_week": "",
"Type": "Tyyppi",
"Type here...": "Kirjoita tähän...",
"Type Hugging Face Resolve (Download) URL": "Kirjoita Hugging Face -resolve-latausosoite",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Varoitus: Tämän käyttöönotto sallii käyttäjien ladata mielivaltaista koodia palvelimelle.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Varoitus: Jupyter käyttö voi mahdollistaa mielivaltaiseen koodin suorittamiseen, mikä voi aiheuttaa tietoturvariskejä - käytä äärimmäisen varoen.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web-API",
"Web Loader Engine": "Verkkolataaja moottori",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI lähettää pyyntöjä osoitteeseen \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI lähettää pyyntöjä osoitteeseen \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI lähettää pyyntöjä osoitteeseen \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Mitä yrität saavuttaa?",
"What are you working on?": "Mitä olet työskentelemässä?",
"What is NOT shared:": "Mitä EI jaeta:",
diff --git a/src/lib/i18n/locales/fr-CA/translation.json b/src/lib/i18n/locales/fr-CA/translation.json
index de0021689b..b0f58be103 100644
--- a/src/lib/i18n/locales/fr-CA/translation.json
+++ b/src/lib/i18n/locales/fr-CA/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "Créer un compte",
"Create Admin Account": "Créer un compte administrateur",
+ "Create and manage scheduled automations": "",
"Create Channel": "Créer un canal",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Nom du réglage personnalisé",
"Custom Parameter Value": "Valeur du réglage personnalisé",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Zone de danger",
"Dark": "Sombre",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Transmet les identifiants de session de l'utilisateur pour l'authentification",
+ "Fr_day_of_week": "",
"Full Context Mode": "Mode avec injection complète dans le Context",
"Function": "Fonction",
"Function Calling": "Appel de fonction",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "Home",
"Host": "Hôte",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Comment puis-je vous aider aujourd'hui ?",
"How would you rate this response?": "Comment évalueriez-vous cette réponse ?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Clé d'API pour Mistral OCR requise",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modèle",
"Model '{{modelName}}' has been successfully downloaded.": "Le modèle '{{modelName}}' a été téléchargé avec succès.",
"Model '{{modelTag}}' is already in queue for downloading.": "Le modèle '{{modelTag}}' est déjà dans la file d'attente pour le téléchargement.",
- "Model {{modelId}} not found": "Modèle {{modelId}} introuvable",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Le modèle {{modelName}} n'a pas de capacités visuelles",
"Model {{name}} is now {{status}}": "Le modèle {{name}} est désormais {{status}}.",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Clé API Mojeek",
+ "Monthly": "",
"More": "Plus",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Version d'Ollama",
"On": "Activé",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Personnalisation",
"Pin": "Épingler",
+ "Pin to Sidebar": "",
"Pinned": "Épinglé",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "Exécution",
"Running...": "Exécution...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Enregistrer",
"Save & Create": "Enregistrer & Créer",
"Save & Update": "Enregistrer & Mettre à jour",
@@ -1891,6 +1899,7 @@
"STT Model": "Modèle de Speech-to-Text",
"STT Settings": "Réglages de Speech-to-Text",
"Stylized PDF Export": "Export de PDF stylisés",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Text Splitter",
"Text-to-Speech": "Text-to-Speech",
"Text-to-Speech Engine": "Moteur de Text-to-Speech",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Merci pour vos commentaires !",
"The Application Account DN you bind with for search": "Le DN du compte de l'application avec lequel vous vous liez pour la recherche",
"The base to search for users": "La base pour rechercher des utilisateurs",
@@ -2049,6 +2059,7 @@
"TTS Model": "Modèle de Text-to-Speech",
"TTS Settings": "Réglages de Text-to-Speech",
"TTS Voice": "Voix de Text-to-Speech",
+ "Tu_day_of_week": "",
"Type": "Type",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Entrez l'URL de Téléchargement Hugging Face Resolve",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Avertissement : Activer cette option permettra aux utilisateurs de télécharger du code arbitraire sur le serveur.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Avertissement : L'exécution Jupyter permet l'exécution de code arbitraire, ce qui présente des risques de sécurité importants. Procédez avec une extrême prudence.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Moteur de chargement Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI effectuera des requêtes vers \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI fera des requêtes à \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI fera des requêtes à \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Que cherchez-vous à accomplir ?",
"What are you working on?": "Sur quoi travaillez-vous ?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/fr-FR/translation.json b/src/lib/i18n/locales/fr-FR/translation.json
index a8a8a8dd52..60c85440d3 100644
--- a/src/lib/i18n/locales/fr-FR/translation.json
+++ b/src/lib/i18n/locales/fr-FR/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "Créer une nouvelle note",
"Create Account": "Créer un compte",
"Create Admin Account": "Créer un compte administrateur",
+ "Create and manage scheduled automations": "",
"Create Channel": "Créer un canal",
"Create Folder": "Créer un dossier",
"Create Image": "Création d'image",
@@ -480,6 +481,7 @@
"Custom Gender": "Genre personnalisé",
"Custom Parameter Name": "Nom du réglage personnalisé",
"Custom Parameter Value": "Valeur du réglage personnalisé",
+ "Daily": "",
"Daily Messages": "Messages par jour",
"Danger Zone": "Zone de danger",
"Dark": "Sombre",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Transfère le jeton d'accès OAuth de l'utilisateur système pour l'authentification",
"Forwards system user session credentials to authenticate": "Transmet les identifiants de session de l'utilisateur pour l'authentification",
+ "Fr_day_of_week": "",
"Full Context Mode": "Mode avec injection complète dans le contexte",
"Function": "Fonction",
"Function Calling": "Appel de fonction",
@@ -1044,6 +1047,7 @@
"History": "Historique",
"Home": "Home",
"Host": "Hôte",
+ "Hourly": "",
"Hourly Messages": "Messages par heure",
"How can I help you today?": "Comment puis-je vous aider aujourd'hui ?",
"How would you rate this response?": "Comment évalueriez-vous cette réponse ?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Clé d'API pour Mistral OCR requise",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Modèle",
"Model '{{modelName}}' has been successfully downloaded.": "Le modèle '{{modelName}}' a été téléchargé avec succès.",
"Model '{{modelTag}}' is already in queue for downloading.": "Le modèle '{{modelTag}}' est déjà dans la file d'attente pour le téléchargement.",
- "Model {{modelId}} not found": "Modèle {{modelId}} introuvable",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Le modèle {{modelName}} n'a pas de fonctionnalité de vision",
"Model {{name}} is now {{status}}": "Le modèle {{name}} est désormais {{status}}.",
@@ -1310,6 +1314,7 @@
"Models Sharing": "Partage des modèles",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Clé API Mojeek",
+ "Monthly": "",
"More": "Plus",
"More Concise": "Plus concis",
"More options": "Plus d'options",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "Clé API Ollama Cloud",
"Ollama Version": "Version d'Ollama",
"On": "Activé",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Uniquement lorsque le paramètre \"Coller un texte volumineux comme fichier\" est activé.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Uniquement lorsque la zone de saisie de la conversation est active et qu'un LLM génère une réponse.",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Personnalisation",
"Pin": "Épingler",
+ "Pin to Sidebar": "",
"Pinned": "Épinglé",
"Pinned Messages": "Messages épinglés",
"Pinned Models": "Modèles épinglés",
@@ -1675,6 +1682,7 @@
"Running": "Exécution",
"Running...": "Exécution...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Exécute les tâches d'embedding en parallèle pour accélérer le traitement. Désactivez si les limites de débit posent problème.",
+ "Sa_day_of_week": "",
"Save": "Enregistrer",
"Save & Create": "Enregistrer & Créer",
"Save & Update": "Enregistrer & Mettre à jour",
@@ -1891,6 +1899,7 @@
"STT Model": "Modèle de Speech-to-Text",
"STT Settings": "Réglages de Speech-to-Text",
"Stylized PDF Export": "Export de PDF stylisés",
+ "Su_day_of_week": "",
"Submit question": "Envoyer la question",
"Submit suggestion": "Soumettre la suggestion",
"Subtitle": "Sous-titre",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Découpage du texte",
"Text-to-Speech": "Text-to-Speech",
"Text-to-Speech Engine": "Moteur de Text-to-Speech",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Merci pour vos commentaires !",
"The Application Account DN you bind with for search": "Le DN du compte de l'application avec lequel vous vous liez pour la recherche",
"The base to search for users": "La base pour rechercher des utilisateurs",
@@ -2049,6 +2059,7 @@
"TTS Model": "Modèle de Text-to-Speech",
"TTS Settings": "Réglages de Text-to-Speech",
"TTS Voice": "Voix de Text-to-Speech",
+ "Tu_day_of_week": "",
"Type": "Type",
"Type here...": "Entrez votre message ici...",
"Type Hugging Face Resolve (Download) URL": "Entrez l'URL de Téléchargement Hugging Face Resolve",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Avertissement : Activer cette option permettra aux utilisateurs de télécharger du code arbitraire sur le serveur.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Avertissement : L'exécution Jupyter permet l'exécution de code arbitraire, ce qui présente des risques de sécurité importants. Procédez avec une extrême prudence.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Moteur de chargement Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI effectuera des requêtes vers \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI fera des requêtes à \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI fera des requêtes à \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Que cherchez-vous à accomplir ?",
"What are you working on?": "Sur quoi travaillez-vous ?",
"What is NOT shared:": "Ce qui n'est PAS partagé :",
diff --git a/src/lib/i18n/locales/gl-ES/translation.json b/src/lib/i18n/locales/gl-ES/translation.json
index d831953ddf..19fad395d1 100644
--- a/src/lib/i18n/locales/gl-ES/translation.json
+++ b/src/lib/i18n/locales/gl-ES/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Xerar unha conta",
"Create Admin Account": "Xerar conta administrativa",
+ "Create and manage scheduled automations": "",
"Create Channel": "Xerar Canal",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Oscuro",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "Función",
"Function Calling": "chamada de función",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "¿Cómo podo axudarche hoxe?",
"How would you rate this response?": "¿Cómo calificarías esta resposta?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modelo",
"Model '{{modelName}}' has been successfully downloaded.": "0 modelo '{{modelName}}' se ha descargado correctamente.",
"Model '{{modelTag}}' is already in queue for downloading.": "0 modelo '{{modelTag}}' ya está en cola para descargar.",
- "Model {{modelId}} not found": "0 modelo {{modelId}} no fue encontrado",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "O modelo {{modelName}} no es capaz de ver",
"Model {{name}} is now {{status}}": "O modelo {{name}} ahora es {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "chave API de Mojeek Search",
+ "Monthly": "",
"More": "mais",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Versión de Ollama",
"On": "Activado",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Personalización",
"Pin": "Fijar",
+ "Pin to Sidebar": "",
"Pinned": "Fijado",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Executando",
"Running...": "Executando...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Gardar",
"Save & Create": "Gardar y xerar",
"Save & Update": "Gardar y Actualizar",
@@ -1889,6 +1897,7 @@
"STT Model": "Modelo STT",
"STT Settings": "Configuracions de STT",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Divisor de texto",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Motor de texto a voz",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "¡Gracias pola tua retroalimentación!",
"The Application Account DN you bind with for search": "A conta de aplicación DN que vincula para a búsqueda",
"The base to search for users": "A base para buscar usuarios",
@@ -2047,6 +2057,7 @@
"TTS Model": "Modelo TTS",
"TTS Settings": "Configuración de TTS",
"TTS Voice": "Voz do TTS",
+ "Tu_day_of_week": "",
"Type": "Tipo",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Escriba la URL (Descarga) de Hugging Face Resolve",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Advertencia: Habilitar esto permitirá a os usuarios subir código arbitrario no servidor.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Aviso: A execución de Jupyter permite a execución de código arbitrario, o que supón riscos de seguridade graves - procede con extrema precaución.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI hará solicitudes a \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI hará solicitudes a \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "¿Qué estás tratando de lograr?",
"What are you working on?": "¿En qué estás trabajando?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/he-IL/translation.json b/src/lib/i18n/locales/he-IL/translation.json
index 99feaf7600..a273c87c71 100644
--- a/src/lib/i18n/locales/he-IL/translation.json
+++ b/src/lib/i18n/locales/he-IL/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "צור חשבון",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "כהה",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "בית",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "כיצד אוכל לעזור לך היום?",
"How would you rate this response?": "",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "המודל '{{modelName}}' הורד בהצלחה.",
"Model '{{modelTag}}' is already in queue for downloading.": "המודל '{{modelTag}}' כבר בתור להורדה.",
- "Model {{modelId}} not found": "המודל {{modelId}} לא נמצא",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "דגם {{modelName}} אינו בעל יכולת ראייה",
"Model {{name}} is now {{status}}": "דגם {{name}} הוא כעת {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "עוד",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "גרסת Ollama",
"On": "פועל",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "תאור",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "",
"Running...": "פועל...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "שמור",
"Save & Create": "שמור וצור",
"Save & Update": "שמור ועדכן",
@@ -1891,6 +1899,7 @@
"STT Model": "",
"STT Settings": "הגדרות חקירה של TTS",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "מנוע טקסט לדיבור",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "תודה על המשוב שלך!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2049,6 +2059,7 @@
"TTS Model": "",
"TTS Settings": "הגדרות TTS",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "סוג",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "הקלד כתובת URL של פתרון פנים מחבק (הורד)",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "רשת",
"Web API": "",
"Web Loader Engine": "",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/hi-IN/translation.json b/src/lib/i18n/locales/hi-IN/translation.json
index 71edceb884..b2d1252c76 100644
--- a/src/lib/i18n/locales/hi-IN/translation.json
+++ b/src/lib/i18n/locales/hi-IN/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "खाता बनाएं",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "डार्क",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "आज मैं आपकी कैसे मदद कर सकता हूँ?",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "मॉडल '{{modelName}}' सफलतापूर्वक डाउनलोड हो गया है।",
"Model '{{modelTag}}' is already in queue for downloading.": "मॉडल '{{modelTag}}' पहले से ही डाउनलोड करने के लिए कतार में है।",
- "Model {{modelId}} not found": "मॉडल {{modelId}} नहीं मिला",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "मॉडल {{modelName}} दृष्टि सक्षम नहीं है",
"Model {{name}} is now {{status}}": "मॉडल {{name}} अब {{status}} है",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "और..",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama Version",
"On": "चालू",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "पेरसनलाइज़मेंट",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "चल रहा है...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "सहेजें",
"Save & Create": "सहेजें और बनाएं",
"Save & Update": "सहेजें और अपडेट करें",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "STT सेटिंग्स ",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "टेक्स्ट-टू-स्पीच इंजन",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "आपकी प्रतिक्रिया के लिए धन्यवाद!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "TTS सेटिंग्स",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "प्रकार",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "हगिंग फेस रिज़ॉल्व (डाउनलोड) यूआरएल टाइप करें",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "वेब",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/hr-HR/translation.json b/src/lib/i18n/locales/hr-HR/translation.json
index 33ee5f734a..5bf1a7e65c 100644
--- a/src/lib/i18n/locales/hr-HR/translation.json
+++ b/src/lib/i18n/locales/hr-HR/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "Stvori račun",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Tamno",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Kako vam mogu pomoći danas?",
"How would you rate this response?": "",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' je uspješno preuzet.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' je već u redu za preuzimanje.",
- "Model {{modelId}} not found": "Model {{modelId}} nije pronađen",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} ne čita vizualne impute",
"Model {{name}} is now {{status}}": "Model {{name}} sada je {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Više",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama verzija",
"On": "Uključeno",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Prilagodba",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "Pokrenuto",
"Running...": "Pokrenuto...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Spremi",
"Save & Create": "Spremi i stvori",
"Save & Update": "Spremi i ažuriraj",
@@ -1891,6 +1899,7 @@
"STT Model": "STT model",
"STT Settings": "STT postavke",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Stroj za pretvorbu teksta u govor",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Hvala na povratnim informacijama!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2049,6 +2059,7 @@
"TTS Model": "TTS model",
"TTS Settings": "TTS postavke",
"TTS Voice": "TTS glas",
+ "Tu_day_of_week": "",
"Type": "Tip",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Upišite Hugging Face Resolve (Download) URL",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/hu-HU/translation.json b/src/lib/i18n/locales/hu-HU/translation.json
index f411786290..362977a7f0 100644
--- a/src/lib/i18n/locales/hu-HU/translation.json
+++ b/src/lib/i18n/locales/hu-HU/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Fiók létrehozása",
"Create Admin Account": "Admin fiók létrehozása",
+ "Create and manage scheduled automations": "",
"Create Channel": "Csatorna létrehozása",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Veszélyzóna",
"Dark": "Sötét",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Továbbítja a rendszer felhasználói munkamenet hitelesítő adatait a hitelesítéshez",
+ "Fr_day_of_week": "",
"Full Context Mode": "Teljes kontextus mód",
"Function": "Funkció",
"Function Calling": "Funkcióhívás",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Kezdőlap",
"Host": "Hoszt",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Hogyan segíthetek ma?",
"How would you rate this response?": "Hogyan értékelnéd ezt a választ?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API kulcs szükséges.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modell",
"Model '{{modelName}}' has been successfully downloaded.": "A '{{modelName}}' modell sikeresen letöltve.",
"Model '{{modelTag}}' is already in queue for downloading.": "A '{{modelTag}}' modell már a letöltési sorban van.",
- "Model {{modelId}} not found": "A {{modelId}} modell nem található",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "A {{modelName}} modell nem képes képfeldolgozásra",
"Model {{name}} is now {{status}}": "A {{name}} modell most {{status}} állapotban van",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API kulcs",
+ "Monthly": "",
"More": "Több",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama verzió",
"On": "Be",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Személyre szabás",
"Pin": "Rögzítés",
+ "Pin to Sidebar": "",
"Pinned": "Rögzítve",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Fut",
"Running...": "Fut...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Mentés",
"Save & Create": "Mentés és létrehozás",
"Save & Update": "Mentés és frissítés",
@@ -1889,6 +1897,7 @@
"STT Model": "STT modell",
"STT Settings": "STT beállítások",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Szöveg felosztó",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Szöveg-beszéd motor",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Köszönjük a visszajelzést!",
"The Application Account DN you bind with for search": "Az alkalmazás fiók DN, amellyel kereséshez kötsz",
"The base to search for users": "A felhasználók keresésének alapja",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS modell",
"TTS Settings": "TTS beállítások",
"TTS Voice": "TTS hang",
+ "Tu_day_of_week": "",
"Type": "Típus",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Adja meg a Hugging Face Resolve (Letöltési) URL-t",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Figyelmeztetés: Ennek engedélyezése lehetővé teszi a felhasználók számára, hogy tetszőleges kódot töltsenek fel a szerverre.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Figyelmeztetés: A Jupyter végrehajtás lehetővé teszi a tetszőleges kód végrehajtását, ami súlyos biztonsági kockázatot jelent – óvatosan folytassa.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "A WebUI kéréseket küld a \"{{url}}\" címre",
"WebUI will make requests to \"{{url}}/api/chat\"": "A WebUI kéréseket küld a \"{{url}}/api/chat\" címre",
"WebUI will make requests to \"{{url}}/chat/completions\"": "A WebUI kéréseket küld a \"{{url}}/chat/completions\" címre",
+ "Weekly": "",
"What are you trying to achieve?": "Mit próbálsz elérni?",
"What are you working on?": "Min dolgozol?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/id-ID/translation.json b/src/lib/i18n/locales/id-ID/translation.json
index 0d36c77484..80317e4361 100644
--- a/src/lib/i18n/locales/id-ID/translation.json
+++ b/src/lib/i18n/locales/id-ID/translation.json
@@ -453,6 +453,7 @@
"Create a new note": "",
"Create Account": "Buat Akun",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -478,6 +479,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Gelap",
@@ -968,6 +970,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1042,6 +1045,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Ada yang bisa saya bantu hari ini?",
"How would you rate this response?": "",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' telah berhasil diunduh.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' sudah berada dalam antrean untuk diunduh.",
- "Model {{modelId}} not found": "Model {{modelId}} tidak ditemukan",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} tidak dapat dilihat",
"Model {{name}} is now {{status}}": "Model {{name}} sekarang menjadi {{status}}",
@@ -1308,6 +1312,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Lainnya",
"More Concise": "",
"More options": "",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Versi Ollama",
"On": "Aktif",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "Personalisasi",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1671,6 +1678,7 @@
"Running": "Berjalan",
"Running...": "Berjalan...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Simpan",
"Save & Create": "Simpan & Buat",
"Save & Update": "Simpan & Perbarui",
@@ -1887,6 +1895,7 @@
"STT Model": "Model STT",
"STT Settings": "Pengaturan STT",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1932,6 +1941,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Mesin Teks-ke-Suara",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Terima kasih atas umpan balik Anda!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2045,6 +2055,7 @@
"TTS Model": "Model TTS",
"TTS Settings": "Pengaturan TTS",
"TTS Voice": "Suara TTS",
+ "Tu_day_of_week": "",
"Type": "Ketik",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Ketik Hugging Face Resolve (Unduh) URL",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ie-GA/translation.json b/src/lib/i18n/locales/ie-GA/translation.json
index ba6fdbc60d..2fa16b9a30 100644
--- a/src/lib/i18n/locales/ie-GA/translation.json
+++ b/src/lib/i18n/locales/ie-GA/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Cruthaigh nóta nua",
"Create Account": "Cruthaigh Cuntas",
"Create Admin Account": "Cruthaigh Cuntas Riaracháin",
+ "Create and manage scheduled automations": "",
"Create Channel": "Cruthaigh Cainéal",
"Create Folder": "Cruthaigh Fillteán",
"Create Image": "Cruthaigh Íomhá",
@@ -479,6 +480,7 @@
"Custom Gender": "Inscne Saincheaptha",
"Custom Parameter Name": "Ainm Paraiméadair Saincheaptha",
"Custom Parameter Value": "Luach Paraiméadair Saincheaptha",
+ "Daily": "",
"Daily Messages": "Teachtaireachtaí Laethúla",
"Danger Zone": "Crios Contúirte",
"Dark": "Dorcha",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Seolann sé comhartha rochtana OAuth úsáideora an chórais ar aghaidh chun fíordheimhniú a dhéanamh",
"Forwards system user session credentials to authenticate": "Cuir dintiúir seisiúin úsáideora córais ar aghaidh lena bhfíordheimhniú",
+ "Fr_day_of_week": "",
"Full Context Mode": "Mód Comhthéacs Iomlán",
"Function": "Feidhm",
"Function Calling": "Glaonna Feidhme",
@@ -1043,6 +1046,7 @@
"History": "Stair",
"Home": "Baile",
"Host": "Óstach",
+ "Hourly": "",
"Hourly Messages": "Teachtaireachtaí Uaireanta",
"How can I help you today?": "Conas is féidir liom cabhrú leat inniu?",
"How would you rate this response?": "Cad é mar a mheasfá an freagra seo?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "OCR Mistral",
"Mistral OCR API Key required.": "Mistral OCR API Eochair ag teastáil.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Samhail",
"Model '{{modelName}}' has been successfully downloaded.": "Rinneadh an tsamhail '{{modelName}}' a íoslódáil go rathúil.",
"Model '{{modelTag}}' is already in queue for downloading.": "Tá samhail '{{modelTag}}' sa scuaine cheana féin le híoslódáil.",
- "Model {{modelId}} not found": "Níor aimsíodh an tsamhail {{modelId}}",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Níl samhail {{modelName}} in ann amharc",
"Model {{name}} is now {{status}}": "Tá samhail {{name}} {{status}} anois",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Roinnt Samhlacha",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Eochair API Cuardach Mojeek",
+ "Monthly": "",
"More": "Tuilleadh",
"More Concise": "Níos Gonta",
"More options": "Tuilleadh roghanna",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Eochair API Ollama Cloud",
"Ollama Version": "Leagan Ollama",
"On": "Ar",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Gníomhach amháin nuair a bhíonn an socrú \"Greamaigh Téacs Mór mar Chomhad\" casta air.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Gníomhach ach amháin nuair a bhíonn an t-ionchur comhrá i bhfócas agus nuair a bhíonn LLM ag giniúint freagra.",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Pearsantú",
"Pin": "Bioráin",
+ "Pin to Sidebar": "",
"Pinned": "Pinneáilte",
"Pinned Messages": "Teachtaireachtaí Pionáilte",
"Pinned Models": "Samhlacha bioráilte",
@@ -1673,6 +1680,7 @@
"Running": "Ag rith",
"Running...": "Ag rith...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Ritheann sé tascanna leabaithe ag an am céanna chun luas a chur leis an bpróiseáil. Múch é má bhíonn teorainneacha ráta ina bhfadhb.",
+ "Sa_day_of_week": "",
"Save": "Sábháil",
"Save & Create": "Sábháil & Cruthaigh",
"Save & Update": "Sábháil & Nuashonraigh",
@@ -1889,6 +1897,7 @@
"STT Model": "Samhail STT",
"STT Settings": "Socruithe STT",
"Stylized PDF Export": "Easpórtáil PDF Stílithe",
+ "Su_day_of_week": "",
"Submit question": "Cuir ceist isteach",
"Submit suggestion": "Cuir moladh isteach",
"Subtitle": "Fotheideal",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Scoilteoir Téacs",
"Text-to-Speech": "Téacs-go-Caint",
"Text-to-Speech Engine": "Inneall téacs-go-labhra",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Go raibh maith agat as do chuid aiseolas!",
"The Application Account DN you bind with for search": "An Cuntas Feidhmchláir DN a nascann tú leis le haghaidh cuardaigh",
"The base to search for users": "An bonn chun cuardach a dhéanamh ar úsáideoirí",
@@ -2047,6 +2057,7 @@
"TTS Model": "Samhail TTS",
"TTS Settings": "Socruithe TTS",
"TTS Voice": "Guth TTS",
+ "Tu_day_of_week": "",
"Type": "Cineál",
"Type here...": "Clóscríobh anseo...",
"Type Hugging Face Resolve (Download) URL": "Cineál Hugging Face Resolve (Íoslódáil) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Rabhadh: Cuirfidh sé seo ar chumas úsáideoirí cód treallach a uaslódáil ar an bhfreastalaí.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Rabhadh: Trí fhorghníomhú Jupyter is féidir cód a fhorghníomhú go treallach, rud a chruthaíonn mór-rioscaí slándála - bí fíorchúramach.",
+ "We_day_of_week": "",
"Web": "Gréasán",
"Web API": "API Gréasáin",
"Web Loader Engine": "Inneall Luchtaithe Gréasáin",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "Déanfaidh WebUI iarratais ar \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "Déanfaidh WebUI iarratais ar \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "Déanfaidh WebUI iarratais ar \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Cad atá tú ag iarraidh a bhaint amach?",
"What are you working on?": "Cad air a bhfuil tú ag obair?",
"What is NOT shared:": "Cad NACH roinntear",
diff --git a/src/lib/i18n/locales/it-IT/translation.json b/src/lib/i18n/locales/it-IT/translation.json
index eb3e42aa68..8cb9822986 100644
--- a/src/lib/i18n/locales/it-IT/translation.json
+++ b/src/lib/i18n/locales/it-IT/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "Crea account",
"Create Admin Account": "Crea account amministratore",
+ "Create and manage scheduled automations": "",
"Create Channel": "Crea canale",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Nome parametro personalizzato",
"Custom Parameter Value": "Valore parametro personalizzato",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Zona di pericolo",
"Dark": "Scuro",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Inoltra le credenziali della sessione utente di sistema per autenticare",
+ "Fr_day_of_week": "",
"Full Context Mode": "Modalità Contesto Completo",
"Function": "Funzione",
"Function Calling": "Chiamata Funzione",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "Home",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Come posso aiutarti oggi?",
"How would you rate this response?": "Come valuteresti questa risposta?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "OCR Mistral",
"Mistral OCR API Key required.": "La Chiave API OCR Mistral è richiesta.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modello",
"Model '{{modelName}}' has been successfully downloaded.": "Il modello '{{modelName}}' è stato scaricato con successo.",
"Model '{{modelTag}}' is already in queue for downloading.": "Il modello '{{modelTag}}' è già in coda per il download.",
- "Model {{modelId}} not found": "Modello {{modelId}} non trovato",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Il modello {{modelName}} non è in grado di vedere",
"Model {{name}} is now {{status}}": "Il modello {{name}} è ora {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Chiave API di Mojeek Search",
+ "Monthly": "",
"More": "Altro",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Versione Ollama",
"On": "Attivato",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Personalizzazione",
"Pin": "Appunta",
+ "Pin to Sidebar": "",
"Pinned": "Appuntato",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "In esecuzione",
"Running...": "In esecuzione...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Salva",
"Save & Create": "Salva e crea",
"Save & Update": "Salva e aggiorna",
@@ -1891,6 +1899,7 @@
"STT Model": "Modello STT",
"STT Settings": "Impostazioni STT",
"Stylized PDF Export": "Esportazione PDF Stilizzata",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Divisore di testo",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Motore da testo a voce",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Grazie per il tuo feedback!",
"The Application Account DN you bind with for search": "L'account dell'applicazione DN con cui ti colleghi per la ricerca",
"The base to search for users": "La base da cercare per gli utenti",
@@ -2049,6 +2059,7 @@
"TTS Model": "Modello TTS",
"TTS Settings": "Impostazioni TTS",
"TTS Voice": "Voce TTS",
+ "Tu_day_of_week": "",
"Type": "Digitare",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Digita URL di Risoluzione (Download) di Hugging Face",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Attenzione: abilitando questo, gli utenti potranno caricare codice arbitrario sul server.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Attenzione: l'esecuzione di Jupyter consente l'esecuzione di codice arbitrario, comportando gravi rischi per la sicurezza: procedere con estrema cautela.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Motore di Caricamento Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI farà richieste a \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI farà richieste a \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI farà richieste a \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Cosa stai cercando di ottenere?",
"What are you working on?": "Su cosa stai lavorando?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ja-JP/translation.json b/src/lib/i18n/locales/ja-JP/translation.json
index c4032c2728..be5c8d6a98 100644
--- a/src/lib/i18n/locales/ja-JP/translation.json
+++ b/src/lib/i18n/locales/ja-JP/translation.json
@@ -11,7 +11,7 @@
"{{ models }}": "{{ モデル }}",
"{{COUNT}} Available Tools": "{{COUNT}} 個の有効なツール",
"{{COUNT}} characters": "{{COUNT}} 文字",
- "{{COUNT}} extracted lines": "",
+ "{{COUNT}} extracted lines": "{{COUNT}} 行を抽出",
"{{COUNT}} files": "",
"{{COUNT}} hidden lines": "{{COUNT}} 行が非表示",
"{{COUNT}} members": "{{COUNT}} メンバー",
@@ -185,7 +185,7 @@
"Are you sure you want to delete all chats? This action cannot be undone.": "すべてのチャットを削除しますか？ この操作は元に戻すことができません。",
"Are you sure you want to delete this channel?": "このチャンネルを削除しますか？",
"Are you sure you want to delete this connection? This action cannot be undone.": "",
- "Are you sure you want to delete this memory? This action cannot be undone.": "",
+ "Are you sure you want to delete this memory? This action cannot be undone.": "このメモリをクリアしますか？ この操作は元に戻すことができません。",
"Are you sure you want to delete this message?": "このメッセージを削除しますか？",
"Are you sure you want to delete this version? Child versions will be relinked to this version's parent.": "",
"Are you sure you want to delete this?": "",
@@ -198,7 +198,7 @@
"Assistant": "アシスタント",
"Async Embedding Processing": "",
"Attach File From Knowledge": "ナレッジからファイルを添付",
- "Attach Files": "",
+ "Attach Files": "ファイルを追加",
"Attach Knowledge": "ナレッジを追加",
"Attach Notes": "ノートを追加",
"Attach Webpage": "ウェブページを追加",
@@ -221,13 +221,13 @@
"AUTOMATIC1111 Base URL": "AUTOMATIC1111 ベース URL",
"AUTOMATIC1111 Base URL is required.": "AUTOMATIC1111 ベース URL が必要です。",
"Automatically inject system tools in native function calling mode (e.g., timestamps, memory, chat history, notes, etc.)": "ネイティブの関数呼び出しモードにおいて、システムツール(例: タイムスタンプ、メモリー、チャット履歴、ノートなど)を自動的に注入します",
- "Automation": "",
- "Automation created": "",
- "Automation Name": "",
- "Automation title": "",
- "Automation triggered": "",
- "Automation updated": "",
- "Automations": "",
+ "Automation": "オートメーション",
+ "Automation created": "オートメーションを作成しました",
+ "Automation Name": "オートメーション名",
+ "Automation title": "オートメーションのタイトル",
+ "Automation triggered": "オートメーションが実行されました",
+ "Automation updated": "オートメーションを更新しました",
+ "Automations": "オートメーション",
"Available list": "利用可能リスト",
"Available models": "",
"Available Tools": "利用可能ツール",
@@ -392,7 +392,7 @@
"Concurrent Requests": "同時リクエスト",
"Config": "",
"Config imported successfully": "設定のインポートに成功しました",
- "Configuration": "",
+ "Configuration": "設定",
"Configure": "設定",
"Confirm": "確認",
"Confirm Password": "パスワードの確認",
@@ -453,6 +453,7 @@
"Create a new note": "新しいノートを作成する",
"Create Account": "アカウントを作成",
"Create Admin Account": "管理者アカウントを作成",
+ "Create and manage scheduled automations": "",
"Create Channel": "チャンネルを作成",
"Create Folder": "フォルダを作成",
"Create Image": "",
@@ -462,7 +463,7 @@
"Create new secret key": "新しいシークレットキーを作成",
"Create note": "ノートを作成",
"Create Note": "ノートを作成",
- "Create scheduled prompts that run automatically on a recurring basis.": "",
+ "Create scheduled prompts that run automatically on a recurring basis.": "定期的に自動実行されるプロンプトを作成します。",
"Create your first note by clicking on the plus button below.": "プラスボタンをクリックして最初のノートを作成します。",
"Created at": "作成日時",
"Created At": "作成日時",
@@ -478,6 +479,7 @@
"Custom Gender": "",
"Custom Parameter Name": "カスタムパラメータ名",
"Custom Parameter Value": "カスタムパラメータ値",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "危険地帯",
"Dark": "ダーク",
@@ -515,13 +517,13 @@
"Delete All": "すべて削除する",
"Delete All Chats": "すべてのチャットを削除",
"Delete all contents inside this folder": "",
- "Delete automation?": "",
+ "Delete automation?": "オートメーションを削除しますか？",
"Delete Chat": "チャットを削除",
"Delete chat?": "チャットを削除しますか？",
"Delete File": "",
"Delete folder?": "フォルダーを削除しますか？",
"Delete function?": "Functionを削除しますか？",
- "Delete Memory?": "",
+ "Delete Memory?": "メモリを削除しますか？",
"Delete Message": "メッセージを削除",
"Delete message?": "メッセージを削除しますか？",
"Delete Model": "",
@@ -752,7 +754,7 @@
"Enter Perplexity Search API URL": "",
"Enter Playwright Timeout": "Playwrightタイムアウトを入力",
"Enter Playwright WebSocket URL": "Playwright WebSocket URLを入力",
- "Enter prompt here.": "",
+ "Enter prompt here.": "ここにプロンプトを入力",
"Enter proxy URL (e.g. https://user:password@host:port)": "プロキシURLを入力 (例: https://user:password@host:port)",
"Enter reasoning effort": "推論の努力を入力",
"Enter Score": "スコアを入力",
@@ -777,7 +779,7 @@
"Enter system prompt here": "システムプロンプトをここに入力",
"Enter Tavily API Key": "Tavily API Keyを入力",
"Enter Tavily Extract Depth": "Tavily Extract Depthを入力",
- "Enter the prompt instructions for this automation...": "",
+ "Enter the prompt instructions for this automation...": "このオートメーションのプロンプトを入力してください...",
"Enter the public URL of your WebUI. This URL will be used to generate links in the notifications.": "WebUIの公開URLを入力してください。このURLは通知でリンクを生成するために使用されます。",
"Enter the URL of the function to import": "インポートするFunctionのURLを入力",
"Enter the URL to import": "インポートするURLを入力",
@@ -835,7 +837,7 @@
"Execute code": "",
"Execute code for analysis": "コードの分析に実行",
"Executing **{{NAME}}**...": "**{{NAME}}**を実行中...",
- "Execution Logs": "",
+ "Execution Logs": "実行ログ",
"Expand": "展開",
"Experimental": "実験的",
"Explain": "説明",
@@ -964,10 +966,11 @@
"Form": "フォーム",
"Format Lines": "出力テキストをフォーマット",
"Format the lines in the output. Defaults to False. If set to True, the lines will be formatted to detect inline math and styles.": "出力をフォーマットする。デフォルトでは無効です。有効にすると、インライン数式やスタイルを検出しフォーマットします。",
- "Formatting may be inconsistent from source.": "",
+ "Formatting may be inconsistent from source.": "元のデータにより、書式が一致しない場合があります。",
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "システムユーザーセッションの資格情報を転送して認証する",
+ "Fr_day_of_week": "",
"Full Context Mode": "フルコンテキストモード",
"Function": "",
"Function Calling": "Function呼び出し",
@@ -1042,6 +1045,7 @@
"History": "",
"Home": "ホーム",
"Host": "ホスト",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "今日はどのようにお手伝いしましょうか？",
"How would you rate this response?": "この応答をどのように評価しますか？",
@@ -1101,7 +1105,7 @@
"Insert Suggestion Prompt to Input": "",
"Install from Github URL": "Github URLからインストール",
"Instant Auto-Send After Voice Transcription": "音声文字変換後に自動送信",
- "Instructions": "",
+ "Instructions": "指示",
"Integration": "連携",
"Integrations": "連携",
"Interface": "インターフェース",
@@ -1161,7 +1165,7 @@
"Last 90 days": "",
"Last Active": "最終アクティブ",
"Last Modified": "最終変更",
- "Last ran": "",
+ "Last ran": "最終実行",
"Last reply": "最終応答",
"LDAP": "LDAP",
"LDAP server updated": "LDAPサーバーの更新に成功しました",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "Mistral OCR APIキーが必要です。",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "モデル",
"Model '{{modelName}}' has been successfully downloaded.": "モデル '{{modelName}}' が正常にダウンロードされました。",
"Model '{{modelTag}}' is already in queue for downloading.": "モデル '{{modelTag}}' はすでにダウンロード待機中です。",
- "Model {{modelId}} not found": "モデル {{modelId}} が見つかりません",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "モデル {{modelName}} は視覚に対応していません",
"Model {{name}} is now {{status}}": "モデル {{name}} は {{status}} になりました。",
@@ -1308,6 +1312,7 @@
"Models Sharing": "モデルの共有",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search APIキー",
+ "Monthly": "",
"More": "もっと見る",
"More Concise": "より簡潔に",
"More options": "",
@@ -1318,11 +1323,11 @@
"Name": "名前",
"Name and ID are required, please fill them out": "名前とIDは必須です。項目を入力してください。",
"Name your knowledge base": "ナレッジベースに名前を付ける",
- "Name, prompt, and model are required": "",
+ "Name, prompt, and model are required": "名前、プロンプト、モデルの選択が必要です",
"Native": "ネイティブ",
- "Never": "",
+ "Never": "なし",
"New": "",
- "New Automation": "",
+ "New Automation": "新しいオートメーション",
"New Button": "新しいボタン",
"New Chat": "新しいチャット",
"New File": "",
@@ -1341,11 +1346,11 @@
"New Webhook": "",
"new-channel": "新しいチャンネル",
"Next message": "次のメッセージ",
- "Next run": "",
+ "Next run": "次回実行",
"No access grants. Private to you.": "アクセス権は付与されていません。あなただけが利用できます。",
"No activity data": "",
"No authentication": "",
- "No automations found": "",
+ "No automations found": "オートメーションが見つかりません",
"No chats found": "チャットが見つかりません。",
"No chats found for this user.": "このユーザーのチャットが見つかりません。",
"No chats found.": "チャットが見つかりません。",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama バージョン",
"On": "オン",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "\"大きなテキストをファイルとして貼り付ける\"がオンの場合にのみ有効です。",
"Only active when the chat input is in focus and an LLM is generating a response.": "チャット入力欄にフォーカスがあり、LLMが応答を生成しているときのみ有効です。",
@@ -1493,7 +1499,7 @@
"Password": "パスワード",
"Passwords do not match.": "パスワードが一致しません。",
"Paste Large Text as File": "大きなテキストをファイルとして貼り付ける",
- "Paused": "",
+ "Paused": "停止中",
"PDF document (.pdf)": "PDF ドキュメント (.pdf)",
"PDF Extract Images (OCR)": "PDF 画像抽出 (OCR)",
"PDF Loader Mode": "",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "パーソナライズ",
"Pin": "ピン留め",
+ "Pin to Sidebar": "",
"Pinned": "ピン留めされています",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1634,7 +1641,7 @@
"Renamed to {{name}}": "",
"Render Markdown in Previews": "",
"Reorder Models": "モデルを並べ替え",
- "Repeats": "",
+ "Repeats": "繰り返し",
"Reply": "",
"Reply in Thread": "スレッドで返信",
"Reply to thread...": "",
@@ -1666,11 +1673,12 @@
"RTL": "RTL",
"Run": "実行",
"Run All": "",
- "Run now": "",
- "Run Now": "",
+ "Run now": "今すぐ実行",
+ "Run Now": "今すぐ実行",
"Running": "実行中",
"Running...": "実行中...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "保存",
"Save & Create": "保存して作成",
"Save & Update": "保存して更新",
@@ -1678,7 +1686,7 @@
"Save Chat": "チャットを保存",
"Saved": "保存しました。",
"Saving chat logs directly to your browser's storage is no longer supported. Please take a moment to download and delete your chat logs by clicking the button below. Don't worry, you can easily re-import your chat logs to the backend through": "チャットログをブラウザのストレージに直接保存する機能はサポートされなくなりました。下のボタンをクリックして、チャットログをダウンロードして削除してください。ご心配なく。チャットログは、次の方法でバックエンドに簡単に再インポートできます。",
- "Schedule": "",
+ "Schedule": "スケジュール",
"Scheduled time must be in the future": "",
"Scroll On Branch Change": "ブランチ変更時にスクロール",
"Search": "検索",
@@ -1686,7 +1694,7 @@
"Search all emojis": "絵文字を検索",
"Search and manage user memories": "",
"Search and view user chat history": "",
- "Search Automations": "",
+ "Search Automations": "オートメーションを検索",
"Search Base": "ベースを検索",
"Search channels and channel messages": "",
"Search Chats": "チャットの検索",
@@ -1702,7 +1710,7 @@
"Search Groups": "グループの検索",
"Search In Models": "モデルを検索",
"Search Knowledge": "ナレッジベースの検索",
- "Search Memories": "",
+ "Search Memories": "メモリの検索",
"Search Models": "モデル検索",
"Search Notes": "ノートを検索",
"Search options": "検索オプション",
@@ -1756,7 +1764,7 @@
"Select how to split message text for TTS requests": "TTSリクエストのテキスト分割方法を選択",
"Select Knowledge": "ナレッジベースの選択",
"Select Method": "",
- "Select model": "",
+ "Select model": "モデルを選択",
"Select only one model to call": "1つのモデルを呼び出すには、1つのモデルを選択してください。",
"Select view": "",
"Selected model: {{modelName}}": "",
@@ -1866,7 +1874,7 @@
"Start of the channel": "チャンネルの開始",
"Start Tag": "",
"Starting kernel...": "",
- "State": "",
+ "State": "状態",
"Status": "ステータス",
"Status cleared successfully": "正常にステータスをクリアしました",
"Status updated successfully": "正常にステータスを更新しました",
@@ -1887,6 +1895,7 @@
"STT Model": "STTモデル",
"STT Settings": "STT設定",
"Stylized PDF Export": "スタイル付きPDFエクスポート",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "サブタイトル",
@@ -1932,6 +1941,7 @@
"Text Splitter": "テキスト分割",
"Text-to-Speech": "テキスト音声変換",
"Text-to-Speech Engine": "テキスト音声変換エンジン",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "ご意見ありがとうございます！",
"The Application Account DN you bind with for search": "LDAP属性を使用してユーザーを検索するためにバインドするアプリケーションアカウントのDN。",
"The base to search for users": "ユーザーを検索するためのベース。",
@@ -2001,7 +2011,7 @@
"To access the WebUI, please reach out to the administrator. Admins can manage user statuses from the Admin Panel.": "WebUIにアクセスするには、管理者にお問い合わせください。管理者は管理者パネルからユーザーのステータスを管理できます。",
"To attach knowledge base here, add them to the \"Knowledge\" workspace first.": "ここにナレッジベースを追加するには、まず \"Knowledge\" ワークスペースに追加してください。",
"To learn more about available endpoints, visit our documentation.": "利用可能なエンドポイントについては、ドキュメントを参照してください。",
- "To select skills here, add them to the \"Skills\" workspace first.": "",
+ "To select skills here, add them to the \"Skills\" workspace first.": "ここでSkillを選択するには、まず\"Skills\" ワークスペースに追加してください。",
"To select toolkits here, add them to the \"Tools\" workspace first.": "ここでツールキットを選択するには、まず \"Tools\" ワークスペースに追加してください。",
"Toast notifications for new updates": "新しい更新のトースト通知",
"Today": "今日",
@@ -2045,6 +2055,7 @@
"TTS Model": "TTSモデル",
"TTS Settings": "TTS 設定",
"TTS Voice": "TTSボイス",
+ "Tu_day_of_week": "",
"Type": "種類",
"Type here...": "ここに入力してください...",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (ダウンロード) URL を入力してください",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "警告: これを有効にすると、ユーザーがサーバー上で任意のコードをアップロードできるようになります。",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "警告: Jupyter 実行は任意のコード実行を可能にし、重大なセキュリティリスクを伴います。極めて慎重に進めてください。",
+ "We_day_of_week": "",
"Web": "ウェブ",
"Web API": "ウェブAPI",
"Web Loader Engine": "ウェブローダーエンジン",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUIは\"{{url}}\"にリクエストを行います",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUIは\"{{url}}/api/chat\"にリクエストを行います",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUIは\"{{url}}/chat/completions\"にリクエストを行います",
+ "Weekly": "",
"What are you trying to achieve?": "何を達成したいですか？",
"What are you working on?": "何に取り組んでいますか？",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ka-GE/translation.json b/src/lib/i18n/locales/ka-GE/translation.json
index 67c99ba548..cdbc392315 100644
--- a/src/lib/i18n/locales/ka-GE/translation.json
+++ b/src/lib/i18n/locales/ka-GE/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "ახალი შენიშვნის შექმნა",
"Create Account": "ანგარიშის შექმნა",
"Create Admin Account": "ადმინისტრატორის ანგარიშის შექმნა",
+ "Create and manage scheduled automations": "",
"Create Channel": "არხის შექმნა",
"Create Folder": "საქაღალდის შექმნა",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "მორგებული პარამეტრის სახელი",
"Custom Parameter Value": "მორგებული პარამეტრის მნიშვნელობა",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "საშიში ზონა",
"Dark": "მუქი",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "სრული კონტექსტის რეჟიმი",
"Function": "ფუნქცია",
"Function Calling": "ფუნქციის გამოძახება",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "მთავარი",
"Host": "ჰოსტი",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "რით შემიძლია დაგეხმაროთ დღეს?",
"How would you rate this response?": "როგორ შეაფასებდით ამ პასუხს?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "მოდელი",
"Model '{{modelName}}' has been successfully downloaded.": "მოდელის „{{modelName}}“ გადმოწერა წარმატებით დასრულდა.",
"Model '{{modelTag}}' is already in queue for downloading.": "მოდელი „{{modelTag}}“ უკვე გადმოწერის რიგშია.",
- "Model {{modelId}} not found": "მოდელი {{modelId}} აღმოჩენილი არაა",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} is not vision capable",
"Model {{name}} is now {{status}}": "Model {{name}} is now {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "მეტი",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama Cloud API-ის გასაღები",
"Ollama Version": "Ollama ვერსია",
"On": "ჩართული",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "პერსონალიზაცია",
"Pin": "მიმაგრება",
+ "Pin to Sidebar": "",
"Pinned": "მიმაგრებულია",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "გაშვებულია",
"Running...": "გაშვებულია...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "შენახვა",
"Save & Create": "შენახვა და შექმნა",
"Save & Update": "შენახვა და განახლება",
@@ -1889,6 +1897,7 @@
"STT Model": "STT მოდელი",
"STT Settings": "STT-ის მორგება",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "ტექსტის გამყოფი",
"Text-to-Speech": "ტექსტის-ხმამაღლა-წაკითხვა",
"Text-to-Speech Engine": "ტექსტურ-ხმოვანი ძრავი",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "მადლობა გამოხმაურებისთვის!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS მოდელი",
"TTS Settings": "TTS პარამეტრები",
"TTS Voice": "TTS ხმა",
+ "Tu_day_of_week": "",
"Type": "ტიპი",
"Type here...": "აკრიფეთ აქ...",
"Type Hugging Face Resolve (Download) URL": "აკრიფეთ HuggingFace-ის ამოხსნის (გადმოწერის) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "ვები",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "რას ცდილობთ, მიაღწიოთ?",
"What are you working on?": "რაზე მუშაობთ?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/kab-DZ/translation.json b/src/lib/i18n/locales/kab-DZ/translation.json
index b38b37e73c..82dcd4c833 100644
--- a/src/lib/i18n/locales/kab-DZ/translation.json
+++ b/src/lib/i18n/locales/kab-DZ/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Snulfu-d amiḍan",
"Create Admin Account": "Snulfu-d amiḍan n unedbal",
+ "Create and manage scheduled automations": "",
"Create Channel": "Snulfu-d abadu",
"Create Folder": "Snulfu-d akaram",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Isem n uɣewwar udmawan",
"Custom Parameter Value": "Azal n uɣewwar udmawan",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Tamnaḍt i iweɛren",
"Dark": "Aberkan",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Welleh inekcam n tɣimit n useqdac i usesteb",
+ "Fr_day_of_week": "",
"Full Context Mode": "Askar n usatal aččuran",
"Function": "Tasɣent",
"Function Calling": "Asiwel n twuriwin",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Agejdan",
"Host": "Asneftaɣ",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Amek i zemreɣ ad k·kem-ɛiwneɣ ass-a?",
"How would you rate this response?": "Amek ara d-teskefleḍ tiririt-a?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "API Mistral OCR Tesri tasarut.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Tamudemt",
"Model '{{modelName}}' has been successfully downloaded.": "Tettwasider-d tmudemt '{{modelName}}' akken iwata.",
"Model '{{modelTag}}' is already in queue for downloading.": "Tamudemt '{{modelTag}}' ha-t-an yakan deg tebdart n usader.",
- "Model {{modelId}} not found": "Tamudemt {{modelId}} ulac-itt",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Mudel Isem}} mačči d tamuɣli izemren ad tili",
"Model {{name}} is now {{status}}": "Tamudemt {{name}} tura {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Tasarut API n Mojeek",
+ "Monthly": "",
"More": "Ugar",
"More Concise": "Awezlan ugar",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Tasarut API n Ollama Cloud",
"Ollama Version": "Lqem n Ollama",
"On": "Irmed",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Asagen",
"Pin": "Senteḍ",
+ "Pin to Sidebar": "",
"Pinned": "Yettwasenteḍ",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Aselkem",
"Running...": "Aselkem...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Sekles",
"Save & Create": "Sekles rnu snulfu-d",
"Save & Update": "Sekles rnu leqqem",
@@ -1889,6 +1897,7 @@
"STT Model": "Tamudemt n uɛqal n taɣect",
"STT Settings": "Iɣewwaren n uɛqal n tavect",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Amebḍay n uḍris",
"Text-to-Speech": "Aḍris-ɣer-taɣect",
"Text-to-Speech Engine": "Amsadday n TTS",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Tanemmirt ɣef tikti-inek·inem!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "Taffa n unadi ɣef yiseqdacen",
@@ -2047,6 +2057,7 @@
"TTS Model": "Tamudemt TTS",
"TTS Settings": "Iɣewwaṛen n TTS",
"TTS Voice": "Taɣect n TTS",
+ "Tu_day_of_week": "",
"Type": "Anaw",
"Type here...": "Aru da…",
"Type Hugging Face Resolve (Download) URL": "Anaw n usefres n wudem amezwer (Download) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Ɣur-k: Asewḥel n waya ad yeǧǧ iseqdacen ad d-salin tangalt tazurant ɣef uqeddac.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Ɣur-k: Asselkem n Jupyter yettaǧǧa asselkem n tengalt tazurant, d tukksa n tmijwin n tɣellist qessiḥen — s leḥder meqqren.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Amsedday n uzdam Web",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI ad ssutreɣ \"{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI ad ssutreɣ i \"{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI ad ssutreɣ i \"{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Sanda ay tettarmed ad tessiwḍed?",
"What are you working on?": "Ɣef wacu ay la tettmahaled?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ko-KR/translation.json b/src/lib/i18n/locales/ko-KR/translation.json
index c4492175ca..96a52e359d 100644
--- a/src/lib/i18n/locales/ko-KR/translation.json
+++ b/src/lib/i18n/locales/ko-KR/translation.json
@@ -453,6 +453,7 @@
"Create a new note": "새 노트 생성",
"Create Account": "계정 생성",
"Create Admin Account": "관리자 계정 생성",
+ "Create and manage scheduled automations": "",
"Create Channel": "채널 생성",
"Create Folder": "폴더 생성",
"Create Image": "이미지 생성",
@@ -478,6 +479,7 @@
"Custom Gender": "",
"Custom Parameter Name": "사용자 정의 매개변수 이름",
"Custom Parameter Value": "사용자 정의 매개변수 값",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "위험 기능",
"Dark": "다크",
@@ -968,6 +970,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "인증을 위해 시스템 사용자 OAuth 액세스 토큰을 전달합니다.",
"Forwards system user session credentials to authenticate": "인증을 위해 시스템 사용자 세션 자격 증명 전달",
+ "Fr_day_of_week": "",
"Full Context Mode": "전체 컨텍스트 모드",
"Function": "함수",
"Function Calling": "함수 호출",
@@ -1042,6 +1045,7 @@
"History": "",
"Home": "홈",
"Host": "호스트",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "무엇을 도와드릴까요?",
"How would you rate this response?": "이 응답을 어떻게 평가하시겠어요?",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "Mistral OCR API Key가 필요합니다.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "모델",
"Model '{{modelName}}' has been successfully downloaded.": "모델 '{{modelName}}'이/가 성공적으로 다운로드되었습니다.",
"Model '{{modelTag}}' is already in queue for downloading.": "모델 '{{modelTag}}'은/는 이미 다운로드 대기열에 있습니다.",
- "Model {{modelId}} not found": "모델 {{modelId}}을/를 찾을 수 없습니다.",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "모델 {{modelName}}은/는 비전을 사용할 수 없습니다.",
"Model {{name}} is now {{status}}": "모델 {{name}}은/는 이제 {{status}} 상태입니다.",
@@ -1308,6 +1312,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API 키",
+ "Monthly": "",
"More": "더보기",
"More Concise": "더 간결하게",
"More options": "",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama 버전",
"On": "켜기",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "\"긴 텍스트를 파일로 붙여넣기\" 설정이 켜져 있을 때만 작동합니다.",
"Only active when the chat input is in focus and an LLM is generating a response.": "채팅 입력창이 선택되어 있고 LLM이 응답을 생성 중일 때만 작동합니다.",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "개인화",
"Pin": "고정",
+ "Pin to Sidebar": "",
"Pinned": "고정됨",
"Pinned Messages": "고정된 메시지",
"Pinned Models": "",
@@ -1671,6 +1678,7 @@
"Running": "실행 중",
"Running...": "실행 중...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "저장",
"Save & Create": "저장 및 생성",
"Save & Update": "저장 및 업데이트",
@@ -1887,6 +1895,7 @@
"STT Model": "STT 모델",
"STT Settings": "STT 설정",
"Stylized PDF Export": "서식이 적용된 PDF 내보내기",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1932,6 +1941,7 @@
"Text Splitter": "텍스트 나누기",
"Text-to-Speech": "텍스트-음성 변환",
"Text-to-Speech Engine": "텍스트-음성 변환 엔진",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "피드백 감사합니다!",
"The Application Account DN you bind with for search": "검색을 위해 바인딩하는 애플리케이션 계정 DN",
"The base to search for users": "사용자를 검색할 수 있는 기반",
@@ -2045,6 +2055,7 @@
"TTS Model": "TTS 모델",
"TTS Settings": "TTS 설정",
"TTS Voice": "TTS 음성",
+ "Tu_day_of_week": "",
"Type": "입력",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (다운로드) URL 입력",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "주의: 이 기능을 활성화하면 사용자가 서버에 임의 코드를 업로드할 수 있습니다.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "경고: Jupyter 실행은 임의의 코드 실행을 가능하게 하여 심각한 보안 위험을 초래합니다. — 매우 신중하게 진행하세요.",
+ "We_day_of_week": "",
"Web": "웹",
"Web API": "웹 API",
"Web Loader Engine": "웹 로더 엔진",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI가 \"{{url}}\"로 요청을 보냅니다",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI가 \"{{url}}/api/chat\"로 요청을 보냅니다",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI가 \"{{url}}/chat/completions\"로 요청을 보냅니다",
+ "Weekly": "",
"What are you trying to achieve?": "무엇을 성취하고 싶으신가요?",
"What are you working on?": "어떤 작업을 하고 계신가요?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/lt-LT/translation.json b/src/lib/i18n/locales/lt-LT/translation.json
index f6e1694dfc..87819d1d2e 100644
--- a/src/lib/i18n/locales/lt-LT/translation.json
+++ b/src/lib/i18n/locales/lt-LT/translation.json
@@ -456,6 +456,7 @@
"Create a new note": "",
"Create Account": "Créer un compte",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -481,6 +482,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Tamsus",
@@ -971,6 +973,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1045,6 +1048,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Kuo galėčiau Jums padėti ?",
"How would you rate this response?": "",
@@ -1267,10 +1271,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "'{{modelName}}' modelis sėkmingai atsisiųstas.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modelis '{{modelTag}}' jau atsisiuntimų eilėje.",
- "Model {{modelId}} not found": "Modelis {{modelId}} nerastas",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Modelis {{modelName}} neturi vaizdo gebėjimų",
"Model {{name}} is now {{status}}": "Modelis {{name}} dabar {{status}}",
@@ -1311,6 +1315,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Daugiau",
"More Concise": "",
"More options": "",
@@ -1431,6 +1436,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama versija",
"On": "Aktyvuota",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1515,6 +1521,7 @@
"Persistent": "",
"Personalization": "Personalizacija",
"Pin": "Smeigtukas",
+ "Pin to Sidebar": "",
"Pinned": "Įsmeigta",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1677,6 +1684,7 @@
"Running": "Veikia",
"Running...": "Veikia...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Išsaugoti",
"Save & Create": "Išsaugoti ir sukurti",
"Save & Update": "Išsaugoti ir atnaujinti",
@@ -1893,6 +1901,7 @@
"STT Model": "STT modelis",
"STT Settings": "STT nustatymai",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1938,6 +1947,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Balso sintezės modelis",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Ačiū už atsiliepimus",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2051,6 +2061,7 @@
"TTS Model": "TTS modelis",
"TTS Settings": "TTS parametrai",
"TTS Voice": "TTS balsas",
+ "Tu_day_of_week": "",
"Type": "Tipas",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Įveskite Hugging Face Resolve nuorodą",
@@ -2153,6 +2164,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2169,6 +2181,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/lv-LV/translation.json b/src/lib/i18n/locales/lv-LV/translation.json
index d3ce9d12f7..648f032641 100644
--- a/src/lib/i18n/locales/lv-LV/translation.json
+++ b/src/lib/i18n/locales/lv-LV/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "Izveidot jaunu piezīmi",
"Create Account": "Izveidot kontu",
"Create Admin Account": "Izveidot administratora kontu",
+ "Create and manage scheduled automations": "",
"Create Channel": "Izveidot kanālu",
"Create Folder": "Izveidot mapi",
"Create Image": "Izveidot attēlu",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Pielāgota parametra nosaukums",
"Custom Parameter Value": "Pielāgota parametra vērtība",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Bīstamā zona",
"Dark": "Tumšs",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Pārsūta sistēmas lietotāja OAuth piekļuves tokenu autentifikācijai",
"Forwards system user session credentials to authenticate": "Pārsūta sistēmas lietotāja sesijas akreditācijas datus autentifikācijai",
+ "Fr_day_of_week": "",
"Full Context Mode": "Pilna konteksta režīms",
"Function": "Funkcija",
"Function Calling": "Funkciju izsaukšana",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "Sākums",
"Host": "Hosts",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Kā es varu jums šodien palīdzēt?",
"How would you rate this response?": "Kā jūs novērtētu šo atbildi?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Nepieciešama Mistral OCR API atslēga.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Modelis",
"Model '{{modelName}}' has been successfully downloaded.": "Modelis '{{modelName}}' ir veiksmīgi lejupielādēts.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modelis '{{modelTag}}' jau ir lejupielādes rindā.",
- "Model {{modelId}} not found": "Modelis {{modelId}} nav atrasts",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Modelis {{modelName}} neatbalsta redzi",
"Model {{name}} is now {{status}}": "Modelis {{name}} tagad ir {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "Modeļu kopīgošana",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Mojeek Search API atslēga",
+ "Monthly": "",
"More": "Vairāk",
"More Concise": "Kodolīgāk",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "Ollama Cloud API atslēga",
"Ollama Version": "Ollama versija",
"On": "Ieslēgts",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Aktīvs tikai, ja ir ieslēgts iestatījums \"Ielīmēt lielu tekstu kā failu\".",
"Only active when the chat input is in focus and an LLM is generating a response.": "Aktīvs tikai, kad tērzēšanas ievade ir fokusā un LLM ģenerē atbildi.",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Personalizācija",
"Pin": "Piespraust",
+ "Pin to Sidebar": "",
"Pinned": "Piesprausts",
"Pinned Messages": "Piespraustie ziņojumi",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "Darbojas",
"Running...": "Darbojas...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Palaiž iegulšanas uzdevumus vienlaicīgi, lai paātrinātu apstrādi. Izslēdziet, ja rodas ātruma ierobežojumu problēmas.",
+ "Sa_day_of_week": "",
"Save": "Saglabāt",
"Save & Create": "Saglabāt un izveidot",
"Save & Update": "Saglabāt un atjaunināt",
@@ -1891,6 +1899,7 @@
"STT Model": "STT modelis",
"STT Settings": "STT iestatījumi",
"Stylized PDF Export": "Stilizēts PDF eksports",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "Apakšvirsraksts",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Teksta sadalītājs",
"Text-to-Speech": "Teksts uz runu",
"Text-to-Speech Engine": "Teksta uz runas dzinējs",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Paldies par jūsu atsauksmēm!",
"The Application Account DN you bind with for search": "Lietojumprogrammas konta DN, ar kuru saistāties meklēšanai",
"The base to search for users": "Bāze lietotāju meklēšanai",
@@ -2049,6 +2059,7 @@
"TTS Model": "TTS modelis",
"TTS Settings": "TTS iestatījumi",
"TTS Voice": "TTS balss",
+ "Tu_day_of_week": "",
"Type": "Tips",
"Type here...": "Rakstiet šeit...",
"Type Hugging Face Resolve (Download) URL": "Ievadiet Hugging Face Resolve (lejupielādes) URL",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Brīdinājums: Šīs opcijas iespējošana ļaus lietotājiem augšupielādēt patvaļīgu kodu serverī.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Brīdinājums: Jupyter izpilde ļauj patvaļīgu koda izpildi, radot nopietnus drošības riskus — rīkojieties ar ārkārtēju piesardzību.",
+ "We_day_of_week": "",
"Web": "Tīmeklis",
"Web API": "Tīmekļa API",
"Web Loader Engine": "Tīmekļa ielādētāja dzinējs",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI veiks pieprasījumus uz \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI veiks pieprasījumus uz \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI veiks pieprasījumus uz \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Ko jūs mēģināt sasniegt?",
"What are you working on?": "Pie kā jūs strādājat?",
"What is NOT shared:": "Kas NETIEK kopīgots:",
diff --git a/src/lib/i18n/locales/ms-MY/translation.json b/src/lib/i18n/locales/ms-MY/translation.json
index a819fa33b4..cf1f755f08 100644
--- a/src/lib/i18n/locales/ms-MY/translation.json
+++ b/src/lib/i18n/locales/ms-MY/translation.json
@@ -453,6 +453,7 @@
"Create a new note": "Buat nota baru",
"Create Account": "Cipta Akaun",
"Create Admin Account": "Buat Akaun Admin",
+ "Create and manage scheduled automations": "",
"Create Channel": "Buat Saluran",
"Create Folder": "Buat Folder",
"Create Image": "Buat Imej",
@@ -478,6 +479,7 @@
"Custom Gender": "Jantina Tersuai",
"Custom Parameter Name": "Nama Parameter Tersuai",
"Custom Parameter Value": "Nilai Parameter Tersuai",
+ "Daily": "",
"Daily Messages": "Mesej Harian",
"Danger Zone": "Zon Bahaya",
"Dark": "Gelap",
@@ -968,6 +970,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Meneruskan token akses OAuth pengguna sistem untuk pengesahan",
"Forwards system user session credentials to authenticate": "Meneruskan bukti kelayakan sesi pengguna sistem untuk pengesahan",
+ "Fr_day_of_week": "",
"Full Context Mode": "Mod Konteks Penuh",
"Function": "Fungsi",
"Function Calling": "Pemanggilan Fungsi",
@@ -1042,6 +1045,7 @@
"History": "Sejarah",
"Home": "Halaman Utama",
"Host": "Hos",
+ "Hourly": "",
"Hourly Messages": "Mesej Setiap Jam",
"How can I help you today?": "Bagaimana saya boleh membantu anda hari ini?",
"How would you rate this response?": "Bagaimana anda menilai respons ini?",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Kunci API Mistral OCR diperlukan.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{ modelName }}' telah berjaya dimuat turun.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{ modelTag }}' sudah dalam baris gilir untuk dimuat turun.",
- "Model {{modelId}} not found": "Model {{ modelId }} tidak dijumpai",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{ modelName }} tidak mempunyai keupayaan penglihatan",
"Model {{name}} is now {{status}}": "Model {{name}} kini {{status}}",
@@ -1308,6 +1312,7 @@
"Models Sharing": "Perkongsian Model",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Kunci API Pencarian Mojeek",
+ "Monthly": "",
"More": "Lagi",
"More Concise": "Lebih Ringkas",
"More options": "Lebih banyak pilihan",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "Kunci API Ollama Cloud",
"Ollama Version": "Versi Ollama",
"On": "Hidup",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Hanya aktif apabila tetapan \"Tampal Teks Besar sebagai Fail\" dihidupkan.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Hanya aktif apabila input sembang berada dalam fokus dan LLM sedang menjana respons.",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "Personalisasi",
"Pin": "Pin",
+ "Pin to Sidebar": "",
"Pinned": "Disemat",
"Pinned Messages": "Mesej Disematkan",
"Pinned Models": "Model Tersapu",
@@ -1671,6 +1678,7 @@
"Running": "Sedang dijalankan",
"Running...": "Sedang dijalankan...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Menjalankan tugas penyisipan secara serentak untuk mempercepatkan pemprosesan. Matikan jika had kadar menjadi isu.",
+ "Sa_day_of_week": "",
"Save": "Simpan",
"Save & Create": "Simpan & Cipta",
"Save & Update": "Simpan & Kemas Kini",
@@ -1887,6 +1895,7 @@
"STT Model": "Model STT",
"STT Settings": "Tetapan STT",
"Stylized PDF Export": "Eksport PDF Bergaya",
+ "Su_day_of_week": "",
"Submit question": "Hantar soalan",
"Submit suggestion": "Hantar cadangan",
"Subtitle": "Subtitle",
@@ -1932,6 +1941,7 @@
"Text Splitter": "Pemisah Teks",
"Text-to-Speech": "Teks-ke-Ucapan",
"Text-to-Speech Engine": "Enjin Teks-ke-Ucapan",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Terima kasih atas maklum balas anda!",
"The Application Account DN you bind with for search": "DN Akaun Aplikasi yang anda ikat untuk carian",
"The base to search for users": "Pangkalan untuk mencari pengguna",
@@ -2045,6 +2055,7 @@
"TTS Model": "Model TTS",
"TTS Settings": "Tetapan TTS",
"TTS Voice": "Suara TTS",
+ "Tu_day_of_week": "",
"Type": "jenis",
"Type here...": "Taip di sini...",
"Type Hugging Face Resolve (Download) URL": "Taip URL 'Hugging Face Resolve (Download)'",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Amaran: Mengaktifkan ini akan membenarkan pengguna memuat naik kod arbitrari pada pelayan.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Amaran: Pelaksanaan Jupyter membolehkan pelaksanaan kod arbitrari, menimbulkan risiko keselamatan yang teruk—teruskan dengan berhati-hati.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Enjin Pemuatan Web",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI akan membuat permintaan ke \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI akan membuat permintaan ke \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI akan membuat permintaan ke \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Apa yang ingin anda capai?",
"What are you working on?": "Apa yang sedang anda kerjakan?",
"What is NOT shared:": "Apa yang TIDAK dikongsi:",
diff --git a/src/lib/i18n/locales/nb-NO/translation.json b/src/lib/i18n/locales/nb-NO/translation.json
index 2c2d6ba0d2..0fdd082794 100644
--- a/src/lib/i18n/locales/nb-NO/translation.json
+++ b/src/lib/i18n/locales/nb-NO/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Opprett konto",
"Create Admin Account": "Opprett administratorkonto",
+ "Create and manage scheduled automations": "",
"Create Channel": "Opprett kanal",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Mørk",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "Modus for full kontekst",
"Function": "Funksjon",
"Function Calling": "Kalling av funksjon",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Hjem",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Hva kan jeg hjelpe deg med i dag?",
"How would you rate this response?": "Hvordan vurderer du dette svaret?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modell",
"Model '{{modelName}}' has been successfully downloaded.": "Modellen {{modelName}} er lastet ned.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modellen {{modelTag}} er allerede i nedlastingskøen.",
- "Model {{modelId}} not found": "Finner ikke modellen {{modelId}}",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Modellen {{modelName}} er ikke egnet til visuelle data",
"Model {{name}} is now {{status}}": "Modellen {{name}} er nå {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "API-nøekkel for Mojeek Search",
+ "Monthly": "",
"More": "Mer",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama-versjon",
"On": "Aktivert",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Tilpassing",
"Pin": "Fest",
+ "Pin to Sidebar": "",
"Pinned": "Festet",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Kjører",
"Running...": "Kjører...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Lagre",
"Save & Create": "Lagre og opprett",
"Save & Update": "Lagre og oppdater",
@@ -1889,6 +1897,7 @@
"STT Model": "STT-modell",
"STT Settings": "STT-innstillinger",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Oppdeling av tekst",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Tekst-til-tale-motor",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Takk for tilbakemeldingen!",
"The Application Account DN you bind with for search": "Applikasjonskontoens DN du binder deg med for søking",
"The base to search for users": "Basen for å søke etter brukere",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS-modell",
"TTS Settings": "TTS-innstillinger",
"TTS Voice": "TTS-stemme",
+ "Tu_day_of_week": "",
"Type": "Type",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Angi nedlastings-Resolve-URL for Hugging Face",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Advarsel: Hvis du aktiverer denne funksjonen, kan brukere laste opp vilkårlig kode på serveren.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Advarsel! Jupyter gjør det mulig å kjøre vilkårlig kode, noe som utgjør en alvorlig sikkerhetsrisiko. Utvis ekstrem forsiktighet.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web-API",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI vil rette forespørsler til \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI vil rette forespørsler til \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Hva prøver du å oppnå?",
"What are you working on?": "Hva jobber du på nå?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/nl-NL/translation.json b/src/lib/i18n/locales/nl-NL/translation.json
index 9feab00ffe..f54b0190b3 100644
--- a/src/lib/i18n/locales/nl-NL/translation.json
+++ b/src/lib/i18n/locales/nl-NL/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Maak account",
"Create Admin Account": "Maak admin-account",
+ "Create and manage scheduled automations": "",
"Create Channel": "Maak kanaal",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Gevarenzone",
"Dark": "Donker",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "Volledige contextmodus",
"Function": "Functie",
"Function Calling": "Functieaanroep",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Thuis",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Hoe kan ik je vandaag helpen?",
"How would you rate this response?": "Hoe zou je dit antwoord beoordelen?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' is succesvol gedownload.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' staat al in de wachtrij voor downloaden.",
- "Model {{modelId}} not found": "Model {{modelId}} niet gevonden",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} is niet geschikt voor visie",
"Model {{name}} is now {{status}}": "Model {{name}} is nu {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API-sleutel",
+ "Monthly": "",
"More": "Meer",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama Versie",
"On": "Aan",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Personalisatie",
"Pin": "Zet vast",
+ "Pin to Sidebar": "",
"Pinned": "Vastgezet",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Aan het uitvoeren",
"Running...": "Aan het uitvoeren...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Opslaan",
"Save & Create": "Opslaan & Creëren",
"Save & Update": "Opslaan & Bijwerken",
@@ -1889,6 +1897,7 @@
"STT Model": "STT Model",
"STT Settings": "STT Instellingen",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Tekst splitser",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Tekst-naar-Spraak Engine",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Bedankt voor je feedback!",
"The Application Account DN you bind with for search": "Het applicatieaccount DN waarmee je zoekt",
"The base to search for users": "De basis om gebruikers te zoeken",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS Model",
"TTS Settings": "TTS instellingen",
"TTS Voice": "TTS Stem",
+ "Tu_day_of_week": "",
"Type": "Type",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Type Hugging Face Resolve (Download) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Waarschuwing: Door dit in te schakelen kunnen gebruikers willekeurige code uploaden naar de server.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Waarschuwing: Jupyter kan willekeurige code uitvoeren, wat ernstige veiligheidsrisico's met zich meebrengt - ga uiterst voorzichtig te werk. ",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web-API",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI zal verzoeken doen aan \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI zal verzoeken doen aan \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Wat probeer je te bereiken?",
"What are you working on?": "Waar werk je aan?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/pa-IN/translation.json b/src/lib/i18n/locales/pa-IN/translation.json
index 952a681610..4e43ac1ff3 100644
--- a/src/lib/i18n/locales/pa-IN/translation.json
+++ b/src/lib/i18n/locales/pa-IN/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "ਖਾਤਾ ਬਣਾਓ",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "ਗੂੜ੍ਹਾ",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "ਮੈਂ ਅੱਜ ਤੁਹਾਡੀ ਕਿਵੇਂ ਮਦਦ ਕਰ ਸਕਦਾ ਹਾਂ?",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "",
"Model '{{modelName}}' has been successfully downloaded.": "ਮਾਡਲ '{{modelName}}' ਸਫਲਤਾਪੂਰਵਕ ਡਾਊਨਲੋਡ ਕੀਤਾ ਗਿਆ ਹੈ।",
"Model '{{modelTag}}' is already in queue for downloading.": "ਮਾਡਲ '{{modelTag}}' ਪਹਿਲਾਂ ਹੀ ਡਾਊਨਲੋਡ ਲਈ ਕਤਾਰ ਵਿੱਚ ਹੈ।",
- "Model {{modelId}} not found": "ਮਾਡਲ {{modelId}} ਨਹੀਂ ਮਿਲਿਆ",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "ਮਾਡਲ {{modelName}} ਦ੍ਰਿਸ਼ਟੀ ਸਮਰੱਥ ਨਹੀਂ ਹੈ",
"Model {{name}} is now {{status}}": "ਮਾਡਲ {{name}} ਹੁਣ {{status}} ਹੈ",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "ਹੋਰ",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "ਓਲਾਮਾ ਵਰਜਨ",
"On": "ਚਾਲੂ",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "ਪਰਸੋਨਲਿਸ਼ਮ",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "ਚੱਲ ਰਿਹਾ ਹੈ...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "ਸੰਭਾਲੋ",
"Save & Create": "ਸੰਭਾਲੋ ਅਤੇ ਬਣਾਓ",
"Save & Update": "ਸੰਭਾਲੋ ਅਤੇ ਅੱਪਡੇਟ ਕਰੋ",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "STT ਸੈਟਿੰਗਾਂ",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "ਪਾਠ-ਤੋਂ-ਬੋਲ ਇੰਜਣ",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "ਤੁਹਾਡੇ ਫੀਡਬੈਕ ਲਈ ਧੰਨਵਾਦ!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "TTS ਸੈਟਿੰਗਾਂ",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "ਕਿਸਮ",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (ਡਾਊਨਲੋਡ) URL ਟਾਈਪ ਕਰੋ",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "ਵੈਬ",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/pl-PL/translation.json b/src/lib/i18n/locales/pl-PL/translation.json
index 5194bf8be7..c41577a730 100644
--- a/src/lib/i18n/locales/pl-PL/translation.json
+++ b/src/lib/i18n/locales/pl-PL/translation.json
@@ -456,6 +456,7 @@
"Create a new note": "Utwórz nową notatkę",
"Create Account": "Utwórz konto",
"Create Admin Account": "Utwórz konto administratora",
+ "Create and manage scheduled automations": "",
"Create Channel": "Utwórz kanał",
"Create Folder": "Utwórz folder",
"Create Image": "Utwórz obraz",
@@ -481,6 +482,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Nazwa parametru niestandardowego",
"Custom Parameter Value": "Wartość parametru niestandardowego",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Strefa krytyczna",
"Dark": "Ciemny",
@@ -971,6 +973,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Przekazuje token dostępu OAuth użytkownika systemowego",
"Forwards system user session credentials to authenticate": "Przekazuje poświadczenia sesji użytkownika systemowego",
+ "Fr_day_of_week": "",
"Full Context Mode": "Tryb pełnego kontekstu",
"Function": "Funkcja",
"Function Calling": "Wywoływanie funkcji",
@@ -1045,6 +1048,7 @@
"History": "",
"Home": "Strona główna",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "W czym mogę pomóc?",
"How would you rate this response?": "Jak oceniasz tę odpowiedź?",
@@ -1267,10 +1271,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Wymagany klucz API Mistral OCR.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Model '{{modelName}}' został pomyślnie pobrany.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' jest już w kolejce pobierania.",
- "Model {{modelId}} not found": "Nie znaleziono modelu {{modelId}}",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} nie obsługuje widzenia (Vision)",
"Model {{name}} is now {{status}}": "Model {{name}} jest teraz {{status}}",
@@ -1311,6 +1315,7 @@
"Models Sharing": "Udostępnianie modeli",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Klucz API Mojeek Search",
+ "Monthly": "",
"More": "Więcej",
"More Concise": "Bardziej zwięzły",
"More options": "",
@@ -1431,6 +1436,7 @@
"Ollama Cloud API Key": "Klucz API Ollama Cloud",
"Ollama Version": "Wersja Ollama",
"On": "Wł.",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Aktywne tylko gdy opcja \"Wklej duży tekst jako plik\" jest włączona.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Aktywne tylko gdy pole czatu jest aktywne a LLM generuje odpowiedź.",
@@ -1515,6 +1521,7 @@
"Persistent": "",
"Personalization": "Personalizacja",
"Pin": "Przypnij",
+ "Pin to Sidebar": "",
"Pinned": "Przypięte",
"Pinned Messages": "Przypięte wiadomości",
"Pinned Models": "",
@@ -1677,6 +1684,7 @@
"Running": "Działa",
"Running...": "Działa...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Uruchamia zadania embeddingu współbieżnie. Wyłącz, jeśli masz limity API.",
+ "Sa_day_of_week": "",
"Save": "Zapisz",
"Save & Create": "Zapisz i utwórz",
"Save & Update": "Zapisz i aktualizuj",
@@ -1893,6 +1901,7 @@
"STT Model": "Model STT",
"STT Settings": "Ustawienia STT",
"Stylized PDF Export": "Stylizowany eksport PDF",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "Podtytuł",
@@ -1938,6 +1947,7 @@
"Text Splitter": "Text Splitter",
"Text-to-Speech": "Silnik syntezy mowy (TTS)",
"Text-to-Speech Engine": "Silnik syntezy mowy (TTS)",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Dzięki za opinię!",
"The Application Account DN you bind with for search": "DN konta aplikacji do wyszukiwania",
"The base to search for users": "Baza wyszukiwania użytkowników",
@@ -2051,6 +2061,7 @@
"TTS Model": "Model TTS",
"TTS Settings": "Ustawienia TTS",
"TTS Voice": "Głos TTS",
+ "Tu_day_of_week": "",
"Type": "Typ",
"Type here...": "Pisz tutaj...",
"Type Hugging Face Resolve (Download) URL": "Wpisz URL Hugging Face Resolve (Pobieranie)",
@@ -2153,6 +2164,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Uwaga: Włączenie tego pozwoli użytkownikom na przesyłanie dowolnego kodu na serwer.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Uwaga: Jupyter umożliwia wykonanie dowolnego kodu (ryzyko bezpieczeństwa) – zachowaj szczególną ostrożność.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "Silnik ładowania stron",
@@ -2169,6 +2181,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI będzie wysyłać żądania do \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI będzie wysyłać żądania do \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI będzie wysyłać żądania do \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Co chcesz osiągnąć?",
"What are you working on?": "Nad czym pracujesz?",
"What is NOT shared:": "Co NIE jest udostępniane:",
diff --git a/src/lib/i18n/locales/pt-BR/translation.json b/src/lib/i18n/locales/pt-BR/translation.json
index 3eb41f721e..2a4e99c95c 100644
--- a/src/lib/i18n/locales/pt-BR/translation.json
+++ b/src/lib/i18n/locales/pt-BR/translation.json
@@ -183,7 +183,7 @@
"Are you sure you want to archive all chats? This action cannot be undone.": "Tem certeza de que deseja arquivar todos os chats? Esta ação não pode ser desfeita.",
"Are you sure you want to clear all memories? This action cannot be undone.": "Tem certeza de que deseja apagar todas as memórias? Esta ação não pode ser desfeita.",
"Are you sure you want to delete \"{{NAME}}\"?": "Tem certeza de que deseja excluir \"{{NAME}}\"?",
- "Are you sure you want to delete **{{modelName}}**?": "",
+ "Are you sure you want to delete **{{modelName}}**?": "Tem certeza de que deseja excluir **{{modelName}}**?",
"Are you sure you want to delete all chats? This action cannot be undone.": "Tem certeza de que deseja excluir todas as conversas? Esta ação não pode ser desfeita.",
"Are you sure you want to delete this channel?": "Tem certeza de que deseja excluir este canal?",
"Are you sure you want to delete this connection? This action cannot be undone.": "Tem certeza de que deseja excluir esta conexão? Esta ação não pode ser desfeita.",
@@ -200,7 +200,7 @@
"Assistant": "Assistente",
"Async Embedding Processing": "Processamento de Embedding assíncrono",
"Attach File From Knowledge": "Anexar arquivo da base de conhecimento",
- "Attach Files": "",
+ "Attach Files": "Anexar arquivos",
"Attach Knowledge": "Anexar Base de Conhecimento",
"Attach Notes": "Anexar Notas",
"Attach Webpage": "Anexar Página Web",
@@ -223,13 +223,13 @@
"AUTOMATIC1111 Base URL": "URL Base AUTOMATIC1111",
"AUTOMATIC1111 Base URL is required.": "URL Base AUTOMATIC1111 é necessária.",
"Automatically inject system tools in native function calling mode (e.g., timestamps, memory, chat history, notes, etc.)": "Injetar automaticamente ferramentas do sistema no modo de chamada de função nativa (por exemplo, carimbos de data/hora, memória, histórico de chat, notas, etc.)",
- "Automation": "",
- "Automation created": "",
- "Automation Name": "",
- "Automation title": "",
- "Automation triggered": "",
- "Automation updated": "",
- "Automations": "",
+ "Automation": "Automação",
+ "Automation created": "Automação criada",
+ "Automation Name": "Nome da automação",
+ "Automation title": "Título de automação",
+ "Automation triggered": "Automação acionada",
+ "Automation updated": "Automação atualizada",
+ "Automations": "Automações",
"Available list": "Lista disponível",
"Available models": "Modelos disponíveis",
"Available Tools": "Ferramentas disponíveis",
@@ -260,7 +260,7 @@
"Boosting or penalizing specific tokens for constrained responses. Bias values will be clamped between -100 and 100 (inclusive). (Default: none)": "Aumentar ou penalizar tokens específicos para respostas restritas. Os valores de viés serão fixados entre -100 e 100 (inclusive). (Padrão: nenhum)",
"Brave": "Brave",
"Brave Search API Key": "Chave API do Brave Search",
- "Break down complex requests into trackable steps": "",
+ "Break down complex requests into trackable steps": "Divida solicitações complexas em etapas rastreáveis.",
"Browse and query knowledge bases": "Navegue e consulte bases de conhecimento.",
"Builtin Tools": "Ferramentas integradas",
"Bullet List": "Lista com marcadores",
@@ -394,7 +394,7 @@
"Concurrent Requests": "Solicitações simultâneas",
"Config": "Configuração",
"Config imported successfully": "Configuração importada com sucesso",
- "Configuration": "",
+ "Configuration": "Configuração",
"Configure": "Configurar",
"Confirm": "Confirmar",
"Confirm Password": "Confirmar Senha",
@@ -455,6 +455,7 @@
"Create a new note": "Criar uma nova nota",
"Create Account": "Criar Conta",
"Create Admin Account": "Criar Conta de Administrador",
+ "Create and manage scheduled automations": "",
"Create Channel": "Criar Canal",
"Create Folder": "Criar Pasta",
"Create Image": "Criar imagem",
@@ -464,7 +465,7 @@
"Create new secret key": "Criar nova chave secreta",
"Create note": "Criar nota",
"Create Note": "Criar Nota",
- "Create scheduled prompts that run automatically on a recurring basis.": "",
+ "Create scheduled prompts that run automatically on a recurring basis.": "Crie prompts agendados que sejam executados automaticamente de forma recorrente.",
"Create your first note by clicking on the plus button below.": "Crie sua primeira nota clicando no botão de adição abaixo.",
"Created at": "Criado em",
"Created At": "Criado Em",
@@ -480,13 +481,14 @@
"Custom Gender": "Gênero personalizado",
"Custom Parameter Name": "Nome do parâmetro personalizado",
"Custom Parameter Value": "Valor do parâmetro personalizado",
+ "Daily": "",
"Daily Messages": "Mensagens Diárias",
"Danger Zone": "Zona de Perigo",
"Dark": "Escuro",
"Data Controls": "Controle de Dados",
"Database": "Banco de Dados",
"Datalab Marker API": "API do Marcador do Datalab",
- "Day": "",
+ "Day": "Dia",
"DD/MM/YYYY": "DD/MM/AAAA",
"DDGS Backend": "Backend DDGS",
"December": "Dezembro",
@@ -517,7 +519,7 @@
"Delete All": "Excluir tudo",
"Delete All Chats": "Excluir Todos os Chats",
"Delete all contents inside this folder": "Apague todo o conteúdo desta pasta.",
- "Delete automation?": "",
+ "Delete automation?": "Excluir automação?",
"Delete Chat": "Excluir Chat",
"Delete chat?": "Excluir chat?",
"Delete File": "Excluir arquivo",
@@ -662,7 +664,7 @@
"Embedding Concurrent Requests": "Solicitações Simultâneas de Embedding",
"Embedding Model": "Modelo de Embedding",
"Embedding Model Engine": "Motor do Modelo de Embedding",
- "Emojis": "",
+ "Emojis": "Emojis",
"Empty message": "Mensagem vazia",
"Enable All": "Ativar tudo",
"Enable API Keys": "Habilitar Chaves de API",
@@ -754,7 +756,7 @@
"Enter Perplexity Search API URL": "Insira a URL da API de pesquisa Perplexity",
"Enter Playwright Timeout": "Insira o tempo limite do Playwright",
"Enter Playwright WebSocket URL": "Insira a URL do WebSocket do Playwright",
- "Enter prompt here.": "",
+ "Enter prompt here.": "Insira o prompt aqui.",
"Enter proxy URL (e.g. https://user:password@host:port)": "Insira a URL do proxy (por exemplo, https://usuário:senha@host:porta)",
"Enter reasoning effort": "Insira o esforço de raciocínio",
"Enter Score": "Digite a Pontuação",
@@ -779,7 +781,7 @@
"Enter system prompt here": "Insira o prompt do sistema aqui",
"Enter Tavily API Key": "Digite a Chave API do Tavily",
"Enter Tavily Extract Depth": "Insira a profundidade de extração do Tavily",
- "Enter the prompt instructions for this automation...": "",
+ "Enter the prompt instructions for this automation...": "Insira as instruções para esta automação...",
"Enter the public URL of your WebUI. This URL will be used to generate links in the notifications.": "Insira a URL pública da sua WebUI. Esta URL será usada para gerar links nas notificações.",
"Enter the URL of the function to import": "Digite a URL da função a ser importada",
"Enter the URL to import": "Digite a URL para importar",
@@ -819,7 +821,7 @@
"Error accessing directory": "Erro ao acessar o diretório",
"Error accessing Google Drive: {{error}}": "Erro ao acessar o Google Drive: {{error}}",
"Error accessing media devices.": "Erro ao acessar dispositivos de mídia.",
- "Error deleting model: {{error}}": "",
+ "Error deleting model: {{error}}": "Erro ao excluir o modelo: {{error}}",
"Error starting recording.": "Erro ao iniciar a gravação.",
"Error unloading model: {{error}}": "Erro ao descarregar modelo: {{error}}",
"Error uploading file: {{error}}": "Erro ao carregar o arquivo: {{error}}",
@@ -837,7 +839,7 @@
"Execute code": "Executar código",
"Execute code for analysis": "Executar código para análise",
"Executing **{{NAME}}**...": "Executando **{{NAME}}**...",
- "Execution Logs": "",
+ "Execution Logs": "Registros de execução",
"Expand": "Expandir",
"Experimental": "Experimental",
"Explain": "Explicar",
@@ -847,8 +849,8 @@
"Export": "Exportar",
"Export All Archived Chats": "Exportar todos os chats arquivados",
"Export All Chats (All Users)": "Exportar Todos os Chats (Todos os Usuários)",
- "Export as CSV": "",
- "Export as JSON": "",
+ "Export as CSV": "Exportar como CSV",
+ "Export as JSON": "Exportar como JSON",
"Export chat (.json)": "Exportar chat (.json)",
"Export Chats": "Exportar Chats",
"Export Config": "Exportar Configuração",
@@ -970,6 +972,7 @@
"Forward": "Encaminhar",
"Forwards system user OAuth access token to authenticate": "Encaminha o token de acesso OAuth do usuário do sistema para autenticação",
"Forwards system user session credentials to authenticate": "Encaminha as credenciais da sessão do usuário do sistema para autenticação",
+ "Fr_day_of_week": "",
"Full Context Mode": "Modo de contexto completo",
"Function": "Função",
"Function Calling": "Chamada de função",
@@ -1044,6 +1047,7 @@
"History": "Histórico",
"Home": "Início",
"Host": "Servidor",
+ "Hourly": "",
"Hourly Messages": "Mensagens por Hora",
"How can I help you today?": "Como posso ajudar você hoje?",
"How would you rate this response?": "Como você avalia essa resposta?",
@@ -1103,7 +1107,7 @@
"Insert Suggestion Prompt to Input": "Inserir prompt de sugestão para entrada",
"Install from Github URL": "Instalar da URL do Github",
"Instant Auto-Send After Voice Transcription": "Envio Automático Instantâneo Após Transcrição de Voz",
- "Instructions": "",
+ "Instructions": "Instruções",
"Integration": "Integração",
"Integrations": "Integrações",
"Interface": "Interface",
@@ -1163,7 +1167,7 @@
"Last 90 days": "Últimos 90 dias",
"Last Active": "Última Atividade",
"Last Modified": "Última Modificação",
- "Last ran": "",
+ "Last ran": "Última execução",
"Last reply": "Última resposta",
"LDAP": "LDAP",
"LDAP server updated": "Servidor LDAP atualizado",
@@ -1266,11 +1270,11 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Chave de API do Mistral OCR necessária.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Modelo",
"Model '{{modelName}}' has been successfully downloaded.": "Modelo '{{modelName}}' foi baixado com sucesso.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modelo '{{modelTag}}' já está na fila para download.",
- "Model {{modelId}} not found": "Modelo {{modelId}} não encontrado",
- "Model {{modelName}} deleted successfully": "",
+ "Model {{modelName}} deleted successfully": "Modelo {{modelName}} excluído com sucesso",
"Model {{modelName}} is not vision capable": "Modelo {{modelName}} não é capaz de visão",
"Model {{name}} is now {{status}}": "Modelo {{name}} está agora {{status}}",
"Model {{name}} is now hidden": "O modelo {{name}} agora está oculto",
@@ -1310,6 +1314,7 @@
"Models Sharing": "Compartilhamento de Modelos",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Chave de API Mojeek Search",
+ "Monthly": "",
"More": "Mais",
"More Concise": "Mais conciso",
"More options": "Mais opções",
@@ -1320,11 +1325,11 @@
"Name": "Nome",
"Name and ID are required, please fill them out": "Nome e ID são obrigatórios, por favor preencha-os",
"Name your knowledge base": "Nome da sua base de conhecimento",
- "Name, prompt, and model are required": "",
+ "Name, prompt, and model are required": "Nome, prompt e modelo são obrigatórios.",
"Native": "Nativo",
- "Never": "",
+ "Never": "Nunca",
"New": "Novo",
- "New Automation": "",
+ "New Automation": "Nova Automação",
"New Button": "Novo Botão",
"New Chat": "Novo Chat",
"New File": "Novo Arquivo",
@@ -1343,11 +1348,11 @@
"New Webhook": "Novo Webhook",
"new-channel": "novo-canal",
"Next message": "Próxima mensagem",
- "Next run": "",
+ "Next run": "Próxima execução",
"No access grants. Private to you.": "Sem permissões de acesso. Privacidade exclusiva para você.",
"No activity data": "Sem dados de atividade",
"No authentication": "Sem autenticação",
- "No automations found": "",
+ "No automations found": "Nenhuma automação encontrada",
"No chats found": "Nenhum chat encontrado",
"No chats found for this user.": "Nenhum chat encontrado para este usuário.",
"No chats found.": "Nenhum chat encontrado.",
@@ -1358,7 +1363,7 @@
"No data": "Sem dados",
"No data found": "Nenhum dado encontrado",
"No distance available": "Sem distância disponível",
- "No execution logs available yet": "",
+ "No execution logs available yet": "Ainda não há registros de execução disponíveis.",
"No expiration can pose security risks.": "A ausência de expiração pode representar riscos de segurança.",
"No feedback found": "Nenhum feedback encontrado",
"No file selected": "Nenhum arquivo selecionado",
@@ -1405,7 +1410,7 @@
"Not factually correct": "Não está factualmente correto",
"Not helpful": "Não é útil",
"Not Registered": "Não registrado",
- "Not scheduled": "",
+ "Not scheduled": "Não agendado",
"Note": "Nota",
"Note deleted successfully": "Nota excluída com sucesso",
"Note: If you set a minimum score, the search will only return documents with a score greater than or equal to the minimum score.": "Nota: Se você definir uma pontuação mínima, a pesquisa retornará apenas documentos com pontuação igual ou superior à pontuação mínima.",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "Chave da API Ollama Cloud",
"Ollama Version": "Versão Ollama",
"On": "Ligado",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Ativo somente quando a configuração \"Colar texto grande como arquivo\" estiver ativada.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Ativo somente quando o campo de entrada do chat está em foco e um LLM está gerando uma resposta.",
@@ -1495,7 +1501,7 @@
"Password": "Senha",
"Passwords do not match.": "As senhas não coincidem.",
"Paste Large Text as File": "Cole Textos Longos como Arquivo",
- "Paused": "",
+ "Paused": "Em pausa",
"PDF document (.pdf)": "Documento PDF (.pdf)",
"PDF Extract Images (OCR)": "Extrair Imagens do PDF (OCR)",
"PDF Loader Mode": "Modo de carregamento de PDF",
@@ -1514,6 +1520,7 @@
"Persistent": "Persistente",
"Personalization": "Personalização",
"Pin": "Fixar",
+ "Pin to Sidebar": "",
"Pinned": "Fixado",
"Pinned Messages": "Mensagens fixadas",
"Pinned Models": "Modelos Fixados",
@@ -1600,7 +1607,7 @@
"Reason": "Razão",
"Reasoning Effort": "Esforço de raciocínio",
"Reasoning Tags": "Tags de raciocínio",
- "Recently Used": "",
+ "Recently Used": "Usado recentemente",
"Record": "Gravar",
"Record voice": "Gravar voz",
"Redirecting you to Open WebUI Community": "Redirecionando você para a Comunidade OpenWebUI",
@@ -1636,7 +1643,7 @@
"Renamed to {{name}}": "Renomeado para {{name}}",
"Render Markdown in Previews": "Renderizar Markdown nas Pré-visualizações",
"Reorder Models": "Reordenar modelos",
- "Repeats": "",
+ "Repeats": "Repetições",
"Reply": "Responder",
"Reply in Thread": "Responder no tópico",
"Reply to thread...": "Responder ao tópico...",
@@ -1670,11 +1677,12 @@
"RTL": "Direita para Esquerda",
"Run": "Executar",
"Run All": "Executar Tudo",
- "Run now": "",
- "Run Now": "",
+ "Run now": "Executar agora",
+ "Run Now": "Executar Agora",
"Running": "Executando",
"Running...": "Executando...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Executa tarefas de incorporação simultaneamente para acelerar o processamento. Desative se os limites de taxa se tornarem um problema.",
+ "Sa_day_of_week": "",
"Save": "Salvar",
"Save & Create": "Salvar e Criar",
"Save & Update": "Salvar e Atualizar",
@@ -1682,15 +1690,15 @@
"Save Chat": "Salvar Chat",
"Saved": "Armazenado",
"Saving chat logs directly to your browser's storage is no longer supported. Please take a moment to download and delete your chat logs by clicking the button below. Don't worry, you can easily re-import your chat logs to the backend through": "Salvar registros de chat diretamente no armazenamento do seu navegador não é mais suportado. Por favor, reserve um momento para baixar e excluir seus registros de chat clicando no botão abaixo. Não se preocupe, você pode facilmente reimportar seus registros de chat para o backend através de",
- "Schedule": "",
- "Scheduled time must be in the future": "",
+ "Schedule": "Agendar",
+ "Scheduled time must be in the future": "O horário agendado deve ser no futuro.",
"Scroll On Branch Change": "Rolar na mudança de ramo",
"Search": "Pesquisar",
"Search a model": "Pesquisar um modelo",
"Search all emojis": "Pesquisar todos os emojis",
"Search and manage user memories": "Pesquisar e gerenciar memórias de usuários",
"Search and view user chat history": "Pesquise e visualize o histórico de chat do usuário",
- "Search Automations": "",
+ "Search Automations": "Pesquisar Automações",
"Search Base": "Pesquisar Base",
"Search channels and channel messages": "Pesquisar canais e mensagens de canais",
"Search Chats": "Pesquisar Chats",
@@ -1760,7 +1768,7 @@
"Select how to split message text for TTS requests": "Selecione como dividir o texto da mensagem para solicitações TTS",
"Select Knowledge": "Selecionar Conhecimento",
"Select Method": "Selecione o método",
- "Select model": "",
+ "Select model": "Selecione o modelo",
"Select only one model to call": "Selecione apenas um modelo para chamar",
"Select view": "Selecionar visualização",
"Selected model: {{modelName}}": "Modelo selecionado: {{modelName}}",
@@ -1870,7 +1878,7 @@
"Start of the channel": "Início do canal",
"Start Tag": "Tag inicial",
"Starting kernel...": "Iniciando kernel...",
- "State": "",
+ "State": "Estado",
"Status": "Status",
"Status cleared successfully": "Status liberado com sucesso",
"Status updated successfully": "Status atualizado com sucesso",
@@ -1891,6 +1899,7 @@
"STT Model": "Modelo STT",
"STT Settings": "Configurações STT",
"Stylized PDF Export": "Exportação de PDF estilizado",
+ "Su_day_of_week": "",
"Submit question": "Enviar pergunta",
"Submit suggestion": "Enviar sugestão",
"Subtitle": "Subtítulo",
@@ -1921,10 +1930,10 @@
"Talk to Model": "Fale com o modelo",
"Tap to interrupt": "Toque para interromper",
"Task List": "Lista de tarefas",
- "Task Management": "",
+ "Task Management": "Gerenciamento de Tarefas",
"Task Model": "Modelo de Tarefa",
"Tasks": "Tarefas",
- "tasks completed": "",
+ "tasks completed": "tarefas concluídas",
"Tavily API Key": "Chave da API Tavily",
"Tavily Extract Depth": "Profundidade de extração do Tavily",
"Tell us more:": "Conte-nos mais:",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Divisor de Texto",
"Text-to-Speech": "Texto-para-Fala",
"Text-to-Speech Engine": "Motor de Texto para Fala",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Obrigado pelo seu comentário!",
"The Application Account DN you bind with for search": "O DN (Distinguished Name) da Conta de Aplicação com a qual você se conecta para pesquisa.",
"The base to search for users": "Base para pesquisar usuários.",
@@ -1991,7 +2001,7 @@
"Tika": "Tika",
"Tika Server URL required.": "URL do servidor Tika necessária.",
"Tiktoken": "Tiktoken",
- "Time": "",
+ "Time": "Tempo",
"Time & Calculation": "Tempo e Cálculo",
"Timeout": "Tempo limite",
"Title": "Título",
@@ -2009,7 +2019,7 @@
"To select toolkits here, add them to the \"Tools\" workspace first.": "Para selecionar kits de ferramentas aqui, adicione-os ao espaço de trabalho \"Ferramentas\" primeiro.",
"Toast notifications for new updates": "Notificações de alerta para novas atualizações",
"Today": "Hoje",
- "Today at": "",
+ "Today at": "Hoje em",
"Today at {{LOCALIZED_TIME}}": "Hoje às {{LOCALIZED_TIME}}",
"Toggle {{COUNT}} sources": "Alternar {{COUNT}} origens",
"Toggle 1 source": "Alternar 1 origem",
@@ -2049,6 +2059,7 @@
"TTS Model": "Modelo TTS",
"TTS Settings": "Configurações TTS",
"TTS Voice": "Voz TTS",
+ "Tu_day_of_week": "",
"Type": "Tipo",
"Type here...": "Digite aqui...",
"Type Hugging Face Resolve (Download) URL": "Digite o URL de download do Hugging Face",
@@ -2148,9 +2159,10 @@
"Waiting for upload...": "Aguardando upload...",
"Warning": "Aviso",
"Warning:": "Aviso:",
- "Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
+ "Warning: Enabling this will allow users to run scheduled prompts automatically.": "Aviso: Habilitar esta opção permitirá que os usuários executem solicitações agendadas automaticamente.",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Aviso: Habilitar isso permitirá que os usuários façam upload de código arbitrário no servidor.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Aviso: a execução do Jupyter permite a execução de código arbitrário, o que representa sérios riscos de segurança. Prossiga com extremo cuidado.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "Motor de carregamento da Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "A WebUI fará requisições para \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "A WebUI fará requisições para \"{{url}}/api/chat\".",
"WebUI will make requests to \"{{url}}/chat/completions\"": "A WebUI fará requisições para \"{{url}}/chat/completions\".",
+ "Weekly": "",
"What are you trying to achieve?": "O que está tentando alcançar?",
"What are you working on?": "No que está trabalhando?",
"What is NOT shared:": "O que NÃO é compartilhado:",
@@ -2183,7 +2196,7 @@
"Width": "Largura",
"Wikipedia": "Wikipédia",
"Won": "Ganhou",
- "Working Directory": "",
+ "Working Directory": "Diretório de Trabalho",
"Works together with top-k. A higher value (e.g., 0.95) will lead to more diverse text, while a lower value (e.g., 0.5) will generate more focused and conservative text.": "Funciona em conjunto com o top-k. Um valor mais alto (por exemplo, 0,95) resultará em um texto mais diverso, enquanto um valor mais baixo (por exemplo, 0,5) gerará um texto mais focado e conservador.",
"Workspace": "Espaço de Trabalho",
"Workspace Permissions": "Permissões do espaço de trabalho",
diff --git a/src/lib/i18n/locales/pt-PT/translation.json b/src/lib/i18n/locales/pt-PT/translation.json
index 7861982f6c..76c97853a2 100644
--- a/src/lib/i18n/locales/pt-PT/translation.json
+++ b/src/lib/i18n/locales/pt-PT/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "Criar uma nova nota",
"Create Account": "Criar Conta",
"Create Admin Account": "Criar Conta de Administrador",
+ "Create and manage scheduled automations": "",
"Create Channel": "Criar Canal",
"Create Folder": "Criar Pasta",
"Create Image": "Criar Imagem",
@@ -480,6 +481,7 @@
"Custom Gender": "Género Personalizado",
"Custom Parameter Name": "Nome do Parâmetro Personalizado",
"Custom Parameter Value": "Valor do Parâmetro Personalizado",
+ "Daily": "",
"Daily Messages": "Mensagens Diárias",
"Danger Zone": "Zona de Perigo",
"Dark": "Escuro",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Encaminha o token de acesso OAuth do utilizador do sistema para autenticação",
"Forwards system user session credentials to authenticate": "Encaminha as credenciais de sessão do utilizador do sistema para autenticação",
+ "Fr_day_of_week": "",
"Full Context Mode": "Modo de Contexto Completo",
"Function": "Função",
"Function Calling": "Chamada de Função",
@@ -1044,6 +1047,7 @@
"History": "Histórico",
"Home": "Início",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "Mensagens Horárias",
"How can I help you today?": "Como posso ajudá-lo hoje?",
"How would you rate this response?": "Como você avaliaria esta resposta?",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Chave API do Mistral OCR necessária.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Modelo",
"Model '{{modelName}}' has been successfully downloaded.": "O modelo '{{modelName}}' foi descarregado com sucesso.",
"Model '{{modelTag}}' is already in queue for downloading.": "O modelo '{{modelTag}}' já está na fila para descarregar.",
- "Model {{modelId}} not found": "Modelo {{modelId}} não foi encontrado",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "O modelo {{modelName}} não é capaz de visão",
"Model {{name}} is now {{status}}": "Modelo {{name}} agora é {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "Partilha de Modelos",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Chave API de Pesquisa Mojeek",
+ "Monthly": "",
"More": "Mais",
"More Concise": "Mais Conciso",
"More options": "Mais opções",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "Chave da API Ollama Cloud",
"Ollama Version": "Versão do Ollama",
"On": "Ligado",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Apenas ativo quando a configuração \"Colar Texto Grande como Arquivo\" está ativada.",
"Only active when the chat input is in focus and an LLM is generating a response.": "Apenas ativo quando a entrada do chat está em foco e um LLM está gerando uma resposta.",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Personalização",
"Pin": "Fixar",
+ "Pin to Sidebar": "",
"Pinned": "Fixado",
"Pinned Messages": "Mensagens Fixadas",
"Pinned Models": "Modelos Fixados",
@@ -1675,6 +1682,7 @@
"Running": "A correr",
"Running...": "A correr...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Corre tarefas de vetorização simultaneamente para acelerar o processamento. Desative se os limites de taxa se tornarem um problema.",
+ "Sa_day_of_week": "",
"Save": "Guardar",
"Save & Create": "Guardar e Criar",
"Save & Update": "Guardar e Atualizar",
@@ -1891,6 +1899,7 @@
"STT Model": "Modelo STT",
"STT Settings": "Configurações STT",
"Stylized PDF Export": "Exportação de PDF Estilizado",
+ "Su_day_of_week": "",
"Submit question": "Enviar pergunta",
"Submit suggestion": "Enviar sugestão",
"Subtitle": "Legenda",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Divisor de Texto",
"Text-to-Speech": "Texto para Fala",
"Text-to-Speech Engine": "Motor de Texto para Fala",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Obrigado pelo seu feedback!",
"The Application Account DN you bind with for search": "A DN de Conta de Aplicação que você vincula para pesquisa",
"The base to search for users": "A base para pesquisar utilizadores",
@@ -2049,6 +2059,7 @@
"TTS Model": "Modelo TTS",
"TTS Settings": "Configurações TTS",
"TTS Voice": "Voz TTS",
+ "Tu_day_of_week": "",
"Type": "Tipo",
"Type here...": "Escreva aqui...",
"Type Hugging Face Resolve (Download) URL": "Escreva o URL do Hugging Face Resolve (Descarregar)",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Aviso: Ativar isso permitirá que os utilizadores carreguem código arbitrário no servidor.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Aviso: A execução do Jupyter permite a execução de código arbitrário, representando riscos de segurança graves - prossiga com extrema cautela.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "Motor de Carregamento Web",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "O WebUI fará solicitações para \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "O WebUI fará solicitações para \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "O WebUI fará solicitações para \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "O que você está tentando alcançar?",
"What are you working on?": "Em que você está trabalhando?",
"What is NOT shared:": "O que NÃO é compartilhado:",
diff --git a/src/lib/i18n/locales/ro-RO/translation.json b/src/lib/i18n/locales/ro-RO/translation.json
index 49e4e7dba2..83ffb71fb9 100644
--- a/src/lib/i18n/locales/ro-RO/translation.json
+++ b/src/lib/i18n/locales/ro-RO/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "Creează Cont",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "Creează canal",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Întunecat",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "Funcție",
"Function Calling": "",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Cum te pot ajuta astăzi?",
"How would you rate this response?": "",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Modelul '{{modelName}}' a fost descărcat cu succes.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modelul '{{modelTag}}' este deja în coada de descărcare.",
- "Model {{modelId}} not found": "Modelul {{modelId}} nu a fost găsit",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Modelul {{modelName}} nu are capacități de viziune",
"Model {{name}} is now {{status}}": "Modelul {{name}} este acum {{status}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Mai multe",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Versiune Ollama",
"On": "Activat",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Personalizare",
"Pin": "Fixează",
+ "Pin to Sidebar": "",
"Pinned": "Fixat",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "Rulează",
"Running...": "Rulează...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Salvează",
"Save & Create": "Salvează & Creează",
"Save & Update": "Salvează & Actualizează",
@@ -1891,6 +1899,7 @@
"STT Model": "Model STT",
"STT Settings": "Setări STT",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Divizor de Text",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Motor de Conversie a Textului în Vorbire",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Mulțumim pentru feedback!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2049,6 +2059,7 @@
"TTS Model": "Model TTS",
"TTS Settings": "Setări TTS",
"TTS Voice": "Voce TTS",
+ "Tu_day_of_week": "",
"Type": "Tip",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Introduceți URL-ul de Rezolvare (Descărcare) Hugging Face",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "API Web",
"Web Loader Engine": "",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ru-RU/translation.json b/src/lib/i18n/locales/ru-RU/translation.json
index ebb4e76654..6c62e7a09d 100644
--- a/src/lib/i18n/locales/ru-RU/translation.json
+++ b/src/lib/i18n/locales/ru-RU/translation.json
@@ -456,6 +456,7 @@
"Create a new note": "Создать новую заметку",
"Create Account": "Создать аккаунт",
"Create Admin Account": "Создать аккаунт Администратора",
+ "Create and manage scheduled automations": "",
"Create Channel": "Создать канал",
"Create Folder": "Создать папку",
"Create Image": "Создать изображение",
@@ -481,6 +482,7 @@
"Custom Gender": "Другой пол",
"Custom Parameter Name": "Название пользовательского параметра",
"Custom Parameter Value": "Значение пользовательского параметра",
+ "Daily": "",
"Daily Messages": "Сообщений в день",
"Danger Zone": "Опасная зона",
"Dark": "Темная",
@@ -971,6 +973,7 @@
"Forward": "Вперёд",
"Forwards system user OAuth access token to authenticate": "Передаёт токен доступа OAuth системного пользователя для аутентификации",
"Forwards system user session credentials to authenticate": "Перенаправляет учетные данные сеанса системного пользователя для проверки подлинности",
+ "Fr_day_of_week": "",
"Full Context Mode": "Режим полного контекста",
"Function": "Функция",
"Function Calling": "Вызов функции",
@@ -1045,6 +1048,7 @@
"History": "История",
"Home": "Домой",
"Host": "Хост",
+ "Hourly": "",
"Hourly Messages": "Сообщений в час",
"How can I help you today?": "Чем я могу помочь вам сегодня?",
"How would you rate this response?": "Как бы вы оценили этот ответ?",
@@ -1267,10 +1271,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Требуется API ключ Mistral OCR.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "Модель",
"Model '{{modelName}}' has been successfully downloaded.": "Модель '{{modelName}}' успешно загружена.",
"Model '{{modelTag}}' is already in queue for downloading.": "Модель '{{modelTag}}' уже находится в очереди на загрузку.",
- "Model {{modelId}} not found": "Модель {{modelId}} не найдена",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Модель {{modelName}} не поддерживает зрение",
"Model {{name}} is now {{status}}": "Модель {{name}} теперь {{status}}",
@@ -1311,6 +1315,7 @@
"Models Sharing": "Общий доступ к моделям",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Ключ API для поиска Mojeek",
+ "Monthly": "",
"More": "Больше",
"More Concise": "Более кратко",
"More options": "Больше опций",
@@ -1431,6 +1436,7 @@
"Ollama Cloud API Key": "API-ключ Ollama Cloud",
"Ollama Version": "Версия Ollama",
"On": "Включено",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "Активно только при включённой настройке «Вставлять большой текст как файл».",
"Only active when the chat input is in focus and an LLM is generating a response.": "Активно только при фокусе на поле ввода и генерации ответа.",
@@ -1515,6 +1521,7 @@
"Persistent": "Постоянный",
"Personalization": "Персонализация",
"Pin": "Закрепить",
+ "Pin to Sidebar": "",
"Pinned": "Закреплено",
"Pinned Messages": "Закреплённые сообщения",
"Pinned Models": "Закреплённые модели",
@@ -1677,6 +1684,7 @@
"Running": "Выполняется",
"Running...": "Выполняется...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "Параллельное выполнение задач эмбеддингов для ускорения. Отключите при проблемах с лимитами запросов.",
+ "Sa_day_of_week": "",
"Save": "Сохранить",
"Save & Create": "Сохранить и создать",
"Save & Update": "Сохранить и обновить",
@@ -1893,6 +1901,7 @@
"STT Model": "Модель распознавания речи",
"STT Settings": "Настройки распознавания речи",
"Stylized PDF Export": "Стилизованный экспорт в формате PDF",
+ "Su_day_of_week": "",
"Submit question": "Отправить вопрос",
"Submit suggestion": "Отправить предложение",
"Subtitle": "Подзаголовок",
@@ -1938,6 +1947,7 @@
"Text Splitter": "Разделитель текста",
"Text-to-Speech": "Текст в речь",
"Text-to-Speech Engine": "Система синтеза речи",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Спасибо за вашу обратную связь!",
"The Application Account DN you bind with for search": "Логин учетной записи приложения, к которому вы привязываетесь для поиска",
"The base to search for users": "База для поиска пользователей",
@@ -2051,6 +2061,7 @@
"TTS Model": "Модель TTS",
"TTS Settings": "Настройки TTS",
"TTS Voice": "Голос TTS",
+ "Tu_day_of_week": "",
"Type": "Тип",
"Type here...": "Введите текст...",
"Type Hugging Face Resolve (Download) URL": "Введите URL-адрес Hugging Face Resolve (загрузки)",
@@ -2153,6 +2164,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Предупреждение. Включение этого параметра позволит пользователям загружать произвольный код на сервер.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Предупреждение: Выполнение Jupyter позволяет выполнять произвольный код, что создает серьезные угрозы безопасности — действуйте с особой осторожностью.",
+ "We_day_of_week": "",
"Web": "Веб",
"Web API": "Веб API",
"Web Loader Engine": "Движок веб-загрузчика",
@@ -2169,6 +2181,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI будет отправлять запросы к \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI будет отправлять запросы к \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI будет отправлять запросы к \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Чего вы пытаетесь достичь?",
"What are you working on?": "Над чем вы работаете?",
"What is NOT shared:": "Что НЕ передаётся:",
diff --git a/src/lib/i18n/locales/sk-SK/translation.json b/src/lib/i18n/locales/sk-SK/translation.json
index a828c7fb3b..cfb4343e49 100644
--- a/src/lib/i18n/locales/sk-SK/translation.json
+++ b/src/lib/i18n/locales/sk-SK/translation.json
@@ -456,6 +456,7 @@
"Create a new note": "Vytvoriť novú poznámku",
"Create Account": "Vytvoriť účet",
"Create Admin Account": "Vytvoriť admin účet",
+ "Create and manage scheduled automations": "",
"Create Channel": "Vytvoriť kanál",
"Create Folder": "Vytvoriť priečinok",
"Create Image": "Vytvoriť obrázok",
@@ -481,6 +482,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Nebezpečná zóna",
"Dark": "Tmavý",
@@ -971,6 +973,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "Funkcia",
"Function Calling": "",
@@ -1045,6 +1048,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Ako vám môžem dnes pomôcť?",
"How would you rate this response?": "",
@@ -1267,10 +1271,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "Model „{{modelName}}“ bol úspešne stiahnutý.",
"Model '{{modelTag}}' is already in queue for downloading.": "Model '{{modelTag}}' je už zaradený do fronty na sťahovanie.",
- "Model {{modelId}} not found": "Model {{modelId}} nebol nájdený",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} nie je schopný spracovávať vizuálne údaje.",
"Model {{name}} is now {{status}}": "Model {{name}} je teraz {{status}}.",
@@ -1311,6 +1315,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Viac",
"More Concise": "",
"More options": "",
@@ -1431,6 +1436,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Verzia Ollama",
"On": "Zapnuté",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1515,6 +1521,7 @@
"Persistent": "",
"Personalization": "Personalizácia",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1677,6 +1684,7 @@
"Running": "Spúšťanie",
"Running...": "Spúšťanie...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Uložiť",
"Save & Create": "Uložiť a Vytvoriť",
"Save & Update": "Uložiť a aktualizovať",
@@ -1893,6 +1901,7 @@
"STT Model": "Model rozpoznávania reči na text (STT)",
"STT Settings": "Nastavenia STT (Rozpoznávanie reči)",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1938,6 +1947,7 @@
"Text Splitter": "Rozdeľovač textu",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Stroj na prevod textu na reč",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Ďakujeme za vašu spätnú väzbu!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2051,6 +2061,7 @@
"TTS Model": "Model prevodu textu na reč (TTS)",
"TTS Settings": "Nastavenia TTS (Text-to-Speech)",
"TTS Voice": "TTS hlas",
+ "Tu_day_of_week": "",
"Type": "Napíšte",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Zadajte URL na úspešné stiahnutie z Hugging Face.",
@@ -2153,6 +2164,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Webové API",
"Web Loader Engine": "",
@@ -2169,6 +2181,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "Čo sa snažíte dosiahnuť?",
"What are you working on?": "Na čom pracujete?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/sr-RS/translation.json b/src/lib/i18n/locales/sr-RS/translation.json
index ddfe1d46fc..7e1b205850 100644
--- a/src/lib/i18n/locales/sr-RS/translation.json
+++ b/src/lib/i18n/locales/sr-RS/translation.json
@@ -455,6 +455,7 @@
"Create a new note": "",
"Create Account": "Направи налог",
"Create Admin Account": "Направи админ налог",
+ "Create and manage scheduled automations": "",
"Create Channel": "Направи канал",
"Create Folder": "",
"Create Image": "",
@@ -480,6 +481,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Тамна",
@@ -970,6 +972,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1044,6 +1047,7 @@
"History": "",
"Home": "",
"Host": "Домаћин",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Како могу да вам помогнем данас?",
"How would you rate this response?": "",
@@ -1266,10 +1270,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Модел",
"Model '{{modelName}}' has been successfully downloaded.": "Модел „{{modelName}}“ је успешно преузет.",
"Model '{{modelTag}}' is already in queue for downloading.": "Модел „{{modelTag}}“ је већ у реду за преузимање.",
- "Model {{modelId}} not found": "Модел {{modelId}} није пронађен",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Модел {{моделНаме}} није способан за вид",
"Model {{name}} is now {{status}}": "Модел {{наме}} је сада {{статус}}",
@@ -1310,6 +1314,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Више",
"More Concise": "",
"More options": "",
@@ -1430,6 +1435,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Издање Ollama-е",
"On": "Укључено",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1514,6 +1520,7 @@
"Persistent": "",
"Personalization": "Прилагођавање",
"Pin": "Закачи",
+ "Pin to Sidebar": "",
"Pinned": "Закачено",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1675,6 +1682,7 @@
"Running": "Покрећем",
"Running...": "Покрећем...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Сачувај",
"Save & Create": "Сачувај и направи",
"Save & Update": "Сачувај и ажурирај",
@@ -1891,6 +1899,7 @@
"STT Model": "STT модел",
"STT Settings": "STT подешавања",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1936,6 +1945,7 @@
"Text Splitter": "Раздвајач текста",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Мотор за текст у говор",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Хвала на вашем коментару!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2049,6 +2059,7 @@
"TTS Model": "TTS модел",
"TTS Settings": "TTS подешавања",
"TTS Voice": "TTS глас",
+ "Tu_day_of_week": "",
"Type": "Тип",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Унесите Hugging Face Resolve (Download) адресу",
@@ -2151,6 +2162,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Веб",
"Web API": "Веб АПИ",
"Web Loader Engine": "",
@@ -2167,6 +2179,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/sv-SE/translation.json b/src/lib/i18n/locales/sv-SE/translation.json
index d8b0674e34..e097438fb6 100644
--- a/src/lib/i18n/locales/sv-SE/translation.json
+++ b/src/lib/i18n/locales/sv-SE/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Skapa en ny anteckning",
"Create Account": "Skapa konto",
"Create Admin Account": "Skapa administratörskonto",
+ "Create and manage scheduled automations": "",
"Create Channel": "Skapa kanal",
"Create Folder": "Skapa mapp",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Anpassat parameternamn",
"Custom Parameter Value": "Anpassat parametervärde",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Fara",
"Dark": "Mörk",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Vidarebefordrar systemanvändarsessionens autentiseringsuppgifter för att autentisera",
+ "Fr_day_of_week": "",
"Full Context Mode": "Fullständigt kontextläge",
"Function": "Funktion",
"Function Calling": "Funktionsanrop",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Hem",
"Host": "Värd",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Hur kan jag hjälpa dig idag?",
"How would you rate this response?": "Hur skulle du betygsätta detta svar?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API-nyckel krävs.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Modell",
"Model '{{modelName}}' has been successfully downloaded.": "Modellen '{{modelName}}' har laddats ner.",
"Model '{{modelTag}}' is already in queue for downloading.": "Modellen '{{modelTag}}' är redan i kö för nedladdning.",
- "Model {{modelId}} not found": "Modell {{modelId}} hittades inte",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Modellen {{modelName}} har inte stöd för vision/syn",
"Model {{name}} is now {{status}}": "Modellen {{name}} är nu {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Sök API-nyckel",
+ "Monthly": "",
"More": "Mer",
"More Concise": "Mer kortfattat",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama-version",
"On": "På",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Personalisering",
"Pin": "Fäst",
+ "Pin to Sidebar": "",
"Pinned": "Fäst",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Kör",
"Running...": "Kör...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Spara",
"Save & Create": "Spara och skapa",
"Save & Update": "Spara och uppdatera",
@@ -1889,6 +1897,7 @@
"STT Model": "Tal-till-text-modell",
"STT Settings": "Tal-till-text-inställningar",
"Stylized PDF Export": "Stiliserad PDF-export",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Textdelare",
"Text-to-Speech": "Text-till-tal",
"Text-to-Speech Engine": "Text-till-tal-motor",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Tack för din feedback!",
"The Application Account DN you bind with for search": "Applikationskontots DN du binder med för sökning",
"The base to search for users": "Basen för att söka efter användare",
@@ -2047,6 +2057,7 @@
"TTS Model": "Text-till-tal-modell",
"TTS Settings": "Text-till-tal-inställningar",
"TTS Voice": "Text-till-tal-röst",
+ "Tu_day_of_week": "",
"Type": "Typ",
"Type here...": "Skriv här...",
"Type Hugging Face Resolve (Download) URL": "Ange Hugging Face Resolve (Download) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Varning för detta: Om du aktiverar detta kan användare ladda upp godtycklig kod på servern.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Varning: Jupyter-exekvering möjliggör godtycklig kodkörning, vilket innebär allvarliga säkerhetsrisker - fortsätt med extrem försiktighet",
+ "We_day_of_week": "",
"Web": "Webb",
"Web API": "Webb-API",
"Web Loader Engine": "Webbladdarmotor",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI kommer att göra förfrågningar till \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI kommer att göra förfrågningar till \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI kommer att göra förfrågningar till \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Vad försöker du uppnå?",
"What are you working on?": "Var arbetar du med?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ta-IN/translation.json b/src/lib/i18n/locales/ta-IN/translation.json
index d4b413514a..4f773125d6 100644
--- a/src/lib/i18n/locales/ta-IN/translation.json
+++ b/src/lib/i18n/locales/ta-IN/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "புதிய குறிப்பை உருவாக்கவும்",
"Create Account": "கணக்கை உருவாக்கவும்",
"Create Admin Account": "நிர்வாகி கணக்கை உருவாக்கவும்",
+ "Create and manage scheduled automations": "",
"Create Channel": "சேனலை உருவாக்கவும்",
"Create Folder": "கோப்புறையை உருவாக்கவும்",
"Create Image": "படத்தை உருவாக்கவும்",
@@ -479,6 +480,7 @@
"Custom Gender": "தனிப்பயன் பாலினம்",
"Custom Parameter Name": "தனிப்பயன் அளவுரு பெயர்",
"Custom Parameter Value": "தனிப்பயன் அளவுரு மதிப்பு",
+ "Daily": "",
"Daily Messages": "தினசரி செய்திகள்",
"Danger Zone": "ஆபத்து மண்டலம்",
"Dark": "இருண்ட",
@@ -969,6 +971,7 @@
"Forward": "முன்னோக்கி",
"Forwards system user OAuth access token to authenticate": "அங்கீகரிக்க கணினி பயனர் OAuth அணுகல் டோக்கனை அனுப்புகிறது",
"Forwards system user session credentials to authenticate": "அங்கீகரிக்க கணினி பயனர் அமர்வு நற்சான்றிதழ்களை அனுப்புகிறது",
+ "Fr_day_of_week": "",
"Full Context Mode": "முழு சூழல் பயன்முறை",
"Function": "செயல்பாடு",
"Function Calling": "செயல்பாடு அழைப்பு",
@@ -1043,6 +1046,7 @@
"History": "வரலாறு",
"Home": "வீடு",
"Host": "புரவலன்",
+ "Hourly": "",
"Hourly Messages": "மணிநேர செய்திகள்",
"How can I help you today?": "இன்று நான் உங்களுக்கு எப்படி உதவலாம்?",
"How would you rate this response?": "இந்த பதிலை எப்படி மதிப்பிடுவீர்கள்?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "மிஸ்ட்ரல் OCR",
"Mistral OCR API Key required.": "மிஸ்ட்ரல் OCR API விசை தேவை.",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "மாதிரி",
"Model '{{modelName}}' has been successfully downloaded.": "மாடல் '{{modelName}}' வெற்றிகரமாகப் பதிவிறக்கப்பட்டது.",
"Model '{{modelTag}}' is already in queue for downloading.": "மாடல் '{{modelTag}}' ஏற்கனவே பதிவிறக்குவதற்கு வரிசையில் உள்ளது.",
- "Model {{modelId}} not found": "மாதிரி {{modelId}} கிடைக்கவில்லை",
"Model {{modelName}} deleted successfully": "மாதிரி {{modelName}} வெற்றிகரமாக நீக்கப்பட்டது",
"Model {{modelName}} is not vision capable": "மாதிரி {{modelName}} பார்வை திறன் இல்லை",
"Model {{name}} is now {{status}}": "மாடல் {{name}} இப்போது {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "மாதிரிகள் பகிர்வு",
"Mojeek": "மொஜீக்",
"Mojeek Search API Key": "Mojeek தேடல் API விசை",
+ "Monthly": "",
"More": "மேலும்",
"More Concise": "மேலும் சுருக்கமானது",
"More options": "மேலும் விருப்பங்கள்",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "Ollama கிளவுட் API விசை",
"Ollama Version": "Ollama பதிப்பு",
"On": "அன்று",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "\"பெரிய உரையை கோப்பாக ஒட்டவும்\" அமைப்பு மாறியிருந்தால் மட்டுமே செயலில் இருக்கும்.",
"Only active when the chat input is in focus and an LLM is generating a response.": "அரட்டை உள்ளீடு ஃபோகஸ் மற்றும் LLM பதிலை உருவாக்கும் போது மட்டுமே செயலில் இருக்கும்.",
@@ -1513,6 +1519,7 @@
"Persistent": "பிடிவாதமான",
"Personalization": "தனிப்பயனாக்கம்",
"Pin": "பின்",
+ "Pin to Sidebar": "",
"Pinned": "நிலைநிறுத்தப்பட்டவை",
"Pinned Messages": "பின் செய்யப்பட்ட செய்திகள்",
"Pinned Models": "பின் செய்யப்பட்ட மாதிரிகள்",
@@ -1673,6 +1680,7 @@
"Running": "ஓடுகிறது",
"Running...": "இயங்குகிறது...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "செயலாக்கத்தை விரைவுபடுத்த ஒரே நேரத்தில் உட்பொதிக்கும் பணிகளை இயக்குகிறது. கட்டண வரம்புகள் சிக்கலாக இருந்தால் அணைக்கவும்.",
+ "Sa_day_of_week": "",
"Save": "சேமி",
"Save & Create": "சேமித்து உருவாக்கவும்",
"Save & Update": "சேமி & புதுப்பிக்கவும்",
@@ -1889,6 +1897,7 @@
"STT Model": "STT மாதிரி",
"STT Settings": "STT அமைப்புகள்",
"Stylized PDF Export": "பகட்டான PDF ஏற்றுமதி",
+ "Su_day_of_week": "",
"Submit question": "கேள்வியை சமர்ப்பிக்கவும்",
"Submit suggestion": "பரிந்துரையைச் சமர்ப்பிக்கவும்",
"Subtitle": "வசனம்",
@@ -1934,6 +1943,7 @@
"Text Splitter": "உரை பிரிப்பான்",
"Text-to-Speech": "உரையிலிருந்து பேச்சு",
"Text-to-Speech Engine": "உரையிலிருந்து பேச்சு இயந்திரம்",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "உங்கள் கருத்துக்கு நன்றி!",
"The Application Account DN you bind with for search": "தேடலுக்காக நீங்கள் இணைக்கும் பயன்பாட்டுக் கணக்கு DN",
"The base to search for users": "பயனர்களைத் தேடுவதற்கான அடிப்படை",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS மாதிரி",
"TTS Settings": "TTS அமைப்புகள்",
"TTS Voice": "TTS குரல்",
+ "Tu_day_of_week": "",
"Type": "வகை",
"Type here...": "இங்கே தட்டச்சு செய்யவும்...",
"Type Hugging Face Resolve (Download) URL": "கட்டிப்பிடிக்கும் முகம் தீர்வு (பதிவிறக்கம்) URL",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "எச்சரிக்கை: இதை இயக்கினால் பயனர்கள் திட்டமிட்ட வினாக்களை தானாக இயக்க அனுமதிக்கும்.",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "எச்சரிக்கை: இதை இயக்குவது பயனர்கள் சர்வரில் தன்னிச்சையான குறியீட்டைப் பதிவேற்ற அனுமதிக்கும்.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "எச்சரிக்கை: வியாழன் இயக்கமானது தன்னிச்சையான குறியீடு செயல்படுத்தலை செயல்படுத்துகிறது, இது கடுமையான பாதுகாப்பு அபாயங்களை ஏற்படுத்துகிறது-அதிக எச்சரிக்கையுடன் தொடரவும்.",
+ "We_day_of_week": "",
"Web": "வலை",
"Web API": "வலை API",
"Web Loader Engine": "வலை ஏற்றி இயந்திரம்",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI \"{{url}}\" க்கு கோரிக்கைகளை வைக்கும்",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI \"{{url}}/api/chat\" க்கு கோரிக்கைகளை வைக்கும்",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI \"{{url}}/chat/completions\" க்கு கோரிக்கைகளை வைக்கும்",
+ "Weekly": "",
"What are you trying to achieve?": "நீங்கள் எதை அடைய முயற்சிக்கிறீர்கள்?",
"What are you working on?": "நீங்கள் என்ன வேலை செய்கிறீர்கள்?",
"What is NOT shared:": "NOT என்பது என்ன பகிரப்பட்டது:",
diff --git a/src/lib/i18n/locales/th-TH/translation.json b/src/lib/i18n/locales/th-TH/translation.json
index 3282b044fe..d337319a8b 100644
--- a/src/lib/i18n/locales/th-TH/translation.json
+++ b/src/lib/i18n/locales/th-TH/translation.json
@@ -453,6 +453,7 @@
"Create a new note": "สร้างบันทึกใหม่",
"Create Account": "สร้างบัญชี",
"Create Admin Account": "สร้างบัญชีผู้ดูแลระบบ",
+ "Create and manage scheduled automations": "",
"Create Channel": "สร้างช่องทาง",
"Create Folder": "สร้างโฟลเดอร์",
"Create Image": "สร้างรูปภาพ",
@@ -478,6 +479,7 @@
"Custom Gender": "",
"Custom Parameter Name": "ชื่อพารามิเตอร์แบบกำหนดเอง",
"Custom Parameter Value": "ค่าพารามิเตอร์กำหนดเอง",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "เขตอันตราย",
"Dark": "มืด",
@@ -968,6 +970,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "ส่งต่อโทเค็นการเข้าถึง OAuth ของผู้ใช้ระบบเพื่อยืนยันตัวตน",
"Forwards system user session credentials to authenticate": "ส่งต่อข้อมูลรับรอง Session ของผู้ใช้ระบบเพื่อใช้ยืนยันตัวตน",
+ "Fr_day_of_week": "",
"Full Context Mode": "โหมดบริบทเต็ม",
"Function": "ฟังก์ชัน",
"Function Calling": "การเรียกใช้ฟังก์ชัน",
@@ -1042,6 +1045,7 @@
"History": "",
"Home": "หน้าแรก",
"Host": "โฮสต์",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "วันนี้ฉันจะช่วยคุณได้อย่างไร?",
"How would you rate this response?": "คุณจะให้คะแนนคำตอบนี้อย่างไร?",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "จำเป็นต้องมี Mistral OCR API Key",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "โมเดล",
"Model '{{modelName}}' has been successfully downloaded.": "โมเดล '{{modelName}}' ถูกดาวน์โหลดเรียบร้อยแล้ว",
"Model '{{modelTag}}' is already in queue for downloading.": "โมเดล '{{modelTag}}' อยู่ในคิวสำหรับการดาวน์โหลดแล้ว",
- "Model {{modelId}} not found": "ไม่พบโมเดล {{modelId}}",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "โมเดล {{modelName}} ไม่รองรับฟีเจอร์ Vision",
"Model {{name}} is now {{status}}": "โมเดล {{name}} ขณะนี้ {{status}}",
@@ -1308,6 +1312,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "API Key สำหรับ Mojeek Search",
+ "Monthly": "",
"More": "เพิ่มเติม",
"More Concise": "กระชับมากขึ้น",
"More options": "",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "API Key ของ Ollama Cloud",
"Ollama Version": "เวอร์ชัน Ollama",
"On": "เปิด",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "จะทำงานเฉพาะเมื่อเปิดการตั้งค่า \"วางข้อความขนาดใหญ่เป็นไฟล์\"",
"Only active when the chat input is in focus and an LLM is generating a response.": "ทำงานเฉพาะเมื่อช่องป้อนข้อความแชทถูกโฟกัสอยู่ และ LLM กำลังสร้างคำตอบ",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "การปรับแต่ง",
"Pin": "ปักหมุด",
+ "Pin to Sidebar": "",
"Pinned": "ปักหมุดแล้ว",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1671,6 +1678,7 @@
"Running": "กำลังทำงาน",
"Running...": "กำลังทำงาน...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "บันทึก",
"Save & Create": "บันทึกและสร้าง",
"Save & Update": "บันทึกและอัปเดต",
@@ -1887,6 +1895,7 @@
"STT Model": "โมเดลแปลงเสียงเป็นข้อความ",
"STT Settings": "การตั้งค่าแปลงเสียงเป็นข้อความ",
"Stylized PDF Export": "ส่งออก PDF แบบมีสไตล์",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1932,6 +1941,7 @@
"Text Splitter": "ตัวแบ่งข้อความ",
"Text-to-Speech": "แปลงข้อความเป็นเสียง",
"Text-to-Speech Engine": "เครื่องมือแปลงข้อความเป็นเสียง",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "ขอบคุณสำหรับคำติชมของคุณ!",
"The Application Account DN you bind with for search": "DN ของบัญชีแอปพลิเคชันที่คุณใช้ bind เพื่อค้นหา",
"The base to search for users": "ฐานสำหรับค้นหาผู้ใช้",
@@ -2045,6 +2055,7 @@
"TTS Model": "โมเดล TTS",
"TTS Settings": "การตั้งค่า TTS",
"TTS Voice": "เสียง TTS",
+ "Tu_day_of_week": "",
"Type": "ประเภท",
"Type here...": "พิมพ์ที่นี่...",
"Type Hugging Face Resolve (Download) URL": "พิมพ์ URL Hugging Face Resolve (ดาวน์โหลด)",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "คำเตือน: การเปิดใช้งานตัวเลือกนี้จะอนุญาตให้ผู้ใช้อัปโหลดโค้ดใดๆ บนเซิร์ฟเวอร์",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "คำเตือน: การรัน Jupyter อนุญาตให้รันโค้ดใดๆ ก็ได้ ซึ่งก่อให้เกิดความเสี่ยงด้านความปลอดภัยอย่างร้ายแรง—โปรดดำเนินการด้วยความระมัดระวังอย่างยิ่ง",
+ "We_day_of_week": "",
"Web": "เว็บ",
"Web API": "เว็บ API",
"Web Loader Engine": "เอนจินโหลดเว็บ",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI จะส่งคำขอไปยัง \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI จะส่งคำขอไปที่ \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI จะส่งคำขอไปยัง \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "คุณพยายามจะทำอะไร?",
"What are you working on?": "คุณกำลังทำอะไรอยู่?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/tk-TM/translation.json b/src/lib/i18n/locales/tk-TM/translation.json
index a39d4984cc..cb777190b4 100644
--- a/src/lib/i18n/locales/tk-TM/translation.json
+++ b/src/lib/i18n/locales/tk-TM/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Hasap döret",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "Garaňky",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Baş Sahypa",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "",
"Model '{{modelTag}}' is already in queue for downloading.": "",
- "Model {{modelId}} not found": "",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "",
"Model {{name}} is now {{status}}": "",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "Has köp",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "",
"On": "Işjeň",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "",
"Pin": "",
+ "Pin to Sidebar": "",
"Pinned": "",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "",
"Running...": "Işleýär...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Sakla",
"Save & Create": "",
"Save & Update": "",
@@ -1889,6 +1897,7 @@
"STT Model": "",
"STT Settings": "",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "",
"Text-to-Speech": "",
"Text-to-Speech Engine": "",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "",
"TTS Settings": "",
"TTS Voice": "",
+ "Tu_day_of_week": "",
"Type": "Typ",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "",
"Web API": "",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/tr-TR/translation.json b/src/lib/i18n/locales/tr-TR/translation.json
index 68f5afaba4..62350354d5 100644
--- a/src/lib/i18n/locales/tr-TR/translation.json
+++ b/src/lib/i18n/locales/tr-TR/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "Yeni bir not oluştur",
"Create Account": "Hesap Oluştur",
"Create Admin Account": "Yönetici Hesabı Oluştur",
+ "Create and manage scheduled automations": "",
"Create Channel": "Kanal Oluştur",
"Create Folder": "Klasör Oluştur",
"Create Image": "Görsel Oluştur",
@@ -479,6 +480,7 @@
"Custom Gender": "Özel Cinsiyet",
"Custom Parameter Name": "Özel Parametre Adı",
"Custom Parameter Value": "Özel Parametre Değeri",
+ "Daily": "",
"Daily Messages": "Günlük Mesajlar",
"Danger Zone": "Tehlikeli Bölge",
"Dark": "Koyu",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "Kimlik doğrulamak için sistem kullanıcı OAuth erişim belirtecini iletir",
"Forwards system user session credentials to authenticate": "Kimlik doğrulamak için sistem kullanıcı oturum kimlik bilgilerini iletir",
+ "Fr_day_of_week": "",
"Full Context Mode": "Tam Bağlam Modu",
"Function": "Fonksiyon",
"Function Calling": "Fonksiyon Çağırma",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Ana Sayfa",
"Host": "Ana bilgisayar",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Bugün size nasıl yardımcı olabilirim?",
"How would you rate this response?": "Bu yanıtı nasıl değerlendirirsiniz?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API Anahtarı Gerekli",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "'{{modelName}}' başarıyla indirildi.",
"Model '{{modelTag}}' is already in queue for downloading.": "'{{modelTag}}' zaten indirme sırasında.",
- "Model {{modelId}} not found": "{{modelId}} bulunamadı",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} görüntü yeteneğine sahip değil",
"Model {{name}} is now {{status}}": "{{name}} modeli artık {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "Model Paylaşımı",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek Search API Anahtarı",
+ "Monthly": "",
"More": "Daha Fazla",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama Sürümü",
"On": "Açık",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Kişiselleştirme",
"Pin": "Sabitle",
+ "Pin to Sidebar": "",
"Pinned": "Sabitlenmiş",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Çalışıyor",
"Running...": "Çalışıyor...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Kaydet",
"Save & Create": "Kaydet ve Oluştur",
"Save & Update": "Kaydet ve Güncelle",
@@ -1889,6 +1897,7 @@
"STT Model": "STT Modeli",
"STT Settings": "STT Ayarları",
"Stylized PDF Export": "Biçimlendirilmiş PDF Dışa Aktarımı",
+ "Su_day_of_week": "",
"Submit question": "Soru gönder",
"Submit suggestion": "Öneri gönder",
"Subtitle": "Altyazı",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Metin Bölücü",
"Text-to-Speech": "Metinden Sese",
"Text-to-Speech Engine": "Metinden Sese Motoru",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Geri bildiriminiz için teşekkürler!",
"The Application Account DN you bind with for search": "Arama için bağlandığınız Uygulama Hesap DN'si",
"The base to search for users": "Kullanıcıları aramak için temel",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS Modeli",
"TTS Settings": "TTS Ayarları",
"TTS Voice": "TTS Sesi",
+ "Tu_day_of_week": "",
"Type": "Tür",
"Type here...": "Buraya yaz...",
"Type Hugging Face Resolve (Download) URL": "HuggingFace Çözümleme (İndirme) URL'sini Yazın",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Uyarı: Bu etkinleştirildiğinde, kullanıcıların sunucuya rastgele kod yüklemesine izin verilecektir.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI, \"{{url}}/api/chat\" adresine istek yapacak",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI, \"{{url}}/chat/completions\" adresine istek yapacak",
+ "Weekly": "",
"What are you trying to achieve?": "Ne yapmaya çalışıyorsunuz?",
"What are you working on?": "Üzerinde çalıştığınız nedir?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ug-CN/translation.json b/src/lib/i18n/locales/ug-CN/translation.json
index 5301df3b2e..b7fee9fb65 100644
--- a/src/lib/i18n/locales/ug-CN/translation.json
+++ b/src/lib/i18n/locales/ug-CN/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "ھېساب قۇرۇش",
"Create Admin Account": "باشقۇرغۇچى ھېساباتى قۇرۇش",
+ "Create and manage scheduled automations": "",
"Create Channel": "قانال قۇرۇش",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "ئۆزلۈك پارامېتىر نامى",
"Custom Parameter Value": "ئۆزلۈك پارامېتىر قىممىتى",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "خەۋپلىك رايون",
"Dark": "قاراڭغۇ",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "سىستېما ئىشلەتكۈچىسى ئۇچۇرلىرىنى دەلىللەشكە يوللايدۇ",
+ "Fr_day_of_week": "",
"Full Context Mode": "تولۇق مەزمۇن ھالىتى",
"Function": "فۇنكسىيە",
"Function Calling": "فۇنكسىيە چاقىرىش",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "باش بەت",
"Host": "مۇلازىمېتىر",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "بۈگۈن سىزگە قانداق ياردەم بېرەي؟",
"How would you rate this response?": "بۇ ئىنكاسقا قانداق باھا بېرىسىز؟",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API ئاچقۇچى زۆرۈر.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "مودېل",
"Model '{{modelName}}' has been successfully downloaded.": "مودېل '{{modelName}}' مۇۋەپپەقىيەتلىك چۈشۈرۈلدى.",
"Model '{{modelTag}}' is already in queue for downloading.": "مودېل '{{modelTag}}' ئاللىقاچان چۈشۈرۈش قاتارىغا قوشۇلغان.",
- "Model {{modelId}} not found": "مودېل {{modelId}} تېپىلمىدى",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "مودېل {{modelName}} كۆرۈنۈش ئىقتىدارى يوق",
"Model {{name}} is now {{status}}": "مودېل {{name}} ھازىر {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek ئىزدەش API ئاچقۇچى",
+ "Monthly": "",
"More": "تېخىمۇ كۆپ",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama نەشرى",
"On": "قوزغىتىلغان",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "شەخسىيلاشتۇرۇش",
"Pin": "مۇقىملا",
+ "Pin to Sidebar": "",
"Pinned": "مۇقىملاندى",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "ئىجرا قىلىنىۋاتىدۇ",
"Running...": "ئىجرا قىلىنىۋاتىدۇ...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "ساقلاش",
"Save & Create": "ساقلا ۋە قۇر",
"Save & Update": "ساقلا ۋە يېڭىلا",
@@ -1889,6 +1897,7 @@
"STT Model": "STT مودېلى",
"STT Settings": "STT تەڭشەكلىرى",
"Stylized PDF Export": "ئۇسلۇبلۇق PDF چىقىرىش",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "تېكست بۆلگۈچ",
"Text-to-Speech": "تېكستتىن ئاۋازغا",
"Text-to-Speech Engine": "تېكست ئاۋاز ماتورى",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "پىكىرىڭىزگە رەھمەت!",
"The Application Account DN you bind with for search": "ئىزدەش ئۈچۈن باغلانغان قوللىنىشچان DN",
"The base to search for users": "ئىشلەتكۈچى ئىزدەش ئاساسى",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS مودېلى",
"TTS Settings": "TTS تەڭشەكلىرى",
"TTS Voice": "TTS ئاۋازى",
+ "Tu_day_of_week": "",
"Type": "تىپى",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face چۈشۈرۈش URL كىرگۈزۈڭ",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "ئاگاھلاندۇرۇش: بۇ قوزغىتىلسا، ئىشلەتكۈچىلەر خالىغان كودنى مۇلازىمېتىرغا چىقىرىدۇ.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "ئاگاھلاندۇرۇش: Jupyter ئىجرا قىلىش خالىغان كود ئىجرا قىلىشقا يول قويىدۇ، بىخەتەرلىك خەۋپى يۇقىرى — ئىنتايىن دىققەت قىلىڭ.",
+ "We_day_of_week": "",
"Web": "تور",
"Web API": "تور API",
"Web Loader Engine": "تور يۈكلىگۈچ ماتورى",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI \"{{url}}\" غا تەلەپ يوللايدۇ",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI \"{{url}}/api/chat\" غا تەلەپ يوللايدۇ",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI \"{{url}}/chat/completions\" غا تەلەپ يوللايدۇ",
+ "Weekly": "",
"What are you trying to achieve?": "نېمىگە ئېرىشمىكچىسىز؟",
"What are you working on?": "نېمە ئىش قىلىۋاتىسىز؟",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/uk-UA/translation.json b/src/lib/i18n/locales/uk-UA/translation.json
index 4c2adc74c3..67ab638362 100644
--- a/src/lib/i18n/locales/uk-UA/translation.json
+++ b/src/lib/i18n/locales/uk-UA/translation.json
@@ -456,6 +456,7 @@
"Create a new note": "",
"Create Account": "Створити обліковий запис",
"Create Admin Account": "Створити обліковий запис адміністратора",
+ "Create and manage scheduled automations": "",
"Create Channel": "Створити канал",
"Create Folder": "",
"Create Image": "",
@@ -481,6 +482,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Зона небезпеки",
"Dark": "Темна",
@@ -971,6 +973,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "Режим повного контексту",
"Function": "Функція",
"Function Calling": "Виклик функцій",
@@ -1045,6 +1048,7 @@
"History": "",
"Home": "Головна",
"Host": "Хост",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Чим я можу допомогти вам сьогодні?",
"How would you rate this response?": "Як би ви оцінили цю відповідь?",
@@ -1267,10 +1271,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Модель",
"Model '{{modelName}}' has been successfully downloaded.": "Модель '{{modelName}}' успішно завантажено.",
"Model '{{modelTag}}' is already in queue for downloading.": "Модель '{{modelTag}}' вже знаходиться в черзі на завантаження.",
- "Model {{modelId}} not found": "Модель {{modelId}} не знайдено",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Модель {{modelName}} не здатна бачити",
"Model {{name}} is now {{status}}": "Модель {{name}} тепер має {{status}}",
@@ -1311,6 +1315,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "API ключ для пошуку Mojeek",
+ "Monthly": "",
"More": "Більше",
"More Concise": "",
"More options": "",
@@ -1431,6 +1436,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Версія Ollama",
"On": "Увімк",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1515,6 +1521,7 @@
"Persistent": "",
"Personalization": "Персоналізація",
"Pin": "Зачепити",
+ "Pin to Sidebar": "",
"Pinned": "Зачеплено",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1677,6 +1684,7 @@
"Running": "Виконується",
"Running...": "Виконується...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Зберегти",
"Save & Create": "Зберегти та створити",
"Save & Update": "Зберегти та оновити",
@@ -1893,6 +1901,7 @@
"STT Model": "Модель STT ",
"STT Settings": "Налаштування STT",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1938,6 +1947,7 @@
"Text Splitter": "Роздільник тексту",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Система синтезу мови",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Дякуємо за ваш відгук!",
"The Application Account DN you bind with for search": "DN облікового запису застосунку, з яким ви здійснюєте прив'язку для пошуку",
"The base to search for users": "База для пошуку користувачів",
@@ -2051,6 +2061,7 @@
"TTS Model": "Модель TTS",
"TTS Settings": "Налаштування TTS",
"TTS Voice": "Голос TTS",
+ "Tu_day_of_week": "",
"Type": "Тип",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Введіть URL ресурсу Hugging Face Resolve (завантаження)",
@@ -2153,6 +2164,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Попередження: Увімкнення цього дозволить користувачам завантажувати довільний код на сервер.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Попередження: Виконання коду в Jupyter дозволяє виконувати任 будь-який код, що становить серйозні ризики для безпеки — дійте з крайньою обережністю.",
+ "We_day_of_week": "",
"Web": "Веб",
"Web API": "Веб-API",
"Web Loader Engine": "",
@@ -2169,6 +2181,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI надсилатиме запити до \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI надсилатиме запити до \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Чого ви прагнете досягти?",
"What are you working on?": "Над чим ти працюєш?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/ur-PK/translation.json b/src/lib/i18n/locales/ur-PK/translation.json
index 7a43ef0db5..8ac674cc4b 100644
--- a/src/lib/i18n/locales/ur-PK/translation.json
+++ b/src/lib/i18n/locales/ur-PK/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "اکاؤنٹ بنائیں",
"Create Admin Account": "",
+ "Create and manage scheduled automations": "",
"Create Channel": "",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "",
"Dark": "ڈارک",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "",
+ "Fr_day_of_week": "",
"Full Context Mode": "",
"Function": "فنکشن",
"Function Calling": "",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "",
"Host": "",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "میں آج آپ کی کس طرح مدد کر سکتا ہوں؟",
"How would you rate this response?": "",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "",
"Mistral OCR API Key required.": "",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "ماڈل",
"Model '{{modelName}}' has been successfully downloaded.": "ماڈل '{{modelName}}' کامیابی سے ڈاؤن لوڈ ہو گیا ہے",
"Model '{{modelTag}}' is already in queue for downloading.": "ماڈل '{{modelTag}}' پہلے ہی ڈاؤن لوڈ کے لیے قطار میں ہے",
- "Model {{modelId}} not found": "ماڈل {{modelId}} نہیں ملا",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "ماڈل {{modelName}} بصری صلاحیت نہیں رکھتا",
"Model {{name}} is now {{status}}": "ماڈل {{name}} اب {{status}} ہے",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "",
+ "Monthly": "",
"More": "مزید",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "اولاما ورژن",
"On": "چالو",
+ "Once": "",
"OneDrive": "",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "شخصی ترتیبات",
"Pin": "پن",
+ "Pin to Sidebar": "",
"Pinned": "پن کیا گیا",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "چل رہا ہے",
"Running...": "چل رہا ہے...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "محفوظ کریں",
"Save & Create": "محفوظ کریں اور تخلیق کریں",
"Save & Update": "محفوظ کریں اور اپ ڈیٹ کریں",
@@ -1889,6 +1897,7 @@
"STT Model": "ایس ٹی ٹی ماڈل",
"STT Settings": "ایس ٹی ٹی ترتیبات",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "متن تقسیم کنندہ",
"Text-to-Speech": "",
"Text-to-Speech Engine": "ٹیکسٹ ٹو اسپیچ انجن",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "آپ کی رائے کا شکریہ!",
"The Application Account DN you bind with for search": "",
"The base to search for users": "",
@@ -2047,6 +2057,7 @@
"TTS Model": "ٹی ٹی ایس ماڈل",
"TTS Settings": "ٹی ٹی ایس ترتیبات",
"TTS Voice": "ٹی ٹی ایس آواز",
+ "Tu_day_of_week": "",
"Type": "ٹائپ کریں",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "قسم ہگنگ فیس ری زولو (ڈاؤن لوڈ) یو آر ایل",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "",
+ "We_day_of_week": "",
"Web": "ویب",
"Web API": "ویب اے پی آئی",
"Web Loader Engine": "",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "",
"WebUI will make requests to \"{{url}}/api/chat\"": "",
"WebUI will make requests to \"{{url}}/chat/completions\"": "",
+ "Weekly": "",
"What are you trying to achieve?": "",
"What are you working on?": "",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/uz-Cyrl-UZ/translation.json b/src/lib/i18n/locales/uz-Cyrl-UZ/translation.json
index b2928870f7..3ffd9ab2fa 100644
--- a/src/lib/i18n/locales/uz-Cyrl-UZ/translation.json
+++ b/src/lib/i18n/locales/uz-Cyrl-UZ/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Ҳисоб яратиш",
"Create Admin Account": "Администратор ҳисобини яратинг",
+ "Create and manage scheduled automations": "",
"Create Channel": "Канал яратиш",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Махсус параметр номи",
"Custom Parameter Value": "Махсус параметр қиймати",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Хавфли зона",
"Dark": "Қоронғи",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Аутентификация қилиш учун тизим фойдаланувчиси сеанси ҳисоб маълумотларини йўналтиради",
+ "Fr_day_of_week": "",
"Full Context Mode": "Тўлиқ контекст режими",
"Function": "Функция",
"Function Calling": "Функцияни чақириш",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Уй",
"Host": "Хост",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Бугун сизга қандай ёрдам бера оламан?",
"How would you rate this response?": "Бу жавобни қандай баҳолайсиз?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral ОCР",
"Mistral OCR API Key required.": "Mistral ОCР АПИ калити талаб қилинади.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Модел",
"Model '{{modelName}}' has been successfully downloaded.": "“{{modelName}}” модели юклаб олинди.",
"Model '{{modelTag}}' is already in queue for downloading.": "“{{modelTag}}” модели аллақачон юклаб олиш учун навбатда турибди.",
- "Model {{modelId}} not found": "{{modelId}} модели топилмади",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "{{modelName}} модели кўриш қобилиятига эга эмас",
"Model {{name}} is now {{status}}": "{{name}} модели энди {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Можеэк қидирув АПИ калити",
+ "Monthly": "",
"More": "Кўпроқ",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama версияси",
"On": "Ёниқ",
+ "Once": "",
"OneDrive": "ОнеДриве",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Шахсийлаштириш",
"Pin": "Пин",
+ "Pin to Sidebar": "",
"Pinned": "Қадалган",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Ишлаётган",
"Running...": "Ишлаётган...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Сақлаш",
"Save & Create": "Сақлаш ва яратиш",
"Save & Update": "Сақлаш ва янгилаш",
@@ -1889,6 +1897,7 @@
"STT Model": "СТТ модели",
"STT Settings": "СТТ созламалари",
"Stylized PDF Export": "Услубий PDF экспорти",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Матн ажратувчи",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Матнни нутққа айлантириш механизми",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Фикр-мулоҳазангиз учун ташаккур!",
"The Application Account DN you bind with for search": "Қидириш учун сиз боғланган илова ҳисоби ДН",
"The base to search for users": "Фойдаланувчиларни қидириш учун асос",
@@ -2047,6 +2057,7 @@
"TTS Model": "ТТС модели",
"TTS Settings": "ТТС созламалари",
"TTS Voice": "ТТС овози",
+ "Tu_day_of_week": "",
"Type": "Тури",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (юклаб олиш) URL манзилини киритинг",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Огоҳлантириш: Буни ёқиш фойдаланувчиларга серверга ихтиёрий кодни юклаш имконини беради.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Огоҳлантириш: Жупйтер ижроси ўзбошимчалик билан код бажарилишини таъминлайди, бу хавфсизликка жиддий хавф туғдиради - жуда эҳтиёткорлик билан давом этинг.",
+ "We_day_of_week": "",
"Web": "Веб",
"Web API": "Web API",
"Web Loader Engine": "Web Loader Engine",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI “{{url}}” манзилига сўров юборади",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI “{{url}}/api/chat” манзилига сўров юборади",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI \"{{url}}/chat/completions\" манзилига сўров юборади",
+ "Weekly": "",
"What are you trying to achieve?": "Нимага эришмоқчисиз?",
"What are you working on?": "Нима устида ишлаяпсиз?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/uz-Latn-Uz/translation.json b/src/lib/i18n/locales/uz-Latn-Uz/translation.json
index 767dd2bb23..06bd7cf1fa 100644
--- a/src/lib/i18n/locales/uz-Latn-Uz/translation.json
+++ b/src/lib/i18n/locales/uz-Latn-Uz/translation.json
@@ -454,6 +454,7 @@
"Create a new note": "",
"Create Account": "Hisob yaratish",
"Create Admin Account": "Administrator hisobini yarating",
+ "Create and manage scheduled automations": "",
"Create Channel": "Kanal yaratish",
"Create Folder": "",
"Create Image": "",
@@ -479,6 +480,7 @@
"Custom Gender": "",
"Custom Parameter Name": "Maxsus parametr nomi",
"Custom Parameter Value": "Maxsus parametr qiymati",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Xavfli zona",
"Dark": "Qorong'i",
@@ -969,6 +971,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Autentifikatsiya qilish uchun tizim foydalanuvchisi seansi hisob ma'lumotlarini yo'naltiradi",
+ "Fr_day_of_week": "",
"Full Context Mode": "To'liq kontekst rejimi",
"Function": "Funktsiya",
"Function Calling": "Funktsiyani chaqirish",
@@ -1043,6 +1046,7 @@
"History": "",
"Home": "Uy",
"Host": "Xost",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Bugun sizga qanday yordam bera olaman?",
"How would you rate this response?": "Bu javobni qanday baholaysiz?",
@@ -1265,10 +1269,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Mistral OCR API kaliti talab qilinadi.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Model",
"Model '{{modelName}}' has been successfully downloaded.": "“{{modelName}}” modeli yuklab olindi.",
"Model '{{modelTag}}' is already in queue for downloading.": "“{{modelTag}}” modeli allaqachon yuklab olish uchun navbatda turibdi.",
- "Model {{modelId}} not found": "{{modelId}} modeli topilmadi",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "{{modelName}} modeli ko'rish qobiliyatiga ega emas",
"Model {{name}} is now {{status}}": "{{name}} modeli endi {{status}}",
@@ -1309,6 +1313,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Mojeek qidiruv API kaliti",
+ "Monthly": "",
"More": "Ko'proq",
"More Concise": "",
"More options": "",
@@ -1429,6 +1434,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Ollama versiyasi",
"On": "Yoniq",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1513,6 +1519,7 @@
"Persistent": "",
"Personalization": "Shaxsiylashtirish",
"Pin": "Pin",
+ "Pin to Sidebar": "",
"Pinned": "Qadalgan",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1673,6 +1680,7 @@
"Running": "Yugurish",
"Running...": "Yugurish...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Saqlash",
"Save & Create": "Saqlash va yaratish",
"Save & Update": "Saqlash va yangilash",
@@ -1889,6 +1897,7 @@
"STT Model": "STT modeli",
"STT Settings": "STT sozlamalari",
"Stylized PDF Export": "Stillashtirilgan PDF eksporti",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1934,6 +1943,7 @@
"Text Splitter": "Matn ajratuvchi",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Matnni nutqqa aylantirish mexanizmi",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Fikr-mulohazangiz uchun tashakkur!",
"The Application Account DN you bind with for search": "Qidirish uchun siz bog'langan ilova hisobi DN",
"The base to search for users": "Foydalanuvchilarni qidirish uchun asos",
@@ -2047,6 +2057,7 @@
"TTS Model": "TTS modeli",
"TTS Settings": "TTS sozlamalari",
"TTS Voice": "TTS ovozi",
+ "Tu_day_of_week": "",
"Type": "Turi",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Hugging Face Resolve (Yuklab olish) URL manzilini kiriting",
@@ -2149,6 +2160,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Ogohlantirish: Buni yoqish foydalanuvchilarga serverga ixtiyoriy kodni yuklash imkonini beradi.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Ogohlantirish: Jupyter ijrosi o'zboshimchalik bilan kod bajarilishini ta'minlaydi, bu xavfsizlikka jiddiy xavf tug'diradi - juda ehtiyotkorlik bilan davom eting.",
+ "We_day_of_week": "",
"Web": "Veb",
"Web API": "Web API",
"Web Loader Engine": "Web Loader Engine",
@@ -2165,6 +2177,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI “{{url}}” manziliga so‘rov yuboradi",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI “{{url}}/api/chat” manziliga so‘rov yuboradi",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI \"{{url}}/chat/completions\" manziliga so'rov yuboradi",
+ "Weekly": "",
"What are you trying to achieve?": "Nimaga erishmoqchisiz?",
"What are you working on?": "Nima ustida ishlayapsiz?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/vi-VN/translation.json b/src/lib/i18n/locales/vi-VN/translation.json
index d2917e0c20..604dac83f6 100644
--- a/src/lib/i18n/locales/vi-VN/translation.json
+++ b/src/lib/i18n/locales/vi-VN/translation.json
@@ -453,6 +453,7 @@
"Create a new note": "",
"Create Account": "Tạo Tài khoản",
"Create Admin Account": "Tạo Tài khoản Quản trị",
+ "Create and manage scheduled automations": "",
"Create Channel": "Tạo Kênh",
"Create Folder": "",
"Create Image": "",
@@ -478,6 +479,7 @@
"Custom Gender": "",
"Custom Parameter Name": "",
"Custom Parameter Value": "",
+ "Daily": "",
"Daily Messages": "",
"Danger Zone": "Vùng Nguy hiểm",
"Dark": "Tối",
@@ -968,6 +970,7 @@
"Forward": "",
"Forwards system user OAuth access token to authenticate": "",
"Forwards system user session credentials to authenticate": "Chuyển tiếp thông tin xác thực phiên người dùng hệ thống để xác thực",
+ "Fr_day_of_week": "",
"Full Context Mode": "Chế độ Ngữ cảnh Đầy đủ",
"Function": "Function",
"Function Calling": "Gọi Function",
@@ -1042,6 +1045,7 @@
"History": "",
"Home": "Trang chủ",
"Host": "Host",
+ "Hourly": "",
"Hourly Messages": "",
"How can I help you today?": "Tôi có thể giúp gì cho bạn hôm nay?",
"How would you rate this response?": "Bạn đánh giá phản hồi này thế nào?",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "Yêu cầu Khóa API Mistral OCR.",
"MistralAI": "",
+ "Mo_day_of_week": "",
"Model": "Mô hình",
"Model '{{modelName}}' has been successfully downloaded.": "Mô hình '{{modelName}}' đã được tải xuống thành công.",
"Model '{{modelTag}}' is already in queue for downloading.": "Mô hình '{{modelTag}}' đã có trong hàng đợi để tải xuống.",
- "Model {{modelId}} not found": "Không tìm thấy Mô hình {{modelId}}",
"Model {{modelName}} deleted successfully": "",
"Model {{modelName}} is not vision capable": "Model {{modelName}} không có khả năng nhìn",
"Model {{name}} is now {{status}}": "Model {{name}} bây giờ là {{status}}",
@@ -1308,6 +1312,7 @@
"Models Sharing": "",
"Mojeek": "",
"Mojeek Search API Key": "Khóa API Mojeek Search",
+ "Monthly": "",
"More": "Thêm",
"More Concise": "",
"More options": "",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "",
"Ollama Version": "Phiên bản Ollama",
"On": "Bật",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "",
"Only active when the chat input is in focus and an LLM is generating a response.": "",
@@ -1512,6 +1518,7 @@
"Persistent": "",
"Personalization": "Cá nhân hóa",
"Pin": "Ghim",
+ "Pin to Sidebar": "",
"Pinned": "Đã ghim",
"Pinned Messages": "",
"Pinned Models": "",
@@ -1671,6 +1678,7 @@
"Running": "Đang chạy",
"Running...": "Đang chạy...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "",
+ "Sa_day_of_week": "",
"Save": "Lưu",
"Save & Create": "Lưu & Tạo",
"Save & Update": "Lưu & Cập nhật",
@@ -1887,6 +1895,7 @@
"STT Model": "Mô hình STT",
"STT Settings": "Cài đặt Nhận dạng Giọng nói",
"Stylized PDF Export": "",
+ "Su_day_of_week": "",
"Submit question": "",
"Submit suggestion": "",
"Subtitle": "",
@@ -1932,6 +1941,7 @@
"Text Splitter": "Bộ chia Văn bản",
"Text-to-Speech": "",
"Text-to-Speech Engine": "Công cụ Chuyển Văn bản thành Giọng nói",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "Cám ơn bạn đã gửi phản hồi!",
"The Application Account DN you bind with for search": "DN Tài khoản Ứng dụng bạn liên kết để tìm kiếm",
"The base to search for users": "Cơ sở để tìm kiếm người dùng",
@@ -2045,6 +2055,7 @@
"TTS Model": "Mô hình TTS",
"TTS Settings": "Cài đặt Chuyển văn bản thành Giọng nói",
"TTS Voice": "Giọng nói TTS",
+ "Tu_day_of_week": "",
"Type": "Kiểu",
"Type here...": "",
"Type Hugging Face Resolve (Download) URL": "Nhập URL Hugging Face Resolve (Tải xuống)",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "Cảnh báo: Bật tính năng này sẽ cho phép người dùng tải lên mã tùy ý trên máy chủ.",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "Cảnh báo: Thực thi Jupyter cho phép thực thi mã tùy ý, gây ra rủi ro bảo mật nghiêm trọng—hãy tiến hành hết sức thận trọng.",
+ "We_day_of_week": "",
"Web": "Web",
"Web API": "Web API",
"Web Loader Engine": "",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI sẽ thực hiện yêu cầu đến \"{{url}}\"",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI sẽ thực hiện yêu cầu đến \"{{url}}/api/chat\"",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI sẽ thực hiện yêu cầu đến \"{{url}}/chat/completions\"",
+ "Weekly": "",
"What are you trying to achieve?": "Bạn đang cố gắng đạt được điều gì?",
"What are you working on?": "Bạn đang làm gì vậy?",
"What is NOT shared:": "",
diff --git a/src/lib/i18n/locales/zh-CN/translation.json b/src/lib/i18n/locales/zh-CN/translation.json
index 52295330c8..a089bb8e01 100644
--- a/src/lib/i18n/locales/zh-CN/translation.json
+++ b/src/lib/i18n/locales/zh-CN/translation.json
@@ -137,7 +137,7 @@
"Allow Text to Speech": "允许文本转语音",
"Allow User Location": "获取您的位置",
"Allow Voice Interruption in Call": "允许语音通话时打断对话",
- "Allow Web Upload": "允许上传文件",
+ "Allow Web Upload": "允许从网络上传内容",
"Allowed Endpoints": "允许的接口",
"Allowed File Extensions": "允许的文件扩展名",
"Allowed file extensions for upload. Separate multiple extensions with commas. Leave empty for all file types.": "文件上传允许的扩展名。多个扩展名用逗号分隔。留空以允许所有文件类型。",
@@ -453,6 +453,7 @@
"Create a new note": "新建笔记",
"Create Account": "创建账号",
"Create Admin Account": "创建管理员账号",
+ "Create and manage scheduled automations": "",
"Create Channel": "创建频道",
"Create Folder": "创建分组",
"Create Image": "图片生成",
@@ -478,6 +479,7 @@
"Custom Gender": "自定义性别",
"Custom Parameter Name": "自定义参数名称",
"Custom Parameter Value": "自定义参数值",
+ "Daily": "",
"Daily Messages": "每日消息数",
"Danger Zone": "危险区域",
"Dark": "暗色",
@@ -968,6 +970,7 @@
"Forward": "前进",
"Forwards system user OAuth access token to authenticate": "转发用户的 OAuth 访问令牌（Access Token）以进行身份验证",
"Forwards system user session credentials to authenticate": "转发用户的会话凭证（Session Credentials）以进行身份验证",
+ "Fr_day_of_week": "",
"Full Context Mode": "完整上下文模式",
"Function": "函数",
"Function Calling": "函数调用 (Function Calling)",
@@ -1042,6 +1045,7 @@
"History": "历史记录",
"Home": "主页",
"Host": "主机",
+ "Hourly": "",
"Hourly Messages": "每小时消息数",
"How can I help you today?": "有什么我能帮您的吗？",
"How would you rate this response?": "您如何评价这个回答？",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "需要 Mistral OCR 接口密钥",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "模型",
"Model '{{modelName}}' has been successfully downloaded.": "模型“{{modelName}}”已成功下载",
"Model '{{modelTag}}' is already in queue for downloading.": "模型“{{modelTag}}”已在下载队列中",
- "Model {{modelId}} not found": "未找到模型 {{modelId}}",
"Model {{modelName}} deleted successfully": "成功删除模型：{{modelName}}",
"Model {{modelName}} is not vision capable": "模型 {{modelName}} 不支持视觉能力",
"Model {{name}} is now {{status}}": "模型 {{name}} 现在是 {{status}}",
@@ -1308,6 +1312,7 @@
"Models Sharing": "分享模型",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Mojeek Search 接口密钥",
+ "Monthly": "",
"More": "更多",
"More Concise": "精炼表达",
"More options": "更多选项",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "Ollama Cloud 接口密钥",
"Ollama Version": "Ollama 版本",
"On": "开启",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "仅在启用“粘贴超长文本为文件”选项时有效。",
"Only active when the chat input is in focus and an LLM is generating a response.": "仅在聚焦对话框且大语言模型正在生成回答时有效。",
@@ -1512,6 +1518,7 @@
"Persistent": "持久化",
"Personalization": "个性化",
"Pin": "置顶",
+ "Pin to Sidebar": "",
"Pinned": "已置顶",
"Pinned Messages": "置顶消息",
"Pinned Models": "固定在侧边栏的模型",
@@ -1671,6 +1678,7 @@
"Running": "运行中",
"Running...": "运行中...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "并行运行嵌入任务以加快处理速度。如果遇到限速问题，请关闭此选项。",
+ "Sa_day_of_week": "",
"Save": "保存",
"Save & Create": "保存并创建",
"Save & Update": "保存并更新",
@@ -1887,6 +1895,7 @@
"STT Model": "语音转文本模型",
"STT Settings": "语音转文本设置",
"Stylized PDF Export": "美化 PDF 导出",
+ "Su_day_of_week": "",
"Submit question": "提交问题",
"Submit suggestion": "提交建议",
"Subtitle": "副标题",
@@ -1932,6 +1941,7 @@
"Text Splitter": "文本切分器",
"Text-to-Speech": "文本转语音",
"Text-to-Speech Engine": "文本转语音引擎",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "感谢您的反馈！",
"The Application Account DN you bind with for search": "您所绑定用于搜索的 Application Account DN",
"The base to search for users": "搜索用户的 Base",
@@ -2045,6 +2055,7 @@
"TTS Model": "文本转语音模型",
"TTS Settings": "文本转语音设置",
"TTS Voice": "文本转语音音色",
+ "Tu_day_of_week": "",
"Type": "类型",
"Type here...": "请输入内容...",
"Type Hugging Face Resolve (Download) URL": "输入 Hugging Face 模型解析（下载）地址",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "警告：启用后将允许用户自动执行定时提示词任务。",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "警告：启用此功能将允许用户在服务器上上传任意代码",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "警告：启用 Jupyter 执行将允许运行任意代码，存在严重安全风险——务必谨慎操作",
+ "We_day_of_week": "",
"Web": "网页",
"Web API": "网页 API",
"Web Loader Engine": "网页加载引擎",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI 将向 \"{{url}}\" 发出请求",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI 将向 \"{{url}}/api/chat\" 发出请求",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI 将向 \"{{url}}/chat/completions\" 发出请求",
+ "Weekly": "",
"What are you trying to achieve?": "您想要达到什么目标？",
"What are you working on?": "您在忙于什么？",
"What is NOT shared:": "不共享的内容：",
diff --git a/src/lib/i18n/locales/zh-TW/translation.json b/src/lib/i18n/locales/zh-TW/translation.json
index 497f7a623b..f90782c3f6 100644
--- a/src/lib/i18n/locales/zh-TW/translation.json
+++ b/src/lib/i18n/locales/zh-TW/translation.json
@@ -137,7 +137,7 @@
"Allow Text to Speech": "允許文字轉語音",
"Allow User Location": "允許使用者位置",
"Allow Voice Interruption in Call": "允許在通話中打斷語音",
- "Allow Web Upload": "允許上傳檔案",
+ "Allow Web Upload": "允許從網路上傳內容",
"Allowed Endpoints": "允許的端點",
"Allowed File Extensions": "允許的檔案副檔名",
"Allowed file extensions for upload. Separate multiple extensions with commas. Leave empty for all file types.": "允許上傳的檔案副檔名。多個副檔名請用逗號分隔，留空則允許所有檔案類型。",
@@ -453,6 +453,7 @@
"Create a new note": "新建筆記",
"Create Account": "建立帳號",
"Create Admin Account": "建立管理員帳號",
+ "Create and manage scheduled automations": "",
"Create Channel": "建立頻道",
"Create Folder": "建立分組",
"Create Image": "產生圖片",
@@ -478,6 +479,7 @@
"Custom Gender": "自訂性別",
"Custom Parameter Name": "自訂參數名稱",
"Custom Parameter Value": "自訂參數值",
+ "Daily": "",
"Daily Messages": "每日訊息數",
"Danger Zone": "危險區域",
"Dark": "深色",
@@ -968,6 +970,7 @@
"Forward": "前進",
"Forwards system user OAuth access token to authenticate": "轉發使用者 OAuth 存取權杖（Access Token）以進行驗證",
"Forwards system user session credentials to authenticate": "轉發使用者工作階段憑證（Session Credentials）以進行驗證",
+ "Fr_day_of_week": "",
"Full Context Mode": "完整上下文模式",
"Function": "函式",
"Function Calling": "函式呼叫",
@@ -1042,6 +1045,7 @@
"History": "歷史紀錄",
"Home": "首頁",
"Host": "主機",
+ "Hourly": "",
"Hourly Messages": "每小時訊息數",
"How can I help you today?": "今天我能為您做些什麼？",
"How would you rate this response?": "您如何評價此回應？",
@@ -1264,10 +1268,10 @@
"Mistral OCR": "Mistral OCR",
"Mistral OCR API Key required.": "需要提供 Mistral OCR API 金鑰。",
"MistralAI": "MistralAI",
+ "Mo_day_of_week": "",
"Model": "模型",
"Model '{{modelName}}' has been successfully downloaded.": "模型「{{modelName}}」已成功下載。",
"Model '{{modelTag}}' is already in queue for downloading.": "模型「{{modelTag}}」已在下載佇列中。",
- "Model {{modelId}} not found": "未找到模型 {{modelId}}",
"Model {{modelName}} deleted successfully": "模型 {{modelName}} 已成功刪除",
"Model {{modelName}} is not vision capable": "模型 {{modelName}} 不具備視覺能力",
"Model {{name}} is now {{status}}": "模型 {{name}} 現在狀態為 {{status}}",
@@ -1308,6 +1312,7 @@
"Models Sharing": "分享模型",
"Mojeek": "Mojeek",
"Mojeek Search API Key": "Mojeek 搜尋 API 金鑰",
+ "Monthly": "",
"More": "更多",
"More Concise": "精煉表達",
"More options": "更多選項",
@@ -1428,6 +1433,7 @@
"Ollama Cloud API Key": "Ollama Cloud API 金鑰",
"Ollama Version": "Ollama 版本",
"On": "開啟",
+ "Once": "",
"OneDrive": "OneDrive",
"Only active when \"Paste Large Text as File\" setting is toggled on.": "僅在啟用「將大型文字作為檔案貼上」設定時有效。",
"Only active when the chat input is in focus and an LLM is generating a response.": "僅在對話輸入框聚焦且大型語言模型正在產生回應時有效。",
@@ -1512,6 +1518,7 @@
"Persistent": "持久性",
"Personalization": "個人化",
"Pin": "釘選",
+ "Pin to Sidebar": "",
"Pinned": "已釘選",
"Pinned Messages": "置頂訊息",
"Pinned Models": "固定於側邊欄的模型",
@@ -1671,6 +1678,7 @@
"Running": "正在執行",
"Running...": "正在執行...",
"Runs embedding tasks concurrently to speed up processing. Turn off if rate limits become an issue.": "同時執行嵌入任務以加快處理速度。如果遇到速率限制問題，請關閉此功能。",
+ "Sa_day_of_week": "",
"Save": "儲存",
"Save & Create": "儲存並建立",
"Save & Update": "儲存並更新",
@@ -1887,6 +1895,7 @@
"STT Model": "語音轉文字 (STT) 模型",
"STT Settings": "語音轉文字 (STT) 設定",
"Stylized PDF Export": "風格化 PDF 匯出",
+ "Su_day_of_week": "",
"Submit question": "提交問題",
"Submit suggestion": "提交建議",
"Subtitle": "副標題",
@@ -1932,6 +1941,7 @@
"Text Splitter": "文字分割器",
"Text-to-Speech": "文字轉語音",
"Text-to-Speech Engine": "文字轉語音引擎",
+ "Th_day_of_week": "",
"Thanks for your feedback!": "感謝您的回饋！",
"The Application Account DN you bind with for search": "您綁定用於搜尋的應用程式帳號 DN",
"The base to search for users": "搜尋使用者的基礎",
@@ -2045,6 +2055,7 @@
"TTS Model": "文字轉語音 (TTS) 模型",
"TTS Settings": "文字轉語音 (TTS) 設定",
"TTS Voice": "文字轉語音 (TTS) 聲音",
+ "Tu_day_of_week": "",
"Type": "類型",
"Type here...": "在此輸入...",
"Type Hugging Face Resolve (Download) URL": "輸入 Hugging Face 的解析（下載）URL",
@@ -2147,6 +2158,7 @@
"Warning: Enabling this will allow users to run scheduled prompts automatically.": "警告：啟用後將允許使用者自動執行排程提示詞。",
"Warning: Enabling this will allow users to upload arbitrary code on the server.": "警告：啟用此功能將允許使用者在伺服器上上傳任意程式碼。",
"Warning: Jupyter execution enables arbitrary code execution, posing severe security risks—proceed with extreme caution.": "警告：Jupyter 執行允許任意程式碼執行，構成嚴重安全風險 —— 請務必極度謹慎。",
+ "We_day_of_week": "",
"Web": "網頁",
"Web API": "網頁 API",
"Web Loader Engine": "網頁載入引擎",
@@ -2163,6 +2175,7 @@
"WebUI will make requests to \"{{url}}\"": "WebUI 將向 \"{{url}}\" 傳送請求",
"WebUI will make requests to \"{{url}}/api/chat\"": "WebUI 將向 \"{{url}}/api/chat\" 傳送請求",
"WebUI will make requests to \"{{url}}/chat/completions\"": "WebUI 將向 \"{{url}}/chat/completions\" 傳送請求",
+ "Weekly": "",
"What are you trying to achieve?": "您正在試圖完成什麼？",
"What are you working on?": "您現在的工作是什麼？",
"What is NOT shared:": "不會分享的內容：",
diff --git a/src/lib/stores/index.ts b/src/lib/stores/index.ts
index edcea62137..a59944f50c 100644
--- a/src/lib/stores/index.ts
+++ b/src/lib/stores/index.ts
@@ -59,6 +59,7 @@ export const channelId = writable(null);
export const chats = writable(null);
export const pinnedChats = writable([]);
+export const pinnedNotes = writable([]);
export const tags = writable([]);
export const folders = writable([]);
@@ -116,8 +117,8 @@ export const temporaryChatEnabled = writable(false);
// Set by +layout.svelte, consumed and cleared by Chat.svelte.
export type DesktopEventFile = { name: string; mimeType: string; dataUrl: string };
export type DesktopEvent = {
- query?: string;
- files?: DesktopEventFile[];
+ type: string;
+ data?: any;
};
export const desktopEvent: Writable = writable(null);
export const scrollPaginationEnabled = writable(false);
diff --git a/src/lib/utils/index.ts b/src/lib/utils/index.ts
index c11c33ac33..ecaeefab04 100644
--- a/src/lib/utils/index.ts
+++ b/src/lib/utils/index.ts
@@ -520,7 +520,7 @@ export const copyToClipboard = async (text, html = null, formatted = false) => {
textArea.style.position = 'fixed';
document.body.appendChild(textArea);
- textArea.focus();
+ textArea.focus({ preventScroll: true });
textArea.select();
try {
@@ -923,8 +923,19 @@ export const processDetails = (content) => {
attributes[attributeMatch[1]] = attributeMatch[2];
}
+ // New format: result in body content; Old format: result in attribute
+ let resultText = '';
if (attributes.result) {
- content = content.replace(match, unescapeHtml(attributes.result));
+ resultText = unescapeHtml(attributes.result);
+ } else {
+ // Extract body content (strip ... )
+ const bodyMatch = match.match(/[\s\S]*?<\/summary>\s*([\s\S]*?)\s*<\/details>/i);
+ if (bodyMatch && bodyMatch[1].trim()) {
+ resultText = unescapeHtml(bodyMatch[1].trim());
+ }
+ }
+ if (resultText) {
+ content = content.replace(match, resultText);
}
}
}
diff --git a/src/lib/utils/marked/extension.ts b/src/lib/utils/marked/extension.ts
index eec17adbd0..13f9f7cdc3 100644
--- a/src/lib/utils/marked/extension.ts
+++ b/src/lib/utils/marked/extension.ts
@@ -60,7 +60,7 @@ function detailsTokenizer(src: string) {
}
function detailsStart(src: string) {
- return src.match(/^/) ? 0 : -1;
+ return src.match(/^]/) ? 0 : -1;
}
function detailsRenderer(token: any) {
diff --git a/src/lib/utils/marked/katex-extension.ts b/src/lib/utils/marked/katex-extension.ts
index 13860fab30..a890ff4135 100644
--- a/src/lib/utils/marked/katex-extension.ts
+++ b/src/lib/utils/marked/katex-extension.ts
@@ -66,6 +66,44 @@ function generateRegexRules(delimiters) {
const { inlineRule, blockRule } = generateRegexRules(DELIMITER_LIST);
+const isAllowedTrailing = (src: string, i: number): boolean =>
+ i >= src.length || ALLOWED_SURROUNDING_CHARS_REGEX.test(src.charAt(i));
+
+const isBlockBoundary = (src: string, i: number): boolean =>
+ /^(?:[ \t]*\r?\n|$)/.test(src.slice(i));
+
+const findClosingDelimiter = (src: string, i: number): number =>
+ i >= src.length - 1
+ ? -1
+ : src[i] === '\\'
+ ? findClosingDelimiter(src, i + 2)
+ : src[i] === '$' && src[i + 1] === '$'
+ ? i
+ : findClosingDelimiter(src, i + 1);
+
+export const tokenizeDisplayMath = (
+ src: string,
+ type: 'inlineKatex' | 'blockKatex',
+ requireBlockBoundary = false
+) => {
+ if (!src.startsWith('$$')) return;
+
+ const endIndex = findClosingDelimiter(src, 2);
+ if (endIndex === -1) return;
+
+ const raw = src.slice(0, endIndex + 2);
+ const text = raw.slice(2, -2);
+ const afterClose = endIndex + 2;
+
+ const validators: Array<() => boolean> = [
+ () => text.trim().length > 0,
+ () => isAllowedTrailing(src, afterClose),
+ () => !requireBlockBoundary || isBlockBoundary(src, afterClose)
+ ];
+
+ return validators.every((v) => v()) ? { type, raw, text, displayMode: true } : undefined;
+};
+
export default function (options = {}) {
return {
extensions: [inlineKatex(options), blockKatex(options)]
@@ -102,6 +140,17 @@ function katexStart(src, displayMode: boolean) {
}
function katexTokenizer(src, tokens, displayMode: boolean) {
+ if (src.startsWith('$$')) {
+ const displayToken = tokenizeDisplayMath(
+ src,
+ displayMode ? 'blockKatex' : 'inlineKatex',
+ displayMode
+ );
+ if (displayToken) {
+ return displayToken;
+ }
+ }
+
const ruleReg = displayMode ? blockRule : inlineRule;
const type = displayMode ? 'blockKatex' : 'inlineKatex';
diff --git a/src/lib/utils/marked/mention-extension.ts b/src/lib/utils/marked/mention-extension.ts
index 4e3865fd89..835ce30714 100644
--- a/src/lib/utils/marked/mention-extension.ts
+++ b/src/lib/utils/marked/mention-extension.ts
@@ -18,24 +18,6 @@ function mentionStart(src: string) {
return src.indexOf('<');
}
-function mentionTokenizer(this: any, src: string, options: MentionOptions = {}) {
- const trigger = options.triggerChar ?? '@';
- // Build dynamic regex for `<@id>`, `<@id|label>`, `<@id|>`
- // Added forward slash (/) to the character class for IDs
- const re = new RegExp(`^<\\${trigger}([\\w.\\-:/]+)(?:\\|([^>]*))?>`);
- const m = re.exec(src);
- if (!m) return;
-
- const [, id, label] = m;
- return {
- type: 'mention',
- raw: m[0],
- triggerChar: trigger,
- id,
- label: label && label.length > 0 ? label : id
- };
-}
-
function mentionRenderer(token: any, options: MentionOptions = {}) {
const trigger = options.triggerChar ?? '@';
const cls = options.className ?? 'mention';
@@ -55,15 +37,34 @@ function mentionRenderer(token: any, options: MentionOptions = {}) {
}
export function mentionExtension(opts: MentionOptions = {}) {
+ // Compile the regex once when the extension is created, not on every tokenizer call.
+ // mentionStart fires on every '<' in the document, making the tokenizer a hot path.
+ const trigger = opts.triggerChar ?? '@';
+ const re = new RegExp(`^<\\${trigger}([\\w.\\-:/]+)(?:\\|([^>]*))?>`);
+ const snapshot: MentionOptions = {
+ triggerChar: trigger,
+ className: opts.className,
+ extraAttrs: opts.extraAttrs
+ };
+
return {
name: 'mention',
level: 'inline' as const,
start: mentionStart,
tokenizer(src: string) {
- return mentionTokenizer.call(this, src, opts);
+ const m = re.exec(src);
+ if (!m) return;
+ const [, id, label] = m;
+ return {
+ type: 'mention',
+ raw: m[0],
+ triggerChar: trigger,
+ id,
+ label: label && label.length > 0 ? label : id
+ };
},
renderer(token: any) {
- return mentionRenderer(token, opts);
+ return mentionRenderer(token, snapshot);
}
};
}
diff --git a/src/lib/utils/onedrive-file-picker.ts b/src/lib/utils/onedrive-file-picker.ts
index 136d239294..198a8bcafb 100644
--- a/src/lib/utils/onedrive-file-picker.ts
+++ b/src/lib/utils/onedrive-file-picker.ts
@@ -78,7 +78,8 @@ class OneDriveConfig {
const msalParams = {
auth: {
authority: `https://login.microsoftonline.com/${authorityEndpoint}`,
- clientId: clientId
+ clientId: clientId,
+ redirectUri: window.location.origin
}
};
diff --git a/src/routes/(app)/automations/+page.svelte b/src/routes/(app)/automations/+page.svelte
index 7919ec43ee..e48dd3be37 100644
--- a/src/routes/(app)/automations/+page.svelte
+++ b/src/routes/(app)/automations/+page.svelte
@@ -47,8 +47,8 @@
let page = 1;
- // Debounce only query changes
- $: if (query !== undefined) {
+ // Debounce only query changes (gate behind loaded to prevent double-fetch on mount)
+ $: if (loaded && query !== undefined) {
loading = true;
clearTimeout(searchDebounceTimer);
searchDebounceTimer = setTimeout(() => {
@@ -57,8 +57,8 @@
}, 300);
}
- // Immediate response to page/filter changes
- $: if (page && statusFilter !== undefined) {
+ // Immediate response to page/filter changes (gate behind loaded)
+ $: if (loaded && page && statusFilter !== undefined) {
getAutomationList();
}
@@ -171,6 +171,8 @@
}
loaded = true;
+ // Explicit initial fetch — reactive blocks will handle subsequent changes
+ await getAutomationList();
return () => {
clearTimeout(searchDebounceTimer);
diff --git a/src/routes/+layout.svelte b/src/routes/+layout.svelte
index b028236533..8aeda16594 100644
--- a/src/routes/+layout.svelte
+++ b/src/routes/+layout.svelte
@@ -712,7 +712,12 @@
return;
}
if (event.type === 'query' && (event.data?.query || event.data?.files?.length)) {
- desktopEvent.set({ query: event.data.query, files: event.data.files });
+ desktopEvent.set(event);
+ await goto('/');
+ return;
+ }
+ if (event.type === 'call') {
+ desktopEvent.set(event);
await goto('/');
return;
}
@@ -723,7 +728,8 @@
// Apply theme classes (mirrors logic from chat/Settings/General.svelte)
const themes = ['dark', 'light', 'oled-dark'];
- let themeToApply = newTheme === 'oled-dark' ? 'dark' : newTheme === 'her' ? 'light' : newTheme;
+ let themeToApply =
+ newTheme === 'oled-dark' ? 'dark' : newTheme === 'her' ? 'light' : newTheme;
if (newTheme === 'system') {
themeToApply = window.matchMedia('(prefers-color-scheme: dark)').matches ? 'dark' : 'light';
}
@@ -982,6 +988,16 @@
} catch (error) {
console.error('Error refreshing backend config:', error);
}
+
+ // Relay auth token to desktop app for API access
+ if (window.electronAPI?.send) {
+ window.electronAPI
+ .send({
+ type: 'token:update',
+ token: localStorage.token
+ })
+ .catch(() => {});
+ }
} else {
// Redirect Invalid Session User to /auth Page
localStorage.removeItem('token');