Merge pull request #414 from BhAem/fix/guardrail

fix: allow streaming in guardrail response

Author: warm-wm

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