scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

chore(guardrails): match main's type-discipline rules in generic_guardrail_api

Browse source 
Main dropped the LIT002 rule, so the mutable-ok comments on the JSON row builds no longer suppress anything and LIT013 now flags them. They are removed. The part-list guard narrows to Sequence[object] instead of list[object], since its callers only need len() and a read, which also keeps it clear of LIT001
This commit is contained in:
Caduri Katzav 2026-10-04 15:11:41 +03:00
parent 5657a8f6e2
commit d1f31a64ac
1 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/guardrails/guardrail_hooks/generic_guardrail_api/generic_guardrail_api.py
View file

@ -156,7 +156,7 @@ def _extract_inbound_headers(
return None
def _is_part_list(value: object) -> TypeIs[list[object]]: # guard-ok: trivial isinstance narrowing
def _is_part_list(value: object) -> TypeIs[Sequence[object]]: # guard-ok: trivial isinstance narrowing
return isinstance(value, list)
@ -177,15 +177,13 @@ def _row_as_sent(dumped: JsonValue, caller: Mapping[str, object]) -> JsonValue:
dumped_content: Final = dumped.get("content")
if isinstance(dumped_content, list) and len(dumped_content) == len(caller_content):
return dumped
return {**dumped, "content": _as_posted_json(caller_content)} # mutable-ok: a JsonValue object is a dict
return {**dumped, "content": _as_posted_json(caller_content)}
def _rows_as_sent(dumped_rows: JsonValue, caller_rows: Sequence[Mapping[str, object]] | None) -> JsonValue:
if caller_rows is None or not isinstance(dumped_rows, list):
return dumped_rows
return [ # mutable-ok: a JsonValue array is a list
_row_as_sent(dumped, caller) for dumped, caller in zip(dumped_rows, caller_rows, strict=True)
]
return [_row_as_sent(dumped, caller) for dumped, caller in zip(dumped_rows, caller_rows, strict=True)]
def _structured_rows_to_write_back(
@ -514,7 +512,7 @@ class GenericGuardrailAPI(CustomGuardrail):
# The model's list content is a lazy iterator that this dump consumes, so it cannot be read again
dumped: Final[Mapping[str, JsonValue]] = guardrail_request.model_dump(mode="json")
sent_messages: Final = _rows_as_sent(dumped.get("structured_messages"), structured_messages)
request_json: Final = {**dumped, "structured_messages": sent_messages} # mutable-ok: post() needs a dict
request_json: Final = {**dumped, "structured_messages": sent_messages}
response: Final = await self.async_handler.post(
url=self.api_base,