Skip to content

feat: add maxResultsPerQuery option to limit search results per query #1056

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
CrossAndHorsesRanch wants to merge 7 commits into
base: master
Choose a base branch
from
+48 −12
Open

feat: add maxResultsPerQuery option to limit search results per query #1056

Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
6e42428
feat: add maxResultsPerQuery option to limit search results per query
CrossAndHorsesRanch Mar 11, 2026
bad78f5
fix: validate maxResultsPerQuery is a positive integer before use
CrossAndHorsesRanch Mar 11, 2026
9daa641
fix: also slice accumulated results at return to enforce maxResultsPe…
CrossAndHorsesRanch Mar 11, 2026
4e5240d
fix: remove incorrect final slice on accumulated results
CrossAndHorsesRanch Mar 12, 2026
bb68ea0
fix: truncate search results in agentMessageHistory to enforce maxRes…
CrossAndHorsesRanch Mar 12, 2026
1e3ad0f
fix: use sanitized positive-integer guard in agentMessageHistory trun…
CrossAndHorsesRanch Mar 12, 2026
00b4e25
feat: add maxTotalResults to cap results across all engines post-merge
CrossAndHorsesRanch Mar 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions src/app/api/search/route.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ interface ChatRequestBody {
history: Array<[string, string]>;
stream?: boolean;
systemInstructions?: string;
maxResultsPerQuery?: number; // Limits results per SearXNG query; reduces LLM prompt size for local/small-context models
}

export const POST = async (req: Request) => {
Expand Down Expand Up @@ -60,6 +61,7 @@ export const POST = async (req: Request) => {
mode: body.optimizationMode,
fileIds: [],
systemInstructions: body.systemInstructions || '',
maxResultsPerQuery: body.maxResultsPerQuery,
},
followUp: body.query,
chatId: crypto.randomUUID(),
Expand Down
6 changes: 4 additions & 2 deletions src/lib/agents/search/researcher/actions/academicSearch.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,9 @@ const academicSearchAction: ResearchAction<typeof schema> = {
engines: ['arxiv', 'google scholar', 'pubmed'],
});

const resultChunks: Chunk[] = res.results.map((r) => ({
const resultChunks: Chunk[] = res.results
.slice(0, additionalConfig.maxResultsPerQuery ?? res.results.length)
.map((r) => ({
content: r.content || r.title,
metadata: {
title: r.title,
Expand Down Expand Up @@ -121,7 +123,7 @@ const academicSearchAction: ResearchAction<typeof schema> = {

return {
type: 'search_results',
results,
results,
};
},
};
Expand Down
6 changes: 4 additions & 2 deletions src/lib/agents/search/researcher/actions/socialSearch.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,9 @@ const socialSearchAction: ResearchAction<typeof schema> = {
engines: ['reddit'],
});

const resultChunks: Chunk[] = res.results.map((r) => ({
const resultChunks: Chunk[] = res.results
.slice(0, additionalConfig.maxResultsPerQuery ?? res.results.length)
.map((r) => ({
content: r.content || r.title,
metadata: {
title: r.title,
Expand Down Expand Up @@ -121,7 +123,7 @@ const socialSearchAction: ResearchAction<typeof schema> = {

return {
type: 'search_results',
results,
results,
};
},
};
Expand Down
6 changes: 4 additions & 2 deletions src/lib/agents/search/researcher/actions/webSearch.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,9 @@ const webSearchAction: ResearchAction<typeof actionSchema> = {
const search = async (q: string) => {
const res = await searchSearxng(q);

const resultChunks: Chunk[] = res.results.map((r) => ({
const resultChunks: Chunk[] = res.results
.slice(0, additionalConfig.maxResultsPerQuery ?? res.results.length)
.map((r) => ({
content: r.content || r.title,
metadata: {
title: r.title,
Expand Down Expand Up @@ -174,7 +176,7 @@ const webSearchAction: ResearchAction<typeof actionSchema> = {

return {
type: 'search_results',
results,
results,
};
},
};
Expand Down
21 changes: 15 additions & 6 deletions src/lib/agents/search/researcher/index.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -167,18 +167,27 @@ class Researcher {
session: session,
researchBlockId: researchBlockId,
fileIds: input.config.fileIds,
maxResultsPerQuery:
Number.isInteger(input.config.maxResultsPerQuery) &&
(input.config.maxResultsPerQuery as number) > 0
? input.config.maxResultsPerQuery
: undefined,
});

actionOutput.push(...actionResults);

actionResults.forEach((action, i) => {
agentMessageHistory.push({
role: 'tool',
id: finalToolCalls[i].id,
name: finalToolCalls[i].name,
content: JSON.stringify(action),
const truncatedAction =
action.type === 'search_results' && input.config.maxResultsPerQuery
? { ...action, results: action.results.slice(0, input.config.maxResultsPerQuery) }
: action;
agentMessageHistory.push({
role: 'tool',
id: finalToolCalls[i].id,
name: finalToolCalls[i].name,
content: JSON.stringify(truncatedAction),
});
});
});
}

const searchResults = actionOutput
Expand Down
2 changes: 2 additions & 0 deletions src/lib/agents/search/types.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ export type SearchAgentConfig = {
embedding: BaseEmbedding<any>;
mode: 'speed' | 'balanced' | 'quality';
systemInstructions: string;
maxResultsPerQuery?: number; // Limits results per SearXNG query; reduces LLM prompt size for local/small-context models
};

export type SearchAgentInput = {
Expand Down Expand Up @@ -66,6 +67,7 @@ export type AdditionalConfig = {
llm: BaseLLM<any>;
embedding: BaseEmbedding<any>;
session: SessionManager;
maxResultsPerQuery?: number;
};

export type ResearcherInput = {
Expand Down