moving

packages/agent/tests/entity-extractor.test.ts

@@ -0,0 +1,27 @@
+import { describe, it, expect } from 'vitest';
+import { extractEntities } from '../src/entity-extractor.js';
+
+describe('Entity Extractor', () => {
+  it('extracts person names', () => {
+    const entities = extractEntities('I had a meeting with Alice Johnson about the Q3 roadmap.');
+    const people = entities.filter(e => e.type === 'person');
+    expect(people.length).toBeGreaterThanOrEqual(1);
+    expect(people.some(p => p.name.includes('Alice'))).toBe(true);
+  });
+
+  it('extracts technology references', () => {
+    const entities = extractEntities('Switch from PostgreSQL to SQLite for the local database.');
+    const techs = entities.filter(e => e.type === 'technology');
+    expect(techs.length).toBeGreaterThanOrEqual(2);
+  });
+
+  it('returns empty array for trivial input', () => {
+    expect(extractEntities('Hi')).toEqual([]);
+  });
+
+  it('deduplicates within same extraction', () => {
+    const entities = extractEntities('Alice Johnson talked to Alice Johnson about Alice Johnson.');
+    const alices = entities.filter(e => e.name.includes('Alice'));
+    expect(alices).toHaveLength(1);
+  });
+});