Plugin AI Generatif Google menyediakan antarmuka ke model Gemini Google melalui Gemini API.
Penginstalan
npm i --save @genkit-ai/googleai
Konfigurasi
Untuk menggunakan plugin ini, tentukan saat Anda menginisialisasi Genkit:
import { genkit } from 'genkit';
import { googleAI } from '@genkit-ai/googleai';
const ai = genkit({
plugins: [googleAI()],
});
Plugin ini memerlukan kunci API untuk Gemini API, yang dapat Anda peroleh dari Google AI Studio.
Konfigurasikan plugin untuk menggunakan kunci API Anda dengan melakukan salah satu hal berikut:
- Tetapkan variabel lingkungan
GOOGLE_GENAI_API_KEY
ke kunci API Anda. Tentukan kunci API saat Anda menginisialisasi plugin:
googleAI({ apiKey: yourKey });
Namun, jangan sematkan kunci API Anda secara langsung di dalam kode. Gunakan fitur ini saja bersama dengan layanan seperti Cloud Secret Manager atau yang serupa.
Penggunaan
Plugin ini mengekspor referensi secara statis ke model yang didukungnya:
import {
gemini15Flash,
gemini15Pro,
textEmbedding004,
} from '@genkit-ai/googleai';
Anda dapat menggunakan referensi ini untuk menentukan model yang digunakan generate()
:
const ai = genkit({
plugins: [googleAI()],
model: gemini15Flash,
});
const llmResponse = await ai.generate('Tell me a joke.');
atau menggunakan penyemat (mis. textEmbedding004
) dengan embed
atau pengambil:
const ai = genkit({
plugins: [googleAI()],
});
const embedding = await ai.embed({
embedder: textEmbedding004,
content: input,
});
Gemini Files API
Anda dapat menggunakan file yang diupload ke Gemini Files API dengan Genkit:
import { GoogleAIFileManager } from '@google/generative-ai/server';
import { genkit } from 'genkit';
import { googleAI } from '@genkit-ai/googleai';
const ai = genkit({
plugins: [googleAI()],
});
const fileManager = new GoogleAIFileManager(process.env.GOOGLE_GENAI_API_KEY);
const uploadResult = await fileManager.uploadFile(
'path/to/file.jpg',
{
mimeType: 'image/jpeg',
displayName: 'Your Image',
}
);
const response = await ai.generate({
model: gemini15Flash,
prompt: [
{text: 'Describe this image:'},
{media: {contentType: uploadResult.file.mimeType, url: uploadResult.file.uri}}
]
});
Model yang disesuaikan
Anda dapat menggunakan model yang disesuaikan dengan Google Gemini API. Ikuti petunjuk dari Gemini API atau sesuaikan model menggunakan AI Studio.
Proses penyesuaian menggunakan model dasar—misalnya, Gemini 1.5 Flash—dan contoh yang Anda berikan untuk membuat model baru yang disesuaikan. Ingat model dasar yang Anda gunakan, dan salin ID model baru.
Saat memanggil model yang dioptimalkan di Genkit, gunakan model dasar sebagai parameter model
, dan teruskan ID model yang dioptimalkan sebagai bagian dari blok config
. Misalnya,
jika Anda menggunakan Gemini 1.5 Flash sebagai model dasar, dan mendapatkan ID model
tunedModels/my-example-model-apbm8oqbvuv2
, Anda dapat memanggilnya dengan:
const ai = genkit({
plugins: [googleAI()],
});
const llmResponse = await ai.generate({
prompt: `Suggest an item for the menu of fish themed restruant`,
model: gemini15Flash.withConfig({
version: "tunedModels/my-example-model-apbm8oqbvuv2",
}),
});