Google 생성형 AI 플러그인은 Gemini API를 통해 Google Gemini 모델에 인터페이스를 제공합니다.
설치
npm i --save @genkit-ai/googleai
구성
이 플러그인을 사용하려면 configureGenkit()
를 호출할 때 지정합니다.
import { googleAI } from '@genkit-ai/googleai';
export default configureGenkit({
plugins: [googleAI()],
// ...
});
플러그인에는 Gemini API의 API 키가 필요하며 Google AI Studio에서 이 키를 가져올 수 있습니다.
다음 중 하나를 수행하여 API 키를 사용하도록 플러그인을 구성합니다.
GOOGLE_GENAI_API_KEY
환경 변수를 API 키에 설정합니다.플러그인을 초기화할 때 API 키를 지정합니다.
googleAI({ apiKey: yourKey });
그러나 코드에 API 키를 직접 삽입하지 마세요. 이 기능은 Cloud Secret Manager 등의 서비스와 함께 사용하세요.
일부 모델 (예: Gemini 1.5 Pro)은 미리보기 상태이며 v1beta
API를 통해서만 사용할 수 있습니다. apiVersion
를 지정하여 이러한 모델에 액세스할 수 있습니다.
configureGenkit({
plugins: [googleAI({ apiVersion: 'v1beta' })],
});
또는 여러 버전의 모델을 동시에 사용하려면 여러 버전을 지정할 수 있습니다
configureGenkit({
plugins: [googleAI({ apiVersion: ['v1', 'v1beta'] })],
});
사용량
이 플러그인은 지원되는 모델에 대한 참조를 정적으로 내보냅니다.
import {
gemini15Flash,
gemini15Pro,
textEmbeddingGecko001,
} from '@genkit-ai/googleai';
다음 참조를 사용하여 generate()
에서 사용할 모델을 지정할 수 있습니다.
const llmResponse = await generate({
model: gemini15Flash,
prompt: 'Tell me a joke.',
});
임베딩을 사용하거나 textEmbeddingGecko001
)를 embed
또는 검색기와 함께 사용합니다.
const embedding = await embed({
embedder: textEmbeddingGecko001,
content: input,
});