Added OllamaEmbeddings component with documentation

src/backend/langflow/components/embeddings/OllamaEmbeddings.py

@@ -0,0 +1,41 @@
+from typing import Optional
+
+from langflow import CustomComponent
+from langchain.embeddings.base import Embeddings
+from langchain_community.embeddings import OllamaEmbeddings
+
+class OllamaEmbeddingsComponent(CustomComponent):
+    """
+    A custom component for implementing an Embeddings Model using Ollama.
+    """
+
+    display_name: str = "Ollama Embeddings"
+    description: str = "Embeddings model from Ollama."
+    documentation = "https://python.langchain.com/docs/integrations/text_embedding/ollama"
+    beta = True
+
+    def build_config(self):
+        return {
+            "model": {
+                "display_name": "Ollama Model",
+            },
+            "base_url": {"display_name": "Ollama Base URL"},
+            "temperature": {"display_name": "Model Temperature"},
+            "code": {"show": False},
+        }
+    
+    def build(
+        self,
+        model: str = "llama2",
+        base_url: str = "http://localhost:11434",
+        temperature: Optional[float] = None,
+    ) -> Embeddings:
+        try:
+            output = OllamaEmbeddings(
+                model=model,
+                base_url=base_url,
+                temperature=temperature
+            )  # type: ignore
+        except Exception as e:
+            raise ValueError("Could not connect to Ollama API.") from e
+        return output

src/backend/langflow/config.yaml

@@ -106,6 +106,8 @@ embeddings:
     documentation: "https://python.langchain.com/docs/modules/data_connection/text_embedding/integrations/google_vertex_ai_palm"
   AmazonBedrockEmbeddings:
     documentation: "https://python.langchain.com/docs/modules/data_connection/text_embedding/integrations/bedrock"
+  OllamaEmbeddings:
+    documentation: "https://python.langchain.com/docs/modules/data_connection/text_embedding/integrations/ollama"
 
 llms:
   OpenAI: