mirror of
https://github.com/simstudioai/sim.git
synced 2026-01-09 06:58:07 -05:00
766279bb8b
* feat(integrations): added parallel ai block/tool and corresponding docs * add postgres block * added mysql block * enrich docs for Postgres and MySQL * make password fields user only for mysql and postgres * fixed build * ack greptile comments * fix PR comments * remove search_id from parallel ai * fix parallel ai params
109 lines
3.0 KiB
TypeScript
109 lines
3.0 KiB
TypeScript
import type { ParallelSearchParams } from '@/tools/parallel/types'
|
|
import type { ToolConfig, ToolResponse } from '@/tools/types'
|
|
|
|
export const searchTool: ToolConfig<ParallelSearchParams, ToolResponse> = {
|
|
id: 'parallel_search',
|
|
name: 'Parallel AI Search',
|
|
description:
|
|
'Search the web using Parallel AI. Provides comprehensive search results with intelligent processing and content extraction.',
|
|
version: '1.0.0',
|
|
|
|
params: {
|
|
objective: {
|
|
type: 'string',
|
|
required: true,
|
|
visibility: 'user-or-llm',
|
|
description: 'The search objective or question to answer',
|
|
},
|
|
search_queries: {
|
|
type: 'string',
|
|
required: false,
|
|
visibility: 'user-or-llm',
|
|
description: 'Optional comma-separated list of search queries to execute',
|
|
},
|
|
processor: {
|
|
type: 'string',
|
|
required: false,
|
|
visibility: 'user-only',
|
|
description: 'Processing method: base or pro (default: base)',
|
|
},
|
|
max_results: {
|
|
type: 'number',
|
|
required: false,
|
|
visibility: 'user-only',
|
|
description: 'Maximum number of results to return (default: 5)',
|
|
},
|
|
max_chars_per_result: {
|
|
type: 'number',
|
|
required: false,
|
|
visibility: 'user-only',
|
|
description: 'Maximum characters per result (default: 1500)',
|
|
},
|
|
apiKey: {
|
|
type: 'string',
|
|
required: true,
|
|
visibility: 'user-only',
|
|
description: 'Parallel AI API Key',
|
|
},
|
|
},
|
|
|
|
request: {
|
|
url: 'https://api.parallel.ai/v1beta/search',
|
|
method: 'POST',
|
|
headers: (params) => ({
|
|
'Content-Type': 'application/json',
|
|
'x-api-key': params.apiKey,
|
|
}),
|
|
body: (params) => {
|
|
const body: Record<string, unknown> = {
|
|
objective: params.objective,
|
|
search_queries: params.search_queries,
|
|
}
|
|
|
|
// Add optional parameters if provided
|
|
if (params.processor) body.processor = params.processor
|
|
if (params.max_results) body.max_results = params.max_results
|
|
if (params.max_chars_per_result) body.max_chars_per_result = params.max_chars_per_result
|
|
|
|
return body
|
|
},
|
|
},
|
|
|
|
transformResponse: async (response: Response) => {
|
|
const data = await response.json()
|
|
|
|
return {
|
|
success: true,
|
|
output: {
|
|
results: data.results.map((result: unknown) => {
|
|
const resultObj = result as Record<string, unknown>
|
|
return {
|
|
url: resultObj.url || '',
|
|
title: resultObj.title || '',
|
|
excerpts: resultObj.excerpts || [],
|
|
}
|
|
}),
|
|
},
|
|
}
|
|
},
|
|
|
|
outputs: {
|
|
results: {
|
|
type: 'array',
|
|
description: 'Search results with excerpts from relevant pages',
|
|
items: {
|
|
type: 'object',
|
|
properties: {
|
|
url: { type: 'string', description: 'The URL of the search result' },
|
|
title: { type: 'string', description: 'The title of the search result' },
|
|
excerpts: {
|
|
type: 'array',
|
|
description: 'Text excerpts from the page',
|
|
items: { type: 'string' },
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|