chore: src/modules/demo/controller/open/plugin.ts

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();
+  }
+}