Compare commits

..

13 Commits

Author SHA1 Message Date
Alexander Whitestone
49296d538e feat: nightly pipeline scheduler — auto-start when inference available (#624)
Some checks failed
Architecture Lint / Lint Repository (pull_request) Blocked by required conditions
Validate Config / Python Test Suite (pull_request) Blocked by required conditions
Architecture Lint / Linter Tests (pull_request) Successful in 36s
PR Checklist / pr-checklist (pull_request) Failing after 6m56s
Smoke Test / smoke (pull_request) Failing after 34s
Validate Config / YAML Lint (pull_request) Failing after 21s
Validate Config / JSON Validate (pull_request) Successful in 9s
Validate Config / Python Syntax & Import Check (pull_request) Failing after 1m40s
Validate Config / Shell Script Lint (pull_request) Failing after 48s
Validate Config / Cron Syntax Check (pull_request) Successful in 18s
Validate Config / Deploy Script Dry Run (pull_request) Successful in 20s
Validate Config / Playbook Schema Validation (pull_request) Successful in 38s
Scheduler that auto-starts batch pipelines when inference is available.

Features:
- Checks inference provider availability (local Ollama, RunPod, OpenRouter)
- Priority ordering: playground > training > knowledge > adversary > genome
- Dependency rules (e.g., knowledge_mine waits for training_factory)
- Daily token budget (5M default, configurable)
- Peak-hour pausing (8am-10pm = interactive mode, no pipelines)
- State persistence via ~/.hermes/pipeline_state.json
- One pipeline per cycle to avoid overload

Usage:
  python3 pipeline/nightly_scheduler.py --status
  python3 pipeline/nightly_scheduler.py --check      # dry-run
  python3 pipeline/nightly_scheduler.py              # live

Cron: */30 22-5 * * * pipeline/nightly_scheduler.py

Closes #624
2026-04-15 08:14:00 -04:00
817785d763 Merge pull request 'feat: training data augmentation — paraphrase and translate pairs (#695)' (#732) from fix/695 into main 2026-04-15 11:56:28 +00:00
Alexander Whitestone
3603030235 feat: training data augmentation — paraphrase and translate pairs (#695)
Some checks failed
Architecture Lint / Lint Repository (pull_request) Blocked by required conditions
Validate Config / Python Test Suite (pull_request) Blocked by required conditions
Architecture Lint / Linter Tests (pull_request) Successful in 22s
Smoke Test / smoke (pull_request) Failing after 18s
Validate Config / YAML Lint (pull_request) Failing after 23s
Validate Config / JSON Validate (pull_request) Successful in 21s
Validate Config / Python Syntax & Import Check (pull_request) Failing after 1m54s
Validate Config / Shell Script Lint (pull_request) Failing after 54s
Validate Config / Cron Syntax Check (pull_request) Successful in 16s
Validate Config / Deploy Script Dry Run (pull_request) Successful in 16s
Validate Config / Playbook Schema Validation (pull_request) Successful in 23s
PR Checklist / pr-checklist (pull_request) Failing after 11m2s
augment_pairs.py: generates paraphrases and translations for any
JSONL training file.

Features:
- Auto-detects text field (rich, terse, text, content, lyric_line, etc.)
- N paraphrases per entry (template-based, or LLM with --llm-endpoint)
- Translations to ES, FR, DE (template dictionary, or LLM)
- Outputs augmented JSONL alongside originals
- Marks each augmented entry with _augmentation, _original, _language

Usage:
  python3 augment_pairs.py --input data.jsonl
  python3 augment_pairs.py --input data.jsonl --paraphrases 5 --langs es,fr
  python3 augment_pairs.py --input data.jsonl --llm-endpoint http://localhost:11434/v1

Closes #695
2026-04-15 07:51:38 -04:00
35a191f7b1 Merge PR #725: feat: Provider health monitor with auto-switch (#509) 2026-04-15 06:10:45 +00:00
e987e1b870 Merge PR #726: feat: Pre-flight provider check for session launch (#508) 2026-04-15 06:10:41 +00:00
19278513b4 Merge PR #727: feat: Three.js-specific glitch detection patterns (#543) 2026-04-15 06:10:38 +00:00
1088bf8983 test: add Three.js pattern tests and update assertions (#543)
Some checks failed
Architecture Lint / Lint Repository (pull_request) Blocked by required conditions
Validate Config / Python Test Suite (pull_request) Blocked by required conditions
Architecture Lint / Linter Tests (pull_request) Successful in 28s
Smoke Test / smoke (pull_request) Failing after 23s
Validate Config / YAML Lint (pull_request) Failing after 21s
Validate Config / JSON Validate (pull_request) Successful in 21s
Validate Config / Python Syntax & Import Check (pull_request) Failing after 1m23s
Validate Config / Shell Script Lint (pull_request) Failing after 50s
Validate Config / Cron Syntax Check (pull_request) Successful in 11s
Validate Config / Deploy Script Dry Run (pull_request) Successful in 26s
Validate Config / Playbook Schema Validation (pull_request) Successful in 32s
PR Checklist / pr-checklist (pull_request) Failing after 11m13s
- Added TestThreeJsPatterns class with 14 tests
- Tests cover: pattern existence, severity inference, vision prompt
- Updated pattern count assertion (14+ patterns now)
- Updated demo test (6 glitches: 4 original + 2 Three.js)
2026-04-15 05:37:17 +00:00
94f0a132d4 feat: add get_threejs_patterns() filter function (#543) 2026-04-15 05:34:17 +00:00
279356bed6 feat: add --threejs flag and Three.js-aware severity inference (#543)
- Added --threejs flag for focused Three.js pattern scanning
- Updated _infer_severity with shader_failure, texture_placeholder,
  uv_mapping_error, frustum_culling, shadow_map_artifact categories
- Added Three.js demo detections (shader failure, shadow map)
- Bumped detector version to 0.2.0
2026-04-15 05:34:16 +00:00
511ff863c2 feat: add Three.js-specific glitch detection patterns (#543)
Adds 6 new Three.js-specific glitch categories and patterns:
- SHADER_FAILURE: Solid black materials from shader compilation errors
- TEXTURE_PLACEHOLDER: 1x1 white pixel stretched surfaces
- UV_MAPPING_ERROR: BufferGeometry UV coordinate errors
- FRUSTUM_CULLING: Objects popping at screen edges
- SHADOW_MAP_ARTIFACT: Pixelated/blocky shadow maps
- BLOOM_OVERFLOW: Excessive post-processing bloom bleed

Closes #543
2026-04-15 05:32:25 +00:00
b6e3a647b0 feat: add pre-flight provider check script (#508)
Some checks failed
Architecture Lint / Lint Repository (pull_request) Blocked by required conditions
Validate Config / Python Test Suite (pull_request) Blocked by required conditions
Architecture Lint / Linter Tests (pull_request) Successful in 29s
PR Checklist / pr-checklist (pull_request) Failing after 7m23s
Smoke Test / smoke (pull_request) Failing after 20s
Validate Config / YAML Lint (pull_request) Failing after 14s
Validate Config / JSON Validate (pull_request) Successful in 15s
Validate Config / Python Syntax & Import Check (pull_request) Failing after 1m1s
Validate Config / Shell Script Lint (pull_request) Failing after 46s
Validate Config / Cron Syntax Check (pull_request) Successful in 9s
Validate Config / Deploy Script Dry Run (pull_request) Successful in 10s
Validate Config / Playbook Schema Validation (pull_request) Successful in 28s
- Checks OpenRouter balance via /api/v1/auth/key
- Tests Nous and Anthropic API keys
- Verifies Ollama is running
- Pre-flight check before session launch
- Returns exit code for automation

Closes #508
2026-04-15 03:55:04 +00:00
e14158676d feat: add provider health monitor script (#509)
Some checks failed
Architecture Lint / Lint Repository (pull_request) Blocked by required conditions
Validate Config / Python Test Suite (pull_request) Blocked by required conditions
Architecture Lint / Linter Tests (pull_request) Successful in 44s
Smoke Test / smoke (pull_request) Failing after 36s
Validate Config / YAML Lint (pull_request) Failing after 21s
Validate Config / JSON Validate (pull_request) Successful in 28s
Validate Config / Python Syntax & Import Check (pull_request) Failing after 2m36s
Validate Config / Shell Script Lint (pull_request) Failing after 1m3s
Validate Config / Cron Syntax Check (pull_request) Successful in 13s
Validate Config / Deploy Script Dry Run (pull_request) Successful in 12s
PR Checklist / pr-checklist (pull_request) Failing after 6m15s
Validate Config / Playbook Schema Validation (pull_request) Successful in 28s
- Tests all configured providers
- Maintains health map in tmux-state.json
- Auto-switches profiles to working providers
- Supports --daemon and --status modes

Closes #509
2026-04-15 03:48:37 +00:00
26e39d8949 feat: add autonomous cron supervisor job (#513)
- Runs every 7 minutes
- Checks dev and timmy sessions
- Loads tmux-supervisor skill
- Telegram only on actionable events
- Silent when all agents busy
2026-04-15 03:33:43 +00:00
9 changed files with 1459 additions and 109 deletions

View File

@@ -31,6 +31,14 @@ class GlitchCategory(Enum):
WATER_REFLECTION = "water_reflection"
SKYBOX_SEAM = "skybox_seam"
# Three.js-specific categories (ref: timmy-config#543)
SHADER_FAILURE = "shader_failure"
TEXTURE_PLACEHOLDER = "texture_placeholder"
UV_MAPPING_ERROR = "uv_mapping_error"
FRUSTUM_CULLING = "frustum_culling"
SHADOW_MAP_ARTIFACT = "shadow_map_artifact"
BLOOM_OVERFLOW = "bloom_overflow"
@dataclass
class GlitchPattern:
@@ -241,6 +249,123 @@ MATRIX_GLITCH_PATTERNS: list[GlitchPattern] = [
],
confidence_threshold=0.45,
),
# --- Three.js-Specific Glitch Patterns (ref: timmy-config#543) ---
GlitchPattern(
category=GlitchCategory.SHADER_FAILURE,
name="Shader Compilation Failure",
description="Three.js shader failed to compile, rendering the material as solid black. "
"Common when custom ShaderMaterial has syntax errors or missing uniforms.",
severity=GlitchSeverity.CRITICAL,
detection_prompts=[
"Look for objects or surfaces rendered as pure black (#000000) that should have visible textures or materials.",
"Identify geometry that appears completely dark while surrounding objects are normally lit.",
"Check for objects where the material seems to 'absorb all light' — flat black with no shading gradient.",
],
visual_indicators=[
"solid black object with no shading",
"geometry rendered as silhouette",
"material appears to absorb light entirely",
"black patch inconsistent with scene lighting",
],
confidence_threshold=0.7,
),
GlitchPattern(
category=GlitchCategory.TEXTURE_PLACEHOLDER,
name="Three.js Texture Not Loaded",
description="Three.js failed to load the texture asset, rendering a 1x1 white pixel "
"stretched across the entire surface. Distinguished from missing-texture by "
"the uniform white/grey appearance rather than magenta.",
severity=GlitchSeverity.CRITICAL,
detection_prompts=[
"Look for surfaces that are uniformly white or light grey with no texture detail, even on large geometry.",
"Identify objects where the texture appears as a single solid color stretched across complex UVs.",
"Check for surfaces that look 'blank' or 'unloaded' — flat white/grey where detail should exist.",
],
visual_indicators=[
"uniform white or light grey surface",
"no texture detail on large geometry",
"stretched single-color appearance",
"1x1 pixel placeholder stretched to fill UV space",
],
confidence_threshold=0.65,
),
GlitchPattern(
category=GlitchCategory.UV_MAPPING_ERROR,
name="BufferGeometry UV Mapping Error",
description="Three.js BufferGeometry has incorrect UV coordinates, causing textures to "
"appear stretched, compressed, or mapped to the wrong faces.",
severity=GlitchSeverity.HIGH,
detection_prompts=[
"Look for textures that appear dramatically stretched in one direction on specific faces.",
"Identify surfaces where the texture pattern is distorted but other nearby surfaces look correct.",
"Check for faces where the texture seems 'smeared' or mapped with incorrect aspect ratio.",
],
visual_indicators=[
"texture stretching on specific faces",
"distorted pattern on geometry",
"smeared texture appearance",
"aspect ratio mismatch between texture and surface",
],
confidence_threshold=0.6,
),
GlitchPattern(
category=GlitchCategory.FRUSTUM_CULLING,
name="Frustum Culling Artifact",
description="Three.js frustum culling incorrectly marks objects as outside the camera "
"frustum, causing them to pop in/out of existence at screen edges.",
severity=GlitchSeverity.MEDIUM,
detection_prompts=[
"Look for objects that are partially visible at the edge of the frame — half-rendered or cut off unnaturally.",
"Identify geometry that seems to 'pop' into existence as the view angle changes.",
"Check screen edges for objects that appear suddenly rather than smoothly entering the viewport.",
],
visual_indicators=[
"half-visible object at screen edge",
"object popping into frame",
"abrupt appearance of geometry",
"bounding box visible but mesh missing",
],
confidence_threshold=0.55,
),
GlitchPattern(
category=GlitchCategory.SHADOW_MAP_ARTIFACT,
name="Shadow Map Resolution Artifact",
description="Three.js shadow map has insufficient resolution, causing pixelated, "
"blocky shadows with visible texel edges instead of smooth shadow gradients.",
severity=GlitchSeverity.MEDIUM,
detection_prompts=[
"Look for shadows with visible blocky or pixelated edges instead of smooth gradients.",
"Identify shadow maps where individual texels (texture pixels) are clearly visible.",
"Check for shadows that appear as jagged stair-stepped patterns rather than soft edges.",
],
visual_indicators=[
"blocky shadow edges",
"visible texel grid in shadows",
"stair-stepped shadow boundary",
"pixelated shadow gradient",
],
confidence_threshold=0.55,
),
GlitchPattern(
category=GlitchCategory.BLOOM_OVERFLOW,
name="Post-Processing Bloom Overflow",
description="Three.js UnrealBloomPass or similar post-processing bloom effect is too "
"intense, causing bright areas to bleed glow into surrounding geometry.",
severity=GlitchSeverity.LOW,
detection_prompts=[
"Look for bright areas that have an unusually large, soft glow bleeding into adjacent surfaces.",
"Identify scenes where light sources appear to have a 'halo' that extends beyond physical plausibility.",
"Check for bright objects whose glow color bleeds onto nearby unrelated geometry.",
],
visual_indicators=[
"excessive glow bleeding from bright surfaces",
"halo around light sources",
"bloom color tinting adjacent geometry",
"glow bleeding beyond object boundaries",
],
confidence_threshold=0.5,
),
]
@@ -289,6 +414,23 @@ def build_vision_prompt(patterns: list[GlitchPattern] | None = None) -> str:
)
# Three.js-specific category set for filtering (ref: timmy-config#543)
THREEJS_CATEGORIES = {
GlitchCategory.SHADER_FAILURE,
GlitchCategory.TEXTURE_PLACEHOLDER,
GlitchCategory.UV_MAPPING_ERROR,
GlitchCategory.FRUSTUM_CULLING,
GlitchCategory.SHADOW_MAP_ARTIFACT,
GlitchCategory.BLOOM_OVERFLOW,
}
def get_threejs_patterns() -> list[GlitchPattern]:
"""Return only Three.js-specific glitch patterns."""
return [p for p in MATRIX_GLITCH_PATTERNS if p.category in THREEJS_CATEGORIES]
if __name__ == "__main__":
import json
print(f"Loaded {len(MATRIX_GLITCH_PATTERNS)} glitch patterns:\n")

View File

@@ -9,7 +9,7 @@ Usage:
python matrix_glitch_detector.py <url> [--angles 4] [--output report.json]
python matrix_glitch_detector.py --demo # Run with synthetic test data
Ref: timmy-config#491
Ref: timmy-config#491, timmy-config#543
"""
import argparse
@@ -33,6 +33,7 @@ from glitch_patterns import (
MATRIX_GLITCH_PATTERNS,
build_vision_prompt,
get_patterns_by_severity,
get_threejs_patterns,
)
@@ -345,14 +346,17 @@ def _parse_vision_response(
def _infer_severity(category: str, confidence: float) -> str:
"""Infer severity from category and confidence when not provided."""
critical_cats = {"missing_textures", "clipping"}
high_cats = {"floating_assets", "broken_normals"}
critical_cats = {"missing_textures", "clipping", "shader_failure", "texture_placeholder"}
high_cats = {"floating_assets", "broken_normals", "uv_mapping_error"}
medium_cats = {"frustum_culling", "shadow_map_artifact"}
cat_lower = category.lower()
if any(c in cat_lower for c in critical_cats):
return "critical" if confidence > 0.7 else "high"
if any(c in cat_lower for c in high_cats):
return "high" if confidence > 0.7 else "medium"
if any(c in cat_lower for c in medium_cats):
return "medium" if confidence > 0.6 else "low"
return "medium" if confidence > 0.6 else "low"
@@ -389,9 +393,9 @@ def build_report(
),
},
metadata={
"detector_version": "0.1.0",
"detector_version": "0.2.0",
"pattern_count": len(MATRIX_GLITCH_PATTERNS),
"reference": "timmy-config#491",
"reference": "timmy-config#491, timmy-config#543",
},
)
@@ -460,6 +464,30 @@ def run_demo(output_path: Optional[Path] = None) -> ScanResult:
screenshot_index=3,
screenshot_angle="left",
),
DetectedGlitch(
id=str(uuid.uuid4())[:8],
category="shader_failure",
name="Black Material on Portal Frame",
description="Portal frame rendered as solid black — shader compilation failed (missing uniform u_time)",
severity="critical",
confidence=0.91,
location_x=45.0,
location_y=30.0,
screenshot_index=0,
screenshot_angle="front",
),
DetectedGlitch(
id=str(uuid.uuid4())[:8],
category="shadow_map_artifact",
name="Pixelated Character Shadow",
description="Character shadow shows visible texel grid — shadow map resolution too low (512x512)",
severity="medium",
confidence=0.78,
location_x=52.0,
location_y=75.0,
screenshot_index=1,
screenshot_angle="right",
),
]
print(f"[*] Detected {len(demo_glitches)} glitches")
@@ -496,6 +524,11 @@ Examples:
help="Minimum severity to include in report",
)
parser.add_argument("--verbose", "-v", action="store_true", help="Verbose output")
parser.add_argument(
"--threejs",
action="store_true",
help="Focus on Three.js-specific glitch patterns only (shader, texture, UV, culling, shadow, bloom)",
)
args = parser.parse_args()
@@ -525,9 +558,13 @@ Examples:
screenshots = capture_screenshots(args.url, angles, screenshots_dir)
print(f"[*] Captured {len(screenshots)} screenshots")
# Filter patterns by severity
# Filter patterns by severity and type
min_sev = GlitchSeverity(args.min_severity)
patterns = get_patterns_by_severity(min_sev)
if args.threejs:
threejs_patterns = get_threejs_patterns()
patterns = [p for p in patterns if p in threejs_patterns]
print(f"[*] Three.js-focused mode: {len(patterns)} patterns")
# Analyze with vision AI
print(f"[*] Analyzing with vision AI ({len(patterns)} patterns)...")

View File

@@ -0,0 +1,271 @@
#!/usr/bin/env python3
"""
Pre-Flight Provider Check Script
Issue #508: [Robustness] Credential drain detection — provider health checks
Pre-flight check before session launch: verifies provider credentials and balance.
Usage:
python3 preflight-provider-check.py # Check all providers
python3 preflight-provider-check.py --launch # Check and return exit code
python3 preflight-provider-check.py --balance # Check OpenRouter balance
"""
import os, sys, json, yaml, urllib.request
from datetime import datetime, timezone
from pathlib import Path
# Configuration
HERMES_HOME = Path(os.environ.get("HERMES_HOME", Path.home() / ".hermes"))
LOG_DIR = Path.home() / ".local" / "timmy" / "fleet-health"
LOG_FILE = LOG_DIR / "preflight-check.log"
def log(msg):
"""Log message to file and optionally console."""
timestamp = datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S")
log_entry = "[" + timestamp + "] " + msg
LOG_DIR.mkdir(parents=True, exist_ok=True)
with open(LOG_FILE, "a") as f:
f.write(log_entry + "\n")
if "--quiet" not in sys.argv:
print(log_entry)
def get_provider_api_key(provider):
"""Get API key for a provider from .env or environment."""
env_file = HERMES_HOME / ".env"
if env_file.exists():
with open(env_file) as f:
for line in f:
line = line.strip()
if line.startswith(provider.upper() + "_API_KEY="):
return line.split("=", 1)[1].strip().strip("'\"")
return os.environ.get(provider.upper() + "_API_KEY")
def check_openrouter_balance(api_key):
"""Check OpenRouter balance via /api/v1/auth/key."""
if not api_key:
return False, "No API key", 0
try:
req = urllib.request.Request(
"https://openrouter.ai/api/v1/auth/key",
headers={"Authorization": "Bearer " + api_key}
)
resp = urllib.request.urlopen(req, timeout=10)
data = json.loads(resp.read())
# Check for credits
credits = data.get("data", {}).get("limit", 0)
usage = data.get("data", {}).get("usage", 0)
remaining = credits - usage if credits else None
if remaining is not None and remaining <= 0:
return False, "No credits remaining", 0
elif remaining is not None:
return True, "Credits available", remaining
else:
return True, "Unlimited or unknown balance", None
except urllib.error.HTTPError as e:
if e.code == 401:
return False, "Invalid API key", 0
else:
return False, "HTTP " + str(e.code), 0
except Exception as e:
return False, str(e)[:100], 0
def check_nous_key(api_key):
"""Check Nous API key with minimal test call."""
if not api_key:
return False, "No API key"
try:
req = urllib.request.Request(
"https://inference.nousresearch.com/v1/models",
headers={"Authorization": "Bearer " + api_key}
)
resp = urllib.request.urlopen(req, timeout=10)
if resp.status == 200:
return True, "Valid key"
else:
return False, "HTTP " + str(resp.status)
except urllib.error.HTTPError as e:
if e.code == 401:
return False, "Invalid API key"
elif e.code == 403:
return False, "Forbidden"
else:
return False, "HTTP " + str(e.code)
except Exception as e:
return False, str(e)[:100]
def check_anthropic_key(api_key):
"""Check Anthropic API key with minimal test call."""
if not api_key:
return False, "No API key"
try:
req = urllib.request.Request(
"https://api.anthropic.com/v1/models",
headers={
"x-api-key": api_key,
"anthropic-version": "2023-06-01"
}
)
resp = urllib.request.urlopen(req, timeout=10)
if resp.status == 200:
return True, "Valid key"
else:
return False, "HTTP " + str(resp.status)
except urllib.error.HTTPError as e:
if e.code == 401:
return False, "Invalid API key"
elif e.code == 403:
return False, "Forbidden"
else:
return False, "HTTP " + str(e.code)
except Exception as e:
return False, str(e)[:100]
def check_ollama():
"""Check if Ollama is running."""
try:
req = urllib.request.Request("http://localhost:11434/api/tags")
resp = urllib.request.urlopen(req, timeout=5)
if resp.status == 200:
data = json.loads(resp.read())
models = data.get("models", [])
return True, str(len(models)) + " models loaded"
else:
return False, "HTTP " + str(resp.status)
except Exception as e:
return False, str(e)[:100]
def get_configured_provider():
"""Get the configured provider from global config."""
config_file = HERMES_HOME / "config.yaml"
if not config_file.exists():
return None
try:
with open(config_file) as f:
config = yaml.safe_load(f)
model_config = config.get("model", {})
if isinstance(model_config, dict):
return model_config.get("provider")
except:
pass
return None
def run_preflight_check():
"""Run pre-flight check on all providers."""
log("=== Pre-Flight Provider Check ===")
results = {}
# Check OpenRouter
or_key = get_provider_api_key("openrouter")
or_ok, or_msg, or_balance = check_openrouter_balance(or_key)
results["openrouter"] = {"healthy": or_ok, "message": or_msg, "balance": or_balance}
# Check Nous
nous_key = get_provider_api_key("nous")
nous_ok, nous_msg = check_nous_key(nous_key)
results["nous"] = {"healthy": nous_ok, "message": nous_msg}
# Check Anthropic
anthropic_key = get_provider_api_key("anthropic")
anthropic_ok, anthropic_msg = check_anthropic_key(anthropic_key)
results["anthropic"] = {"healthy": anthropic_ok, "message": anthropic_msg}
# Check Ollama
ollama_ok, ollama_msg = check_ollama()
results["ollama"] = {"healthy": ollama_ok, "message": ollama_msg}
# Get configured provider
configured = get_configured_provider()
# Summary
healthy_count = sum(1 for r in results.values() if r["healthy"])
total_count = len(results)
log("Results: " + str(healthy_count) + "/" + str(total_count) + " providers healthy")
for provider, result in results.items():
status = "HEALTHY" if result["healthy"] else "UNHEALTHY"
extra = ""
if provider == "openrouter" and result.get("balance") is not None:
extra = " (balance: " + str(result["balance"]) + ")"
log(" " + provider + ": " + status + " - " + result["message"] + extra)
if configured:
log("Configured provider: " + configured)
if configured in results and not results[configured]["healthy"]:
log("WARNING: Configured provider " + configured + " is UNHEALTHY!")
return results, configured
def check_launch_readiness():
"""Check if we're ready to launch sessions."""
results, configured = run_preflight_check()
# Check if configured provider is healthy
if configured and configured in results:
if not results[configured]["healthy"]:
log("LAUNCH BLOCKED: Configured provider " + configured + " is unhealthy")
return False, configured + " is unhealthy"
# Check if at least one provider is healthy
healthy_providers = [p for p, r in results.items() if r["healthy"]]
if not healthy_providers:
log("LAUNCH BLOCKED: No healthy providers available")
return False, "No healthy providers"
log("LAUNCH READY: " + str(len(healthy_providers)) + " healthy providers available")
return True, "Ready"
def show_balance():
"""Show OpenRouter balance."""
api_key = get_provider_api_key("openrouter")
if not api_key:
print("No OpenRouter API key found")
return
ok, msg, balance = check_openrouter_balance(api_key)
if ok:
if balance is not None:
print("OpenRouter balance: " + str(balance) + " credits")
else:
print("OpenRouter: " + msg)
else:
print("OpenRouter: " + msg)
def main():
if "--balance" in sys.argv:
show_balance()
elif "--launch" in sys.argv:
ready, message = check_launch_readiness()
if ready:
print("READY")
sys.exit(0)
else:
print("BLOCKED: " + message)
sys.exit(1)
else:
run_preflight_check()
if __name__ == "__main__":
main()

View File

@@ -0,0 +1,411 @@
#!/usr/bin/env python3
"""
Provider Health Monitor Script
Issue #509: [Robustness] Provider-aware profile config — auto-switch on failure
Monitors provider health and automatically switches profiles to working providers.
Usage:
python3 provider-health-monitor.py # Run once
python3 provider-health-monitor.py --daemon # Run continuously
python3 provider-health-monitor.py --status # Show provider health
"""
import os, sys, json, yaml, urllib.request, time
from datetime import datetime, timezone
from pathlib import Path
# Configuration
HERMES_HOME = Path(os.environ.get("HERMES_HOME", Path.home() / ".hermes"))
PROFILES_DIR = HERMES_HOME / "profiles"
LOG_DIR = Path.home() / ".local" / "timmy" / "fleet-health"
STATE_FILE = LOG_DIR / "tmux-state.json"
LOG_FILE = LOG_DIR / "provider-health.log"
# Provider test endpoints
PROVIDER_TESTS = {
"openrouter": {
"url": "https://openrouter.ai/api/v1/models",
"method": "GET",
"headers": lambda api_key: {"Authorization": "Bearer " + api_key},
"timeout": 10
},
"anthropic": {
"url": "https://api.anthropic.com/v1/models",
"method": "GET",
"headers": lambda api_key: {"x-api-key": api_key, "anthropic-version": "2023-06-01"},
"timeout": 10
},
"nous": {
"url": "https://inference.nousresearch.com/v1/models",
"method": "GET",
"headers": lambda api_key: {"Authorization": "Bearer " + api_key},
"timeout": 10
},
"kimi-coding": {
"url": "https://api.kimi.com/coding/v1/models",
"method": "GET",
"headers": lambda api_key: {"x-api-key": api_key, "x-api-provider": "kimi-coding"},
"timeout": 10
},
"ollama": {
"url": "http://localhost:11434/api/tags",
"method": "GET",
"headers": lambda api_key: {},
"timeout": 5
}
}
def log(msg):
"""Log message to file and optionally console."""
timestamp = datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S")
log_entry = "[" + timestamp + "] " + msg
LOG_DIR.mkdir(parents=True, exist_ok=True)
with open(LOG_FILE, "a") as f:
f.write(log_entry + "\n")
if "--quiet" not in sys.argv:
print(log_entry)
def get_provider_api_key(provider):
"""Get API key for a provider from .env or environment."""
env_file = HERMES_HOME / ".env"
if env_file.exists():
with open(env_file) as f:
for line in f:
line = line.strip()
if line.startswith(provider.upper() + "_API_KEY="):
return line.split("=", 1)[1].strip().strip("'\"")
return os.environ.get(provider.upper() + "_API_KEY")
def test_provider(provider, api_key=None):
"""Test if a provider is healthy."""
config = PROVIDER_TESTS.get(provider)
if not config:
return False, "Unknown provider: " + provider
headers = config["headers"](api_key or "")
try:
req = urllib.request.Request(
config["url"],
headers=headers,
method=config["method"]
)
resp = urllib.request.urlopen(req, timeout=config["timeout"])
if resp.status == 200:
return True, "Healthy"
else:
return False, "HTTP " + str(resp.status)
except urllib.error.HTTPError as e:
if e.code == 401:
return False, "Unauthorized (401)"
elif e.code == 403:
return False, "Forbidden (403)"
elif e.code == 429:
return True, "Rate limited but accessible"
else:
return False, "HTTP " + str(e.code)
except Exception as e:
return False, str(e)[:100]
def get_all_providers():
"""Get all providers from profiles and global config."""
providers = set()
# Global config
global_config = HERMES_HOME / "config.yaml"
if global_config.exists():
try:
with open(global_config) as f:
config = yaml.safe_load(f)
# Primary model provider
model_config = config.get("model", {})
if isinstance(model_config, dict):
provider = model_config.get("provider", "")
if provider:
providers.add(provider)
# Auxiliary providers
auxiliary = config.get("auxiliary", {})
for aux_config in auxiliary.values():
if isinstance(aux_config, dict):
provider = aux_config.get("provider", "")
if provider and provider != "auto":
providers.add(provider)
except:
pass
# Profile configs
if PROFILES_DIR.exists():
for profile_dir in PROFILES_DIR.iterdir():
if profile_dir.is_dir():
config_file = profile_dir / "config.yaml"
if config_file.exists():
try:
with open(config_file) as f:
config = yaml.safe_load(f)
model_config = config.get("model", {})
if isinstance(model_config, dict):
provider = model_config.get("provider", "")
if provider:
providers.add(provider)
auxiliary = config.get("auxiliary", {})
for aux_config in auxiliary.values():
if isinstance(aux_config, dict):
provider = aux_config.get("provider", "")
if provider and provider != "auto":
providers.add(provider)
except:
pass
# Add common providers even if not configured
providers.update(["openrouter", "nous", "ollama"])
return list(providers)
def build_health_map():
"""Build a health map of all providers."""
providers = get_all_providers()
health_map = {}
log("Testing " + str(len(providers)) + " providers...")
for provider in providers:
api_key = get_provider_api_key(provider)
healthy, message = test_provider(provider, api_key)
health_map[provider] = {
"healthy": healthy,
"message": message,
"last_test": datetime.now(timezone.utc).isoformat(),
"api_key_present": bool(api_key)
}
status = "HEALTHY" if healthy else "UNHEALTHY"
log(" " + provider + ": " + status + " - " + message)
return health_map
def get_fallback_providers(health_map):
"""Get list of healthy providers in priority order."""
# Priority order: nous, openrouter, ollama, others
priority_order = ["nous", "openrouter", "ollama", "anthropic", "kimi-coding"]
healthy = []
for provider in priority_order:
if provider in health_map and health_map[provider]["healthy"]:
healthy.append(provider)
# Add any other healthy providers not in priority list
for provider, info in health_map.items():
if info["healthy"] and provider not in healthy:
healthy.append(provider)
return healthy
def update_profile_config(profile_name, new_provider):
"""Update a profile's config to use a new provider."""
config_file = PROFILES_DIR / profile_name / "config.yaml"
if not config_file.exists():
return False, "Config file not found"
try:
with open(config_file) as f:
config = yaml.safe_load(f)
# Update model provider
if "model" not in config:
config["model"] = {}
old_provider = config["model"].get("provider", "unknown")
config["model"]["provider"] = new_provider
# Update auxiliary providers if they were using the old provider
auxiliary = config.get("auxiliary", {})
for aux_name, aux_config in auxiliary.items():
if isinstance(aux_config, dict) and aux_config.get("provider") == old_provider:
aux_config["provider"] = new_provider
# Write back
with open(config_file, "w") as f:
yaml.dump(config, f, default_flow_style=False)
log("Updated " + profile_name + ": " + old_provider + " -> " + new_provider)
return True, "Updated"
except Exception as e:
return False, str(e)
def check_profiles(health_map):
"""Check all profiles and update unhealthy providers."""
if not PROFILES_DIR.exists():
return
fallback_providers = get_fallback_providers(health_map)
if not fallback_providers:
log("CRITICAL: No healthy providers available!")
return
updated_profiles = []
for profile_dir in PROFILES_DIR.iterdir():
if not profile_dir.is_dir():
continue
profile_name = profile_dir.name
config_file = profile_dir / "config.yaml"
if not config_file.exists():
continue
try:
with open(config_file) as f:
config = yaml.safe_load(f)
model_config = config.get("model", {})
if not isinstance(model_config, dict):
continue
current_provider = model_config.get("provider", "")
if not current_provider:
continue
# Check if current provider is healthy
if current_provider in health_map and health_map[current_provider]["healthy"]:
continue # Provider is healthy, no action needed
# Find best fallback
best_fallback = None
for provider in fallback_providers:
if provider != current_provider:
best_fallback = provider
break
if not best_fallback:
log("No fallback for " + profile_name + " (current: " + current_provider + ")")
continue
# Update profile
success, message = update_profile_config(profile_name, best_fallback)
if success:
updated_profiles.append({
"profile": profile_name,
"old_provider": current_provider,
"new_provider": best_fallback
})
except Exception as e:
log("Error processing " + profile_name + ": " + str(e))
return updated_profiles
def load_state():
"""Load state from tmux-state.json."""
if STATE_FILE.exists():
try:
with open(STATE_FILE) as f:
return json.load(f)
except:
pass
return {}
def save_state(state):
"""Save state to tmux-state.json."""
LOG_DIR.mkdir(parents=True, exist_ok=True)
with open(STATE_FILE, "w") as f:
json.dump(state, f, indent=2)
def run_once():
"""Run provider health check once."""
log("=== Provider Health Check ===")
state = load_state()
# Build health map
health_map = build_health_map()
# Check profiles and update if needed
updated_profiles = check_profiles(health_map)
# Update state
state["provider_health"] = health_map
state["last_provider_check"] = datetime.now(timezone.utc).isoformat()
if updated_profiles:
state["last_profile_updates"] = updated_profiles
save_state(state)
# Summary
healthy_count = sum(1 for p in health_map.values() if p["healthy"])
total_count = len(health_map)
log("Health: " + str(healthy_count) + "/" + str(total_count) + " providers healthy")
if updated_profiles:
log("Updated " + str(len(updated_profiles)) + " profiles:")
for update in updated_profiles:
log(" " + update["profile"] + ": " + update["old_provider"] + " -> " + update["new_provider"])
def show_status():
"""Show provider health status."""
state = load_state()
health_map = state.get("provider_health", {})
if not health_map:
print("No provider health data available. Run without --status first.")
return
print("Provider Health (last updated: " + str(state.get("last_provider_check", "unknown")) + ")")
print("=" * 80)
for provider, info in sorted(health_map.items()):
status = "HEALTHY" if info["healthy"] else "UNHEALTHY"
message = info.get("message", "")
api_key = "yes" if info.get("api_key_present") else "no"
print(provider.ljust(20) + " " + status.ljust(10) + " API key: " + api_key + " - " + message)
# Show recent updates
updates = state.get("last_profile_updates", [])
if updates:
print()
print("Recent Profile Updates:")
for update in updates:
print(" " + update["profile"] + ": " + update["old_provider"] + " -> " + update["new_provider"])
def daemon_mode():
"""Run continuously."""
log("Starting provider health daemon (check every 300s)")
while True:
try:
run_once()
time.sleep(300) # Check every 5 minutes
except KeyboardInterrupt:
log("Daemon stopped by user")
break
except Exception as e:
log("Error: " + str(e))
time.sleep(60)
def main():
if "--status" in sys.argv:
show_status()
elif "--daemon" in sys.argv:
daemon_mode()
else:
run_once()
if __name__ == "__main__":
main()

View File

@@ -196,7 +196,37 @@
"paused_reason": null,
"skills": [],
"skill": null
},
{
"id": "tmux-supervisor-513",
"name": "Autonomous Cron Supervisor",
"prompt": "Load the tmux-supervisor skill and execute the monitoring protocol.\n\nCheck both `dev` and `timmy` tmux sessions for idle panes. Only send Telegram notifications on actionable events (idle, overflow, failure). Be silent when all agents are working.\n\nSteps:\n1. List all tmux sessions (skip 'Alexander')\n2. For each session, list windows and panes\n3. Capture each pane and classify state (idle vs active)\n4. For idle panes: read context, craft context-aware prompt\n5. Send /queue prompts to idle panes\n6. Verify prompts landed\n7. Only notify via Telegram if:\n - A pane was prompted (idle detected)\n - A pane shows context overflow (>80%)\n - A pane is stuck or crashed\n8. If all panes are active: respond with [SILENT]",
"schedule": {
"kind": "interval",
"minutes": 7,
"display": "every 7m"
},
"schedule_display": "every 7m",
"repeat": {
"times": null,
"completed": 0
},
"enabled": true,
"created_at": "2026-04-15T03:00:00.000000+00:00",
"next_run_at": null,
"last_run_at": null,
"last_status": null,
"last_error": null,
"deliver": "telegram",
"origin": null,
"state": "scheduled",
"paused_at": null,
"paused_reason": null,
"skills": [
"tmux-supervisor"
],
"skill": "tmux-supervisor"
}
],
"updated_at": "2026-04-13T02:00:00+00:00"
}
}

331
pipeline/nightly_scheduler.py Executable file
View File

@@ -0,0 +1,331 @@
#!/usr/bin/env python3
"""
nightly_scheduler.py — Nightly Pipeline Scheduler
Auto-starts batch pipelines when inference is available, respecting
priority ordering, token budgets, and peak-hour pausing.
Usage:
python3 nightly_scheduler.py # run scheduler
python3 nightly_scheduler.py --check # dry-run: show what would start
python3 nightly_scheduler.py --status # show pipeline status
python3 nightly_scheduler.py --reset # reset daily budget
Crontab:
# Run every 30 minutes during off-peak hours (10pm-6am)
*/30 22-5 * * * cd /path/to/timmy-config && python3 pipeline/nightly_scheduler.py >> ~/.hermes/pipeline-logs/nightly.log 2>&1
"""
import json
import os
import sys
import time
import urllib.request
import urllib.error
from datetime import datetime, timezone, timedelta
from pathlib import Path
# --- Config ---
STATE_FILE = Path.home() / ".hermes" / "pipeline_state.json"
LOG_DIR = Path.home() / ".hermes" / "pipeline-logs"
DAILY_TOKEN_BUDGET = 5_000_000 # 5M tokens per day
PEAK_HOURS = list(range(8, 22)) # 8am-10pm = peak interactive usage
CHECK_INTERVAL = 1800 # 30 minutes
INFERENCE_ENDPOINTS = [
{"name": "local_ollama", "url": "http://localhost:11434/v1/models", "type": "local"},
{"name": "runpod", "url": "https://8lfr3j47a5r3gn-11434.proxy.runpod.net/v1/models", "type": "gpu"},
{"name": "openrouter", "url": "https://openrouter.ai/api/v1/models", "type": "cloud"},
]
# Pipeline priority order (highest first)
PIPELINE_PRIORITY = [
{"name": "playground_factory", "script": "pipeline/playground_factory.py", "priority": 1},
{"name": "training_factory", "script": "pipeline/training_factory.py", "priority": 2},
{"name": "knowledge_mine", "script": "pipeline/knowledge_mine.py", "priority": 3},
{"name": "adversary", "script": "pipeline/adversary_runner.py", "priority": 4},
{"name": "codebase_genome", "script": "pipeline/codebase_genome.py", "priority": 5},
]
# Dependency rules: some pipelines only start after others are running
DEPENDENCY_RULES = {
"playground_factory": [], # no deps, start immediately
"training_factory": [], # no deps, start in parallel
"knowledge_mine": ["training_factory"], # start after training is running
"adversary": ["knowledge_mine"], # start after knowledge is halfway
"codebase_genome": [], # continuous, one repo per night
}
def load_state():
"""Load pipeline state from disk."""
if STATE_FILE.exists():
with open(STATE_FILE) as f:
return json.load(f)
return {
"last_run": None,
"daily_tokens_used": 0,
"budget_reset_date": None,
"pipelines": {},
"active_sessions": [],
}
def save_state(state):
"""Save pipeline state to disk."""
STATE_FILE.parent.mkdir(parents=True, exist_ok=True)
with open(STATE_FILE, "w") as f:
json.dump(state, f, indent=2)
def check_provider(endpoint):
"""Check if an inference provider is available."""
try:
req = urllib.request.Request(endpoint["url"], headers={"Authorization": "Bearer ollama"})
with urllib.request.urlopen(req, timeout=10) as resp:
return resp.status == 200
except Exception:
return False
def get_available_providers():
"""Check all inference endpoints and return available ones."""
available = []
for ep in INFERENCE_ENDPOINTS:
if check_provider(ep):
available.append(ep["name"])
return available
def is_peak_hours():
"""Check if current time is during peak interactive usage."""
now = datetime.now()
return now.hour in PEAK_HOURS
def check_token_budget(state):
"""Check if daily token budget allows starting new work."""
today = datetime.now().strftime("%Y-%m-%d")
if state.get("budget_reset_date") != today:
# New day, reset budget
state["daily_tokens_used"] = 0
state["budget_reset_date"] = today
save_state(state)
return state["daily_tokens_used"] < DAILY_TOKEN_BUDGET
def get_pipeline_status(state, pipeline_name):
"""Get the status of a specific pipeline."""
return state.get("pipelines", {}).get(pipeline_name, {
"status": "not_started",
"last_run": None,
"last_success": None,
"progress": 0,
})
def check_dependencies(state, pipeline_name):
"""Check if pipeline dependencies are satisfied."""
deps = DEPENDENCY_RULES.get(pipeline_name, [])
for dep in deps:
dep_status = get_pipeline_status(state, dep)
if dep_status["status"] not in ("running", "completed"):
return False
return True
def start_pipeline(pipeline, state, dry_run=False):
"""Start a pipeline process."""
name = pipeline["name"]
script = pipeline["script"]
log(f"Starting pipeline: {name}")
if dry_run:
log(f" DRY RUN — would run: python3 {script}")
return True
# Check if script exists
script_path = Path(script)
if not script_path.exists():
log(f" Script not found: {script_path}")
# Update state anyway so we track the attempt
state["pipelines"][name] = {
"status": "script_missing",
"last_run": datetime.now(timezone.utc).isoformat(),
"progress": 0,
}
save_state(state)
return False
# Run the pipeline script
import subprocess
log_dir = LOG_DIR / name
log_dir.mkdir(parents=True, exist_ok=True)
log_file = log_dir / f"{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
try:
proc = subprocess.Popen(
["python3", str(script_path)],
stdout=open(log_file, "w"),
stderr=subprocess.STDOUT,
cwd=str(Path(script).parent.parent),
)
state["pipelines"][name] = {
"status": "running",
"pid": proc.pid,
"last_run": datetime.now(timezone.utc).isoformat(),
"log_file": str(log_file),
"progress": 0,
}
save_state(state)
log(f" Started PID {proc.pid}, log: {log_file}")
return True
except Exception as e:
log(f" Failed to start: {e}")
state["pipelines"][name] = {
"status": "failed",
"last_run": datetime.now(timezone.utc).isoformat(),
"error": str(e),
}
save_state(state)
return False
def check_running_pipelines(state):
"""Check status of running pipelines and update state."""
import subprocess
for name, info in state.get("pipelines", {}).items():
if info.get("status") == "running":
pid = info.get("pid")
if pid:
try:
os.kill(pid, 0) # Check if process exists
except ProcessLookupError:
# Process finished
info["status"] = "completed"
info["completed_at"] = datetime.now(timezone.utc).isoformat()
log(f"Pipeline {name} completed (PID {pid} exited)")
save_state(state)
def run_scheduler(dry_run=False, check_only=False):
"""Main scheduler loop."""
state = load_state()
log("=" * 50)
log(f"Pipeline Scheduler — {datetime.now().isoformat()}")
log(f"Mode: {'CHECK' if check_only else 'DRY RUN' if dry_run else 'LIVE'}")
# Check peak hours
if is_peak_hours():
log("Peak hours detected. Pausing pipeline starts.")
log("Pipelines will resume at 10pm.")
return
# Check token budget
if not check_token_budget(state):
log(f"Daily token budget exhausted ({state['daily_tokens_used']}/{DAILY_TOKEN_BUDGET})")
return
log(f"Token budget: {state['daily_tokens_used']}/{DAILY_TOKEN_BUDGET}")
# Check providers
providers = get_available_providers()
if not providers:
log("No inference providers available. Skipping.")
return
log(f"Available providers: {', '.join(providers)}")
# Check running pipelines
check_running_pipelines(state)
# Find next pipeline to start
started = 0
for pipeline in sorted(PIPELINE_PRIORITY, key=lambda p: p["priority"]):
name = pipeline["name"]
status = get_pipeline_status(state, name)
# Skip if already running or completed
if status["status"] in ("running", "completed"):
log(f" {name}: {status['status']} (skipping)")
continue
# Check dependencies
if not check_dependencies(state, name):
deps = DEPENDENCY_RULES.get(name, [])
log(f" {name}: waiting for dependencies: {deps}")
continue
# Start the pipeline
if check_only:
log(f" {name}: READY to start (priority {pipeline['priority']})")
else:
if start_pipeline(pipeline, state, dry_run):
started += 1
# Only start one pipeline per run to avoid overload
if started >= 1:
log("Started 1 pipeline. Will check again next cycle.")
break
if started == 0 and not check_only:
log("No pipelines to start. All are running, completed, or blocked.")
log("=" * 50)
def show_status():
"""Show current pipeline status."""
state = load_state()
print(f"\nPipeline Status — {datetime.now().strftime('%Y-%m-%d %H:%M')}")
print(f"Token budget: {state.get('daily_tokens_used', 0)}/{DAILY_TOKEN_BUDGET}")
print(f"Last run: {state.get('last_run', 'never')}")
print()
for pipeline in sorted(PIPELINE_PRIORITY, key=lambda p: p["priority"]):
name = pipeline["name"]
status = get_pipeline_status(state, name)
st = status["status"]
icon = {"running": "", "completed": "", "failed": "", "not_started": "", "script_missing": "?"}.get(st, "?")
print(f" {icon} {name:25} {st:15} last={(status.get('last_run') or 'never')[:19]}")
def reset_budget():
"""Reset daily token budget."""
state = load_state()
state["daily_tokens_used"] = 0
state["budget_reset_date"] = datetime.now().strftime("%Y-%m-%d")
save_state(state)
print("Budget reset.")
def log(msg):
"""Log to stdout and file."""
timestamp = datetime.now().strftime("%H:%M:%S")
line = f"[{timestamp}] {msg}"
print(line)
LOG_DIR.mkdir(parents=True, exist_ok=True)
log_file = LOG_DIR / "nightly.log"
with open(log_file, "a") as f:
f.write(line + "\n")
def main():
import argparse
parser = argparse.ArgumentParser(description="Nightly Pipeline Scheduler")
parser.add_argument("--check", action="store_true", help="Dry-run: show what would start")
parser.add_argument("--status", action="store_true", help="Show pipeline status")
parser.add_argument("--reset", action="store_true", help="Reset daily token budget")
parser.add_argument("--dry-run", action="store_true", help="Dry-run mode")
args = parser.parse_args()
if args.status:
show_status()
elif args.reset:
reset_budget()
else:
run_scheduler(dry_run=args.dry_run or args.check, check_only=args.check)
if __name__ == "__main__":
main()

View File

@@ -19,9 +19,11 @@ from glitch_patterns import (
GlitchPattern,
GlitchSeverity,
MATRIX_GLITCH_PATTERNS,
THREEJS_CATEGORIES,
build_vision_prompt,
get_pattern_by_category,
get_patterns_by_severity,
get_threejs_patterns,
)
from matrix_glitch_detector import (
@@ -40,7 +42,7 @@ class TestGlitchPatterns(unittest.TestCase):
def test_pattern_count(self):
"""Verify we have a reasonable number of defined patterns."""
self.assertGreaterEqual(len(MATRIX_GLITCH_PATTERNS), 8)
self.assertGreaterEqual(len(MATRIX_GLITCH_PATTERNS), 14) # 10 generic + 6 Three.js
def test_all_patterns_have_required_fields(self):
"""Every pattern must have category, name, description, severity, prompts."""
@@ -88,6 +90,9 @@ class TestGlitchPatterns(unittest.TestCase):
self.assertIn("Floating Object", prompt)
self.assertIn("Z-Fighting", prompt)
self.assertIn("Missing", prompt)
# Three.js patterns should be included
self.assertIn("Shader Compilation Failure", prompt)
self.assertIn("Bloom Overflow", prompt)
def test_build_vision_prompt_subset(self):
"""Vision prompt with subset should only include specified patterns."""
@@ -248,7 +253,7 @@ class TestGlitchDetector(unittest.TestCase):
try:
report = run_demo(output_path)
self.assertEqual(len(report.glitches), 4)
self.assertEqual(len(report.glitches), 6) # 4 original + 2 Three.js
self.assertGreater(report.summary["total_glitches"], 0)
self.assertTrue(output_path.exists())
@@ -260,6 +265,93 @@ class TestGlitchDetector(unittest.TestCase):
output_path.unlink(missing_ok=True)
class TestThreeJsPatterns(unittest.TestCase):
"""Tests for Three.js-specific glitch patterns (timmy-config#543)."""
def test_get_threejs_patterns_returns_only_threejs(self):
"""get_threejs_patterns() should return only Three.js categories."""
patterns = get_threejs_patterns()
self.assertEqual(len(patterns), 6)
for p in patterns:
self.assertIn(p.category, THREEJS_CATEGORIES)
def test_threejs_patterns_have_required_fields(self):
"""All Three.js patterns must have valid fields."""
for p in get_threejs_patterns():
self.assertIsInstance(p.category, GlitchCategory)
self.assertTrue(p.name)
self.assertTrue(p.description)
self.assertIsInstance(p.severity, GlitchSeverity)
self.assertGreater(len(p.detection_prompts), 0)
self.assertGreater(len(p.visual_indicators), 0)
def test_shader_failure_is_critical(self):
"""Shader compilation failure should be CRITICAL severity."""
p = get_pattern_by_category(GlitchCategory.SHADER_FAILURE)
self.assertIsNotNone(p)
self.assertEqual(p.severity, GlitchSeverity.CRITICAL)
def test_texture_placeholder_is_critical(self):
"""Texture placeholder (1x1 white) should be CRITICAL severity."""
p = get_pattern_by_category(GlitchCategory.TEXTURE_PLACEHOLDER)
self.assertIsNotNone(p)
self.assertEqual(p.severity, GlitchSeverity.CRITICAL)
def test_infer_severity_shader_failure(self):
"""Shader failure should infer critical/high."""
self.assertEqual(_infer_severity("shader_failure", 0.8), "critical")
self.assertEqual(_infer_severity("shader_failure", 0.5), "high")
def test_infer_severity_texture_placeholder(self):
"""Texture placeholder should infer critical/high."""
self.assertEqual(_infer_severity("texture_placeholder", 0.8), "critical")
self.assertEqual(_infer_severity("texture_placeholder", 0.5), "high")
def test_infer_severity_uv_mapping(self):
"""UV mapping error should infer high/medium."""
self.assertEqual(_infer_severity("uv_mapping_error", 0.8), "high")
self.assertEqual(_infer_severity("uv_mapping_error", 0.5), "medium")
def test_infer_severity_frustum_culling(self):
"""Frustum culling should infer medium/low."""
self.assertEqual(_infer_severity("frustum_culling", 0.7), "medium")
self.assertEqual(_infer_severity("frustum_culling", 0.4), "low")
def test_infer_severity_shadow_map(self):
"""Shadow map artifact should infer medium/low."""
self.assertEqual(_infer_severity("shadow_map_artifact", 0.7), "medium")
self.assertEqual(_infer_severity("shadow_map_artifact", 0.4), "low")
def test_infer_severity_bloom_overflow(self):
"""Bloom overflow should infer medium/low (default path)."""
self.assertEqual(_infer_severity("bloom_overflow", 0.7), "medium")
self.assertEqual(_infer_severity("bloom_overflow", 0.4), "low")
def test_threejs_patterns_in_vision_prompt(self):
"""Three.js patterns should appear in the composite vision prompt."""
prompt = build_vision_prompt()
self.assertIn("shader_failure", prompt)
self.assertIn("texture_placeholder", prompt)
self.assertIn("uv_mapping_error", prompt)
self.assertIn("frustum_culling", prompt)
self.assertIn("shadow_map_artifact", prompt)
self.assertIn("bloom_overflow", prompt)
def test_threejs_subset_prompt(self):
"""Building prompt from Three.js-only patterns should work."""
threejs = get_threejs_patterns()
prompt = build_vision_prompt(threejs)
self.assertIn("Shader Compilation Failure", prompt)
self.assertNotIn("Floating Object", prompt) # generic, not Three.js
def test_report_metadata_version(self):
"""Report metadata should reference both issues."""
report = run_demo()
self.assertEqual(report.metadata["detector_version"], "0.2.0")
self.assertIn("543", report.metadata["reference"])
class TestIntegration(unittest.TestCase):
"""Integration-level tests."""
@@ -276,6 +368,13 @@ class TestIntegration(unittest.TestCase):
expected = {"floating_assets", "z_fighting", "missing_textures", "clipping", "broken_normals"}
self.assertTrue(expected.issubset(category_values))
def test_patterns_cover_threejs_themes(self):
"""Patterns should cover Three.js-specific glitch themes (#543)."""
category_values = {p.category.value for p in MATRIX_GLITCH_PATTERNS}
threejs_expected = {"shader_failure", "texture_placeholder", "uv_mapping_error",
"frustum_culling", "shadow_map_artifact", "bloom_overflow"}
self.assertTrue(threejs_expected.issubset(category_values))
if __name__ == "__main__":
unittest.main()

View File

@@ -1,100 +0,0 @@
{"song": "Dirt Road Home", "beat": 1, "lyric_line": "Gravel crunches under the tires slow", "scene": {"mood": "warmth", "colors": ["navy", "silver", "white"], "composition": "close-up", "camera": "slow pan", "description": "A warmth scene: 'Gravel crunches under the tires slow'. close-up with navy, silver, white. Camera: slow pan."}}
{"song": "Dirt Road Home", "beat": 2, "lyric_line": "The oak tree still leans toward the road", "scene": {"mood": "memory", "colors": ["dusty rose", "gold", "brown"], "composition": "wide shot", "camera": "steady", "description": "A memory scene: 'The oak tree still leans toward the road'. wide shot with dusty rose, gold, brown. Camera: steady."}}
{"song": "Dirt Road Home", "beat": 3, "lyric_line": "Mama's porch light through the fog", "scene": {"mood": "bittersweet", "colors": ["orange", "red", "brown"], "composition": "medium shot", "camera": "handheld", "description": "A bittersweet scene: 'Mama's porch light through the fog'. medium shot with orange, red, brown. Camera: handheld."}}
{"song": "Dirt Road Home", "beat": 4, "lyric_line": "Daddy's truck rusting by the barn", "scene": {"mood": "tender", "colors": ["navy", "silver", "white"], "composition": "low angle", "camera": "slow zoom", "description": "A tender scene: 'Daddy's truck rusting by the barn'. low angle with navy, silver, white. Camera: slow zoom."}}
{"song": "Dirt Road Home", "beat": 5, "lyric_line": "The creek remembers my bare feet", "scene": {"mood": "fading", "colors": ["amber", "brown", "cream"], "composition": "high angle", "camera": "tracking", "description": "A fading scene: 'The creek remembers my bare feet'. high angle with amber, brown, cream. Camera: tracking."}}
{"song": "Dirt Road Home", "beat": 6, "lyric_line": "Fireflies spell out summer's name", "scene": {"mood": "warmth", "colors": ["green", "olive", "tan"], "composition": "over-the-shoulder", "camera": "static", "description": "A warmth scene: 'Fireflies spell out summer's name'. over-the-shoulder with green, olive, tan. Camera: static."}}
{"song": "Dirt Road Home", "beat": 7, "lyric_line": "The church bell rings for no one new", "scene": {"mood": "memory", "colors": ["dusty rose", "gold", "brown"], "composition": "profile", "camera": "crane up", "description": "A memory scene: 'The church bell rings for no one new'. profile with dusty rose, gold, brown. Camera: crane up."}}
{"song": "Dirt Road Home", "beat": 8, "lyric_line": "Honeysuckle climbs the fence again", "scene": {"mood": "bittersweet", "colors": ["navy", "silver", "white"], "composition": "bird's eye", "camera": "dolly in", "description": "A bittersweet scene: 'Honeysuckle climbs the fence again'. bird's eye with navy, silver, white. Camera: dolly in."}}
{"song": "Dirt Road Home", "beat": 9, "lyric_line": "The field where we learned to drive", "scene": {"mood": "tender", "colors": ["warm yellow", "barn red", "cream"], "composition": "tracking shot", "camera": "gentle drift", "description": "A tender scene: 'The field where we learned to drive'. tracking shot with warm yellow, barn red, cream. Camera: gentle drift."}}
{"song": "Dirt Road Home", "beat": 10, "lyric_line": "Some roads only go one way home", "scene": {"mood": "fading", "colors": ["forest green", "brown", "gold"], "composition": "establishing", "camera": "locked-off", "description": "A fading scene: 'Some roads only go one way home'. establishing with forest green, brown, gold. Camera: locked-off."}}
{"song": "Last Call Honky-Tonk", "beat": 1, "lyric_line": "Neon beer sign flickers twice", "scene": {"mood": "regret", "colors": ["warm yellow", "barn red", "cream"], "composition": "close-up", "camera": "slow pan", "description": "A regret scene: 'Neon beer sign flickers twice'. close-up with warm yellow, barn red, cream. Camera: slow pan."}}
{"song": "Last Call Honky-Tonk", "beat": 2, "lyric_line": "The jukebox plays what we used to be", "scene": {"mood": "fondness", "colors": ["copper", "dust", "gold"], "composition": "wide shot", "camera": "steady", "description": "A fondness scene: 'The jukebox plays what we used to be'. wide shot with copper, dust, gold. Camera: steady."}}
{"song": "Last Call Honky-Tonk", "beat": 3, "lyric_line": "Boots scuff the dance floor marks", "scene": {"mood": "aging", "colors": ["sunset orange", "purple", "pink"], "composition": "medium shot", "camera": "handheld", "description": "A aging scene: 'Boots scuff the dance floor marks'. medium shot with sunset orange, purple, pink. Camera: handheld."}}
{"song": "Last Call Honky-Tonk", "beat": 4, "lyric_line": "She orders water for the drive", "scene": {"mood": "memory", "colors": ["grey", "mud brown", "green"], "composition": "low angle", "camera": "slow zoom", "description": "A memory scene: 'She orders water for the drive'. low angle with grey, mud brown, green. Camera: slow zoom."}}
{"song": "Last Call Honky-Tonk", "beat": 5, "lyric_line": "The bartender knows both our names", "scene": {"mood": "music", "colors": ["orange", "red", "brown"], "composition": "high angle", "camera": "tracking", "description": "A music scene: 'The bartender knows both our names'. high angle with orange, red, brown. Camera: tracking."}}
{"song": "Last Call Honky-Tonk", "beat": 6, "lyric_line": "Two-step where the floor dips low", "scene": {"mood": "regret", "colors": ["navy", "silver", "white"], "composition": "over-the-shoulder", "camera": "static", "description": "A regret scene: 'Two-step where the floor dips low'. over-the-shoulder with navy, silver, white. Camera: static."}}
{"song": "Last Call Honky-Tonk", "beat": 7, "lyric_line": "Closing time writes the last song", "scene": {"mood": "fondness", "colors": ["sky blue", "wheat", "brown"], "composition": "profile", "camera": "crane up", "description": "A fondness scene: 'Closing time writes the last song'. profile with sky blue, wheat, brown. Camera: crane up."}}
{"song": "Last Call Honky-Tonk", "beat": 8, "lyric_line": "We leave our shadows at the bar", "scene": {"mood": "aging", "colors": ["green", "olive", "tan"], "composition": "bird's eye", "camera": "dolly in", "description": "A aging scene: 'We leave our shadows at the bar'. bird's eye with green, olive, tan. Camera: dolly in."}}
{"song": "Last Call Honky-Tonk", "beat": 9, "lyric_line": "The parking lot smells like rain", "scene": {"mood": "memory", "colors": ["dusty rose", "gold", "brown"], "composition": "tracking shot", "camera": "gentle drift", "description": "A memory scene: 'The parking lot smells like rain'. tracking shot with dusty rose, gold, brown. Camera: gentle drift."}}
{"song": "Last Call Honky-Tonk", "beat": 10, "lyric_line": "Some nights end before the music", "scene": {"mood": "music", "colors": ["navy", "silver", "white"], "composition": "establishing", "camera": "locked-off", "description": "A music scene: 'Some nights end before the music'. establishing with navy, silver, white. Camera: locked-off."}}
{"song": "Church on Sunday", "beat": 1, "lyric_line": "White steeple catches the morning sun", "scene": {"mood": "sacred", "colors": ["forest green", "brown", "gold"], "composition": "close-up", "camera": "slow pan", "description": "A sacred scene: 'White steeple catches the morning sun'. close-up with forest green, brown, gold. Camera: slow pan."}}
{"song": "Church on Sunday", "beat": 2, "lyric_line": "Hymnal pages turn like falling leaves", "scene": {"mood": "quiet", "colors": ["grey", "mud brown", "green"], "composition": "wide shot", "camera": "steady", "description": "A quiet scene: 'Hymnal pages turn like falling leaves'. wide shot with grey, mud brown, green. Camera: steady."}}
{"song": "Church on Sunday", "beat": 3, "lyric_line": "The organ hums beneath the prayers", "scene": {"mood": "devotion", "colors": ["orange", "red", "brown"], "composition": "medium shot", "camera": "handheld", "description": "A devotion scene: 'The organ hums beneath the prayers'. medium shot with orange, red, brown. Camera: handheld."}}
{"song": "Church on Sunday", "beat": 4, "lyric_line": "Grandma's hat blocks the stained glass light", "scene": {"mood": "stillness", "colors": ["warm yellow", "barn red", "cream"], "composition": "low angle", "camera": "slow zoom", "description": "A stillness scene: 'Grandma's hat blocks the stained glass light'. low angle with warm yellow, barn red, cream. Camera: slow zoom."}}
{"song": "Church on Sunday", "beat": 5, "lyric_line": "We kneel on cushions worn by years", "scene": {"mood": "awe", "colors": ["navy", "silver", "white"], "composition": "high angle", "camera": "tracking", "description": "A awe scene: 'We kneel on cushions worn by years'. high angle with navy, silver, white. Camera: tracking."}}
{"song": "Church on Sunday", "beat": 6, "lyric_line": "The preacher's voice fills the gaps", "scene": {"mood": "sacred", "colors": ["sky blue", "wheat", "brown"], "composition": "over-the-shoulder", "camera": "static", "description": "A sacred scene: 'The preacher's voice fills the gaps'. over-the-shoulder with sky blue, wheat, brown. Camera: static."}}
{"song": "Church on Sunday", "beat": 7, "lyric_line": "Offering plate circles like a prayer", "scene": {"mood": "quiet", "colors": ["white", "blue", "brown"], "composition": "profile", "camera": "crane up", "description": "A quiet scene: 'Offering plate circles like a prayer'. profile with white, blue, brown. Camera: crane up."}}
{"song": "Church on Sunday", "beat": 8, "lyric_line": "Amen echoes off the wooden beams", "scene": {"mood": "devotion", "colors": ["dusty rose", "gold", "brown"], "composition": "bird's eye", "camera": "dolly in", "description": "A devotion scene: 'Amen echoes off the wooden beams'. bird's eye with dusty rose, gold, brown. Camera: dolly in."}}
{"song": "Church on Sunday", "beat": 9, "lyric_line": "Children wiggle through the sermon", "scene": {"mood": "stillness", "colors": ["white", "blue", "brown"], "composition": "tracking shot", "camera": "gentle drift", "description": "A stillness scene: 'Children wiggle through the sermon'. tracking shot with white, blue, brown. Camera: gentle drift."}}
{"song": "Church on Sunday", "beat": 10, "lyric_line": "Faith smells like old wood and coffee", "scene": {"mood": "awe", "colors": ["sunset orange", "purple", "pink"], "composition": "establishing", "camera": "locked-off", "description": "A awe scene: 'Faith smells like old wood and coffee'. establishing with sunset orange, purple, pink. Camera: locked-off."}}
{"song": "Tractor at Dawn", "beat": 1, "lyric_line": "Headlights cut the fog at five", "scene": {"mood": "resolve", "colors": ["dusty rose", "gold", "brown"], "composition": "close-up", "camera": "slow pan", "description": "A resolve scene: 'Headlights cut the fog at five'. close-up with dusty rose, gold, brown. Camera: slow pan."}}
{"song": "Tractor at Dawn", "beat": 2, "lyric_line": "The diesel coughs to life like faith", "scene": {"mood": "effort", "colors": ["dusty rose", "gold", "brown"], "composition": "wide shot", "camera": "steady", "description": "A effort scene: 'The diesel coughs to life like faith'. wide shot with dusty rose, gold, brown. Camera: steady."}}
{"song": "Tractor at Dawn", "beat": 3, "lyric_line": "Rows stretch past the tree line", "scene": {"mood": "strength", "colors": ["dusty rose", "gold", "brown"], "composition": "medium shot", "camera": "handheld", "description": "A strength scene: 'Rows stretch past the tree line'. medium shot with dusty rose, gold, brown. Camera: handheld."}}
{"song": "Tractor at Dawn", "beat": 4, "lyric_line": "Dust follows the disc like a ghost", "scene": {"mood": "persistence", "colors": ["warm yellow", "barn red", "cream"], "composition": "low angle", "camera": "slow zoom", "description": "A persistence scene: 'Dust follows the disc like a ghost'. low angle with warm yellow, barn red, cream. Camera: slow zoom."}}
{"song": "Tractor at Dawn", "beat": 5, "lyric_line": "Hands grip leather worn to skin", "scene": {"mood": "focus", "colors": ["amber", "brown", "cream"], "composition": "high angle", "camera": "tracking", "description": "A focus scene: 'Hands grip leather worn to skin'. high angle with amber, brown, cream. Camera: tracking."}}
{"song": "Tractor at Dawn", "beat": 6, "lyric_line": "The sun rises behind the work", "scene": {"mood": "resolve", "colors": ["forest green", "brown", "gold"], "composition": "over-the-shoulder", "camera": "static", "description": "A resolve scene: 'The sun rises behind the work'. over-the-shoulder with forest green, brown, gold. Camera: static."}}
{"song": "Tractor at Dawn", "beat": 7, "lyric_line": "Crows circle what the plow reveals", "scene": {"mood": "effort", "colors": ["amber", "brown", "cream"], "composition": "profile", "camera": "crane up", "description": "A effort scene: 'Crows circle what the plow reveals'. profile with amber, brown, cream. Camera: crane up."}}
{"song": "Tractor at Dawn", "beat": 8, "lyric_line": "Sweat salts the steering wheel", "scene": {"mood": "strength", "colors": ["grey", "mud brown", "green"], "composition": "bird's eye", "camera": "dolly in", "description": "A strength scene: 'Sweat salts the steering wheel'. bird's eye with grey, mud brown, green. Camera: dolly in."}}
{"song": "Tractor at Dawn", "beat": 9, "lyric_line": "Planting season has no snooze", "scene": {"mood": "persistence", "colors": ["green", "olive", "tan"], "composition": "tracking shot", "camera": "gentle drift", "description": "A persistence scene: 'Planting season has no snooze'. tracking shot with green, olive, tan. Camera: gentle drift."}}
{"song": "Tractor at Dawn", "beat": 10, "lyric_line": "Every seed is a prayer for rain", "scene": {"mood": "focus", "colors": ["dusty rose", "gold", "brown"], "composition": "establishing", "camera": "locked-off", "description": "A focus scene: 'Every seed is a prayer for rain'. establishing with dusty rose, gold, brown. Camera: locked-off."}}
{"song": "Letters from Boot Camp", "beat": 1, "lyric_line": "Her handwriting shakes on the envelope", "scene": {"mood": "love", "colors": ["sky blue", "wheat", "brown"], "composition": "close-up", "camera": "slow pan", "description": "A love scene: 'Her handwriting shakes on the envelope'. close-up with sky blue, wheat, brown. Camera: slow pan."}}
{"song": "Letters from Boot Camp", "beat": 2, "lyric_line": "The postmark is two weeks old", "scene": {"mood": "vulnerability", "colors": ["green", "olive", "tan"], "composition": "wide shot", "camera": "steady", "description": "A vulnerability scene: 'The postmark is two weeks old'. wide shot with green, olive, tan. Camera: steady."}}
{"song": "Letters from Boot Camp", "beat": 3, "lyric_line": "He reads it sitting on his bunk", "scene": {"mood": "distance", "colors": ["sky blue", "wheat", "brown"], "composition": "medium shot", "camera": "handheld", "description": "A distance scene: 'He reads it sitting on his bunk'. medium shot with sky blue, wheat, brown. Camera: handheld."}}
{"song": "Letters from Boot Camp", "beat": 4, "lyric_line": "She describes the dog getting bigger", "scene": {"mood": "hope", "colors": ["white", "blue", "brown"], "composition": "low angle", "camera": "slow zoom", "description": "A hope scene: 'She describes the dog getting bigger'. low angle with white, blue, brown. Camera: slow zoom."}}
{"song": "Letters from Boot Camp", "beat": 5, "lyric_line": "Censored words leave blank stares", "scene": {"mood": "longing", "colors": ["sunset orange", "purple", "pink"], "composition": "high angle", "camera": "tracking", "description": "A longing scene: 'Censored words leave blank stares'. high angle with sunset orange, purple, pink. Camera: tracking."}}
{"song": "Letters from Boot Camp", "beat": 6, "lyric_line": "The stamp is peeling at the corner", "scene": {"mood": "love", "colors": ["white", "blue", "brown"], "composition": "over-the-shoulder", "camera": "static", "description": "A love scene: 'The stamp is peeling at the corner'. over-the-shoulder with white, blue, brown. Camera: static."}}
{"song": "Letters from Boot Camp", "beat": 7, "lyric_line": "He folds it back exactly right", "scene": {"mood": "vulnerability", "colors": ["forest green", "brown", "gold"], "composition": "profile", "camera": "crane up", "description": "A vulnerability scene: 'He folds it back exactly right'. profile with forest green, brown, gold. Camera: crane up."}}
{"song": "Letters from Boot Camp", "beat": 8, "lyric_line": "Pictures curl at the edges now", "scene": {"mood": "distance", "colors": ["forest green", "brown", "gold"], "composition": "bird's eye", "camera": "dolly in", "description": "A distance scene: 'Pictures curl at the edges now'. bird's eye with forest green, brown, gold. Camera: dolly in."}}
{"song": "Letters from Boot Camp", "beat": 9, "lyric_line": "Her voice lives between the lines", "scene": {"mood": "hope", "colors": ["orange", "red", "brown"], "composition": "tracking shot", "camera": "gentle drift", "description": "A hope scene: 'Her voice lives between the lines'. tracking shot with orange, red, brown. Camera: gentle drift."}}
{"song": "Letters from Boot Camp", "beat": 10, "lyric_line": "Some letters carry more than words", "scene": {"mood": "longing", "colors": ["sunset orange", "purple", "pink"], "composition": "establishing", "camera": "locked-off", "description": "A longing scene: 'Some letters carry more than words'. establishing with sunset orange, purple, pink. Camera: locked-off."}}
{"song": "Flood Stage", "beat": 1, "lyric_line": "The river rose past the marker", "scene": {"mood": "loss", "colors": ["white", "blue", "brown"], "composition": "close-up", "camera": "slow pan", "description": "A loss scene: 'The river rose past the marker'. close-up with white, blue, brown. Camera: slow pan."}}
{"song": "Flood Stage", "beat": 2, "lyric_line": "Sandbags line the church basement door", "scene": {"mood": "resilience", "colors": ["green", "olive", "tan"], "composition": "wide shot", "camera": "steady", "description": "A resilience scene: 'Sandbags line the church basement door'. wide shot with green, olive, tan. Camera: steady."}}
{"song": "Flood Stage", "beat": 3, "lyric_line": "Photo albums float in the living room", "scene": {"mood": "community", "colors": ["white", "blue", "brown"], "composition": "medium shot", "camera": "handheld", "description": "A community scene: 'Photo albums float in the living room'. medium shot with white, blue, brown. Camera: handheld."}}
{"song": "Flood Stage", "beat": 4, "lyric_line": "The bridge went out Tuesday night", "scene": {"mood": "grief", "colors": ["navy", "silver", "white"], "composition": "low angle", "camera": "slow zoom", "description": "A grief scene: 'The bridge went out Tuesday night'. low angle with navy, silver, white. Camera: slow zoom."}}
{"song": "Flood Stage", "beat": 5, "lyric_line": "Neighbors carry what they can hold", "scene": {"mood": "rebuilding", "colors": ["green", "olive", "tan"], "composition": "high angle", "camera": "tracking", "description": "A rebuilding scene: 'Neighbors carry what they can hold'. high angle with green, olive, tan. Camera: tracking."}}
{"song": "Flood Stage", "beat": 6, "lyric_line": "Water stains reach the second step", "scene": {"mood": "loss", "colors": ["amber", "brown", "cream"], "composition": "over-the-shoulder", "camera": "static", "description": "A loss scene: 'Water stains reach the second step'. over-the-shoulder with amber, brown, cream. Camera: static."}}
{"song": "Flood Stage", "beat": 7, "lyric_line": "The well is tasting like the river", "scene": {"mood": "resilience", "colors": ["forest green", "brown", "gold"], "composition": "profile", "camera": "crane up", "description": "A resilience scene: 'The well is tasting like the river'. profile with forest green, brown, gold. Camera: crane up."}}
{"song": "Flood Stage", "beat": 8, "lyric_line": "Mud lines mark where hope used to sit", "scene": {"mood": "community", "colors": ["warm yellow", "barn red", "cream"], "composition": "bird's eye", "camera": "dolly in", "description": "A community scene: 'Mud lines mark where hope used to sit'. bird's eye with warm yellow, barn red, cream. Camera: dolly in."}}
{"song": "Flood Stage", "beat": 9, "lyric_line": "We rebuild on the same ground", "scene": {"mood": "grief", "colors": ["sunset orange", "purple", "pink"], "composition": "tracking shot", "camera": "gentle drift", "description": "A grief scene: 'We rebuild on the same ground'. tracking shot with sunset orange, purple, pink. Camera: gentle drift."}}
{"song": "Flood Stage", "beat": 10, "lyric_line": "Some floods are just the land remembering", "scene": {"mood": "rebuilding", "colors": ["dusty rose", "gold", "brown"], "composition": "establishing", "camera": "locked-off", "description": "A rebuilding scene: 'Some floods are just the land remembering'. establishing with dusty rose, gold, brown. Camera: locked-off."}}
{"song": "Barbed Wire Waltz", "beat": 1, "lyric_line": "The fence line needs mending again", "scene": {"mood": "endurance", "colors": ["orange", "red", "brown"], "composition": "close-up", "camera": "slow pan", "description": "A endurance scene: 'The fence line needs mending again'. close-up with orange, red, brown. Camera: slow pan."}}
{"song": "Barbed Wire Waltz", "beat": 2, "lyric_line": "Wire cuts through leather gloves", "scene": {"mood": "patience", "colors": ["amber", "brown", "cream"], "composition": "wide shot", "camera": "steady", "description": "A patience scene: 'Wire cuts through leather gloves'. wide shot with amber, brown, cream. Camera: steady."}}
{"song": "Barbed Wire Waltz", "beat": 3, "lyric_line": "Posts lean like tired old men", "scene": {"mood": "repair", "colors": ["grey", "mud brown", "green"], "composition": "medium shot", "camera": "handheld", "description": "A repair scene: 'Posts lean like tired old men'. medium shot with grey, mud brown, green. Camera: handheld."}}
{"song": "Barbed Wire Waltz", "beat": 4, "lyric_line": "The cattle watch from the far hill", "scene": {"mood": "land", "colors": ["warm yellow", "barn red", "cream"], "composition": "low angle", "camera": "slow zoom", "description": "A land scene: 'The cattle watch from the far hill'. low angle with warm yellow, barn red, cream. Camera: slow zoom."}}
{"song": "Barbed Wire Waltz", "beat": 5, "lyric_line": "We stretch and twist and tie", "scene": {"mood": "duty", "colors": ["amber", "brown", "cream"], "composition": "high angle", "camera": "tracking", "description": "A duty scene: 'We stretch and twist and tie'. high angle with amber, brown, cream. Camera: tracking."}}
{"song": "Barbed Wire Waltz", "beat": 6, "lyric_line": "Sunburn writes on the back of necks", "scene": {"mood": "endurance", "colors": ["sunset orange", "purple", "pink"], "composition": "over-the-shoulder", "camera": "static", "description": "A endurance scene: 'Sunburn writes on the back of necks'. over-the-shoulder with sunset orange, purple, pink. Camera: static."}}
{"song": "Barbed Wire Waltz", "beat": 7, "lyric_line": "The wind carries fence wire songs", "scene": {"mood": "patience", "colors": ["dusty rose", "gold", "brown"], "composition": "profile", "camera": "crane up", "description": "A patience scene: 'The wind carries fence wire songs'. profile with dusty rose, gold, brown. Camera: crane up."}}
{"song": "Barbed Wire Waltz", "beat": 8, "lyric_line": "We repair what the storm broke", "scene": {"mood": "repair", "colors": ["white", "blue", "brown"], "composition": "bird's eye", "camera": "dolly in", "description": "A repair scene: 'We repair what the storm broke'. bird's eye with white, blue, brown. Camera: dolly in."}}
{"song": "Barbed Wire Waltz", "beat": 9, "lyric_line": "Patience wears like good boots", "scene": {"mood": "land", "colors": ["copper", "dust", "gold"], "composition": "tracking shot", "camera": "gentle drift", "description": "A land scene: 'Patience wears like good boots'. tracking shot with copper, dust, gold. Camera: gentle drift."}}
{"song": "Barbed Wire Waltz", "beat": 10, "lyric_line": "Some fences are promises to the land", "scene": {"mood": "duty", "colors": ["amber", "brown", "cream"], "composition": "establishing", "camera": "locked-off", "description": "A duty scene: 'Some fences are promises to the land'. establishing with amber, brown, cream. Camera: locked-off."}}
{"song": "Tailgate Sunset", "beat": 1, "lyric_line": "The truck bed holds two lawn chairs", "scene": {"mood": "calm", "colors": ["grey", "mud brown", "green"], "composition": "close-up", "camera": "slow pan", "description": "A calm scene: 'The truck bed holds two lawn chairs'. close-up with grey, mud brown, green. Camera: slow pan."}}
{"song": "Tailgate Sunset", "beat": 2, "lyric_line": "Cooler rattles with ice and hope", "scene": {"mood": "companionship", "colors": ["orange", "red", "brown"], "composition": "wide shot", "camera": "steady", "description": "A companionship scene: 'Cooler rattles with ice and hope'. wide shot with orange, red, brown. Camera: steady."}}
{"song": "Tailgate Sunset", "beat": 3, "lyric_line": "Crickets start before the stars", "scene": {"mood": "simplicity", "colors": ["sky blue", "wheat", "brown"], "composition": "medium shot", "camera": "handheld", "description": "A simplicity scene: 'Crickets start before the stars'. medium shot with sky blue, wheat, brown. Camera: handheld."}}
{"song": "Tailgate Sunset", "beat": 4, "lyric_line": "The highway hums its evening song", "scene": {"mood": "nature", "colors": ["copper", "dust", "gold"], "composition": "low angle", "camera": "slow zoom", "description": "A nature scene: 'The highway hums its evening song'. low angle with copper, dust, gold. Camera: slow zoom."}}
{"song": "Tailgate Sunset", "beat": 5, "lyric_line": "We share a blanket past dark", "scene": {"mood": "contentment", "colors": ["forest green", "brown", "gold"], "composition": "high angle", "camera": "tracking", "description": "A contentment scene: 'We share a blanket past dark'. high angle with forest green, brown, gold. Camera: tracking."}}
{"song": "Tailgate Sunset", "beat": 6, "lyric_line": "Fireflies answer the sunset", "scene": {"mood": "calm", "colors": ["green", "olive", "tan"], "composition": "over-the-shoulder", "camera": "static", "description": "A calm scene: 'Fireflies answer the sunset'. over-the-shoulder with green, olive, tan. Camera: static."}}
{"song": "Tailgate Sunset", "beat": 7, "lyric_line": "Country station plays the old ones", "scene": {"mood": "companionship", "colors": ["navy", "silver", "white"], "composition": "profile", "camera": "crane up", "description": "A companionship scene: 'Country station plays the old ones'. profile with navy, silver, white. Camera: crane up."}}
{"song": "Tailgate Sunset", "beat": 8, "lyric_line": "The tailgate is our table now", "scene": {"mood": "simplicity", "colors": ["sunset orange", "purple", "pink"], "composition": "bird's eye", "camera": "dolly in", "description": "A simplicity scene: 'The tailgate is our table now'. bird's eye with sunset orange, purple, pink. Camera: dolly in."}}
{"song": "Tailgate Sunset", "beat": 9, "lyric_line": "Silence sits between the songs", "scene": {"mood": "nature", "colors": ["sky blue", "wheat", "brown"], "composition": "tracking shot", "camera": "gentle drift", "description": "A nature scene: 'Silence sits between the songs'. tracking shot with sky blue, wheat, brown. Camera: gentle drift."}}
{"song": "Tailgate Sunset", "beat": 10, "lyric_line": "Some nights are church without walls", "scene": {"mood": "contentment", "colors": ["sunset orange", "purple", "pink"], "composition": "establishing", "camera": "locked-off", "description": "A contentment scene: 'Some nights are church without walls'. establishing with sunset orange, purple, pink. Camera: locked-off."}}
{"song": "Granddad's Pocket Knife", "beat": 1, "lyric_line": "Bone handle smooth from his pocket", "scene": {"mood": "sacred", "colors": ["forest green", "brown", "gold"], "composition": "close-up", "camera": "slow pan", "description": "A sacred scene: 'Bone handle smooth from his pocket'. close-up with forest green, brown, gold. Camera: slow pan."}}
{"song": "Granddad's Pocket Knife", "beat": 2, "lyric_line": "Three blades none of them new", "scene": {"mood": "quiet", "colors": ["green", "olive", "tan"], "composition": "wide shot", "camera": "steady", "description": "A quiet scene: 'Three blades none of them new'. wide shot with green, olive, tan. Camera: steady."}}
{"song": "Granddad's Pocket Knife", "beat": 3, "lyric_line": "He carved initials in the oak", "scene": {"mood": "devotion", "colors": ["warm yellow", "barn red", "cream"], "composition": "medium shot", "camera": "handheld", "description": "A devotion scene: 'He carved initials in the oak'. medium shot with warm yellow, barn red, cream. Camera: handheld."}}
{"song": "Granddad's Pocket Knife", "beat": 4, "lyric_line": "The steel remembers every cut", "scene": {"mood": "stillness", "colors": ["warm yellow", "barn red", "cream"], "composition": "low angle", "camera": "slow zoom", "description": "A stillness scene: 'The steel remembers every cut'. low angle with warm yellow, barn red, cream. Camera: slow zoom."}}
{"song": "Granddad's Pocket Knife", "beat": 5, "lyric_line": "I open it the way he taught", "scene": {"mood": "awe", "colors": ["sunset orange", "purple", "pink"], "composition": "high angle", "camera": "tracking", "description": "A awe scene: 'I open it the way he taught'. high angle with sunset orange, purple, pink. Camera: tracking."}}
{"song": "Granddad's Pocket Knife", "beat": 6, "lyric_line": "The smallest blade is for detail", "scene": {"mood": "sacred", "colors": ["green", "olive", "tan"], "composition": "over-the-shoulder", "camera": "static", "description": "A sacred scene: 'The smallest blade is for detail'. over-the-shoulder with green, olive, tan. Camera: static."}}
{"song": "Granddad's Pocket Knife", "beat": 7, "lyric_line": "Whittling shavings catch the light", "scene": {"mood": "quiet", "colors": ["amber", "brown", "cream"], "composition": "profile", "camera": "crane up", "description": "A quiet scene: 'Whittling shavings catch the light'. profile with amber, brown, cream. Camera: crane up."}}
{"song": "Granddad's Pocket Knife", "beat": 8, "lyric_line": "He said a knife is a promise", "scene": {"mood": "devotion", "colors": ["sunset orange", "purple", "pink"], "composition": "bird's eye", "camera": "dolly in", "description": "A devotion scene: 'He said a knife is a promise'. bird's eye with sunset orange, purple, pink. Camera: dolly in."}}
{"song": "Granddad's Pocket Knife", "beat": 9, "lyric_line": "I carry it in my front pocket", "scene": {"mood": "stillness", "colors": ["forest green", "brown", "gold"], "composition": "tracking shot", "camera": "gentle drift", "description": "A stillness scene: 'I carry it in my front pocket'. tracking shot with forest green, brown, gold. Camera: gentle drift."}}
{"song": "Granddad's Pocket Knife", "beat": 10, "lyric_line": "Some tools are heirlooms of skill", "scene": {"mood": "awe", "colors": ["orange", "red", "brown"], "composition": "establishing", "camera": "locked-off", "description": "A awe scene: 'Some tools are heirlooms of skill'. establishing with orange, red, brown. Camera: locked-off."}}
{"song": "County Fair", "beat": 1, "lyric_line": "Ferris wheel lights the harvest sky", "scene": {"mood": "celebration", "colors": ["green", "olive", "tan"], "composition": "close-up", "camera": "slow pan", "description": "A celebration scene: 'Ferris wheel lights the harvest sky'. close-up with green, olive, tan. Camera: slow pan."}}
{"song": "County Fair", "beat": 2, "lyric_line": "Corn dogs and lemon shake-ups", "scene": {"mood": "community", "colors": ["orange", "red", "brown"], "composition": "wide shot", "camera": "steady", "description": "A community scene: 'Corn dogs and lemon shake-ups'. wide shot with orange, red, brown. Camera: steady."}}
{"song": "County Fair", "beat": 3, "lyric_line": "The livestock barn smells like home", "scene": {"mood": "youth", "colors": ["sunset orange", "purple", "pink"], "composition": "medium shot", "camera": "handheld", "description": "A youth scene: 'The livestock barn smells like home'. medium shot with sunset orange, purple, pink. Camera: handheld."}}
{"song": "County Fair", "beat": 4, "lyric_line": "Blue ribbons pinned to quilt squares", "scene": {"mood": "harvest", "colors": ["dusty rose", "gold", "brown"], "composition": "low angle", "camera": "slow zoom", "description": "A harvest scene: 'Blue ribbons pinned to quilt squares'. low angle with dusty rose, gold, brown. Camera: slow zoom."}}
{"song": "County Fair", "beat": 5, "lyric_line": "Kids scream on the zipper ride", "scene": {"mood": "fun", "colors": ["grey", "mud brown", "green"], "composition": "high angle", "camera": "tracking", "description": "A fun scene: 'Kids scream on the zipper ride'. high angle with grey, mud brown, green. Camera: tracking."}}
{"song": "County Fair", "beat": 6, "lyric_line": "Band plays covers by the stage", "scene": {"mood": "celebration", "colors": ["navy", "silver", "white"], "composition": "over-the-shoulder", "camera": "static", "description": "A celebration scene: 'Band plays covers by the stage'. over-the-shoulder with navy, silver, white. Camera: static."}}
{"song": "County Fair", "beat": 7, "lyric_line": "Hay bales line the midway path", "scene": {"mood": "community", "colors": ["sky blue", "wheat", "brown"], "composition": "profile", "camera": "crane up", "description": "A community scene: 'Hay bales line the midway path'. profile with sky blue, wheat, brown. Camera: crane up."}}
{"song": "County Fair", "beat": 8, "lyric_line": "We split the funnel cake in half", "scene": {"mood": "youth", "colors": ["copper", "dust", "gold"], "composition": "bird's eye", "camera": "dolly in", "description": "A youth scene: 'We split the funnel cake in half'. bird's eye with copper, dust, gold. Camera: dolly in."}}
{"song": "County Fair", "beat": 9, "lyric_line": "The demolition derby is last", "scene": {"mood": "harvest", "colors": ["sunset orange", "purple", "pink"], "composition": "tracking shot", "camera": "gentle drift", "description": "A harvest scene: 'The demolition derby is last'. tracking shot with sunset orange, purple, pink. Camera: gentle drift."}}
{"song": "County Fair", "beat": 10, "lyric_line": "Some summers live in a single night", "scene": {"mood": "fun", "colors": ["warm yellow", "barn red", "cream"], "composition": "establishing", "camera": "locked-off", "description": "A fun scene: 'Some summers live in a single night'. establishing with warm yellow, barn red, cream. Camera: locked-off."}}

129
training/scripts/augment_pairs.py Executable file
View File

@@ -0,0 +1,129 @@
#!/usr/bin/env python3
"""
augment_pairs.py — Training data augmentation: paraphrase and translate.
Usage:
python3 augment_pairs.py --input data.jsonl
python3 augment_pairs.py --input data.jsonl --paraphrases 3 --langs es,fr,de
python3 augment_pairs.py --input data.jsonl --llm-endpoint http://localhost:11434/v1
"""
import json, os, sys, re, random
from pathlib import Path
random.seed(42)
PARAPHRASE_TRANSFORMS = [
lambda s: re.sub(r"(\w+), (\w+)", r"\2, \1", s, count=1),
lambda s: f"A beautifully rendered scene: {s[0].lower()}{s[1:]}" if len(s) > 10 else s,
lambda s: s.replace("A ", "The ").replace("An ", "The ") if s.startswith(("A ", "An ")) else f"Here, {s[0].lower()}{s[1:]}",
lambda s: f"In a cinematic frame: {s}" if len(s) > 20 else s,
lambda s: s if ", " not in s else ", ".join(s.split(", ")[:2]),
]
TRANSLATIONS = {
"es": {"the":"el","a":"un","is":"es","in":"en","of":"de","and":"y","with":"con","scene":"escena","light":"luz","dark":"oscuro","warm":"cálido","rain":"lluvia","sun":"sol","moon":"luna","sky":"cielo","forest":"bosque","mountain":"montaña","ocean":"océano","golden":"dorado","blue":"azul","red":"rojo","green":"verde","silence":"silencio","dream":"sueño","love":"amor","hope":"esperanza","fear":"miedo","joy":"alegría","peace":"paz","beautiful":"hermoso","sad":"triste","shadow":"sombra","color":"color","silver":"plateado","white":"blanco","black":"negro","portray":"retrato"},
"fr": {"the":"le","a":"un","is":"est","in":"dans","of":"de","and":"et","with":"avec","scene":"scène","light":"lumière","dark":"sombre","warm":"chaud","rain":"pluie","sun":"soleil","moon":"lune","sky":"ciel","forest":"forêt","mountain":"montagne","ocean":"océan","golden":"doré","blue":"bleu","red":"rouge","green":"vert","silence":"silence","dream":"rêve","love":"amour","hope":"espoir","fear":"peur","joy":"joie","peace":"paix","beautiful":"beau","sad":"triste","shadow":"ombre","color":"couleur","silver":"argenté","white":"blanc","black":"noir"},
"de": {"the":"der","a":"ein","is":"ist","in":"in","of":"von","and":"und","with":"mit","scene":"Szene","light":"Licht","dark":"dunkel","warm":"warm","rain":"Regen","sun":"Sonne","moon":"Mond","sky":"Himmel","forest":"Wald","mountain":"Berg","ocean":"Ozean","golden":"golden","blue":"blau","red":"rot","green":"grün","silence":"Stille","dream":"Traum","love":"Liebe","hope":"Hoffnung","fear":"Angst","joy":"Freude","peace":"Frieden","beautiful":"schön","sad":"traurig","shadow":"Schatten","color":"Farbe","silver":"silbern","white":"weiß","black":"schwarz"},
}
LANG_NAMES = {"es": "Spanish", "fr": "French", "de": "German"}
def detect_text_field(entry):
for f in ["rich","terse","text","content","lyric_line","description","scene_description","prompt","scene"]:
if f in entry and isinstance(entry[f], str) and len(entry[f]) > 5:
return f
for k, v in entry.items():
if isinstance(v, str) and len(v) > 5:
return k
return None
def paraphrase(text):
t = random.choice(PARAPHRASE_TRANSFORMS)(text)
if t == text:
t = text.replace(" and ", " & ").replace(" with ", " alongside ")
if t == text:
t = f"In this scene: {text[0].lower()}{text[1:]}" if text[0].isupper() else text
return t
def translate(text, lang):
d = TRANSLATIONS.get(lang, {})
words = text.split()
out = []
for w in words:
lo = w.lower().strip(".,;:!?")
suf = w[len(w.rstrip(".,;:!?")):]
if lo in d:
out.append(d[lo] + suf)
else:
out.append(w)
return " ".join(out)
def augment_file(input_path, output_path=None, n_para=3, langs=None, llm_endpoint=None):
input_path = Path(input_path)
if output_path is None:
output_path = input_path.parent / f"{input_path.stem}_augmented{input_path.suffix}"
entries = [json.loads(l) for l in open(input_path) if l.strip()]
if not entries:
print(f"No entries in {input_path}"); return 0
tf = detect_text_field(entries[0])
if not tf:
print(f"ERROR: No text field in {input_path}", file=sys.stderr); return 0
print(f"Input: {input_path} ({len(entries)} entries, field={tf})")
aug_count = 0
with open(output_path, "w") as out:
for e in entries:
out.write(json.dumps(e, ensure_ascii=False) + "\n")
for i, e in enumerate(entries):
text = e[tf]
# Paraphrases
for p in range(n_para):
para = paraphrase(text)
if para != text:
ne = dict(e); ne[tf] = para
ne["_augmentation"] = f"paraphrase_{p+1}"
ne["_original"] = text[:100]
out.write(json.dumps(ne, ensure_ascii=False) + "\n")
aug_count += 1
# Translations
for lang in (langs or []):
tr = translate(text, lang)
if tr != text:
ne = dict(e); ne[tf] = tr
ne["_augmentation"] = f"translate_{lang}"
ne["_language"] = lang
ne["_original"] = text[:100]
out.write(json.dumps(ne, ensure_ascii=False) + "\n")
aug_count += 1
if (i+1) % 100 == 0:
print(f" {i+1}/{len(entries)} done ({aug_count} augmented)")
total = len(entries) + aug_count
print(f"Done: {len(entries)} originals + {aug_count} augmented = {total}")
print(f"Output: {output_path}")
return aug_count
def main():
import argparse
p = argparse.ArgumentParser()
p.add_argument("--input", required=True)
p.add_argument("--output", default=None)
p.add_argument("--paraphrases", type=int, default=3)
p.add_argument("--langs", default="es,fr,de")
p.add_argument("--llm-endpoint", default=None)
args = p.parse_args()
langs = [l.strip() for l in args.langs.split(",") if l.strip()] if args.langs else []
augment_file(args.input, args.output, args.paraphrases, langs, args.llm_endpoint)
if __name__ == "__main__":
main()