treeship-sdk 0.1.0__tar.gz

treeship_sdk-0.1.0/.gitignore

@@ -0,0 +1,40 @@
+# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
+
+# dependencies
+/node_modules
+/.pnp
+.pnp.js
+
+# testing
+/coverage
+
+# next.js
+/.next/
+/out/
+
+# production
+/build
+/dist
+
+# misc
+.DS_Store
+*.pem
+.env
+.env.local
+.env.development.local
+.env.test.local
+.env.production.local
+
+# debug
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# vercel
+.vercel

treeship_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,87 @@
+Metadata-Version: 2.4
+Name: treeship-sdk
+Version: 0.1.0
+Summary: Cryptographic attestations for AI agents
+Project-URL: Homepage, https://treeship.dev
+Project-URL: Documentation, https://docs.treeship.dev
+Project-URL: Repository, https://github.com/treeship-dev/treeship
+Author-email: Treeship <hello@treeship.dev>
+License: MIT
+Keywords: agents,ai,attestation,cryptography,treeship,verification
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.27.0
+Description-Content-Type: text/markdown
+
+# treeship-sdk
+
+Cryptographic attestations for AI agents.
+
+## Installation
+
+```bash
+pip install treeship-sdk
+```
+
+## Quick Start
+
+```python
+from treeship import Treeship
+
+# Initialize (uses TREESHIP_API_KEY and TREESHIP_AGENT env vars)
+ts = Treeship()
+
+# Create an attestation
+result = ts.attest(
+    action="Approved loan application #12345",
+    inputs_hash=ts.hash({"customer_id": "cust_123", "amount": 50000})
+)
+
+print(f"Attestation: {result.attestation_id}")
+print(f"Verify at: {result.url}")
+```
+
+## Environment Variables
+
+| Variable | Description |
+|----------|-------------|
+| `TREESHIP_API_KEY` | Your API key (required for attest) |
+| `TREESHIP_AGENT` | Default agent slug |
+| `TREESHIP_API_URL` | API URL (default: https://api.treeship.dev) |
+
+## Async Support
+
+```python
+from treeship import AsyncTreeship
+
+async with AsyncTreeship() as ts:
+    result = await ts.attest(
+        action="Processed request",
+        inputs_hash=ts.hash(request_data)
+    )
+```
+
+## Verification
+
+```python
+# Verify an attestation
+result = ts.verify("attestation-id-here")
+print(f"Valid: {result['valid']}")
+```
+
+## Documentation
+
+- [Full Documentation](https://docs.treeship.dev)
+- [API Reference](https://docs.treeship.dev/api-reference)
+
+## License
+
+MIT

treeship_sdk-0.1.0/README.md

@@ -0,0 +1,64 @@
+# treeship-sdk
+
+Cryptographic attestations for AI agents.
+
+## Installation
+
+```bash
+pip install treeship-sdk
+```
+
+## Quick Start
+
+```python
+from treeship import Treeship
+
+# Initialize (uses TREESHIP_API_KEY and TREESHIP_AGENT env vars)
+ts = Treeship()
+
+# Create an attestation
+result = ts.attest(
+    action="Approved loan application #12345",
+    inputs_hash=ts.hash({"customer_id": "cust_123", "amount": 50000})
+)
+
+print(f"Attestation: {result.attestation_id}")
+print(f"Verify at: {result.url}")
+```
+
+## Environment Variables
+
+| Variable | Description |
+|----------|-------------|
+| `TREESHIP_API_KEY` | Your API key (required for attest) |
+| `TREESHIP_AGENT` | Default agent slug |
+| `TREESHIP_API_URL` | API URL (default: https://api.treeship.dev) |
+
+## Async Support
+
+```python
+from treeship import AsyncTreeship
+
+async with AsyncTreeship() as ts:
+    result = await ts.attest(
+        action="Processed request",
+        inputs_hash=ts.hash(request_data)
+    )
+```
+
+## Verification
+
+```python
+# Verify an attestation
+result = ts.verify("attestation-id-here")
+print(f"Valid: {result['valid']}")
+```
+
+## Documentation
+
+- [Full Documentation](https://docs.treeship.dev)
+- [API Reference](https://docs.treeship.dev/api-reference)
+
+## License
+
+MIT

treeship_sdk-0.1.0/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "treeship-sdk"
+version = "0.1.0"
+description = "Cryptographic attestations for AI agents"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.9"
+dependencies = ["httpx>=0.27.0"]
+keywords = ["treeship", "attestation", "ai", "agents", "cryptography", "verification"]
+authors = [{ name = "Treeship", email = "hello@treeship.dev" }]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+
+[project.urls]
+Homepage = "https://treeship.dev"
+Documentation = "https://docs.treeship.dev"
+Repository = "https://github.com/treeship-dev/treeship"
+
+[tool.hatch.build.targets.wheel]
+packages = ["treeship_sdk"]

treeship_sdk-0.1.0/treeship_sdk/__init__.py

@@ -0,0 +1,5 @@
+from .client import Treeship, AttestResult
+from .async_client import AsyncTreeship
+
+__version__ = "0.1.0"
+__all__ = ["Treeship", "AsyncTreeship", "AttestResult"]

treeship_sdk-0.1.0/treeship_sdk/async_client.py

@@ -0,0 +1,162 @@
+from __future__ import annotations
+
+import hashlib
+import json
+import os
+from typing import Any, Optional, Dict
+
+import httpx
+
+from .client import AttestResult, _get_api_url
+
+
+class AsyncTreeship:
+    """
+    Async Treeship client for creating cryptographic attestations.
+    
+    Example:
+        >>> from treeship import AsyncTreeship
+        >>> async with AsyncTreeship() as ts:
+        ...     result = await ts.attest(
+        ...         action="Approved loan application #12345",
+        ...         inputs_hash=ts.hash({"customer_id": "cust_123"})
+        ...     )
+        ...     print(result.url)
+    """
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        agent: Optional[str] = None,
+        api_url: Optional[str] = None,
+    ):
+        """
+        Initialize the async Treeship client.
+        
+        Args:
+            api_key: API key for authentication. Defaults to TREESHIP_API_KEY env var.
+            agent: Default agent slug. Defaults to TREESHIP_AGENT env var.
+            api_url: API base URL. Defaults to TREESHIP_API_URL env var or https://api.treeship.dev.
+        """
+        self.api_key = api_key or os.environ.get("TREESHIP_API_KEY", "")
+        self.agent = agent or os.getenv("TREESHIP_AGENT", "")
+        self._api_url = api_url or _get_api_url()
+        self._client = httpx.AsyncClient(
+            base_url=self._api_url,
+            headers={
+                "Authorization": f"Bearer {self.api_key}",
+                "User-Agent": "treeship-sdk/0.1.0",
+            },
+            timeout=30.0,
+        )
+
+    @staticmethod
+    def hash(data: Any) -> str:
+        """
+        Create a SHA256 hash of data for use as inputs_hash.
+        
+        Args:
+            data: Any JSON-serializable data, string, or bytes.
+            
+        Returns:
+            Hex-encoded SHA256 hash.
+        """
+        if isinstance(data, (dict, list)):
+            data = json.dumps(data, sort_keys=True, separators=(",", ":"))
+        if isinstance(data, str):
+            data = data.encode()
+        return hashlib.sha256(data).hexdigest()
+
+    async def attest(
+        self,
+        action: str,
+        inputs_hash: Optional[str] = None,
+        agent: Optional[str] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> AttestResult:
+        """
+        Create a cryptographic attestation.
+        
+        Args:
+            action: Human-readable description of the action.
+            inputs_hash: SHA256 hash of inputs. If not provided, hashes the action string.
+            agent: Agent slug. Uses default agent if not provided.
+            metadata: Optional key-value metadata.
+            
+        Returns:
+            AttestResult with attestation details and verification URL.
+        """
+        slug = agent or self.agent
+        if not slug:
+            raise ValueError(
+                "Agent slug required: pass agent= or set TREESHIP_AGENT environment variable"
+            )
+
+        if not self.api_key:
+            raise ValueError(
+                "API key required: pass api_key= or set TREESHIP_API_KEY environment variable"
+            )
+
+        response = await self._client.post(
+            "/v1/attest",
+            json={
+                "agent_slug": slug,
+                "action": action,
+                "inputs_hash": inputs_hash or self.hash(action),
+                "metadata": metadata or {},
+            },
+        )
+        response.raise_for_status()
+        return AttestResult.from_dict(response.json())
+
+    async def verify(self, attestation_id: str) -> Dict[str, Any]:
+        """
+        Verify an attestation by ID.
+        
+        Args:
+            attestation_id: The attestation UUID.
+            
+        Returns:
+            Dict with verification result including 'valid' boolean.
+        """
+        response = await self._client.get(f"/v1/verify/{attestation_id}")
+        response.raise_for_status()
+        return response.json()
+
+    async def get_agent(self, slug: Optional[str] = None) -> Dict[str, Any]:
+        """
+        Get agent feed with recent attestations.
+        
+        Args:
+            slug: Agent slug. Uses default agent if not provided.
+            
+        Returns:
+            Dict with agent info and attestations list.
+        """
+        agent_slug = slug or self.agent
+        if not agent_slug:
+            raise ValueError("Agent slug required")
+        response = await self._client.get(f"/v1/agent/{agent_slug}")
+        response.raise_for_status()
+        return response.json()
+
+    async def get_pubkey(self) -> Dict[str, Any]:
+        """
+        Get the public signing key for independent verification.
+        
+        Returns:
+            Dict with key_id, algorithm, and public_key_pem.
+        """
+        response = await self._client.get("/v1/pubkey")
+        response.raise_for_status()
+        return response.json()
+
+    async def close(self) -> None:
+        """Close the HTTP client."""
+        await self._client.aclose()
+
+    async def __aenter__(self) -> AsyncTreeship:
+        return self
+
+    async def __aexit__(self, *args: Any) -> None:
+        await self.close()

treeship_sdk-0.1.0/treeship_sdk/client.py

@@ -0,0 +1,203 @@
+from __future__ import annotations
+
+import hashlib
+import json
+import os
+from dataclasses import dataclass
+from typing import Any, Optional, Dict
+
+import httpx
+
+
+def _get_api_url() -> str:
+    return os.getenv("TREESHIP_API_URL", "https://api.treeship.dev")
+
+
+@dataclass
+class AttestResult:
+    """Result of creating an attestation."""
+    
+    attestation_id: str
+    signature: str
+    payload_hash: str
+    key_id: str
+    timestamp: str
+    url: str
+    verify_command: str
+    agent_slug: str
+    action: str
+    inputs_hash: str
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> AttestResult:
+        return cls(
+            attestation_id=d["attestation_id"],
+            signature=d["signature"],
+            payload_hash=d["payload_hash"],
+            key_id=d["key_id"],
+            timestamp=d["timestamp"],
+            url=d["public_url"],
+            verify_command=d["verify_command"],
+            agent_slug=d["agent_slug"],
+            action=d["action"],
+            inputs_hash=d["inputs_hash"],
+        )
+
+
+class Treeship:
+    """
+    Treeship client for creating cryptographic attestations.
+    
+    Example:
+        >>> from treeship import Treeship
+        >>> ts = Treeship()
+        >>> result = ts.attest(
+        ...     action="Approved loan application #12345",
+        ...     inputs_hash=ts.hash({"customer_id": "cust_123"})
+        ... )
+        >>> print(result.url)
+    """
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        agent: Optional[str] = None,
+        api_url: Optional[str] = None,
+    ):
+        """
+        Initialize the Treeship client.
+        
+        Args:
+            api_key: API key for authentication. Defaults to TREESHIP_API_KEY env var.
+            agent: Default agent slug. Defaults to TREESHIP_AGENT env var.
+            api_url: API base URL. Defaults to TREESHIP_API_URL env var or https://api.treeship.dev.
+        """
+        self.api_key = api_key or os.environ.get("TREESHIP_API_KEY", "")
+        self.agent = agent or os.getenv("TREESHIP_AGENT", "")
+        self._api_url = api_url or _get_api_url()
+        self._client = httpx.Client(
+            base_url=self._api_url,
+            headers={
+                "Authorization": f"Bearer {self.api_key}",
+                "User-Agent": "treeship-sdk/0.1.0",
+            },
+            timeout=30.0,
+        )
+
+    @staticmethod
+    def hash(data: Any) -> str:
+        """
+        Create a SHA256 hash of data for use as inputs_hash.
+        
+        Args:
+            data: Any JSON-serializable data, string, or bytes.
+            
+        Returns:
+            Hex-encoded SHA256 hash.
+        """
+        if isinstance(data, (dict, list)):
+            data = json.dumps(data, sort_keys=True, separators=(",", ":"))
+        if isinstance(data, str):
+            data = data.encode()
+        return hashlib.sha256(data).hexdigest()
+
+    def attest(
+        self,
+        action: str,
+        inputs_hash: Optional[str] = None,
+        agent: Optional[str] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> AttestResult:
+        """
+        Create a cryptographic attestation.
+        
+        Args:
+            action: Human-readable description of the action.
+            inputs_hash: SHA256 hash of inputs. If not provided, hashes the action string.
+            agent: Agent slug. Uses default agent if not provided.
+            metadata: Optional key-value metadata.
+            
+        Returns:
+            AttestResult with attestation details and verification URL.
+            
+        Raises:
+            ValueError: If no agent slug is available.
+            httpx.HTTPStatusError: If the API request fails.
+        """
+        slug = agent or self.agent
+        if not slug:
+            raise ValueError(
+                "Agent slug required: pass agent= or set TREESHIP_AGENT environment variable"
+            )
+
+        if not self.api_key:
+            raise ValueError(
+                "API key required: pass api_key= or set TREESHIP_API_KEY environment variable"
+            )
+
+        response = self._client.post(
+            "/v1/attest",
+            json={
+                "agent_slug": slug,
+                "action": action,
+                "inputs_hash": inputs_hash or self.hash(action),
+                "metadata": metadata or {},
+            },
+        )
+        response.raise_for_status()
+        return AttestResult.from_dict(response.json())
+
+    def verify(self, attestation_id: str) -> Dict[str, Any]:
+        """
+        Verify an attestation by ID.
+        
+        Args:
+            attestation_id: The attestation UUID.
+            
+        Returns:
+            Dict with verification result including 'valid' boolean.
+            
+        Raises:
+            httpx.HTTPStatusError: If the attestation is not found or request fails.
+        """
+        response = self._client.get(f"/v1/verify/{attestation_id}")
+        response.raise_for_status()
+        return response.json()
+
+    def get_agent(self, slug: Optional[str] = None) -> Dict[str, Any]:
+        """
+        Get agent feed with recent attestations.
+        
+        Args:
+            slug: Agent slug. Uses default agent if not provided.
+            
+        Returns:
+            Dict with agent info and attestations list.
+        """
+        agent_slug = slug or self.agent
+        if not agent_slug:
+            raise ValueError("Agent slug required")
+        response = self._client.get(f"/v1/agent/{agent_slug}")
+        response.raise_for_status()
+        return response.json()
+
+    def get_pubkey(self) -> Dict[str, Any]:
+        """
+        Get the public signing key for independent verification.
+        
+        Returns:
+            Dict with key_id, algorithm, and public_key_pem.
+        """
+        response = self._client.get("/v1/pubkey")
+        response.raise_for_status()
+        return response.json()
+
+    def close(self) -> None:
+        """Close the HTTP client."""
+        self._client.close()
+
+    def __enter__(self) -> Treeship:
+        return self
+
+    def __exit__(self, *args: Any) -> None:
+        self.close()