Schema

.github/workflows/ci.yml

@@ -0,0 +1,34 @@
+name: CI
+
+on:
+  push:
+    branches: [main, master]
+  pull_request:
+    branches: [main, master]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Run linter
+        run: ruff check src/ tests/
+
+      - name: Run tests
+        run: pytest tests/ -v

.gitignore

@@ -56,3 +56,7 @@ htmlcov/
 # Build output
 dist/
 build/
+
+# Agent/workflow tracking
+.agent/
+.beads/

src/tpuff/cli.py

@@ -3,13 +3,13 @@
 import click
 
 from tpuff import __version__
-from tpuff.commands.list import list_cmd
-from tpuff.commands.search import search
 from tpuff.commands.delete import delete
 from tpuff.commands.edit import edit
-from tpuff.commands.get import get
 from tpuff.commands.export import export
-
+from tpuff.commands.get import get
+from tpuff.commands.list import list_cmd
+from tpuff.commands.schema import schema
+from tpuff.commands.search import search
 
 # Context settings to enable -h as help alias for all commands
 CONTEXT_SETTINGS = {"help_option_names": ["-h", "--help"]}
@@ -35,6 +35,7 @@ def cli(ctx: click.Context, debug: bool) -> None:
 cli.add_command(get)
 cli.add_command(export)
 cli.add_command(export, name="metrics")  # alias
+cli.add_command(schema)
 
 
 def main() -> None:

src/tpuff/client.py

@@ -8,7 +8,6 @@
 from tpuff.utils.debug import debug_log
 from tpuff.utils.regions import DEFAULT_REGION
 
-
 # Global client cache to avoid re-creating clients for the same region
 _client_cache: dict[str, Turbopuffer] = {}
 

src/tpuff/commands/delete.py

@@ -8,7 +8,6 @@
 from tpuff.client import get_turbopuffer_client
 from tpuff.utils.debug import debug_log
 
-
 console = Console()
 
 
@@ -91,7 +90,7 @@ def delete(
                 console.print(f"[dim]  - {ns.id}[/dim]")
 
             console.print("\n[bold yellow]💀 This is your last chance to back out! 💀[/bold yellow]")
-            console.print(f"[dim]To confirm, please type: [bold red]yolo[/bold red][/dim]\n")
+            console.print("[dim]To confirm, please type: [bold red]yolo[/bold red][/dim]\n")
 
             answer = prompt_user(">")
 

src/tpuff/commands/edit.py

@@ -12,7 +12,6 @@
 from tpuff.client import get_namespace
 from tpuff.utils.debug import debug_log
 
-
 console = Console()
 
 

src/tpuff/commands/export.py

@@ -6,7 +6,7 @@
 import threading
 import time
 from datetime import datetime
-from http.server import HTTPServer, BaseHTTPRequestHandler
+from http.server import BaseHTTPRequestHandler, HTTPServer
 from typing import Any
 
 import click
@@ -21,14 +21,13 @@
     get_unindexed_bytes,
 )
 from tpuff.utils.metrics import (
-    PrometheusMetric,
     MetricValue,
-    format_prometheus_metrics,
+    PrometheusMetric,
     create_simple_gauge,
+    format_prometheus_metrics,
     get_current_timestamp,
 )
 
-
 console = Console()
 
 
@@ -382,11 +381,11 @@ def shutdown_handler(signum: int, frame: Any) -> None:
         console.print("[dim]                  Note: Recall estimation runs queries and incurs costs[/dim]")
     else:
         console.print("[dim]  Recall:         disabled (use --include-recall to enable)[/dim]")
-    console.print(f"\n[dim]  Endpoints:[/dim]")
+    console.print("\n[dim]  Endpoints:[/dim]")
     console.print(f"[dim]    http://localhost:{port}/metrics[/dim]")
     console.print(f"[dim]    http://localhost:{port}/health[/dim]")
     console.print(f"[dim]    http://localhost:{port}/[/dim]")
-    console.print(f"\n[dim]  Press Ctrl+C to stop[/dim]\n")
+    console.print("\n[dim]  Press Ctrl+C to stop[/dim]\n")
 
     # Start server
     try:

src/tpuff/commands/get.py

@@ -9,7 +9,6 @@
 from tpuff.client import get_namespace
 from tpuff.utils.debug import debug_log
 
-
 console = Console()
 
 

src/tpuff/commands/list.py

@@ -8,7 +8,7 @@
 from rich.console import Console
 from rich.table import Table
 
-from tpuff.client import get_namespace, get_turbopuffer_client
+from tpuff.client import get_namespace
 from tpuff.utils.debug import debug_log
 from tpuff.utils.metadata_fetcher import (
     NamespaceWithMetadata,
@@ -17,7 +17,6 @@
     get_unindexed_bytes,
 )
 
-
 console = Console()