embedchainjs入门指南 - 轻松创建基于任何数据集的LLM驱动聊天机器人-CSDN博客

本文链接：https://blog.csdn.net/m0_56734068/article/details/142139500

embedchainjs

embedchainjs简介

embedchainjs是embedchain框架的JavaScript版本,旨在帮助开发者快速构建基于大语言模型(LLM)的聊天机器人,可以对任意数据集进行智能问答。它抽象了加载数据集、分块、创建嵌入向量、存储向量等复杂流程,提供了简单易用的API接口。

embedchainjs logo

核心功能

支持添加多种格式的数据源:网页、PDF文件、问答对等
自动处理数据加载、分块、创建嵌入向量等流程
使用向量数据库存储嵌入向量,支持高效相似度搜索
基于OpenAI API实现智能问答
提供简单易用的API,几行代码即可创建聊天机器人

快速开始

安装embedchainjs:

npm install embedchain && npm install -S openai@^3.3.0

创建.env文件,设置OpenAI API密钥:

OPENAI_API_KEY = "sk-xxxx"

创建聊天机器人:

const dotenv = require("dotenv");
dotenv.config();
const { App } = require("embedchain");

async function testApp() {
  const bot = await App();
  
  // 添加数据源
  await bot.add("web_page", "https://example.com/page1");
  await bot.add("pdf_file", "https://example.com/doc.pdf");
  
  // 查询
  const result = await bot.query("如何使用embedchainjs?");
  console.log(result);
}

testApp();