Nodejs Pinecone远程向量库文件上传(需分割文档)

Pinecone远程向量库文件上传

安装包

npm i @pinecone-database/pinecone langchain

示例代码

const { PineconeClient } = require("@pinecone-database/pinecone");
const { PineconeStore } = require("langchain/vectorstores/pinecone");
const { OpenAIEmbeddings } = require("langchain/embeddings/openai");
/**
     * Pinecone数据库向量上传
     * @param {Object}  data 需要的参数对象 
     * @param {Array}  data.splitterDocs langchain已经分割好的文档
     * @param {String}  data.namespace pinecone数据库中的命名空间
     * @returns {promises} upload_status 上传状态
     */
  async fileUpload({ splitterDocs,namespace }) {
        let upload_status ;
        const { apiKey, environment } = this.ctx.app.config.pinecone; // Pinecone 连接需要的参数
        const client = new PineconeClient();
        await client.init({ apiKey, environment });
        const pineconeIndex = client.Index('first-database'); //Index 再Pinecone官网控制台可以看到 
        await PineconeStore.fromDocuments(splitterDocs, new OpenAIEmbeddings({
            openAIApiKey
        }, {
            basePath
        }), {
            pineconeIndex,
            namespace,
        }).then(res => {
            console.log('操作成功');
            upload_status = 'Success'
        }).catch(err => {
            console.log(err, 'err');
            upload_status = 'Fail'
        }).finally(() => {
        let { filepath } = this.ctx.request.files[0]; 
            fs.unlink(filepath, (err) => {
                console.log(err, 'File deleted!');
            });
        });
        return upload_status
    }