@accomplish_ai/agent-core 0.0.0-pr-353-20260205225158

package/mcp-tools/dev-browser-mcp/src/snapshot/priority.ts

@@ -0,0 +1,84 @@
+export interface TruncatableElement {
+  ref: string;
+  role: string;
+  name: string;
+  inViewport: boolean;
+  [key: string]: unknown;
+}
+
+export interface TruncateOptions {
+  maxElements?: number;
+}
+
+export interface TruncateResult<T extends TruncatableElement> {
+  elements: T[];
+  totalElements: number;
+  includedElements: number;
+  truncated: boolean;
+}
+
+export const ROLE_PRIORITIES: Record<string, number> = {
+  button: 100,
+  textbox: 95,
+  searchbox: 95,
+  checkbox: 90,
+  radio: 90,
+  switch: 90,
+  combobox: 85,
+  listbox: 85,
+  slider: 85,
+  spinbutton: 85,
+  link: 80,
+  tab: 75,
+  menuitem: 70,
+  menuitemcheckbox: 70,
+  menuitemradio: 70,
+  option: 70,
+  navigation: 60,
+  menu: 60,
+  tablist: 55,
+  form: 50,
+  dialog: 50,
+  alertdialog: 50,
+};
+
+const VIEWPORT_BONUS = 50;
+const DEFAULT_PRIORITY = 50;
+
+export function getElementPriority(role: string, inViewport: boolean): number {
+  const basePriority = ROLE_PRIORITIES[role] ?? DEFAULT_PRIORITY;
+  return inViewport ? basePriority + VIEWPORT_BONUS : basePriority;
+}
+
+export function truncateElements<T extends TruncatableElement>(
+  elements: T[],
+  options: TruncateOptions
+): TruncateResult<T> {
+  const maxElements = options.maxElements ?? 300;
+  const totalElements = elements.length;
+
+  if (totalElements <= maxElements) {
+    return {
+      elements,
+      totalElements,
+      includedElements: totalElements,
+      truncated: false,
+    };
+  }
+
+  const scored = elements.map(element => ({
+    element,
+    score: getElementPriority(element.role, element.inViewport),
+  }));
+
+  scored.sort((a, b) => b.score - a.score);
+
+  const truncatedElements = scored.slice(0, maxElements).map(s => s.element);
+
+  return {
+    elements: truncatedElements,
+    totalElements,
+    includedElements: maxElements,
+    truncated: true,
+  };
+}

package/mcp-tools/dev-browser-mcp/src/snapshot/tokens.test.ts

@@ -0,0 +1,64 @@
+import { describe, it, expect } from 'vitest';
+import { estimateTokens, estimateElementTokens } from './tokens';
+
+describe('token estimation', () => {
+  describe('estimateElementTokens', () => {
+    it('should estimate basic element at ~15 tokens', () => {
+      const tokens = estimateElementTokens({
+        role: 'button',
+        name: 'Submit',
+        ref: 'e1',
+      });
+      // role (1) + name (2) + ref (2) + yaml overhead (5) + attributes (2-5)
+      expect(tokens).toBeGreaterThanOrEqual(10);
+      expect(tokens).toBeLessThanOrEqual(20);
+    });
+
+    it('should cap long names at 50 token contribution', () => {
+      const shortName = estimateElementTokens({
+        role: 'button',
+        name: 'OK',
+        ref: 'e1',
+      });
+      const longName = estimateElementTokens({
+        role: 'button',
+        name: 'A'.repeat(1000), // Very long name
+        ref: 'e2',
+      });
+      // Difference should be at most 50 (capped)
+      expect(longName - shortName).toBeLessThanOrEqual(50);
+    });
+
+    it('should add tokens for extra attributes', () => {
+      const basic = estimateElementTokens({
+        role: 'checkbox',
+        name: 'Accept',
+        ref: 'e1',
+      });
+      const withAttrs = estimateElementTokens({
+        role: 'checkbox',
+        name: 'Accept',
+        ref: 'e1',
+        checked: true,
+        disabled: true,
+      });
+      expect(withAttrs).toBeGreaterThan(basic);
+    });
+  });
+
+  describe('estimateTokens', () => {
+    it('should estimate YAML string tokens', () => {
+      const yaml = `- button "Submit" [ref=e1]
+- textbox "Email" [ref=e2]
+- link "Home" [ref=e3]`;
+      const tokens = estimateTokens(yaml);
+      // ~15 tokens per element * 3 = ~45
+      expect(tokens).toBeGreaterThanOrEqual(30);
+      expect(tokens).toBeLessThanOrEqual(60);
+    });
+
+    it('should handle empty string', () => {
+      expect(estimateTokens('')).toBe(0);
+    });
+  });
+});

