UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-07 00:59:52 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #4701 from ndrsfel/fix-rag-embedding-openai-batch-size-environment-variable

Browse Source 
fix: RAG with OpenAI embedding models and batch_size environment variable fails silently
This commit is contained in:
Timothy Jaeryang Baek 2024-08-18 13:24:18 +02:00 committed by GitHub
commit 079f37a2d6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/config.py
View File

@ -1055,7 +1055,7 @@ RAG_EMBEDDING_MODEL_TRUST_REMOTE_CODE = (
RAG_EMBEDDING_OPENAI_BATCH_SIZE = PersistentConfig( RAG_EMBEDDING_OPENAI_BATCH_SIZE = PersistentConfig(
"RAG_EMBEDDING_OPENAI_BATCH_SIZE", "RAG_EMBEDDING_OPENAI_BATCH_SIZE",
"rag.embedding_openai_batch_size", "rag.embedding_openai_batch_size",
os.environ.get("RAG_EMBEDDING_OPENAI_BATCH_SIZE", 1), int(os.environ.get("RAG_EMBEDDING_OPENAI_BATCH_SIZE", "1")),
) )
RAG_RERANKING_MODEL = PersistentConfig( RAG_RERANKING_MODEL = PersistentConfig(