From f80fe7d4c1e8f1bb4e4db62f429db8c9d97a6403 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Fri, 21 Feb 2025 11:53:58 +0000 Subject: [PATCH] fix: use unquoted type names in model descriptions Co-Authored-By: Joe Moura --- src/crewai/utilities/converter.py | 4 ++-- tests/utilities/test_converter.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/crewai/utilities/converter.py b/src/crewai/utilities/converter.py index 108bad34d..79f979d3f 100644 --- a/src/crewai/utilities/converter.py +++ b/src/crewai/utilities/converter.py @@ -327,9 +327,9 @@ def generate_model_description(model: Type[BaseModel]) -> str: elif isinstance(field_type, type) and issubclass(field_type, BaseModel): return generate_model_description(field_type) elif hasattr(field_type, "__name__"): - return f'"{field_type.__name__}"' + return field_type.__name__ else: - return f'"{str(field_type)}"' + return str(field_type) fields = model.model_fields field_descriptions = [] diff --git a/tests/utilities/test_converter.py b/tests/utilities/test_converter.py index 0ed2ad666..dfee01548 100644 --- a/tests/utilities/test_converter.py +++ b/tests/utilities/test_converter.py @@ -369,7 +369,7 @@ def test_generate_model_description_with_empty_description(): age: int = Field(..., description=None) description = generate_model_description(ModelWithEmptyDescription) - expected = '{\n "name": "str",\n "age": "int"\n}' + expected = '{\n "name": str,\n "age": int\n}' assert description == expected