scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-29 06:33:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 201

fix(guardrails): preserve the existing stream sink value types

Browse source
This commit is contained in:
albertbausili 2026-10-16 09:43:10 +02:00
parent 95348ba610
commit c99cc59dda
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/base_llm/guardrail_translation/base_translation.py
View file

@ -31,12 +31,12 @@ class StreamTransformSink:
single call, not shared state.
"""
mutated_text_per_choice: Mapping[int, str] = field(default_factory=dict)
holdback_per_choice: Mapping[int, int] = field(default_factory=dict)
mutated_text_per_choice: dict[int, str] = field(default_factory=dict)
holdback_per_choice: dict[int, int] = field(default_factory=dict)
def record(self, texts: Mapping[int, str], holdback: Mapping[int, int]) -> None:
self.mutated_text_per_choice = texts
self.holdback_per_choice = holdback
self.mutated_text_per_choice = dict(texts)
self.holdback_per_choice = dict(holdback)
@dataclass(frozen=True, slots=True)