fix comment, remove unused var, add backward-compat alias for _build_guardrail_inputs

This commit is contained in:
joereyna 2026-04-16 21:31:30 -07:00
parent 91ac86ec17
commit 2c3ab41248
No known key found for this signature in database
GPG key ID: 37E09E2BDB5920E5
2 changed files with 5 additions and 2 deletions

View file

@ -245,8 +245,8 @@ class A2AGuardrailHandler(BaseTranslation):
# Collect text from each chunk in order (by original index in responses_so_far)
text_parts: List[str] = []
chunk_indices_with_text: List[int] = [] # indices into valid_parsed
for idx, (orig_i, obj) in enumerate(valid_parsed):
chunk_indices_with_text: List[int] = [] # indices into responses_so_far
for orig_i, obj in valid_parsed:
t = extract_text_from_a2a_response(obj)
if t:
text_parts.append(t)

View file

@ -408,6 +408,9 @@ class AnthropicMessagesHandler(BaseTranslation):
inputs["model"] = response_model
return inputs
# Backward-compat alias — subclasses that overrode _build_guardrail_inputs still work
_build_guardrail_inputs = _build_output_guardrail_inputs
async def process_output_streaming_response(
self,
responses_so_far: List[Any],