agentlinkops 0.5.0 → 0.6.0

package/src/citations/browser/gen/core/providers/ai-overview/lib/expand.js

@@ -0,0 +1,59 @@
+import { logger } from "../../../../../shims/utils.js";
+const AI_OVERVIEW_SOURCE_ROOT_SELECTOR = '[data-container-id="rhs-col"]';
+const AI_OVERVIEW_BUTTON_SELECTOR = '[role="button"], button';
+function shouldClickSourceExpansionButton(text, aria) {
+  const t = `${text} ${aria}`.toLowerCase();
+  if (t.includes("videos")) return false;
+  if (t.includes("more")) return true;
+  if (t.includes("all")) return true;
+  if (t.includes("expand")) return true;
+  return false;
+}
+async function expandAIOverviewSources(page) {
+  const matches = await page.evaluate(
+    ({
+      rootSelector,
+      buttonSelector
+    }) => {
+      const root = document.querySelector(rootSelector);
+      if (!(root instanceof HTMLElement)) return [];
+      return Array.from(root.querySelectorAll(buttonSelector)).map((button, index) => {
+        if (!(button instanceof HTMLElement)) {
+          return null;
+        }
+        const style = window.getComputedStyle(button);
+        const rect = button.getBoundingClientRect();
+        const visible = button.isConnected && style.display !== "none" && style.visibility !== "hidden" && style.opacity !== "0" && rect.width > 0 && rect.height > 0;
+        if (!visible) return null;
+        return {
+          index,
+          text: (button.textContent || "").trim(),
+          aria: button.getAttribute("aria-label")?.trim() || ""
+        };
+      }).filter(
+        (match) => match !== null
+      );
+    },
+    {
+      rootSelector: AI_OVERVIEW_SOURCE_ROOT_SELECTOR,
+      buttonSelector: AI_OVERVIEW_BUTTON_SELECTOR
+    }
+  );
+  for (const match of matches) {
+    if (!shouldClickSourceExpansionButton(match.text, match.aria)) {
+      continue;
+    }
+    const button = page.locator(`${AI_OVERVIEW_SOURCE_ROOT_SELECTOR} ${AI_OVERVIEW_BUTTON_SELECTOR}`).nth(match.index);
+    const visible = await button.isVisible().catch(() => false);
+    if (!visible) continue;
+    logger.debug(
+      `[ai-overview] expanding sources text="${match.text}" aria="${match.aria}"`
+    );
+    await button.click({ timeout: 5e3 }).catch(() => {
+    });
+    await page.waitForTimeout(700);
+  }
+}
+export {
+  expandAIOverviewSources
+};

package/src/citations/browser/gen/core/providers/ai-overview/lib/extractSources.js

@@ -0,0 +1,60 @@
+import { toErrorMessage } from "../../../../../shims/errors.js";
+import { logger } from "../../../../../shims/utils.js";
+import { buildSources } from "../../_shared/sourceUtils.js";
+const AI_OVERVIEW_RAW_SOURCES_DOM_EXTRACTOR = String.raw`(_helpers) => {
+	const results = [];
+	const rhsCol = document.querySelector('[data-container-id="rhs-col"]');
+	if (!rhsCol) {
+		return { rawSources: results, containerFound: false };
+	}
+
+	for (const card of Array.from(rhsCol.querySelectorAll("div[data-src-id]"))) {
+		if (!(card instanceof HTMLElement)) continue;
+
+		const link = card.querySelector('a[href^="http"]');
+		if (!(link instanceof HTMLAnchorElement)) continue;
+
+		const title =
+			link
+				.getAttribute("aria-label")
+				?.replace(/\.\s*Opens in new tab\.?$/i, "")
+				.trim() || link.href;
+		const citedText =
+			card.querySelector("[data-crb-snippet-text]")?.textContent?.trim() || "";
+
+		results.push({
+			rawHref: link.href,
+			title,
+			citedText,
+		});
+	}
+
+	return { rawSources: results, containerFound: true };
+}`;
+function normalizeAIOverviewTitle(title) {
+  return title.replace(/\s*\.?\s*opens in new tab\.?\s*$/i, "").trim();
+}
+async function extractAIOverviewSources(page) {
+  try {
+    const { rawSources, containerFound } = await page.runDomOp("raw-sources", {
+      provider: "ai-overview"
+    });
+    if (!containerFound) {
+      logger.warn("AI Overview container not found \u2014 no sources extracted");
+    }
+    const normalizedSources = rawSources.map((source) => ({
+      ...source,
+      title: normalizeAIOverviewTitle(source.title ?? "") || source.rawHref
+    }));
+    return buildSources(normalizedSources, { provider: "ai-overview" });
+  } catch (err) {
+    logger.error(
+      `Failed to extract AI Overview sources: ${toErrorMessage(err)}`
+    );
+    return [];
+  }
+}
+export {
+  AI_OVERVIEW_RAW_SOURCES_DOM_EXTRACTOR,
+  extractAIOverviewSources
+};

