test(ai): run prompt templates integration tests against prod endpoint

packages/ai/integration/constants.ts

@@ -94,6 +94,21 @@ export const liveTestConfigs: readonly TestConfig[] = backends.flatMap(
   }
 );
 
+/**
+ * Test configurations used for server prompt templates integration tests.
+ * Server prompt templates don't define the model name from the client, so these test configs
+ * do not define a model string.
+ * These tests should only run once per backend, rather than once per backend *per model*.
+ */
+export const promptTemplatesTestConfigs: readonly TestConfig[] = backends.flatMap(backend => {
+  const ai = getAI(app, { backend });
+  return {
+    ai,
+    model: '', // Unused by prompt templates tests
+    toString: () => formatConfigAsString({ ai, model: '' })
+  };
+});
+
 export const TINY_IMG_BASE64 =
   'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEUAAACnej3aAAAAAXRSTlMAQObYZgAAAApJREFUCNdjYAAAAAIAAeIhvDMAAAAASUVORK5CYII=';
 export const IMAGE_MIME_TYPE = 'image/png';

packages/ai/integration/prompt-templates.test.ts

@@ -21,8 +21,7 @@ import {
   getTemplateGenerativeModel,
   getTemplateImagenModel
 } from '../src';
-import { testConfigs } from './constants';
-import { STAGING_URL } from '../src/constants';
+import { promptTemplatesTestConfigs } from './constants';
 
 const templateBackendSuffix = (
   backendType: BackendType
@@ -31,13 +30,11 @@ const templateBackendSuffix = (
 
 describe('Prompt templates', function () {
   this.timeout(20_000);
-  testConfigs.forEach(testConfig => {
-    describe(`${testConfig.toString()}`, () => {
+  promptTemplatesTestConfigs.forEach(testConfig => {
+    describe(`${promptTemplatesTestConfigs.toString()}`, () => {
       describe('Generative Model', () => {
         it('successfully generates content', async () => {
-          const model = getTemplateGenerativeModel(testConfig.ai, {
-            baseUrl: STAGING_URL
-          });
+          const model = getTemplateGenerativeModel(testConfig.ai);
           const { response } = await model.generateContent(
             `sassy-greeting-${templateBackendSuffix(
               testConfig.ai.backend.backendType
@@ -49,16 +46,14 @@ describe('Prompt templates', function () {
       });
       describe('Imagen model', async () => {
         it('successfully generates images', async () => {
-          const model = getTemplateImagenModel(testConfig.ai, {
-            baseUrl: STAGING_URL
-          });
+          const model = getTemplateImagenModel(testConfig.ai);
           const { images } = await model.generateImages(
             `portrait-${templateBackendSuffix(
               testConfig.ai.backend.backendType
             )}`,
             { animal: 'Rhino' }
           );
-          expect(images.length).to.equal(2); // We ask for two images in the prompt template
+          expect(images.length).to.equal(1); // We ask for two images in the prompt template
         });
       });
     });