TensorFlow
此嵌入集成使用TensorFlow.js，完全在您的浏览器或 Node.js 环境中运行嵌入。这意味着您的数据不会发送到任何第三方，您也不需要注册任何 API 密钥。但是，它确实需要比其他集成更多的内存和处理能力。
- npm
- Yarn
- pnpm
npm install @tensorflow/[email protected] @tensorflow/[email protected] @tensorflow-models/[email protected] @tensorflow/tfjs-backend-cpu
yarn add @tensorflow/[email protected] @tensorflow/[email protected] @tensorflow-models/[email protected] @tensorflow/tfjs-backend-cpu
pnpm add @tensorflow/[email protected] @tensorflow/[email protected] @tensorflow-models/[email protected] @tensorflow/tfjs-backend-cpu
import "@tensorflow/tfjs-backend-cpu";
import { TensorFlowEmbeddings } from "langchain/embeddings/tensorflow";
const embeddings = new TensorFlowEmbeddings();
此示例使用 CPU 后端，可在任何 JS 环境中运行。但是，您可以使用 TensorFlow.js 支持的任何后端，包括 GPU 和 WebAssembly，它们的速度将快得多。对于 Node.js，您可以使用 @tensorflow/tfjs-node 包，对于浏览器，您可以使用 @tensorflow/tfjs-backend-webgl 包。有关更多信息，请参阅TensorFlow.js 文档。