@librechat/agents 2.4.30 → 2.4.311

package/dist/esm/tools/search/search.mjs

@@ -0,0 +1,407 @@
+import axios from 'axios';
+import { RecursiveCharacterTextSplitter } from '@langchain/textsplitters';
+import { getAttribution } from './firecrawl.mjs';
+
+/* eslint-disable no-console */
+const chunker = {
+    cleanText: (text) => {
+        if (!text)
+            return '';
+        /** Normalized all line endings to '\n' */
+        const normalizedText = text.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
+        /** Handle multiple backslashes followed by newlines
+         * This replaces patterns like '\\\\\\n' with a single newline */
+        const fixedBackslashes = normalizedText.replace(/\\+\n/g, '\n');
+        /** Cleaned up consecutive newlines, tabs, and spaces around newlines */
+        const cleanedNewlines = fixedBackslashes.replace(/[\t ]*\n[\t \n]*/g, '\n');
+        /** Cleaned up excessive spaces and tabs */
+        const cleanedSpaces = cleanedNewlines.replace(/[ \t]+/g, ' ');
+        return cleanedSpaces.trim();
+    },
+    splitText: async (text, options) => {
+        const chunkSize = options?.chunkSize ?? 150;
+        const chunkOverlap = options?.chunkOverlap ?? 50;
+        const separators = options?.separators || ['\n\n', '\n'];
+        const splitter = new RecursiveCharacterTextSplitter({
+            separators,
+            chunkSize,
+            chunkOverlap,
+        });
+        return await splitter.splitText(text);
+    },
+    splitTexts: async (texts, options) => {
+        // Split multiple texts
+        const promises = texts.map((text) => chunker.splitText(text, options).catch((error) => {
+            console.error('Error splitting text:', error);
+            return [text];
+        }));
+        return Promise.all(promises);
+    },
+};
+const createSourceUpdateCallback = (sourceMap) => {
+    return (link, update) => {
+        const source = sourceMap.get(link);
+        if (source) {
+            sourceMap.set(link, {
+                ...source,
+                ...update,
+            });
+        }
+    };
+};
+const getHighlights = async ({ query, content, reranker, topResults = 5, }) => {
+    if (!content) {
+        console.warn('No content provided for highlights');
+        return;
+    }
+    if (!reranker) {
+        console.warn('No reranker provided for highlights');
+        return;
+    }
+    try {
+        const documents = await chunker.splitText(content);
+        if (Array.isArray(documents)) {
+            return await reranker.rerank(query, documents, topResults);
+        }
+        else {
+            console.error('Expected documents to be an array, got:', typeof documents);
+            return;
+        }
+    }
+    catch (error) {
+        console.error('Error in content processing:', error);
+        return;
+    }
+};
+const createSerperAPI = (apiKey) => {
+    const config = {
+        apiKey: apiKey ?? process.env.SERPER_API_KEY,
+        apiUrl: 'https://google.serper.dev/search',
+        defaultLocation: 'us',
+        timeout: 10000,
+    };
+    if (config.apiKey == null || config.apiKey === '') {
+        throw new Error('SERPER_API_KEY is required for SerperAPI');
+    }
+    const getSources = async (query, numResults = 8, storedLocation) => {
+        if (!query.trim()) {
+            return { success: false, error: 'Query cannot be empty' };
+        }
+        try {
+            const searchLocation = (storedLocation ?? config.defaultLocation).toLowerCase();
+            const payload = {
+                q: query,
+                num: Math.min(Math.max(1, numResults), 10),
+                gl: searchLocation,
+            };
+            const response = await axios.post(config.apiUrl, payload, {
+                headers: {
+                    'X-API-KEY': config.apiKey,
+                    'Content-Type': 'application/json',
+                },
+                timeout: config.timeout,
+            });
+            const data = response.data;
+            const results = {
+                organic: data.organic,
+                images: data.images ?? [],
+                topStories: data.topStories ?? [],
+                knowledgeGraph: data.knowledgeGraph,
+                answerBox: data.answerBox,
+                peopleAlsoAsk: data.peopleAlsoAsk,
+                relatedSearches: data.relatedSearches,
+            };
+            return { success: true, data: results };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return { success: false, error: `API request failed: ${errorMessage}` };
+        }
+    };
+    return { getSources };
+};
+const createSearXNGAPI = (instanceUrl, apiKey) => {
+    const config = {
+        instanceUrl: instanceUrl ?? process.env.SEARXNG_INSTANCE_URL,
+        apiKey: apiKey ?? process.env.SEARXNG_API_KEY,
+        timeout: 10000,
+    };
+    if (config.instanceUrl == null || config.instanceUrl === '') {
+        throw new Error('SEARXNG_INSTANCE_URL is required for SearXNG API');
+    }
+    const getSources = async (query, numResults = 8, storedLocation) => {
+        if (!query.trim()) {
+            return { success: false, error: 'Query cannot be empty' };
+        }
+        try {
+            // Ensure the instance URL ends with /search
+            if (config.instanceUrl == null || config.instanceUrl === '') {
+                return { success: false, error: 'Instance URL is not defined' };
+            }
+            let searchUrl = config.instanceUrl;
+            if (!searchUrl.endsWith('/search')) {
+                searchUrl = searchUrl.replace(/\/$/, '') + '/search';
+            }
+            // Prepare parameters for SearXNG
+            const params = {
+                q: query,
+                format: 'json',
+                pageno: 1,
+                categories: 'general',
+                language: 'all',
+                safesearch: 0,
+                engines: 'google,bing,duckduckgo',
+                max_results: Math.min(Math.max(1, numResults), 20),
+            };
+            if (storedLocation != null && storedLocation !== 'all') {
+                params.language = storedLocation;
+            }
+            const headers = {
+                'Content-Type': 'application/json',
+            };
+            if (config.apiKey != null && config.apiKey !== '') {
+                headers['X-API-Key'] = config.apiKey;
+            }
+            const response = await axios.get(searchUrl, {
+                headers,
+                params,
+                timeout: config.timeout,
+            });
+            const data = response.data;
+            // Transform SearXNG results to match SerperAPI format
+            const organicResults = (data.results ?? [])
+                .slice(0, numResults)
+                .map((result) => ({
+                title: result.title ?? '',
+                link: result.url ?? '',
+                snippet: result.content ?? '',
+                date: result.publishedDate ?? '',
+            }));
+            // Extract image results if available
+            const imageResults = (data.results ?? [])
+                .filter((result) => result.img_src)
+                .slice(0, 6)
+                .map((result) => ({
+                title: result.title ?? '',
+                imageUrl: result.img_src ?? '',
+            }));
+            // Format results to match SerperAPI structure
+            const results = {
+                organic: organicResults,
+                images: imageResults,
+                topStories: [],
+                // Use undefined instead of null for optional properties
+                relatedSearches: data.suggestions ?? [],
+            };
+            return { success: true, data: results };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return {
+                success: false,
+                error: `SearXNG API request failed: ${errorMessage}`,
+            };
+        }
+    };
+    return { getSources };
+};
+const createSearchAPI = (config) => {
+    const { searchProvider = 'serper', serperApiKey, searxngInstanceUrl, searxngApiKey, } = config;
+    if (searchProvider.toLowerCase() === 'serper') {
+        return createSerperAPI(serperApiKey);
+    }
+    else if (searchProvider.toLowerCase() === 'searxng') {
+        return createSearXNGAPI(searxngInstanceUrl, searxngApiKey);
+    }
+    else {
+        throw new Error(`Invalid search provider: ${searchProvider}. Must be 'serper' or 'searxng'`);
+    }
+};
+const createSourceProcessor = (config = {}, scraperInstance) => {
+    if (!scraperInstance) {
+        throw new Error('Firecrawl scraper instance is required');
+    }
+    const { topResults = 5, 
+    // strategies = ['no_extraction'],
+    // filterContent = true,
+    reranker, } = config;
+    const firecrawlScraper = scraperInstance;
+    const webScraper = {
+        scrapeMany: async ({ query, links, }) => {
+            console.log(`Scraping ${links.length} links with Firecrawl`);
+            const promises = [];
+            try {
+                for (const currentLink of links) {
+                    const promise = firecrawlScraper
+                        .scrapeUrl(currentLink, {})
+                        .then(([url, response]) => {
+                        const attribution = getAttribution(url, response.data?.metadata);
+                        if (response.success && response.data) {
+                            const content = firecrawlScraper.extractContent(response);
+                            return {
+                                url,
+                                attribution,
+                                content: chunker.cleanText(content),
+                            };
+                        }
+                        return {
+                            url,
+                            attribution,
+                            error: true,
+                            content: `Failed to scrape ${url}: ${response.error ?? 'Unknown error'}`,
+                        };
+                    })
+                        .then(async (result) => {
+                        try {
+                            if (result.error != null) {
+                                console.error(`Error scraping ${result.url}: ${result.content}`);
+                                return {
+                                    ...result,
+                                };
+                            }
+                            const highlights = await getHighlights({
+                                query,
+                                reranker,
+                                content: result.content,
+                            });
+                            return {
+                                ...result,
+                                highlights,
+                            };
+                        }
+                        catch (error) {
+                            console.error('Error processing scraped content:', error);
+                            return {
+                                ...result,
+                            };
+                        }
+                    })
+                        .catch((error) => {
+                        console.error(`Error scraping ${currentLink}:`, error);
+                        return {
+                            url: currentLink,
+                            error: true,
+                            content: `Failed to scrape ${currentLink}: ${error.message ?? 'Unknown error'}`,
+                        };
+                    });
+                    promises.push(promise);
+                }
+                return await Promise.all(promises);
+            }
+            catch (error) {
+                console.error('Error in scrapeMany:', error);
+                return [];
+            }
+        },
+    };
+    const fetchContents = async ({ links, query, target, onContentScraped, }) => {
+        const initialLinks = links.slice(0, target);
+        // const remainingLinks = links.slice(target).reverse();
+        const results = await webScraper.scrapeMany({ query, links: initialLinks });
+        for (const result of results) {
+            if (result.error === true) {
+                continue;
+            }
+            const { url, content, attribution, highlights } = result;
+            onContentScraped?.(url, {
+                content,
+                attribution,
+                highlights,
+            });
+        }
+    };
+    const processSources = async (result, numElements, query, proMode = false) => {
+        try {
+            if (!result.data) {
+                return {
+                    organic: [],
+                    topStories: [],
+                    images: [],
+                    relatedSearches: [],
+                };
+            }
+            else if (!result.data.organic) {
+                return result.data;
+            }
+            if (!proMode) {
+                const wikiSources = result.data.organic.filter((source) => source.link.includes('wikipedia.org'));
+                if (!wikiSources.length) {
+                    return result.data;
+                }
+                const wikiSourceMap = new Map();
+                wikiSourceMap.set(wikiSources[0].link, wikiSources[0]);
+                const onContentScraped = createSourceUpdateCallback(wikiSourceMap);
+                await fetchContents({
+                    query,
+                    target: 1,
+                    onContentScraped,
+                    links: [wikiSources[0].link],
+                });
+                for (let i = 0; i < result.data.organic.length; i++) {
+                    const source = result.data.organic[i];
+                    const updatedSource = wikiSourceMap.get(source.link);
+                    if (updatedSource) {
+                        result.data.organic[i] = {
+                            ...source,
+                            ...updatedSource,
+                        };
+                    }
+                }
+                return result.data;
+            }
+            const sourceMap = new Map();
+            const allLinks = [];
+            for (const source of result.data.organic) {
+                if (source.link) {
+                    allLinks.push(source.link);
+                    sourceMap.set(source.link, source);
+                }
+            }
+            if (allLinks.length === 0) {
+                return result.data;
+            }
+            const onContentScraped = createSourceUpdateCallback(sourceMap);
+            await fetchContents({
+                links: allLinks,
+                query,
+                onContentScraped,
+                target: numElements,
+            });
+            for (let i = 0; i < result.data.organic.length; i++) {
+                const source = result.data.organic[i];
+                const updatedSource = sourceMap.get(source.link);
+                if (updatedSource) {
+                    result.data.organic[i] = {
+                        ...source,
+                        ...updatedSource,
+                    };
+                }
+            }
+            const successfulSources = result.data.organic
+                .filter((source) => source.content != null && !source.content.startsWith('Failed'))
+                .slice(0, numElements);
+            if (successfulSources.length > 0) {
+                result.data.organic = successfulSources;
+            }
+            return result.data;
+        }
+        catch (error) {
+            console.error('Error in processSources:', error);
+            return {
+                organic: [],
+                topStories: [],
+                images: [],
+                relatedSearches: [],
+                ...result.data,
+                error: error instanceof Error ? error.message : String(error),
+            };
+        }
+    };
+    return {
+        processSources,
+        topResults,
+    };
+};
+
+export { createSearchAPI, createSourceProcessor };
+//# sourceMappingURL=search.mjs.map

