q-koa 11.0.5 → 11.0.7

package/core/file/plugins/common/controller.js

@@ -563,3 +563,15 @@ exports.sql = async (ctx) => {
 
   ctx.SUCCESS(result)
 }
+
+exports.gpt = async (ctx) => {
+  const { app, appName } = getAppByCtx(ctx)
+  const appConfig = getConfig(app)
+  const { content } = ctx.request.body
+  if (!content) return ctx.ERROR('?content')
+  if (!app.service.common) throw new Error(`common可能没有service.js`)
+  const result = await app.service.common.gpt({
+    content,
+  })
+  return ctx.SUCCESS(result)
+}

package/core/file/plugins/common/service.js

@@ -0,0 +1,36 @@
+const axios = require('axios')
+exports.gpt = async ({ content }) => {
+  let result = ''
+  let messages = [
+    {
+      content,
+      role: 'user',
+    },
+  ]
+
+  for (let i = 0; i < 1000; i++) {
+    try {
+      const { choices } = await axios
+        .post('https://gpt.likangjian.cn:3001/gpt/chat', {
+          messages,
+        })
+        .then((res) => res.data)
+      result += choices[0].message.content
+      if (choices.finish_reason === 'stop') {
+        break
+      }
+      messages = [
+        ...messages,
+        choices[0].message,
+        {
+          content: '还有呢',
+          role: 'user',
+        },
+      ]
+    } catch (e) {
+      throw new Error('ai开小差了')
+    }
+  }
+
+  return result
+}

package/core/file/plugins/user/controller.js

@@ -325,7 +325,7 @@ exports.checkLogin = async (ctx) => {
         model: app.model.github_user,
       },
       config === 'weixin_h5' ? h5_user_include : {},
-      config === 'weixin_mp' ? mp_user_include : {},
+      config.includes('weixin_mp') ? mp_user_include : {},
       {
         model: app.model.toutiao_user,
       },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "q-koa",
-  "version": "11.0.5",
+  "version": "11.0.7",
   "description": "",
   "main": "index.js",
   "scripts": {