feat: enrich custom companies (#3311)

Author: AmarTrebinjac

.infra/Pulumi.prod.yaml

@@ -196,3 +196,7 @@ config:
       secure: AAABAOiIbsLPyafEKkSV7RxVIOPz9GrNR7jMRmSlHcRyLaX7pqSgVqR5J8N7yL/GT4kgyvxRvA9hQPysJ+VWc5XlLdKGoKP4YH5kwMoPFpnGBJbQVUXDHTjNq6AHjdzhGXG8+SrVK3D3TnTNgPeEHWQo1lwfBQp6nGVUo3cySwtNpnNcnYfG3/CQMd0aVL+/YfiB60rRyP/ngEOqNf/tY/yjzElbBMryJnbC693eElVOUxqvyiPUs9dMCs1X0oNnsMk4f09oKLAoA8ksYdpid6Tz9zdu4mlWCD4TT9M8LvDWJu5UczFUZQt5M8UuyzM2nshgYwFiuhaer9bRQCELknV/vx02NSUpGSYJBvRUhhpGdMNhNBTlxDX4FTXKx1vvSaFfvBzXJDh+S7cvI0yzabWto+u3vuXEzKyYJVgFUNX+AgAcMbEhTuetFGUCjfcUtRBhdk8Q/A64U2KScEqvxV9k+NyySQ64raoqR91lgn1ORMgbnZA5ukAvRIQ8/1eYXPkNVz3RG55WPxtSnC9vD1wIyngfSu3QhY7y7FWe3oeuKOhZNZHVejW+cljRZnGU9nAA3uk9tGl+k121HSMKiMWBulA9orPSW76XknpgHdlF9mutfvvBT0KEuZx8+DRkcuEGPKtJtSPmfYg8mLdywOwKog6ImoF3drDGnnkEJLyO0F83sPW1+5gPKqWh7BNSECKK3PHkbjsqlBL2LoOocjNKmX61KoXvBE3CqBLfK7nHsTZtI8OhInlaf5nxYir4Uvw+SCIe1FXmhA46joySKUSI2tyOVS2pyPjbyXxHZSmt/7HWJqCOkrQvHsnk7mEmQJ3CK75gOOiV5ylJBmWdKqW46EnlDoNLCumta2zOaFUovqJAhQNnOXLv0btu/SVUp/ovkFpsKqs1VllNDRJZ/O4B9Uh2UEOZLakxIUyDnCv3Oy7nxUxmLZEoLXzgP2tjQbpxDkYt1ffltMf7vlDrKY/DRRMa9sdPVeechW/eaiCZNilR3SLtrQ3bYMLhbcSE6+URhHA/cHlSebRu2os61BI4WDe2zW6DEjCeT7zEijVRCvNgqrygDIajB00QVfYnuoc6rZEQV0ElqA6bj7mALGGpfj9cNlpbjSbu95lOtBR8CIlW/TzwHKoz0HOBbS3sLcWIm7w9JLcYkQsdqdJ+J3ou8EcNpK5jTMRBp7KLm1EtOjjKrrQJ6RqBo5j9xVs/8jJXL/79XNzS7JCt+bm/U+euc9oFalo7pw4BYf8XopS6947fG2bXX30ragdzBm2og2KKdSTdslr3eHVLWOMp3WtFYFyJZPc1gO67itNmp501NrIu/8EEHR3E9bKlRkFO8tmEqTcQo2k1LtZ555RlVfrPusneYmAKSZSo8NGNWK/V5L7Hmey9lPL+fG5qM97ZwV5oDJTMTgKcFpJf8Dq19MmIrE67xBqRnbhnD4d69IqBozICUPtiJDJ4OIx+AXczLsyNzxvH593TA2uzkwIV956vKBBiCdmKNZbRJXGPfaOz+BDBTWS7cnLbJzm1J8sNESJ7mZMZbFiGh2YpTzA8zYZvE1SQvuawOz+r5J6/i1CQIvAqrTo8/rauNm2WZXNnL8Hm1Awo3e9j7xUSJekQFiP4A4TgEU7TWYMrHfGM9UsE6VCTx17q2/DY61k1FrEhkETC9K/FHakchgnMJY/3BhdwN+h1nHBYe/KZKoOkTXBkXisP1m6baZMO9AZl4qWiVHKrlfymEwn7aNQFlgEGyjM02lZS6mJ+3JL71vGBbbXO9rfZU8NNeb9Fh+fQ7m1wkMbGyd7OWZXqa3wnx8EGH4gw9Ro8zK6uIgA8SYpaSAFKP4drNX3eryRm0NdEIJPqauecE2/Q2Z6jxkw1aQH8LLpEQvTxanDJivFb6gqzdE8ois9/wzYTVwp9sM/GrhHxJSiVADNWLrjKcTjK62o/VUU6Cpzh7HDNPyqkI1zyRwh0wxjq2Y+9ClNUETvHeekOetASngEFEjH1jvZB2zqs87KkZV7KdqPHV2+CSwHC1TVV7tloHOoGlq9MujsGpFDPmBbcFqfRGR5SNuNjie79WiM9nJsnh7o4yFHrD8OKKVCKFfvYvekWwFACOYHiYwQGRih3TRf44obENKEdbyNs6ysIT2TYnG/HTip5fAQeOjO0BTR8aKT1QpL94Hpv83lUnDiik59qoerINK+xPj6XiYMuaBJ8Qp/voP5J6wN3xw9ynx/jUNPDdgon7hEBvCdlyOEnNtXVRUAFkYqqtUlQmXDveAx2biA19kru1p3U
   gcp:project: devkit-prod
   gcp:region: us-central1
+  api:anthropicApiKey:
+    secure: AAABAHEzcSWbWl8xVDhOLgPCopvQnihNX6MIzyG/JbaYeGA3p1qrJ3UEQhOvBg7kMoIbx9u+0CRC102IldakBAlxx0l9l9kCDSE/JqqfCfjiLT5mjdLISRE5q2dQz+MaqqVzqm0MzaIQQkWBmOxeJAxRxQ/+/dWdla8RMWKG+Q7PnrH8vS8PeNKASRQ=
+  api:anthropicApiUrl: https://api.anthropic.com/v1/messages
+  api:anthropicVersion: 2023-06-01

__tests__/integrations/anthropic/client.ts

@@ -0,0 +1,230 @@
+import nock from 'nock';
+import { AnthropicClient } from '../../../src/integrations/anthropic/client';
+import {
+  GarmrService,
+  GarmrNoopService,
+} from '../../../src/integrations/garmr';
+import type { AnthropicResponse } from '../../../src/integrations/anthropic/types';
+
+describe('AnthropicClient', () => {
+  const API_KEY = 'test-api-key';
+  let client: AnthropicClient;
+
+  beforeAll(() => {
+    process.env.ANTHROPIC_API_URL = 'https://api.anthropic.com/v1/messages';
+    process.env.ANTHROPIC_VERSION = '2023-06-01';
+  });
+
+  beforeEach(() => {
+    nock.cleanAll();
+    client = new AnthropicClient(API_KEY);
+  });
+
+  afterEach(() => {
+    nock.cleanAll();
+  });
+
+  describe('constructor', () => {
+    it('should use GarmrNoopService by default', () => {
+      const defaultClient = new AnthropicClient(API_KEY);
+      expect(defaultClient.garmr).toBeInstanceOf(GarmrNoopService);
+    });
+
+    it('should use provided Garmr service', () => {
+      const customGarmr = new GarmrService({
+        service: 'test',
+        breakerOpts: {
+          halfOpenAfter: 10000,
+          threshold: 0.2,
+          duration: 30000,
+        },
+        retryOpts: {
+          maxAttempts: 3,
+          backoff: 500,
+        },
+      });
+
+      const customClient = new AnthropicClient(API_KEY, { garmr: customGarmr });
+      expect(customClient.garmr).toBe(customGarmr);
+    });
+  });
+
+  describe('createMessage', () => {
+    const mockResponse: AnthropicResponse = {
+      content: [
+        {
+          input: {},
+        },
+      ],
+    };
+
+    it('should send request with correct headers', async () => {
+      const scope = nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .matchHeader('Content-Type', 'application/json')
+        .matchHeader('x-api-key', API_KEY)
+        .matchHeader('anthropic-version', '2023-06-01')
+        .reply(200, mockResponse);
+
+      await client.createMessage({
+        model: 'claude-sonnet-4-5-20250929',
+        max_tokens: 1024,
+        system: 'You are a helpful assistant.',
+        messages: [{ role: 'user', content: 'Hello' }],
+      });
+
+      expect(scope.isDone()).toBe(true);
+    });
+
+    it('should return parsed response on success', async () => {
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(200, mockResponse);
+
+      const result = await client.createMessage({
+        model: 'claude-sonnet-4-5-20250929',
+        max_tokens: 1024,
+        system: 'You are a helpful assistant.',
+        messages: [{ role: 'user', content: 'Hello' }],
+      });
+
+      expect(result).toEqual(mockResponse);
+    });
+
+    it('should handle tool use response', async () => {
+      const toolUseResponse: AnthropicResponse = {
+        content: [
+          {
+            input: {
+              englishName: 'Google',
+              nativeName: 'Google',
+              domain: 'google.com',
+            },
+          },
+        ],
+      };
+
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(200, toolUseResponse);
+
+      const result = await client.createMessage({
+        model: 'claude-sonnet-4-5-20250929',
+        max_tokens: 1024,
+        system: 'You are a helpful assistant.',
+        messages: [{ role: 'user', content: 'Google' }],
+        tools: [
+          {
+            name: 'organization_info',
+            description: 'Gets information about the given organization',
+            input_schema: {
+              type: 'object',
+              properties: {
+                englishName: { type: 'string' },
+                nativeName: { type: 'string' },
+                domain: { type: 'string' },
+              },
+              required: ['englishName', 'nativeName', 'domain'],
+            },
+          },
+        ],
+        tool_choice: {
+          type: 'tool',
+          name: 'organization_info',
+        },
+      });
+
+      expect(result.content[0].input).toEqual({
+        englishName: 'Google',
+        nativeName: 'Google',
+        domain: 'google.com',
+      });
+    });
+
+    it('should throw error on API failure', async () => {
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(400, 'Bad Request');
+
+      await expect(
+        client.createMessage({
+          model: 'claude-sonnet-4-5-20250929',
+          max_tokens: 1024,
+          system: 'You are a helpful assistant.',
+          messages: [{ role: 'user', content: 'Hello' }],
+        }),
+      ).rejects.toThrow('Anthropic API error: 400 Bad Request - Bad Request');
+    });
+
+    it('should throw error on 401 unauthorized', async () => {
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(401, 'Unauthorized');
+
+      await expect(
+        client.createMessage({
+          model: 'claude-sonnet-4-5-20250929',
+          max_tokens: 1024,
+          system: 'You are a helpful assistant.',
+          messages: [{ role: 'user', content: 'Hello' }],
+        }),
+      ).rejects.toThrow('Anthropic API error: 401 Unauthorized - Unauthorized');
+    });
+
+    it('should throw error on 429 rate limit', async () => {
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(429, 'Rate limit exceeded');
+
+      await expect(
+        client.createMessage({
+          model: 'claude-sonnet-4-5-20250929',
+          max_tokens: 1024,
+          system: 'You are a helpful assistant.',
+          messages: [{ role: 'user', content: 'Hello' }],
+        }),
+      ).rejects.toThrow(
+        'Anthropic API error: 429 Too Many Requests - Rate limit exceeded',
+      );
+    });
+
+    it('should throw error on 500 server error', async () => {
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(500, 'Internal Server Error');
+
+      await expect(
+        client.createMessage({
+          model: 'claude-sonnet-4-5-20250929',
+          max_tokens: 1024,
+          system: 'You are a helpful assistant.',
+          messages: [{ role: 'user', content: 'Hello' }],
+        }),
+      ).rejects.toThrow(
+        'Anthropic API error: 500 Internal Server Error - Internal Server Error',
+      );
+    });
+
+    it('should include error details in error message', async () => {
+      const errorResponse = JSON.stringify({
+        error: {
+          type: 'invalid_request_error',
+          message: 'max_tokens must be greater than 0',
+        },
+      });
+
+      nock('https://api.anthropic.com')
+        .post('/v1/messages')
+        .reply(400, errorResponse);
+
+      await expect(
+        client.createMessage({
+          model: 'claude-sonnet-4-5-20250929',
+          max_tokens: 0,
+          system: 'You are a helpful assistant.',
+          messages: [{ role: 'user', content: 'Hello' }],
+        }),
+      ).rejects.toThrow(errorResponse);
+    });
+  });
+});

__tests__/schema/autocompletes.ts

@@ -861,6 +861,22 @@ describe('query autocompleteCompany', () => {
         domains: ['amazon.com'],
         type: CompanyType.Company,
       },
+      {
+        id: 'samsung',
+        name: 'Samsung Electronics',
+        altName: '삼성전자',
+        image: 'https://example.com/samsung.png',
+        domains: ['samsung.com'],
+        type: CompanyType.Company,
+      },
+      {
+        id: 'toyota',
+        name: 'Toyota Motor Corporation',
+        altName: 'トヨタ自動車',
+        image: 'https://example.com/toyota.png',
+        domains: ['toyota.com'],
+        type: CompanyType.Company,
+      },
       {
         id: 'mit',
         name: 'Massachusetts Institute of Technology',
@@ -889,6 +905,14 @@ describe('query autocompleteCompany', () => {
         domains: ['berkeley.edu'],
         type: CompanyType.School,
       },
+      {
+        id: 'todai',
+        name: 'The University of Tokyo',
+        altName: '東京大学',
+        image: 'https://example.com/todai.png',
+        domains: ['u-tokyo.ac.jp'],
+        type: CompanyType.School,
+      },
     ]);
   });
 
