fix: add support in Agent to fix ollama

Author: edwinjosechittilappilly

src/lfx/src/lfx/_assets/component_index.json

@@ -101,6 +101,7 @@ class ChatOllamaComponent(LCModelComponent):
             info="Refer to https://ollama.com/library for more models.",
             refresh_button=True,
             real_time_refresh=True,
+            required=True,
         ),
         SecretStrInput(
             name="api_key",
@@ -125,6 +126,7 @@ class ChatOllamaComponent(LCModelComponent):
             advanced=False,
             table_schema=TABLE_SCHEMA,
             value=default_table_row,
+            show=False,
         ),
         DictInput(name="metadata", display_name="Metadata", info="Metadata to add to the run trace.", advanced=True),
         DropdownInput(
@@ -215,6 +217,14 @@ class ChatOllamaComponent(LCModelComponent):
         MessageTextInput(
             name="template", display_name="Template", info="Template to use for generating text.", advanced=True
         ),
+        BoolInput(
+            name="enable_structured_output",
+            display_name="Enable Structured Output",
+            info="Whether to enable structured output in the model.",
+            value=False,
+            advanced=True,
+            real_time_refresh=True,
+        ),
         *LCModelComponent.get_base_inputs(),
     ]
 
@@ -254,7 +264,7 @@ def build_model(self) -> LanguageModel:  # type: ignore[type-var]
             )
 
         try:
-            output_format = self._parse_format_field(self.format)
+            output_format = self._parse_format_field(self.format) if self.enable_structured_output else None
         except Exception as e:
             msg = f"Failed to parse the format field: {e}"
             raise ValueError(msg) from e
@@ -264,7 +274,7 @@ def build_model(self) -> LanguageModel:  # type: ignore[type-var]
             "base_url": transformed_base_url,
             "model": self.model_name,
             "mirostat": mirostat_value,
-            "format": output_format,
+            "format": output_format or None,
             "metadata": self.metadata,
             "tags": self.tags.split(",") if self.tags else None,
             "mirostat_eta": mirostat_eta,
@@ -319,6 +329,13 @@ async def is_valid_ollama_url(self, url: str) -> bool:
             return False
 
     async def update_build_config(self, build_config: dict, field_value: Any, field_name: str | None = None):
+        if field_name == "enable_structured_output":
+            if field_value:
+                build_config["format"]["show"] = True
+            else:
+                build_config["format"]["show"] = False
+            return build_config
+
         if field_name == "mirostat":
             if field_value == "Disabled":
                 build_config["mirostat_eta"]["advanced"] = True