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