remi/Perplexica
1
0
Fork 0
mirror of https://github.com/ItzCrazyKns/Perplexica.git synced 2025-12-03 10:18:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(widgets): add LLM context to prevent context overflow

Browse Source
This commit is contained in:
ItzCrazyKns
2025-11-24 15:35:00 +05:30
parent 956a768a86
commit 0a62c60da2
5 changed files with 99 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/lib/agents/search/index.ts
View File

@@ -65,14 +65,13 @@ class SearchAgent {
const widgetContext = widgetOutputs
.map((o) => {
return `${o.type}: ${JSON.stringify(o.data)}`;
return `${o.type}: ${o.llmContext}`;
})
.join('\n-------------\n');
const finalContextWithWidgets = `<search_results note="These are the search results and you can cite these">${finalContext}</search_results>\n<widgets_result noteForAssistant="Its output is already showed to the user, you can use this information to answer the query but do not CITE this as a souce">${widgetContext}</widgets_result>`;
const writerPrompt = getWriterPrompt(finalContextWithWidgets);
const answerStream = input.config.llm.streamText({
messages: [
{

1
src/lib/agents/search/types.ts
View File

@@ -43,6 +43,7 @@ export type WidgetConfig = {
export type WidgetOutput = {
type: string;
llmContext: string;
data: any;
};

2
src/lib/agents/search/widgets/calculationWidget.ts
View File

@@ -45,6 +45,7 @@ const calculationWidget: Widget<typeof schema> = {
return {
type: 'calculation_result',
llmContext: `The result of the expression "${params.expression}" is ${result}.`,
data: {
expression: params.expression,
result: result,
@@ -53,6 +54,7 @@ const calculationWidget: Widget<typeof schema> = {
} catch (error) {
return {
type: 'calculation_result',
llmContext: 'Failed to evaluate mathematical expression.',
data: {
expression: params.expression,
result: `Error evaluating expression: ${error}`,

9
src/lib/agents/search/widgets/stockWidget.ts
View File

@@ -395,11 +395,20 @@ You can set skipSearch to true if the stock widget can fully answer the user's q
return {
type: 'stock',
llmContext: `Current price of ${stockData.shortName} (${stockData.symbol}) is ${stockData.regularMarketPrice} ${stockData.currency}. Other details: ${JSON.stringify({
marketState: stockData.marketState,
regularMarketChange: stockData.regularMarketChange,
regularMarketChangePercent: stockData.regularMarketChangePercent,
marketCap: stockData.marketCap,
peRatio: stockData.trailingPE,
dividendYield: stockData.dividendYield,
})}`,
data: stockData,
};
} catch (error: any) {
return {
type: 'stock',
llmContext: 'Failed to fetch stock data.',
data: {
error: `Error fetching stock data: ${error.message || error}`,
ticker: params.ticker,

4
src/lib/agents/search/widgets/weatherWidget.ts
View File

@@ -94,6 +94,7 @@ You can set skipSearch to true if the weather widget can fully answer the user's
return {
type: 'weather',
llmContext: `Weather in ${params.location} is ${weatherData.current}`,
data: {
location: params.location,
latitude: location.lat,
@@ -138,6 +139,7 @@ You can set skipSearch to true if the weather widget can fully answer the user's
return {
type: 'weather',
llmContext: `Weather in ${locationData.display_name} is ${weatherData.current}`,
data: {
location: locationData.display_name,
latitude: params.lat,
@@ -159,11 +161,13 @@ You can set skipSearch to true if the weather widget can fully answer the user's
return {
type: 'weather',
llmContext: 'No valid location or coordinates provided.',
data: null,
};
} catch (err) {
return {
type: 'weather',
llmContext: 'Failed to fetch weather data.',
data: {
error: `Error fetching weather data: ${err}`,
},