package/src/citations/browser/gen/core/providers/ai-overview/lib/pageLifecycle.js

@@ -0,0 +1,18 @@
+import { resetProviderPage } from "../../_shared/resetProviderPage.js";
+import {
+  assertAIOverviewPageNotBlocked,
+  dismissGoogleConsentDialog
+} from "./session.js";
+const AI_OVERVIEW_URL = "https://www.google.com/";
+async function resetAIOverviewPage(page) {
+  await resetProviderPage(page, "ai-overview", AI_OVERVIEW_URL, {
+    postNavigationHook: async (currentPage) => {
+      assertAIOverviewPageNotBlocked(currentPage);
+      await dismissGoogleConsentDialog(currentPage);
+    }
+  });
+}
+export {
+  AI_OVERVIEW_URL,
+  resetAIOverviewPage
+};

package/src/citations/browser/gen/core/providers/ai-overview/lib/session.js

@@ -0,0 +1,62 @@
+import { ExternalServiceError } from "../../../../../shims/errors.js";
+import { logger } from "../../../../../shims/utils.js";
+import { navigateWithRetry } from "../../../../lib/browser/navigate.js";
+const GOOGLE_CONSENT_SELECTOR = "button#L2AGLb, button#W0wltc, form[action*='consent.google.com'] button";
+const SEARCH_RESULTS_WAIT_MS = 8e3;
+const warmedPages = /* @__PURE__ */ new WeakSet();
+async function dismissGoogleConsentDialog(page) {
+  const consentBtn = page.locator(GOOGLE_CONSENT_SELECTOR).first();
+  const visible = await consentBtn.isVisible({ timeout: 2500 }).catch(() => false);
+  if (!visible) return;
+  await consentBtn.click({ timeout: 4e3 }).catch(() => {
+  });
+  await page.waitForTimeout(1e3);
+}
+function assertAIOverviewPageNotBlocked(page) {
+  const url = page.url();
+  if (url.includes("/sorry/")) {
+    throw new ExternalServiceError(
+      "ai-overview",
+      "Google bot detection triggered (sorry page) \u2014 proxy IP blocked",
+      429
+    );
+  }
+  if (url.includes("accounts.google.com")) {
+    throw new ExternalServiceError(
+      "ai-overview",
+      "Google redirected to login page \u2014 session cookie missing or expired",
+      401
+    );
+  }
+}
+async function ensureAIOverviewGoogleSession(page) {
+  if (warmedPages.has(page)) return;
+  logger.log("[ai-overview] warming up Google cookies");
+  await navigateWithRetry(page, "https://www.google.com/", {
+    waitUntil: "domcontentloaded",
+    timeout: 3e4
+  });
+  assertAIOverviewPageNotBlocked(page);
+  await dismissGoogleConsentDialog(page);
+  warmedPages.add(page);
+}
+async function waitForAIOverviewSearchResults(page) {
+  const deadline = Date.now() + SEARCH_RESULTS_WAIT_MS;
+  while (Date.now() < deadline) {
+    if (page.url().includes("/search?")) {
+      return;
+    }
+    assertAIOverviewPageNotBlocked(page);
+    await page.waitForTimeout(150);
+  }
+  throw new ExternalServiceError(
+    "ai-overview",
+    `Not on search results page after submission (url: ${page.url()})`
+  );
+}
+export {
+  assertAIOverviewPageNotBlocked,
+  dismissGoogleConsentDialog,
+  ensureAIOverviewGoogleSession,
+  waitForAIOverviewSearchResults
+};

