zeroshot-sdk 0.0.1__tar.gz

zeroshot_sdk-0.0.1/.env

@@ -0,0 +1 @@
+ZEROSHOT_API_KEY=

zeroshot_sdk-0.0.1/PKG-INFO

@@ -0,0 +1,60 @@
+Metadata-Version: 2.4
+Name: zeroshot-sdk
+Version: 0.0.1
+Summary: ZeroShot SDK - AI System Testing CLI and Client Library
+Author: Lab42 Team
+License: MIT
+Keywords: ai,llm,red-teaming,sdk,security,testing
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Security
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.25
+Requires-Dist: rich>=13.0
+Requires-Dist: toml>=0.10
+Requires-Dist: typer>=0.9
+Description-Content-Type: text/markdown
+
+# ZeroShot SDK
+
+Official SDK for ZeroShot - AI Security Testing Platform.
+
+## Installation
+
+```bash
+pip install zeroshot-sdk
+```
+
+## Quick Start
+
+```bash
+# Login (stores API key locally)
+zeroshot auth login
+
+# Run a scan
+zeroshot scan --target http://target-url
+
+# View results
+zeroshot scans list
+```
+
+## Python Usage
+
+```python
+from zeroshot_sdk import ZeroShotClient
+
+client = ZeroShotClient(api_key="zsk_...")
+
+# Run a scan
+result = client.scan(
+    target="http://target-url",
+    categories=["jailbreak", "prompt_injection"],
+)
+
+print(f"Found {result['vulnerabilities_found']} vulnerabilities")
+```

zeroshot_sdk-0.0.1/README.md

@@ -0,0 +1,38 @@
+# ZeroShot SDK
+
+Official SDK for ZeroShot - AI Security Testing Platform.
+
+## Installation
+
+```bash
+pip install zeroshot-sdk
+```
+
+## Quick Start
+
+```bash
+# Login (stores API key locally)
+zeroshot auth login
+
+# Run a scan
+zeroshot scan --target http://target-url
+
+# View results
+zeroshot scans list
+```
+
+## Python Usage
+
+```python
+from zeroshot_sdk import ZeroShotClient
+
+client = ZeroShotClient(api_key="zsk_...")
+
+# Run a scan
+result = client.scan(
+    target="http://target-url",
+    categories=["jailbreak", "prompt_injection"],
+)
+
+print(f"Found {result['vulnerabilities_found']} vulnerabilities")
+```

zeroshot_sdk-0.0.1/pyproject.toml

@@ -0,0 +1,38 @@
+[project]
+name = "zeroshot-sdk"
+version = "0.0.1"
+description = "ZeroShot SDK - AI System Testing CLI and Client Library"
+readme = "README.md"
+requires-python = ">=3.10"
+license = {text = "MIT"}
+authors = [
+    {name = "Lab42 Team"}
+]
+keywords = ["ai", "security", "testing", "llm", "red-teaming", "sdk"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Security",
+]
+
+dependencies = [
+    "httpx>=0.25",
+    "typer>=0.9",
+    "rich>=13.0",
+    "toml>=0.10",
+]
+
+[project.scripts]
+zeroshot = "zeroshot_sdk.cli:app"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/zeroshot_sdk"]

zeroshot_sdk-0.0.1/src/zeroshot_sdk/__init__.py

@@ -0,0 +1,13 @@
+"""
+ZeroShot SDK - AI Security Testing Client.
+"""
+
+
+from zeroshot_sdk.config import API_URL, API_KEY, get_api_url, get_api_key, get_headers
+
+
+__version__ = "0.1.0"
+__all__ = ["API_URL", "API_KEY", "get_api_url", "get_api_key", "get_headers"]
+
+
+

zeroshot_sdk-0.0.1/src/zeroshot_sdk/auth.py

@@ -0,0 +1,127 @@
+"""
+ZeroShot SDK - API Key management.
+
+Stores credentials in ~/.zeroshot/config.toml
+"""
+
+import os
+from pathlib import Path
+from typing import Optional
+
+try:
+    import toml
+except ImportError:
+    toml = None
+
+
+CONFIG_DIR = Path.home() / ".zeroshot"
+CONFIG_FILE = CONFIG_DIR / "config.toml"
+
+
+def get_config_path() -> Path:
+    """Get the config file path."""
+    return CONFIG_FILE
+
+
+def ensure_config_dir():
+    """Ensure the config directory exists."""
+    CONFIG_DIR.mkdir(exist_ok=True)
+
+
+def load_config() -> dict:
+    """Load configuration from file."""
+    if not CONFIG_FILE.exists():
+        return {}
+    
+    if toml is None:
+        # Fallback: simple key=value parsing
+        config = {}
+        with open(CONFIG_FILE) as f:
+            for line in f:
+                line = line.strip()
+                if "=" in line and not line.startswith("#"):
+                    key, value = line.split("=", 1)
+                    config[key.strip()] = value.strip().strip('"')
+        return config
+    
+    with open(CONFIG_FILE) as f:
+        return toml.load(f)
+
+
+def save_config(config: dict):
+    """Save configuration to file."""
+    ensure_config_dir()
+    
+    if toml is None:
+        # Fallback: simple key=value format
+        with open(CONFIG_FILE, "w") as f:
+            for key, value in config.items():
+                f.write(f'{key} = "{value}"\n')
+    else:
+        with open(CONFIG_FILE, "w") as f:
+            toml.dump(config, f)
+
+
+def get_api_key() -> Optional[str]:
+    """
+    Get the stored API key.
+    
+    Checks:
+    1. ZEROSHOT_API_KEY environment variable
+    2. ~/.zeroshot/config.toml
+    
+    Returns:
+        API key string or None if not found
+    """
+    # Check environment variable first
+    env_key = os.environ.get("ZEROSHOT_API_KEY")
+    if env_key:
+        return env_key
+    
+    # Check config file
+    config = load_config()
+    return config.get("api_key")
+
+
+def save_api_key(api_key: str):
+    """
+    Save API key to config file.
+    
+    Args:
+        api_key: The API key to save (zsk_...)
+    """
+    config = load_config()
+    config["api_key"] = api_key
+    save_config(config)
+
+
+def get_api_url() -> str:
+    """
+    Get the API base URL.
+    
+    Checks:
+    1. ZEROSHOT_API_URL environment variable
+    2. ~/.zeroshot/config.toml
+    3. Default: https://api.0eroshot.com
+    """
+    # Check environment variable
+    env_url = os.environ.get("ZEROSHOT_API_URL")
+    if env_url:
+        return env_url
+    
+    # Check config file
+    config = load_config()
+    return config.get("api_url", "https://api.0eroshot.com")
+
+
+def save_api_url(api_url: str):
+    """Save API URL to config file."""
+    config = load_config()
+    config["api_url"] = api_url
+    save_config(config)
+
+
+def clear_config():
+    """Clear all stored configuration."""
+    if CONFIG_FILE.exists():
+        CONFIG_FILE.unlink()