diff options
| author | A.J. Shulman <Shulman.aj@gmail.com> | 2024-12-18 11:46:14 -0500 |
|---|---|---|
| committer | A.J. Shulman <Shulman.aj@gmail.com> | 2024-12-18 11:46:14 -0500 |
| commit | ad1e0cf62187e0f8bbb19b4720b7681585361de9 (patch) | |
| tree | 673dd63ddc1808e6e89dab5021c2136cbbe843c8 /src/client/views/nodes/chatbot/tools/SearchTool.ts | |
| parent | 9e447814b551c352709296ae562f1f50480320f5 (diff) | |
better
Diffstat (limited to 'src/client/views/nodes/chatbot/tools/SearchTool.ts')
| -rw-r--r-- | src/client/views/nodes/chatbot/tools/SearchTool.ts | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/src/client/views/nodes/chatbot/tools/SearchTool.ts b/src/client/views/nodes/chatbot/tools/SearchTool.ts index d22f4c189..5fc6ab768 100644 --- a/src/client/views/nodes/chatbot/tools/SearchTool.ts +++ b/src/client/views/nodes/chatbot/tools/SearchTool.ts @@ -2,13 +2,14 @@ import { v4 as uuidv4 } from 'uuid'; import { Networking } from '../../../../Network'; import { BaseTool } from './BaseTool'; import { Observation } from '../types/types'; -import { ParametersType } from '../types/tool_types'; +import { ParametersType, ToolInfo } from '../types/tool_types'; const searchToolParams = [ { name: 'queries', type: 'string[]', - description: 'The search query or queries to use for finding websites', + description: + 'The search query or queries to use for finding websites. Provide up to 3 search queries to find a broad range of websites. Should be in the form of a TypeScript array of strings (e.g. <queries>["search term 1", "search term 2", "search term 3"]</queries>).', required: true, max_inputs: 3, }, @@ -16,18 +17,19 @@ const searchToolParams = [ type SearchToolParamsType = typeof searchToolParams; +const searchToolInfo: ToolInfo<SearchToolParamsType> = { + name: 'searchTool', + citationRules: 'No citation needed. Cannot cite search results for a response. Use web scraping tools to cite specific information.', + parameterRules: searchToolParams, + description: 'Search the web to find a wide range of websites related to a query or multiple queries. Returns a list of websites and their overviews based on the search queries.', +}; + export class SearchTool extends BaseTool<SearchToolParamsType> { private _addLinkedUrlDoc: (url: string, id: string) => void; private _max_results: number; constructor(addLinkedUrlDoc: (url: string, id: string) => void, max_results: number = 4) { - super( - 'searchTool', - 'Search the web to find a wide range of websites related to a query or multiple queries', - searchToolParams, - 'Provide up to 3 search queries to find a broad range of websites.', - 'Returns a list of websites and their overviews based on the search queries.' - ); + super(searchToolInfo); this._addLinkedUrlDoc = addLinkedUrlDoc; this._max_results = max_results; } |
