@oh-my-pi/pi-coding-agent 16.0.5 → 16.0.7

package/src/advisor/advise-tool.ts

@@ -1,19 +1,23 @@
-import type { AgentTool, AgentToolContext, AgentToolResult, AgentToolUpdateCallback } from "@oh-my-pi/pi-agent-core";
+import type {
+	AgentIdentity,
+	AgentTelemetryConfig,
+	AgentTool,
+	AgentToolContext,
+	AgentToolResult,
+	AgentToolUpdateCallback,
+} from "@oh-my-pi/pi-agent-core";
 import { escapeXmlText } from "@oh-my-pi/pi-utils";
-import { z } from "zod/v4";
+import { type } from "arktype";
 import adviseDescription from "../prompts/advisor/advise-tool.md" with { type: "text" };
 
-const adviseSchema = z.object({
-	note: z
-		.string()
-		.describe("One concrete piece of advice for the agent you are watching. Terse, specific, actionable."),
-	severity: z
-		.enum(["nit", "concern", "blocker"])
-		.optional()
-		.describe("How strongly to weigh this. Omit for a plain nit."),
+const adviseSchema = type({
+	note: type("string").describe(
+		"One concrete piece of advice for the agent you are watching. Terse, specific, actionable.",
+	),
+	"severity?": type("'nit' | 'concern' | 'blocker'").describe("How strongly to weigh this. Omit for a plain nit."),
 });
 
-export type AdviseParams = z.infer<typeof adviseSchema>;
+export type AdviseParams = typeof adviseSchema.infer;
 
 export type AdvisorSeverity = "nit" | "concern" | "blocker";
 
@@ -109,6 +113,25 @@ export function resolveAdvisorDeliveryChannel(opts: {
 	return "steer";
 }
 
+/**
+ * Derive the advisor loop's telemetry from the primary session's config so the
+ * advisor model's GenAI spans and usage/cost hooks (onChatUsage, onCostDelta,
+ * costEstimator) fire under the same pipeline as every other model call —
+ * stamped with the advisor's own agent identity. `conversationId` is cleared so
+ * the advisor loop falls back to its own `-advisor` session id for
+ * `gen_ai.conversation.id` instead of inheriting the primary's conversation.
+ *
+ * Returns undefined when the primary has no telemetry (instrumentation off), so
+ * the advisor `Agent` stays a zero-overhead no-op as well.
+ */
+export function deriveAdvisorTelemetry(
+	primaryTelemetry: AgentTelemetryConfig | undefined,
+	identity: AgentIdentity,
+): AgentTelemetryConfig | undefined {
+	if (!primaryTelemetry) return undefined;
+	return { ...primaryTelemetry, agent: identity, conversationId: undefined };
+}
+
 /**
  * Side-effect-free investigation tools handed to the advisor agent so it can
  * inspect the workspace before weighing in. Names match the primary session's

package/src/autoresearch/tools/init-experiment.ts

@@ -1,7 +1,7 @@
 import * as path from "node:path";
 
 import { Text } from "@oh-my-pi/pi-tui";
-import { z } from "zod/v4";
+import { type } from "arktype";
 import type { ToolDefinition } from "../../extensibility/extensions";
 import type { Theme } from "../../modes/theme/theme";
 import { replaceTabs, truncateToWidth } from "../../tools/render-utils";
@@ -16,21 +16,18 @@ export const HARNESS_FILENAME = "autoresearch.sh";
 export const DEFAULT_HARNESS_COMMAND = `bash ${HARNESS_FILENAME}`;
 const HARNESS_COMMIT_TITLE = "autoresearch: harness setup";
 
-const initExperimentSchema = z.object({
-	name: z.string().describe("experiment name"),
-	goal: z.string().describe("session goal").optional(),
-	primary_metric: z.string().describe("primary metric name"),
-	metric_unit: z.string().describe("metric unit (e.g. ms, µs, mb)").optional(),
-	direction: z
-		.enum(["lower", "higher"] as const)
-		.describe("better direction (default lower)")
-		.optional(),
-	secondary_metrics: z.array(z.string()).describe("secondary metric names").optional(),
-	scope_paths: z.array(z.string()).describe("expected-to-modify paths").optional(),
-	off_limits: z.array(z.string()).describe("off-limits paths").optional(),
-	constraints: z.array(z.string()).describe("free-form constraints").optional(),
-	max_iterations: z.number().describe("soft iteration cap per segment").optional(),
-	new_segment: z.boolean().describe("bump to a new segment in existing session").optional(),
+const initExperimentSchema = type({
+	name: type("string").describe("experiment name"),
+	"goal?": type("string").describe("session goal"),
+	primary_metric: type("string").describe("primary metric name"),
+	"metric_unit?": type("string").describe("metric unit (e.g. ms, µs, mb)"),
+	"direction?": type("'lower' | 'higher'").describe("better direction (default lower)"),
+	"secondary_metrics?": type("string[]").describe("secondary metric names"),
+	"scope_paths?": type("string[]").describe("expected-to-modify paths"),
+	"off_limits?": type("string[]").describe("off-limits paths"),
+	"constraints?": type("string[]").describe("free-form constraints"),
+	"max_iterations?": type("number").describe("soft iteration cap per segment"),
+	"new_segment?": type("boolean").describe("bump to a new segment in existing session"),
 });
 
 interface InitExperimentDetails {

package/src/autoresearch/tools/log-experiment.ts

@@ -2,7 +2,7 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 
 import { Text } from "@oh-my-pi/pi-tui";
-import { z } from "zod/v4";
+import { type } from "arktype";
 import type { ToolDefinition } from "../../extensibility/extensions";
 import type { Theme } from "../../modes/theme/theme";
 import { replaceTabs, truncateToWidth } from "../../tools/render-utils";
@@ -36,23 +36,20 @@ import type {
 
 const EXPERIMENT_TOOL_NAMES = ["init_experiment", "run_experiment", "log_experiment", "update_notes"];
 
-const logExperimentSchema = z.object({
-	metric: z.number().describe("primary metric value"),
-	status: z.enum(["keep", "discard", "crash", "checks_failed"] as const).describe("run outcome"),
-	description: z.string().describe("short run description"),
-	metrics: z.record(z.string(), z.number()).describe("secondary metrics").optional(),
-	asi: z.object({}).passthrough().describe("free-form structured metadata").optional(),
-	commit: z.string().describe("override recorded commit hash").optional(),
-	justification: z.string().describe("required when keeping a scope-deviating run").optional(),
-	flag_runs: z
-		.array(
-			z.object({
-				run_id: z.number().describe("run id to flag"),
-				reason: z.string().describe("why this run is suspect"),
-			}),
-		)
-		.describe("flag earlier runs as suspect")
-		.optional(),
+const logExperimentSchema = type({
+	metric: type("number").describe("primary metric value"),
+	status: type("'keep'|'discard'|'crash'|'checks_failed'").describe("run outcome"),
+	description: type("string").describe("short run description"),
+	"metrics?": type({ "[string]": "number" }).describe("secondary metrics"),
+	"asi?": type({ "[string]": "unknown" }).describe("free-form structured metadata"),
+	"commit?": type("string").describe("override recorded commit hash"),
+	"justification?": type("string").describe("required when keeping a scope-deviating run"),
+	"flag_runs?": type({
+		run_id: type("number.integer").describe("run id to flag"),
+		reason: type("string").describe("why this run is suspect"),
+	})
+		.array()
+		.describe("flag earlier runs as suspect"),
 });
 
 export function createLogExperimentTool(

package/src/autoresearch/tools/run-experiment.ts

@@ -2,7 +2,7 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 import { Text } from "@oh-my-pi/pi-tui";
 import { formatBytes } from "@oh-my-pi/pi-utils";
-import { z } from "zod/v4";
+import { type } from "arktype";
 import { executeBash } from "../../exec/bash-executor";
 import type { ToolDefinition } from "../../extensibility/extensions";
 import type { Theme } from "../../modes/theme/theme";
@@ -25,8 +25,8 @@ import { openAutoresearchStorageIfExists } from "../storage";
 import type { AutoresearchToolFactoryOptions, RunDetails, RunExperimentProgressDetails } from "../types";
 import { DEFAULT_HARNESS_COMMAND } from "./init-experiment";
 
-const runExperimentSchema = z.object({
-	timeout_seconds: z.number().describe("timeout in seconds (default 600)").optional(),
+const runExperimentSchema = type({
+	"timeout_seconds?": type("number").describe("timeout in seconds (default 600)"),
 });
 
 interface ProcessExecutionResult {

package/src/autoresearch/tools/update-notes.ts

@@ -1,5 +1,5 @@
 import { Text } from "@oh-my-pi/pi-tui";
-import { z } from "zod/v4";
+import { type } from "arktype";
 import type { ToolDefinition } from "../../extensibility/extensions";
 import type { Theme } from "../../modes/theme/theme";
 import { replaceTabs, truncateToWidth } from "../../tools/render-utils";
@@ -8,9 +8,9 @@ import { buildExperimentState } from "../state";
 import { openAutoresearchStorageIfExists } from "../storage";
 import type { AutoresearchToolFactoryOptions } from "../types";
 
-const updateNotesSchema = z.object({
-	body: z.string().describe("replacement notes body"),
-	append_idea: z.string().describe("append as bullet under Ideas instead of replacing body").optional(),
+const updateNotesSchema = type({
+	body: type("string").describe("replacement notes body"),
+	"append_idea?": type("string").describe("append as bullet under Ideas instead of replacing body"),
 });
 
 interface UpdateNotesDetails {