梯度 AI

GradientEmbeddings 类使用 Gradient AI API 为给定文本生成嵌入。

设置

您需要将官方 Gradient Node SDK 安装为对等依赖项

npm

npm i @gradientai/nodejs-sdk

Yarn

yarn add @gradientai/nodejs-sdk

pnpm

pnpm add @gradientai/nodejs-sdk

您需要设置以下环境变量才能使用 Gradient AI API。

export GRADIENT_ACCESS_TOKEN=<YOUR_ACCESS_TOKEN>
export GRADIENT_WORKSPACE_ID=<YOUR_WORKSPACE_ID>

或者，这些可以在 GradientAI 类实例化期间分别设置为 gradientAccessKey 和 workspaceId。例如

const model = new GradientEmbeddings({
  gradientAccessKey: "My secret Access Token"
  workspaceId: "My secret workspace id"
});

使用

import { GradientEmbeddings } from "@langchain/community/embeddings/gradient_ai";

const model = new GradientEmbeddings({});
const res = await model.embedQuery(
  "What would be a good company name a company that makes colorful socks?"
);
console.log({ res });

API 参考

• GradientEmbeddings 来自 @langchain/community/embeddings/gradient_ai

相关

• 嵌入模型 概念指南
• 嵌入模型 操作指南