package/src/citations/browser/gen/core/providers/chatgpt/index.js

@@ -0,0 +1,28 @@
+import { extractSourcesFromChatgpt } from "./lib/extractSources.js";
+import { dismissChatgptAuthModal } from "./lib/dismissAuthModal.js";
+import { extractAssistantMarkdown } from "../../../lib/input/markdown/toMarkdown.js";
+import { openSourcesPanel } from "../../../lib/input/sources/openPanel.js";
+import { findSourcesButton } from "../../../lib/input/sources/findButton.js";
+import { waitForAssistantToFinish } from "../../../lib/input/response/waitForFinish.js";
+import { CHATGPT_URL, resetChatgptPage } from "./lib/pageLifecycle.js";
+const chatgptConfig = {
+  url: CHATGPT_URL,
+  label: "ChatGPT",
+  displayName: "ChatGPT",
+  waitForResponse: (page) => waitForAssistantToFinish(page, "chatgpt"),
+  extractResponse: (page) => extractAssistantMarkdown(page, "chatgpt"),
+  beforePromptHook: (page) => dismissChatgptAuthModal(page, { waitForAppearanceMs: 500 }),
+  afterTypingHook: (page) => dismissChatgptAuthModal(page, { waitForAppearanceMs: 1500 }),
+  beforeSubmitHook: (page) => dismissChatgptAuthModal(page, { waitForAppearanceMs: 1500 }),
+  afterSubmitHook: (page) => dismissChatgptAuthModal(page, { waitForAppearanceMs: 1500 }),
+  beforeRetryHook: resetChatgptPage,
+  extractSources: async (page) => {
+    const btn = await findSourcesButton(page);
+    if (!btn) return [];
+    await openSourcesPanel(page, btn);
+    return extractSourcesFromChatgpt(page, btn);
+  }
+};
+export {
+  chatgptConfig
+};

package/src/citations/browser/gen/core/providers/chatgpt/lib/dismissAuthModal.js

@@ -0,0 +1,39 @@
+import { logger } from "../../../../../shims/utils.js";
+import {
+  canUseOsLevelInput,
+  clickLocatorLikeUser
+} from "../../../../lib/browser/humanBehavior.js";
+const AUTH_DIALOG_SELECTOR = '[role="dialog"][data-state="open"]';
+const AUTH_DIALOG_TEXT_RE = /Thanks for trying ChatGPT|Log in or sign up/i;
+const DISMISS_MODAL_TEXT_RE = /^Stay logged out$/i;
+const MODAL_POLL_INTERVAL_MS = 200;
+async function dismissChatgptAuthModal(page, options = {}) {
+  const waitForAppearanceMs = options.waitForAppearanceMs ?? 0;
+  const deadline = Date.now() + waitForAppearanceMs;
+  do {
+    const dialog = page.locator(AUTH_DIALOG_SELECTOR).filter({ hasText: AUTH_DIALOG_TEXT_RE }).first();
+    const dialogVisible = await dialog.isVisible().catch(() => false);
+    if (!dialogVisible) {
+      if (Date.now() >= deadline) return;
+      await page.waitForTimeout(MODAL_POLL_INTERVAL_MS);
+      continue;
+    }
+    const dismissTarget = dialog.getByText(DISMISS_MODAL_TEXT_RE).first();
+    const dismissVisible = await dismissTarget.isVisible().catch(() => false);
+    if (!dismissVisible) return;
+    logger.log("[chatgpt] dismissing auth modal via Stay logged out");
+    const clicked = await clickLocatorLikeUser(page, dismissTarget, {
+      timeout: 5e3
+    }).catch(() => false);
+    if (!clicked && canUseOsLevelInput(page)) {
+      return;
+    }
+    await dialog.waitFor({ state: "hidden", timeout: 5e3 }).catch(() => {
+    });
+    await page.waitForTimeout(300);
+    return;
+  } while (Date.now() <= deadline);
+}
+export {
+  dismissChatgptAuthModal
+};

