From cac8a0a4147707095d5d72233dc51e85e6058ace Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Sat, 2 Mar 2024 16:39:29 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BC=98=E5=8C=96=E5=AF=BC=E5=85=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/core/boot.py | 20 -------------------- pkg/core/stage.py | 2 +- 2 files changed, 1 insertion(+), 21 deletions(-) diff --git a/pkg/core/boot.py b/pkg/core/boot.py index 6a395c7..2e5a2b6 100644 --- a/pkg/core/boot.py +++ b/pkg/core/boot.py @@ -1,27 +1,7 @@ from __future__ import print_function -import sys - -from .bootutils import files -from .bootutils import log -from .bootutils import config - from . import app -from ..pipeline import pool -from ..pipeline import controller -from ..pipeline import stagemgr from ..audit import identifier -from ..provider.session import sessionmgr as llm_session_mgr -from ..provider.requester import modelmgr as llm_model_mgr -from ..provider.sysprompt import sysprompt as llm_prompt_mgr -from ..provider.tools import toolmgr as llm_tool_mgr -from ..platform import manager as im_mgr -from ..command import cmdmgr -from ..plugin import manager as plugin_mgr -from ..audit.center import v2 as center_v2 -from ..utils import version, proxy, announce - -from .stages import build_app, load_config, setup_logger from . import stage diff --git a/pkg/core/stage.py b/pkg/core/stage.py index cb2e486..cb8e90e 100644 --- a/pkg/core/stage.py +++ b/pkg/core/stage.py @@ -6,7 +6,7 @@ import typing from . import app -preregistered_stages: list[typing.Type[BootingStage]] = {} +preregistered_stages: dict[str, typing.Type[BootingStage]] = {} def stage_class( name: str