This commit is contained in:
ItzCrazyKns
2026-03-26 22:34:18 +05:30
6 changed files with 33 additions and 9 deletions

View File

@@ -37,7 +37,8 @@ const getStepTitle = (
if (step.type === 'reasoning') { if (step.type === 'reasoning') {
return isStreaming && !step.reasoning ? 'Thinking...' : 'Thinking'; return isStreaming && !step.reasoning ? 'Thinking...' : 'Thinking';
} else if (step.type === 'searching') { } else if (step.type === 'searching') {
return `Searching ${step.searching.length} ${step.searching.length === 1 ? 'query' : 'queries'}`; const queries = Array.isArray(step.searching) ? step.searching : [];
return `Searching ${queries.length} ${queries.length === 1 ? 'query' : 'queries'}`;
} else if (step.type === 'search_results') { } else if (step.type === 'search_results') {
return `Found ${step.reading.length} ${step.reading.length === 1 ? 'result' : 'results'}`; return `Found ${step.reading.length} ${step.reading.length === 1 ? 'result' : 'results'}`;
} else if (step.type === 'reading') { } else if (step.type === 'reading') {
@@ -160,6 +161,7 @@ const AssistantSteps = ({
)} )}
{step.type === 'searching' && {step.type === 'searching' &&
Array.isArray(step.searching) &&
step.searching.length > 0 && ( step.searching.length > 0 && (
<div className="flex flex-wrap gap-1.5 mt-1.5"> <div className="flex flex-wrap gap-1.5 mt-1.5">
{step.searching.map((query, idx) => ( {step.searching.map((query, idx) => (

View File

@@ -19,6 +19,9 @@ class APISearchAgent {
chatHistory: input.chatHistory, chatHistory: input.chatHistory,
followUp: input.followUp, followUp: input.followUp,
llm: input.config.llm, llm: input.config.llm,
}).catch((err) => {
console.error(`Error executing widgets: ${err}`);
return [];
}); });
let searchPromise: Promise<ResearcherOutput> | null = null; let searchPromise: Promise<ResearcherOutput> | null = null;

View File

@@ -30,7 +30,7 @@ const academicSearchAction: ResearchAction<typeof schema> = {
config.classification.classification.skipSearch === false && config.classification.classification.skipSearch === false &&
config.classification.classification.academicSearch === true, config.classification.classification.academicSearch === true,
execute: async (input, additionalConfig) => { execute: async (input, additionalConfig) => {
input.queries = input.queries.slice(0, 3); input.queries = (Array.isArray(input.queries) ? input.queries : [input.queries]).slice(0, 3);
const researchBlock = additionalConfig.session.getBlock( const researchBlock = additionalConfig.session.getBlock(
additionalConfig.researchBlockId, additionalConfig.researchBlockId,

View File

@@ -30,7 +30,7 @@ const socialSearchAction: ResearchAction<typeof schema> = {
config.classification.classification.skipSearch === false && config.classification.classification.skipSearch === false &&
config.classification.classification.discussionSearch === true, config.classification.classification.discussionSearch === true,
execute: async (input, additionalConfig) => { execute: async (input, additionalConfig) => {
input.queries = input.queries.slice(0, 3); input.queries = (Array.isArray(input.queries) ? input.queries : [input.queries]).slice(0, 3);
const researchBlock = additionalConfig.session.getBlock( const researchBlock = additionalConfig.session.getBlock(
additionalConfig.researchBlockId, additionalConfig.researchBlockId,

View File

@@ -85,7 +85,7 @@ const webSearchAction: ResearchAction<typeof actionSchema> = {
config.sources.includes('web') && config.sources.includes('web') &&
config.classification.classification.skipSearch === false, config.classification.classification.skipSearch === false,
execute: async (input, additionalConfig) => { execute: async (input, additionalConfig) => {
input.queries = input.queries.slice(0, 3); input.queries = (Array.isArray(input.queries) ? input.queries : [input.queries]).slice(0, 3);
const researchBlock = additionalConfig.session.getBlock( const researchBlock = additionalConfig.session.getBlock(
additionalConfig.researchBlockId, additionalConfig.researchBlockId,

View File

@@ -38,11 +38,30 @@ export const searchSearxng = async (
}); });
} }
const res = await fetch(url); const controller = new AbortController();
const data = await res.json(); const timeoutId = setTimeout(() => controller.abort(), 10000);
const results: SearxngSearchResult[] = data.results; try {
const suggestions: string[] = data.suggestions; const res = await fetch(url, {
signal: controller.signal,
});
return { results, suggestions }; if (!res.ok) {
throw new Error(`SearXNG error: ${res.statusText}`);
}
const data = await res.json();
const results: SearxngSearchResult[] = data.results;
const suggestions: string[] = data.suggestions;
return { results, suggestions };
} catch (err: any) {
if (err.name === 'AbortError') {
throw new Error('SearXNG search timed out');
}
throw err;
} finally {
clearTimeout(timeoutId);
}
}; };