pi-finder-subagent 1.0.0

package/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

package/README.md

@@ -0,0 +1,71 @@
+# pi-finder
+
+Read-only codebase scout subagent package for [pi](https://github.com/badlogic/pi-mono).
+
+## Installation
+
+From npm (after publish):
+
+```bash
+pi install npm:pi-finder-subagent
+```
+
+From git:
+
+```bash
+pi install git:github.com/default-anton/pi-finder
+```
+
+Or run without installing:
+
+```bash
+pi -e npm:pi-finder-subagent
+# or
+pi -e git:github.com/default-anton/pi-finder
+```
+
+## What it does
+
+- Registers a `finder` tool.
+- Runs a dedicated subagent session with strict turn budget enforcement.
+- Uses only `bash` + `read` in the subagent.
+- Enforces read-only scouting behavior (rg/fd/ls + targeted reads).
+- Returns structured Markdown output (`Summary`, `Locations`, optional `Evidence`/`Searched`/`Next steps`).
+- Selects subagent model dynamically using shared package `pi-subagent-model-selection`.
+- Emits compact selection diagnostics (`authMode`, `authSource`, `reason`) in tool details.
+
+## Tool interface
+
+```ts
+finder({
+  query: string,
+})
+```
+
+## Model selection policy
+
+Finder delegates model selection to `pi-subagent-model-selection`, which uses the same policy as pi-librarian:
+
+- OAuth mode:
+  1. `google-antigravity/gemini-3-flash`
+  2. fallback strategy
+- API-key mode:
+  1. `google-vertex` Gemini 3 Flash (`gemini-3-flash*`)
+  2. `google` Gemini 3 Flash (`gemini-3-flash*`)
+  3. fallback strategy
+
+Fallback strategy:
+1. Gemini 3 Flash on current provider
+2. Claude Haiku 4.5 on current provider
+3. Current model with `thinkingLevel: low`
+
+## Development
+
+```bash
+npm run test:model-selection
+npm run pack:check
+```
+
+## License
+
+Apache-2.0

package/extensions/finder-core.ts

@@ -0,0 +1,142 @@
+import events from "node:events";
+
+import { Type } from "@sinclair/typebox";
+
+export const DEFAULT_MAX_TURNS = 10;
+export const MAX_TOOL_CALLS_TO_KEEP = 80;
+
+const DEFAULT_EVENTTARGET_MAX_LISTENERS = 100;
+const EVENTTARGET_MAX_LISTENERS_STATE_KEY = Symbol.for("pi.eventTargetMaxListenersState");
+
+type EventTargetMaxListenersState = { depth: number; savedDefault?: number };
+
+export type FinderStatus = "running" | "done" | "error" | "aborted";
+
+export type ToolCall = {
+  id: string;
+  name: string;
+  args: unknown;
+  startedAt: number;
+  endedAt?: number;
+  isError?: boolean;
+};
+
+export interface FinderRunDetails {
+  status: FinderStatus;
+  query: string;
+  turns: number;
+  toolCalls: ToolCall[];
+  summaryText?: string;
+  error?: string;
+  startedAt: number;
+  endedAt?: number;
+}
+
+export interface SubagentSelectionInfo {
+  authMode: "oauth" | "api-key";
+  authSource: "runtime" | "api_key" | "oauth" | "env" | "fallback" | "none";
+  reason: string;
+}
+
+export interface FinderDetails {
+  status: FinderStatus;
+  workspace?: string;
+  subagentProvider?: string;
+  subagentModelId?: string;
+  subagentSelection?: SubagentSelectionInfo;
+  runs: FinderRunDetails[];
+}
+
+export const FinderParams = Type.Object({
+  query: Type.String({
+    description: [
+      "Describe what to find in the codebase. Include: (1) specific goal, (2) keywords/identifiers to search, (3) desired output type (paths, line ranges, directory structure), (4) what counts as 'found'.",
+      "Finder uses rg/fd/ls and read — do not request grep or find.",
+      "Example: 'Find where user authentication is implemented. Look for functions named login, auth, or authenticate. Return paths with line ranges for the main entry point and token handling. Search in src/auth and src/api directories.'",
+    ].join("\n"),
+  }),
+});
+
+function getEventTargetMaxListenersState(): EventTargetMaxListenersState {
+  const g = globalThis as any;
+  if (!g[EVENTTARGET_MAX_LISTENERS_STATE_KEY]) g[EVENTTARGET_MAX_LISTENERS_STATE_KEY] = { depth: 0 };
+  return g[EVENTTARGET_MAX_LISTENERS_STATE_KEY] as EventTargetMaxListenersState;
+}
+
+export function bumpDefaultEventTargetMaxListeners(): () => void {
+  const state = getEventTargetMaxListenersState();
+
+  const raw = process.env.PI_EVENTTARGET_MAX_LISTENERS ?? process.env.PI_ABORT_MAX_LISTENERS;
+  const desired = raw !== undefined ? Number(raw) : DEFAULT_EVENTTARGET_MAX_LISTENERS;
+  if (!Number.isFinite(desired) || desired < 0) return () => { };
+
+  if (state.depth === 0) state.savedDefault = events.defaultMaxListeners;
+  state.depth += 1;
+
+  if (events.defaultMaxListeners < desired) events.setMaxListeners(desired);
+
+  return () => {
+    state.depth = Math.max(0, state.depth - 1);
+    if (state.depth !== 0) return;
+    if (state.savedDefault === undefined) return;
+
+    events.setMaxListeners(state.savedDefault);
+    state.savedDefault = undefined;
+  };
+}
+
+export function shorten(text: string, max: number): string {
+  if (text.length <= max) return text;
+  return `${text.slice(0, max)}…`;
+}
+
+export function getLastAssistantText(messages: any[]): string {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const msg = messages[i];
+    if (msg?.role !== "assistant") continue;
+    const parts = msg.content;
+    if (!Array.isArray(parts)) continue;
+
+    const blocks: string[] = [];
+    for (const part of parts) {
+      if (part?.type === "text" && typeof part.text === "string") blocks.push(part.text);
+    }
+
+    if (blocks.length > 0) return blocks.join("");
+  }
+  return "";
+}
+
+export function computeOverallStatus(runs: FinderRunDetails[]): FinderStatus {
+  if (runs.some((r) => r.status === "running")) return "running";
+  if (runs.some((r) => r.status === "error")) return "error";
+  if (runs.every((r) => r.status === "aborted")) return "aborted";
+  return "done";
+}
+
+export function renderCombinedMarkdown(runs: FinderRunDetails[]): string {
+  const r = runs[0];
+  return (r.summaryText ?? (r.status === "running" ? "(searching...)" : "(no output)")).trim();
+}
+
+export function formatToolCall(call: ToolCall): string {
+  const args = call.args && typeof call.args === "object" ? (call.args as Record<string, any>) : undefined;
+
+  if (call.name === "read") {
+    const p = typeof args?.path === "string" ? args.path : "";
+    const offset = typeof args?.offset === "number" ? args.offset : undefined;
+    const limit = typeof args?.limit === "number" ? args.limit : undefined;
+    const range = offset || limit ? `:${offset ?? 1}${limit ? `-${(offset ?? 1) + limit - 1}` : ""}` : "";
+    return `read ${p}${range}`;
+  }
+
+  if (call.name === "bash") {
+    const command = typeof args?.command === "string" ? args.command : "";
+    const timeout = typeof args?.timeout === "number" ? args.timeout : undefined;
+    const normalized = command.replace(/\s+/g, " ").trim();
+    const suffix = timeout ? ` (timeout ${timeout}s)` : "";
+    return `bash ${shorten(normalized, 120)}${suffix}`.trimEnd();
+  }
+
+  return call.name;
+}

