diff --git a/scaffoldforge/templates/python/default/models.py.j2 b/scaffoldforge/templates/python/default/models.py.j2 index d65ecd9..8065859 100644 --- a/scaffoldforge/templates/python/default/models.py.j2 +++ b/scaffoldforge/templates/python/default/models.py.j2 @@ -3,12 +3,14 @@ Generated from GitHub Issue #{{ issue_number }} """ - -class DataModel: - """TODO: Implement data model class.""" - pass +from dataclasses import dataclass -class AnotherModel: - """TODO: Implement another model class.""" - pass +@dataclass +class Item: + """TODO: Define item model.""" + name: str + description: str = "" + + +# TODO: Add more models based on requirements