midscene/packages/core/tests/ai/llm-inspect.test.ts

import { AiLocateElement, AiLocateSection } from '@/ai-model';
import { getContextFromFixture } from 'tests/evaluation';
import { expect, test, vi } from 'vitest';

vi.setConfig({
  testTimeout: 60 * 1000,
});

test(
  'basic inspect',
  async () => {
    const { context } = await getContextFromFixture('todo');

    const startTime = Date.now();
    const { parseResult } = await AiLocateElement({
      context,
      targetElementDescription: 'input 输入框',
    });
    console.log('parseResult', JSON.stringify(parseResult, null, 2));
    const endTime = Date.now();
    const cost = endTime - startTime;
    expect(parseResult.elements.length).toBe(1);
  },
  {
    timeout: 1000000,
  },
);

test('locate section', async () => {
  const { context } = await getContextFromFixture('todo');
  const { rect } = await AiLocateSection({
    context,
    sectionDescription: '搜索框',
  });
  expect(rect).toBeDefined();
});

test('use quick answer', async () => {
  const { context } = await getContextFromFixture('todo');

  const startTime = Date.now();
  const { parseResult } = await AiLocateElement({
    context,
    targetElementDescription: 'never mind',
  });
  console.log('parseResult', parseResult);
  const endTime = Date.now();
  const cost = endTime - startTime;
  expect(parseResult.elements.length).toBe(1);
  expect(cost).toBeLessThan(100);
});