make it possible to configure multiple engines

This commit is contained in:
HadiCherkaoui
2025-02-28 15:50:59 +01:00
parent 89a6e7fbb1
commit aba702c51b
5 changed files with 28 additions and 9 deletions

View File

@ -12,7 +12,7 @@ import { searchGooglePSE } from '../lib/searchEngines/google_pse';
import { searchBraveAPI } from '../lib/searchEngines/brave';
import { searchYaCy } from '../lib/searchEngines/yacy';
import { searchBingAPI } from '../lib/searchEngines/bing';
import { getSearchEngineBackend } from '../config';
import { getImageSearchEngineBackend } from '../config';
import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
const imageSearchChainPrompt = `
@ -42,7 +42,7 @@ type ImageSearchChainInput = {
};
async function performImageSearch(query: string) {
const searchEngine = getSearchEngineBackend();
const searchEngine = getImageSearchEngineBackend();
let images = [];
switch (searchEngine) {

View File

@ -11,7 +11,7 @@ import { searchSearxng } from '../lib/searchEngines/searxng';
import { searchGooglePSE } from '../lib/searchEngines/google_pse';
import { searchBraveAPI } from '../lib/searchEngines/brave';
import { searchBingAPI } from '../lib/searchEngines/bing';
import { getSearchEngineBackend } from '../config';
import { getVideoSearchEngineBackend } from '../config';
import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
const VideoSearchChainPrompt = `
@ -43,7 +43,7 @@ type VideoSearchChainInput = {
const strParser = new StringOutputParser();
async function performVideoSearch(query: string) {
const searchEngine = getSearchEngineBackend();
const searchEngine = getVideoSearchEngineBackend();
const youtubeQuery = `${query} site:youtube.com`;
let videos = [];