package/src/citations/browser/gen/core/providers/chatgpt/lib/extractSources.js

@@ -0,0 +1,66 @@
+import {
+  buildSources,
+  clickButtonViaDispatch
+} from "../../_shared/sourceUtils.js";
+const CHATGPT_RAW_SOURCES_DOM_EXTRACTOR = String.raw`(_helpers) => {
+	const results = [];
+
+	const getLeafTexts = (anchor) => {
+		const texts = [];
+
+		for (const el of anchor.querySelectorAll("*")) {
+			const text = (el.textContent || "").trim();
+			if (!text) continue;
+
+			if (
+				Array.from(el.children).some(
+					(child) => (child.textContent || "").trim().length > 0,
+				)
+			) {
+				continue;
+			}
+
+			texts.push(text);
+		}
+
+		return Array.from(new Set(texts));
+	};
+
+	for (const anchor of Array.from(
+		document.querySelectorAll(
+			'ul li > a[target="_blank"][rel*="noopener"][href^="http"]',
+		),
+	)) {
+		if (!(anchor instanceof HTMLAnchorElement)) continue;
+
+		const rawHref = anchor.href.replace(/#.*$/, "");
+		if (!rawHref) continue;
+
+		const texts = getLeafTexts(anchor);
+		if (texts.length === 0) continue;
+		const sorted = [...texts].sort((a, b) => a.length - b.length);
+		const citedText = sorted[sorted.length - 1] || "";
+		const title =
+			sorted.length >= 2 ? sorted[sorted.length - 2] || "" : sorted[0] || "";
+
+		results.push({
+			rawHref,
+			title,
+			citedText,
+		});
+	}
+
+	return results;
+}`;
+async function extractSourcesFromChatgpt(page, sourcesButton) {
+  const rawSources = await page.runDomOp("raw-sources", {
+    provider: "chatgpt"
+  });
+  if (!await clickButtonViaDispatch(page, sourcesButton)) return [];
+  await page.waitForTimeout(300);
+  return buildSources(rawSources, { provider: "chatgpt" });
+}
+export {
+  CHATGPT_RAW_SOURCES_DOM_EXTRACTOR,
+  extractSourcesFromChatgpt
+};

package/src/citations/browser/gen/core/providers/chatgpt/lib/pageLifecycle.js

@@ -0,0 +1,11 @@
+import { resetProviderPage } from "../../_shared/resetProviderPage.js";
+import { dismissChatgptAuthModal } from "./dismissAuthModal.js";
+const CHATGPT_URL = "https://chatgpt.com/";
+async function resetChatgptPage(page) {
+  await resetProviderPage(page, "chatgpt", CHATGPT_URL);
+  await dismissChatgptAuthModal(page, { waitForAppearanceMs: 1e3 });
+}
+export {
+  CHATGPT_URL,
+  resetChatgptPage
+};

package/src/citations/browser/gen/core/providers/claude/index.js

@@ -0,0 +1,16 @@
+import { extractAssistantMarkdown } from "../../../lib/input/markdown/toMarkdown.js";
+import { waitForAssistantToFinish } from "../../../lib/input/response/waitForFinish.js";
+import { extractSourcesFromClaude } from "./lib/extractSources.js";
+import { CLAUDE_URL, resetClaudePage } from "./lib/pageLifecycle.js";
+const claudeConfig = {
+  url: CLAUDE_URL,
+  label: "Claude",
+  displayName: "Claude",
+  beforeRetryHook: resetClaudePage,
+  waitForResponse: (page) => waitForAssistantToFinish(page, "claude"),
+  extractResponse: (page) => extractAssistantMarkdown(page, "claude"),
+  extractSources: (page) => extractSourcesFromClaude(page)
+};
+export {
+  claudeConfig
+};