@@ -939,6 +963,11 @@ describe('query autocompleteCompany', () => {
         name: 'Stanford University',
         image: 'https://example.com/stanford.png',
       },
+      {
+        id: 'todai',
+        name: 'The University of Tokyo',
+        image: 'https://example.com/todai.png',
+      },
       {
         id: 'berkeley',
         name: 'University of California, Berkeley',
@@ -976,6 +1005,16 @@ describe('query autocompleteCompany', () => {
         name: 'Microsoft Corporation',
         image: 'https://example.com/microsoft.png',
       },
+      {
+        id: 'samsung',
+        name: 'Samsung Electronics',
+        image: 'https://example.com/samsung.png',
+      },
+      {
+        id: 'toyota',
+        name: 'Toyota Motor Corporation',
+        image: 'https://example.com/toyota.png',
+      },
     ]);
   });
 
@@ -1050,11 +1089,21 @@ describe('query autocompleteCompany', () => {
         name: 'Microsoft Corporation',
         image: 'https://example.com/microsoft.png',
       },
+      {
+        id: 'samsung',
+        name: 'Samsung Electronics',
+        image: 'https://example.com/samsung.png',
+      },
       {
         id: 'stanford',
         name: 'Stanford University',
         image: 'https://example.com/stanford.png',
       },
+      {
+        id: 'toyota',
+        name: 'Toyota Motor Corporation',
+        image: 'https://example.com/toyota.png',
+      },
       {
         id: 'berkeley',
         name: 'University of California, Berkeley',
@@ -1187,4 +1236,60 @@ describe('query autocompleteCompany', () => {
       },
     ]);
   });