package/extensions/finder-prompts.md.ts

@@ -0,0 +1,46 @@
+export function buildFinderSystemPrompt(maxTurns: number): string {
+  return `You are Finder, an evidence-first repository scout.
+You operate in a read-only environment and may only use the provided tools (bash/read).
+
+Use bash for repository scouting (e.g., rg, fd, ls). Never use grep (use rg) or find (use fd).
+Use read to open specific file ranges for line-level citations.
+
+Turn budget: at most ${maxTurns} turns total (hard cap).
+Tool use is disabled on the final turn; reserve final turn for synthesis.
+
+Non-negotiable constraints:
+- Do not modify files, propose patches, or refactor.
+- No side effects: do not run commands that modify files or repository state.
+- Do not guess. Every claim must be backed by tool output.
+- Keep snippets short (about 5-15 lines).
+
+How to work:
+1) Translate the query into a concrete search checklist.
+2) Search broadly with bash using rg + fd, then narrow.
+3) Use read with offset+limit for line-level evidence.
+4) Stop as soon as evidence is sufficient.
+
+Citations:
+- For file-content claims, cite path:lineStart-lineEnd from explicit read ranges.
+- For path-only claims, cite paths from bash outputs.
+
+Output format (Markdown, exact order):
+## Summary
+(1-3 sentences)
+## Locations
+- \`path\` or \`path:lineStart-lineEnd\` — what is here and why it matters
+## Evidence (optional)
+(snippets, each preceded by a citation)
+## Searched (only if incomplete / not found)
+(patterns and directories tried)
+## Next steps (optional)
+(what to check next if ambiguous)`.trim();
+}
+
+export function buildFinderUserPrompt(query: string): string {
+  return `Task: locate and cite exact code locations that answer the query.
+Follow the system instructions for tools, citations, and output format.
+
+Query:
+${query.trim()}`.trim();
+}

