From b242460874967d3e0e9d50bce17ad6f1dff9e7b4 Mon Sep 17 00:00:00 2001 From: Peter De-Ath Date: Sun, 6 Oct 2024 23:09:07 +0100 Subject: [PATCH] formatting --- backend/open_webui/config.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py index fb6f7282d..22b3d385b 100644 --- a/backend/open_webui/config.py +++ b/backend/open_webui/config.py @@ -989,7 +989,10 @@ 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") or os.environ.get("RAG_EMBEDDING_OPENAI_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(