ปลั๊กอิน Generative AI ของ Google มอบอินเทอร์เฟซสำหรับโมเดล Gemini ของ Google ผ่าน Gemini API
การติดตั้ง
npm i --save @genkit-ai/googleai
การกำหนดค่า
หากต้องการใช้ปลั๊กอินนี้ ให้ระบุปลั๊กอินเมื่อเรียกใช้ configureGenkit()
:
import { googleAI } from '@genkit-ai/googleai';
export default configureGenkit({
plugins: [googleAI()],
// ...
});
ปลั๊กอินต้องใช้คีย์ API สำหรับ Gemini API ซึ่งหาได้จาก Google AI Studio
กำหนดค่าปลั๊กอินให้ใช้คีย์ API โดยทำอย่างใดอย่างหนึ่งต่อไปนี้
ตั้งค่าตัวแปรสภาพแวดล้อม
GOOGLE_GENAI_API_KEY
เป็นคีย์ API ของคุณระบุคีย์ API เมื่อเริ่มต้นปลั๊กอิน
googleAI({ apiKey: yourKey });
อย่างไรก็ตาม อย่าฝังคีย์ API ลงในโค้ดโดยตรง ใช้ฟีเจอร์นี้ร่วมกับบริการอย่างเช่น Cloud Secret Manager หรือบริการอื่นๆ ที่คล้ายกันเท่านั้น
บางรุ่น (เช่น Gemini 1.5 Pro) เป็นเวอร์ชันตัวอย่างและใช้ได้ผ่าน API ของ v1beta
เท่านั้น คุณระบุ apiVersion
เพื่อรับสิทธิ์เข้าถึงรุ่นเหล่านั้นได้
configureGenkit({
plugins: [googleAI({ apiVersion: 'v1beta' })],
});
หรือจะระบุหลายเวอร์ชันหากต้องการใช้โมเดลหลายเวอร์ชันพร้อมกันก็ได้
configureGenkit({
plugins: [googleAI({ apiVersion: ['v1', 'v1beta'] })],
});
การใช้งาน
ปลั๊กอินนี้จะส่งออกการอ้างอิงไปยังโมเดลที่รองรับแบบคงที่ดังนี้
import {
geminiPro,
geminiProVision,
textEmbeddingGecko001,
} from '@genkit-ai/googleai';
คุณใช้การอ้างอิงเหล่านี้เพื่อระบุโมเดลที่ generate()
ใช้ได้
const llmResponse = await generate({
model: geminiPro,
prompt: 'Tell me a joke.',
});
หรือใช้เครื่องมือฝัง (เช่น textEmbeddingGecko001
) ที่มี embed
หรือรีทรีฟเวอร์:
const embedding = await embed({
embedder: textEmbeddingGecko001,
content: input,
});