feat: 重构数据库导入路径,移除旧的数据库模块并添加新的数据库实现

This commit is contained in:
墨梓柒
2025-05-14 19:42:53 +08:00
parent ac6f96f805
commit 307d5a73a6
18 changed files with 18 additions and 16 deletions

View File

@@ -10,7 +10,7 @@ from PIL import Image
import io import io
import re import re
from ...common.database import db from ...common.database.database import db
from ...config.config import global_config from ...config.config import global_config
from ..utils.utils_image import image_path_to_base64, image_manager from ..utils.utils_image import image_path_to_base64, image_manager
from ..models.utils_model import LLMRequest from ..models.utils_model import LLMRequest

View File

@@ -7,7 +7,7 @@ from src.chat.person_info.relationship_manager import relationship_manager
from src.chat.utils.utils import get_embedding from src.chat.utils.utils import get_embedding
import time import time
from typing import Union, Optional, Dict, Any from typing import Union, Optional, Dict, Any
from src.common.database import db from common.database.database import db
from src.chat.utils.utils import get_recent_group_speaker from src.chat.utils.utils import get_recent_group_speaker
from src.manager.mood_manager import mood_manager from src.manager.mood_manager import mood_manager
from src.chat.memory_system.Hippocampus import HippocampusManager from src.chat.memory_system.Hippocampus import HippocampusManager

View File

@@ -10,7 +10,7 @@ import jieba
import networkx as nx import networkx as nx
import numpy as np import numpy as np
from collections import Counter from collections import Counter
from ...common.database import db from ...common.database.database import db
from ...chat.models.utils_model import LLMRequest from ...chat.models.utils_model import LLMRequest
from src.common.logger_manager import get_logger from src.common.logger_manager import get_logger
from src.chat.memory_system.sample_distribution import MemoryBuildScheduler # 分布生成器 from src.chat.memory_system.sample_distribution import MemoryBuildScheduler # 分布生成器

View File

@@ -34,7 +34,7 @@ root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
sys.path.append(root_path) sys.path.append(root_path)
from src.common.logger import get_module_logger # noqa E402 from src.common.logger import get_module_logger # noqa E402
from src.common.database import db # noqa E402 from common.database.database import db # noqa E402
logger = get_module_logger("mem_alter") logger = get_module_logger("mem_alter")
console = Console() console = Console()

View File

@@ -5,7 +5,7 @@ import copy
from typing import Dict, Optional from typing import Dict, Optional
from ...common.database import db from ...common.database.database import db
from maim_message import GroupInfo, UserInfo from maim_message import GroupInfo, UserInfo
from src.common.logger_manager import get_logger from src.common.logger_manager import get_logger

View File

@@ -1,7 +1,7 @@
import re import re
from typing import Union from typing import Union
from ...common.database import db from ...common.database.database import db
from .message import MessageSending, MessageRecv from .message import MessageSending, MessageRecv
from .chat_stream import ChatStream from .chat_stream import ChatStream
from src.common.logger import get_module_logger from src.common.logger import get_module_logger

View File

@@ -12,7 +12,7 @@ import base64
from PIL import Image from PIL import Image
import io import io
import os import os
from ...common.database import db from ...common.database.database import db
from ...config.config import global_config from ...config.config import global_config
from rich.traceback import install from rich.traceback import install

View File

@@ -1,5 +1,5 @@
from src.common.logger_manager import get_logger from src.common.logger_manager import get_logger
from ...common.database import db from ...common.database.database import db
import copy import copy
import hashlib import hashlib
from typing import Any, Callable, Dict from typing import Any, Callable, Dict

View File

@@ -1,6 +1,6 @@
from src.config.config import global_config from src.config.config import global_config
from src.chat.message_receive.message import MessageRecv, MessageSending, Message from src.chat.message_receive.message import MessageRecv, MessageSending, Message
from src.common.database import db from common.database.database import db
import time import time
import traceback import traceback
from typing import List from typing import List

View File

@@ -5,7 +5,7 @@ from typing import Any, Dict, Tuple, List
from src.common.logger import get_module_logger from src.common.logger import get_module_logger
from src.manager.async_task_manager import AsyncTask from src.manager.async_task_manager import AsyncTask
from ...common.database import db from ...common.database.database import db
from src.manager.local_store_manager import local_storage from src.manager.local_store_manager import local_storage
logger = get_module_logger("maibot_statistic") logger = get_module_logger("maibot_statistic")

View File

@@ -13,7 +13,7 @@ from src.manager.mood_manager import mood_manager
from ..message_receive.message import MessageRecv from ..message_receive.message import MessageRecv
from ..models.utils_model import LLMRequest from ..models.utils_model import LLMRequest
from .typo_generator import ChineseTypoGenerator from .typo_generator import ChineseTypoGenerator
from ...common.database import db from ...common.database.database import db
from ...config.config import global_config from ...config.config import global_config
logger = get_module_logger("chat_utils") logger = get_module_logger("chat_utils")

View File

@@ -8,7 +8,7 @@ import io
import numpy as np import numpy as np
from ...common.database import db from ...common.database.database import db
from ...config.config import global_config from ...config.config import global_config
from ..models.utils_model import LLMRequest from ..models.utils_model import LLMRequest

View File

@@ -16,7 +16,7 @@ root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
sys.path.append(root_path) sys.path.append(root_path)
# 现在可以导入src模块 # 现在可以导入src模块
from src.common.database import db # noqa E402 from common.database.database import db # noqa E402
# 加载根目录下的env.edv文件 # 加载根目录下的env.edv文件

View File

@@ -0,0 +1,2 @@
from peewee import *

View File

@@ -1,4 +1,4 @@
from src.common.database import db from common.database.database import db
from src.common.logger import get_module_logger from src.common.logger import get_module_logger
import traceback import traceback
from typing import List, Any, Optional from typing import List, Any, Optional

View File

@@ -1,6 +1,6 @@
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import List, Dict, Any from typing import List, Dict, Any
from src.common.database import db from common.database.database import db
class MessageStorage(ABC): class MessageStorage(ABC):

View File

@@ -1,6 +1,6 @@
from src.tools.tool_can_use.base_tool import BaseTool from src.tools.tool_can_use.base_tool import BaseTool
from src.chat.utils.utils import get_embedding from src.chat.utils.utils import get_embedding
from src.common.database import db from common.database.database import db
from src.common.logger_manager import get_logger from src.common.logger_manager import get_logger
from typing import Any, Union from typing import Any, Union