From 429eb656ab6dee2e7b6da2bb3a7fcba29030c0aa Mon Sep 17 00:00:00 2001
From: "wuqingfu.528" <wuqingfu.528@bytedance.com>
Date: Tue, 30 Dec 2025 10:13:58 +0800
Subject: [PATCH] fix: allow streaming in guardrail response

---
 veadk/tools/builtin_tools/llm_shield.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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