调整目录,FIXME 待解决插件加载时,只读取目录下的main.py方法,其他过滤
This commit is contained in:
92
utils/compress_chat_data.py
Normal file
92
utils/compress_chat_data.py
Normal file
@@ -0,0 +1,92 @@
|
||||
import re
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
def compress_chat_data(chat_data_str, time_threshold=5):
|
||||
"""
|
||||
压缩聊天数据,减少 token 使用,格式为时间,发信人,内容。
|
||||
|
||||
:param chat_data_str: 原始聊天记录的长字符串
|
||||
:param time_threshold: 同一发信人连续发言间隔小于该值(秒),则合并
|
||||
:return: 压缩后的聊天数据的长字符串
|
||||
"""
|
||||
# 如果字符串长度超过30000,则去除前面的聊天记录
|
||||
if len(chat_data_str) > 30000:
|
||||
lines = chat_data_str.splitlines()
|
||||
total_length = 0
|
||||
cut_index = 0
|
||||
|
||||
# 从后往前计算,找到保留哪些行
|
||||
for i in range(len(lines) - 1, -1, -1):
|
||||
line_length = len(lines[i]) + 1 # +1 是为了计入换行符
|
||||
total_length += line_length
|
||||
if total_length > 30000:
|
||||
cut_index = i + 1 # 保留这个索引之后的行
|
||||
break
|
||||
|
||||
# 只保留后面的聊天记录
|
||||
chat_data_str = '\n'.join(lines[cut_index:])
|
||||
|
||||
# 解析原始聊天数据为列表
|
||||
chat_data = []
|
||||
for line in chat_data_str.splitlines():
|
||||
# 跳过空行
|
||||
if not line.strip():
|
||||
continue
|
||||
|
||||
# 分割每一行,确保有3部分(时间, 发信人, 内容)
|
||||
parts = line.split(',', 2)
|
||||
if len(parts) < 3:
|
||||
continue # 如果没有完整的三部分,跳过该行
|
||||
|
||||
timestamp, sender, content = parts
|
||||
chat_data.append((timestamp, sender, content))
|
||||
|
||||
# 其余代码保持不变
|
||||
if not chat_data:
|
||||
return "" # 如果没有有效数据,返回空字符串
|
||||
|
||||
compressed_data = []
|
||||
current_date = None
|
||||
prev_time_obj = None
|
||||
prev_sender = None
|
||||
prev_content = None
|
||||
|
||||
for timestamp, sender, content in chat_data:
|
||||
try:
|
||||
time_obj = datetime.strptime(timestamp.strip(), "%Y-%m-%d %H:%M:%S")
|
||||
|
||||
# 去除无意义的空格
|
||||
content = re.sub(r"\s+", " ", content).strip()
|
||||
if not content:
|
||||
continue
|
||||
|
||||
# 检查日期是否变化(不包含年份)
|
||||
date_str = time_obj.strftime('%m-%d')
|
||||
|
||||
# 如果是第一条消息或日期变化,添加日期标记
|
||||
if current_date is None or date_str != current_date:
|
||||
current_date = date_str
|
||||
compressed_data.append(f"{date_str}")
|
||||
|
||||
# 检查是否需要合并消息
|
||||
if (prev_sender == sender and prev_time_obj is not None and
|
||||
(time_obj - prev_time_obj).total_seconds() <= time_threshold):
|
||||
# 合并消息,更新最后一条消息
|
||||
compressed_data[-1] = f"{prev_time_obj.strftime('%H:%M:%S')},{sender},{prev_content} {content}"
|
||||
else:
|
||||
# 添加新消息
|
||||
compressed_data.append(f"{time_obj.strftime('%H:%M:%S')},{sender},{content}")
|
||||
|
||||
# 更新前一条消息的信息
|
||||
prev_time_obj = time_obj
|
||||
prev_sender = sender
|
||||
prev_content = content
|
||||
|
||||
except ValueError as e:
|
||||
# 处理日期格式错误
|
||||
print(f"日期格式错误: {timestamp}, 错误: {e}")
|
||||
continue
|
||||
|
||||
# 返回压缩后的数据长字符串
|
||||
return '\n'.join(compressed_data)
|
||||
Reference in New Issue
Block a user