diff --git a/veadk/tools/builtin_tools/llm_shield.py b/veadk/tools/builtin_tools/llm_shield.py
index 76317bfe..29978305 100644
--- a/veadk/tools/builtin_tools/llm_shield.py
+++ b/veadk/tools/builtin_tools/llm_shield.py
@@ -266,7 +266,8 @@ def before_model_callback(
                 content=types.Content(
                     role="model",
                     parts=[types.Part(text=response)],
-                )
+                ),
+                partial=True,
             )
         return None
 
@@ -313,7 +314,8 @@ def after_model_callback(
                 content=types.Content(
                     role="model",
                     parts=[types.Part(text=response)],
-                )
+                ),
+                partial=True,
             )
         return None