Update dev-ui gallery to use the latest in prompt technology

js/testapps/dev-ui-gallery/prompts/hello.first-last-name.prompt

@@ -1,5 +1,5 @@
 ---
-model: googleai/gemini-1.5-flash-latest
+model: googleai/gemini-1.5-flash
 input:
   schema:
     firstName: string

js/testapps/dev-ui-gallery/prompts/hello.history.prompt

@@ -1,5 +1,5 @@
 ---
-model: googleai/gemini-1.5-flash-latest
+model: googleai/gemini-1.5-flash
 config:
   maxOutputTokens: 2048
   temperature: 0.6

js/testapps/dev-ui-gallery/prompts/hello.json-output.prompt

@@ -1,5 +1,5 @@
 ---
-model: googleai/gemini-1.5-flash-latest
+model: googleai/gemini-1.5-flash
 input:
   schema:
     name: string

js/testapps/dev-ui-gallery/prompts/hello.prompt

@@ -1,5 +1,5 @@
 ---
-model: googleai/gemini-1.5-flash-latest
+model: googleai/gemini-1.5-flash
 config:
   maxOutputTokens: 2048
   temperature: 0.6

js/testapps/dev-ui-gallery/prompts/hello.system.prompt

@@ -1,5 +1,5 @@
 ---
-model: googleai/gemini-1.5-flash-latest
+model: googleai/gemini-1.5-flash
 config:
   maxOutputTokens: 2048
   temperature: 0.6

js/testapps/dev-ui-gallery/prompts/tools.prompt → js/testapps/dev-ui-gallery/prompts/tools/weather.prompt

@@ -1,5 +1,5 @@
 ---
-model: googleai/gemini-1.5-flash-latest
+model: googleai/gemini-1.5-flash
 config:
   maxOutputTokens: 2048
   temperature: 0.6

js/testapps/dev-ui-gallery/src/genkit.ts

@@ -26,10 +26,13 @@ import {
   VertexAIEvaluationMetricType,
 } from '@genkit-ai/vertexai';
 import { genkit } from 'genkit';
+import { logger } from 'genkit/logging';
 import { chroma } from 'genkitx-chromadb';
 import { ollama } from 'genkitx-ollama';
 import { pinecone } from 'genkitx-pinecone';
 
+logger.setLogLevel('info');
+
 // Turn off safety checks for evaluation so that the LLM as an evaluator can
 // respond appropriately to potentially harmful content without error.
 export const PERMISSIVE_SAFETY_SETTINGS: any = {
@@ -57,9 +60,7 @@ export const ai = genkit({
   // load at least one plugin representing each action type
   plugins: [
     // model providers
-    googleAI({
-      apiVersion: ['v1', 'v1beta'],
-    }),
+    googleAI(),
     ollama({
       models: [
         { name: 'llama2' },

js/testapps/dev-ui-gallery/src/main/prompts.ts

@@ -96,15 +96,52 @@ ai.defineStreamingFlow(
     outputSchema: z.string(),
   },
   async (input) => {
-    const prompt = await ai.prompt('codeDefinedPrompt');
-    const response = await prompt.generate({
-      input,
-    });
-
+    const response = await codeDefinedPrompt(input);
     return response.text;
   }
 );
 
+//
+// Function(al) prompts
+//
+
+export const promptFn = ai.definePrompt(
+  {
+    name: 'functionalPrompt',
+    input: {
+      schema: HelloSchema,
+      default: {
+        persona: 'Space Pirate',
+      },
+    },
+    model: gemini15Flash,
+  },
+  async (input) => ({
+    messages: [
+      {
+        role: 'user',
+        content: [
+          {
+            text: `say hello to ${input.name} in the voice of ${input.persona}`,
+          },
+        ],
+      },
+    ],
+  })
+);
+
+ai.defineFlow(
+  {
+    name: 'flowFunctionalPrompt',
+    inputSchema: HelloSchema,
+    outputSchema: z.string(),
+  },
+  async (input) => {
+    const hello = await ai.prompt('functionalPrompt');
+    return (await hello(input)).text;
+  }
+);
+
 //
 // Dotprompt file - text output
 //
@@ -116,8 +153,8 @@ ai.defineFlow(
     outputSchema: z.string(),
   },
   async (input) => {
-    const prompt = await ai.prompt('hello');
-    return (await prompt.generate({ input })).text;
+    const hello = await ai.prompt('hello');
+    return (await hello(input)).text;
   }
 );
 
@@ -132,8 +169,10 @@ ai.defineFlow(
     outputSchema: z.string(),
   },
   async (input) => {
-    const prompt = await ai.prompt('hello', { variant: 'first-last-name' });
-    return (await prompt.generate({ input })).text;
+    const hello = await ai.prompt('hello', {
+      variant: 'first-last-name',
+    });
+    return (await hello(input)).text;
   }
 );
 
@@ -148,8 +187,10 @@ ai.defineFlow(
     outputSchema: z.any(),
   },
   async (input) => {
-    const prompt = await ai.prompt('hello', { variant: 'json-output' });
-    return (await prompt.generate({ input })).output;
+    const hello = await ai.prompt('hello', {
+      variant: 'json-output',
+    });
+    return (await hello(input)).output;
   }
 );
 
@@ -164,8 +205,10 @@ ai.defineFlow(
     outputSchema: z.any(),
   },
   async (input) => {
-    const prompt = await ai.prompt('hello', { variant: 'system' });
-    return (await prompt.generate({ input })).text;
+    const hello = await ai.prompt('hello', {
+      variant: 'system',
+    });
+    return (await hello(input)).text;
   }
 );
 
@@ -180,8 +223,10 @@ ai.defineFlow(
     outputSchema: z.any(),
   },
   async (input) => {
-    const prompt = await ai.prompt('hello', { variant: 'history' });
-    return (await prompt.generate({ input })).text;
+    const hello = await ai.prompt('hello', {
+      variant: 'history',
+    });
+    return (await hello(input)).text;
   }
 );
 

js/testapps/dev-ui-gallery/src/main/tools.ts

@@ -92,8 +92,5 @@ ai.defineFlow(
     inputSchema: WeatherSchema,
     outputSchema: z.string(),
   },
-  async (input) => {
-    const { text } = await weatherPrompt(input);
-    return text;
-  }
+  async (input) => (await weatherPrompt(input)).text
 );