diff --git a/litellm/llms/vertex_ai/agent_engine/sse_iterator.py b/litellm/llms/vertex_ai/agent_engine/sse_iterator.py index 66c5b220870..4b03556e1f6 100644 --- a/litellm/llms/vertex_ai/agent_engine/sse_iterator.py +++ b/litellm/llms/vertex_ai/agent_engine/sse_iterator.py @@ -44,6 +44,10 @@ class VertexAgentEngineResponseIterator(BaseModelResponseIterator): "BLOCKLIST", "PROHIBITED_CONTENT", "SPII", + "LANGUAGE", + "OTHER", + "IMAGE_SAFETY", + "IMAGE_PROHIBITED_CONTENT", } )