|
32 | 32 | import copy |
33 | 33 | import logging |
34 | 34 | import threading |
35 | | -from typing import Any, Optional, Union |
| 35 | +from typing import Any, Literal, Optional, Union |
36 | 36 |
|
37 | 37 | # Third-Party |
38 | 38 | from pydantic import BaseModel, RootModel |
@@ -108,7 +108,7 @@ def __init__( |
108 | 108 | timeout: int = DEFAULT_PLUGIN_TIMEOUT, |
109 | 109 | observability: Optional[ObservabilityProvider] = None, |
110 | 110 | hook_policies: Optional[dict[str, HookPayloadPolicy]] = None, |
111 | | - default_hook_policy: Optional[str] = None, |
| 111 | + default_hook_policy: Optional[Literal["allow", "deny"]] = None, |
112 | 112 | ): |
113 | 113 | """Initialize the plugin executor. |
114 | 114 |
|
@@ -890,7 +890,7 @@ def __init__( |
890 | 890 | timeout: int = DEFAULT_PLUGIN_TIMEOUT, |
891 | 891 | observability: Optional[ObservabilityProvider] = None, |
892 | 892 | hook_policies: Optional[dict[str, HookPayloadPolicy]] = None, |
893 | | - default_hook_policy: Optional[str] = None, |
| 893 | + default_hook_policy: Optional[Literal["allow", "deny"]] = None, |
894 | 894 | ): |
895 | 895 | """Initialize plugin manager. |
896 | 896 |
|
|
0 commit comments