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

52 lines
1.3 KiB
TypeScript
Raw Permalink Normal View History

import { AiLocateElement, AiLocateSection } from '@/ai-model';
import { getContextFromFixture } from 'tests/evaluation';
2025-04-30 16:07:31 +08:00
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);
});