fix(test): 'get_model'-> 'get_language_model'

Author: EmanueleDeRossi1

tests/sdk/metrics/test_metric_scope.py

@@ -96,9 +96,9 @@ class TestMetricWithScope:
 
     def test_numeric_judge_with_metric_scope(self):
         """Test NumericJudge with metric_scope."""
-        with patch("rhesis.sdk.models.factory.get_model") as mock_get_model:
+        with patch("rhesis.sdk.models.factory.get_language_model") as mock_get_language_model:
             mock_model = Mock()
-            mock_get_model.return_value = mock_model
+            mock_get_language_model.return_value = mock_model
 
             metric = NumericJudge(
                 name="test_numeric",
@@ -112,9 +112,9 @@ def test_numeric_judge_with_metric_scope(self):
 
     def test_numeric_judge_to_config_includes_metric_scope(self):
         """Test that NumericJudge.to_config() includes metric_scope."""
-        with patch("rhesis.sdk.models.factory.get_model") as mock_get_model:
+        with patch("rhesis.sdk.models.factory.get_language_model") as mock_get_language_model:
             mock_model = Mock()
-            mock_get_model.return_value = mock_model
+            mock_get_language_model.return_value = mock_model
 
             metric = NumericJudge(
                 name="test_numeric",
@@ -138,9 +138,9 @@ def test_push_metric_with_scope(self, mock_client_class):
         mock_client = Mock()
         mock_client_class.return_value = mock_client
 
-        with patch("rhesis.sdk.models.factory.get_model") as mock_get_model:
+        with patch("rhesis.sdk.models.factory.get_language_model") as mock_get_language_model:
             mock_model = Mock()
-            mock_get_model.return_value = mock_model
+            mock_get_language_model.return_value = mock_model
 
             metric = NumericJudge(
                 name="test_metric",
@@ -182,9 +182,9 @@ def test_pull_metric_with_scope(self, mock_client_class):
         ]
         mock_client.send_request.return_value = backend_response
 
-        with patch("rhesis.sdk.models.factory.get_model") as mock_get_model:
+        with patch("rhesis.sdk.models.factory.get_language_model") as mock_get_language_model:
             mock_model = Mock()
-            mock_get_model.return_value = mock_model
+            mock_get_language_model.return_value = mock_model
 
             # Mock the from_config method to avoid actual instantiation issues
             with patch.object(NumericJudge, "from_config") as mock_from_config:

tests/sdk/services/mcp/test_agent.py

@@ -65,14 +65,14 @@ def test_agent_initialization_without_client_raises(self, mock_model):
 
     def test_agent_initialization_with_string_model(self, mock_mcp_client):
         """Test agent initialization with string model name"""
-        with patch("rhesis.sdk.services.mcp.agent.get_model") as mock_get_model:
+        with patch("rhesis.sdk.services.mcp.agent.get_language_model") as mock_get_language_model:
             mock_model = Mock()
-            mock_get_model.return_value = mock_model
+            mock_get_language_model.return_value = mock_model
 
             agent = MCPAgent(model="gpt-4", mcp_client=mock_mcp_client)
 
             assert agent.model == mock_model
-            mock_get_model.assert_called_once_with("gpt-4")
+            mock_get_language_model.assert_called_once_with("gpt-4")
 
     def test_agent_loads_default_system_prompt(self, mock_mcp_client, mock_model):
         """Test agent loads default system prompt"""