Handled upgraded genai library

Author: carsongee

template/{{agent_app_name}}/tests/test_mcp_templates/test_mcp_crewai.j2

@@ -175,7 +175,7 @@ class TestMyAgentMCPIntegration:
             assert len(writer.tools) == 1  # 1 MCP tool
             assert len(editor.tools) == 1  # 1 MCP tool
 
-    @patch("datarobot_genai.crewai.base.mcp_tools_context")
+    @patch("datarobot_genai.crewai.base.mcp_tools_context", autospec=True)
     def test_agent_works_without_mcp_tools(
         self, mock_mcp_tools_context, crewai_common_mocks
     ):
@@ -232,7 +232,7 @@ class TestMyAgentMCPIntegration:
 
         assert access_count["count"] >= 3
 
-    @patch("datarobot_genai.crewai.base.mcp_tools_context")
+    @patch("datarobot_genai.crewai.base.mcp_tools_context", autospec=True)
     def test_mcp_tool_execution_makes_request_to_server(
         self, mock_mcp_tools_context, crewai_common_mocks
     ):

template/{{agent_app_name}}/tests/test_mcp_templates/test_mcp_langgraph.j2

@@ -71,7 +71,10 @@ def langgraph_common_mocks():
     mock_execution_graph.astream.side_effect = _create_mock_astream()
 
     with (
-        patch("datarobot_genai.langgraph.agent.mcp_tools_context") as mock_mcp_context,
+        patch(
+            "datarobot_genai.langgraph.agent.mcp_tools_context",
+            autospec=True,
+        ) as mock_mcp_context,
         patch.object(
             MyAgent,
             "workflow",
@@ -120,7 +123,8 @@ class TestMyAgentLangGraphMCPIntegration:
             except (StopIteration, AttributeError, TypeError, ValueError):
                 pass
 
-            mock_context.assert_called_once_with(authorization_context={})
+            mock_context.assert_called_once()
+            assert mock_context.call_args.kwargs.get("authorization_context") == {}
             assert agent.mcp_tools == mock_tools
 
     def test_agent_loads_mcp_tools_from_datarobot_deployment_in_invoke(
@@ -155,7 +159,8 @@ class TestMyAgentLangGraphMCPIntegration:
             except (StopIteration, AttributeError, TypeError, ValueError):
                 pass
 
-            mock_context.assert_called_once_with(authorization_context={})
+            mock_context.assert_called_once()
+            assert mock_context.call_args.kwargs.get("authorization_context") == {}
             assert agent.mcp_tools == mock_tools
 
     def test_agent_works_without_mcp_tools(self, langgraph_common_mocks):

template/{{agent_app_name}}/tests/test_mcp_templates/test_mcp_llamaindex.j2

@@ -110,8 +110,12 @@ class TestMyAgentMCPIntegration:
                 # Expected when workflow is mocked
                 pass
 
-            # Verify load_mcp_tools was called with correct parameters
-            mock_load_mcp_tools.assert_called_once_with(authorization_context={})
+            # Verify load_mcp_tools was called and provided authorization_context
+            mock_load_mcp_tools.assert_called_once()
+            assert (
+                mock_load_mcp_tools.call_args.kwargs.get("authorization_context")
+                == {}
+            )
 
             # Verify set_mcp_tools was called with the tools from MCP server
             assert agent.mcp_tools == mock_tools
@@ -164,8 +168,12 @@ class TestMyAgentMCPIntegration:
                 # Expected when workflow is mocked
                 pass
 
-            # Verify load_mcp_tools was called with correct parameters
-            mock_load_mcp_tools.assert_called_once_with(authorization_context={})
+            # Verify load_mcp_tools was called and provided authorization_context
+            mock_load_mcp_tools.assert_called_once()
+            assert (
+                mock_load_mcp_tools.call_args.kwargs.get("authorization_context")
+                == {}
+            )
 
             # Verify set_mcp_tools was called with the tools from MCP server
             assert agent.mcp_tools == mock_tools