Add test for validating model output without code block markers

crewAIInc · Nov 22, 2024 · 948a79d · 948a79d
1 parent 3ee8165
commit 948a79d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/utilities/test_converter.py b/tests/utilities/test_converter.py
@@ -142,6 +142,13 @@ def test_validate_model_json_output():
     assert output == {"name": "Bob", "age": 40}
 
 
+def test_validate_model_remove_code_block_markers():
+    result = '```json {"name": "Charlie", "age": 35}```'
+    output = validate_model(result, SimpleModel, True, None)
+    assert isinstance(output, dict)
+    assert output == {"name": "Charlie", "age": 35}
+
+
 # Tests for handle_partial_json
 def test_handle_partial_json_with_valid_partial():
     result = 'Some text {"name": "Charlie", "age": 35} more text'