entroplain 0.2.0 → 0.2.1

package/entroplain/shared_state.py

@@ -0,0 +1,72 @@
+"""
+Shared state for real-time data sharing between proxy and dashboard.
+"""
+
+import asyncio
+import json
+from typing import Dict, Any, List, Callable, Optional
+from dataclasses import dataclass, field
+from datetime import datetime
+
+
+@dataclass
+class SharedState:
+    """
+    Shared state between proxy and dashboard.
+    
+    Usage:
+        state = SharedState()
+        
+        # In proxy:
+        await state.update({
+            "trajectory": [...],
+            "token_count": 10,
+            ...
+        })
+        
+        # In dashboard:
+        data = await state.get_update()
+    """
+    _data: Dict[str, Any] = field(default_factory=dict)
+    _subscribers: List[Callable] = field(default_factory=list)
+    _lock: asyncio.Lock = field(default_factory=asyncio.Lock)
+    _last_update: Optional[datetime] = None
+    
+    async def update(self, data: Dict[str, Any]) -> None:
+        """Update state and notify subscribers."""
+        async with self._lock:
+            self._data = data
+            self._last_update = datetime.utcnow()
+        
+        # Notify all subscribers
+        for callback in self._subscribers:
+            try:
+                await callback(data)
+            except Exception:
+                pass
+    
+    async def get(self) -> Dict[str, Any]:
+        """Get current state."""
+        async with self._lock:
+            return self._data.copy()
+    
+    def subscribe(self, callback: Callable) -> None:
+        """Subscribe to state updates."""
+        self._subscribers.append(callback)
+    
+    def unsubscribe(self, callback: Callable) -> None:
+        """Unsubscribe from state updates."""
+        if callback in self._subscribers:
+            self._subscribers.remove(callback)
+
+
+# Global shared state instance
+_shared_state: Optional[SharedState] = None
+
+
+def get_shared_state() -> SharedState:
+    """Get or create the global shared state."""
+    global _shared_state
+    if _shared_state is None:
+        _shared_state = SharedState()
+    return _shared_state

package/package.json

