ปลั๊กอิน Generative AI ของ Google

ปลั๊กอิน 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,
});