Prem AI
PremEmbeddings 类使用 Prem AI API 为给定文本生成嵌入(embeddings)。
准备工作
要使用 Prem API,您需要一个 API 密钥。您可以在此处注册 Prem 账户并创建 API 密钥:注册链接。
首先,您需要安装 @langchain/community 包:
:::提示 请参阅安装集成包的一般说明部分。 :::
- npm
- Yarn
- pnpm
npm install @langchain/community @langchain/core
yarn add @langchain/community @langchain/core
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 Reference:
- PremEmbeddings from
@langchain/community/embeddings/premai
相关内容
Related
- Embedding model conceptual guide
- Embedding model how-to guides