main.py

import importlib
import json
import logging
import sys
import os

from Muice import Muice
from ws import QQBot

logging.basicConfig(format='[%(levelname)s] %(message)s', level=logging.INFO)

logger = logging.getLogger(__name__)

if not os.path.exists('configs.json'):
    logger.warning("未找到配置configs.json文件，尝试配置")
    try:
        os.system("python configuration_gui.py")
    except:
        logger.error("配置失败，请手动配置configs.json文件")
        os._exit(1)

configs = json.load(open('configs.json', 'r', encoding='utf-8'))

# 模型配置
model_loader = configs["model_loader"]
model_name_or_path = configs["model_name_or_path"]
adapter_name_or_path = configs["adapter_name_or_path"]

# 模型加载
model_adapter = importlib.import_module(f"llm.{model_loader}")
model = model_adapter.llm(model_name_or_path, adapter_name_or_path)

# Faiss配置
enable_faiss = configs.get('enable_faiss', False)
if enable_faiss:
    from llm.faiss_memory import FAISSMemory
    import signal
    memory = FAISSMemory(model_path=configs["sentence_transformer_model_name_or_path"],db_path="./memory/faiss_index.faiss",top_k=2)
    def handle_interrupt(faiss_memory: FAISSMemory):
        """处理中断信号"""
        logging.info("接收到中断信号，正在保存数据...")
        faiss_memory.save_all_data()
        sys.exit(0)
    signal.signal(signal.SIGINT, lambda sig, frame: handle_interrupt(memory))
else:
    memory = None



# OFA图像模型
enable_ofa_image = configs["enable_ofa_image"]
if enable_ofa_image:
    from utils.ofa_image_process import ImageCaptioningPipeline
    ofa_image_model_name_or_path = configs["ofa_image_model_name_or_path"]
    ImageCaptioningPipeline.load_model(ofa_image_model_name_or_path)

# QQBot
muice_app = Muice(model,memory,configs['read_memory_from_file'], configs['known_topic_probability'],
                  configs['time_topic_probability'])
qqbot_app = QQBot(muice_app)
qqbot_app.run()