mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-07 02:59:05 +00:00
test(guardrails): type the recorder hook's request_data as a Mapping
This commit is contained in:
parent
2e83871d54
commit
19d77e2442
1 changed files with 2 additions and 1 deletions
|
|
@ -6,6 +6,7 @@ with guardrail transformations, including tool calls.
|
|||
"""
|
||||
|
||||
import json
|
||||
from collections.abc import Mapping
|
||||
from typing import Any, Literal, Optional
|
||||
|
||||
import pytest
|
||||
|
|
@ -1390,7 +1391,7 @@ class TestOpenAIChatCompletionsHandlerStreamingOutput:
|
|||
async def apply_guardrail(
|
||||
self,
|
||||
inputs: GenericGuardrailAPIInputs,
|
||||
request_data: dict,
|
||||
request_data: Mapping[str, object],
|
||||
input_type: Literal["request", "response"],
|
||||
logging_obj: object = None,
|
||||
) -> GenericGuardrailAPIInputs:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue