relaysecurity-dev 0.1.0__tar.gz

relaysecurity_dev-0.1.0/.gitignore

@@ -0,0 +1,6 @@
+.env
+.env.local
+node_modules/
+dist/
+.vite/
+.next/

relaysecurity_dev-0.1.0/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,18 @@
+# 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/examples/langgraph_guard.py

@@ -0,0 +1,8 @@
+from relaysecurity_dev import Relay
+
+relay = Relay()
+
+
+def relay_guard_node(state):
+    relay.assert_allowed(state["tool"], state.get("arguments", {}))
+    return state

relaysecurity_dev-0.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "relaysecurity-dev"
+version = "0.1.0"
+description = "Python SDK for checking AI agent tool calls with Relay."
+readme = "README.md"
+requires-python = ">=3.9"
+license = "MIT"
+authors = [
+  { name = "Relay Security" }
+]
+keywords = ["ai", "agents", "guardrails", "security", "relay"]
+dependencies = [
+  "requests>=2.31.0"
+]
+
+[project.urls]
+Homepage = "https://relay-security-lemon.vercel.app"
+Repository = "https://github.com/aniiketvarshney/Relay-Security"

relaysecurity_dev-0.1.0/src/relaysecurity_dev/__init__.py

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

relaysecurity_dev-0.1.0/src/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