pi-jev-guard 0.1.4 → 0.1.6

package/README.md

@@ -45,6 +45,9 @@ In chat, l'LLM può chiamare `jev_validate` per controlli mirati.
 
 Nessun login extra: il twin vive nello stesso provider e riusa la credenziale già configurata (env, stored, OAuth/SSO come Codex business plan). I modelli originali restano intatti e selezionabili.
 
+In `automatic` il twin viene registrato già all'avvio (prima del ripristino del
+modello di sessione), così la sessione può riaprire direttamente su `__jev`.
+
 Torna normale con `/model` (modello normale) + `/jev mode on-demand`.
 In `automatic`, selezionare un modello non protetto lo fa ritornare
 subito al twin (enforcement a livello selezione).

package/extensions/index.ts

@@ -1,8 +1,7 @@
 import type { ExtensionAPI, ExtensionContext, ModelRegistry } from "@earendil-works/pi-coding-agent";
 import type { Model, Provider } from "@earendil-works/pi-ai";
 import type { Api } from "@earendil-works/pi-ai";
-import { builtinProviders } from "@earendil-works/pi-ai/providers/all";
-import { registerJevCommand, type UpstreamResult } from "../src/commands.ts";
+import { builtinProviders } from "@earendil-works/pi-ai/providers/all";import { registerJevCommand, type UpstreamResult } from "../src/commands.ts";
 import { registerJevAskTool } from "../src/ask.ts";
 import {
   configSnapshot,
@@ -22,8 +21,8 @@ import {
   twinIdFor,
 } from "../src/automatic/overlay.ts";
 import type { GateVerdictReport } from "../src/automatic/guardian.ts";
-import { DEFAULT_GATE_RULES } from "../src/reviewer.ts";
-import { elideWithMarker } from "../src/reviewer.ts";
+import { readStoreModels } from "../src/store-models.ts";
+import { DEFAULT_GATE_RULES, elideWithMarker } from "../src/reviewer.ts";
 import {
   TOOL_POLICY_REVISION,
   checkFilePath,
@@ -307,35 +306,67 @@ export default function (pi: ExtensionAPI) {
     getLastGate: () => lastGate,
   });
 
-  // Seed headless: --print risolve --model prima di session_start, quindi per
-  // il solo caso deepseek/deepseek-flash si pre-registra un overlay statico
-  // (sostituito dal clone live appena il registry è disponibile).
-  // Richiede JEV_AUTO_UPSTREAM: senza, nessun guarded (opt-in).
-  const bootAuto = process.env.JEV_AUTO_UPSTREAM?.trim().toLowerCase();
-  if (bootAuto === "deepseek/deepseek-flash") {
+  // Seed all'avvio: pi ripristina il modello di sessione PRIMA di
+  // session_start, quindi un twin persistito non esisterebbe ancora e
+  // cadrebbe su "Could not restore model ...__jev". Pre-registriamo un
+  // overlay statico dai provider builtin quando la config persiste
+  // automatic (o JEV_AUTO_UPSTREAM è impostato); session_start lo
+  // sostituisce col clone live dal registry.
+  const bootAuto = process.env.JEV_AUTO_UPSTREAM?.trim();
+  const autoOff = bootAuto === "0" || bootAuto?.toLowerCase() === "off";
+  let seedTarget: { provider: string; model: string } | undefined;
+  if (bootAuto && !autoOff) {
+    const parsed = parseUpstreamRef([bootAuto]);
+    if (!("error" in parsed)) seedTarget = parsed;
+  } else if (!bootAuto && config.mode === "automatic") {
+    seedTarget = {
+      provider: config.automatic.upstreamProvider,
+      model: config.automatic.upstreamModel,
+    };
+  }
+  if (seedTarget) {
     try {
       // providers/all è mappato dagli alias di pi; il subpath del singolo
       // provider non lo è e fallirebbe a load (solo installazioni npm).
-      const base = builtinProviders().find((p) => p.id === "deepseek");
-      if (!base) throw new Error("deepseek provider not in builtins");
+      const base = builtinProviders().find((p) => p.id === seedTarget?.provider);
+      if (!base) throw new Error(`provider ${seedTarget.provider} not in builtins`);
+      // Il catalogo builtin può non avere il modello salvato: provider
+      // dinamici (es. openai-codex) tengono la lista nel models-store,
+      // altri (deepseek-flash su versioni vecchie) in una definizione statica.
+      const storeModels = readStoreModels(seedTarget.provider);
+      const extra = [
+        ...(seedTarget.provider === "deepseek" && seedTarget.model === "deepseek-flash"
+          ? [STATIC_DEEPSEEK_FLASH]
+          : []),
+        ...storeModels,
+      ];
       const seedUpstream: Provider = {
         ...base,
-        getModels: () => [...base.getModels(), { ...STATIC_DEEPSEEK_FLASH }],
+        getModels: () => {
+          const models = base.getModels();
+          const missing = extra.filter(
+            (m) => !models.some((existing) => existing.id === m.id),
+          );
+          return [...models, ...missing];
+        },
       };
+      const hasModel = seedUpstream
+        .getModels()
+        .some((m) => m.id === seedTarget?.model);
+      if (!hasModel) throw new Error(`model ${seedTarget.model} not in builtins`);
       const overlay = buildOverlaidProvider({
         upstream: seedUpstream,
-        twinModelIds: ["deepseek-flash"],
+        twinModelIds: [seedTarget.model],
         getConfig,
         getReview: getGateReview,
         onVerdict: recordGateVerdict,
       });
       pi.registerProvider(overlay);
-      upstreamRefs.set("deepseek", seedUpstream);
-      overlaidProviderId = "deepseek";
-      twinTarget = { provider: "deepseek", model: "deepseek-flash" };
+      upstreamRefs.set(seedTarget.provider, seedUpstream);
+      overlaidProviderId = seedTarget.provider;
+      twinTarget = { provider: seedTarget.provider, model: seedTarget.model };
     } catch {
-      overlaidProviderId = undefined;
-      twinTarget = undefined;
+      // Nessun seed: il twin arriverà da session_start se possibile.
     }
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-jev-guard",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "type": "module",
   "description": "Jev validation guard for pi: on-demand tool + automatic provider gate. Dual backend: TypeSafe direct + OpenRouter.",
   "keywords": [
@@ -46,5 +46,11 @@
     "typebox": "1.3.34",
     "typescript": "5.9.3",
     "vitest": "5.0.1"
+  },
+  "allowScripts": {
+    "@google/genai@1.52.0": true,
+    "@openrouter/sdk@1.3.1": true,
+    "protobufjs@7.6.6": true,
+    "esbuild@0.28.2": true
   }
 }

package/src/store-models.ts

@@ -0,0 +1,47 @@
+/**
+ * Lettura read-only dei modelli dinamici dal models-store di pi.
+ *
+ * Alcuni provider (es. openai-codex) non hanno modelli nel catalogo statico:
+ * la lista arriva da `<agentDir>/models-store.json` e viene popolata al
+ * refresh con le credenziali. Senza questo, il seed all'avvio non troverebbe
+ * il modello e il twin non esisterebbe al momento del ripristino sessione.
+ *
+ * Solo lettura, best-effort: file assente/corrotto → nessun modello.
+ */
+import { readFileSync } from "node:fs";
+import { join } from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+import type { Api, Model } from "@earendil-works/pi-ai";
+
+export function modelsStorePath(agentDir?: string): string {
+  return join(agentDir ?? getAgentDir(), "models-store.json");
+}
+
+function isUsableModel(value: unknown, providerId: string): value is Model<Api> {
+  if (typeof value !== "object" || value === null) return false;
+  const m = value as Record<string, unknown>;
+  return (
+    typeof m.id === "string" &&
+    m.id.length > 0 &&
+    typeof m.api === "string" &&
+    m.api.length > 0 &&
+    (m.provider === undefined || m.provider === providerId)
+  );
+}
+
+/** Modelli salvati nello store per un provider, normalizzati sull'id richiesto. */
+export function readStoreModels(providerId: string, agentDir?: string): Model<Api>[] {
+  try {
+    const raw = JSON.parse(readFileSync(modelsStorePath(agentDir), "utf8")) as Record<
+      string,
+      { models?: unknown }
+    >;
+    const entry = raw?.[providerId];
+    const models = Array.isArray(entry?.models) ? entry.models : [];
+    return models
+      .filter((m) => isUsableModel(m, providerId))
+      .map((m) => ({ ...(m as Model<Api>), provider: providerId }));
+  } catch {
+    return [];
+  }
+}