draft-protocol 0.1.0__py3-none-any.whl

draft_protocol/__init__.py

@@ -0,0 +1,65 @@
+"""DRAFT Protocol — Intake Governance for AI Tool Calls.
+
+Ensures AI understands human intent before execution begins.
+
+Usage:
+    # As MCP server (primary use case):
+    pip install draft-protocol
+    python -m draft_protocol
+
+    # As library:
+    from draft_protocol.engine import classify_tier, map_dimensions, check_gate
+    from draft_protocol.storage import create_session, get_session
+    from draft_protocol.providers import llm_available, embed_available
+"""
+
+__version__ = "0.1.0"
+
+# Public API — importable from `draft_protocol` directly
+from draft_protocol.engine import (
+    add_assumption,
+    check_gate,
+    classify_tier,
+    confirm_field,
+    elicitation_review,
+    generate_assumptions,
+    generate_elicitation,
+    map_dimensions,
+    override_gate,
+    unscreen_dimension,
+    verify_assumption,
+)
+from draft_protocol.providers import (
+    embed_available,
+    llm_available,
+)
+from draft_protocol.storage import (
+    close_session,
+    create_session,
+    get_active_session,
+    get_session,
+)
+
+__all__ = [
+    "__version__",
+    # Engine
+    "classify_tier",
+    "map_dimensions",
+    "generate_elicitation",
+    "generate_assumptions",
+    "check_gate",
+    "confirm_field",
+    "unscreen_dimension",
+    "add_assumption",
+    "override_gate",
+    "verify_assumption",
+    "elicitation_review",
+    # Storage
+    "create_session",
+    "get_session",
+    "get_active_session",
+    "close_session",
+    # Providers
+    "llm_available",
+    "embed_available",
+]

draft_protocol/__main__.py

@@ -0,0 +1,61 @@
+"""Entry point: python -m draft_protocol
+
+Supports multiple MCP transports:
+  stdio            — Default. For Claude Desktop, Cursor, Windsurf, Continue.
+  sse              — Server-Sent Events. For web-based MCP clients.
+  streamable-http  — New MCP standard. For HTTP-native MCP clients.
+
+Usage:
+  python -m draft_protocol                          # stdio (default)
+  python -m draft_protocol --transport sse          # SSE on port 8420
+  python -m draft_protocol --transport streamable-http --port 8420
+  python -m draft_protocol --transport rest         # REST API on port 8420
+
+Environment variables (override CLI defaults):
+  DRAFT_TRANSPORT  — stdio | sse | streamable-http
+  DRAFT_HOST       — Bind address (default: 127.0.0.1)
+  DRAFT_PORT       — Port for SSE/HTTP (default: 8420)
+"""
+import argparse
+import os
+
+from draft_protocol.server import mcp
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="draft-protocol",
+        description="DRAFT Protocol — Intake governance for AI tool calls.",
+    )
+    parser.add_argument(
+        "--transport", "-t",
+        choices=["stdio", "sse", "streamable-http", "rest"],
+        default=os.environ.get("DRAFT_TRANSPORT", "stdio"),
+        help="MCP transport protocol (default: stdio)",
+    )
+    parser.add_argument(
+        "--host",
+        default=os.environ.get("DRAFT_HOST", "127.0.0.1"),
+        help="Bind address for SSE/HTTP (default: 127.0.0.1)",
+    )
+    parser.add_argument(
+        "--port", "-p",
+        type=int,
+        default=int(os.environ.get("DRAFT_PORT", "8420")),
+        help="Port for SSE/HTTP (default: 8420)",
+    )
+    args = parser.parse_args()
+
+    if args.transport == "stdio":
+        mcp.run(transport="stdio")
+    elif args.transport == "sse":
+        mcp.run(transport="sse", host=args.host, port=args.port)
+    elif args.transport == "streamable-http":
+        mcp.run(transport="streamable-http", host=args.host, port=args.port)
+    elif args.transport == "rest":
+        from draft_protocol.rest import run_rest_server
+        run_rest_server(host=args.host, port=args.port)
+
+
+if __name__ == "__main__":
+    main()

draft_protocol/config.py

