feat: add support for chatglm

utils/chatglm.js

@@ -0,0 +1,106 @@
+import { Config } from './config.js'
+import fetch from 'node-fetch'
+import { v4 as uuidv4 } from 'uuid'
+async function getKeyv () {
+  let Keyv
+  try {
+    Keyv = (await import('keyv')).default
+  } catch (error) {
+    throw new Error('keyv依赖未安装，请使用pnpm install keyv安装')
+  }
+  return Keyv
+}
+
+export default class ChatGLMClient {
+  constructor (opts) {
+    // user: qq号
+    this.opts = opts
+  }
+
+  async initCache () {
+    if (!this.conversationsCache) {
+      const cacheOptions = this.opts.cache || {}
+      cacheOptions.namespace = cacheOptions.namespace || 'chatglm'
+      let Keyv = await getKeyv()
+      this.conversationsCache = new Keyv(cacheOptions)
+    }
+  }
+
+  async sendMessage (prompt, opts) {
+    const {
+      conversationId = uuidv4(),
+      messageId = uuidv4(),
+      parentMessageId,
+      temperature = Config.temperature
+    } = opts
+    await this.initCache()
+    let url = Config.chatglmBaseUrl + '/api/chat'
+    if (Config.debug) {
+      logger.info('use chatglm api server endpoint: ' + url)
+    }
+    const conversationKey = `ChatGLMUser_${this.opts.user}`
+    const conversation = (await this.conversationsCache.get(conversationKey)) || {
+      messages: [],
+      createdAt: Date.now()
+    }
+    let history = getMessagesForConversation(conversation.messages, parentMessageId)
+    if (Config.debug) {
+      logger.info(history)
+    }
+    console.log(history)
+    let option = {
+      method: 'POST',
+      body: JSON.stringify({
+        prompt,
+        temperature,
+        history
+      }),
+      headers: {
+        'content-type': 'application/json',
+        library: 'chatgpt-plugin'
+      }
+    }
+    let response = await fetch(url, option)
+    let result = await response.text()
+    try {
+      result = JSON.parse(result)
+      conversation.messages.push({
+        id: messageId,
+        role: 'user',
+        content: prompt,
+        parentMessageId
+      })
+      let responseId = uuidv4()
+      conversation.messages.push({
+        id: responseId,
+        role: 'AI',
+        content: result.data,
+        parentMessageId: messageId
+      })
+      await this.conversationsCache.set(conversationKey, conversation)
+      return {
+        conversationId,
+        id: responseId,
+        text: result.data
+      }
+    } catch (e) {
+      console.error(result)
+      throw new Error(result)
+    }
+  }
+}
+
+function getMessagesForConversation (messages, parentMessageId) {
+  const orderedMessages = []
+  let currentMessageId = parentMessageId
+  while (currentMessageId) {
+    const message = messages.find((m) => m.id === currentMessageId)
+    if (!message) {
+      break
+    }
+    orderedMessages.unshift(message)
+    currentMessageId = message.parentMessageId
+  }
+
+  return orderedMessages
+}