wehub-resource-sync
59 lines
1.9 KiB
TypeScript
59 lines
1.9 KiB
TypeScript
import { tool, type Tool } from 'ai'
|
|
import { z } from 'zod'
|
|
import { bdclient } from '@brightdata/sdk'
|
|
|
|
type BrightDataTools = 'amazonProduct'
|
|
|
|
interface BrightDataToolsConfig {
|
|
apiKey: string
|
|
excludeTools?: BrightDataTools[]
|
|
}
|
|
|
|
export const brightDataTools = (
|
|
config: BrightDataToolsConfig
|
|
): Partial<Record<BrightDataTools, Tool>> => {
|
|
const client = new bdclient({
|
|
apiKey: config.apiKey,
|
|
autoCreateZones: true
|
|
})
|
|
|
|
const tools: Partial<Record<BrightDataTools, Tool>> = {
|
|
amazonProduct: tool({
|
|
description:
|
|
'Get detailed Amazon product information including price, ratings, reviews, and specifications. Requires a valid Amazon product URL.',
|
|
inputSchema: z.object({
|
|
url: z
|
|
.string()
|
|
.url()
|
|
.describe('Amazon product URL (must contain /dp/ or /gp/product/)'),
|
|
zipcode: z
|
|
.string()
|
|
.optional()
|
|
.describe('ZIP code for location-specific pricing and availability'),
|
|
}),
|
|
execute: async ({ url, zipcode }) => {
|
|
try {
|
|
const result = await client.datasets.amazon.collectProducts(
|
|
[{ url, zipcode }],
|
|
{
|
|
format: 'json',
|
|
async: false
|
|
}
|
|
)
|
|
return JSON.stringify(result, null, 2)
|
|
} catch (error) {
|
|
return `Error fetching Amazon product data: ${String(error)}`
|
|
}
|
|
},
|
|
}),
|
|
}
|
|
|
|
// Remove excluded tools
|
|
for (const toolName in tools) {
|
|
if (config.excludeTools?.includes(toolName as BrightDataTools)) {
|
|
delete tools[toolName as BrightDataTools]
|
|
}
|
|
}
|
|
|
|
return tools
|
|
} |