+
+  it('should return results when searching by English name for company with altName', async () => {
+    const res = await client.query(QUERY, {
+      variables: { query: 'samsung' },
+    });
+
+    expect(res.errors).toBeFalsy();
+    expect(res.data.autocompleteCompany).toMatchObject([
+      {
+        id: 'samsung',
+        name: 'Samsung Electronics',
+        image: 'https://example.com/samsung.png',
+      },
+    ]);
+  });
+
+  it('should return results when searching by non-Latin characters (Korean)', async () => {
+    const res = await client.query(QUERY, {
+      variables: { query: '삼성' },
+    });
+
+    expect(res.errors).toBeFalsy();
+    expect(res.data.autocompleteCompany).toMatchObject([
+      {
+        id: 'samsung',
+        name: 'Samsung Electronics',
+        image: 'https://example.com/samsung.png',
+      },
+    ]);
+  });
+
+  it('should return results when searching by non-Latin characters (Japanese)', async () => {
+    const res = await client.query(QUERY, {
+      variables: { query: 'トヨタ' },
+    });
+
+    expect(res.errors).toBeFalsy();
+    expect(res.data.autocompleteCompany).toMatchObject([
+      {
+        id: 'toyota',
+        name: 'Toyota Motor Corporation',
+        image: 'https://example.com/toyota.png',
+      },
+    ]);
+  });
+
+  it('should not return unrelated companies when searching with non-Latin characters', async () => {
+    const res = await client.query(QUERY, {
+      variables: { query: '삼성' },
+    });
+
+    expect(res.errors).toBeFalsy();
+    // Should only return Samsung, not other companies with non-Latin altNames
+    expect(res.data.autocompleteCompany.length).toBe(1);
+    expect(res.data.autocompleteCompany[0].id).toBe('samsung');
+  });
 });