Gradient AI
GradientEmbeddings 类使用 Gradient AI API 为给定文本生成嵌入(embeddings)。
准备工作
你需要安装官方的 Gradient Node SDK 作为对等依赖:
- npm
- Yarn
- pnpm
npm i @langchain/community @langchain/core @gradientai/nodejs-sdk
yarn add @langchain/community @langchain/core @gradientai/nodejs-sdk
pnpm add @langchain/community @langchain/core @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 Reference:
- GradientEmbeddings from
@langchain/community/embeddings/gradient_ai
相关内容
Related
- Embedding model conceptual guide
- Embedding model how-to guides