modelfuzz 0.1.1__tar.gz → 0.1.2__tar.gz

modelfuzz-0.1.2/.claude/settings.local.json

@@ -0,0 +1,27 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(git -C /Users/gagandeep/agentshield status --short --branch)",
+      "Bash(uv --version)",
+      "Bash(uv sync *)",
+      "Bash(uv run python -c ' *)",
+      "Bash(uv run *)",
+      "Bash(git add *)",
+      "Bash(python3 -c \"import modelfuzz; print\\(modelfuzz.__file__\\)\")",
+      "Bash(pip show *)",
+      "Bash(python3 test_agent.py)",
+      "Bash(python3 -c \"import sys; print\\(sys.path\\)\")",
+      "Bash(pip list *)",
+      "Bash(python3 -c \"import sys,site; print\\(site.getsitepackages\\(\\)\\)\")",
+      "Bash(find / -maxdepth 6 -iname \"modelfuzz*\" -not -path \"*/modelfuzz/*\")",
+      "Bash(./.venv/bin/pip show *)",
+      "Bash(./.venv/bin/python3 test_agent.py)",
+      "Bash(python3 -c ' *)",
+      "Bash(python3 -m pytest -q)",
+      "Bash(./.venv/bin/pip install *)",
+      "Bash(./.venv/bin/python test_agent.py)",
+      "Bash(python3 -m pip show build twine)",
+      "Bash(env)"
+    ]
+  }
+}

{modelfuzz-0.1.1 → modelfuzz-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modelfuzz
-Version: 0.1.1
+Version: 0.1.2
 Summary: Runtime guardrails for AI agents.
 Project-URL: Homepage, https://github.com/higagan/modelfuzz
 Project-URL: Repository, https://github.com/higagan/modelfuzz

{modelfuzz-0.1.1 → modelfuzz-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "modelfuzz"
-version = "0.1.1"
+version = "0.1.2"
 description = "Runtime guardrails for AI agents."
 readme = "README.md"
 license = { text = "MIT" }

modelfuzz-0.1.2/src/modelfuzz/decorator.py

@@ -0,0 +1,64 @@
+"""Decorators for shielding agent tool functions."""
+
+import functools
+from collections.abc import Callable
+from typing import ParamSpec, TypeVar, overload
+
+from modelfuzz.engine import PolicyEngine
+from modelfuzz.exceptions import ModelFuzzBlockError
+from modelfuzz.rules import SensitiveDataFilter
+
+P = ParamSpec("P")
+R = TypeVar("R")
+
+# Default policy engine for the decorator
+default_policies = [SensitiveDataFilter()]
+_default_engine = PolicyEngine(default_policies)
+
+
+@overload
+def shield_tool(engine: Callable[P, R]) -> Callable[P, R]: ...
+@overload
+def shield_tool(
+    engine: PolicyEngine | None = None,
+) -> Callable[[Callable[P, R]], Callable[P, R]]: ...
+
+
+def shield_tool(engine=None):
+    """Wrap a tool function so every call is intercepted before execution.
+
+    Usable bare (``@shield_tool``) or called (``@shield_tool()`` /
+    ``@shield_tool(engine)``).
+
+    Args:
+        engine: The policy engine to use for validation. If None, a default
+            engine with a SensitiveDataFilter is used. When applied bare,
+            this receives the function being decorated instead.
+
+    Returns:
+        The wrapped function, or a decorator that produces it.
+    """
+    if callable(engine) and not isinstance(engine, PolicyEngine):
+        return _wrap(engine, _default_engine)
+
+    actual_engine = engine if engine is not None else _default_engine
+
+    def decorator(func: Callable[P, R]) -> Callable[P, R]:
+        return _wrap(func, actual_engine)
+
+    return decorator
+
+
+def _wrap(func: Callable[P, R], actual_engine: PolicyEngine) -> Callable[P, R]:
+    @functools.wraps(func)
+    def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
+        # Check all arguments against the policy engine
+        for arg in list(args) + list(kwargs.values()):
+            result = actual_engine.run(arg)
+            if not result.allowed:
+                raise ModelFuzzBlockError(result.reason or "Call blocked by policy")
+
+        print(f"ModelFuzz Intercepted: {func.__name__}")
+        return func(*args, **kwargs)
+
+    return wrapper

modelfuzz-0.1.1/.claude/settings.local.json

@@ -1,12 +0,0 @@
-{
-  "permissions": {
-    "allow": [
-      "Bash(git -C /Users/gagandeep/agentshield status --short --branch)",
-      "Bash(uv --version)",
-      "Bash(uv sync *)",
-      "Bash(uv run python -c ' *)",
-      "Bash(uv run *)",
-      "Bash(git add *)"
-    ]
-  }
-}

modelfuzz-0.1.1/src/modelfuzz/decorator.py

@@ -1,45 +0,0 @@
-"""Decorators for shielding agent tool functions."""
-
-import functools
-from collections.abc import Callable
-from typing import ParamSpec, TypeVar
-
-from modelfuzz.engine import PolicyEngine
-from modelfuzz.exceptions import ModelFuzzBlockError
-from modelfuzz.rules import SensitiveDataFilter
-
-P = ParamSpec("P")
-R = TypeVar("R")
-
-# Default policy engine for the decorator
-default_policies = [SensitiveDataFilter()]
-_default_engine = PolicyEngine(default_policies)
-
-
-def shield_tool(engine: PolicyEngine | None = None) -> Callable[[Callable[P, R]], Callable[P, R]]:
-    """Wrap a tool function so every call is intercepted before execution.
-
-    Args:
-        engine: The policy engine to use for validation. If None, a default
-            engine with a SensitiveDataFilter is used.
-
-    Returns:
-        A decorator that wraps the function.
-    """
-    actual_engine = engine if engine is not None else _default_engine
-
-    def decorator(func: Callable[P, R]) -> Callable[P, R]:
-        @functools.wraps(func)
-        def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
-            # Check all arguments against the policy engine
-            for arg in list(args) + list(kwargs.values()):
-                result = actual_engine.run(arg)
-                if not result.allowed:
-                    raise ModelFuzzBlockError(result.reason or "Call blocked by policy")
-
-            print(f"ModelFuzz Intercepted: {func.__name__}")
-            return func(*args, **kwargs)
-
-        return wrapper
-
-    return decorator