diff --git a/libs/langchain/langchain/embeddings/base.py b/libs/langchain/langchain/embeddings/base.py index 2f1f832f3bc..986634c0d30 100644 --- a/libs/langchain/langchain/embeddings/base.py +++ b/libs/langchain/langchain/embeddings/base.py @@ -13,6 +13,7 @@ _SUPPORTED_PROVIDERS = { "google_vertexai": "langchain_google_vertexai", "huggingface": "langchain_huggingface", "mistralai": "langchain_mistralai", + "ollama": "langchain_ollama", "openai": "langchain_openai", } @@ -209,6 +210,10 @@ def init_embeddings( from langchain_huggingface import HuggingFaceEmbeddings return HuggingFaceEmbeddings(model_name=model_name, **kwargs) + elif provider == "ollama": + from langchain_ollama import OllamaEmbeddings + + return OllamaEmbeddings(model=model_name, **kwargs) else: raise ValueError( f"Provider '{provider}' is not supported.\n"