fix: fix typing import

litellm/types/proxy/guardrails/guardrail_hooks/azure/azure_prompt_shield.py

@@ -1,4 +1,6 @@
-from typing import Any, Dict, List, TypedDict
+from typing import Any, Dict, List
+
+from typing_extensions import TypedDict
 
 from litellm.types.proxy.guardrails.guardrail_hooks.base import GuardrailConfigModel
 

litellm/types/proxy/guardrails/guardrail_hooks/azure/azure_text_moderation.py

@@ -1,6 +1,7 @@
-from typing import Any, Dict, List, Literal, Optional, Required, TypedDict
+from typing import Any, Dict, List, Literal, Optional
 
 from pydantic import BaseModel, Field
+from typing_extensions import Required, TypedDict
 
 from ..base import GuardrailConfigModel
 from .base import AzureContentSafetyConfigModel

litellm/types/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py

@@ -1,4 +1,6 @@
-from typing import Any, Dict, List, Literal, Optional, TypedDict, Union
+from typing import Any, Dict, List, Literal, Optional, Union
+
+from typing_extensions import TypedDict
 
 
 class BedrockTextContent(TypedDict, total=False):

litellm/types/proxy/guardrails/guardrail_hooks/lakera_ai_v2.py

@@ -1,4 +1,6 @@
-from typing import Dict, List, Optional, TypedDict
+from typing import Dict, List, Optional
+
+from typing_extensions import TypedDict
 
 from litellm.types.llms.openai import AllMessageValues
 

litellm/types/proxy/guardrails/guardrail_hooks/presidio.py

@@ -1,4 +1,6 @@
-from typing import Any, Dict, List, Optional, TypedDict, Union
+from typing import Any, Dict, List, Optional, Union
+
+from typing_extensions import TypedDict
 
 from litellm.types.guardrails import PiiEntityType