PaLM API: Kurzanleitung für Einbettungen mit Node.js

In dieser Kurzanleitung werden die ersten Schritte mit dem Einbettungsdienst beschrieben.

API-Client installieren

Initialisieren Sie in einem neuen Verzeichnis ein Node.js-Projekt mit npm und installieren Sie die Bibliothek google-auth:

npm init -y
npm install google-auth-library

Als Nächstes müssen Sie die PaLM API-Clientbibliothek installieren:

npm install @google-ai/generativelanguage

Nachrichten erstellen

Erstellen Sie eine neue Datei index.js, fügen Sie den folgenden Code ein und geben Sie Ihren API-Schlüssel über die Umgebungsvariable API_KEY an:

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));
  });

Führen Sie dann das Skript aus:

node index.js