This commit is contained in:
Pliosauroidea
2025-03-12 08:14:15 +08:00
3 changed files with 8 additions and 4 deletions

View File

@@ -235,10 +235,10 @@ class ChatBot:
is_head=not mark_head, is_head=not mark_head,
is_emoji=False, is_emoji=False,
) )
print(f"bot_message: {bot_message}") logger.debug(f"bot_message: {bot_message}")
if not mark_head: if not mark_head:
mark_head = True mark_head = True
print(f"添加消息到message_set: {bot_message}") logger.debug(f"添加消息到message_set: {bot_message}")
message_set.add_message(bot_message) message_set.add_message(bot_message)
# message_set 可以直接加入 message_manager # message_set 可以直接加入 message_manager

View File

@@ -235,7 +235,7 @@ class LLM_request:
delta_content = "" delta_content = ""
accumulated_content += delta_content accumulated_content += delta_content
# 检测流式输出文本是否结束 # 检测流式输出文本是否结束
finish_reason = chunk["choices"][0]["finish_reason"] finish_reason = chunk["choices"][0].get("finish_reason")
if finish_reason == "stop": if finish_reason == "stop":
usage = chunk.get("usage", None) usage = chunk.get("usage", None)
if usage: if usage:

View File

@@ -13,6 +13,8 @@ from pathlib import Path
import jieba import jieba
from pypinyin import Style, pinyin from pypinyin import Style, pinyin
from loguru import logger
class ChineseTypoGenerator: class ChineseTypoGenerator:
def __init__(self, def __init__(self,
@@ -38,7 +40,9 @@ class ChineseTypoGenerator:
self.max_freq_diff = max_freq_diff self.max_freq_diff = max_freq_diff
# 加载数据 # 加载数据
print("正在加载汉字数据库,请稍候...") # print("正在加载汉字数据库,请稍候...")
logger.info("正在加载汉字数据库,请稍候...")
self.pinyin_dict = self._create_pinyin_dict() self.pinyin_dict = self._create_pinyin_dict()
self.char_frequency = self._load_or_create_char_frequency() self.char_frequency = self._load_or_create_char_frequency()