Plug-in da IA generativa do Google

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 {
  geminiPro,
  geminiProVision,
  textEmbeddingGecko001,
} from '@genkit-ai/googleai';

Use essas referências para especificar qual modelo generate() usa:

const llmResponse = await generate({
  model: geminiPro,
  prompt: 'Tell me a joke.',
});

ou use embedders (por exemplo, textEmbeddingGecko001) com embed ou retrievers:

const embedding = await embed({
  embedder: textEmbeddingGecko001,
  content: input,
});