backwards comp: use RAG_EMBEDDING_OPENAI_BATCH_SIZE

This commit is contained in:
Peter De-Ath 2024-10-06 23:03:34 +01:00
parent 885b9f1ece
commit 3fcdca2f28

View File

@ -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(