Add integration test

hsubox76 · hsubox76 · commit 33bd2572b272 · 2025-09-12T09:54:17.000-07:00
diff --git a/packages/ai/integration/generate-content.test.ts b/packages/ai/integration/generate-content.test.ts
@@ -21,7 +21,9 @@ import {
   GenerationConfig,
   HarmBlockThreshold,
   HarmCategory,
+  Language,
   Modality,
+  Outcome,
   SafetySetting,
   getGenerativeModel
 } from '../src';
@@ -188,6 +190,32 @@ describe('Generate Content', () => {
         });
       });
 
+      it('generateContent: code execution', async () => {
+        const model = getGenerativeModel(testConfig.ai, {
+          model: testConfig.model,
+          generationConfig: commonGenerationConfig,
+          safetySettings: commonSafetySettings,
+          tools: [{ codeExecution: {} }]
+        });
+        const prompt =
+          'What is the sum of the first 50 prime numbers? ' +
+          'Generate and run code for the calculation, and make sure you get all 50.';
+
+        const result = await model.generateContent(prompt);
+        const parts = result.response.candidates?.[0].content.parts;
+        expect(
+          parts?.some(part => part.executableCode?.language === Language.PYTHON)
+        ).to.be.true;
+        expect(
+          parts?.some(part => part.codeExecutionResult?.outcome === Outcome.OK)
+        ).to.be.true;
+        // Expect these to be truthy (!= null)
+        expect(parts?.some(part => part.executableCode?.code != null)).to.be
+          .true;
+        expect(parts?.some(part => part.codeExecutionResult?.output != null)).to
+          .be.true;
+      });
+
       it('generateContentStream: text input, text output', async () => {
         const model = getGenerativeModel(testConfig.ai, {
           model: testConfig.model,
diff --git a/packages/ai/src/methods/generate-content.test.ts b/packages/ai/src/methods/generate-content.test.ts
@@ -262,7 +262,6 @@ describe('generateContent()', () => {
       fakeRequestParams
     );
     const parts = result.response.candidates?.[0].content.parts;
-    parts?.forEach(part => console.log(Object.keys(part)));
     expect(
       parts?.some(part => part.codeExecutionResult?.outcome === Outcome.OK)
     ).to.be.true;
