O plug-in de IA generativa do Google fornece interfaces para os modelos Gemini do Google por meio da API Generative.
Instalação
npm i --save @genkit-ai/googleai
Configuração
Para usar esse plug-in, especifique-o ao chamar configureGenkit()
:
import { googleAI } from '@genkit-ai/googleai';
export default configureGenkit({
plugins: [googleAI()],
// ...
});
O plug-in requer uma chave de API para a API Gemini, que pode ser encontrada no Google AI Studio.
Configure o plug-in para usar sua chave de API seguindo um destes procedimentos:
Defina a variável de ambiente
GOOGLE_GENAI_API_KEY
como sua chave de API.Especifique a chave de API ao inicializar o plug-in:
googleAI({ apiKey: yourKey });
No entanto, não incorpore a chave de API diretamente no código. Use esse recurso apenas em conjunto com um serviço como o Cloud Secret Manager ou semelhante.
Alguns modelos, como o Gemini 1.5 Pro, estão em pré-lançamento e só podem ser acessados pela
API v1beta
. Especifique apiVersion
para ter acesso a esses modelos:
configureGenkit({
plugins: [googleAI({ apiVersion: 'v1beta' })],
});
ou especificar várias versões para usar diferentes versões de modelos ao mesmo tempo.
configureGenkit({
plugins: [googleAI({ apiVersion: ['v1', 'v1beta'] })],
});
Uso
Este plug-in exporta estaticamente referências para os modelos compatíveis:
import {
gemini15Flash,
gemini15Pro,
textEmbeddingGecko001,
} from '@genkit-ai/googleai';
Use essas referências para especificar qual modelo generate()
usa:
const llmResponse = await generate({
model: gemini15Flash,
prompt: 'Tell me a joke.',
});
ou use embedders (por exemplo, textEmbeddingGecko001
) com embed
ou retrievers:
const embedding = await embed({
embedder: textEmbeddingGecko001,
content: input,
});