package/src/citations/browser/gen/core/providers/claude/lib/extractSources.js

@@ -0,0 +1,54 @@
+import { toErrorMessage } from "../../../../../shims/errors.js";
+import { logger, PROVIDER_MODEL_RESPONSE_SELECTORS } from "../../../../../shims/utils.js";
+import { buildSources } from "../../_shared/sourceUtils.js";
+const CLAUDE_RAW_SOURCES_DOM_EXTRACTOR = String.raw`(
+	{
+		getCachedRawSources,
+		setCachedRawSources,
+		findLatestResponseElement,
+		extractClaudeRawSourcesFromResponseElement,
+	},
+	selectors,
+) => {
+	const cached = getCachedRawSources("claude");
+	if (cached) return cached;
+
+	const responseEl = findLatestResponseElement(selectors)?.element;
+	if (!responseEl) return [];
+
+	const rawSources = extractClaudeRawSourcesFromResponseElement(responseEl);
+	setCachedRawSources("claude", rawSources);
+	return rawSources;
+}`;
+async function fetchTitle(url) {
+  try {
+    const res = await fetch(url);
+    const html = await res.text();
+    const match = html.match(/<title>(.*?)<\/title>/i);
+    return match?.[1]?.trim() || null;
+  } catch {
+    return null;
+  }
+}
+async function extractSourcesFromClaude(page) {
+  try {
+    const rawSources = await page.runDomOp("raw-sources", {
+      provider: "claude",
+      selectors: PROVIDER_MODEL_RESPONSE_SELECTORS.claude || []
+    });
+    const rawSourcesWithFetchedTitles = await Promise.all(
+      rawSources.map(async (source) => ({
+        ...source,
+        title: await fetchTitle(source.rawHref) || source.title
+      }))
+    );
+    return buildSources(rawSourcesWithFetchedTitles, { provider: "claude" });
+  } catch (error) {
+    logger.error(`Failed to extract Claude sources: ${toErrorMessage(error)}`);
+    return [];
+  }
+}
+export {
+  CLAUDE_RAW_SOURCES_DOM_EXTRACTOR,
+  extractSourcesFromClaude
+};

package/src/citations/browser/gen/core/providers/claude/lib/pageLifecycle.js

@@ -0,0 +1,9 @@
+import { resetProviderPage } from "../../_shared/resetProviderPage.js";
+const CLAUDE_URL = "https://claude.ai/new";
+async function resetClaudePage(page) {
+  await resetProviderPage(page, "claude", CLAUDE_URL);
+}
+export {
+  CLAUDE_URL,
+  resetClaudePage
+};

package/src/citations/browser/gen/core/providers/gemini/index.js

@@ -0,0 +1,40 @@
+import { extractAssistantMarkdown } from "../../../lib/input/markdown/toMarkdown.js";
+import { openSourcesPanel } from "../../../lib/input/sources/openPanel.js";
+import { findSourcesButton } from "../../../lib/input/sources/findButton.js";
+import { waitForAssistantToFinish } from "../../../lib/input/response/waitForFinish.js";
+import { extractSourcesFromGemini } from "./lib/extractSources.js";
+import { GEMINI_URL, resetGeminiPage } from "./lib/pageLifecycle.js";
+import {
+  handleGeminiConsentPage,
+  waitForGeminiConversationUrl
+} from "./lib/session.js";
+const geminiConfig = {
+  url: GEMINI_URL,
+  label: "Gemini",
+  displayName: "Gemini",
+  // Detect consent pages before attempting to locate the editor.
+  // consent.google.com has no Gemini composer, so without this check
+  // waitForEditorReady times out and misclassifies it as "no_editor".
+  postNavigationHook: async (page) => {
+    await handleGeminiConsentPage(page);
+  },
+  beforePromptHook: async (page) => {
+    await handleGeminiConsentPage(page);
+  },
+  checkSubmitSuccess: async (page, { preSubmitUrl }) => waitForGeminiConversationUrl(page, preSubmitUrl),
+  waitForResponse: (page) => waitForAssistantToFinish(page, "gemini"),
+  extractResponse: (page) => extractAssistantMarkdown(page, "gemini"),
+  beforeRetryHook: resetGeminiPage,
+  // No reset between prompts — session is reused in the same conversation.
+  // Navigating back to gemini.google.com on each prompt adds unnecessary
+  // round-trips and increases detection surface.
+  extractSources: async (page) => {
+    const btn = await findSourcesButton(page);
+    if (!btn) return [];
+    await openSourcesPanel(page, btn);
+    return extractSourcesFromGemini(page, btn);
+  }
+};
+export {
+  geminiConfig
+};

