refactor: Update OllamaEmbeddingsComponent to use new Inputs/Outputs format

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

@@ -1,34 +1,46 @@
 from typing import Optional
 
 from langchain_community.embeddings import OllamaEmbeddings
-from langchain_core.embeddings import Embeddings
+from langflow.base.models.model import LCModelComponent
-
+from langflow.field_typing import Embeddings
-from langflow.custom import CustomComponent
+from langflow.io import FloatInput, Output, TextInput
 
 
-class OllamaEmbeddingsComponent(CustomComponent):
+class OllamaEmbeddingsComponent(LCModelComponent):
     display_name: str = "Ollama Embeddings"
     description: str = "Generate embeddings using Ollama models."
     documentation = "https://python.langchain.com/docs/integrations/text_embedding/ollama"
+    icon = "Ollama"
 
-    def build_config(self):
+    inputs = [
-        return {
+        TextInput(
-            "model": {
+            name="model",
-                "display_name": "Ollama Model",
+            display_name="Ollama Model",
-            },
+            value="llama2",
-            "base_url": {"display_name": "Ollama Base URL"},
+        ),
-            "temperature": {"display_name": "Model Temperature"},
+        TextInput(
-            "code": {"show": False},
+            name="base_url",
-        }
+            display_name="Ollama Base URL",
+            value="http://localhost:11434",
+        ),
+        FloatInput(
+            name="temperature",
+            display_name="Model Temperature",
+            advanced=True,
+        ),
+    ]
 
-    def build(
+    outputs = [
-        self,
+        Output(display_name="Embeddings", name="embeddings", method="build_embeddings"),
-        model: str = "llama2",
+    ]
-        base_url: str = "http://localhost:11434",
+
-        temperature: Optional[float] = None,
+    def build_embeddings(self) -> Embeddings:
-    ) -> Embeddings:
         try:
-            output = OllamaEmbeddings(model=model, base_url=base_url, temperature=temperature)  # type: ignore
+            output = OllamaEmbeddings(
+                model=self.model,
+                base_url=self.base_url,
+                temperature=self.temperature,
+            )  # type: ignore
         except Exception as e:
             raise ValueError("Could not connect to Ollama API.") from e
         return output