chore: import upstream snapshot with attribution
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
/**
|
||||
* @vitest-environment node
|
||||
*/
|
||||
import { describe, expect, it } from 'vitest'
|
||||
import { companyDomainEnrichment } from '@/enrichments/company-domain/company-domain'
|
||||
import type { EnrichmentProvider } from '@/enrichments/types'
|
||||
|
||||
function provider(id: string): EnrichmentProvider {
|
||||
const p = companyDomainEnrichment.providers.find((x) => x.id === id)
|
||||
if (!p) throw new Error(`Provider ${id} not found in company-domain cascade`)
|
||||
return p
|
||||
}
|
||||
|
||||
const nameInput = { companyName: 'Acme Inc' }
|
||||
|
||||
describe('company-domain enrichment cascade', () => {
|
||||
it('chains PDL then Datagma', () => {
|
||||
expect(companyDomainEnrichment.providers.map((p) => p.id)).toEqual(['pdl', 'datagma'])
|
||||
})
|
||||
|
||||
describe('pdl', () => {
|
||||
const p = provider('pdl')
|
||||
it('matches by name and normalizes the returned website', () => {
|
||||
expect(p.toolId).toBe('pdl_company_enrich')
|
||||
expect(p.buildParams(nameInput)).toEqual({ name: 'Acme Inc', required: 'website' })
|
||||
expect(p.buildParams({ companyName: '' })).toBeNull()
|
||||
expect(p.mapOutput({ company: { website: 'https://www.acme.com' } })).toEqual({
|
||||
domain: 'acme.com',
|
||||
})
|
||||
expect(p.mapOutput({ company: {} })).toBeNull()
|
||||
})
|
||||
})
|
||||
|
||||
describe('datagma', () => {
|
||||
const p = provider('datagma')
|
||||
it('enriches by company name and normalizes the returned website', () => {
|
||||
expect(p.toolId).toBe('datagma_enrich_company')
|
||||
expect(p.buildParams(nameInput)).toEqual({ data: 'Acme Inc' })
|
||||
expect(p.buildParams({ companyName: '' })).toBeNull()
|
||||
expect(p.mapOutput({ website: 'https://www.acme.com/' })).toEqual({ domain: 'acme.com' })
|
||||
expect(p.mapOutput({})).toBeNull()
|
||||
})
|
||||
})
|
||||
})
|
||||
@@ -0,0 +1,50 @@
|
||||
import { Globe } from 'lucide-react'
|
||||
import { normalizeDomain, str, toolProvider } from '@/enrichments/providers'
|
||||
import type { EnrichmentConfig } from '@/enrichments/types'
|
||||
|
||||
/**
|
||||
* Company Domain enrichment. Resolves a company's website domain from its name
|
||||
* via a People Data Labs company match, falling back to Datagma's company enrich.
|
||||
*/
|
||||
export const companyDomainEnrichment: EnrichmentConfig = {
|
||||
id: 'company-domain',
|
||||
name: 'Company Domain',
|
||||
description: "Find a company's website domain from its name.",
|
||||
icon: Globe,
|
||||
inputs: [{ id: 'companyName', name: 'Company name', type: 'string', required: true }],
|
||||
outputs: [{ id: 'domain', name: 'domain', type: 'string' }],
|
||||
providers: [
|
||||
toolProvider({
|
||||
id: 'pdl',
|
||||
label: 'People Data Labs',
|
||||
toolId: 'pdl_company_enrich',
|
||||
buildParams: (inputs) => {
|
||||
const name = str(inputs.companyName)
|
||||
if (!name) return null
|
||||
// `required` makes PDL 404 (free) when the match has no website,
|
||||
// instead of charging a credit for a match we'd discard as a no-match.
|
||||
return { name, required: 'website' }
|
||||
},
|
||||
mapOutput: (output) => {
|
||||
const company = output.company as Record<string, unknown> | undefined
|
||||
const domain = normalizeDomain(company?.website)
|
||||
return domain ? { domain } : null
|
||||
},
|
||||
}),
|
||||
toolProvider({
|
||||
id: 'datagma',
|
||||
label: 'Datagma',
|
||||
toolId: 'datagma_enrich_company',
|
||||
buildParams: (inputs) => {
|
||||
// Datagma's `data` accepts a company name and returns its website.
|
||||
const data = str(inputs.companyName)
|
||||
if (!data) return null
|
||||
return { data }
|
||||
},
|
||||
mapOutput: (output) => {
|
||||
const domain = normalizeDomain(output.website)
|
||||
return domain ? { domain } : null
|
||||
},
|
||||
}),
|
||||
],
|
||||
}
|
||||
@@ -0,0 +1 @@
|
||||
export { companyDomainEnrichment } from './company-domain'
|
||||
Reference in New Issue
Block a user