diff --git a/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/competitor_intent/airline.py b/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/competitor_intent/airline.py index 583814de1ef..5da0bd25fcf 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/competitor_intent/airline.py +++ b/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/competitor_intent/airline.py @@ -11,7 +11,7 @@ brand_self so all other major airlines are treated as competitors. import json from pathlib import Path -from typing import Any, Dict, List, Tuple +from typing import Any, Dict, List, Set, Tuple from litellm.proxy.guardrails.guardrail_hooks.litellm_content_filter.competitor_intent.base import ( BaseCompetitorIntentChecker, diff --git a/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/guardrail_benchmarks/test_eval.py b/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/guardrail_benchmarks/test_eval.py index 2c872258cec..5fbdd1456eb 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/guardrail_benchmarks/test_eval.py +++ b/litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/guardrail_benchmarks/test_eval.py @@ -20,7 +20,7 @@ import json import os import time from datetime import datetime, timezone -from typing import Any, List +from typing import Any, Dict, List import pytest from fastapi import HTTPException