fix: fixed test typo

camel-ai · Feb 4, 2025 · cd7d105 · cd7d105
1 parent 0546514
commit cd7d105
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/models/test_cohere_model.py b/test/models/test_cohere_model.py
@@ -178,7 +178,7 @@ def test_cohere_model_run(mock_cohere, mock_record, model_type):
     "model_type",
     model_types,
 )
-@patch("camel.models.cohere_model.record")
+@patch("cohere.ClientV2")
 def test_cohere_model_run_unsupported_role(mock_cohere, model_type):
     # Mock the Cohere create client function
     mock_cohere_client = MagicMock()