@illuma-ai/agents 1.5.1 → 2.1.1

package/src/tools/search/search.ts

@@ -2,7 +2,6 @@ import axios from 'axios';
 import { RecursiveCharacterTextSplitter } from '@langchain/textsplitters';
 import type * as t from './types';
 import { getAttribution, createDefaultLogger } from './utils';
-import { createTavilyAPI } from './tavily-search';
 import { BaseReranker } from './rerankers';
 
 const chunker = {
@@ -439,9 +438,6 @@ export const createSearchAPI = (
     serperApiKey,
     searxngInstanceUrl,
     searxngApiKey,
-    tavilyApiKey,
-    tavilySearchUrl,
-    tavilySearchOptions,
     domainBlocklist,
     countryBlocklist,
   } = config;
@@ -450,11 +446,9 @@ export const createSearchAPI = (
     return createSerperAPI(serperApiKey, domainBlocklist, countryBlocklist);
   } else if (searchProvider.toLowerCase() === 'searxng') {
     return createSearXNGAPI(searxngInstanceUrl, searxngApiKey);
-  } else if (searchProvider.toLowerCase() === 'tavily') {
-    return createTavilyAPI(tavilyApiKey, tavilySearchUrl, tavilySearchOptions);
   } else {
     throw new Error(
-      `Invalid search provider: ${searchProvider}. Must be 'serper', 'searxng', or 'tavily'`
+      `Invalid search provider: ${searchProvider}. Must be 'serper' or 'searxng'`
     );
   }
 };
@@ -500,13 +494,9 @@ export const createSourceProcessor = (
           const promise: Promise<t.ScrapeResult> = scraper
             .scrapeUrl(currentLink, {})
             .then(([url, response]) => {
-              /* metadata exists on Firecrawl/Serper responses but not on
-               * Tavily's. Read it via the scraper's extractMetadata abstraction
-               * to stay polymorphic across all scraper providers. */
-              const rawMeta = scraper.extractMetadata(response);
               const attribution = getAttribution(
                 url,
-                Object.keys(rawMeta).length > 0 ? rawMeta : undefined,
+                response.data?.metadata,
                 logger_
               );
               if (response.success && response.data) {