diff --git a/src/novelai_python/sdk/ai/generate_image/__init__.py b/src/novelai_python/sdk/ai/generate_image/__init__.py
index 8eff6fb..cd896ef 100755
--- a/src/novelai_python/sdk/ai/generate_image/__init__.py
+++ b/src/novelai_python/sdk/ai/generate_image/__init__.py
@@ -65,6 +65,7 @@ def set_mutual_exclusion(self, value: bool):
     model: ModelTypeAlias = "nai-diffusion-3"
     action: Union[str, Action] = Field(Action.GENERATE, description="Mode for img generate")
     parameters: Union[Params]
+    use_new_shared_trial: bool = False
     model_config = ConfigDict(extra="ignore")
 
     @override
@@ -422,6 +423,7 @@ def build_generate(
             decrisp_mode: bool = None,
             variety_boost: bool = None,
             furry_mode: bool = None,
+            use_trial: bool = False
     ):
         """
         Quickly construct a parameter class that meets the requirements.
@@ -492,7 +494,8 @@ def build_generate(
             input=prompt,
             model=model,
             action=Action.GENERATE,
-            parameters=params
+            parameters=params,
+            use_new_shared_trial=use_trial
         )
 
     @staticmethod