index.ts•3.66 kB
/**
* Copyright 2024 Google LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { googleAI } from '@genkit-ai/google-genai';
import { genkit, z } from 'genkit';
const ai = genkit({
plugins: [googleAI()],
});
// This example generates greetings for a customer at our new AI-powered coffee shop,
// demonstrating how to use prompts in Genkit flows.
// A flow to greet a customer by name
const CustomerNameSchema = z.object({
customerName: z.string(),
});
const simpleGreetingPrompt = ai.definePrompt(
{
name: 'simpleGreeting',
model: googleAI.model('gemini-2.5-flash'),
input: { schema: CustomerNameSchema },
output: {
format: 'text',
},
},
`
You're a barista at a nice coffee shop.
A regular customer named {{customerName}} enters.
Greet the customer in one sentence, and recommend a coffee drink.
`
);
export const simpleGreetingFlow = ai.defineFlow(
{
name: 'simpleGreeting',
inputSchema: CustomerNameSchema,
outputSchema: z.string(),
},
async (input) => (await simpleGreetingPrompt(input)).text
);
// Another flow to recommend a drink based on the time of day and a previous order.
// This prompt uses multiple messages, alternating roles
// to make the response more conversational.
const CustomerTimeAndHistorySchema = z.object({
customerName: z.string(),
currentTime: z.string(),
previousOrder: z.string(),
});
const greetingWithHistoryPrompt = ai.definePrompt(
{
name: 'greetingWithHistory',
model: googleAI.model('gemini-2.5-flash'),
input: { schema: CustomerTimeAndHistorySchema },
output: {
format: 'text',
},
},
`
{{role "user"}}
Hi, my name is {{customerName}}. The time is {{currentTime}}. Who are you?
{{role "model"}}
I am Barb, a barista at this nice underwater-themed coffee shop called Krabby Kooffee.
I know pretty much everything there is to know about coffee,
and I can cheerfully recommend delicious coffee drinks to you based on whatever you like.
{{role "user"}}
Great. Last time I had {{previousOrder}}.
I want you to greet me in one sentence, and recommend a drink.
`
);
export const greetingWithHistoryFlow = ai.defineFlow(
{
name: 'greetingWithHistory',
inputSchema: CustomerTimeAndHistorySchema,
outputSchema: z.string(),
},
async (input) => (await greetingWithHistoryPrompt(input)).text
);
// A flow to quickly test all the above flows
// Run on the CLI with `$ genkit flow:run testAllCoffeeFlows`
// View the trace in the Developer UI to see the llm responses.
export const testAllCoffeeFlows = ai.defineFlow(
{
name: 'testAllCoffeeFlows',
outputSchema: z.object({
pass: z.boolean(),
error: z.string().optional(),
}),
},
async () => {
const test1 = simpleGreetingFlow({ customerName: 'Sam' });
const test2 = greetingWithHistoryFlow({
customerName: 'Sam',
currentTime: '09:45am',
previousOrder: 'Caramel Macchiato',
});
return Promise.all([test1, test2])
.then((unused) => {
return { pass: true };
})
.catch((e: Error) => {
return { pass: false, error: e.toString() };
});
}
);