Skip to content

Commit 61ad2e5

Browse files
chore(guardrails): type the masked litellm_params dicts and ratchet the basedpyright budget
1 parent d83ee02 commit 61ad2e5

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

basedpyright-code-budget.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@
9999
"limit": 0
100100
},
101101
"reportUnknownArgumentType": {
102-
"limit": 44655
102+
"limit": 44526
103103
},
104104
"reportUnknownLambdaType": {
105105
"limit": 109

litellm/proxy/guardrails/guardrail_endpoints.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ async def list_guardrails_v2(
264264
if isinstance(litellm_params, LitellmParams)
265265
else litellm_params
266266
) or {}
267-
masked_litellm_params_dict = _get_masked_values(
267+
masked_litellm_params_dict: dict[str, object] = _get_masked_values(
268268
litellm_params_dict,
269269
unmasked_length=4,
270270
number_of_asterisks=4,
@@ -303,7 +303,7 @@ async def list_guardrails_v2(
303303
if isinstance(in_memory_litellm_params_raw, LitellmParams)
304304
else in_memory_litellm_params_raw
305305
) or {}
306-
masked_in_memory_litellm_params = _get_masked_values(
306+
masked_in_memory_litellm_params: dict[str, object] = _get_masked_values(
307307
in_memory_litellm_params_dict,
308308
unmasked_length=4,
309309
number_of_asterisks=4,
@@ -1320,7 +1320,7 @@ async def get_guardrail_info(guardrail_id: str):
13201320
if isinstance(litellm_params, LitellmParams)
13211321
else litellm_params
13221322
) or {}
1323-
masked_litellm_params_dict: Final = _get_masked_values(
1323+
masked_litellm_params_dict: Final[dict[str, object]] = _get_masked_values(
13241324
result_litellm_params_dict,
13251325
unmasked_length=4,
13261326
number_of_asterisks=4,

0 commit comments

Comments
 (0)