Initial commit

templates/code-execution.ts

@@ -0,0 +1,206 @@
+/**
+ * Google Gemini API - Code Execution Example
+ *
+ * Demonstrates how to enable code execution so the model can generate
+ * and run Python code to solve computational problems.
+ *
+ * Features:
+ * - Basic code execution
+ * - Data analysis with code
+ * - Chart generation
+ * - Error handling for failed execution
+ *
+ * Requirements:
+ * - @google/genai@1.27.0+
+ * - GEMINI_API_KEY environment variable
+ *
+ * Note: Code Execution is NOT available on gemini-2.5-flash-lite
+ */
+
+import { GoogleGenAI } from '@google/genai';
+
+const ai = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY! });
+
+async function basicCodeExecution() {
+  console.log('=== Basic Code Execution Example ===\n');
+
+  const response = await ai.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents:
+      'What is the sum of the first 50 prime numbers? Generate and run code for the calculation, and make sure you get all 50.',
+    config: {
+      tools: [{ codeExecution: {} }],
+    },
+  });
+
+  // Parse response parts
+  console.log('Response parts:\n');
+  for (const part of response.candidates[0].content.parts) {
+    if (part.text) {
+      console.log('📝 Text:', part.text);
+    }
+    if (part.executableCode) {
+      console.log('\n💻 Generated Code:');
+      console.log(part.executableCode.code);
+    }
+    if (part.codeExecutionResult) {
+      console.log('\n✅ Execution Output:');
+      console.log(part.codeExecutionResult.output);
+    }
+  }
+
+  console.log('\n=== Basic Code Execution Complete ===');
+}
+
+async function dataAnalysisExample() {
+  console.log('\n=== Data Analysis Example ===\n');
+
+  const prompt = `
+    Analyze this sales data and calculate:
+    1. Total revenue
+    2. Average sale price
+    3. Best-selling month
+    4. Month with highest revenue
+
+    Use pandas or numpy for analysis.
+
+    Data (CSV format):
+    month,sales,revenue
+    Jan,150,45000
+    Feb,200,62000
+    Mar,175,53000
+    Apr,220,68000
+    May,190,58000
+  `;
+
+  const response = await ai.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents: prompt,
+    config: {
+      tools: [{ codeExecution: {} }],
+    },
+  });
+
+  for (const part of response.candidates[0].content.parts) {
+    if (part.text) {
+      console.log('📊 Analysis:', part.text);
+    }
+    if (part.executableCode) {
+      console.log('\n💻 Analysis Code:');
+      console.log(part.executableCode.code);
+    }
+    if (part.codeExecutionResult) {
+      console.log('\n📈 Results:');
+      console.log(part.codeExecutionResult.output);
+    }
+  }
+
+  console.log('\n=== Data Analysis Complete ===');
+}
+
+async function chartGenerationExample() {
+  console.log('\n=== Chart Generation Example ===\n');
+
+  const response = await ai.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents:
+      'Create a bar chart showing the distribution of prime numbers under 100 by their last digit. Generate the chart code and describe any patterns you see.',
+    config: {
+      tools: [{ codeExecution: {} }],
+    },
+  });
+
+  for (const part of response.candidates[0].content.parts) {
+    if (part.text) {
+      console.log('📊 Chart Description:', part.text);
+    }
+    if (part.executableCode) {
+      console.log('\n📉 Chart Code:');
+      console.log(part.executableCode.code);
+    }
+    if (part.codeExecutionResult) {
+      console.log('\n✓ Chart generated');
+      // Note: Image data would be in output
+    }
+  }
+
+  console.log('\n=== Chart Generation Complete ===');
+}
+
+async function chatWithCodeExecution() {
+  console.log('\n=== Chat with Code Execution Example ===\n');
+
+  const chat = await ai.chats.create({
+    model: 'gemini-2.5-flash',
+    config: {
+      tools: [{ codeExecution: {} }],
+    },
+  });
+
+  // First message
+  console.log('User: I have a math question for you.');
+  let response = await chat.sendMessage('I have a math question for you.');
+  console.log(`Assistant: ${response.text}\n`);
+
+  // Second message (will generate and execute code)
+  console.log('User: Calculate the Fibonacci sequence up to the 20th number and sum them.');
+  response = await chat.sendMessage(
+    'Calculate the Fibonacci sequence up to the 20th number and sum them.'
+  );
+
+  for (const part of response.candidates[0].content.parts) {
+    if (part.text) {
+      console.log('Assistant:', part.text);
+    }
+    if (part.executableCode) {
+      console.log('\nCode:');
+      console.log(part.executableCode.code);
+    }
+    if (part.codeExecutionResult) {
+      console.log('\nOutput:');
+      console.log(part.codeExecutionResult.output);
+    }
+  }
+
+  console.log('\n=== Chat with Code Execution Complete ===');
+}
+
+async function errorHandlingExample() {
+  console.log('\n=== Error Handling Example ===\n');
+
+  const response = await ai.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents: 'Write code that divides by zero and see what happens',
+    config: {
+      tools: [{ codeExecution: {} }],
+    },
+  });
+
+  for (const part of response.candidates[0].content.parts) {
+    if (part.codeExecutionResult) {
+      if (part.codeExecutionResult.outcome === 'OUTCOME_FAILED') {
+        console.error('❌ Code execution failed:');
+        console.error(part.codeExecutionResult.output);
+      } else {
+        console.log('✅ Success:');
+        console.log(part.codeExecutionResult.output);
+      }
+    }
+  }
+
+  console.log('\n=== Error Handling Example Complete ===');
+}
+
+// Run all examples
+async function main() {
+  await basicCodeExecution();
+  await dataAnalysisExample();
+  await chartGenerationExample();
+  await chatWithCodeExecution();
+  await errorHandlingExample();
+}
+
+main().catch((error) => {
+  console.error('Error:', error.message);
+  process.exit(1);
+});