diff --git a/bin/scratch.ts b/bin/scratch.ts index 3d447c09..2d503e86 100644 --- a/bin/scratch.ts +++ b/bin/scratch.ts @@ -2,7 +2,6 @@ import 'dotenv/config' import restoreCursor from 'restore-cursor' -import { TavilyClient } from 'tavily' // import { SearxngClient } from '../src/services/searxng-client.js' // import { ClearbitClient } from '../src/index.js' @@ -19,6 +18,7 @@ import { TavilyClient } from 'tavily' // } from '../src/services/twitter/index.js' // import { MidjourneyClient } from '../src/index.js' // import { BingClient } from '../src/index.js' +import { TavilyClient } from '../src/index.js' /** * Scratch pad for testing. @@ -113,9 +113,10 @@ async function main() { // console.log(JSON.stringify(res, null, 2)) const tavily = new TavilyClient() - const res = await tavily.search( - 'when do experts predict that OpenAI will release GPT-5?' - ) + const res = await tavily.search({ + query: 'when do experts predict that OpenAI will release GPT-5?', + include_answer: true + }) console.log(JSON.stringify(res, null, 2)) } diff --git a/src/services/tavily-client.ts b/src/services/tavily-client.ts index 83a4e8ba..021707e3 100644 --- a/src/services/tavily-client.ts +++ b/src/services/tavily-client.ts @@ -167,6 +167,9 @@ export class TavilyClient extends AIFunctionsProvider { }) .json() - return pruneNullOrUndefined(res).results?.map(pruneNullOrUndefined) + return pruneNullOrUndefined({ + ...res, + results: res.results?.map(pruneNullOrUndefined) + }) } }