diff --git a/tests/unit/llms/anthropic/messages/adapters/test_anthropic_messages_adapters_transformation.py b/tests/unit/llms/anthropic/messages/adapters/test_anthropic_messages_adapters_transformation.py index 4b9d371255f..d7c763d18c5 100644 --- a/tests/unit/llms/anthropic/messages/adapters/test_anthropic_messages_adapters_transformation.py +++ b/tests/unit/llms/anthropic/messages/adapters/test_anthropic_messages_adapters_transformation.py @@ -5925,17 +5925,21 @@ class _FailureLog(CustomLogger): super().__init__() self.rejections: tuple[str, ...] = () - def _record(self, kwargs) -> None: + def _record(self, kwargs: Mapping[str, object]) -> None: error: Final = str(kwargs.get("exception")) self.rejections = ( *self.rejections, next((phrase for phrase in _UPSTREAM_REJECTION_PHRASES if phrase in error), error), ) - def log_failure_event(self, kwargs, response_obj, start_time, end_time) -> None: + def log_failure_event( + self, kwargs: Mapping[str, object], response_obj: object, start_time: object, end_time: object + ) -> None: self._record(kwargs) - async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time) -> None: + async def async_log_failure_event( + self, kwargs: Mapping[str, object], response_obj: object, start_time: object, end_time: object + ) -> None: self._record(kwargs) diff --git a/tests/unit/llms/anthropic/messages/responses_adapters/test_responses_adapters_handler.py b/tests/unit/llms/anthropic/messages/responses_adapters/test_responses_adapters_handler.py index 869b352e585..9890cb9b186 100644 --- a/tests/unit/llms/anthropic/messages/responses_adapters/test_responses_adapters_handler.py +++ b/tests/unit/llms/anthropic/messages/responses_adapters/test_responses_adapters_handler.py @@ -1233,17 +1233,21 @@ class _FailureLog(CustomLogger): super().__init__() self.rejections: tuple[str, ...] = () - def _record(self, kwargs) -> None: + def _record(self, kwargs: Mapping[str, object]) -> None: error: Final = str(kwargs.get("exception")) self.rejections = ( *self.rejections, next((phrase for phrase in ("encrypted content", "organization") if phrase in error), error), ) - def log_failure_event(self, kwargs, response_obj, start_time, end_time) -> None: + def log_failure_event( + self, kwargs: Mapping[str, object], response_obj: object, start_time: object, end_time: object + ) -> None: self._record(kwargs) - async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time) -> None: + async def async_log_failure_event( + self, kwargs: Mapping[str, object], response_obj: object, start_time: object, end_time: object + ) -> None: self._record(kwargs)