API PaLM: guia de início rápido de embeddings com o Node.js

Neste guia de início rápido, mostramos como começar a usar o serviço de incorporação.

Instalar o cliente da API

Em um novo diretório, inicialize um projeto Node.js usando npm e instale a biblioteca google-auth:

npm init -y
npm install google-auth-library

Em seguida, instale a biblioteca de cliente da API PaLM:

npm install @google-ai/generativelanguage

Gerar mensagens

Crie um novo arquivo index.js e adicione o seguinte código, fornecendo sua chave de API por meio da variável de ambiente API_KEY:

const { TextServiceClient } =
  require("@google-ai/generativelanguage").v1;

const { GoogleAuth } = require("google-auth-library");

const MODEL_NAME = "models/embedding-gecko-001";
const API_KEY = process.env.API_KEY;

const client = new TextServiceClient({
  authClient: new GoogleAuth().fromAPIKey(API_KEY),
});

const text = "Repeat after me: one, two,";

client
  .embedText({
    model: MODEL_NAME,
    text: text,
  })
  .then((result) => {
    console.log(JSON.stringify(result));
  });

Em seguida, execute o script:

node index.js