@maogo/rag-file NPM

调试相关的代码

config:

// config.default.ts
import { MidwayConfig } from '@midwayjs/core';
import { uploadWhiteList } from '@midwayjs/upload';

export default {
  // use for cookie sign key, should change to your own and keep security
  keys: '1710008275436_4273',
  koa: {
    port: 7001,
  },
  upload: {
    mode: 'stream',
    whitelist: uploadWhiteList.concat(['.txt', '.log']),
  },
} as MidwayConfig;

// configuration.ts
import * as maogoUpload from '@maogo/upload';
import * as upload from '@midwayjs/upload';

imports: [
  // ...
  upload,
  maogoUpload,
]
// ...

controller:

import { Controller, Get, Query, Inject, Post, Files, Fields } from '@midwayjs/core';
import { Context } from '@midwayjs/koa';
import { UploadService as MaogoUploadService } from '@maogo/upload';

@Controller('/')
export class HomeController {
  @Inject()
  ctx: Context;

  @Inject()
  maogoUploadService: MaogoUploadService;

  @Get('/text')
  async text(@Query('text') text: string): Promise<any> {
    console.log('----------- 收到请求 ----------');
    const result = await this.maogoUploadService.textSplit(text, {chunkSize: 10, chunkOverlap: 1});
    console.log('text result: ', result);

    return result;
  }

  @Post('/upload')
  async upload(@Files() files, @Fields() fields): Promise<any> {
    console.log({files, fields});
    return await this.maogoUploadService.fileSplitAndUpload(files[0].data, files[0].filename);
  }
}