fix: add default models to Anthropic and make sure template is updated (#5839)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

src/backend/tests/unit/components/agents/test_agent_component.py

@@ -1,8 +1,99 @@
 import os
+from typing import Any
+from uuid import uuid4
 
 import pytest
+from langflow.base.models.model_input_constants import MODEL_PROVIDERS_DICT
 from langflow.components.agents.agent import AgentComponent
 from langflow.components.tools.calculator import CalculatorToolComponent
+from langflow.custom import Component
+from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI
+
+from tests.base import ComponentTestBaseWithoutClient
+from tests.unit.mock_language_model import MockLanguageModel
+
+
+class TestAgentComponent(ComponentTestBaseWithoutClient):
+    @pytest.fixture
+    def component_class(self):
+        return AgentComponent
+
+    @pytest.fixture
+    def file_names_mapping(self):
+        return []
+
+    def component_setup(self, component_class: type[Any], default_kwargs: dict[str, Any]) -> Component:
+        component_instance = super().component_setup(component_class, default_kwargs)
+        # Mock _should_process_output method
+        component_instance._should_process_output = lambda output: False  # noqa: ARG005
+        return component_instance
+
+    @pytest.fixture
+    def default_kwargs(self):
+        return {
+            "_type": "Agent",
+            "add_current_date_tool": True,
+            "agent_description": "A helpful agent",
+            "agent_llm": MockLanguageModel(),
+            "handle_parsing_errors": True,
+            "input_value": "",
+            "max_iterations": 10,
+            "system_prompt": "You are a helpful assistant.",
+            "tools": [],
+            "verbose": True,
+            "session_id": str(uuid4()),
+            "sender": MESSAGE_SENDER_AI,
+            "sender_name": MESSAGE_SENDER_NAME_AI,
+        }
+
+    async def test_build_config_update(self, component_class, default_kwargs):
+        component = self.component_setup(component_class, default_kwargs)
+        frontend_node = component.to_frontend_node()
+        build_config = frontend_node["data"]["node"]["template"]
+        # Test updating build config for OpenAI
+        component.set(agent_llm="OpenAI")
+        updated_config = await component.update_build_config(build_config, "OpenAI", "agent_llm")
+        assert "agent_llm" in updated_config
+        assert updated_config["agent_llm"]["value"] == "OpenAI"
+        assert isinstance(updated_config["agent_llm"]["options"], list)
+        assert len(updated_config["agent_llm"]["options"]) > 0
+        assert all(provider in updated_config["agent_llm"]["options"] for provider in MODEL_PROVIDERS_DICT)
+        assert "Custom" in updated_config["agent_llm"]["options"]
+
+        # Verify model_name field is populated for OpenAI
+
+        assert "model_name" in updated_config
+        model_name_dict = updated_config["model_name"]
+        assert isinstance(model_name_dict["options"], list)
+        assert len(model_name_dict["options"]) > 0  # OpenAI should have available models
+        assert "gpt-4o" in model_name_dict["options"]
+
+        # Test Anthropic
+        component.set(agent_llm="Anthropic")
+        updated_config = await component.update_build_config(build_config, "Anthropic", "agent_llm")
+        assert "agent_llm" in updated_config
+        assert updated_config["agent_llm"]["value"] == "Anthropic"
+        assert isinstance(updated_config["agent_llm"]["options"], list)
+        assert len(updated_config["agent_llm"]["options"]) > 0
+        assert all(provider in updated_config["agent_llm"]["options"] for provider in MODEL_PROVIDERS_DICT)
+        assert "Anthropic" in updated_config["agent_llm"]["options"]
+        assert updated_config["agent_llm"]["input_types"] == []
+        assert any("sonnet" in option.lower() for option in updated_config["model_name"]["options"]), (
+            f"Options: {updated_config['model_name']['options']}"
+        )
+
+        # Test updating build config for Custom
+        updated_config = await component.update_build_config(build_config, "Custom", "agent_llm")
+        assert "agent_llm" in updated_config
+        assert updated_config["agent_llm"]["value"] == "Custom"
+        assert isinstance(updated_config["agent_llm"]["options"], list)
+        assert len(updated_config["agent_llm"]["options"]) > 0
+        assert all(provider in updated_config["agent_llm"]["options"] for provider in MODEL_PROVIDERS_DICT)
+        assert "Custom" in updated_config["agent_llm"]["options"]
+        assert updated_config["agent_llm"]["input_types"] == ["LanguageModel"]
+
+        # Verify model_name field is cleared for Custom
+        assert "model_name" not in updated_config
 
 
 @pytest.mark.api_key_required