Market Research Agent with Gemini and the AI SDK by Vercel

import{google}from"@ai-sdk/google";
import{generateText}from"ai";
asyncfunctionmain(){
const{text}=awaitgenerateText({
model:google("gemini-2.5-flash"),
prompt:'What is plant-based milk?',
});
console.log(text);
}
main().catch(console.error);

import{google}from"@ai-sdk/google";
import{generateText}from"ai";
asyncfunctionmain(){
// Step 1: Search market trends
const{text:marketTrends,sources}=awaitgenerateText({
model:google("gemini-2.5-flash"),
tools:{
google_search:google.tools.googleSearch({}),
},
prompt:`Search the web for market trends for plant-based milk in North America for 2024-2025.
 I need to know the market size, key players and their market share, and primary consumer drivers.
 `,
});
console.log("Market trends found:\n",marketTrends);
// To see the sources, uncomment the following line:
// console.log("Sources:\n", sources);
}
main().catch(console.error);

import{google}from"@ai-sdk/google";
import{generateText,generateObject}from"ai";
import{z}from"zod/v4";
import{ChartConfiguration}from"chart.js";
// Helper function to create Chart.js configurations
functioncreateChartConfig({labels,data,label,type,colors,}:{
labels:string[];
data:number[];
label:string;
type:"bar"|"line";
colors:string[];
}):ChartConfiguration{
return{
type:type,
data:{
labels:labels,
datasets:[
{
label:label,
data:data,
borderWidth:1,
...(type==="bar" && {backgroundColor:colors}),
...(type==="line" && colors.length > 0 && {borderColor:colors[0]}),
},
],
},
options:{
animation:{duration:0},// Disable animations for static PDF rendering
},
};
}
asyncfunctionmain(){
// Step 1: Search market trends
const{text:marketTrends,sources}=awaitgenerateText({
model:google("gemini-2.5-flash"),
tools:{
google_search:google.tools.googleSearch({}),
},
prompt:`Search the web for market trends for plant-based milk in North America for 2024-2025.
 I need to know the market size, key players and their market share, and primary consumer drivers.
 `,
});
console.log("Market trends found.");
// Step 2: Extract chart data
const{object:chartData}=awaitgenerateObject({
model:google("gemini-2.5-flash"),
schema:z.object({
chartConfigurations:z
.array(
z.object({
type:z.enum(["bar","line"]).describe('The type of chart to generate. Either "bar" or "line"',),
labels:z.array(z.string()).describe("A list of chart labels"),
data:z.array(z.number()).describe("A list of the chart data"),
label:z.string().describe("A label for the chart"),
colors:z.array(z.string()).describe('A list of colors to use for the chart, e.g. "rgba(255, 99, 132, 0.8)"',),
}),
)
.describe("A list of chart configurations"),
}),
prompt:`Given the following market trends text, come up with a list of 1-3 meaningful bar or line charts
 and generate chart data.

Market Trends:
${marketTrends}
`,
});
constchartConfigs=chartData.chartConfigurations.map(createChartConfig);
console.log("Chart configurations generated.");
}
main().catch(console.error);

// ... (imports from previous step)
importpuppeteerfrom"puppeteer";
// ... (createChartConfig helper function from previous step)
asyncfunctionmain(){
// ... (Step 1 and 2 from previous step)
// Step 3: Generate the final HTML report and save it as a PDF
const{text:htmlReport}=awaitgenerateText({
model:google("gemini-2.5-flash"),
prompt:`You are an expert financial analyst and report writer.
 Your task is to generate a comprehensive market analysis report in HTML format.
 **Instructions:**
 1. Write a full HTML document.
 2. Use the provided "Market Trends" text to write the main body of the report. Structure it with clear headings and paragraphs.
 3. Incorporate the provided "Chart Configurations" to visualize the data. For each chart, you MUST create a unique <canvas> element and a corresponding <script> block to render it using Chart.js.
 4. Reference the "Sources" at the end of the report.
 5. Do not include any placeholder data; use only the information provided.
 6. Return only the raw HTML code.
 **Chart Rendering Snippet:**
 Include this script in the head of the HTML: <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
 For each chart, use a structure like below, ensuring the canvas 'id' is unique for each chart, and apply the correspinding config:
 ---
 <div style="width: 800px; height: 600px;">
 <canvas id="chart1"></canvas>
 </div>
 <script>
 new Chart(document.getElementById('chart1'), config);
 </script>
 ---
 (For the second chart, use 'chart2' and the corresponding config, and so on.)
 **Data:**
 - Market Trends: ${marketTrends}
 - Chart Configurations: ${JSON.stringify(chartConfigs)}
 - Sources: ${JSON.stringify(sources)}
 `,
});
// LLMs may wrap the HTML in a markdown code block, so strip it.
constfinalHtml=htmlReport.replace(/^```html\n/,"").replace(/\n```$/,"");
constbrowser=awaitpuppeteer.launch();
constpage=awaitbrowser.newPage();
awaitpage.setContent(finalHtml);
awaitpage.pdf({path:"report.pdf",format:"A4"});
awaitbrowser.close();
console.log("\nReport generated successfully: report.pdf");
}
main().catch(console.error);