package/mcp-tools/dev-browser-mcp/src/snapshot/tokens.ts

@@ -0,0 +1,36 @@
+import type { SnapshotElement } from './types';
+
+const CHARS_PER_TOKEN = 2;
+const YAML_OVERHEAD = 5;
+const ATTRIBUTE_TOKENS = 2;
+const MAX_NAME_TOKENS = 50;
+
+export function estimateElementTokens(element: Partial<SnapshotElement>): number {
+  let tokens = YAML_OVERHEAD;
+
+  tokens += Math.ceil((element.role?.length ?? 0) / CHARS_PER_TOKEN);
+
+  const nameLength = element.name?.length ?? 0;
+  const nameTokens = Math.ceil(nameLength / CHARS_PER_TOKEN);
+  tokens += Math.min(nameTokens, MAX_NAME_TOKENS);
+
+  tokens += 2;
+
+  if (element.checked !== undefined) tokens += ATTRIBUTE_TOKENS;
+  if (element.disabled !== undefined) tokens += ATTRIBUTE_TOKENS;
+  if (element.expanded !== undefined) tokens += ATTRIBUTE_TOKENS;
+  if (element.selected !== undefined) tokens += ATTRIBUTE_TOKENS;
+  if (element.pressed !== undefined) tokens += ATTRIBUTE_TOKENS;
+
+  if (element.value) {
+    const valueTokens = Math.ceil(element.value.length / CHARS_PER_TOKEN);
+    tokens += Math.min(valueTokens, MAX_NAME_TOKENS);
+  }
+
+  return tokens;
+}
+
+export function estimateTokens(yaml: string): number {
+  if (!yaml) return 0;
+  return Math.ceil(yaml.length / CHARS_PER_TOKEN);
+}

package/mcp-tools/dev-browser-mcp/src/snapshot/types.ts

@@ -0,0 +1,89 @@
+export interface SnapshotElement {
+  ref: string;
+  role: string;
+  name: string;
+  value?: string;
+  checked?: boolean | 'mixed';
+  disabled?: boolean;
+  expanded?: boolean;
+  selected?: boolean;
+  level?: number;
+  pressed?: boolean | 'mixed';
+  url?: string;
+  placeholder?: string;
+}
+
+export interface ElementPriority {
+  ref: string;
+  score: number;
+  inViewport: boolean;
+}
+
+export interface SnapshotMetadata {
+  totalElements: number;
+  includedElements: number;
+  truncated: boolean;
+  estimatedTokens: number;
+}
+
+export interface ParsedSnapshot {
+  url: string;
+  title: string;
+  timestamp: number;
+  elements: Map<string, SnapshotElement>;
+  rawYaml: string;
+  metadata?: SnapshotMetadata;
+}
+
+export interface SnapshotOptions {
+  fullSnapshot?: boolean;
+  interactiveOnly?: boolean;
+  maxElements?: number;
+  viewportOnly?: boolean;
+  maxTokens?: number;
+  includeHistory?: boolean;
+}
+
+export const DEFAULT_SNAPSHOT_OPTIONS: Required<SnapshotOptions> = {
+  fullSnapshot: false,
+  interactiveOnly: true,
+  maxElements: 300,
+  viewportOnly: false,
+  maxTokens: 8000,
+  includeHistory: true,
+};
+
+export interface ElementChange {
+  ref: string;
+  element: SnapshotElement;
+  previousValue?: string;
+  previousChecked?: boolean | 'mixed';
+  previousDisabled?: boolean;
+  previousExpanded?: boolean;
+  previousSelected?: boolean;
+  changeType: 'added' | 'modified' | 'removed';
+}
+
+export interface SnapshotDiff {
+  unchangedRefs: string[];
+  changes: ElementChange[];
+  addedRefs: string[];
+  removedRefs: string[];
+}
+
+export type SnapshotResult =
+  | { type: 'full'; content: string }
+  | { type: 'diff'; content: string; unchangedRefs: string[] };
+
+export interface SessionHistoryEntry {
+  url: string;
+  title: string;
+  timestamp: number;
+  actionsTaken: string[];
+}
+
+export interface SessionSummary {
+  history: string;
+  pagesVisited: number;
+  navigationPatternHash?: string;
+}

package/mcp-tools/dev-browser-mcp/tsconfig.json

@@ -0,0 +1,15 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "esModuleInterop": true,
+    "strict": true,
+    "skipLibCheck": true,
+    "resolveJsonModule": true,
+    "isolatedModules": true,
+    "noEmit": true
+  },
+  "include": ["src/**/*.ts"],
+  "exclude": ["node_modules"]
+}