package/dist/esm/tools/search/search.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.mjs","sources":["../../../../src/tools/search/search.ts"],"sourcesContent":["/* eslint-disable no-console */\nimport axios from 'axios';\nimport { RecursiveCharacterTextSplitter } from '@langchain/textsplitters';\nimport type * as t from './types';\nimport { getAttribution, FirecrawlScraper } from './firecrawl';\nimport { BaseReranker } from './rerankers';\n\nconst chunker = {\n  cleanText: (text: string): string => {\n    if (!text) return '';\n\n    /** Normalized all line endings to '\\n' */\n    const normalizedText = text.replace(/\\r\\n/g, '\\n').replace(/\\r/g, '\\n');\n\n    /** Handle multiple backslashes followed by newlines\n     * This replaces patterns like '\\\\\\\\\\\\n' with a single newline */\n    const fixedBackslashes = normalizedText.replace(/\\\\+\\n/g, '\\n');\n\n    /** Cleaned up consecutive newlines, tabs, and spaces around newlines */\n    const cleanedNewlines = fixedBackslashes.replace(/[\\t ]*\\n[\\t \\n]*/g, '\\n');\n\n    /** Cleaned up excessive spaces and tabs */\n    const cleanedSpaces = cleanedNewlines.replace(/[ \\t]+/g, ' ');\n\n    return cleanedSpaces.trim();\n  },\n  splitText: async (\n    text: string,\n    options?: {\n      chunkSize?: number;\n      chunkOverlap?: number;\n      separators?: string[];\n    }\n  ): Promise<string[]> => {\n    const chunkSize = options?.chunkSize ?? 150;\n    const chunkOverlap = options?.chunkOverlap ?? 50;\n    const separators = options?.separators || ['\\n\\n', '\\n'];\n\n    const splitter = new RecursiveCharacterTextSplitter({\n      separators,\n      chunkSize,\n      chunkOverlap,\n    });\n\n    return await splitter.splitText(text);\n  },\n\n  splitTexts: async (\n    texts: string[],\n    options?: {\n      chunkSize?: number;\n      chunkOverlap?: number;\n      separators?: string[];\n    }\n  ): Promise<string[][]> => {\n    // Split multiple texts\n    const promises = texts.map((text) =>\n      chunker.splitText(text, options).catch((error) => {\n        console.error('Error splitting text:', error);\n        return [text];\n      })\n    );\n    return Promise.all(promises);\n  },\n};\n\nconst createSourceUpdateCallback = (sourceMap: Map<string, t.ValidSource>) => {\n  return (link: string, update?: Partial<t.ValidSource>): void => {\n    const source = sourceMap.get(link);\n    if (source) {\n      sourceMap.set(link, {\n        ...source,\n        ...update,\n      });\n    }\n  };\n};\n\nconst getHighlights = async ({\n  query,\n  content,\n  reranker,\n  topResults = 5,\n}: {\n  content: string;\n  query: string;\n  reranker?: BaseReranker;\n  topResults?: number;\n}): Promise<t.Highlight[] | undefined> => {\n  if (!content) {\n    console.warn('No content provided for highlights');\n    return;\n  }\n  if (!reranker) {\n    console.warn('No reranker provided for highlights');\n    return;\n  }\n\n  try {\n    const documents = await chunker.splitText(content);\n    if (Array.isArray(documents)) {\n      return await reranker.rerank(query, documents, topResults);\n    } else {\n      console.error(\n        'Expected documents to be an array, got:',\n        typeof documents\n      );\n      return;\n    }\n  } catch (error) {\n    console.error('Error in content processing:', error);\n    return;\n  }\n};\n\nconst createSerperAPI = (\n  apiKey?: string\n): {\n  getSources: (\n    query: string,\n    numResults?: number,\n    storedLocation?: string\n  ) => Promise<t.SearchResult>;\n} => {\n  const config = {\n    apiKey: apiKey ?? process.env.SERPER_API_KEY,\n    apiUrl: 'https://google.serper.dev/search',\n    defaultLocation: 'us',\n    timeout: 10000,\n  };\n\n  if (config.apiKey == null || config.apiKey === '') {\n    throw new Error('SERPER_API_KEY is required for SerperAPI');\n  }\n\n  const getSources = async (\n    query: string,\n    numResults: number = 8,\n    storedLocation?: string\n  ): Promise<t.SearchResult> => {\n    if (!query.trim()) {\n      return { success: false, error: 'Query cannot be empty' };\n    }\n\n    try {\n      const searchLocation = (\n        storedLocation ?? config.defaultLocation\n      ).toLowerCase();\n\n      const payload = {\n        q: query,\n        num: Math.min(Math.max(1, numResults), 10),\n        gl: searchLocation,\n      };\n\n      const response = await axios.post(config.apiUrl, payload, {\n        headers: {\n          'X-API-KEY': config.apiKey,\n          'Content-Type': 'application/json',\n        },\n        timeout: config.timeout,\n      });\n\n      const data = response.data;\n      const results: t.SearchResultData = {\n        organic: data.organic,\n        images: data.images ?? [],\n        topStories: data.topStories ?? [],\n        knowledgeGraph: data.knowledgeGraph as t.KnowledgeGraphResult,\n        answerBox: data.answerBox as t.AnswerBoxResult,\n        peopleAlsoAsk: data.peopleAlsoAsk as t.PeopleAlsoAskResult[],\n        relatedSearches: data.relatedSearches as string[],\n      };\n\n      return { success: true, data: results };\n    } catch (error) {\n      const errorMessage =\n        error instanceof Error ? error.message : String(error);\n      return { success: false, error: `API request failed: ${errorMessage}` };\n    }\n  };\n\n  return { getSources };\n};\n\nconst createSearXNGAPI = (\n  instanceUrl?: string,\n  apiKey?: string\n): {\n  getSources: (\n    query: string,\n    numResults?: number,\n    storedLocation?: string\n  ) => Promise<t.SearchResult>;\n} => {\n  const config = {\n    instanceUrl: instanceUrl ?? process.env.SEARXNG_INSTANCE_URL,\n    apiKey: apiKey ?? process.env.SEARXNG_API_KEY,\n    defaultLocation: 'all',\n    timeout: 10000,\n  };\n\n  if (config.instanceUrl == null || config.instanceUrl === '') {\n    throw new Error('SEARXNG_INSTANCE_URL is required for SearXNG API');\n  }\n\n  const getSources = async (\n    query: string,\n    numResults: number = 8,\n    storedLocation?: string\n  ): Promise<t.SearchResult> => {\n    if (!query.trim()) {\n      return { success: false, error: 'Query cannot be empty' };\n    }\n\n    try {\n      // Ensure the instance URL ends with /search\n      if (config.instanceUrl == null || config.instanceUrl === '') {\n        return { success: false, error: 'Instance URL is not defined' };\n      }\n\n      let searchUrl = config.instanceUrl;\n      if (!searchUrl.endsWith('/search')) {\n        searchUrl = searchUrl.replace(/\\/$/, '') + '/search';\n      }\n\n      // Prepare parameters for SearXNG\n      const params: Record<string, string | number> = {\n        q: query,\n        format: 'json',\n        pageno: 1,\n        categories: 'general',\n        language: 'all',\n        safesearch: 0,\n        engines: 'google,bing,duckduckgo',\n        max_results: Math.min(Math.max(1, numResults), 20),\n      };\n\n      if (storedLocation != null && storedLocation !== 'all') {\n        params.language = storedLocation;\n      }\n\n      const headers: Record<string, string> = {\n        'Content-Type': 'application/json',\n      };\n\n      if (config.apiKey != null && config.apiKey !== '') {\n        headers['X-API-Key'] = config.apiKey;\n      }\n\n      const response = await axios.get(searchUrl, {\n        headers,\n        params,\n        timeout: config.timeout,\n      });\n\n      const data = response.data;\n\n      // Transform SearXNG results to match SerperAPI format\n      const organicResults = (data.results ?? [])\n        .slice(0, numResults)\n        .map((result: t.SearXNGResult) => ({\n          title: result.title ?? '',\n          link: result.url ?? '',\n          snippet: result.content ?? '',\n          date: result.publishedDate ?? '',\n        }));\n\n      // Extract image results if available\n      const imageResults = (data.results ?? [])\n        .filter((result: t.SearXNGResult) => result.img_src)\n        .slice(0, 6)\n        .map((result: t.SearXNGResult) => ({\n          title: result.title ?? '',\n          imageUrl: result.img_src ?? '',\n        }));\n\n      // Format results to match SerperAPI structure\n      const results: t.SearchResultData = {\n        organic: organicResults,\n        images: imageResults,\n        topStories: [],\n        // Use undefined instead of null for optional properties\n        relatedSearches: data.suggestions ?? [],\n      };\n\n      return { success: true, data: results };\n    } catch (error) {\n      const errorMessage =\n        error instanceof Error ? error.message : String(error);\n      return {\n        success: false,\n        error: `SearXNG API request failed: ${errorMessage}`,\n      };\n    }\n  };\n\n  return { getSources };\n};\n\nexport const createSearchAPI = (\n  config: t.SearchConfig\n): {\n  getSources: (\n    query: string,\n    numResults?: number,\n    storedLocation?: string\n  ) => Promise<t.SearchResult>;\n} => {\n  const {\n    searchProvider = 'serper',\n    serperApiKey,\n    searxngInstanceUrl,\n    searxngApiKey,\n  } = config;\n\n  if (searchProvider.toLowerCase() === 'serper') {\n    return createSerperAPI(serperApiKey);\n  } else if (searchProvider.toLowerCase() === 'searxng') {\n    return createSearXNGAPI(searxngInstanceUrl, searxngApiKey);\n  } else {\n    throw new Error(\n      `Invalid search provider: ${searchProvider}. Must be 'serper' or 'searxng'`\n    );\n  }\n};\n\nexport const createSourceProcessor = (\n  config: t.ProcessSourcesConfig = {},\n  scraperInstance?: FirecrawlScraper\n): {\n  processSources: (\n    result: t.SearchResult,\n    numElements: number,\n    query: string,\n    proMode?: boolean\n  ) => Promise<t.SearchResultData>;\n  topResults: number;\n} => {\n  if (!scraperInstance) {\n    throw new Error('Firecrawl scraper instance is required');\n  }\n  const {\n    topResults = 5,\n    // strategies = ['no_extraction'],\n    // filterContent = true,\n    reranker,\n  } = config;\n\n  const firecrawlScraper = scraperInstance;\n\n  const webScraper = {\n    scrapeMany: async ({\n      query,\n      links,\n    }: {\n      query: string;\n      links: string[];\n    }): Promise<Array<t.ScrapeResult>> => {\n      console.log(`Scraping ${links.length} links with Firecrawl`);\n      const promises: Array<Promise<t.ScrapeResult>> = [];\n      try {\n        for (const currentLink of links) {\n          const promise: Promise<t.ScrapeResult> = firecrawlScraper\n            .scrapeUrl(currentLink, {})\n            .then(([url, response]) => {\n              const attribution = getAttribution(url, response.data?.metadata);\n              if (response.success && response.data) {\n                const content = firecrawlScraper.extractContent(response);\n                return {\n                  url,\n                  attribution,\n                  content: chunker.cleanText(content),\n                };\n              }\n\n              return {\n                url,\n                attribution,\n                error: true,\n                content: `Failed to scrape ${url}: ${response.error ?? 'Unknown error'}`,\n              };\n            })\n            .then(async (result) => {\n              try {\n                if (result.error != null) {\n                  console.error(\n                    `Error scraping ${result.url}: ${result.content}`\n                  );\n                  return {\n                    ...result,\n                  };\n                }\n                const highlights = await getHighlights({\n                  query,\n                  reranker,\n                  content: result.content,\n                });\n                return {\n                  ...result,\n                  highlights,\n                };\n              } catch (error) {\n                console.error('Error processing scraped content:', error);\n                return {\n                  ...result,\n                };\n              }\n            })\n            .catch((error) => {\n              console.error(`Error scraping ${currentLink}:`, error);\n              return {\n                url: currentLink,\n                error: true,\n                content: `Failed to scrape ${currentLink}: ${error.message ?? 'Unknown error'}`,\n              };\n            });\n          promises.push(promise);\n        }\n        return await Promise.all(promises);\n      } catch (error) {\n        console.error('Error in scrapeMany:', error);\n        return [];\n      }\n    },\n  };\n\n  const fetchContents = async ({\n    links,\n    query,\n    target,\n    onContentScraped,\n  }: {\n    links: string[];\n    query: string;\n    target: number;\n    onContentScraped?: (link: string, update?: Partial<t.ValidSource>) => void;\n  }): Promise<void> => {\n    const initialLinks = links.slice(0, target);\n    // const remainingLinks = links.slice(target).reverse();\n    const results = await webScraper.scrapeMany({ query, links: initialLinks });\n    for (const result of results) {\n      if (result.error === true) {\n        continue;\n      }\n      const { url, content, attribution, highlights } = result;\n      onContentScraped?.(url, {\n        content,\n        attribution,\n        highlights,\n      });\n    }\n  };\n\n  const processSources = async (\n    result: t.SearchResult,\n    numElements: number,\n    query: string,\n    proMode: boolean = false\n  ): Promise<t.SearchResultData> => {\n    try {\n      if (!result.data) {\n        return {\n          organic: [],\n          topStories: [],\n          images: [],\n          relatedSearches: [],\n        };\n      } else if (!result.data.organic) {\n        return result.data;\n      }\n\n      if (!proMode) {\n        const wikiSources = result.data.organic.filter((source) =>\n          source.link.includes('wikipedia.org')\n        );\n\n        if (!wikiSources.length) {\n          return result.data;\n        }\n\n        const wikiSourceMap = new Map<string, t.ValidSource>();\n        wikiSourceMap.set(wikiSources[0].link, wikiSources[0]);\n        const onContentScraped = createSourceUpdateCallback(wikiSourceMap);\n        await fetchContents({\n          query,\n          target: 1,\n          onContentScraped,\n          links: [wikiSources[0].link],\n        });\n\n        for (let i = 0; i < result.data.organic.length; i++) {\n          const source = result.data.organic[i];\n          const updatedSource = wikiSourceMap.get(source.link);\n          if (updatedSource) {\n            result.data.organic[i] = {\n              ...source,\n              ...updatedSource,\n            };\n          }\n        }\n\n        return result.data;\n      }\n\n      const sourceMap = new Map<string, t.ValidSource>();\n      const allLinks: string[] = [];\n\n      for (const source of result.data.organic) {\n        if (source.link) {\n          allLinks.push(source.link);\n          sourceMap.set(source.link, source);\n        }\n      }\n\n      if (allLinks.length === 0) {\n        return result.data;\n      }\n\n      const onContentScraped = createSourceUpdateCallback(sourceMap);\n      await fetchContents({\n        links: allLinks,\n        query,\n        onContentScraped,\n        target: numElements,\n      });\n\n      for (let i = 0; i < result.data.organic.length; i++) {\n        const source = result.data.organic[i];\n        const updatedSource = sourceMap.get(source.link);\n        if (updatedSource) {\n          result.data.organic[i] = {\n            ...source,\n            ...updatedSource,\n          };\n        }\n      }\n\n      const successfulSources = result.data.organic\n        .filter(\n          (source) =>\n            source.content != null && !source.content.startsWith('Failed')\n        )\n        .slice(0, numElements);\n\n      if (successfulSources.length > 0) {\n        result.data.organic = successfulSources;\n      }\n      return result.data;\n    } catch (error) {\n      console.error('Error in processSources:', error);\n      return {\n        organic: [],\n        topStories: [],\n        images: [],\n        relatedSearches: [],\n        ...result.data,\n        error: error instanceof Error ? error.message : String(error),\n      };\n    }\n  };\n\n  return {\n    processSources,\n    topResults,\n  };\n};\n"],"names":[],"mappings":";;;;AAAA;AAOA,MAAM,OAAO,GAAG;AACd,IAAA,SAAS,EAAE,CAAC,IAAY,KAAY;AAClC,QAAA,IAAI,CAAC,IAAI;AAAE,YAAA,OAAO,EAAE;;AAGpB,QAAA,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC;AAEvE;AACiE;QACjE,MAAM,gBAAgB,GAAG,cAAc,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC;;QAG/D,MAAM,eAAe,GAAG,gBAAgB,CAAC,OAAO,CAAC,mBAAmB,EAAE,IAAI,CAAC;;QAG3E,MAAM,aAAa,GAAG,eAAe,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC;AAE7D,QAAA,OAAO,aAAa,CAAC,IAAI,EAAE;KAC5B;AACD,IAAA,SAAS,EAAE,OACT,IAAY,EACZ,OAIC,KACoB;AACrB,QAAA,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,GAAG;AAC3C,QAAA,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,EAAE;QAChD,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC;AAExD,QAAA,MAAM,QAAQ,GAAG,IAAI,8BAA8B,CAAC;YAClD,UAAU;YACV,SAAS;YACT,YAAY;AACb,SAAA,CAAC;AAEF,QAAA,OAAO,MAAM,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC;KACtC;AAED,IAAA,UAAU,EAAE,OACV,KAAe,EACf,OAIC,KACsB;;QAEvB,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,KAC9B,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,KAAI;AAC/C,YAAA,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC;SACd,CAAC,CACH;AACD,QAAA,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;KAC7B;CACF;AAED,MAAM,0BAA0B,GAAG,CAAC,SAAqC,KAAI;AAC3E,IAAA,OAAO,CAAC,IAAY,EAAE,MAA+B,KAAU;QAC7D,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC;QAClC,IAAI,MAAM,EAAE;AACV,YAAA,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE;AAClB,gBAAA,GAAG,MAAM;AACT,gBAAA,GAAG,MAAM;AACV,aAAA,CAAC;;AAEN,KAAC;AACH,CAAC;AAED,MAAM,aAAa,GAAG,OAAO,EAC3B,KAAK,EACL,OAAO,EACP,QAAQ,EACR,UAAU,GAAG,CAAC,GAMf,KAAwC;IACvC,IAAI,CAAC,OAAO,EAAE;AACZ,QAAA,OAAO,CAAC,IAAI,CAAC,oCAAoC,CAAC;QAClD;;IAEF,IAAI,CAAC,QAAQ,EAAE;AACb,QAAA,OAAO,CAAC,IAAI,CAAC,qCAAqC,CAAC;QACnD;;AAGF,IAAA,IAAI;QACF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;AAClD,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;YAC5B,OAAO,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,UAAU,CAAC;;aACrD;YACL,OAAO,CAAC,KAAK,CACX,yCAAyC,EACzC,OAAO,SAAS,CACjB;YACD;;;IAEF,OAAO,KAAK,EAAE;AACd,QAAA,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC;QACpD;;AAEJ,CAAC;AAED,MAAM,eAAe,GAAG,CACtB,MAAe,KAOb;AACF,IAAA,MAAM,MAAM,GAAG;AACb,QAAA,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;AAC5C,QAAA,MAAM,EAAE,kCAAkC;AAC1C,QAAA,eAAe,EAAE,IAAI;AACrB,QAAA,OAAO,EAAE,KAAK;KACf;AAED,IAAA,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI,IAAI,MAAM,CAAC,MAAM,KAAK,EAAE,EAAE;AACjD,QAAA,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC;;IAG7D,MAAM,UAAU,GAAG,OACjB,KAAa,EACb,UAAA,GAAqB,CAAC,EACtB,cAAuB,KACI;AAC3B,QAAA,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE;YACjB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,uBAAuB,EAAE;;AAG3D,QAAA,IAAI;AACF,YAAA,MAAM,cAAc,GAAG,CACrB,cAAc,IAAI,MAAM,CAAC,eAAe,EACxC,WAAW,EAAE;AAEf,YAAA,MAAM,OAAO,GAAG;AACd,gBAAA,CAAC,EAAE,KAAK;AACR,gBAAA,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC;AAC1C,gBAAA,EAAE,EAAE,cAAc;aACnB;AAED,YAAA,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE;AACxD,gBAAA,OAAO,EAAE;oBACP,WAAW,EAAE,MAAM,CAAC,MAAM;AAC1B,oBAAA,cAAc,EAAE,kBAAkB;AACnC,iBAAA;gBACD,OAAO,EAAE,MAAM,CAAC,OAAO;AACxB,aAAA,CAAC;AAEF,YAAA,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI;AAC1B,YAAA,MAAM,OAAO,GAAuB;gBAClC,OAAO,EAAE,IAAI,CAAC,OAAO;AACrB,gBAAA,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;AACzB,gBAAA,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;gBACjC,cAAc,EAAE,IAAI,CAAC,cAAwC;gBAC7D,SAAS,EAAE,IAAI,CAAC,SAA8B;gBAC9C,aAAa,EAAE,IAAI,CAAC,aAAwC;gBAC5D,eAAe,EAAE,IAAI,CAAC,eAA2B;aAClD;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE;;QACvC,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;YACxD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAuB,oBAAA,EAAA,YAAY,CAAE,CAAA,EAAE;;AAE3E,KAAC;IAED,OAAO,EAAE,UAAU,EAAE;AACvB,CAAC;AAED,MAAM,gBAAgB,GAAG,CACvB,WAAoB,EACpB,MAAe,KAOb;AACF,IAAA,MAAM,MAAM,GAAG;AACb,QAAA,WAAW,EAAE,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB;AAC5D,QAAA,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe;AAC7C,QACA,OAAO,EAAE,KAAK;KACf;AAED,IAAA,IAAI,MAAM,CAAC,WAAW,IAAI,IAAI,IAAI,MAAM,CAAC,WAAW,KAAK,EAAE,EAAE;AAC3D,QAAA,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC;;IAGrE,MAAM,UAAU,GAAG,OACjB,KAAa,EACb,UAAA,GAAqB,CAAC,EACtB,cAAuB,KACI;AAC3B,QAAA,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE;YACjB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,uBAAuB,EAAE;;AAG3D,QAAA,IAAI;;AAEF,YAAA,IAAI,MAAM,CAAC,WAAW,IAAI,IAAI,IAAI,MAAM,CAAC,WAAW,KAAK,EAAE,EAAE;gBAC3D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,6BAA6B,EAAE;;AAGjE,YAAA,IAAI,SAAS,GAAG,MAAM,CAAC,WAAW;YAClC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;gBAClC,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,SAAS;;;AAItD,YAAA,MAAM,MAAM,GAAoC;AAC9C,gBAAA,CAAC,EAAE,KAAK;AACR,gBAAA,MAAM,EAAE,MAAM;AACd,gBAAA,MAAM,EAAE,CAAC;AACT,gBAAA,UAAU,EAAE,SAAS;AACrB,gBAAA,QAAQ,EAAE,KAAK;AACf,gBAAA,UAAU,EAAE,CAAC;AACb,gBAAA,OAAO,EAAE,wBAAwB;AACjC,gBAAA,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC;aACnD;YAED,IAAI,cAAc,IAAI,IAAI,IAAI,cAAc,KAAK,KAAK,EAAE;AACtD,gBAAA,MAAM,CAAC,QAAQ,GAAG,cAAc;;AAGlC,YAAA,MAAM,OAAO,GAA2B;AACtC,gBAAA,cAAc,EAAE,kBAAkB;aACnC;AAED,YAAA,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI,IAAI,MAAM,CAAC,MAAM,KAAK,EAAE,EAAE;AACjD,gBAAA,OAAO,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC,MAAM;;YAGtC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE;gBAC1C,OAAO;gBACP,MAAM;gBACN,OAAO,EAAE,MAAM,CAAC,OAAO;AACxB,aAAA,CAAC;AAEF,YAAA,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI;;YAG1B,MAAM,cAAc,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE;AACvC,iBAAA,KAAK,CAAC,CAAC,EAAE,UAAU;AACnB,iBAAA,GAAG,CAAC,CAAC,MAAuB,MAAM;AACjC,gBAAA,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;AACzB,gBAAA,IAAI,EAAE,MAAM,CAAC,GAAG,IAAI,EAAE;AACtB,gBAAA,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;AAC7B,gBAAA,IAAI,EAAE,MAAM,CAAC,aAAa,IAAI,EAAE;AACjC,aAAA,CAAC,CAAC;;YAGL,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE;iBACrC,MAAM,CAAC,CAAC,MAAuB,KAAK,MAAM,CAAC,OAAO;AAClD,iBAAA,KAAK,CAAC,CAAC,EAAE,CAAC;AACV,iBAAA,GAAG,CAAC,CAAC,MAAuB,MAAM;AACjC,gBAAA,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;AACzB,gBAAA,QAAQ,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;AAC/B,aAAA,CAAC,CAAC;;AAGL,YAAA,MAAM,OAAO,GAAuB;AAClC,gBAAA,OAAO,EAAE,cAAc;AACvB,gBAAA,MAAM,EAAE,YAAY;AACpB,gBAAA,UAAU,EAAE,EAAE;;AAEd,gBAAA,eAAe,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;aACxC;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE;;QACvC,OAAO,KAAK,EAAE;AACd,YAAA,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;YACxD,OAAO;AACL,gBAAA,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,CAA+B,4BAAA,EAAA,YAAY,CAAE,CAAA;aACrD;;AAEL,KAAC;IAED,OAAO,EAAE,UAAU,EAAE;AACvB,CAAC;AAEY,MAAA,eAAe,GAAG,CAC7B,MAAsB,KAOpB;AACF,IAAA,MAAM,EACJ,cAAc,GAAG,QAAQ,EACzB,YAAY,EACZ,kBAAkB,EAClB,aAAa,GACd,GAAG,MAAM;AAEV,IAAA,IAAI,cAAc,CAAC,WAAW,EAAE,KAAK,QAAQ,EAAE;AAC7C,QAAA,OAAO,eAAe,CAAC,YAAY,CAAC;;AAC/B,SAAA,IAAI,cAAc,CAAC,WAAW,EAAE,KAAK,SAAS,EAAE;AACrD,QAAA,OAAO,gBAAgB,CAAC,kBAAkB,EAAE,aAAa,CAAC;;SACrD;AACL,QAAA,MAAM,IAAI,KAAK,CACb,4BAA4B,cAAc,CAAA,+BAAA,CAAiC,CAC5E;;AAEL;AAEa,MAAA,qBAAqB,GAAG,CACnC,SAAiC,EAAE,EACnC,eAAkC,KAShC;IACF,IAAI,CAAC,eAAe,EAAE;AACpB,QAAA,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC;;IAE3D,MAAM,EACJ,UAAU,GAAG,CAAC;;;IAGd,QAAQ,GACT,GAAG,MAAM;IAEV,MAAM,gBAAgB,GAAG,eAAe;AAExC,IAAA,MAAM,UAAU,GAAG;QACjB,UAAU,EAAE,OAAO,EACjB,KAAK,EACL,KAAK,GAIN,KAAoC;YACnC,OAAO,CAAC,GAAG,CAAC,CAAA,SAAA,EAAY,KAAK,CAAC,MAAM,CAAuB,qBAAA,CAAA,CAAC;YAC5D,MAAM,QAAQ,GAAmC,EAAE;AACnD,YAAA,IAAI;AACF,gBAAA,KAAK,MAAM,WAAW,IAAI,KAAK,EAAE;oBAC/B,MAAM,OAAO,GAA4B;AACtC,yBAAA,SAAS,CAAC,WAAW,EAAE,EAAE;yBACzB,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,QAAQ,CAAC,KAAI;AACxB,wBAAA,MAAM,WAAW,GAAG,cAAc,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;wBAChE,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE;4BACrC,MAAM,OAAO,GAAG,gBAAgB,CAAC,cAAc,CAAC,QAAQ,CAAC;4BACzD,OAAO;gCACL,GAAG;gCACH,WAAW;AACX,gCAAA,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;6BACpC;;wBAGH,OAAO;4BACL,GAAG;4BACH,WAAW;AACX,4BAAA,KAAK,EAAE,IAAI;4BACX,OAAO,EAAE,oBAAoB,GAAG,CAAA,EAAA,EAAK,QAAQ,CAAC,KAAK,IAAI,eAAe,CAAE,CAAA;yBACzE;AACH,qBAAC;AACA,yBAAA,IAAI,CAAC,OAAO,MAAM,KAAI;AACrB,wBAAA,IAAI;AACF,4BAAA,IAAI,MAAM,CAAC,KAAK,IAAI,IAAI,EAAE;AACxB,gCAAA,OAAO,CAAC,KAAK,CACX,CAAA,eAAA,EAAkB,MAAM,CAAC,GAAG,CAAA,EAAA,EAAK,MAAM,CAAC,OAAO,CAAA,CAAE,CAClD;gCACD,OAAO;AACL,oCAAA,GAAG,MAAM;iCACV;;AAEH,4BAAA,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC;gCACrC,KAAK;gCACL,QAAQ;gCACR,OAAO,EAAE,MAAM,CAAC,OAAO;AACxB,6BAAA,CAAC;4BACF,OAAO;AACL,gCAAA,GAAG,MAAM;gCACT,UAAU;6BACX;;wBACD,OAAO,KAAK,EAAE;AACd,4BAAA,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC;4BACzD,OAAO;AACL,gCAAA,GAAG,MAAM;6BACV;;AAEL,qBAAC;AACA,yBAAA,KAAK,CAAC,CAAC,KAAK,KAAI;wBACf,OAAO,CAAC,KAAK,CAAC,CAAA,eAAA,EAAkB,WAAW,CAAG,CAAA,CAAA,EAAE,KAAK,CAAC;wBACtD,OAAO;AACL,4BAAA,GAAG,EAAE,WAAW;AAChB,4BAAA,KAAK,EAAE,IAAI;4BACX,OAAO,EAAE,oBAAoB,WAAW,CAAA,EAAA,EAAK,KAAK,CAAC,OAAO,IAAI,eAAe,CAAE,CAAA;yBAChF;AACH,qBAAC,CAAC;AACJ,oBAAA,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;;AAExB,gBAAA,OAAO,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;;YAClC,OAAO,KAAK,EAAE;AACd,gBAAA,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC;AAC5C,gBAAA,OAAO,EAAE;;SAEZ;KACF;AAED,IAAA,MAAM,aAAa,GAAG,OAAO,EAC3B,KAAK,EACL,KAAK,EACL,MAAM,EACN,gBAAgB,GAMjB,KAAmB;QAClB,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC;;AAE3C,QAAA,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;AAC3E,QAAA,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE;AAC5B,YAAA,IAAI,MAAM,CAAC,KAAK,KAAK,IAAI,EAAE;gBACzB;;YAEF,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,MAAM;YACxD,gBAAgB,GAAG,GAAG,EAAE;gBACtB,OAAO;gBACP,WAAW;gBACX,UAAU;AACX,aAAA,CAAC;;AAEN,KAAC;AAED,IAAA,MAAM,cAAc,GAAG,OACrB,MAAsB,EACtB,WAAmB,EACnB,KAAa,EACb,OAAmB,GAAA,KAAK,KACO;AAC/B,QAAA,IAAI;AACF,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;gBAChB,OAAO;AACL,oBAAA,OAAO,EAAE,EAAE;AACX,oBAAA,UAAU,EAAE,EAAE;AACd,oBAAA,MAAM,EAAE,EAAE;AACV,oBAAA,eAAe,EAAE,EAAE;iBACpB;;AACI,iBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE;gBAC/B,OAAO,MAAM,CAAC,IAAI;;YAGpB,IAAI,CAAC,OAAO,EAAE;gBACZ,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,KACpD,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CACtC;AAED,gBAAA,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE;oBACvB,OAAO,MAAM,CAAC,IAAI;;AAGpB,gBAAA,MAAM,aAAa,GAAG,IAAI,GAAG,EAAyB;AACtD,gBAAA,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;AACtD,gBAAA,MAAM,gBAAgB,GAAG,0BAA0B,CAAC,aAAa,CAAC;AAClE,gBAAA,MAAM,aAAa,CAAC;oBAClB,KAAK;AACL,oBAAA,MAAM,EAAE,CAAC;oBACT,gBAAgB;oBAChB,KAAK,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC7B,iBAAA,CAAC;AAEF,gBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;oBACnD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;oBACrC,MAAM,aAAa,GAAG,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;oBACpD,IAAI,aAAa,EAAE;AACjB,wBAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;AACvB,4BAAA,GAAG,MAAM;AACT,4BAAA,GAAG,aAAa;yBACjB;;;gBAIL,OAAO,MAAM,CAAC,IAAI;;AAGpB,YAAA,MAAM,SAAS,GAAG,IAAI,GAAG,EAAyB;YAClD,MAAM,QAAQ,GAAa,EAAE;YAE7B,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE;AACxC,gBAAA,IAAI,MAAM,CAAC,IAAI,EAAE;AACf,oBAAA,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBAC1B,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC;;;AAItC,YAAA,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;gBACzB,OAAO,MAAM,CAAC,IAAI;;AAGpB,YAAA,MAAM,gBAAgB,GAAG,0BAA0B,CAAC,SAAS,CAAC;AAC9D,YAAA,MAAM,aAAa,CAAC;AAClB,gBAAA,KAAK,EAAE,QAAQ;gBACf,KAAK;gBACL,gBAAgB;AAChB,gBAAA,MAAM,EAAE,WAAW;AACpB,aAAA,CAAC;AAEF,YAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBACnD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;gBACrC,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;gBAChD,IAAI,aAAa,EAAE;AACjB,oBAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;AACvB,wBAAA,GAAG,MAAM;AACT,wBAAA,GAAG,aAAa;qBACjB;;;AAIL,YAAA,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC;iBACnC,MAAM,CACL,CAAC,MAAM,KACL,MAAM,CAAC,OAAO,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;AAEjE,iBAAA,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC;AAExB,YAAA,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;AAChC,gBAAA,MAAM,CAAC,IAAI,CAAC,OAAO,GAAG,iBAAiB;;YAEzC,OAAO,MAAM,CAAC,IAAI;;QAClB,OAAO,KAAK,EAAE;AACd,YAAA,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC;YAChD,OAAO;AACL,gBAAA,OAAO,EAAE,EAAE;AACX,gBAAA,UAAU,EAAE,EAAE;AACd,gBAAA,MAAM,EAAE,EAAE;AACV,gBAAA,eAAe,EAAE,EAAE;gBACnB,GAAG,MAAM,CAAC,IAAI;AACd,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;aAC9D;;AAEL,KAAC;IAED,OAAO;QACL,cAAc;QACd,UAAU;KACX;AACH;;;;"}

package/dist/esm/tools/search/tool.mjs

@@ -0,0 +1,101 @@
+import { z } from 'zod';
+import { tool } from '@langchain/core/tools';
+import { createSearchAPI, createSourceProcessor } from './search.mjs';
+import { createFirecrawlScraper } from './firecrawl.mjs';
+import { expandHighlights } from './highlights.mjs';
+import { formatResultsForLLM } from './format.mjs';
+import { createReranker } from './rerankers.mjs';
+import { Constants } from '../../common/enum.mjs';
+
+/* eslint-disable no-console */
+const SearchToolSchema = z.object({
+    query: z
+        .string()
+        .describe('The search query string that specifies what should be searched for.'),
+});
+const createSearchTool = (config = {}) => {
+    const { searchProvider = 'serper', serperApiKey, searxngInstanceUrl, searxngApiKey, rerankerType = 'cohere', topResults = 5, strategies = ['no_extraction'], filterContent = true, firecrawlApiKey, firecrawlApiUrl, firecrawlFormats = ['markdown', 'html'], jinaApiKey, cohereApiKey, onSearchResults: _onSearchResults, } = config;
+    const searchAPI = createSearchAPI({
+        searchProvider,
+        serperApiKey,
+        searxngInstanceUrl,
+        searxngApiKey,
+    });
+    const firecrawlScraper = createFirecrawlScraper({
+        apiKey: firecrawlApiKey ?? process.env.FIRECRAWL_API_KEY,
+        apiUrl: firecrawlApiUrl,
+        formats: firecrawlFormats,
+    });
+    const selectedReranker = createReranker({
+        rerankerType,
+        jinaApiKey,
+        cohereApiKey,
+    });
+    if (!selectedReranker) {
+        console.warn('No reranker selected. Using default ranking.');
+    }
+    const sourceProcessor = createSourceProcessor({
+        reranker: selectedReranker,
+        topResults}, firecrawlScraper);
+    const search = async ({ query, proMode = true, maxSources = 5, onSearchResults, }) => {
+        try {
+            const sources = await searchAPI.getSources(query);
+            onSearchResults?.(sources);
+            if (!sources.success) {
+                throw new Error(sources.error ?? 'Search failed');
+            }
+            const processedSources = await sourceProcessor.processSources(sources, maxSources, query, proMode);
+            return expandHighlights(processedSources);
+        }
+        catch (error) {
+            console.error('Error in search:', error);
+            return {
+                organic: [],
+                topStories: [],
+                images: [],
+                relatedSearches: [],
+                error: error instanceof Error ? error.message : String(error),
+            };
+        }
+    };
+    return tool(async ({ query }, runnableConfig) => {
+        const searchResult = await search({
+            query,
+            onSearchResults: _onSearchResults
+                ? (result) => {
+                    _onSearchResults(result, runnableConfig);
+                }
+                : undefined,
+        });
+        const output = formatResultsForLLM(searchResult);
+        return [output, { [Constants.WEB_SEARCH]: { ...searchResult } }];
+    }, {
+        name: Constants.WEB_SEARCH,
+        description: `
+Real-time search. Results have required unique citation anchors.
+
+Anchors:
+- \\ue202turn0searchN (web), \\ue202turn0newsN (news), \\ue202turn0imageN (image)
+
+Special Markers:
+- \\ue203...\\ue204 — mark start/end of cited span
+- \\ue200...\\ue201 — composite/group block (e.g. \\ue200cite\\ue202turn0search1\\ue202turn0news2\\ue201)
+- \\ue206 — marks grouped/summary citation areas
+
+**CITE EVERY NON-OBVIOUS FACT/QUOTE:**
+Insert the anchor marker(s) immediately after the statement:
+- "Pure functions produce same output \\ue202turn0search0."
+- Multiple: "Benefits \\ue202turn0search0\\ue202turn0news0."
+- Span: \\ue203Key: first-class functions\\ue204\\ue202turn0news1
+- Group: "Functional languages."\\ue206 or \\ue200cite\\ue202turn0search0\\ue202turn0news1\\ue201
+- Image: "See photo \\ue202turn0image0."
+
+**NEVER use markdown links, [1], or footnotes. CITE ONLY with anchors provided.**
+`.trim(),
+        schema: SearchToolSchema,
+        responseFormat: Constants.CONTENT_AND_ARTIFACT,
+    });
+};
+
+export { createSearchTool };
+//# sourceMappingURL=tool.mjs.map

package/dist/esm/tools/search/tool.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.mjs","sources":["../../../../src/tools/search/tool.ts"],"sourcesContent":["/* eslint-disable no-console */\nimport { z } from 'zod';\nimport { tool, DynamicStructuredTool } from '@langchain/core/tools';\nimport type * as t from './types';\nimport { createSearchAPI, createSourceProcessor } from './search';\nimport { createFirecrawlScraper } from './firecrawl';\nimport { expandHighlights } from './highlights';\nimport { formatResultsForLLM } from './format';\nimport { createReranker } from './rerankers';\nimport { Constants } from '@/common';\n\nconst SearchToolSchema = z.object({\n  query: z\n    .string()\n    .describe(\n      'The search query string that specifies what should be searched for.'\n    ),\n});\n\nexport const createSearchTool = (\n  config: t.SearchToolConfig = {}\n): DynamicStructuredTool<typeof SearchToolSchema> => {\n  const {\n    searchProvider = 'serper',\n    serperApiKey,\n    searxngInstanceUrl,\n    searxngApiKey,\n    rerankerType = 'cohere',\n    topResults = 5,\n    strategies = ['no_extraction'],\n    filterContent = true,\n    firecrawlApiKey,\n    firecrawlApiUrl,\n    firecrawlFormats = ['markdown', 'html'],\n    jinaApiKey,\n    cohereApiKey,\n    onSearchResults: _onSearchResults,\n  } = config;\n\n  const searchAPI = createSearchAPI({\n    searchProvider,\n    serperApiKey,\n    searxngInstanceUrl,\n    searxngApiKey,\n  });\n\n  const firecrawlScraper = createFirecrawlScraper({\n    apiKey: firecrawlApiKey ?? process.env.FIRECRAWL_API_KEY,\n    apiUrl: firecrawlApiUrl,\n    formats: firecrawlFormats,\n  });\n\n  const selectedReranker = createReranker({\n    rerankerType,\n    jinaApiKey,\n    cohereApiKey,\n  });\n\n  if (!selectedReranker) {\n    console.warn('No reranker selected. Using default ranking.');\n  }\n\n  const sourceProcessor = createSourceProcessor(\n    {\n      reranker: selectedReranker,\n      topResults,\n      strategies,\n      filterContent,\n    },\n    firecrawlScraper\n  );\n\n  const search = async ({\n    query,\n    proMode = true,\n    maxSources = 5,\n    onSearchResults,\n  }: {\n    query: string;\n    proMode?: boolean;\n    maxSources?: number;\n    onSearchResults?: (sources: t.SearchResult) => void;\n  }): Promise<t.SearchResultData> => {\n    try {\n      const sources = await searchAPI.getSources(query);\n      onSearchResults?.(sources);\n\n      if (!sources.success) {\n        throw new Error(sources.error ?? 'Search failed');\n      }\n\n      const processedSources = await sourceProcessor.processSources(\n        sources,\n        maxSources,\n        query,\n        proMode\n      );\n      return expandHighlights(processedSources);\n    } catch (error) {\n      console.error('Error in search:', error);\n      return {\n        organic: [],\n        topStories: [],\n        images: [],\n        relatedSearches: [],\n        error: error instanceof Error ? error.message : String(error),\n      };\n    }\n  };\n\n  return tool<typeof SearchToolSchema>(\n    async ({ query }, runnableConfig) => {\n      const searchResult = await search({\n        query,\n        onSearchResults: _onSearchResults\n          ? (result): void => {\n            _onSearchResults(result, runnableConfig);\n          }\n          : undefined,\n      });\n      const output = formatResultsForLLM(searchResult);\n      return [output, { [Constants.WEB_SEARCH]: { ...searchResult } }];\n    },\n    {\n      name: Constants.WEB_SEARCH,\n      description: `\nReal-time search. Results have required unique citation anchors.\n\nAnchors:\n- \\\\ue202turn0searchN (web), \\\\ue202turn0newsN (news), \\\\ue202turn0imageN (image)\n\nSpecial Markers:\n- \\\\ue203...\\\\ue204 — mark start/end of cited span\n- \\\\ue200...\\\\ue201 — composite/group block (e.g. \\\\ue200cite\\\\ue202turn0search1\\\\ue202turn0news2\\\\ue201)\n- \\\\ue206 — marks grouped/summary citation areas\n\n**CITE EVERY NON-OBVIOUS FACT/QUOTE:**\nInsert the anchor marker(s) immediately after the statement:\n- \"Pure functions produce same output \\\\ue202turn0search0.\"\n- Multiple: \"Benefits \\\\ue202turn0search0\\\\ue202turn0news0.\"\n- Span: \\\\ue203Key: first-class functions\\\\ue204\\\\ue202turn0news1\n- Group: \"Functional languages.\"\\\\ue206 or \\\\ue200cite\\\\ue202turn0search0\\\\ue202turn0news1\\\\ue201\n- Image: \"See photo \\\\ue202turn0image0.\"\n\n**NEVER use markdown links, [1], or footnotes. CITE ONLY with anchors provided.**\n`.trim(),\n      schema: SearchToolSchema,\n      responseFormat: Constants.CONTENT_AND_ARTIFACT,\n    }\n  );\n};\n"],"names":[],"mappings":";;;;;;;;;AAAA;AAWA,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;AAChC,IAAA,KAAK,EAAE;AACJ,SAAA,MAAM;SACN,QAAQ,CACP,qEAAqE,CACtE;AACJ,CAAA,CAAC;MAEW,gBAAgB,GAAG,CAC9B,MAA6B,GAAA,EAAE,KACmB;IAClD,MAAM,EACJ,cAAc,GAAG,QAAQ,EACzB,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,YAAY,GAAG,QAAQ,EACvB,UAAU,GAAG,CAAC,EACd,UAAU,GAAG,CAAC,eAAe,CAAC,EAC9B,aAAa,GAAG,IAAI,EACpB,eAAe,EACf,eAAe,EACf,gBAAgB,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,EACvC,UAAU,EACV,YAAY,EACZ,eAAe,EAAE,gBAAgB,GAClC,GAAG,MAAM;IAEV,MAAM,SAAS,GAAG,eAAe,CAAC;QAChC,cAAc;QACd,YAAY;QACZ,kBAAkB;QAClB,aAAa;AACd,KAAA,CAAC;IAEF,MAAM,gBAAgB,GAAG,sBAAsB,CAAC;AAC9C,QAAA,MAAM,EAAE,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB;AACxD,QAAA,MAAM,EAAE,eAAe;AACvB,QAAA,OAAO,EAAE,gBAAgB;AAC1B,KAAA,CAAC;IAEF,MAAM,gBAAgB,GAAG,cAAc,CAAC;QACtC,YAAY;QACZ,UAAU;QACV,YAAY;AACb,KAAA,CAAC;IAEF,IAAI,CAAC,gBAAgB,EAAE;AACrB,QAAA,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC;;IAG9D,MAAM,eAAe,GAAG,qBAAqB,CAC3C;AACE,QAAA,QAAQ,EAAE,gBAAgB;QAC1B,WAGD,EACD,gBAAgB,CACjB;AAED,IAAA,MAAM,MAAM,GAAG,OAAO,EACpB,KAAK,EACL,OAAO,GAAG,IAAI,EACd,UAAU,GAAG,CAAC,EACd,eAAe,GAMhB,KAAiC;AAChC,QAAA,IAAI;YACF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC;AACjD,YAAA,eAAe,GAAG,OAAO,CAAC;AAE1B,YAAA,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;gBACpB,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,eAAe,CAAC;;AAGnD,YAAA,MAAM,gBAAgB,GAAG,MAAM,eAAe,CAAC,cAAc,CAC3D,OAAO,EACP,UAAU,EACV,KAAK,EACL,OAAO,CACR;AACD,YAAA,OAAO,gBAAgB,CAAC,gBAAgB,CAAC;;QACzC,OAAO,KAAK,EAAE;AACd,YAAA,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC;YACxC,OAAO;AACL,gBAAA,OAAO,EAAE,EAAE;AACX,gBAAA,UAAU,EAAE,EAAE;AACd,gBAAA,MAAM,EAAE,EAAE;AACV,gBAAA,eAAe,EAAE,EAAE;AACnB,gBAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;aAC9D;;AAEL,KAAC;IAED,OAAO,IAAI,CACT,OAAO,EAAE,KAAK,EAAE,EAAE,cAAc,KAAI;AAClC,QAAA,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC;YAChC,KAAK;AACL,YAAA,eAAe,EAAE;AACf,kBAAE,CAAC,MAAM,KAAU;AACjB,oBAAA,gBAAgB,CAAC,MAAM,EAAE,cAAc,CAAC;;AAE1C,kBAAE,SAAS;AACd,SAAA,CAAC;AACF,QAAA,MAAM,MAAM,GAAG,mBAAmB,CAAC,YAAY,CAAC;AAChD,QAAA,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,SAAS,CAAC,UAAU,GAAG,EAAE,GAAG,YAAY,EAAE,EAAE,CAAC;AAClE,KAAC,EACD;QACE,IAAI,EAAE,SAAS,CAAC,UAAU;AAC1B,QAAA,WAAW,EAAE;;;;;;;;;;;;;;;;;;;;AAoBlB,CAAA,CAAC,IAAI,EAAE;AACF,QAAA,MAAM,EAAE,gBAAgB;QACxB,cAAc,EAAE,SAAS,CAAC,oBAAoB;AAC/C,KAAA,CACF;AACH;;;;"}

package/dist/types/common/enum.d.ts

@@ -112,6 +112,7 @@ export declare enum Callback {
 export declare enum Constants {
     OFFICIAL_CODE_BASEURL = "https://api.librechat.ai/v1",
     EXECUTE_CODE = "execute_code",
+    WEB_SEARCH = "web_search",
     CONTENT_AND_ARTIFACT = "content_and_artifact"
 }
 export declare enum EnvVar {

package/dist/types/index.d.ts

@@ -5,6 +5,7 @@ export * from './events';
 export * from './messages';
 export * from './graphs';
 export * from './tools/CodeExecutor';
+export * from './tools/search';
 export * from './common';
 export * from './utils';
 export type * from './types';

package/dist/types/scripts/search.d.ts

@@ -0,0 +1 @@
+export {};