fix: improve AllergyCommand definition

canvas_sdk/commands/commands/allergy.py

@@ -1,9 +1,24 @@
 from datetime import date
-from enum import Enum
+from enum import Enum, IntEnum
+
+from typing_extensions import TypedDict
 
 from canvas_sdk.commands.base import _BaseCommand as BaseCommand
 
 
+class AllergenType(IntEnum):
+    ALLERGEN_GROUP = 1
+    MEDICATION = 2
+    INGREDIENT = 6
+
+
+class Allergen(TypedDict):
+    """A TypedDict representing an allergen."""
+
+    concept_id: int
+    concept_type: AllergenType
+
+
 class AllergyCommand(BaseCommand):
     """A class for managing an Allergy command within a specific note."""
 
@@ -16,7 +31,7 @@ class Severity(Enum):
         MODERATE = "moderate"
         SEVERE = "severe"
 
-    allergy: str | None = None
+    allergy: Allergen | None = None
     severity: Severity | None = None
     narrative: str | None = None
     approximate_date: date | None = None