diff --git a/sdk/python/kfp/components/_structures.py b/sdk/python/kfp/components/_structures.py
index 2afc2be562..ea766a3b60 100644
--- a/sdk/python/kfp/components/_structures.py
+++ b/sdk/python/kfp/components/_structures.py
@@ -80,6 +80,7 @@ class InputSpec(ModelBase):
         description: Optional[str] = None,
         default: Optional[PrimitiveTypes] = None,
         optional: Optional[bool] = False,
+        annotations: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(locals())
 
@@ -90,6 +91,7 @@ class OutputSpec(ModelBase):
         name: str,
         type: Optional[TypeSpecType] = None,
         description: Optional[str] = None,
+        annotations: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(locals())