programasweights 0.1.0.dev2__tar.gz → 0.1.0.dev4__tar.gz

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: programasweights
-Version: 0.1.0.dev2
+Version: 0.1.0.dev4
 Summary: Compile natural language specifications into neural programs that run locally via llama.cpp.
 Author-email: ProgramAsWeights <support@programasweights.com>
 License: MIT
@@ -17,8 +17,8 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: httpx>=0.27.0
-Requires-Dist: llama-cpp-python>=0.3.0
+Requires-Dist: httpx<1.0,>=0.27.0
+Requires-Dist: llama-cpp-python<1.0,>=0.3.0
 Provides-Extra: test
 Requires-Dist: pytest; extra == 'test'
 Description-Content-Type: text/markdown

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/programasweights/__init__.py

@@ -19,7 +19,7 @@ API reference:
     paw.api_key                API key (set via login() or PAW_API_KEY env var)
 """
 
-__version__ = "0.1.0.dev2"
+__version__ = "0.1.0.dev4"
 
 from .config import get_api_url, get_api_key, set_api_key
 

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/programasweights/cache.py

@@ -22,7 +22,7 @@ import httpx
 from . import config
 
 BASE_MODEL_URLS = {
-    "qwen3-0.6b-q6_k": "https://programasweights.com/models/qwen3-0.6b-q6_k.gguf",
+    "qwen3-0.6b-q6_k": "https://huggingface.co/yuntian-deng/Qwen3-0.6B-GGUF-Q6_K/resolve/main/qwen3-0.6b-q6_k.gguf",
 }
 
 INTERPRETER_TO_GGUF = {

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "programasweights"
-version = "0.1.0.dev2"
+version = "0.1.0.dev4"
 description = "Compile natural language specifications into neural programs that run locally via llama.cpp."
 readme = "PYPI_README.md"
 requires-python = ">=3.9"
@@ -27,8 +27,8 @@ classifiers = [
   "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
-  "httpx>=0.27.0",
-  "llama-cpp-python>=0.3.0",
+  "httpx>=0.27.0,<1.0",
+  "llama-cpp-python>=0.3.0,<1.0",
 ]
 
 [project.optional-dependencies]

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/server/api/models/schemas.py

@@ -52,6 +52,7 @@ class CompileResponse(BaseModel):
     status: Literal["queued", "compiling", "ready", "failed"]
     program_id: Optional[str] = None
     compiler_snapshot: Optional[str] = None
+    pseudo_program: Optional[str] = None
     queue_position: Optional[int] = None
     estimated_wait_s: Optional[float] = None
     timings: Optional[CompileTimings] = None

programasweights-0.1.0.dev4/server/api/routes/auth.py

@@ -0,0 +1,184 @@
+"""
+Auth endpoints: GitHub OAuth + session management.
+"""
+
+import hashlib
+import secrets
+import time
+
+import httpx
+import structlog
+from fastapi import APIRouter, HTTPException, Request
+from fastapi.responses import RedirectResponse
+
+from api.config import get_settings
+
+logger = structlog.get_logger()
+router = APIRouter()
+
+# In-memory session store (replace with Redis/DB for production)
+_sessions: dict[str, dict] = {}
+_github_states: dict[str, float] = {}
+
+GITHUB_AUTHORIZE_URL = "https://github.com/login/oauth/authorize"
+GITHUB_TOKEN_URL = "https://github.com/login/oauth/access_token"
+GITHUB_USER_URL = "https://api.github.com/user"
+
+
+@router.get("/auth/github")
+async def github_login(request: Request):
+    settings = get_settings()
+    if not settings.GITHUB_CLIENT_ID:
+        raise HTTPException(status_code=501, detail={
+            "error": "not_configured",
+            "message": "GitHub OAuth is not configured. Set GITHUB_CLIENT_ID and GITHUB_CLIENT_SECRET.",
+        })
+
+    state = secrets.token_urlsafe(32)
+    _github_states[state] = time.time()
+
+    # Clean up states older than 10 minutes
+    cutoff = time.time() - 600
+    for k in [k for k, v in _github_states.items() if v < cutoff]:
+        del _github_states[k]
+
+    params = (
+        f"?client_id={settings.GITHUB_CLIENT_ID}"
+        f"&redirect_uri=https://programasweights.com/api/v1/auth/github/callback"
+        f"&scope=read:user"
+        f"&state={state}"
+    )
+    return RedirectResponse(url=f"{GITHUB_AUTHORIZE_URL}{params}")
+
+
+@router.get("/auth/github/callback")
+async def github_callback(code: str, state: str, request: Request):
+    if state not in _github_states:
+        raise HTTPException(status_code=400, detail={
+            "error": "invalid_state",
+            "message": "Invalid or expired OAuth state. Please try signing in again.",
+        })
+    del _github_states[state]
+
+    settings = get_settings()
+
+    async with httpx.AsyncClient() as client:
+        token_resp = await client.post(
+            GITHUB_TOKEN_URL,
+            data={
+                "client_id": settings.GITHUB_CLIENT_ID,
+                "client_secret": settings.GITHUB_CLIENT_SECRET,
+                "code": code,
+                "redirect_uri": "https://programasweights.com/api/v1/auth/github/callback",
+            },
+            headers={"Accept": "application/json"},
+        )
+        token_data = token_resp.json()
+
+        if "access_token" not in token_data:
+            logger.warning("github_oauth_failed", error=token_data.get("error"))
+            raise HTTPException(status_code=400, detail={
+                "error": "oauth_failed",
+                "message": token_data.get("error_description", "GitHub OAuth failed"),
+            })
+
+        access_token = token_data["access_token"]
+
+        user_resp = await client.get(
+            GITHUB_USER_URL,
+            headers={
+                "Authorization": f"Bearer {access_token}",
+                "Accept": "application/json",
+            },
+        )
+        user_data = user_resp.json()
+
+    session_token = secrets.token_urlsafe(48)
+    user_info = {
+        "authenticated": True,
+        "user_id": str(user_data.get("id")),
+        "username": user_data.get("login"),
+        "display_name": user_data.get("name") or user_data.get("login"),
+        "email": user_data.get("email"),
+        "avatar_url": user_data.get("avatar_url"),
+        "tier": "github",
+        "github_access_token": access_token,
+        "created_at": time.time(),
+    }
+    _sessions[session_token] = user_info
+
+    logger.info("github_login_success",
+                username=user_info["username"],
+                user_id=user_info["user_id"])
+
+    response = RedirectResponse(url="/playground")
+    response.set_cookie(
+        key="paw_session",
+        value=session_token,
+        httponly=True,
+        secure=True,
+        samesite="lax",
+        max_age=30 * 24 * 3600,
+    )
+    return response
+
+
+@router.get("/auth/me")
+async def get_current_user(request: Request):
+    session_token = request.cookies.get("paw_session")
+    api_key = request.headers.get("X-API-Key")
+
+    if session_token and session_token in _sessions:
+        user = _sessions[session_token]
+        return {
+            "authenticated": True,
+            "user_id": user["user_id"],
+            "username": user["username"],
+            "display_name": user["display_name"],
+            "email": user.get("email"),
+            "avatar_url": user.get("avatar_url"),
+            "tier": user["tier"],
+        }
+
+    if api_key and api_key == "test_key_for_development":
+        return {
+            "authenticated": True,
+            "user_id": "dev_user",
+            "username": "dev",
+            "display_name": "Developer",
+            "tier": "github",
+        }
+
+    return {"authenticated": False}
+
+
+@router.post("/auth/logout")
+async def logout(request: Request):
+    session_token = request.cookies.get("paw_session")
+    if session_token and session_token in _sessions:
+        del _sessions[session_token]
+
+    response = RedirectResponse(url="/", status_code=303)
+    response.delete_cookie("paw_session")
+    return response
+
+
+@router.post("/auth/email")
+async def send_email_code(request: Request):
+    return {"message": "Email auth coming soon. Please use GitHub sign-in."}
+
+
+@router.post("/auth/verify")
+async def verify_code(request: Request):
+    raise HTTPException(status_code=501, detail={
+        "error": "not_implemented",
+        "message": "Email auth coming soon. Please use GitHub sign-in.",
+    })
+
+
+@router.post("/auth/rotate-key")
+async def rotate_api_key(request: Request):
+    raise HTTPException(status_code=501, detail={
+        "error": "not_implemented",
+        "message": "API key rotation coming soon.",
+    })

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/server/api/routes/compile.py

@@ -91,6 +91,7 @@ async def submit_compile(req: CompileRequest, request: Request):
             status="ready",
             program_id=result["program_id"],
             compiler_snapshot=snapshot,
+            pseudo_program=result.get("pseudo_program"),
             queue_position=0,
             estimated_wait_s=0,
             timings=CompileTimings(

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/server/api/services/compile_service.py

@@ -139,6 +139,8 @@ class CompileService:
         self._prefix_steps = None
         self._compiler_fingerprint = None
 
+        self._pseudo_cache: dict[str, str] = {}
+
     def _ensure_tokenizers_loaded(self, checkpoint_dir: str):
         if self._compiler_tok is not None:
             return
@@ -212,6 +214,10 @@ class CompileService:
         content = id_bytes + (self._compiler_fingerprint or "unknown").encode()
         return hashlib.sha256(content).hexdigest()[:20]
 
+    def _pseudo_cache_key(self, rendered_prompt: str, max_tokens: int) -> str:
+        content = f"{rendered_prompt}|max_tokens={max_tokens}|temp=0".encode()
+        return hashlib.sha256(content).hexdigest()[:32]
+
     async def step1_generate_pseudo_program(self, spec: str) -> tuple[str, int]:
         prompt = compiler_prompt_examples(spec)
         rendered = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n<think>\n\n</think>\n\n"
@@ -220,6 +226,12 @@ class CompileService:
         else:
             prompt_tokens = len(rendered) // 4
         max_pseudo = min(COMPILER_MAX_PSEUDO_TOKENS, COMPILER_TOKEN_BUDGET - prompt_tokens)
+        max_tokens = max(max_pseudo, 64)
+
+        cache_key = self._pseudo_cache_key(rendered, max_tokens)
+        if cache_key in self._pseudo_cache:
+            logger.info("pseudo_gen_cache_hit", spec_len=len(spec))
+            return self._pseudo_cache[cache_key], prompt_tokens
 
         async with httpx.AsyncClient(timeout=60.0) as client:
             resp = await client.post(
@@ -227,13 +239,15 @@ class CompileService:
                 json={
                     "model": "Qwen/Qwen3-4B-Instruct-2507",
                     "prompt": rendered,
-                    "max_tokens": max(max_pseudo, 64),
+                    "max_tokens": max_tokens,
                     "temperature": 0,
                 },
             )
             resp.raise_for_status()
             data = resp.json()
-            return data["choices"][0]["text"].strip(), prompt_tokens
+            pseudo_program = data["choices"][0]["text"].strip()
+            self._pseudo_cache[cache_key] = pseudo_program
+            return pseudo_program, prompt_tokens
 
     def _build_compiler_input(self, spec: str, pseudo_program: str) -> str:
         """Build the full input text for the compiler model."""
@@ -359,6 +373,27 @@ class CompileService:
         timings["pseudo_gen_s"] = time.time() - t0
         logger.info("compile_step1_done", elapsed_s=round(timings["pseudo_gen_s"], 3))
 
+        # Generate content-addressable program ID early (needs pseudo_program)
+        compiler_input = self._build_compiler_input(spec, pseudo_program)
+        program_id = self._generate_program_id(compiler_input)
+
+        # Cache check: if this exact program already exists, return immediately
+        adapter_dir = self._storage_path / program_id
+        paw_path = str(adapter_dir / f"{program_id}.paw")
+        if (adapter_dir / "adapter.gguf").exists() and (adapter_dir / "prompt_template.txt").exists():
+            timings["total_s"] = time.time() - t_start
+            logger.info("compile_cache_hit", program_id=program_id,
+                        total_s=round(timings["total_s"], 3))
+            return {
+                "program_id": program_id,
+                "paw_path": paw_path,
+                "pseudo_program": pseudo_program,
+                "adapter_path": str(adapter_dir / "adapter.gguf"),
+                "peft_adapter_dir": str(adapter_dir),
+                "timings": timings,
+                "cached": True,
+            }
+
         # Step 2: Extract LoRA via vLLM compiler
         logger.info("compile_step2_start")
         t0 = time.time()
@@ -371,10 +406,6 @@ class CompileService:
             torch.load, lora_params_path, map_location="cpu", weights_only=True,
         )
 
-        # Generate content-addressable program ID
-        compiler_input = self._build_compiler_input(spec, pseudo_program)
-        program_id = self._generate_program_id(compiler_input)
-
         # Copy PEFT adapter + raw params to program storage
         adapter_dir = self._storage_path / program_id
         adapter_dir.mkdir(parents=True, exist_ok=True)

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/web-app/frontend/src/App.tsx

@@ -8,7 +8,7 @@ import HubPage from './components/HubPage';
 import HubUploadPage from './components/HubUploadPage';
 import HubProgramPage from './components/HubProgramPage';
 import ProfilePage from './components/ProfilePage';
-import PricingPage from './components/PricingPage';
+
 import DocsPage from './components/DocsPage';
 import TermsPage from './components/TermsPage';
 import PrivacyPage from './components/PrivacyPage';
@@ -38,7 +38,7 @@ function App() {
             <Route path="/hub/:id" element={<HubProgramPage />} />
             <Route path="/profile" element={<ProfilePage />} />
             <Route path="/about" element={<AboutPage />} />
-            <Route path="/pricing" element={<PricingPage />} />
+            {/* pricing removed — free during internal testing */}
             <Route path="/docs" element={<DocsPage />} />
             <Route path="/terms" element={<TermsPage />} />
             <Route path="/privacy" element={<PrivacyPage />} />

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/web-app/frontend/src/components/DocsPage.tsx

@@ -85,7 +85,7 @@ const DocsPage: React.FC = () => {
         <div className="space-y-8">
           <section>
             <h2 className="text-xl font-bold text-gray-900 mb-4">1. Install</h2>
-            <CodeBlock code="pip install programasweights" id="install" lang="bash" />
+            <CodeBlock code="pip install --pre programasweights" id="install" lang="bash" />
           </section>
 
           <section>

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/web-app/frontend/src/components/Footer.tsx

@@ -18,7 +18,7 @@ const Footer: React.FC = () => {
               Define functions in English. Compile them into neural programs. Run them locally.
             </p>
             <div className="mt-4 inline-flex items-center gap-2 bg-white/5 rounded px-3 py-1.5 font-mono text-xs text-gray-300 border border-white/10">
-              pip install programasweights
+              pip install --pre programasweights
             </div>
           </div>
 

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/web-app/frontend/src/components/Header.tsx

@@ -1,6 +1,6 @@
 import { useState, useEffect } from 'react';
 import { Link, useLocation } from 'react-router-dom';
-import { BookOpen, Package, LogOut, Play, Menu, X, DollarSign, Github } from 'lucide-react';
+import { BookOpen, Package, LogOut, Play, Menu, X, Github } from 'lucide-react';
 import { apiClient } from '../utils/api';
 import type { AuthUser } from '../types';
 
@@ -14,7 +14,6 @@ const Header: React.FC = () => {
     { path: '/playground', label: 'Playground', icon: Play },
     { path: '/hub', label: 'Hub', icon: Package },
     { path: '/docs', label: 'Docs', icon: BookOpen },
-    { path: '/pricing', label: 'Pricing', icon: DollarSign },
   ];
 
   const isActive = (path: string) => location.pathname === path;
@@ -110,13 +109,13 @@ const Header: React.FC = () => {
                 )}
               </div>
             ) : (
-              <Link
-                to="/playground"
+              <a
+                href="/api/v1/auth/github"
                 className="hidden sm:flex items-center space-x-2 px-4 py-2 bg-gray-900 text-white rounded-lg hover:bg-gray-800 transition-colors text-sm font-medium"
               >
-                <Play className="w-4 h-4" />
-                <span>Get Started</span>
-              </Link>
+                <Github className="w-4 h-4" />
+                <span>Sign in</span>
+              </a>
             )}
 
             <button

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/web-app/frontend/src/components/HubProgramPage.tsx

@@ -214,7 +214,7 @@ print(result)`;
         </div>
         <pre className="bg-gray-900 text-gray-100 p-4 rounded-lg text-sm overflow-x-auto leading-relaxed">{usageCode}</pre>
         <p className="mt-3 text-xs text-gray-400">
-          Install with <code className="bg-gray-100 px-1.5 py-0.5 rounded text-gray-600">pip install programasweights</code>
+          Install with <code className="bg-gray-100 px-1.5 py-0.5 rounded text-gray-600">pip install --pre programasweights</code>
         </p>
       </div>
 

{programasweights-0.1.0.dev2 → programasweights-0.1.0.dev4}/web-app/frontend/src/components/LandingPage.tsx

@@ -63,7 +63,7 @@ const LandingPage: React.FC = () => {
     },
   ];
 
-  const installCmd = 'pip install programasweights';
+  const installCmd = 'pip install --pre programasweights';
   const codeExample = `import programasweights as paw
 
 # Define a function in plain English