package/src/citations/browser/gen/core/providers/gemini/lib/extractSources.js

@@ -0,0 +1,65 @@
+import {
+  buildSources,
+  clickButtonViaDispatch
+} from "../../_shared/sourceUtils.js";
+const GEMINI_RAW_SOURCES_DOM_EXTRACTOR = String.raw`(_helpers) => {
+	const results = [];
+
+	const getLeafTexts = (anchor) => {
+		const texts = [];
+
+		for (const el of anchor.querySelectorAll("*")) {
+			const text = (el.textContent || "").trim();
+			if (!text) continue;
+
+			if (
+				Array.from(el.children).some(
+					(child) => (child.textContent || "").trim().length > 0,
+				)
+			) {
+				continue;
+			}
+
+			texts.push(text);
+		}
+
+		return Array.from(new Set(texts));
+	};
+
+	for (const anchor of Array.from(
+		document.querySelectorAll(
+			'inline-source-card a[href^="http"]',
+		),
+	)) {
+		if (!(anchor instanceof HTMLAnchorElement)) continue;
+
+		const rawHref = anchor.href.replace(/#.*$/, "");
+		if (!rawHref) continue;
+		const texts = getLeafTexts(anchor);
+		if (texts.length === 0) continue;
+		const sorted = [...texts].sort((a, b) => a.length - b.length);
+		const citedText = sorted[sorted.length - 1] || "";
+		const title =
+			sorted.length >= 2 ? sorted[sorted.length - 2] || "" : sorted[0] || "";
+
+		results.push({
+			rawHref,
+			title,
+			citedText,
+		});
+	}
+
+	return results;
+}`;
+async function extractSourcesFromGemini(page, sourcesButton) {
+  const rawSources = await page.runDomOp("raw-sources", {
+    provider: "gemini"
+  });
+  if (!await clickButtonViaDispatch(page, sourcesButton)) return [];
+  await page.waitForTimeout(300);
+  return buildSources(rawSources, { provider: "gemini" });
+}
+export {
+  GEMINI_RAW_SOURCES_DOM_EXTRACTOR,
+  extractSourcesFromGemini
+};

package/src/citations/browser/gen/core/providers/gemini/lib/pageLifecycle.js

@@ -0,0 +1,14 @@
+import { resetProviderPage } from "../../_shared/resetProviderPage.js";
+import { handleGeminiConsentPage } from "./session.js";
+const GEMINI_URL = "https://gemini.google.com/";
+async function resetGeminiPage(page) {
+  await resetProviderPage(page, "gemini", GEMINI_URL, {
+    postNavigationHook: async (currentPage) => {
+      await handleGeminiConsentPage(currentPage);
+    }
+  });
+}
+export {
+  GEMINI_URL,
+  resetGeminiPage
+};

package/src/citations/browser/gen/core/providers/gemini/lib/session.js

