修复部分 lint

This commit is contained in:
Asankilp 2025-02-17 01:05:19 +08:00
parent 0e72880167
commit 0b6ac9f73e
5 changed files with 15 additions and 19 deletions

View File

@ -1,4 +1,3 @@
import shutil
from io import StringIO from io import StringIO
from pathlib import Path from pathlib import Path
@ -81,15 +80,15 @@ destination_folder = Path("config/marshoai/")
destination_file = destination_folder / "config.yaml" destination_file = destination_folder / "config.yaml"
def dump_config_to_yaml(config: ConfigModel): def dump_config_to_yaml(cfg: ConfigModel):
return yaml_.dump(config.model_dump(), allow_unicode=True, default_flow_style=False) return yaml_.dump(cfg.model_dump(), allow_unicode=True, default_flow_style=False)
def write_default_config(destination_file): def write_default_config(dest_file):
""" """
写入默认配置 写入默认配置
""" """
with open(destination_file, "w", encoding="utf-8") as f: with open(dest_file, "w", encoding="utf-8") as f:
with StringIO(dump_config_to_yaml(ConfigModel())) as f2: with StringIO(dump_config_to_yaml(ConfigModel())) as f2:
f.write(f2.read()) f.write(f2.read())
@ -110,17 +109,17 @@ def check_yaml_is_changed():
return True return True
def merge_configs(old_config, new_config): def merge_configs(existing_cfg, new_cfg):
""" """
合并配置文件 合并配置文件
""" """
for key, value in new_config.items(): for key, value in new_cfg.items():
if key in old_config: if key in existing_cfg:
continue continue
else: else:
logger.info(f"新增配置项: {key} = {value}") logger.info(f"新增配置项: {key} = {value}")
old_config[key] = value existing_cfg[key] = value
return old_config return existing_cfg
config: ConfigModel = get_plugin_config(ConfigModel) config: ConfigModel = get_plugin_config(ConfigModel)

View File

@ -1,6 +1,4 @@
# Marsho 的类实例以及全局变量 # Marsho 的类实例以及全局变量
from azure.ai.inference.aio import ChatCompletionsClient
from azure.core.credentials import AzureKeyCredential
from nonebot import get_driver from nonebot import get_driver
from openai import AsyncOpenAI from openai import AsyncOpenAI

View File

@ -2,7 +2,6 @@ import contextlib
import traceback import traceback
from typing import Optional from typing import Optional
import openai
from arclet.alconna import Alconna, AllParam, Args from arclet.alconna import Alconna, AllParam, Args
from azure.ai.inference.models import ( from azure.ai.inference.models import (
AssistantMessage, AssistantMessage,
@ -18,10 +17,11 @@ from nonebot.adapters import Bot, Event, Message
from nonebot.matcher import Matcher from nonebot.matcher import Matcher
from nonebot.params import CommandArg from nonebot.params import CommandArg
from nonebot.permission import SUPERUSER from nonebot.permission import SUPERUSER
from nonebot.rule import Rule, to_me from nonebot.rule import to_me
from nonebot.typing import T_State from nonebot.typing import T_State
from nonebot_plugin_alconna import MsgTarget, UniMessage, UniMsg, on_alconna from nonebot_plugin_alconna import MsgTarget, UniMessage, UniMsg, on_alconna
from .constants import INTRODUCTION, OPENAI_NEW_MODELS, SUPPORT_IMAGE_MODELS
from .hooks import * from .hooks import *
from .instances import * from .instances import *
from .metadata import metadata from .metadata import metadata

View File

@ -9,7 +9,6 @@ import traceback
from nonebot import logger from nonebot import logger
from .config import config from .config import config
from .util import *
class MarshoContext: class MarshoContext:

View File

@ -21,12 +21,12 @@ from zhDateTime import DateTime
from ._types import DeveloperMessage from ._types import DeveloperMessage
from .config import config from .config import config
from .constants import * from .constants import CODE_BLOCK_PATTERN, IMG_LATEX_PATTERN, OPENAI_NEW_MODELS
from .deal_latex import ConvertLatex from .deal_latex import ConvertLatex
nickname_json = None # 记录昵称 nickname_json = None # 记录昵称
praises_json = None # 记录夸赞名单 praises_json = None # 记录夸赞名单
loaded_target_list = [] # 记录已恢复备份的上下文的列表 loaded_target_list: List[str] = [] # 记录已恢复备份的上下文的列表
NOT_GIVEN = NotGiven() NOT_GIVEN = NotGiven()
@ -219,7 +219,7 @@ async def get_nicknames():
try: try:
async with aiofiles.open(filename, "r", encoding="utf-8") as f: async with aiofiles.open(filename, "r", encoding="utf-8") as f:
nickname_json = json.loads(await f.read()) nickname_json = json.loads(await f.read())
except Exception: except (json.JSONDecodeError, FileNotFoundError):
nickname_json = {} nickname_json = {}
return nickname_json return nickname_json
@ -249,7 +249,7 @@ async def refresh_nickname_json():
store.get_plugin_data_file("nickname.json"), "r", encoding="utf-8" store.get_plugin_data_file("nickname.json"), "r", encoding="utf-8"
) as f: ) as f:
nickname_json = json.loads(await f.read()) nickname_json = json.loads(await f.read())
except Exception: except (json.JSONDecodeError, FileNotFoundError):
logger.error("刷新 nickname_json 表错误:无法载入 nickname.json 文件") logger.error("刷新 nickname_json 表错误:无法载入 nickname.json 文件")