From 7762404e6ffd7a1a52299bb5dcbb737f4bc28f3f Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Wed, 12 Jun 2024 10:45:29 -0300
Subject: [PATCH] refactor: Update OllamaModel.py to use field_type instead of
 type for input definition

---
 .../langflow/components/models/OllamaModel.py | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/src/backend/base/langflow/components/models/OllamaModel.py b/src/backend/base/langflow/components/models/OllamaModel.py
index 12db058c2..2c7d3bae7 100644
--- a/src/backend/base/langflow/components/models/OllamaModel.py
+++ b/src/backend/base/langflow/components/models/OllamaModel.py
@@ -70,14 +70,14 @@ class ChatOllamaComponent(LCModelComponent):
     inputs = [
         Input(
             name="base_url",
-            type=Optional[str],
+            field_type=Optional[str],
             display_name="Base URL",
             info="Endpoint of the Ollama API. Defaults to 'http://localhost:11434' if not specified.",
             value="http://localhost:11434",
         ),
         Input(
             name="model",
-            type=str,
+            field_type=str,
             display_name="Model Name",
             options=[],  # This should be dynamically loaded if possible
             info="Refer to https://ollama.ai/library for more models.",
@@ -86,7 +86,7 @@ class ChatOllamaComponent(LCModelComponent):
         ),
         Input(
             name="mirostat",
-            type=str,
+            field_type=str,
             display_name="Mirostat",
             options=["Disabled", "Mirostat", "Mirostat 2.0"],
             info="Enable/disable Mirostat sampling for controlling perplexity.",
@@ -97,7 +97,7 @@ class ChatOllamaComponent(LCModelComponent):
         ),
         Input(
             name="mirostat_eta",
-            type=Optional[float],
+            field_type=Optional[float],
             display_name="Mirostat Eta",
             info="Learning rate for Mirostat algorithm.",
             advanced=True,
@@ -106,7 +106,7 @@ class ChatOllamaComponent(LCModelComponent):
         ),
         Input(
             name="mirostat_tau",
-            type=Optional[float],
+            field_type=Optional[float],
             display_name="Mirostat Tau",
             info="Controls the balance between coherence and diversity of the output.",
             advanced=True,
@@ -115,7 +115,7 @@ class ChatOllamaComponent(LCModelComponent):
         ),
         Input(
             name="temperature",
-            type=float,
+            field_type=float,
             display_name="Temperature",
             info="Controls the creativity of model responses.",
             value=0.8,
@@ -124,7 +124,7 @@ class ChatOllamaComponent(LCModelComponent):
         Input(name="stream", type=bool, display_name="Stream", info=STREAM_INFO_TEXT, value=False),
         Input(
             name="system_message",
-            type=Optional[str],
+            field_type=Optional[str],
             display_name="System Message",
             info="System message to pass to the model.",
             advanced=True,
@@ -132,14 +132,14 @@ class ChatOllamaComponent(LCModelComponent):
         ),
         Input(
             name="headers",
-            type=dict,
+            field_type=dict,
             display_name="Headers",
             info="Additional headers to send with the request.",
             advanced=True,
         ),
         Input(
             name="keep_alive_flag",
-            type=str,
+            field_type=str,
             display_params=["Keep", "Immediately", "Minute", "Hour", "sec"],
             display_name="Unload interval",
             info="Controls how the model unload interval is managed.",
@@ -148,7 +148,7 @@ class ChatOllamaComponent(LCModelComponent):
         ),
         Input(
             name="keep_alive",
-            type=int,
+            field_type=int,
             display_name="Interval",
             info="How long the model will stay loaded into memory.",
             value=None,