fix ruff
This commit is contained in:
@@ -2,7 +2,6 @@ import traceback
|
|||||||
import time
|
import time
|
||||||
import asyncio
|
import asyncio
|
||||||
import random
|
import random
|
||||||
import ast
|
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from typing import List, Optional, Dict, Any, Tuple
|
from typing import List, Optional, Dict, Any, Tuple
|
||||||
|
|||||||
@@ -1,36 +1,14 @@
|
|||||||
import ast
|
import ast
|
||||||
import random
|
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
import hashlib
|
import hashlib
|
||||||
from typing import List, Optional, Dict, Any, Tuple
|
|
||||||
from datetime import datetime
|
|
||||||
|
|
||||||
from src.common.logger import get_logger
|
from src.common.logger import get_logger
|
||||||
from src.config.config import global_config
|
from src.config.config import global_config
|
||||||
from src.llm_models.utils_model import LLMRequest
|
from src.llm_models.utils_model import LLMRequest
|
||||||
from src.chat.message_receive.message import UserInfo, Seg, MessageRecv, MessageSending
|
|
||||||
from src.chat.message_receive.chat_stream import ChatStream
|
|
||||||
from src.chat.message_receive.uni_message_sender import HeartFCSender
|
|
||||||
from src.chat.utils.timer_calculator import Timer # <--- Import Timer
|
|
||||||
from src.chat.utils.utils import get_chat_type_and_target_info
|
|
||||||
from src.chat.utils.prompt_builder import Prompt, global_prompt_manager
|
|
||||||
from src.chat.utils.chat_message_builder import build_readable_messages, get_raw_msg_before_timestamp_with_chat
|
|
||||||
from src.chat.express.expression_selector import expression_selector
|
|
||||||
from src.chat.knowledge.knowledge_lib import qa_manager
|
|
||||||
from src.chat.memory_system.memory_activator import MemoryActivator
|
|
||||||
from src.mood.mood_manager import mood_manager
|
|
||||||
from src.person_info.relationship_fetcher import relationship_fetcher_manager
|
|
||||||
from src.person_info.person_info import get_person_info_manager
|
from src.person_info.person_info import get_person_info_manager
|
||||||
from src.tools.tool_executor import ToolExecutor
|
|
||||||
from src.plugin_system.base.component_types import ActionInfo
|
|
||||||
from typing import Optional
|
|
||||||
from rich.traceback import install
|
from rich.traceback import install
|
||||||
|
|
||||||
from src.common.logger import get_logger
|
|
||||||
from src.config.config import global_config
|
|
||||||
from src.llm_models.utils_model import LLMRequest
|
|
||||||
from src.person_info.person_info import get_person_info_manager
|
|
||||||
from .personality import Personality
|
from .personality import Personality
|
||||||
|
|
||||||
install(extra_lines=3)
|
install(extra_lines=3)
|
||||||
|
|||||||
@@ -1,8 +1,6 @@
|
|||||||
import json
|
|
||||||
|
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import Dict, List, Optional
|
from typing import Dict, List
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
|
|||||||
@@ -5,8 +5,6 @@ from datetime import datetime
|
|||||||
from typing import Dict, List, Optional
|
from typing import Dict, List, Optional
|
||||||
from aiohttp import web, WSMsgType
|
from aiohttp import web, WSMsgType
|
||||||
import aiohttp_cors
|
import aiohttp_cors
|
||||||
from threading import Thread
|
|
||||||
import weakref
|
|
||||||
|
|
||||||
from src.chat.message_receive.message import MessageRecv
|
from src.chat.message_receive.message import MessageRecv
|
||||||
from src.common.logger import get_logger
|
from src.common.logger import get_logger
|
||||||
@@ -599,7 +597,7 @@ class ContextWebManager:
|
|||||||
logger.info(f"✅ 添加消息到上下文 [总数: {total_messages}]: [{context_msg.group_name}] {context_msg.user_name}: {context_msg.content}")
|
logger.info(f"✅ 添加消息到上下文 [总数: {total_messages}]: [{context_msg.group_name}] {context_msg.user_name}: {context_msg.content}")
|
||||||
|
|
||||||
# 调试:打印当前所有消息
|
# 调试:打印当前所有消息
|
||||||
logger.info(f"📝 当前上下文中的所有消息:")
|
logger.info("📝 当前上下文中的所有消息:")
|
||||||
for cid, contexts in self.contexts.items():
|
for cid, contexts in self.contexts.items():
|
||||||
logger.info(f" 聊天 {cid}: {len(contexts)} 条消息")
|
logger.info(f" 聊天 {cid}: {len(contexts)} 条消息")
|
||||||
for i, msg in enumerate(contexts):
|
for i, msg in enumerate(contexts):
|
||||||
|
|||||||
@@ -1,14 +1,4 @@
|
|||||||
import asyncio
|
|
||||||
import json
|
|
||||||
import time
|
|
||||||
|
|
||||||
from src.chat.message_receive.message import MessageRecv
|
|
||||||
from src.llm_models.utils_model import LLMRequest
|
|
||||||
from src.common.logger import get_logger
|
|
||||||
from src.chat.utils.chat_message_builder import build_readable_messages, get_raw_msg_by_timestamp_with_chat_inclusive
|
|
||||||
from src.config.config import global_config
|
|
||||||
from src.chat.utils.prompt_builder import Prompt, global_prompt_manager
|
|
||||||
from src.manager.async_task_manager import AsyncTask, async_task_manager
|
|
||||||
from src.plugin_system.apis import send_api
|
from src.plugin_system.apis import send_api
|
||||||
|
|
||||||
async def send_loading(chat_id: str, content: str):
|
async def send_loading(chat_id: str, content: str):
|
||||||
|
|||||||
@@ -250,7 +250,7 @@ class S4UChat:
|
|||||||
await super_chat_manager.add_superchat(message)
|
await super_chat_manager.add_superchat(message)
|
||||||
else:
|
else:
|
||||||
await self.relationship_builder.build_relation(20)
|
await self.relationship_builder.build_relation(20)
|
||||||
except Exception as e:
|
except Exception:
|
||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
|
|
||||||
logger.info(f"[{self.stream_name}] 消息处理完毕,消息内容:{message.processed_plain_text}")
|
logger.info(f"[{self.stream_name}] 消息处理完毕,消息内容:{message.processed_plain_text}")
|
||||||
|
|||||||
@@ -394,7 +394,7 @@ class MoodRegressionTask(AsyncTask):
|
|||||||
if regression_executed > 0:
|
if regression_executed > 0:
|
||||||
logger.info(f"[回归任务] 本次执行了{regression_executed}个聊天的情绪回归")
|
logger.info(f"[回归任务] 本次执行了{regression_executed}个聊天的情绪回归")
|
||||||
else:
|
else:
|
||||||
logger.debug(f"[回归任务] 本次没有符合回归条件的聊天")
|
logger.debug("[回归任务] 本次没有符合回归条件的聊天")
|
||||||
|
|
||||||
|
|
||||||
class MoodManager:
|
class MoodManager:
|
||||||
|
|||||||
@@ -10,10 +10,10 @@ from datetime import datetime
|
|||||||
import asyncio
|
import asyncio
|
||||||
from src.mais4u.s4u_config import s4u_config
|
from src.mais4u.s4u_config import s4u_config
|
||||||
import ast
|
import ast
|
||||||
from src.chat.message_receive.message import MessageSending, MessageRecvS4U
|
from src.chat.message_receive.message import MessageRecvS4U
|
||||||
from src.person_info.person_info import get_person_info_manager
|
from src.person_info.person_info import get_person_info_manager
|
||||||
from src.person_info.relationship_manager import get_relationship_manager
|
from src.person_info.relationship_manager import get_relationship_manager
|
||||||
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
|
from src.chat.message_receive.chat_stream import ChatStream
|
||||||
from src.mais4u.mais4u_chat.super_chat_manager import get_super_chat_manager
|
from src.mais4u.mais4u_chat.super_chat_manager import get_super_chat_manager
|
||||||
|
|
||||||
logger = get_logger("prompt")
|
logger = get_logger("prompt")
|
||||||
|
|||||||
@@ -3,8 +3,7 @@ import time
|
|||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import Dict, List, Optional
|
from typing import Dict, List, Optional
|
||||||
from src.common.logger import get_logger
|
from src.common.logger import get_logger
|
||||||
from src.chat.message_receive.message import MessageRecvS4U, MessageRecv
|
from src.chat.message_receive.message import MessageRecvS4U
|
||||||
from src.mais4u.s4u_config import s4u_config
|
|
||||||
|
|
||||||
logger = get_logger("super_chat_manager")
|
logger = get_logger("super_chat_manager")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user