From 980225e5823808de7ace86a8ee73a0b880b37023 Mon Sep 17 00:00:00 2001 From: Classic298 <27028174+Classic298@users.noreply.github.com> Date: Wed, 14 Oct 2026 22:00:08 +0200 Subject: [PATCH] feat: tag Valkey vector store connections as Open WebUI (#32428) On a shared Valkey or ElastiCache, Open WebUI's vector store connections identified themselves only as the generic GLIDE Python client, the same as every other app using that library, so operators could not tell its traffic apart in server-side logs. Both connections now report the library name as GlidePySync(open-webui). The tag needs valkey-glide-sync 2.5.2 or newer, so the pinned install version moves from 2.3.1 to 2.5.2, and an older manual install fails at startup until it is upgraded. Fixes #31654 --- backend/open_webui/retrieval/vector/dbs/valkey.py | 4 +++- backend/requirements.txt | 2 +- pyproject.toml | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/retrieval/vector/dbs/valkey.py b/backend/open_webui/retrieval/vector/dbs/valkey.py index bf202a22c0..c315a0a5fb 100644 --- a/backend/open_webui/retrieval/vector/dbs/valkey.py +++ b/backend/open_webui/retrieval/vector/dbs/valkey.py @@ -56,7 +56,7 @@ def _import_glide(): ) except ImportError as e: raise ImportError( - 'valkey-glide-sync is required when VECTOR_DB=valkey. Install it with: pip install valkey-glide-sync==2.3.1' + 'valkey-glide-sync is required when VECTOR_DB=valkey. Install it with: pip install valkey-glide-sync==2.5.2' ) from e return { 'Batch': Batch, @@ -193,6 +193,7 @@ class ValkeyClient(VectorDBBase): database_id=db if db else None, request_timeout=5000, client_name='open_webui_vector_store_client', + client_info_tag='open-webui', ) try: self.client = GlideClient.create(config) @@ -206,6 +207,7 @@ class ValkeyClient(VectorDBBase): database_id=db if db else None, request_timeout=10000, # 10s — HNSW indexing can take 1-4s per vector client_name='open_webui_vector_store_batch_client', + client_info_tag='open-webui', ) try: self.batch_client = GlideClient.create(batch_config) diff --git a/backend/requirements.txt b/backend/requirements.txt index e26cd08c67..c5c95fa427 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -116,7 +116,7 @@ pgvector==0.4.2 PyMySQL==1.2.0 boto3==1.42.62 # mariadb==1.1.14 should be added if you want to support MariaDB -# valkey-glide-sync==2.3.1 # optional: install manually if VECTOR_DB=valkey +# valkey-glide-sync==2.5.2 # optional: install manually if VECTOR_DB=valkey pymilvus==2.6.14 qdrant-client==1.18.0 diff --git a/pyproject.toml b/pyproject.toml index 202bf0a352..78f256eff8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -43,7 +43,7 @@ dependencies = [ "pycrdt==0.14.8", "redis==8.0.1", "hiredis==3.4.2", - # "valkey-glide-sync==2.3.1", # optional: install manually if VECTOR_DB=valkey + # "valkey-glide-sync==2.5.2", # optional: install manually if VECTOR_DB=valkey "pytz==2026.2",