Prem AI

PremEmbeddings 类使用 Prem AI API 为给定的文本生成嵌入。

设置

为了使用 Prem API，您需要一个 API 密钥。您可以注册 Prem 帐户并创建一个 API 密钥 此处。

首先需要安装 @langchain/community 包

提示

有关安装集成包的一般说明，请参阅本节。

npm

npm install @langchain/community @langchain/core

Yarn

yarn add @langchain/community @langchain/core

pnpm

pnpm add @langchain/community @langchain/core

用法

import { PremEmbeddings } from "@langchain/community/embeddings/premai";

const embeddings = new PremEmbeddings({
  // In Node.js defaults to process.env.PREM_API_KEY
  apiKey: "YOUR-API-KEY",
  // In Node.js defaults to process.env.PREM_PROJECT_ID
  project_id: "YOUR-PROJECT_ID",
  model: "@cf/baai/bge-small-en-v1.5", // The model to generate the embeddings
});

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

API 参考

• PremEmbeddings 来自 @langchain/community/embeddings/premai

相关

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