@fitchmultz/pi-subagents 0.50.0 → 0.52.0

package/src/agent/host.ts

@@ -3,6 +3,42 @@ import * as path from "node:path";
 import { findPackageJSON } from "node:module";
 import { fileURLToPath, pathToFileURL } from "node:url";
 import type * as SDK from "@earendil-works/pi-coding-agent";
+import type { Api, Model } from "@earendil-works/pi-ai";
+import type { ThinkingLevel } from "@earendil-works/pi-agent-core";
+import { readIntercomLineage } from "../pi-intercom/identity.ts";
+
+export { errorMessage as errorText } from "../shared/unknown.ts";
+
+/** Fork-only exact native working state; official Pi lacks it, so its children stay resident. */
+interface LocalWorkingState {
+  readonly header: { readonly id: string };
+  readonly sessionFile?: string;
+  readonly leafId: string | null;
+  readonly entries: readonly SDK.SessionEntry[];
+}
+type NativeWorkingState = typeof SDK extends { readWorkingSession: (file: string) => infer S }
+  ? S
+  : never;
+export type WorkingState = [NativeWorkingState] extends [never]
+  ? LocalWorkingState
+  : NativeWorkingState;
+// Working sessions stay optional even when the selected graph declares them, so both hosts
+// keep one checked path. The native resolver is optional until official Pi exports it.
+type HostSDK = Omit<
+  typeof SDK,
+  "readWorkingSession" | "writeWorkingSession" | "openWorkingSession"
+> & {
+  readWorkingSession?: (file: string) => WorkingState;
+  writeWorkingSession?: (file: string, state: WorkingState) => void;
+  openWorkingSession?: (state: WorkingState) => SDK.SessionManager;
+  resolveModelReference?: (
+    request: { text: string; provider: string; thinkingSuffix: "strict"; fuzzy: false },
+    models: readonly Model<Api>[],
+    usable: (provider: string) => boolean,
+  ) =>
+    | { status: "resolved"; model: Model<Api>; thinkingLevel?: ThinkingLevel }
+    | { status: "ambiguous" | "not-found" };
+};
 
 function isPackage(root: string): boolean {
   const file = path.join(root, "package.json");
@@ -49,10 +85,10 @@ export function packageRoot(): string {
   }
   return path.dirname(manifest);
 }
-export async function loadSDK(root: string): Promise<typeof SDK> {
+export async function loadSDK(root: string): Promise<HostSDK> {
   // Selected host SDK, not a second package-local runtime hidden behind Jiti aliases.
   // oxlint-disable-next-line typescript/no-unsafe-assignment
-  const sdk: typeof SDK = await import(pathToFileURL(path.join(root, "dist/index.js")).href);
+  const sdk: HostSDK = await import(pathToFileURL(path.join(root, "dist/index.js")).href);
   return sdk;
 }
 export function runtimeEntry(): string {
@@ -61,22 +97,5 @@ export function runtimeEntry(): string {
   );
 }
 export function isChild(): boolean {
-  const root = process.env.PI_INTERCOM_ROOT_SESSION_ID;
-  const parent = process.env.PI_INTERCOM_PARENT_SESSION_ID;
-  if (root === undefined && parent === undefined) {
-    return false;
-  }
-  if (
-    root === undefined ||
-    parent === undefined ||
-    root.length === 0 ||
-    parent.length === 0 ||
-    root !== parent
-  ) {
-    throw new Error("Malformed child identity; delegation is disabled.");
-  }
-  return true;
-}
-export function errorText(error: unknown): string {
-  return error instanceof Error ? error.message : JSON.stringify(error);
+  return readIntercomLineage() !== undefined;
 }