feat(Tavily): update sample configuration for Tavily integration

This commit is contained in:
OTYAK
2025-04-08 10:41:00 +01:00
parent 2c56aa3cb3
commit 54c71e33e0
9 changed files with 13 additions and 124 deletions

View File

@ -7,7 +7,7 @@ import { PromptTemplate } from '@langchain/core/prompts';
import formatChatHistoryAsString from '../utils/formatHistory';
import { BaseMessage } from '@langchain/core/messages';
import { StringOutputParser } from '@langchain/core/output_parsers';
import { searchSearxng } from '../searxng';
import { searchSearxng } from '../searchEngines/searxng';
import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
const imageSearchChainPrompt = `

View File

@ -7,7 +7,7 @@ import { PromptTemplate } from '@langchain/core/prompts';
import formatChatHistoryAsString from '../utils/formatHistory';
import { BaseMessage } from '@langchain/core/messages';
import { StringOutputParser } from '@langchain/core/output_parsers';
import { searchSearxng } from '../searxng';
import { searchSearxng } from '../searchEngines/searxng';
import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
const VideoSearchChainPrompt = `

View File

@ -17,8 +17,8 @@ import LineListOutputParser from '../outputParsers/listLineOutputParser';
import LineOutputParser from '../outputParsers/lineOutputParser';
import { getDocumentsFromLinks } from '../utils/documents';
import { Document } from 'langchain/document';
import { searchTavily } from '../tavily';
import { searchSearxng } from '../searxng';
import { searchTavily } from '../searchEngines/tavily';
import { searchSearxng } from '../searchEngines/searxng';
import { getSearchEngine } from '../config';
import path from 'node:path';
import fs from 'node:fs';
@ -208,7 +208,6 @@ class MetaSearchAgent implements MetaSearchAgentType {
question = question.replace(/<think>.*?<\/think>/g, '');
const searchEngine = getSearchEngine();
console.log(`Using search engine: ${searchEngine}`);
let res;
@ -226,12 +225,8 @@ class MetaSearchAgent implements MetaSearchAgentType {
});
}
// If we have an AI-generated answer from Tavily, create a document for it
let documents: Document[] = [];
// Add the regular search results
documents = documents.concat(
res.results.map(
(result) =>

View File

@ -1,5 +1,5 @@
import axios from 'axios';
import { getSearxngApiEndpoint } from './config';
import { getSearxngApiEndpoint } from '../config';
interface SearxngSearchOptions {
categories?: string[];

View File

@ -1,5 +1,5 @@
import axios from 'axios';
import { getTavilyApiKey } from './config';
import { getTavilyApiKey } from '../config';
interface TavilySearchOptions {
topic?: 'general' | 'news';