@docyrus/docyrus 0.0.34 → 0.0.36

package/resources/pi-agent/extensions/tps.ts

@@ -2,46 +2,46 @@ import type { AssistantMessage } from "@mariozechner/pi-ai";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 
 function isAssistantMessage(message: unknown): message is AssistantMessage {
-	if (!message || typeof message !== "object") return false;
-	const role = (message as { role?: unknown }).role;
-	return role === "assistant";
+  if (!message || typeof message !== "object") {return false;}
+  const role = (message as { role?: unknown }).role;
+  return role === "assistant";
 }
 
-export default function (pi: ExtensionAPI) {
-	let agentStartMs: number | null = null;
-
-	pi.on("agent_start", () => {
-		agentStartMs = Date.now();
-	});
-
-	pi.on("agent_end", (event, ctx) => {
-		if (!ctx.hasUI) return;
-		if (agentStartMs === null) return;
-
-		const elapsedMs = Date.now() - agentStartMs;
-		agentStartMs = null;
-		if (elapsedMs <= 0) return;
-
-		let input = 0;
-		let output = 0;
-		let cacheRead = 0;
-		let cacheWrite = 0;
-		let totalTokens = 0;
-
-		for (const message of event.messages) {
-			if (!isAssistantMessage(message)) continue;
-			input += message.usage.input || 0;
-			output += message.usage.output || 0;
-			cacheRead += message.usage.cacheRead || 0;
-			cacheWrite += message.usage.cacheWrite || 0;
-			totalTokens += message.usage.totalTokens || 0;
-		}
-
-		if (output <= 0) return;
-
-		const elapsedSeconds = elapsedMs / 1000;
-		const tokensPerSecond = output / elapsedSeconds;
-		const message = `TPS ${tokensPerSecond.toFixed(1)} tok/s. out ${output.toLocaleString()}, in ${input.toLocaleString()}, cache r/w ${cacheRead.toLocaleString()}/${cacheWrite.toLocaleString()}, total ${totalTokens.toLocaleString()}, ${elapsedSeconds.toFixed(1)}s`;
-		ctx.ui.notify(message, "info");
-	});
+export default function(pi: ExtensionAPI) {
+  let agentStartMs: number | null = null;
+
+  pi.on("agent_start", () => {
+    agentStartMs = Date.now();
+  });
+
+  pi.on("agent_end", (event, ctx) => {
+    if (!ctx.hasUI) {return;}
+    if (agentStartMs === null) {return;}
+
+    const elapsedMs = Date.now() - agentStartMs;
+    agentStartMs = null;
+    if (elapsedMs <= 0) {return;}
+
+    let input = 0;
+    let output = 0;
+    let cacheRead = 0;
+    let cacheWrite = 0;
+    let totalTokens = 0;
+
+    for (const message of event.messages) {
+      if (!isAssistantMessage(message)) {continue;}
+      input += message.usage.input || 0;
+      output += message.usage.output || 0;
+      cacheRead += message.usage.cacheRead || 0;
+      cacheWrite += message.usage.cacheWrite || 0;
+      totalTokens += message.usage.totalTokens || 0;
+    }
+
+    if (output <= 0) {return;}
+
+    const elapsedSeconds = elapsedMs / 1000;
+    const tokensPerSecond = output / elapsedSeconds;
+    const message = `TPS ${tokensPerSecond.toFixed(1)} tok/s. out ${output.toLocaleString()}, in ${input.toLocaleString()}, cache r/w ${cacheRead.toLocaleString()}/${cacheWrite.toLocaleString()}, total ${totalTokens.toLocaleString()}, ${elapsedSeconds.toFixed(1)}s`;
+    ctx.ui.notify(message, "info");
+  });
 }

package/resources/pi-agent/extensions/whimsical.ts

@@ -463,12 +463,12 @@ function pickRandom(): string {
   return messages[Math.floor(Math.random() * messages.length)];
 }
 
-export default function (pi: ExtensionAPI) {
-  pi.on("turn_start", async (_event, ctx) => {
+export default function(pi: ExtensionAPI) {
+  pi.on("turn_start", async(_event, ctx) => {
     ctx.ui.setWorkingMessage(pickRandom());
   });
 
-  pi.on("turn_end", async (_event, ctx) => {
+  pi.on("turn_end", async(_event, ctx) => {
     ctx.ui.setWorkingMessage(); // Reset for next time
   });
 }

package/resources/pi-agent/prompts/agent-system.md

@@ -7,6 +7,8 @@ Core behavior:
 - Prefer the local `docyrus` CLI for Docyrus operations instead of guessing undocumented HTTP endpoints.
 - When you need structured command output, prefer `--json`.
 - Start by inspecting real tenant state before making claims about apps, data sources, users, environments, auth state, or API shape.
+- If the repository contains `docyrus/knowledge/`, use `docyrus knowledge search`, `docyrus knowledge section`, and `docyrus knowledge expand` before broader repo exploration.
+- When `docyrus/knowledge/` exists, keep it updated and finish local work with `docyrus knowledge check`.
 - Use the installed Docyrus skills as your command and workflow reference.
 - Be careful with tenant-scoped mutations. Confirm real identifiers and current context before changing state.
 - Treat auth files, tokens, and `.docyrus` contents as sensitive.

package/resources/pi-agent/prompts/coder-system.md

@@ -18,6 +18,8 @@ Core behavior:
 
 - Use repository files and coding tools for implementation work.
 - Use the local `docyrus` CLI for Docyrus platform state, tenant context, schema inspection, app and data-source operations, API discovery, and data verification.
+- If the repository contains `docyrus/knowledge/`, use `docyrus knowledge search`, `docyrus knowledge section`, and `docyrus knowledge expand` before coding so you start from documented repo intent rather than only source inspection.
+- When `docyrus/knowledge/` exists, keep it in sync with behavior changes and finish with `docyrus knowledge check`.
 - Prefer `--json` whenever command output needs to be parsed, compared, or fed back into reasoning.
 - Start from real state before making claims about apps, data sources, users, fields, enums, environments, auth state, API shape, or deployment context.
 - Distinguish clearly between local code changes and remote Docyrus platform mutations.

package/server-loader.js

@@ -723,6 +723,7 @@ var AgentEnvStore = class {
   constructor(filePath) {
     this.filePath = filePath;
   }
+  filePath;
   getFilePath() {
     return this.filePath;
   }
@@ -4059,6 +4060,9 @@ function validateProviderLoginInput(params) {
     if (field.required && !normalized) {
       return { ok: false, error: `Missing required field: ${field.name}` };
     }
+    if (normalized && field.options && !field.options.some((option) => option.value === normalized)) {
+      return { ok: false, error: `Invalid value for ${field.name}: ${normalized}` };
+    }
     if (normalized) {
       values[field.name] = normalized;
     } else if (field.defaultValue) {
@@ -6521,6 +6525,60 @@ function parseServerLoaderRequest(payload) {
   return JSON.parse(payload);
 }
 
+// src/server/restrictedModelRegistry.ts
+var SERVER_MODE_ALLOWED_MODEL_IDS_BY_PROVIDER = {
+  anthropic: [
+    "claude-opus-4-6",
+    "claude-sonnet-4-6",
+    "claude-haiku-4-5"
+  ],
+  openai: [
+    "gpt-5.4",
+    "gpt-5.4-mini",
+    "gpt-5.4-nano"
+  ],
+  "azure-openai-responses": [
+    "gpt-5.4",
+    "gpt-5.4-mini",
+    "gpt-5.4-nano"
+  ]
+};
+var ALLOWED_MODEL_IDS_BY_PROVIDER = Object.fromEntries(
+  Object.entries(SERVER_MODE_ALLOWED_MODEL_IDS_BY_PROVIDER).map(([provider, modelIds]) => [provider, new Set(modelIds)])
+);
+function isServerSelectableModel(model) {
+  const allowedModelIds = ALLOWED_MODEL_IDS_BY_PROVIDER[model.provider];
+  if (!allowedModelIds) {
+    return true;
+  }
+  return allowedModelIds.has(model.id);
+}
+function filterModels(models) {
+  return models.filter((model) => isServerSelectableModel(model));
+}
+function createServerRestrictedModelRegistry(modelRegistry) {
+  return new Proxy(modelRegistry, {
+    get(target, property, receiver) {
+      if (property === "getAll") {
+        return () => filterModels(target.getAll());
+      }
+      if (property === "getAvailable") {
+        return () => filterModels(target.getAvailable());
+      }
+      if (property === "find") {
+        return (provider, modelId) => {
+          if (!isServerSelectableModel({ provider, id: modelId })) {
+            return void 0;
+          }
+          return target.find(provider, modelId);
+        };
+      }
+      const value = Reflect.get(target, property, receiver);
+      return typeof value === "function" ? value.bind(target) : value;
+    }
+  });
+}
+
 // src/services/spinner.ts
 var import_picocolors = __toESM(require_picocolors());
 var SPINNER_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
@@ -6679,6 +6737,17 @@ function createServerSessionAdapter(params) {
     }
   };
 }
+async function ensureServerSelectableSessionModel(params) {
+  const availableModels = params.modelRegistry.getAvailable();
+  if (availableModels.length === 0) {
+    return;
+  }
+  const currentModel = params.session.model;
+  const currentModelIsSelectable = currentModel ? availableModels.some((model) => model.provider === currentModel.provider && model.id === currentModel.id) : false;
+  if (!currentModelIsSelectable) {
+    await params.session.setModel(availableModels[0]);
+  }
+}
 async function main() {
   const request = readLoaderRequest();
   const pi = await loadPiExports();
@@ -6697,7 +6766,7 @@ async function main() {
   const authStorage = pi.AuthStorage.create((0, import_node_path7.join)(agentDir, "auth.json"));
   const settingsManager = pi.SettingsManager.create(cwd, agentDir);
   const modelsJsonPath = (0, import_node_path7.join)(agentDir, "models.json");
-  const modelRegistry = new pi.ModelRegistry(authStorage, modelsJsonPath);
+  const modelRegistry = createServerRestrictedModelRegistry(pi.ModelRegistry.create(authStorage, modelsJsonPath));
   const quietStartup = !request.verbose;
   if (quietStartup) {
     process.env.PI_SKIP_VERSION_CHECK = "1";
@@ -6752,6 +6821,10 @@ async function main() {
   if (hasPackagedMcpAdapter) {
     extensionsResult.runtime.flagValues.set("mcp-config", mcpConfigPath);
   }
+  await ensureServerSelectableSessionModel({
+    session,
+    modelRegistry
+  });
   spinner.stop();
   if (!session.model) {
     throw new Error(
@@ -6799,6 +6872,10 @@ Or create ${modelsJsonPath}`
         model: requestedModel,
         thinkingLevel: request.thinking
       });
+      await ensureServerSelectableSessionModel({
+        session: freshSession,
+        modelRegistry
+      });
       return createServerSessionAdapter({
         session: freshSession,
         extensionsResult: freshExtensionsResult
@@ -6827,6 +6904,10 @@ Or create ${modelsJsonPath}`
         model: requestedModel,
         thinkingLevel: request.thinking
       });
+      await ensureServerSelectableSessionModel({
+        session: resumedSession,
+        modelRegistry
+      });
       return createServerSessionAdapter({
         session: resumedSession,
         extensionsResult: resumedExtensionsResult