Skip to content

Commit 25a1e2b

Browse files
committed
refactor: improve default_hook_policy typing
Signed-off-by: Frederico Araujo <frederico.araujo@ibm.com>
1 parent 22904cc commit 25a1e2b

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

cpex/framework/manager.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
import copy
3333
import logging
3434
import threading
35-
from typing import Any, Optional, Union
35+
from typing import Any, Literal, Optional, Union
3636

3737
# Third-Party
3838
from pydantic import BaseModel, RootModel
@@ -108,7 +108,7 @@ def __init__(
108108
timeout: int = DEFAULT_PLUGIN_TIMEOUT,
109109
observability: Optional[ObservabilityProvider] = None,
110110
hook_policies: Optional[dict[str, HookPayloadPolicy]] = None,
111-
default_hook_policy: Optional[str] = None,
111+
default_hook_policy: Optional[Literal["allow", "deny"]] = None,
112112
):
113113
"""Initialize the plugin executor.
114114
@@ -890,7 +890,7 @@ def __init__(
890890
timeout: int = DEFAULT_PLUGIN_TIMEOUT,
891891
observability: Optional[ObservabilityProvider] = None,
892892
hook_policies: Optional[dict[str, HookPayloadPolicy]] = None,
893-
default_hook_policy: Optional[str] = None,
893+
default_hook_policy: Optional[Literal["allow", "deny"]] = None,
894894
):
895895
"""Initialize plugin manager.
896896

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "cpex"
3-
version = "0.1.0.dev5"
3+
version = "0.1.0.dev6"
44
description = "CPEX - ContextForge Plugin Extensibility Framework"
55
classifiers = [
66
"Topic :: Scientific/Engineering :: Artificial Intelligence",

uv.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)