@oh-my-pi/pi-coding-agent 14.5.12 → 14.5.14

package/src/web/search/providers/gemini.ts

@@ -8,11 +8,11 @@
 import {
 	ANTIGRAVITY_SYSTEM_INSTRUCTION,
 	extractRetryDelay,
-	getAntigravityHeaders,
+	getAntigravityUserAgent,
 	getGeminiCliHeaders,
-	refreshAntigravityToken,
-	refreshGoogleCloudToken,
 } from "@oh-my-pi/pi-ai";
+import { refreshAntigravityToken } from "@oh-my-pi/pi-ai/utils/oauth/google-antigravity";
+import { refreshGoogleCloudToken } from "@oh-my-pi/pi-ai/utils/oauth/google-gemini-cli";
 import { getAgentDbPath } from "@oh-my-pi/pi-utils";
 import { AgentStorage } from "../../../session/agent-storage";
 import type { SearchCitation, SearchResponse, SearchSource } from "../../../web/search/types";
@@ -248,7 +248,7 @@ async function callGeminiSearch(
 	usage?: { inputTokens: number; outputTokens: number; totalTokens: number };
 }> {
 	const endpoints = auth.isAntigravity ? ANTIGRAVITY_ENDPOINT_FALLBACKS : [DEFAULT_ENDPOINT];
-	const headers = auth.isAntigravity ? getAntigravityHeaders() : getGeminiCliHeaders();
+	const headers = auth.isAntigravity ? { "User-Agent": getAntigravityUserAgent() } : getGeminiCliHeaders();
 
 	const requestMetadata = auth.isAntigravity
 		? {

package/src/web/search/render.ts

@@ -22,7 +22,7 @@ import {
 } from "../../tools/render-utils";
 import { renderStatusLine, renderTreeList } from "../../tui";
 import { CachedOutputBlock } from "../../tui/output-block";
-import { getSearchProvider } from "./provider";
+import { getSearchProviderLabel } from "./provider";
 import type { SearchResponse } from "./types";
 
 const MAX_COLLAPSED_ANSWER_LINES = PREVIEW_LIMITS.COLLAPSED_LINES;
@@ -112,7 +112,7 @@ export function renderSearchResult(
 		: [];
 	const totalAnswerLines = answerLines.length;
 
-	const providerLabel = provider !== "none" ? getSearchProvider(provider).label : "None";
+	const providerLabel = provider !== "none" ? getSearchProviderLabel(provider) : "None";
 	const queryPreview = args?.query
 		? truncateToWidth(args.query, 80)
 		: searchQueries[0]

package/src/ipy/modules.ts

@@ -1,144 +0,0 @@
-import * as fs from "node:fs/promises";
-import * as path from "node:path";
-import { getAgentModulesDir, getProjectDir, getProjectModulesDir } from "@oh-my-pi/pi-utils";
-import { getExecutionCancellationError } from "./cancellation";
-
-export type PythonModuleSource = "user" | "project";
-
-export interface PythonModuleEntry {
-	path: string;
-	content: string;
-	source: PythonModuleSource;
-}
-
-export interface PythonModuleExecuteResult {
-	status: "ok" | "error";
-	cancelled: boolean;
-	timedOut?: boolean;
-	error?: { name: string; value: string; traceback: string[] };
-}
-
-export interface PythonModuleExecutor {
-	execute: (
-		code: string,
-		options?: { signal?: AbortSignal; timeoutMs?: number; silent?: boolean; storeHistory?: boolean },
-	) => Promise<PythonModuleExecuteResult>;
-}
-
-export interface DiscoverPythonModulesOptions {
-	/** Working directory for project-level modules. Default: getProjectDir() */
-	cwd?: string;
-	/** Agent directory for user-level modules. Default: from getAgentDir() */
-	agentDir?: string;
-}
-
-export interface LoadPythonModulesOptions extends DiscoverPythonModulesOptions {
-	signal?: AbortSignal;
-	timeoutMs?: number;
-	deadlineMs?: number;
-}
-
-interface ModuleCandidate {
-	name: string;
-	path: string;
-	source: PythonModuleSource;
-}
-
-async function listModuleCandidates(dir: string, source: PythonModuleSource): Promise<ModuleCandidate[]> {
-	try {
-		const entries = await fs.readdir(dir, { withFileTypes: true });
-		return entries
-			.filter(entry => entry.isFile() && entry.name.endsWith(".py"))
-			.map(entry => ({
-				name: entry.name,
-				path: path.resolve(dir, entry.name),
-				source,
-			}));
-	} catch {
-		return [];
-	}
-}
-
-async function readModuleContent(candidate: ModuleCandidate): Promise<PythonModuleEntry> {
-	try {
-		const content = await Bun.file(candidate.path).text();
-		return { path: candidate.path, content, source: candidate.source };
-	} catch (err) {
-		const message = err instanceof Error ? err.message : String(err);
-		throw new Error(`Failed to read Python module ${candidate.path}: ${message}`);
-	}
-}
-
-function createTimeoutError(message: string): Error {
-	const error = new Error(message);
-	error.name = "TimeoutError";
-	return error;
-}
-
-function requireModuleExecutionTimeoutMs(options: LoadPythonModulesOptions): number | undefined {
-	if (options.deadlineMs === undefined) {
-		return options.timeoutMs;
-	}
-
-	const remainingMs = options.deadlineMs - Date.now();
-	if (remainingMs <= 0) {
-		throw createTimeoutError("Python module loading timed out");
-	}
-
-	return remainingMs;
-}
-
-/**
- * Discover Python prelude extension modules from user and project directories.
- */
-export async function discoverPythonModules(options: DiscoverPythonModulesOptions = {}): Promise<PythonModuleEntry[]> {
-	const cwd = options.cwd ?? getProjectDir();
-
-	const userDir = getAgentModulesDir(options.agentDir);
-	const projectDir = getProjectModulesDir(cwd);
-
-	const userCandidates = await listModuleCandidates(userDir, "user");
-	const projectCandidates = await listModuleCandidates(projectDir, "project");
-
-	const byName = new Map<string, ModuleCandidate>();
-	for (const candidate of userCandidates) {
-		if (!byName.has(candidate.name)) {
-			byName.set(candidate.name, candidate);
-		}
-	}
-	for (const candidate of projectCandidates) {
-		const existing = byName.get(candidate.name);
-		if (!existing || existing.source === "user") {
-			byName.set(candidate.name, candidate);
-		}
-	}
-
-	const sorted = Array.from(byName.values()).sort((a, b) => a.name.localeCompare(b.name));
-	return Promise.all(sorted.map(candidate => readModuleContent(candidate)));
-}
-
-/**
- * Load Python prelude extension modules into an active kernel.
- */
-export async function loadPythonModules(
-	executor: PythonModuleExecutor,
-	options: LoadPythonModulesOptions = {},
-): Promise<PythonModuleEntry[]> {
-	const modules = await discoverPythonModules(options);
-	for (const module of modules) {
-		const result = await executor.execute(module.content, {
-			signal: options.signal,
-			timeoutMs: requireModuleExecutionTimeoutMs(options),
-			silent: true,
-			storeHistory: false,
-		});
-		if (result.cancelled) {
-			throw getExecutionCancellationError(result, options.signal, `Failed to load Python module ${module.path}`);
-		}
-		if (result.status === "error") {
-			const details = result.error ? `${result.error.name}: ${result.error.value}` : "unknown error";
-			throw new Error(`Failed to load Python module ${module.path}: ${details}`);
-		}
-	}
-	return modules;
-}

package/src/prompts/tools/python.md

@@ -1,57 +0,0 @@
-Runs Python cells sequentially in persistent IPython kernel.
-
-<instruction>
-Kernel persists across calls and cells; **imports, variables, and functions survive — use this.**
-
-**Work incrementally:** one logical step per cell (imports, define, test, use). Pass multiple small cells in one call. Define small reusable functions you can debug individually. You **MUST** put workflow explanations in the assistant message or cell title — never inside cell code.
-
-**On failure:** errors identify the failing cell (e.g., "Cell 3 failed"). Resubmit only the fixed cell (or fixed cell + remaining cells).
-</instruction>
-
-{{#if categories.length}}
-<prelude>
-All helpers auto-print results and return values for chaining.
-
-{{#each categories}}
-### {{name}}
-
-```
-{{#each functions}}
-{{name}}{{signature}}
-    {{docstring}}
-{{/each}}
-```
-{{/each}}
-</prelude>
-{{/if}}
-
-<output>
-User sees output like Jupyter notebook; rich displays render fully:
-- `display(JSON(data))` → interactive JSON tree
-- `display(HTML(…))` → rendered HTML
-- `display(Markdown(…))` → formatted markdown
-- `plt.show()` → inline figures
-
-**You will see object repr** (e.g., `<IPython.core.display.JSON object>`). Trust `display()`; you **MUST NOT** assume the user sees only the repr.
-</output>
-
-<caution>
-- Per-call mode uses a fresh kernel each call
-- You **MUST** use `reset: true` to clear state when session mode is active
-</caution>
-
-<critical>
-- You **MUST** use `run()` for shell commands; you **MUST NOT** use raw `subprocess`
-</critical>
-
-<examples>
-# Multiple small cells
-```python
-cells: [
-    {"title": "imports", "code": "import json\nfrom pathlib import Path"},
-    {"title": "parse helper", "code": "def parse_config(path):\n    return json.loads(Path(path).read_text())"},
-    {"title": "test helper", "code": "parse_config('config.json')"},
-    {"title": "use helper", "code": "configs = [parse_config(p) for p in Path('.').glob('*.json')]"}
-]
-```
-</examples>