package/extensions/index.ts

@@ -0,0 +1,403 @@
+import type { ExtensionAPI, ExtensionContext, ExtensionFactory } from "@mariozechner/pi-coding-agent";
+import {
+  DefaultResourceLoader,
+  SessionManager,
+  createAgentSession,
+  createBashTool,
+  createReadTool,
+  getMarkdownTheme,
+} from "@mariozechner/pi-coding-agent";
+import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
+import { getSmallModelFromProvider } from "pi-subagent-model-selection";
+
+import {
+  DEFAULT_MAX_TURNS,
+  FinderParams,
+  MAX_TOOL_CALLS_TO_KEEP,
+  bumpDefaultEventTargetMaxListeners,
+  computeOverallStatus,
+  formatToolCall,
+  getLastAssistantText,
+  renderCombinedMarkdown,
+  shorten,
+  type FinderDetails,
+  type FinderRunDetails,
+} from "./finder-core";
+import { buildFinderSystemPrompt, buildFinderUserPrompt } from "./finder-prompts.md.ts";
+
+function createTurnBudgetExtension(maxTurns: number): ExtensionFactory {
+  return (pi) => {
+    let turnIndex = 0;
+
+    pi.on("turn_start", async (event) => {
+      turnIndex = event.turnIndex;
+    });
+
+    pi.on("tool_call", async () => {
+      if (turnIndex < maxTurns - 1) return undefined;
+
+      const humanTurn = Math.min(turnIndex + 1, maxTurns);
+      return {
+        block: true,
+        reason: `Tool use is disabled on the final turn (turn ${humanTurn}/${maxTurns}). Provide your final answer now without calling tools.`,
+      };
+    });
+
+    pi.on("tool_result", async (event) => {
+      const remainingAfter = Math.max(0, maxTurns - (turnIndex + 1));
+      const humanTurn = Math.min(turnIndex + 1, maxTurns);
+      const budgetLine = `[turn budget] turn ${humanTurn}/${maxTurns}; remaining after this turn: ${remainingAfter}`;
+
+      return {
+        content: [...(event.content ?? []), { type: "text", text: `\n\n${budgetLine}` }],
+      };
+    });
+  };
+}
+
+export default function finderExtension(pi: ExtensionAPI) {
+  pi.registerTool({
+    name: "finder",
+    label: "Finder",
+    description:
+      "Read-only codebase scout: searches repositories using rg/fd/ls and read, returns structured Markdown with Summary, Locations (path:lineStart-lineEnd), Evidence, and Searched sections.",
+    parameters: FinderParams,
+
+    async execute(_toolCallId, params, signal, onUpdate, ctx: ExtensionContext) {
+      const restoreMaxListeners = bumpDefaultEventTargetMaxListeners();
+      let abortListenerAdded = false;
+      let onAbort: (() => void) | undefined;
+      try {
+        const maxTurns = DEFAULT_MAX_TURNS;
+        const query = typeof (params as any).query === "string" ? ((params as any).query as string).trim() : "";
+
+        if (!query) {
+          const error = "Invalid parameters: expected `query` to be a non-empty string.";
+          return {
+            content: [{ type: "text", text: error }],
+            details: { status: "error", runs: [] } satisfies FinderDetails,
+            isError: true,
+          };
+        }
+
+        const runs: FinderRunDetails[] = [
+          {
+            status: "running",
+            query,
+            turns: 0,
+            toolCalls: [],
+            startedAt: Date.now(),
+          },
+        ];
+
+        const modelRegistry = ctx.modelRegistry;
+        const subModelSelection = getSmallModelFromProvider(modelRegistry, ctx.model);
+
+        if (!subModelSelection) {
+          const error = "No models available. Configure credentials (e.g. /login or auth.json) and try again.";
+          runs[0].status = "error";
+          runs[0].error = error;
+          runs[0].summaryText = error;
+          runs[0].endedAt = Date.now();
+          return {
+            content: [{ type: "text", text: error }],
+            details: {
+              status: "error",
+              workspace: ctx.cwd,
+              runs,
+            } satisfies FinderDetails,
+            isError: true,
+          };
+        }
+
+        const subModel = subModelSelection.model;
+        const subagentThinkingLevel = subModelSelection.thinkingLevel;
+        const subagentSelection = {
+          authMode: subModelSelection.authMode,
+          authSource: subModelSelection.authSource,
+          reason: subModelSelection.reason,
+        } as const;
+
+        let lastUpdate = 0;
+        const emitAll = (force = false) => {
+          const now = Date.now();
+          if (!force && now - lastUpdate < 120) return;
+          lastUpdate = now;
+
+          const status = computeOverallStatus(runs);
+          const text = renderCombinedMarkdown(runs);
+
+          onUpdate?.({
+            content: [{ type: "text", text }],
+            details: {
+              status,
+              workspace: ctx.cwd,
+              subagentProvider: subModel.provider,
+              subagentModelId: subModel.id,
+              subagentSelection,
+              runs,
+            } satisfies FinderDetails,
+          });
+        };
+
+        emitAll(true);
+
+        const systemPrompt = buildFinderSystemPrompt(maxTurns);
+
+        let toolAborted = false;
+        const activeSessions = new Set<{ abort: () => Promise<void> }>();
+
+        const markAllAborted = () => {
+          for (const run of runs) {
+            if (run.status !== "running") continue;
+            run.status = "aborted";
+            run.summaryText = run.summaryText ?? "Aborted";
+            run.endedAt = Date.now();
+          }
+        };
+
+        const abortAll = async () => {
+          if (toolAborted) return;
+          toolAborted = true;
+          markAllAborted();
+          emitAll(true);
+          await Promise.allSettled([...activeSessions].map((session) => session.abort()));
+        };
+
+        onAbort = () => void abortAll();
+
+        if (signal?.aborted) {
+          await abortAll();
+          const status = computeOverallStatus(runs);
+          const text = renderCombinedMarkdown(runs);
+          return {
+            content: [{ type: "text", text }],
+            details: {
+              status,
+              workspace: ctx.cwd,
+              runs,
+              subagentProvider: subModel.provider,
+              subagentModelId: subModel.id,
+              subagentSelection,
+            } satisfies FinderDetails,
+            isError: status === "error",
+          };
+        }
+
+        if (signal) {
+          signal.addEventListener("abort", onAbort);
+          abortListenerAdded = true;
+        }
+
+        const wasAborted = () => toolAborted || signal?.aborted;
+        const run = runs[0];
+
+        let session: any;
+        let unsubscribe: (() => void) | undefined;
+
+        try {
+          const resourceLoader = new DefaultResourceLoader({
+            noExtensions: true,
+            additionalExtensionPaths: ["npm:pi-subdir-context"],
+            noSkills: true,
+            noPromptTemplates: true,
+            noThemes: true,
+            extensionFactories: [createTurnBudgetExtension(maxTurns)],
+            systemPromptOverride: () => systemPrompt,
+            skillsOverride: () => ({ skills: [], diagnostics: [] }),
+          });
+          await resourceLoader.reload();
+
+          run.status = "running";
+          run.turns = 0;
+          run.toolCalls = [];
+          run.startedAt = Date.now();
+          run.endedAt = undefined;
+          run.error = undefined;
+          run.summaryText = undefined;
+
+          const { session: createdSession } = await createAgentSession({
+            cwd: ctx.cwd,
+            modelRegistry,
+            resourceLoader,
+            sessionManager: SessionManager.inMemory(ctx.cwd),
+            model: subModel,
+            thinkingLevel: subagentThinkingLevel,
+            tools: [createReadTool(ctx.cwd), createBashTool(ctx.cwd)],
+          });
+
+          session = createdSession;
+          activeSessions.add(session as any);
+
+          unsubscribe = session.subscribe((event) => {
+            switch (event.type) {
+              case "turn_end": {
+                run.turns += 1;
+                emitAll();
+                break;
+              }
+              case "tool_execution_start": {
+                run.toolCalls.push({
+                  id: event.toolCallId,
+                  name: event.toolName,
+                  args: event.args,
+                  startedAt: Date.now(),
+                });
+                if (run.toolCalls.length > MAX_TOOL_CALLS_TO_KEEP) {
+                  run.toolCalls.splice(0, run.toolCalls.length - MAX_TOOL_CALLS_TO_KEEP);
+                }
+                emitAll(true);
+                break;
+              }
+              case "tool_execution_end": {
+                const call = run.toolCalls.find((c) => c.id === event.toolCallId);
+                if (call) {
+                  call.endedAt = Date.now();
+                  call.isError = event.isError;
+                }
+                emitAll(true);
+                break;
+              }
+            }
+          });
+
+          await session.prompt(buildFinderUserPrompt(query), {
+            expandPromptTemplates: false,
+          });
+          run.summaryText = getLastAssistantText(session.state.messages as any[]).trim();
+          if (!run.summaryText) run.summaryText = wasAborted() ? "Aborted" : "(no output)";
+          run.status = wasAborted() ? "aborted" : "done";
+          run.endedAt = Date.now();
+          emitAll(true);
+        } catch (error) {
+          const message = wasAborted() ? "Aborted" : error instanceof Error ? error.message : String(error);
+          run.status = wasAborted() ? "aborted" : "error";
+          run.error = wasAborted() ? undefined : message;
+          run.summaryText = message;
+          run.endedAt = Date.now();
+          emitAll(true);
+        } finally {
+          if (session) activeSessions.delete(session as any);
+          unsubscribe?.();
+          session?.dispose();
+        }
+
+        const status = computeOverallStatus(runs);
+        const text = renderCombinedMarkdown(runs);
+
+        return {
+          content: [{ type: "text", text }],
+          details: {
+            status,
+            workspace: ctx.cwd,
+            runs,
+            subagentProvider: subModel.provider,
+            subagentModelId: subModel.id,
+            subagentSelection,
+          } satisfies FinderDetails,
+          isError: status === "error",
+        };
+      } finally {
+        if (signal && abortListenerAdded && onAbort) signal.removeEventListener("abort", onAbort);
+        restoreMaxListeners();
+      }
+    },
+
+    renderCall(args, theme) {
+      const query = typeof (args as any)?.query === "string" ? ((args as any).query as string).trim() : "";
+      const preview = shorten(query.replace(/\s+/g, " ").trim(), 70);
+
+      const title = theme.fg("toolTitle", theme.bold("finder"));
+      const text = title + (preview ? `\n${theme.fg("muted", preview)}` : "");
+      return new Text(text, 0, 0);
+    },
+
+    renderResult(result, { expanded, isPartial }, theme) {
+      const details = result.details as FinderDetails | undefined;
+      if (!details) {
+        const text = result.content[0];
+        return new Text(text?.type === "text" ? text.text : "(no output)", 0, 0);
+      }
+
+      const status = isPartial ? "running" : details.status;
+      const icon =
+        status === "done"
+          ? theme.fg("success", "✓")
+          : status === "error"
+            ? theme.fg("error", "✗")
+            : status === "aborted"
+              ? theme.fg("warning", "◼")
+              : theme.fg("warning", "⏳");
+
+      const run = details.runs[0];
+      const totalToolCalls = run?.toolCalls.length ?? 0;
+      const totalTurns = run?.turns ?? 0;
+
+      const selectionSummary = details.subagentSelection
+        ? `${details.subagentSelection.authMode}/${details.subagentSelection.authSource}`
+        : "?/?";
+
+      const header =
+        icon +
+        " " +
+        theme.fg("toolTitle", theme.bold("finder ")) +
+        theme.fg(
+          "dim",
+          `${details.subagentProvider ?? "?"}/${details.subagentModelId ?? "?"} • ${selectionSummary} • ${totalTurns} turns • ${totalToolCalls} tool call${totalToolCalls === 1 ? "" : "s"}`,
+        );
+
+      const workspaceLine = details.workspace
+        ? `${theme.fg("muted", "workspace: ")}${theme.fg("toolOutput", details.workspace)}`
+        : theme.fg("muted", "workspace: (none)");
+
+      const selectionReasonLine = details.subagentSelection
+        ? `${theme.fg("muted", "selection: ")}${theme.fg("toolOutput", details.subagentSelection.reason)}`
+        : undefined;
+
+      let toolsText = "";
+      if (run && run.toolCalls.length > 0) {
+        const calls = expanded ? run.toolCalls : run.toolCalls.slice(-6);
+        const lines: string[] = [theme.fg("muted", "Tools:")];
+        for (const call of calls) {
+          const callIcon = call.isError ? theme.fg("error", "✗") : theme.fg("dim", "→");
+          lines.push(`${callIcon} ${theme.fg("toolOutput", formatToolCall(call))}`);
+        }
+        if (!expanded && run.toolCalls.length > 6) lines.push(theme.fg("muted", "(Ctrl+O to expand)"));
+        toolsText = lines.join("\n");
+      }
+
+      if (status === "running") {
+        let text = `${header}\n${workspaceLine}`;
+        if (expanded && selectionReasonLine) text += `\n${selectionReasonLine}`;
+        if (toolsText) text += `\n\n${toolsText}`;
+        text += `\n\n${theme.fg("muted", "Searching codebase…")}`;
+        return new Text(text, 0, 0);
+      }
+
+      const mdTheme = getMarkdownTheme();
+      const combined =
+        (result.content[0]?.type === "text" ? result.content[0].text : renderCombinedMarkdown(details.runs)).trim() ||
+        "(no output)";
+
+      if (!expanded) {
+        const previewLines = combined.split("\n").slice(0, 18).join("\n");
+        let text = `${header}\n${workspaceLine}\n\n${theme.fg("toolOutput", previewLines)}`;
+        if (combined.split("\n").length > 18) text += `\n${theme.fg("muted", "(Ctrl+O to expand)")}`;
+        if (toolsText) text += `\n\n${toolsText}`;
+        return new Text(text, 0, 0);
+      }
+
+      const container = new Container();
+      container.addChild(new Text(header, 0, 0));
+      container.addChild(new Text(workspaceLine, 0, 0));
+      if (selectionReasonLine) container.addChild(new Text(selectionReasonLine, 0, 0));
+      if (toolsText) {
+        container.addChild(new Spacer(1));
+        container.addChild(new Text(toolsText, 0, 0));
+      }
+      container.addChild(new Spacer(1));
+      container.addChild(new Markdown(combined, 0, 0, mdTheme));
+      return container;
+    },
+  });
+}

