diff --git a/src/modules/demo/controller/open/plugin.ts b/src/modules/demo/controller/open/plugin.ts new file mode 100644 index 0000000..c2d3310 --- /dev/null +++ b/src/modules/demo/controller/open/plugin.ts @@ -0,0 +1,29 @@ +import { CoolController, BaseController } from '@cool-midway/core'; +import { PluginService } from '../../../plugin/service/info'; +import { Get, Inject } from '@midwayjs/core'; + +/** + * 插件 + */ +@CoolController() +export class OpenDemoPluginController extends BaseController { + @Inject() + pluginService: PluginService; + + @Get('/invoke', { summary: '调用插件' }) + async invoke() { + // 获取插件实例 + const instance: any = await this.pluginService.getInstance('ollama'); + // 调用chat + const messages = [ + { role: 'system', content: '你叫小酷,是一个智能助理' }, + { role: 'user', content: '写一个1000字的关于春天的文章' }, + ]; + for (let i = 0; i < 3; i++) { + instance.chat(messages, { stream: true }, res => { + console.log(i, res.content); + }); + } + return this.ok(); + } +}