From 3fcdca2f283378f37d55237df5ea112eaa0e0e99 Mon Sep 17 00:00:00 2001 From: Peter De-Ath Date: Sun, 6 Oct 2024 23:03:34 +0100 Subject: [PATCH] backwards comp: use RAG_EMBEDDING_OPENAI_BATCH_SIZE --- backend/open_webui/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py index eac4855ce..fb6f7282d 100644 --- a/backend/open_webui/config.py +++ b/backend/open_webui/config.py @@ -989,7 +989,7 @@ RAG_EMBEDDING_MODEL_TRUST_REMOTE_CODE = ( RAG_EMBEDDING_BATCH_SIZE = PersistentConfig( "RAG_EMBEDDING_BATCH_SIZE", "rag.embedding_batch_size", - int(os.environ.get("RAG_EMBEDDING_BATCH_SIZE", "1")), + int(os.environ.get("RAG_EMBEDDING_BATCH_SIZE") or os.environ.get("RAG_EMBEDDING_OPENAI_BATCH_SIZE", "1")) ) RAG_RERANKING_MODEL = PersistentConfig(