relaysecurity-dev 0.1.0__py3-none-any.whl

relaysecurity_dev/__init__.py

@@ -0,0 +1,3 @@
+from .client import Relay, RelayPolicyError
+
+__all__ = ["Relay", "RelayPolicyError"]

relaysecurity_dev/client.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+import os
+from collections.abc import Callable
+from typing import Any, TypeVar
+
+import requests
+
+DEFAULT_ENDPOINT = "https://relay-security-lemon.vercel.app/api/execute"
+
+T = TypeVar("T")
+
+
+class RelayPolicyError(RuntimeError):
+    def __init__(self, message: str, result: dict[str, Any]):
+        super().__init__(message)
+        self.result = result
+
+
+class Relay:
+    def __init__(
+        self,
+        api_key: str | None = None,
+        endpoint: str | None = None,
+        timeout: float = 15,
+    ) -> None:
+        self.api_key = api_key or os.getenv("RELAY_API_KEY")
+        self.endpoint = endpoint or os.getenv("RELAY_ENDPOINT") or DEFAULT_ENDPOINT
+        self.timeout = timeout
+
+    def check(
+        self,
+        tool: str,
+        arguments: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
+        headers = {"Content-Type": "application/json"}
+
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+
+        response = requests.post(
+            self.endpoint,
+            headers=headers,
+            json={"tool": tool, "arguments": arguments or {}},
+            timeout=self.timeout,
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def assert_allowed(
+        self,
+        tool: str,
+        arguments: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
+        result = self.check(tool, arguments)
+
+        if result.get("status") == "blocked":
+            raise RelayPolicyError(
+                result.get("reason", "Blocked by Relay policy"),
+                result,
+            )
+
+        return result
+
+    def guard_tool(
+        self,
+        tool: str,
+        handler: Callable[[dict[str, Any]], T],
+    ) -> Callable[[dict[str, Any] | None], T]:
+        def guarded(arguments: dict[str, Any] | None = None) -> T:
+            safe_arguments = arguments or {}
+            self.assert_allowed(tool, safe_arguments)
+            return handler(safe_arguments)
+
+        return guarded

relaysecurity_dev-0.1.0.dist-info/METADATA

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: relaysecurity-dev
+Version: 0.1.0
+Summary: Python SDK for checking AI agent tool calls with Relay.
+Project-URL: Homepage, https://relay-security-lemon.vercel.app
+Project-URL: Repository, https://github.com/aniiketvarshney/Relay-Security
+Author: Relay Security
+License-Expression: MIT
+Keywords: agents,ai,guardrails,relay,security
+Requires-Python: >=3.9
+Requires-Dist: requests>=2.31.0
+Description-Content-Type: text/markdown
+
+# Relay Python SDK
+
+Protect risky AI agent tool calls with Relay from Python, LangGraph, and custom
+agent stacks.
+
+```python
+from relaysecurity_dev import Relay
+
+relay = Relay()
+
+relay.assert_allowed("github_delete_repo", {"repo_name": "myrepo"})
+```
+
+Set your API key before running:
+
+```bash
+RELAY_API_KEY=relay_sk_your_key_here
+```

relaysecurity_dev-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+relaysecurity_dev/__init__.py,sha256=EPLBIbyOkn72sAmG0kI6Cb4d_iLbyqZOLJRyQI9EvM4,85
+relaysecurity_dev/client.py,sha256=ksQHlUk28ZaiFWzxppOK9nvaFJAJs6AdH72Kfkrr_wE,2030
+relaysecurity_dev-0.1.0.dist-info/METADATA,sha256=cowZLSn17Cb19-JkOTAk7AU__3eU9fOPy3DaDjppwgI,788
+relaysecurity_dev-0.1.0.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+relaysecurity_dev-0.1.0.dist-info/RECORD,,

relaysecurity_dev-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.30.1
+Root-Is-Purelib: true
+Tag: py3-none-any