@@ -0,0 +1,112 @@
+"""DRAFT Protocol Configuration."""
+import os
+from pathlib import Path
+
+# ── Storage ───────────────────────────────────────────────
+DB_PATH = Path(os.environ.get("DRAFT_DB_PATH", "~/.draft_protocol/draft.db")).expanduser()
+DB_PATH.parent.mkdir(parents=True, exist_ok=True)
+
+# ── LLM Provider (optional — enhances classification accuracy) ──
+# Supported: "none" (default), "ollama", "openai", "anthropic"
+# "openai" works with any OpenAI-compatible API (Together, Groq, LM Studio, etc.)
+LLM_PROVIDER = os.environ.get("DRAFT_LLM_PROVIDER", "none")
+LLM_MODEL = os.environ.get("DRAFT_LLM_MODEL", "")
+EMBED_MODEL = os.environ.get("DRAFT_EMBED_MODEL", "")
+API_KEY = os.environ.get("DRAFT_API_KEY", "")
+API_BASE = os.environ.get("DRAFT_API_BASE", "")
+
+# Backward compatibility: if old DRAFT_OLLAMA_URL is set, use it
+if not API_BASE and os.environ.get("DRAFT_OLLAMA_URL"):
+    API_BASE = os.environ["DRAFT_OLLAMA_URL"]
+
+# Auto-detect provider from model name if not explicitly set
+if LLM_PROVIDER == "none" and LLM_MODEL:
+    if "gpt" in LLM_MODEL or "o1" in LLM_MODEL or "o3" in LLM_MODEL:
+        LLM_PROVIDER = "openai"
+    elif "claude" in LLM_MODEL:
+        LLM_PROVIDER = "anthropic"
+    elif LLM_MODEL:
+        LLM_PROVIDER = "ollama"  # Default to Ollama for unknown models
+
+# ── Tier Classification Triggers ──────────────────────────
+# Keyword-based fast path. LLM classification (if available) handles ambiguous cases.
+
+CONSEQUENTIAL_TRIGGERS = [
+    "canonical", "governance", "constitution", "guardian", "authority",
+    "ip ", "intellectual property", "classification level",
+    "consciousness", "self-model", "phenomenological",
+    "restructure", "architecture decision", "merge domains",
+    "amendment", "hard constraint", "prohibition",
+    "production deployment", "security policy", "auth modification",
+]
+
+STANDARD_TRIGGERS = [
+    "implement", "specification", "draft", "build", "create",
+    "design", "analyze", "recommend", "evaluate", "compare",
+    "refactor", "migrate", "integrate", "deploy", "configure",
+    "document", "spec", "proposal", "pipeline", "workflow",
+    # Security: extraction-pattern triggers (OWASP LLM07)
+    "ignore previous instructions", "ignore all previous", "ignore above",
+    "repeat above", "repeat everything", "verbatim",
+    "system prompt", "print environment", "environment variables",
+    "show me your instructions", "what are your rules",
+    "dump your config", "reveal your prompt", "debug mode",
+]
+
+# ── Dimensions ────────────────────────────────────────────
+# D and T are mandatory; R, A, F can be screened out when inapplicable.
+
+MANDATORY_DIMENSIONS = ["D", "T"]
+
+DRAFT_FIELDS = {
+    "D": {
+        "D1": "What exactly is being created?",
+        "D2": "What domain does it belong to?",
+        "D3": "What fails without it?",
+        "D4": "Replacement test — what existing thing could serve?",
+        "D5": "What are the explicit non-goals?",
+    },
+    "R": {
+        "R1": "Who is the human authority source?",
+        "R2": "What decisions is this allowed to make?",
+        "R3": "What decisions are forbidden?",
+        "R4": "What are the stop conditions (need >= 3)?",
+        "R5": "What interfaces does it interact with?",
+    },
+    "A": {
+        "A1": "What inputs are allowed?",
+        "A2": "What inputs are forbidden?",
+        "A3": "What outputs are allowed?",
+        "A4": "What outputs are forbidden?",
+        "A5": "Provide a correct example.",
+        "A6": "Provide an incorrect example.",
+    },
+    "F": {
+        "F1": "Who has change authority?",
+        "F2": "What changes are permitted?",
+        "F3": "What changes are forbidden?",
+        "F4": "What triggers a review?",
+    },
+    "T": {
+        "T1": "How is success defined?",
+        "T2": "How is failure defined?",
+        "T3": "What review questions apply (need >= 3)?",
+        "T4": "What evidence is required?",
+    },
+}
+
+DIMENSION_SCREEN_QUESTIONS = {
+    "D": None,  # Never screened
+    "R": "Does this task involve delegated decisions, authority, or operational limits?",
+    "A": "Does this task consume or produce specific artifacts (files, data, outputs)?",
+    "F": "Does this task have a lifecycle — will it need to change or adapt over time?",
+    "T": None,  # Never screened
+}
+
+DIMENSION_NAMES = {
+    "D": "Define (Existence & ROI)",
+    "R": "Rules (Operation & Limits)",
+    "A": "Artifacts (Inputs & Outputs)",
+    "F": "Flex (Change Without Drift)",
+    "T": "Test (Evaluation)",
+}