@coldiq/mcp 0.1.0 → 0.1.1

package/test-web-live.ts

@@ -1,8 +1,9 @@
 import { initClient } from './src/client.js'
 import { searchWebHandler } from './src/tools/search-web.js'
 
-const API_URL = 'https://api.coldiq.com'
-const API_KEY = 'ciq_test_3797d4ddc5542ceb'
+const API_URL = process.env.COLDIQ_API_URL ?? 'https://api.coldiq.com'
+const API_KEY = process.env.COLDIQ_API_KEY
+if (!API_KEY) { console.error('COLDIQ_API_KEY env var is required'); process.exit(1) }
 
 initClient(API_URL, API_KEY)
 

package/tests/client.test.ts

@@ -62,7 +62,7 @@ describe('client', () => {
 
     expect(result.ok).toBe(false)
     expect(result.status).toBe(0)
-    expect((result.data as Record<string, unknown>).error).toBe('Network timeout')
+    expect((result.data as Record<string, unknown>).error).toBe('Network error')
   })
 
   it('returns parsed JSON on success', async () => {

package/tests/executor.test.ts

@@ -1,6 +1,233 @@
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
 import { initClient } from '../src/client.js'
 import { executeWithFallback } from '../src/executor.js'
+import * as registry from '../src/registry.js'
+import type { ProviderEntry } from '../src/registry.js'
+
+// ---------------------------------------------------------------------------
+// Helpers for synthetic provider injection
+// ---------------------------------------------------------------------------
+
+function makeProvider(overrides: Partial<ProviderEntry> & { id: string }): ProviderEntry {
+  return {
+    endpoint: '/test',
+    method: 'POST',
+    priority: 1,
+    mapParams: () => ({ body: {} }),
+    hasResult: (data) => (data as Record<string, unknown>).ok === true,
+    ...overrides,
+  }
+}
+
+function stubProviders(providers: ProviderEntry[]) {
+  return vi.spyOn(registry, 'getProviders').mockReturnValueOnce(providers)
+}
+
+// ---------------------------------------------------------------------------
+// Exception safety — mapParams / hasResult / postFilter / isApplicable
+// ---------------------------------------------------------------------------
+
+describe('executor exception safety', () => {
+  const originalFetch = globalThis.fetch
+
+  beforeEach(() => {
+    initClient('http://test-api.local', 'test-key-123')
+  })
+
+  afterEach(() => {
+    globalThis.fetch = originalFetch
+    vi.restoreAllMocks()
+  })
+
+  it('mapParams throws → skips to next provider', async () => {
+    stubProviders([
+      makeProvider({
+        id: 'bad',
+        mapParams: () => { throw new Error('mapParams explosion') },
+        hasResult: () => false,
+      }),
+      makeProvider({
+        id: 'good',
+        mapParams: () => ({ body: {} }),
+        hasResult: (data) => (data as Record<string, unknown>).ok === true,
+      }),
+    ])
+
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ ok: true }), { status: 200 }),
+    ) as typeof fetch
+
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('data' in result).toBe(true)
+    if ('data' in result) {
+      expect(result._meta.provider).toBe('good')
+    }
+  })
+
+  it('hasResult throws → treated as no result, next provider tried', async () => {
+    stubProviders([
+      makeProvider({
+        id: 'exploding-has-result',
+        mapParams: () => ({ body: {} }),
+        hasResult: () => { throw new Error('hasResult explosion') },
+      }),
+      makeProvider({
+        id: 'safe',
+        mapParams: () => ({ body: {} }),
+        hasResult: (data) => (data as Record<string, unknown>).ok === true,
+      }),
+    ])
+
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ ok: true }), { status: 200 }),
+    ) as typeof fetch
+
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('data' in result).toBe(true)
+    if ('data' in result) {
+      expect(result._meta.provider).toBe('safe')
+    }
+  })
+
+  it('postFilter throws → falls back to raw payload, success path continues', async () => {
+    stubProviders([
+      makeProvider({
+        id: 'exploding-post-filter',
+        mapParams: () => ({ body: {} }),
+        hasResult: (data) => (data as Record<string, unknown>).ok === true,
+        postFilter: () => { throw new Error('postFilter explosion') },
+      }),
+    ])
+
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ ok: true }), { status: 200 }),
+    ) as typeof fetch
+
+    // postFilter throws → raw payload is used → hasResult(raw) should still return true
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('data' in result).toBe(true)
+    if ('data' in result) {
+      expect(result._meta.provider).toBe('exploding-post-filter')
+    }
+  })
+
+  it('isApplicable=false → provider skipped and NOT recorded in providers_tried', async () => {
+    stubProviders([
+      makeProvider({
+        id: 'skipped',
+        isApplicable: () => false,
+        hasResult: () => false,
+      }),
+    ])
+
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ ok: false }), { status: 200 }),
+    ) as typeof fetch
+
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('error' in result).toBe(true)
+    if ('error' in result) {
+      // Skipped providers do not appear in providers_tried
+      expect(result.providers_tried.length).toBe(0)
+    }
+  })
+
+  it('isApplicable throws → provider skipped gracefully', async () => {
+    stubProviders([
+      makeProvider({
+        id: 'throwing-applicable',
+        isApplicable: () => { throw new Error('isApplicable explosion') },
+        hasResult: () => false,
+      }),
+      makeProvider({
+        id: 'safe',
+        hasResult: (data) => (data as Record<string, unknown>).ok === true,
+      }),
+    ])
+
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ ok: true }), { status: 200 }),
+    ) as typeof fetch
+
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('data' in result).toBe(true)
+    if ('data' in result) {
+      expect(result._meta.provider).toBe('safe')
+    }
+  })
+
+  it('all providers fail → providers_tried uses provider_N keys (not internal ids)', async () => {
+    stubProviders([
+      makeProvider({ id: 'alpha', hasResult: () => false }),
+      makeProvider({ id: 'beta', hasResult: () => false }),
+    ])
+
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ ok: false }), { status: 200 }),
+    ) as typeof fetch
+
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('error' in result).toBe(true)
+    if ('error' in result) {
+      expect(result.providers_tried).toHaveLength(2)
+      expect(result.providers_tried[0].provider).toBe('provider_1')
+      expect(result.providers_tried[1].provider).toBe('provider_2')
+      expect('id' in result.providers_tried[0]).toBe(false)
+    }
+  })
+
+  it('async create returns 4xx → recorded as failure, next provider tried', async () => {
+    stubProviders([
+      makeProvider({
+        id: 'async-fail',
+        hasResult: () => false,
+        async: {
+          extractId: () => 'job-1',
+          pollEndpoint: (id) => `/poll/${id}`,
+          pollIntervalMs: 10,
+          timeoutMs: 50,
+          isComplete: () => true,
+        },
+      }),
+      makeProvider({
+        id: 'sync-success',
+        hasResult: (data) => (data as Record<string, unknown>).ok === true,
+      }),
+    ])
+
+    let callCount = 0
+    globalThis.fetch = vi.fn(async (url: string | URL | Request) => {
+      callCount++
+      const u = url.toString()
+      if (u.includes('/test') && !u.includes('/poll/')) {
+        if (callCount === 1) {
+          // async-fail create — returns 4xx so executor skips polling
+          return new Response(JSON.stringify({ error: 'quota exceeded' }), { status: 429 })
+        }
+        // sync-success
+        return new Response(JSON.stringify({ ok: true }), { status: 200 })
+      }
+      return new Response(JSON.stringify({}), { status: 200 })
+    }) as typeof fetch
+
+    const result = await executeWithFallback('enrich_company', { domain: 'coldiq.com' })
+
+    expect('data' in result).toBe(true)
+    if ('data' in result) {
+      expect(result._meta.provider).toBe('sync-success')
+    }
+  })
+})
+
+// ---------------------------------------------------------------------------
+// postFilter integration (existing tests)
+// ---------------------------------------------------------------------------
 
 describe('executor postFilter integration', () => {
   const originalFetch = globalThis.fetch

package/tests/registry.test.ts

@@ -384,53 +384,6 @@ describe('registry', () => {
       expect(sumble.isApplicable!({ countries: ['FR'] })).toBe(false)
     })
 
-    it('Wiza creates a prospect list with [{v}] filter shape, translates ISO codes, and has async config', () => {
-      const providers = getProviders('search_companies')
-      const wiza = providers.find((p) => p.id === 'wiza')!
-      const result = wiza.mapParams({
-        industries: ['SaaS'],
-        countries: ['FR'],
-        min_founded_year: 2015,
-        limit: 25,
-      })
-      expect(result.body).toEqual({
-        list: { name: 'mcp-search-companies', max_profiles: 25 },
-        filters: {
-          company_industry: [{ v: 'SaaS' }],
-          company_location: [{ v: 'France' }],
-          year_founded_start: [{ v: '2015' }],
-        },
-        enrichment_level: 'none',
-      })
-      expect(wiza.async).toBeDefined()
-      expect(wiza.async!.pollEndpoint('123')).toBe('/wiza/lists/123')
-      expect(wiza.async!.extractId({ data: { id: 123 } })).toBe('123')
-    })
-
-    it('Wiza isComplete accepts both finished and completed terminal states', () => {
-      const providers = getProviders('search_companies')
-      const wiza = providers.find((p) => p.id === 'wiza')!
-      expect(wiza.async!.isComplete({ data: { status: 'finished' } })).toBe(true)
-      expect(wiza.async!.isComplete({ data: { status: 'completed' } })).toBe(true)
-      expect(wiza.async!.isComplete({ data: { status: 'failed' } })).toBe(true)
-      expect(wiza.async!.isComplete({ data: { status: 'scraping' } })).toBe(false)
-      expect(wiza.async!.isComplete({ data: { status: 'queued' } })).toBe(false)
-    })
-
-    it('Wiza extractId throws when no list id is present', () => {
-      const providers = getProviders('search_companies')
-      const wiza = providers.find((p) => p.id === 'wiza')!
-      expect(() => wiza.async!.extractId({})).toThrow('Wiza response has no list id')
-      expect(() => wiza.async!.extractId({ data: {} })).toThrow('Wiza response has no list id')
-    })
-
-    it('Wiza hasResult checks data.stats.people > 0', () => {
-      const providers = getProviders('search_companies')
-      const wiza = providers.find((p) => p.id === 'wiza')!
-      expect(wiza.hasResult({ data: { status: 'finished', stats: { people: 10 } } })).toBe(true)
-      expect(wiza.hasResult({ data: { status: 'finished', stats: { people: 0 } } })).toBe(false)
-      expect(wiza.hasResult({ data: { status: 'finished' } })).toBe(false)
-    })
 
     it('LimaData prospect filter maps employees to company_headcount filter_type', () => {
       const providers = getProviders('search_companies')
@@ -464,14 +417,13 @@ describe('registry', () => {
       expect(ld.isApplicable!({ linkedin_search_url: 'https://...' })).toBe(true)
     })
 
-    it('search_companies has 18 providers in priority order', () => {
+    it('search_companies has 17 providers in priority order', () => {
       const providers = getProviders('search_companies')
       expect(providers.map((p) => p.id)).toEqual([
         'companyenrich',
         'fullenrich',
         'pdl',
         'theirstack',
-        'wiza',
         'signalbase',
         'blitzapi',
         'apollo',
@@ -1176,16 +1128,6 @@ describe('registry', () => {
       expect(ts.isApplicable!({ countries: ['FR'] })).toBe(false)
     })
 
-    it('Wiza isApplicable: fires for min/max_founded_year and funding filters', () => {
-      const wiza = getProviders('search_companies').find((p) => p.id === 'wiza')!
-      expect(wiza.isApplicable!({})).toBe(false)
-      expect(wiza.isApplicable!({ keywords: ['x'] })).toBe(false)
-      expect(wiza.isApplicable!({ min_founded_year: 2015 })).toBe(true)
-      expect(wiza.isApplicable!({ max_founded_year: 2020 })).toBe(true)
-      expect(wiza.isApplicable!({ funding_stages: ['seed'] })).toBe(true)
-      expect(wiza.isApplicable!({ min_funding_amount: 500_000 })).toBe(true)
-      expect(wiza.isApplicable!({ max_funding_amount: 10_000_000 })).toBe(true)
-    })
 
     it('Apollo postFilter strips accounts key entirely', () => {
       const apollo = getProviders('search_companies').find((p) => p.id === 'apollo')!
@@ -1327,24 +1269,6 @@ describe('registry', () => {
       expect((result.body as Record<string, unknown>).exclude).toEqual({ domains: ['a.com'] })
     })
 
-    it('Wiza mapParams adds funding_stage, funding_min, funding_max in [{v}] shape', () => {
-      const wiza = getProviders('search_companies').find((p) => p.id === 'wiza')!
-      const result = wiza.mapParams({
-        funding_stages: ['seed', 'series_a'],
-        min_funding_amount: 500_000,
-        max_funding_amount: 20_000_000,
-        limit: 25,
-      })
-      expect(result.body).toEqual({
-        list: { name: 'mcp-search-companies', max_profiles: 25 },
-        filters: {
-          funding_stage: [{ v: 'seed' }, { v: 'series_a' }],
-          funding_min: [{ v: '500000' }],
-          funding_max: [{ v: '20000000' }],
-        },
-        enrichment_level: 'none',
-      })
-    })
 
     it('LimaData passes has_jobs: true when is_hiring is true', () => {
       const ld = getProviders('search_companies').find((p) => p.id === 'limadata')!

package/tests/tools/{enrich-email.test.ts → find-email.test.ts}

@@ -1,6 +1,6 @@
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
 import { initClient } from '../../src/client.js'
-import { findEmailHandler as enrichEmailHandler } from '../../src/tools/find-email.js'
+import { findEmailHandler } from '../../src/tools/find-email.js'
 
 describe('find_email handler (waterfall)', () => {
   const originalFetch = globalThis.fetch
@@ -18,7 +18,7 @@ describe('find_email handler (waterfall)', () => {
       new Response(JSON.stringify({ email: 'michel@coldiq.com' }), { status: 200 })
     ) as typeof fetch
 
-    const result = await enrichEmailHandler({
+    const result = await findEmailHandler({
       first_name: 'Michel',
       last_name: 'Lieben',
       domain: 'coldiq.com',
@@ -41,7 +41,7 @@ describe('find_email handler (waterfall)', () => {
       return new Response(JSON.stringify({ email: 'michel@coldiq.com' }), { status: 200 })
     }) as typeof fetch
 
-    const result = await enrichEmailHandler({
+    const result = await findEmailHandler({
       first_name: 'Michel',
       last_name: 'Lieben',
       domain: 'coldiq.com',
@@ -65,7 +65,7 @@ describe('find_email handler (waterfall)', () => {
       )
     }) as typeof fetch
 
-    const result = await enrichEmailHandler({
+    const result = await findEmailHandler({
       first_name: 'Michel',
       last_name: 'Lieben',
       domain: 'coldiq.com',
@@ -80,7 +80,7 @@ describe('find_email handler (waterfall)', () => {
       new Response(JSON.stringify({ error: 'Not found' }), { status: 404 })
     ) as typeof fetch
 
-    const result = await enrichEmailHandler({
+    const result = await findEmailHandler({
       first_name: 'Unknown',
       last_name: 'Person',
       domain: 'unknown.xyz',

package/tests/tools/{enrich-emails.test.ts → find-emails.test.ts}

@@ -1,6 +1,6 @@
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
 import { initClient } from '../../src/client.js'
-import { findEmailsHandler as enrichEmailsHandler } from '../../src/tools/find-emails.js'
+import { findEmailsHandler } from '../../src/tools/find-emails.js'
 
 describe('find_emails handler (bulk)', () => {
   const originalFetch = globalThis.fetch
@@ -32,7 +32,7 @@ describe('find_emails handler (bulk)', () => {
       return new Response(JSON.stringify({ error: 'unexpected call' }), { status: 500 })
     }) as typeof fetch
 
-    const result = await enrichEmailsHandler({
+    const result = await findEmailsHandler({
       people: [
         { id: 'p1', first_name: 'Alice', last_name: 'Smith', domain: 'example.com' },
         { id: 'p2', first_name: 'Bob', last_name: 'Jones', domain: 'example.com' },
@@ -71,7 +71,7 @@ describe('find_emails handler (bulk)', () => {
       return new Response(JSON.stringify({ error: 'unexpected' }), { status: 500 })
     }) as typeof fetch
 
-    const result = await enrichEmailsHandler({
+    const result = await findEmailsHandler({
       people: [{ id: 'p1', first_name: 'Alice', last_name: 'Smith', domain: 'example.com' }],
     })
 
@@ -98,7 +98,7 @@ describe('find_emails handler (bulk)', () => {
       return new Response(JSON.stringify({ error: 'unexpected' }), { status: 500 })
     }) as typeof fetch
 
-    const result = await enrichEmailsHandler({
+    const result = await findEmailsHandler({
       people: [{ id: 'p1', first_name: 'Alice', last_name: 'Smith', domain: 'example.com' }],
     })
 
@@ -111,7 +111,7 @@ describe('find_emails handler (bulk)', () => {
       new Response(JSON.stringify({ error: true }), { status: 200 }),
     ) as typeof fetch
 
-    const result = await enrichEmailsHandler({
+    const result = await findEmailsHandler({
       people: [{ id: 'p1', first_name: 'Unknown', last_name: 'Person', domain: 'nowhere.xyz' }],
     })
 
@@ -153,7 +153,7 @@ describe('find_emails handler (bulk)', () => {
       return new Response(JSON.stringify({ error: 'unexpected' }), { status: 500 })
     }) as typeof fetch
 
-    const result = await enrichEmailsHandler({
+    const result = await findEmailsHandler({
       people: [
         { id: 'p1', first_name: 'Alice', last_name: 'Smith', domain: 'a.com' },
         { id: 'p2', first_name: 'Bob', last_name: 'Jones', domain: 'b.com' },
@@ -187,7 +187,7 @@ describe('find_emails handler (bulk)', () => {
       return new Response(JSON.stringify({ email: null }), { status: 200 })
     }) as typeof fetch
 
-    await enrichEmailsHandler({
+    await findEmailsHandler({
       people: [{ id: 'p1', linkedin_url: 'https://linkedin.com/in/michel-lieben' }],
     })
 
@@ -196,13 +196,13 @@ describe('find_emails handler (bulk)', () => {
     expect(body.data[0].first_name).toBeUndefined()
   })
 
-  it('uses FullEnrich batch for Prospeo misses and matches by index', async () => {
+  it('uses FullEnrich batch for Prospeo misses and matches by custom_id', async () => {
     vi.useFakeTimers()
 
-    let feCreateCalled = false
+    let feCreateBody: unknown
     let fePollCalled = false
 
-    globalThis.fetch = vi.fn(async (url: string | URL | Request) => {
+    globalThis.fetch = vi.fn(async (url: string | URL | Request, init?: RequestInit) => {
       const u = url.toString()
 
       if (u.includes('/prospeo/bulk-enrich-person')) {
@@ -217,19 +217,19 @@ describe('find_emails handler (bulk)', () => {
         fePollCalled = true
         return new Response(JSON.stringify({
           status: 'DONE',
-          data: [{ emails: ['alice@example.com'] }],
+          data: [{ custom_id: 'p1', emails: ['alice@example.com'] }],
         }), { status: 200 })
       }
 
       if (u.includes('/fullenrich/contact/enrich/bulk')) {
-        feCreateCalled = true
+        feCreateBody = JSON.parse(init?.body as string)
         return new Response(JSON.stringify({ enrichment_id: 'abc-fe-123' }), { status: 200 })
       }
 
       return new Response(JSON.stringify({ error: 'unexpected' }), { status: 500 })
     }) as typeof fetch
 
-    const handlerPromise = enrichEmailsHandler({
+    const handlerPromise = findEmailsHandler({
       people: [{ id: 'p1', first_name: 'Alice', last_name: 'Smith', domain: 'example.com' }],
     })
 
@@ -237,8 +237,9 @@ describe('find_emails handler (bulk)', () => {
     const result = await handlerPromise
     vi.useRealTimers()
 
-    expect(feCreateCalled).toBe(true)
     expect(fePollCalled).toBe(true)
+    const body = feCreateBody as { data: Array<Record<string, unknown>> }
+    expect(body.data[0].custom_id).toBe('p1')
     const parsed = JSON.parse(result.content[0].text)
     expect(parsed.data.results[0]).toEqual({ id: 'p1', email: 'alice@example.com', provider: 'fullenrich' })
   })
@@ -255,7 +256,7 @@ describe('find_emails handler (bulk)', () => {
       return new Response(JSON.stringify({ error: 'unexpected' }), { status: 500 })
     }) as typeof fetch
 
-    const result = await enrichEmailsHandler({
+    const result = await findEmailsHandler({
       people: [{ id: 'p1', first_name: 'Alice', last_name: 'Smith', domain: 'example.com' }],
     })
 

package/tests/tools/find-people.test.ts

@@ -43,10 +43,10 @@ describe('find_people handler', () => {
     const lf = providers.find((p) => p.id === 'leadsfactory')!
     const originalTimeout = lf.async!.timeoutMs
     const originalInterval = lf.async!.pollIntervalMs
-    lf.async!.timeoutMs = 100
-    lf.async!.pollIntervalMs = 20
 
     try {
+      lf.async!.timeoutMs = 100
+      lf.async!.pollIntervalMs = 20
       const result = await findPeopleHandler({
         company_domains: ['coldiq.com'],
         job_titles: ['CEO'],
@@ -96,10 +96,10 @@ describe('find_people handler', () => {
     const lf = providers.find((p) => p.id === 'leadsfactory')!
     const originalTimeout = lf.async!.timeoutMs
     const originalInterval = lf.async!.pollIntervalMs
-    lf.async!.timeoutMs = 500
-    lf.async!.pollIntervalMs = 20
 
     try {
+      lf.async!.timeoutMs = 500
+      lf.async!.pollIntervalMs = 20
       const result = await findPeopleHandler({
         company_domains: ['coldiq.com'],
         job_titles: ['CEO'],
@@ -150,10 +150,10 @@ describe('find_people handler', () => {
     const lf = providers.find((p) => p.id === 'leadsfactory')!
     const originalTimeout = lf.async!.timeoutMs
     const originalInterval = lf.async!.pollIntervalMs
-    lf.async!.timeoutMs = 500
-    lf.async!.pollIntervalMs = 20
 
     try {
+      lf.async!.timeoutMs = 500
+      lf.async!.pollIntervalMs = 20
       const result = await findPeopleHandler({
         company_domains: ['coldiq.com', 'folk.app'],
         job_titles: ['CEO'],
@@ -203,10 +203,10 @@ describe('find_people handler', () => {
     const lf = providers.find((p) => p.id === 'leadsfactory')!
     const originalTimeout = lf.async!.timeoutMs
     const originalInterval = lf.async!.pollIntervalMs
-    lf.async!.timeoutMs = 500
-    lf.async!.pollIntervalMs = 20
 
     try {
+      lf.async!.timeoutMs = 500
+      lf.async!.pollIntervalMs = 20
       const result = await findPeopleHandler({
         company_linkedin_urls: ['https://linkedin.com/company/coldiq', 'https://linkedin.com/company/folk'],
         job_titles: ['CEO'],
@@ -249,10 +249,10 @@ describe('find_people handler', () => {
     const lf = providers.find((p) => p.id === 'leadsfactory')!
     const originalTimeout = lf.async!.timeoutMs
     const originalInterval = lf.async!.pollIntervalMs
-    lf.async!.timeoutMs = 500
-    lf.async!.pollIntervalMs = 20
 
     try {
+      lf.async!.timeoutMs = 500
+      lf.async!.pollIntervalMs = 20
       await findPeopleHandler({
         company_linkedin_urls: ['https://linkedin.com/company/coldiq', 'https://linkedin.com/company/folk'],
         company_domains: ['coldiq.com', 'folk.app'],
@@ -295,10 +295,10 @@ describe('find_people handler', () => {
     const lf = providers.find((p) => p.id === 'leadsfactory')!
     const originalTimeout = lf.async!.timeoutMs
     const originalInterval = lf.async!.pollIntervalMs
-    lf.async!.timeoutMs = 500
-    lf.async!.pollIntervalMs = 20
 
     try {
+      lf.async!.timeoutMs = 500
+      lf.async!.pollIntervalMs = 20
       await findPeopleHandler({
         company_domains: ['coldiq.com'],
         job_titles: ['CEO', 'VP of Sales'],

package/tests/tools/search-places.test.ts

@@ -49,10 +49,10 @@ describe('search_places handler', () => {
     const providers = getProviders('search_places')
     const gm = providers.find((p) => p.id === 'google_maps')!
     const orig = { t: gm.async!.timeoutMs, i: gm.async!.pollIntervalMs }
-    gm.async!.timeoutMs = 500
-    gm.async!.pollIntervalMs = 20
 
     try {
+      gm.async!.timeoutMs = 500
+      gm.async!.pollIntervalMs = 20
       const result = await searchPlacesHandler({ query: 'café', country: 'FR', limit: 5 })
 
       expect(result.isError).toBeFalsy()
@@ -84,10 +84,10 @@ describe('search_places handler', () => {
     const providers = getProviders('search_places')
     const gm = providers.find((p) => p.id === 'google_maps')!
     const orig = { t: gm.async!.timeoutMs, i: gm.async!.pollIntervalMs }
-    gm.async!.timeoutMs = 500
-    gm.async!.pollIntervalMs = 20
 
     try {
+      gm.async!.timeoutMs = 500
+      gm.async!.pollIntervalMs = 20
       const result = await searchPlacesHandler({ query: 'coffee', country: 'US', provider: 'google_maps', limit: 5 })
 
       expect(result.isError).toBeFalsy()

package/tests/tools/search-reddit.test.ts

@@ -33,10 +33,10 @@ describe('search_reddit handler', () => {
     const providers = getProviders('search_reddit')
     const reddit = providers.find((p) => p.id === 'reddit')!
     const orig = { t: reddit.async!.timeoutMs, i: reddit.async!.pollIntervalMs }
-    reddit.async!.timeoutMs = 500
-    reddit.async!.pollIntervalMs = 20
 
     try {
+      reddit.async!.timeoutMs = 500
+      reddit.async!.pollIntervalMs = 20
       const result = await searchRedditHandler({ start_urls: ['https://www.reddit.com/r/sales/'], query: 'best CRM for startups', limit: 5 })
 
       expect(result.isError).toBeFalsy()
@@ -72,10 +72,10 @@ describe('search_reddit handler', () => {
     const providers = getProviders('search_reddit')
     const reddit = providers.find((p) => p.id === 'reddit')!
     const orig = { t: reddit.async!.timeoutMs, i: reddit.async!.pollIntervalMs }
-    reddit.async!.timeoutMs = 500
-    reddit.async!.pollIntervalMs = 20
 
     try {
+      reddit.async!.timeoutMs = 500
+      reddit.async!.pollIntervalMs = 20
       await searchRedditHandler({
         start_urls: ['https://www.reddit.com/r/entrepreneur/'],
         limit: 5,
@@ -108,10 +108,10 @@ describe('search_reddit handler', () => {
     const providers = getProviders('search_reddit')
     const reddit = providers.find((p) => p.id === 'reddit')!
     const orig = { t: reddit.async!.timeoutMs, i: reddit.async!.pollIntervalMs }
-    reddit.async!.timeoutMs = 100
-    reddit.async!.pollIntervalMs = 20
 
     try {
+      reddit.async!.timeoutMs = 100
+      reddit.async!.pollIntervalMs = 20
       const result = await searchRedditHandler({ start_urls: ['https://www.reddit.com/r/sales/'], query: 'cold email', limit: 5 })
 
       expect(result.isError).toBe(true)