@@ -0,0 +1,45 @@
+import { ExternalServiceError } from "../../../../../shims/errors.js";
+import { clickLocatorLikeUser } from "../../../../lib/browser/humanBehavior.js";
+const GEMINI_CONSENT_SELECTOR = "button#L2AGLb, button#W0wltc, form[action*='consent.google.com'] button";
+async function handleGeminiConsentPage(page) {
+  const url = page.url();
+  if (!url.includes("consent.google.com")) return;
+  const consentBtn = page.locator(GEMINI_CONSENT_SELECTOR).first();
+  const visible = await consentBtn.isVisible({ timeout: 3e3 }).catch(() => false);
+  if (visible) {
+    await clickLocatorLikeUser(page, consentBtn, { timeout: 4e3 }).catch(() => {
+    });
+    await page.waitForTimeout(1e3);
+    if (!page.url().includes("consent.google.com")) return;
+  }
+  throw new ExternalServiceError(
+    "gemini",
+    "Google consent page not dismissible \u2014 proxy IP requires Google consent",
+    429
+  );
+}
+function isGeminiAppUrl(rawUrl) {
+  try {
+    const url = new URL(rawUrl);
+    return url.hostname === "gemini.google.com" && url.pathname.startsWith("/app/") && url.pathname.length > "/app/".length;
+  } catch {
+    return false;
+  }
+}
+async function waitForGeminiConversationUrl(page, preSubmitUrl) {
+  if (isGeminiAppUrl(preSubmitUrl)) {
+    return void 0;
+  }
+  const deadline = Date.now() + 4e3;
+  while (Date.now() < deadline) {
+    if (isGeminiAppUrl(await page.getUrl().catch(() => page.url()))) {
+      return true;
+    }
+    await page.waitForTimeout(100);
+  }
+  return false;
+}
+export {
+  handleGeminiConsentPage,
+  waitForGeminiConversationUrl
+};

package/src/citations/browser/gen/core/providers/index.js

@@ -0,0 +1,15 @@
+import { aiOverviewConfig } from "./ai-overview/index.js";
+import { chatgptConfig } from "./chatgpt/index.js";
+import { claudeConfig } from "./claude/index.js";
+import { geminiConfig } from "./gemini/index.js";
+import { perplexityConfig } from "./perplexity/index.js";
+const PROVIDER_CONFIGS = {
+  gemini: geminiConfig,
+  chatgpt: chatgptConfig,
+  perplexity: perplexityConfig,
+  claude: claudeConfig,
+  "ai-overview": aiOverviewConfig
+};
+export {
+  PROVIDER_CONFIGS
+};

package/src/citations/browser/gen/core/providers/perplexity/index.js

@@ -0,0 +1,31 @@
+import { extractSourcesFromPerplexity } from "./lib/extractSources.js";
+import { extractAssistantMarkdown } from "../../../lib/input/markdown/toMarkdown.js";
+import { openSourcesPanel } from "../../../lib/input/sources/openPanel.js";
+import { findSourcesButton } from "../../../lib/input/sources/findButton.js";
+import { waitForAssistantToFinish } from "../../../lib/input/response/waitForFinish.js";
+import {
+  PERPLEXITY_URL,
+  perplexityPostNavigationHook,
+  resetPerplexityPage,
+  waitForPerplexitySearchUrl
+} from "./lib/navigation.js";
+const perplexityConfig = {
+  url: PERPLEXITY_URL,
+  label: "Perplexity",
+  displayName: "Perplexity",
+  beforeRetryHook: resetPerplexityPage,
+  betweenPromptsHook: resetPerplexityPage,
+  checkSubmitSuccess: async (page, { preSubmitUrl }) => waitForPerplexitySearchUrl(page, preSubmitUrl),
+  waitForResponse: (page) => waitForAssistantToFinish(page, "perplexity"),
+  extractResponse: (page) => extractAssistantMarkdown(page, "perplexity"),
+  postNavigationHook: perplexityPostNavigationHook,
+  extractSources: async (page) => {
+    const btn = await findSourcesButton(page);
+    if (!btn) return [];
+    await openSourcesPanel(page, btn);
+    return extractSourcesFromPerplexity(page, btn);
+  }
+};
+export {
+  perplexityConfig
+};