pi-vision-guard 0.1.0 → 0.2.1

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.2.1
+
+- Ranking tries each registered classifier in order until one answers: a keyless or failing first pick (e.g. a built-in cloud entry with no API key) no longer blocks a working backend later in the list. Recency fallback now only fires when no classifier answers at all.
+
+## 0.2.0
+
+- Ranking uses the first available registered classifier with no built-in model preference (previously preferred a local clef provider); fully backend-agnostic, pairs with `pi-local-classifier` for offline use.
+
 ## 0.1.0 — Initial release
 
 - Cap image payloads per request (default 5, `/vision-guard 1..6`), dropping to 1 past 35% context usage.

package/README.md

@@ -32,7 +32,7 @@ Three layers, all automatic:
 
 ## Classifier ranking
 
-Ranking lights up automatically with any registered Pi classifier — no settings. A local `clef-local/clef-flash` provider is preferred when present, otherwise the first available classifier is used. The questions are plain keep/drop judgments, so any Jev-family decision model works.
+Ranking lights up automatically with your registered Pi classifiers — no settings. Each is tried in order until one answers, so a keyless cloud entry never blocks a working local backend. The questions are plain keep/drop judgments, so any Jev-family decision model works. With no classifier present, the guard silently uses newest-first order. Pairs with `pi-local-classifier` for fully offline ranking with no API key.
 
 ## ⚠️ Cache-churn caveat
 

package/extensions/vision-context-guard.ts

@@ -8,11 +8,8 @@ const MAX_EDGE = 1600;
 const DEFAULT_RETAINED_IMAGES = 5;
 const LONG_CONTEXT_FRACTION = 0.35;
 const MAX_SOURCE_BYTES = 64 * 1024 * 1024;
-// Preferred registry classifier for relevance ranking. Any registered
-// classifier works — local clef first when present, otherwise the first
-// available one. Recency fallback covers no-classifier setups.
-const CLEF_PROVIDER = "clef-local";
-const CLEF_MODEL = "clef-flash";
+// Relevance ranking uses the first available registry classifier, if any.
+// Recency fallback covers no-classifier setups.
 const OMITTED_TEXT =
 	"[Older image omitted from this request by the local vision context guard. Use existing textual observations or reread one specific image if it is essential.]";
 
@@ -158,16 +155,19 @@ function lastUserText(messages: any[]): string {
 	return "(unknown)";
 }
 
-/** One batched classifier call scoring every candidate. Returns scores aligned with slots, or undefined on any failure. */
+/**
+ * One batched classifier call scoring every candidate. Tries each registered
+ * classifier in order until one returns usable scores — a keyless or failing
+ * first pick must not block a working backend later in the list. Returns
+ * scores aligned with slots, or undefined when none answer. Abort still throws.
+ */
 async function scoreKeepers(
 	slots: ImageSlot[],
 	taskHint: string,
 	ctx: ExtensionContext,
 ): Promise<number[] | undefined> {
-	const clf =
-		ctx.modelRegistry.findOfType("classifier", CLEF_PROVIDER, CLEF_MODEL) ??
-		ctx.modelRegistry.getModelsOfType("classifier")[0];
-	if (!clf) return undefined;
+	const models = ctx.modelRegistry.getModelsOfType("classifier");
+	if (models.length === 0) return undefined;
 	const questions: Record<string, ClassifierQuestion> = {};
 	for (let i = 0; i < slots.length; i++) {
 		questions[`keep_${i}`] = {
@@ -179,44 +179,51 @@ async function scoreKeepers(
 			},
 		};
 	}
-	try {
-		const result = await ctx.modelRegistry.classify(
-			clf,
-			{
-				state: {
-					task: taskHint,
-					images: slots.map((s, i) => ({
-						index: i,
-						role: s.role,
-						context: s.text.slice(0, 300) || "(no surrounding text)",
-					})),
+	for (const clf of models) {
+		if (ctx.signal?.aborted) throw new Error("aborted");
+		try {
+			const result = await ctx.modelRegistry.classify(
+				clf,
+				{
+					state: {
+						task: taskHint,
+						images: slots.map((s, i) => ({
+							index: i,
+							role: s.role,
+							context: s.text.slice(0, 300) || "(no surrounding text)",
+						})),
+					},
+					questions,
 				},
-				questions,
-			},
-			{ signal: ctx.signal },
-		);
-		if (result.stopReason !== "stop") {
+				{ signal: ctx.signal },
+			);
 			if (result.stopReason === "aborted" || ctx.signal?.aborted) throw new Error("aborted");
-			return undefined;
-		}
-		const scores: number[] = [];
-		for (let i = 0; i < slots.length; i++) {
-			const a = result.answers[`keep_${i}`];
-			if (a?.type !== "bool") return undefined;
-			scores.push(a.probability);
+			if (result.stopReason !== "stop") continue;
+			const scores: number[] = [];
+			let usable = true;
+			for (let i = 0; i < slots.length; i++) {
+				const a = result.answers[`keep_${i}`];
+				if (a?.type !== "bool") {
+					usable = false;
+					break;
+				}
+				scores.push(a.probability);
+			}
+			if (!usable) continue;
+			return scores;
+		} catch (e) {
+			if (ctx.signal?.aborted) throw e;
+			continue;
 		}
-		return scores;
-	} catch (e) {
-		if (ctx.signal?.aborted) throw e;
-		return undefined;
 	}
+	return undefined;
 }
 
 /**
  * Relevance-ranked prune within the same budget the recency guard used.
- * Newest image is always kept; remaining slots go to the highest clef
+ * Newest image is always kept; remaining slots go to the highest classifier
  * keep-scores, falling back to recency when there is no ranking signal
- * (bare images) or the sidecar is unreachable.
+ * (bare images) or no classifier answers.
  */
 async function pruneRanked(
 	messages: any[],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-vision-guard",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "description": "Pi extension that caps, shrinks, and relevance-ranks image payloads so long vision sessions stop 413ing",
   "author": "Jesse Wise",
   "license": "MIT",
@@ -12,14 +12,28 @@
   "bugs": {
     "url": "https://github.com/jwise7/pi-vision-guard/issues"
   },
-  "keywords": ["pi-package", "pi", "vision", "images", "context", "coding-agent"],
+  "keywords": [
+    "pi-package",
+    "pi",
+    "vision",
+    "images",
+    "context",
+    "coding-agent"
+  ],
   "publishConfig": {
     "access": "public"
   },
   "pi": {
-    "extensions": ["./extensions"]
+    "extensions": [
+      "./extensions"
+    ]
   },
-  "files": ["extensions/", "README.md", "CHANGELOG.md", "LICENSE"],
+  "files": [
+    "extensions/",
+    "README.md",
+    "CHANGELOG.md",
+    "LICENSE"
+  ],
   "scripts": {
     "test": "node --test tests/vision-guard/scoping.test.mjs tests/vision-guard/ranking.test.mjs"
   },