diff --git a/litellm/integrations/asqav/asqav.py b/litellm/integrations/asqav/asqav.py index 5bcce14b34a..5f0235fecd8 100644 --- a/litellm/integrations/asqav/asqav.py +++ b/litellm/integrations/asqav/asqav.py @@ -277,8 +277,7 @@ class AsqavLogger(CustomLogger): self, kwargs: dict[str, Any], response_obj: object, - start_time: datetime | None, - end_time: datetime | None, + timing: tuple[datetime | None, datetime | None], status: str, ) -> None: """Build one audit record and append it to the JSONL log. @@ -286,6 +285,7 @@ class AsqavLogger(CustomLogger): seq/prev_hash assignment and the file write happen under _lock so the on-disk order always matches the chain order. """ + start_time, end_time = timing try: loggable = _extract_loggable( kwargs, response_obj, start_time, end_time, status @@ -376,7 +376,7 @@ class AsqavLogger(CustomLogger): start_time: datetime | None, end_time: datetime | None, ) -> None: - self._build_and_append(kwargs, response_obj, start_time, end_time, "success") + self._build_and_append(kwargs, response_obj, (start_time, end_time), "success") def log_failure_event( self, @@ -385,7 +385,7 @@ class AsqavLogger(CustomLogger): start_time: datetime | None, end_time: datetime | None, ) -> None: - self._build_and_append(kwargs, response_obj, start_time, end_time, "failure") + self._build_and_append(kwargs, response_obj, (start_time, end_time), "failure") async def async_log_success_event( self, @@ -398,8 +398,7 @@ class AsqavLogger(CustomLogger): self._build_and_append, kwargs, response_obj, - start_time, - end_time, + (start_time, end_time), "success", ) @@ -414,8 +413,7 @@ class AsqavLogger(CustomLogger): self._build_and_append, kwargs, response_obj, - start_time, - end_time, + (start_time, end_time), "failure", ) diff --git a/litellm/rag/ingestion/milvus_ingestion.py b/litellm/rag/ingestion/milvus_ingestion.py index 67cc3b79e80..e00739229a8 100644 --- a/litellm/rag/ingestion/milvus_ingestion.py +++ b/litellm/rag/ingestion/milvus_ingestion.py @@ -219,9 +219,9 @@ class MilvusRAGIngestion(BaseRAGIngestion): self, file_content: bytes | None, filename: str | None, - content_type: str | None, chunks: list[str], embeddings: list[list[float]] | None, + **_: object, # absorbs content_type and any extra base-class params ) -> tuple[str | None, str | None]: """ Insert chunks + embeddings into a Milvus collection. diff --git a/tests/proxy_unit_tests/test_guardrail_usage_config.py b/tests/proxy_unit_tests/test_guardrail_usage_config.py index 559b3721773..a2b26ce0cd5 100644 --- a/tests/proxy_unit_tests/test_guardrail_usage_config.py +++ b/tests/proxy_unit_tests/test_guardrail_usage_config.py @@ -30,7 +30,6 @@ def _metric( requests_evaluated: int = 10, passed_count: int = 8, blocked_count: int = 2, - flagged_count: int = 0, ): return SimpleNamespace( guardrail_id=guardrail_id, @@ -38,7 +37,7 @@ def _metric( requests_evaluated=requests_evaluated, passed_count=passed_count, blocked_count=blocked_count, - flagged_count=flagged_count, + flagged_count=0, )