@@ -1,46 +1,47 @@
-{
-  "name": "entroplain",
-  "version": "0.2.0",
-  "description": "Entropy-based early exit for efficient agent reasoning",
-  "main": "dist/index.js",
-  "types": "dist/index.d.ts",
-  "scripts": {
-    "build": "tsc",
-    "test": "jest",
-    "lint": "eslint src/",
-    "prepublishOnly": "npm run build"
-  },
-  "keywords": [
-    "llm",
-    "agent",
-    "entropy",
-    "early-exit",
-    "reasoning",
-    "efficiency",
-    "cost-tracking",
-    "dashboard"
-  ],
-  "author": "Entroplain Contributors",
-  "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/entroplain/entroplain.git"
-  },
-  "bugs": {
-    "url": "https://github.com/entroplain/entroplain/issues"
-  },
-  "homepage": "https://github.com/entroplain/entroplain#readme",
-  "devDependencies": {
-    "@types/jest": "^29.5.0",
-    "@types/node": "^20.0.0",
-    "jest": "^29.5.0",
-    "ts-jest": "^29.1.0",
-    "typescript": "^5.0.0"
-  },
-  "dependencies": {
-    "openai": "^4.0.0"
-  },
-  "engines": {
-    "node": ">=18.0.0"
-  }
-}
+{
+  "name": "entroplain",
+  "version": "0.2.1",
+  "description": "Entropy-based early exit for efficient agent reasoning",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "tsc",
+    "test": "jest",
+    "lint": "eslint src/",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "llm",
+    "agent",
+    "entropy",
+    "early-exit",
+    "reasoning",
+    "efficiency",
+    "cost-tracking",
+    "dashboard"
+  ],
+  "author": "Entroplain Contributors",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/entroplain/entroplain.git"
+  },
+  "bugs": {
+    "url": "https://github.com/entroplain/entroplain/issues"
+  },
+  "homepage": "https://github.com/entroplain/entroplain#readme",
+  "devDependencies": {
+    "@types/jest": "^29.5.0",
+    "@types/node": "^20.0.0",
+    "jest": "^29.5.0",
+    "ts-jest": "^29.1.0",
+    "typescript": "^5.0.0"
+  },
+  "dependencies": {
+    "entroplain": "^0.2.0",
+    "openai": "^4.0.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}

package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "entroplain"
-version = "0.2.0"
+version = "0.2.3"
 description = "Entropy-based early exit for efficient agent reasoning"
 readme = "README.md"
 license = "MIT"

package/scripts/setup.bat

@@ -0,0 +1,89 @@
+@echo off
+REM Entroplain Setup Script for Windows
+REM Run from PowerShell or Command Prompt
+
+echo ==============================================================
+echo   ENTROPAIN SETUP - Windows
+echo ==============================================================
+
+REM Check Python
+python --version >nul 2>&1
+if errorlevel 1 (
+    echo X Python is required but not installed.
+    echo   Download from: https://www.python.org/downloads/
+    echo   Make sure to check "Add Python to PATH" during installation.
+    exit /b 1
+)
+
+for /f "tokens=2" %%i in ('python --version 2^>^&1') do set PYTHON_VERSION=%%i
+echo + Python %PYTHON_VERSION% found
+
+REM Install entroplain
+echo.
+echo Installing entroplain...
+pip install --upgrade entroplain
+
+REM Verify installation
+python -c "import entroplain" 2>nul
+if errorlevel 1 (
+    echo X Installation failed
+    exit /b 1
+)
+echo + entroplain installed successfully
+
+REM Check for API keys
+echo.
+echo ==============================================================
+echo   API KEY SETUP
+echo ==============================================================
+
+set PROVIDERS=
+if defined OPENAI_API_KEY (
+    echo + OPENAI_API_KEY is set
+    set PROVIDERS=%PROVIDERS% openai
+)
+if defined ANTHROPIC_API_KEY (
+    echo + ANTHROPIC_API_KEY is set
+    set PROVIDERS=%PROVIDERS% anthropic
+)
+if defined NVIDIA_API_KEY (
+    echo + NVIDIA_API_KEY is set
+    set PROVIDERS=%PROVIDERS% nvidia
+)
+if defined GOOGLE_API_KEY (
+    echo + GOOGLE_API_KEY is set
+    set PROVIDERS=%PROVIDERS% google
+)
+
+if "%PROVIDERS%"=="" (
+    echo.
+    echo ! No API keys found. Set at least one:
+    echo.
+    echo    set OPENAI_API_KEY=your-key
+    echo    set ANTHROPIC_API_KEY=your-key
+    echo    set NVIDIA_API_KEY=your-key
+    echo    set GOOGLE_API_KEY=your-key
+    echo.
+    echo    To make permanent, add to System Environment Variables:
+    echo    - Search "Environment Variables" in Start menu
+    echo    - Edit "User variables" or "System variables"
+    echo    - Add new variable with name and key
+)
+
+echo.
+echo ==============================================================
+echo   QUICK START
+echo ==============================================================
+echo.
+echo   REM Analyze entropy:
+echo   entroplain analyze "Your prompt here"
+echo.
+echo   REM Stream with early exit:
+echo   entroplain stream --exit-on-converge "Your prompt here"
+echo.
+echo   REM Start proxy with dashboard:
+echo   entroplain-proxy --port 8765 --provider openai
+echo   REM Then open http://localhost:8765/dashboard
+echo.
+echo ==============================================================
+echo Setup complete!

package/scripts/setup.sh

@@ -0,0 +1,98 @@
+#!/bin/bash
+# Entroplain Setup Script for macOS/Linux
+# Usage: curl -sSL https://raw.githubusercontent.com/entroplain/entroplain/main/scripts/setup.sh | bash
+
+set -e
+
+echo "=============================================================="
+echo "  ENTROPAIN SETUP - macOS/Linux"
+echo "=============================================================="
+
+# Colors
+GREEN='\033[0;32m'
+BLUE='\033[0;34m'
+NC='\033[0m'
+
+# Check Python
+if ! command -v python3 &> /dev/null; then
+    echo "❌ Python 3 is required but not installed."
+    echo "   Install with: brew install python3 (macOS) or sudo apt install python3 (Linux)"
+    exit 1
+fi
+
+PYTHON_VERSION=$(python3 --version 2>&1 | awk '{print $2}')
+echo -e "${GREEN}✓${NC} Python $PYTHON_VERSION found"
+
+# Check pip
+if ! command -v pip3 &> /dev/null; then
+    echo "❌ pip3 is required but not installed."
+    exit 1
+fi
+
+# Install entroplain
+echo ""
+echo -e "${BLUE}Installing entroplain...${NC}"
+pip3 install --upgrade entroplain
+
+# Verify installation
+if python3 -c "import entroplain" 2>/dev/null; then
+    echo -e "${GREEN}✓${NC} entroplain installed successfully"
+else
+    echo "❌ Installation failed"
+    exit 1
+fi
+
+# Check for API keys
+echo ""
+echo "=============================================================="
+echo "  API KEY SETUP"
+echo "=============================================================="
+
+PROVIDERS=""
+if [ -n "$OPENAI_API_KEY" ]; then
+    echo -e "${GREEN}✓${NC} OPENAI_API_KEY is set"
+    PROVIDERS="$PROVIDERS openai"
+fi
+if [ -n "$ANTHROPIC_API_KEY" ]; then
+    echo -e "${GREEN}✓${NC} ANTHROPIC_API_KEY is set"
+    PROVIDERS="$PROVIDERS anthropic"
+fi
+if [ -n "$NVIDIA_API_KEY" ]; then
+    echo -e "${GREEN}✓${NC} NVIDIA_API_KEY is set"
+    PROVIDERS="$PROVIDERS nvidia"
+fi
+if [ -n "$GOOGLE_API_KEY" ]; then
+    echo -e "${GREEN}✓${NC} GOOGLE_API_KEY is set"
+    PROVIDERS="$PROVIDERS google"
+fi
+
+if [ -z "$PROVIDERS" ]; then
+    echo ""
+    echo "⚠️  No API keys found. Set at least one:"
+    echo ""
+    echo "   export OPENAI_API_KEY='your-key'"
+    echo "   export ANTHROPIC_API_KEY='your-key'"
+    echo "   export NVIDIA_API_KEY='your-key'"
+    echo "   export GOOGLE_API_KEY='your-key'"
+    echo ""
+    echo "   Add to ~/.bashrc or ~/.zshrc to persist"
+fi
+
+# Usage examples
+echo ""
+echo "=============================================================="
+echo "  QUICK START"
+echo "=============================================================="
+echo ""
+echo "  # Analyze entropy:"
+echo "  entroplain analyze \"Your prompt here\""
+echo ""
+echo "  # Stream with early exit:"
+echo "  entroplain stream --exit-on-converge \"Your prompt here\""
+echo ""
+echo "  # Start proxy with dashboard:"
+echo "  entroplain-proxy --port 8765 --provider openai"
+echo "  # Then open http://localhost:8765/dashboard"
+echo ""
+echo "=============================================================="
+echo -e "${GREEN}Setup complete!${NC}"

package/test_nvidia.py

@@ -1,56 +1,56 @@
-"""Test the proxy with NVIDIA API."""
-
-import requests
-import json
-import os
-
-# Get API key from environment
-api_key = os.environ.get("NVIDIA_API_KEY", "")
-
-if not api_key:
-    print("ERROR: NVIDIA_API_KEY not set")
-    exit(1)
-
-# Make request through proxy
-response = requests.post(
-    "http://localhost:8767/v1/chat/completions",
-    headers={
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {api_key}"
-    },
-    json={
-        "model": "meta/llama-3.1-70b-instruct",
-        "messages": [{"role": "user", "content": "What is 2+2? Just answer the number."}],
-        "max_tokens": 50,
-        "temperature": 0.1,
-        "stream": True
-    },
-    stream=True
-)
-
-print(f"Status: {response.status_code}")
-print("Streaming response:")
-print("-" * 40)
-
-for line in response.iter_lines():
-    if line:
-        line = line.decode('utf-8')
-        if line.startswith("data: "):
-            data = line[6:]
-            if data == "[DONE]":
-                print("\n[DONE]")
-                break
-            try:
-                chunk = json.loads(data)
-                if chunk.get("choices"):
-                    delta = chunk["choices"][0].get("delta", {})
-                    if delta.get("content"):
-                        print(delta["content"], end="", flush=True)
-            except json.JSONDecodeError:
-                pass
-
-print("\n" + "-" * 40)
-
-# Check proxy health
-health = requests.get("http://localhost:8767/health")
-print(f"\nProxy stats: {health.json()}")
+"""Test the proxy with NVIDIA API."""
+
+import requests
+import json
+import os
+
+# Get API key from environment
+api_key = os.environ.get("NVIDIA_API_KEY", "")
+
+if not api_key:
+    print("ERROR: NVIDIA_API_KEY not set")
+    exit(1)
+
+# Make request through proxy
+response = requests.post(
+    "http://localhost:8767/v1/chat/completions",
+    headers={
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    },
+    json={
+        "model": "meta/llama-3.1-70b-instruct",
+        "messages": [{"role": "user", "content": "What is 2+2? Just answer the number."}],
+        "max_tokens": 50,
+        "temperature": 0.1,
+        "stream": True
+    },
+    stream=True
+)
+
+print(f"Status: {response.status_code}")
+print("Streaming response:")
+print("-" * 40)
+
+for line in response.iter_lines():
+    if line:
+        line = line.decode('utf-8')
+        if line.startswith("data: "):
+            data = line[6:]
+            if data == "[DONE]":
+                print("\n[DONE]")
+                break
+            try:
+                chunk = json.loads(data)
+                if chunk.get("choices"):
+                    delta = chunk["choices"][0].get("delta", {})
+                    if delta.get("content"):
+                        print(delta["content"], end="", flush=True)
+            except json.JSONDecodeError:
+                pass
+
+print("\n" + "-" * 40)
+
+# Check proxy health
+health = requests.get("http://localhost:8767/health")
+print(f"\nProxy stats: {health.json()}")

package/test_proxy.py

@@ -1,16 +1,16 @@
-"""Test the proxy with a real API call."""
-
-import requests
-import json
-
-# Test health endpoint
-try:
-    response = requests.get("http://localhost:8765/health")
-    print(f"Health check: {response.status_code}")
-    print(response.json())
-except Exception as e:
-    print(f"Proxy not running: {e}")
-    print("\nTo test the proxy, run:")
-    print("  entroplain-proxy --port 8765")
-    print("\nThen in another terminal:")
-    print("  python test_proxy.py")
+"""Test the proxy with a real API call."""
+
+import requests
+import json
+
+# Test health endpoint
+try:
+    response = requests.get("http://localhost:8765/health")
+    print(f"Health check: {response.status_code}")
+    print(response.json())
+except Exception as e:
+    print(f"Proxy not running: {e}")
+    print("\nTo test the proxy, run:")
+    print("  entroplain-proxy --port 8765")
+    print("\nThen in another terminal:")
+    print("  python test_proxy.py")

package/vercel.json

@@ -0,0 +1,6 @@
+{
+  "buildCommand": "cd website && npm install && npm run build",
+  "outputDirectory": "website/out",
+  "framework": "nextjs",
+  "installCommand": "npm install"
+}

package/website/README.md

@@ -0,0 +1,14 @@
+# Entroplain Website
+
+Built with Next.js, deployed on Vercel.
+
+## Development
+
+```bash
+npm install
+npm run dev
+```
+
+## Deploy
+
+Push to main branch - Vercel auto-deploys.

package/website/app/globals.css

@@ -0,0 +1,88 @@
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
+
+:root {
+  --bg: #ffffff;
+  --bg-dark: #0a0a0a;
+  --fg: #171717;
+  --fg-muted: #666666;
+  --accent: #4ade80;
+  --accent-hover: #22c55e;
+  --border: rgba(0, 0, 0, 0.08);
+  --card: #fafafa;
+  --code-bg: #1e1e1e;
+}
+
+* {
+  box-sizing: border-box;
+  margin: 0;
+  padding: 0;
+}
+
+html {
+  scroll-behavior: smooth;
+}
+
+body {
+  font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+  background: var(--bg);
+  color: var(--fg);
+  line-height: 1.5;
+  -webkit-font-smoothing: antialiased;
+}
+
+/* Vercel-style shadow-as-border */
+.border-shadow {
+  box-shadow: rgba(0, 0, 0, 0.08) 0px 0px 0px 1px;
+}
+
+.card-shadow {
+  box-shadow: 
+    rgba(0, 0, 0, 0.08) 0px 0px 0px 1px,
+    rgba(0, 0, 0, 0.04) 0px 2px 2px,
+    rgba(0, 0, 0, 0.04) 0px 8px 8px -8px;
+}
+
+/* Code blocks */
+code, pre {
+  font-family: 'SF Mono', 'Fira Code', 'Consolas', monospace;
+}
+
+pre {
+  background: var(--code-bg);
+  border-radius: 8px;
+  padding: 16px;
+  overflow-x: auto;
+}
+
+code {
+  color: #e0e0e0;
+}
+
+/* Tight tracking for headings */
+h1, h2, h3 {
+  letter-spacing: -0.02em;
+}
+
+/* Selection */
+::selection {
+  background: rgba(74, 222, 128, 0.2);
+}
+
+/* Scrollbar */
+::-webkit-scrollbar {
+  width: 8px;
+  height: 8px;
+}
+
+::-webkit-scrollbar-track {
+  background: transparent;
+}
+
+::-webkit-scrollbar-thumb {
+  background: rgba(0, 0, 0, 0.2);
+  border-radius: 4px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+  background: rgba(0, 0, 0, 0.3);
+}

package/website/app/layout.tsx

@@ -0,0 +1,34 @@
+import type { Metadata } from 'next'
+import { Analytics } from '@vercel/analytics/next'
+import { SpeedInsights } from '@vercel/speed-insights/next'
+import './globals.css'
+
+export const metadata: Metadata = {
+  title: 'Entroplain - Early Exit for Efficient Agent Reasoning',
+  description: 'Stop wasting tokens. Entroplain detects when LLMs have confidently answered and exits early, saving up to 50% on API costs.',
+  openGraph: {
+    title: 'Entroplain - Early Exit for Efficient Agent Reasoning',
+    description: 'Stop wasting tokens. Detect confident answers and exit early.',
+    type: 'website',
+  },
+}
+
+export default function RootLayout({
+  children,
+}: {
+  children: React.ReactNode
+}) {
+  return (
+    <html lang="en">
+      <head>
+        <link rel="preconnect" href="https://fonts.googleapis.com" />
+        <link rel="preconnect" href="https://fonts.gstatic.com" crossOrigin="anonymous" />
+        <link href="https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap" rel="stylesheet" />
+      </head>
+      <body style={{ margin: 0, padding: 0 }}>{children}
+        <Analytics />
+        <SpeedInsights />
+      </body>
+    </html>
+  )
+}