feat(lasso): add tool-calling support to LassoGuardrail - #27648

Merged

krrish-berri-2 merged 15 commits into

BerriAI:litellm_internal_stagingfrom

lasso-security:rnd-5748-lasso-tool-calling

May 14, 2026

Commits on May 14, 2026

feat(lasso): extend LassoGuardrail to support tool calling (RND-5748)
vladpolevoi
committed
316dc8b
fix(lasso): PR review followups for tool-calling guardrail (RND-5748)
vladpolevoi
committed
ba1a73b
fix(lasso): handle object-style tool_calls in _update_tool_calls_from_masked (RND-5748)
vladpolevoi
committed
5091beb
fix(lasso): use model role for tool_use blocks (RND-5748)
vladpolevoi
committed
ca93fbd
test(lasso): add round-trip tests for message transformation (RND-5748)
vladpolevoi
committed
d1911e6
fix(lasso): remove unused imports, handle Responses-API input masking, flatten multimodal content (RND-5748)
vladpolevoi
committed
e446244
fix(lasso): inspect Responses-API input field (RND-5748)
vladpolevoi
committed
8d938b0
fix(lasso): guard text-cursor remap against Lasso count mismatch (RND-5748)
vladpolevoi
committed
e2b1085
fix(lasso): flatten list content in tool_result.content (RND-5748)
vladpolevoi
committed
1e98338
fix(lasso): remap multimodal list content during masking (RND-5748)
vladpolevoi
committed
33189df
refactor(lasso): extract _get_field and _extract_tool_call_fields helpers (RND-5748)
vladpolevoi
committed
b945328
fix(lasso): add count guard to _apply_masking_to_model_response (RND-5748)
vladpolevoi
committed
a4b7f85
fix(lasso): close two guardrail-bypass paths flagged in review (RND-5748)
vladpolevoi
committed
78332d8
chore(lasso): black formatting + comment on expand skip branch (RND-5748)
vladpolevoi
committed
87d92ca
fix(lasso): satisfy mypy in _handle_masking, _update_tool_calls_from_masked, _apply_masking_to_model_response (RND-5748)
vladpolevoi
committed
c26bbb2

Navigation Menu

Uh oh!

feat(lasso): add tool-calling support to LassoGuardrail - #27648

feat(lasso): add tool-calling support to LassoGuardrail#27648
krrish-berri-2 merged 15 commits into
BerriAI:litellm_internal_stagingfrom
lasso-security:rnd-5748-lasso-tool-calling

Commits on May 14, 2026

feat(lasso): extend LassoGuardrail to support tool calling (RND-5748)

fix(lasso): PR review followups for tool-calling guardrail (RND-5748)

fix(lasso): handle object-style tool_calls in _update_tool_calls_from_masked (RND-5748)

fix(lasso): use model role for tool_use blocks (RND-5748)

test(lasso): add round-trip tests for message transformation (RND-5748)

fix(lasso): remove unused imports, handle Responses-API input masking, flatten multimodal content (RND-5748)

fix(lasso): inspect Responses-API input field (RND-5748)

fix(lasso): guard text-cursor remap against Lasso count mismatch (RND-5748)

fix(lasso): flatten list content in tool_result.content (RND-5748)

fix(lasso): remap multimodal list content during masking (RND-5748)

refactor(lasso): extract _get_field and _extract_tool_call_fields helpers (RND-5748)

fix(lasso): add count guard to _apply_masking_to_model_response (RND-5748)

fix(lasso): close two guardrail-bypass paths flagged in review (RND-5748)

chore(lasso): black formatting + comment on expand skip branch (RND-5748)

fix(lasso): satisfy mypy in _handle_masking, _update_tool_calls_from_masked, _apply_masking_to_model_response (RND-5748)