PaLM API:Node.js 嵌入快速入门

本快速入门介绍了如何开始使用嵌入服务。

安装 API 客户端

在新目录中,使用 npm 初始化 Node.js 项目并安装 google-auth 库:

npm init -y
npm install google-auth-library

接下来,您需要安装 PaLM API 客户端库:

npm install @google-ai/generativelanguage

生成消息

创建一个新文件 index.js,并添加以下代码,并通过 API_KEY 环境变量提供您的 API 密钥:

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

然后运行该脚本:

node index.js