1. 记录保存定时清楚

2. 总结评率控制

Author: sineom

README.md

@@ -0,0 +1,36 @@
+## 插件说明
+
+## 一个基于[ChatGPT-on-Wechat](https://github.com/zhayujie/chatgpt-on-wechat)项目的简单插件，
+由于[原插件](https://github.com/lanvent/plugin_summary)安装始终失败，故在此基础上进行了改写。
+
+该插件也是用户根据需求开发自定义插件的示例插件，参考[插件开发说明](https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins)
+
+## 插件配置
+
+将 `plugins/plugin_summary` 目录下的 `config.json.template` 配置模板复制为最终生效的 `config.json`。 (如果未配置则会默认使用`config.json.template`模板中配置)。
+
+
+以下是插件配置项说明：
+
+```bash
+{
+ "rate_limit_summary":60, # 总结间隔时间(单位分钟)，防止同一时间多次触发总结，浪费token
+ "save_time":  1440 # 聊天记录保存时间(单位分钟)，默认保留12小时，凌晨12点将过去12小时之前的记录清楚.-1表示永久保留
+}
+
+```
+
+## 指令参考
+- $总结 999
+- $总结 3 小时内消息
+- $总结 开启
+- $总结 关闭
+
+
+注意：
+ - 总结默认针对所有群开放，关闭请在对应群发送关闭指令 
+ - 实际 `config.json` 配置中应保证json格式，不应携带 '#' 及后面的注释
+ - 如果是`docker`部署，可通过映射 `plugins/config.json` 到容器中来完成插件配置，参考[文档](https://github.com/zhayujie/chatgpt-on-wechat#3-%E6%8F%92%E4%BB%B6%E4%BD%BF%E7%94%A8)
+
+
+

config.json.template

@@ -1,3 +1,4 @@
 {
- "rate_limit_summary":60
+ "rate_limit_summary":60,
+ "save_time": 1440
 }

db.py

@@ -0,0 +1,135 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+"""
+@Author sineom
+@Date 2024/7/23-09:20
+@Email [email protected]
+@description  sqlite操作
+@Copyright (c) 2022 by sineom, All Rights Reserved.
+"""
+import os
+import sqlite3
+
+from common.log import logger
+
+
+class Db:
+    def __init__(self):
+        curdir = os.path.dirname(__file__)
+        db_path = os.path.join(curdir, "chat.db")
+        self.conn = sqlite3.connect(db_path, check_same_thread=False)
+        c = self.conn.cursor()
+        c.execute('''CREATE TABLE IF NOT EXISTS chat_records
+                            (sessionid TEXT, msgid INTEGER, user TEXT, content TEXT, type TEXT, timestamp INTEGER, is_triggered INTEGER,
+                            PRIMARY KEY (sessionid, msgid))''')
+
+        # 创建一个总结时间表，记录合适开始了总结的时间
+        c.execute('''CREATE TABLE IF NOT EXISTS summary_time
+                            (sessionid TEXT, summary_time INTEGER, PRIMARY KEY (sessionid))''')
+
+        # 创建一个关闭保存聊天记录的表
+        c.execute('''CREATE TABLE IF NOT EXISTS summary_stop
+                            (sessionid TEXT, PRIMARY KEY (sessionid))''')
+
+        # 后期增加了is_triggered字段，这里做个过渡，这段代码某天会删除
+        c = c.execute("PRAGMA table_info(chat_records);")
+        column_exists = False
+        for column in c.fetchall():
+            logger.debug("[Summary] column: {}".format(column))
+            if column[1] == 'is_triggered':
+                column_exists = True
+                break
+        if not column_exists:
+            self.conn.execute("ALTER TABLE chat_records ADD COLUMN is_triggered INTEGER DEFAULT 0;")
+            self.conn.execute("UPDATE chat_records SET is_triggered = 0;")
+
+        self.conn.commit()
+        # 禁用的群聊
+        self.disable_group = self._get_summary_stop()
+
+    def insert_record(self, session_id, msg_id, user, content, msg_type, timestamp, is_triggered=0):
+        c = self.conn.cursor()
+        logger.debug("[Summary] insert record: {} {} {} {} {} {} {}".format(session_id, msg_id, user, content, msg_type,
+                                                                            timestamp, is_triggered))
+        c.execute("INSERT OR REPLACE INTO chat_records VALUES (?,?,?,?,?,?,?)",
+                  (session_id, msg_id, user, content, msg_type, timestamp, is_triggered))
+        self.conn.commit()
+
+    # 根据时间删除记录
+    def delete_records(self, start_timestamp):
+        try:
+            c = self.conn.cursor()
+            c.execute('''
+                        DELETE FROM chat_records
+                        WHERE timestamp < ?
+                    ''', start_timestamp,)
+            self.conn.commit()
+            logger.info("Records older have been cleaned.")
+        except Exception as e:
+            logger.error(e)
+
+    # 保存总结时间，如果表中不存在则插入，如果存在则更新
+    def save_summary_time(self, session_id, summary_time):
+        if self.get_summary_time(session_id) is None:
+            self._insert_summary_time(session_id, summary_time)
+        else:
+            self._update_summary_time(session_id, summary_time)
+
+    # 插入总结时间
+    def _insert_summary_time(self, session_id, summary_time):
+        c = self.conn.cursor()
+        logger.debug("[Summary] insert summary time: {} {}".format(session_id, summary_time))
+        c.execute("INSERT OR REPLACE INTO summary_time VALUES (?,?)",
+                  (session_id, summary_time))
+        self.conn.commit()
+
+    # 更新总结时间
+    def _update_summary_time(self, session_id, summary_time):
+        c = self.conn.cursor()
+        logger.debug("[Summary] update summary time: {} {}".format(session_id, summary_time))
+        c.execute("UPDATE summary_time SET summary_time = ? WHERE sessionid = ?",
+                  (summary_time, session_id))
+        self.conn.commit()
+
+    # 获取总结时间，如果不存在返回None
+    def get_summary_time(self, session_id):
+        c = self.conn.cursor()
+        c.execute("SELECT summary_time FROM summary_time WHERE sessionid=?", (session_id,))
+        row = c.fetchone()
+        if row is None:
+            return None
+        return row[0]
+
+    def get_records(self, session_id, start_timestamp=0, limit=9999) -> list:
+        c = self.conn.cursor()
+        c.execute("SELECT * FROM chat_records WHERE sessionid=? and timestamp>? ORDER BY timestamp DESC LIMIT ?",
+                  (session_id, start_timestamp, limit))
+        return c.fetchall()
+
+    # 删除禁用的群聊
+    def delete_summary_stop(self, session_id):
+        try:
+            c = self.conn.cursor()
+            c.execute("DELETE FROM summary_stop WHERE sessionid=?", (session_id,))
+            self.conn.commit()
+            if session_id in self.disable_group:
+                self.disable_group.remove(session_id)
+        except Exception as e:
+            logger.error(e)
+
+    # 保存禁用的群聊
+    def save_summary_stop(self, session_id):
+        try:
+            c = self.conn.cursor()
+            c.execute("INSERT OR REPLACE INTO summary_stop VALUES (?)",
+                      (session_id,))
+            self.conn.commit()
+            self.disable_group.add(session_id)
+        except Exception as e:
+            logger.error(e)
+
+    # 获取所有禁用的群聊
+    def _get_summary_stop(self):
+        c = self.conn.cursor()
+        c.execute("SELECT sessionid FROM summary_stop")
+        return set(c.fetchall())