@apmantza/greedysearch-pi 1.7.2 → 1.7.4

package/extractors/google-ai.mjs

@@ -16,53 +16,17 @@ import {
 	handleError,
 	outputJson,
 	parseArgs,
+	TIMING,
 	validateQuery,
+	waitForStreamComplete,
 } from "./common.mjs";
 import { dismissConsent, handleVerification } from "./consent.mjs";
 import { SELECTORS } from "./selectors.mjs";
 
 const S = SELECTORS.google;
 
-const STREAM_POLL_INTERVAL = 600;
-const STREAM_STABLE_ROUNDS = 3;
-const STREAM_TIMEOUT = 45000;
 const MIN_ANSWER_LENGTH = 50;
 
-// ============================================================================
-// Google AI-specific helpers
-// ============================================================================
-
-async function waitForGoogleStreamComplete(tab) {
-	const deadline = Date.now() + STREAM_TIMEOUT;
-	let stableCount = 0;
-	let lastLen = -1;
-
-	while (Date.now() < deadline) {
-		await new Promise((r) => setTimeout(r, STREAM_POLL_INTERVAL));
-
-		const lenStr = await cdp([
-			"eval",
-			tab,
-			`(document.querySelector('${S.answerContainer}')?.innerText?.length || 0) + ''`,
-		]).catch(() => "0");
-
-		const len = parseInt(lenStr, 10) || 0;
-
-		if (len >= MIN_ANSWER_LENGTH && len === lastLen) {
-			stableCount++;
-			if (stableCount >= STREAM_STABLE_ROUNDS) return len;
-		} else {
-			stableCount = 0;
-			lastLen = len;
-		}
-	}
-
-	if (lastLen >= MIN_ANSWER_LENGTH) return lastLen;
-	throw new Error(
-		`Google AI answer did not stabilise within ${STREAM_TIMEOUT}ms`,
-	);
-}
-
 async function extractAnswer(tab) {
 	const excludeFilter = S.sourceExclude
 		.map((e) => `!a.href.includes('${e}')`)
@@ -105,9 +69,9 @@ async function main() {
 		await cdp(["list"]);
 		const tab = await getOrOpenTab(tabPrefix);
 
-		const url = `https://www.google.com/search?q=${encodeURIComponent(query)}&udm=50`;
+		const url = `https://www.google.com/search?q=${encodeURIComponent(query)}&udm=50&hl=en`;
 		await cdp(["nav", tab, url], 35000);
-		await new Promise((r) => setTimeout(r, 1500));
+		await new Promise((r) => setTimeout(r, TIMING.postNav));
 		await dismissConsent(tab, cdp);
 
 		// If consent redirected us away, navigate back
@@ -116,7 +80,7 @@ async function main() {
 		);
 		if (!currentUrl.includes("google.com/search")) {
 			await cdp(["nav", tab, url], 35000);
-			await new Promise((r) => setTimeout(r, 1500));
+			await new Promise((r) => setTimeout(r, TIMING.postNav));
 		}
 
 		// Handle "verify you're human" — auto-click simple buttons, wait for user on hard CAPTCHA
@@ -128,10 +92,14 @@ async function main() {
 		if (verifyResult === "clicked" || verifyResult === "cleared-by-user") {
 			// Re-navigate to the search URL after verification
 			await cdp(["nav", tab, url], 35000);
-			await new Promise((r) => setTimeout(r, 1500));
+			await new Promise((r) => setTimeout(r, TIMING.postNav));
 		}
 
-		await waitForGoogleStreamComplete(tab);
+		await waitForStreamComplete(tab, {
+			timeout: 45000,
+			selector: `document.querySelector('${S.answerContainer}')`,
+			minLength: MIN_ANSWER_LENGTH,
+		});
 
 		const { answer, sources } = await extractAnswer(tab);
 		if (!answer)

package/index.ts

@@ -23,9 +23,9 @@ const __dir = dirname(fileURLToPath(import.meta.url));
 
 const ALL_ENGINES = ["perplexity", "bing", "google"] as const;
 
-function cdpAvailable(): boolean {
-	return existsSync(join(__dir, "bin", "cdp.mjs"));
-}
+function cdpAvailable(): boolean {
+	return existsSync(join(__dir, "bin", "cdp.mjs"));
+}
 
 function runSearch(
 	engine: string,
@@ -34,14 +34,14 @@ function runSearch(
 	signal?: AbortSignal,
 	onProgress?: (engine: string, status: "done" | "error") => void,
 ): Promise<Record<string, unknown>> {
-	return new Promise((resolve, reject) => {
-		const proc = spawn(
-			"node",
-			[join(__dir, "bin", "search.mjs"), engine, "--inline", ...flags, query],
-			{
-				stdio: ["ignore", "pipe", "pipe"],
-			},
-		);
+	return new Promise((resolve, reject) => {
+		const proc = spawn(
+			"node",
+			[join(__dir, "bin", "search.mjs"), engine, "--inline", ...flags, query],
+			{
+				stdio: ["ignore", "pipe", "pipe"],
+			},
+		);
 		let out = "";
 		let err = "";
 
@@ -355,13 +355,13 @@ export default function greedySearchExtension(pi: ExtensionAPI) {
 
 				const data = await new Promise<Record<string, unknown>>(
 					(resolve, reject) => {
-					const proc = spawn(
-						"node",
-						[join(__dir, "bin", "coding-task.mjs"), task, ...flags],
-						{
-							stdio: ["ignore", "pipe", "pipe"],
-						},
-					);
+					const proc = spawn(
+						"node",
+						[join(__dir, "bin", "coding-task.mjs"), task, ...flags],
+						{
+							stdio: ["ignore", "pipe", "pipe"],
+						},
+					);
 						let out = "";
 						let err = "";
 

package/package.json

@@ -1,46 +1,46 @@
-{
-	"name": "@apmantza/greedysearch-pi",
-	"version": "1.7.2",
-	"description": "Pi extension: multi-engine AI search (Perplexity, Bing Copilot, Google AI) via browser automation -- NO API KEYS needed. Extracts answers with sources, optional Gemini synthesis. Grounded AI answers from real browser interactions.",
-	"type": "module",
-	"keywords": [
-		"pi-package"
-	],
-	"repository": {
-		"type": "git",
-		"url": "git+https://github.com/apmantza/GreedySearch-pi.git"
-	},
-	"author": "Apostolos Mantzaris",
-	"license": "MIT",
-	"scripts": {
-		"test": "./test.sh",
-		"test:quick": "./test.sh quick",
-		"test:smoke": "./test.sh smoke"
-	},
-	"files": [
-		"index.ts",
-		"bin/",
-		"src/",
-		"skills/",
-		"extractors/",
-		"CHANGELOG.md",
-		"README.md"
-	],
-	"pi": {
-		"extensions": [
-			"./index.ts"
-		],
-		"skills": [
-			"./skills"
-		]
-	},
-	"dependencies": {
-		"jsdom": "^24.0.0",
-		"@mozilla/readability": "^0.5.0",
-		"turndown": "^7.1.2"
-	},
-	"peerDependencies": {
-		"@mariozechner/pi-coding-agent": "*",
-		"@sinclair/typebox": "*"
-	}
-}
+{
+	"name": "@apmantza/greedysearch-pi",
+	"version": "1.7.4",
+	"description": "Pi extension: multi-engine AI search (Perplexity, Bing Copilot, Google AI) via browser automation -- NO API KEYS needed. Extracts answers with sources, optional Gemini synthesis. Grounded AI answers from real browser interactions.",
+	"type": "module",
+	"keywords": [
+		"pi-package"
+	],
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/apmantza/GreedySearch-pi.git"
+	},
+	"author": "Apostolos Mantzaris",
+	"license": "MIT",
+	"scripts": {
+		"test": "./test.sh",
+		"test:quick": "./test.sh quick",
+		"test:smoke": "./test.sh smoke"
+	},
+	"files": [
+		"index.ts",
+		"bin/",
+		"src/",
+		"skills/",
+		"extractors/",
+		"CHANGELOG.md",
+		"README.md"
+	],
+	"pi": {
+		"extensions": [
+			"./index.ts"
+		],
+		"skills": [
+			"./skills"
+		]
+	},
+	"dependencies": {
+		"jsdom": "^24.0.0",
+		"@mozilla/readability": "^0.5.0",
+		"turndown": "^7.1.2"
+	},
+	"peerDependencies": {
+		"@mariozechner/pi-coding-agent": "*",
+		"@sinclair/typebox": "*"
+	}
+}