scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

fix(mypy): add type annotations in test_eval.py

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-21 10:00:50 -08:00
parent f4cf2fce55
commit 96e12ac926
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/guardrail_benchmarks/test_eval.py
View file

@ -20,7 +20,7 @@ import json
import os
import time
from datetime import datetime, timezone
from typing import List
from typing import Any, List
import pytest
from fastapi import HTTPException
@ -59,7 +59,7 @@ def _run(checker, text: str) -> dict:
return {"decision": "ALLOW", "score": 0.0, "matched_topic": None}
except HTTPException as e:
if e.status_code == 403:
detail = e.detail if isinstance(e.detail, dict) else {}
detail: dict[str, Any] = e.detail if isinstance(e.detail, dict) else {}
return {
"decision": "BLOCK",
"score": detail.get("score", 1.0),
@ -216,7 +216,7 @@ def _content_filter(category: str):
guardrail = ContentFilterGuardrail(
guardrail_name=f"{category}_eval",
categories=[ # type: ignore[arg-type]
categories=[ # type: ignore[arg-type,list-item]
{
"category": category,
"enabled": True,