Function registerFunctionTools

index.js:1505–1596

()

}

function registerFunctionTools() {
    try {
        const { registerFunctionTool, unregisterFunctionTool } = SillyTavern.getContext();

        if (!registerFunctionTool || !unregisterFunctionTool) {
            console.log('WebSearch: Function tools are not supported');
            return;
        }

        if (!extension_settings.websearch.use_function_tool || !extension_settings.websearch.enabled) {
            unregisterFunctionTool('WebSearch');
            unregisterFunctionTool('VisitLinks');
            return;
        }

        const webSearchSchema = Object.freeze({
            $schema: 'http://json-schema.org/draft-04/schema#',
            type: 'object',
            properties: {
                query: {
                    type: 'string',
                    description: 'Web Query used in search engine.',
                },
            },
            required: [
                'query',
            ],
        });

        const visitLinksSchema = Object.freeze({
            $schema: 'http://json-schema.org/draft-04/schema#',
            type: 'object',
            properties: {
                links: {
                    type: 'array',
                    items: {
                        type: 'string',
                    },
                    description: 'Web links to visit.',
                },
            },
            required: [
                'links',
            ],
        });

        registerFunctionTool({
            name: 'WebSearch',
            displayName: 'Web Search',
            description: 'Search the web and get the content of the relevant pages. Search for unknown knowledge, public personalities, up-to-date information, weather, news, etc.',
            parameters: webSearchSchema,
            formatMessage: (args) => args?.query ? `Searching the web for: ${args?.query}` : '',
            action: async (args) => {
                if (!args) throw new Error('No arguments provided');
                if (!args.query) throw new Error('No query provided');
                if (!(await isSearchAvailable())) throw new Error('Search is not available');
                const search = await performSearchRequest(args.query, { useCache: true });
                return search;