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

src/lib/agents/search/index.ts

@@ -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: [
         {

src/lib/agents/search/types.ts

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

src/lib/agents/search/widgets/calculationWidget.ts

@@ -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}`,

src/lib/agents/search/widgets/stockWidget.ts

@@ -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,

src/lib/agents/search/widgets/weatherWidget.ts

@@ -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}`,
         },