package/extensions/model-selection.test.ts

@@ -0,0 +1,68 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+
+import type { AuthSource } from "pi-subagent-model-selection";
+import { getSmallModelFromProvider } from "pi-subagent-model-selection";
+
+type Model = { provider: string; id: string };
+
+type RegistryOptions = {
+  authSource?: AuthSource;
+  usingOAuth?: boolean;
+};
+
+function makeRegistry(models: Model[], options: RegistryOptions = {}) {
+  return {
+    getAvailable() {
+      return models;
+    },
+    getAuthSource() {
+      return options.authSource ?? "none";
+    },
+    isUsingOAuth() {
+      return options.usingOAuth ?? false;
+    },
+  };
+}
+
+test("oauth mode selects antigravity gemini flash", () => {
+  const selected = getSmallModelFromProvider(
+    makeRegistry([{ provider: "google-antigravity", id: "gemini-3-flash" }], { authSource: "oauth", usingOAuth: true }),
+    { provider: "openai", id: "gpt-5.1-codex" },
+  );
+
+  assert.ok(selected);
+  assert.equal(selected.model.provider, "google-antigravity");
+  assert.equal(selected.authMode, "oauth");
+  assert.equal(selected.authSource, "oauth");
+});
+
+test("api-key mode prefers vertex gemini flash", () => {
+  const selected = getSmallModelFromProvider(
+    makeRegistry(
+      [
+        { provider: "google-vertex", id: "gemini-3-flash-preview" },
+        { provider: "google", id: "gemini-3-flash" },
+      ],
+      { authSource: "api_key" },
+    ),
+    { provider: "anthropic", id: "claude-opus-4-6" },
+  );
+
+  assert.ok(selected);
+  assert.equal(selected.model.provider, "google-vertex");
+  assert.equal(selected.authMode, "api-key");
+  assert.equal(selected.authSource, "api_key");
+});
+
+test("falls back to current model if no preferred small models exist", () => {
+  const selected = getSmallModelFromProvider(
+    makeRegistry([{ provider: "openai", id: "gpt-5.1-codex" }], { authSource: "runtime" }),
+    { provider: "openai", id: "gpt-5.1-codex" },
+  );
+
+  assert.ok(selected);
+  assert.equal(selected.model.provider, "openai");
+  assert.equal(selected.model.id, "gpt-5.1-codex");
+  assert.equal(selected.thinkingLevel, "low");
+});

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "pi-finder-subagent",
+  "version": "1.0.0",
+  "description": "Read-only codebase scout subagent package for pi coding agent",
+  "type": "module",
+  "keywords": [
+    "pi-package",
+    "pi-extension",
+    "finder",
+    "subagent",
+    "code-search"
+  ],
+  "license": "Apache-2.0",
+  "author": "Anton Kuzmenko",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/default-anton/pi-finder.git"
+  },
+  "bugs": {
+    "url": "https://github.com/default-anton/pi-finder/issues"
+  },
+  "homepage": "https://github.com/default-anton/pi-finder#readme",
+  "pi": {
+    "extensions": [
+      "./extensions/index.ts"
+    ]
+  },
+  "scripts": {
+    "test:model-selection": "node --test extensions/model-selection.test.ts",
+    "pack:check": "npm pack --dry-run",
+    "prepublishOnly": "npm run test:model-selection"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": "*",
+    "@mariozechner/pi-tui": "*",
+    "@sinclair/typebox": "*"
+  },
+  "dependencies": {
+    "pi-subagent-model-selection": "^0.1.1"
+  },
+  "files": [
+    "extensions/",
+    "README.md",
+    "LICENSE"
+  ]
+}