diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py index 97faf4f38..69b980688 100644 --- a/backend/open_webui/config.py +++ b/backend/open_webui/config.py @@ -2072,6 +2072,35 @@ DEFAULT_RAG_SETTINGS = PersistentConfig( os.getenv("DEFAULT_RAG_SETTINGS", "True").lower() == "true", ) +DOWNLOADED_EMBEDDING_MODELS = PersistentConfig( + "DOWNLOADED_EMBEDDING_MODELS", + "rag.downloaded_embedding_models", + os.getenv("DOWNLOADED_EMBEDDING_MODELS", {"":["sentence-transformers/all-MiniLM-L6-v2"], + "openai":["text-embedding-3-small"], + "ollama":[]}) +) + +DOWNLOADED_RERANKING_MODELS = PersistentConfig( + "DOWNLOADED_RERANKING_MODELS", + "rag.downloaded_reranking_models", + os.getenv("DOWNLOADED_RERANKING_MODELS", []) +) + +LOADED_EMBEDDING_MODELS = PersistentConfig( + "LOADED_EMBEDDING_MODELS", + "rag.loaded_embedding_models", + os.getenv("LOADED_EMBEDDING_MODELS", {"":[], + "openai":[], + "ollama":[]}) +) + +LOADED_RERANKING_MODELS = PersistentConfig( + "LOADED_RERANKING_MODELS", + "rag.loaded_reranking_models", + os.getenv("LOADED_RERANKING_MODELS", []) +) + + #################################### # Web Search (RAG) ####################################