diff --git a/src/celeste/modalities/audio/parameters.py b/src/celeste/modalities/audio/parameters.py
index 6589a37..82a5718 100644
--- a/src/celeste/modalities/audio/parameters.py
+++ b/src/celeste/modalities/audio/parameters.py
@@ -15,7 +15,7 @@ class AudioParameter(StrEnum):
     LANGUAGE = "language"
 
 
-class AudioParameters(Parameters):
+class AudioParameters(Parameters, total=False):
     """Parameters for audio operations."""
 
     voice: str
diff --git a/src/celeste/modalities/embeddings/parameters.py b/src/celeste/modalities/embeddings/parameters.py
index 86da04e..fcbad6f 100644
--- a/src/celeste/modalities/embeddings/parameters.py
+++ b/src/celeste/modalities/embeddings/parameters.py
@@ -14,7 +14,7 @@ class EmbeddingsParameter(StrEnum):
     AUDIO = "audio"
 
 
-class EmbeddingsParameters(Parameters):
+class EmbeddingsParameters(Parameters, total=False):
     """Parameters for embeddings operations."""
 
     dimensions: int | None
diff --git a/src/celeste/modalities/images/parameters.py b/src/celeste/modalities/images/parameters.py
index 9b60f9b..655d06f 100644
--- a/src/celeste/modalities/images/parameters.py
+++ b/src/celeste/modalities/images/parameters.py
@@ -26,7 +26,7 @@ class ImageParameter(StrEnum):
     THINKING_LEVEL = "thinking_level"
 
 
-class ImageParameters(Parameters):
+class ImageParameters(Parameters, total=False):
     """Parameters for images operations."""
 
     aspect_ratio: str
diff --git a/src/celeste/modalities/text/parameters.py b/src/celeste/modalities/text/parameters.py
index 1278783..93352b0 100644
--- a/src/celeste/modalities/text/parameters.py
+++ b/src/celeste/modalities/text/parameters.py
@@ -41,7 +41,7 @@ class TextParameter(StrEnum):
     DOCUMENT = "document"
 
 
-class TextParameters(Parameters):
+class TextParameters(Parameters, total=False):
     """Parameters for text operations."""
 
     # Common parameters
diff --git a/src/celeste/modalities/videos/parameters.py b/src/celeste/modalities/videos/parameters.py
index c4680e4..e69c45f 100644
--- a/src/celeste/modalities/videos/parameters.py
+++ b/src/celeste/modalities/videos/parameters.py
@@ -17,7 +17,7 @@ class VideoParameter(StrEnum):
     LAST_FRAME = "last_frame"
 
 
-class VideoParameters(Parameters):
+class VideoParameters(Parameters, total=False):
     """Parameters for video generation operations."""
 
     aspect_ratio: str
diff --git a/templates/modalities/{modality_slug}/parameters.py.template b/templates/modalities/{modality_slug}/parameters.py.template
index a97a721..2d37b01 100644
--- a/templates/modalities/{modality_slug}/parameters.py.template
+++ b/templates/modalities/{modality_slug}/parameters.py.template
@@ -32,7 +32,7 @@ class {Modality}Parameter(StrEnum):
     # AUDIO = "audio"
 
 
-class {Modality}Parameters(Parameters):
+class {Modality}Parameters(Parameters, total=False):
     """Parameters for {modality} operations."""
 
     # Common parameters