feat: prompt 加载器的扩展性

2026-08-16 15:30:59 +00:00 · 2024-03-12 16:22:07 +00:00
parent 8c6ce1f030
commit b9fa11c0c3
4 changed files with 23 additions and 5 deletions
@@ -20,12 +20,14 @@ class PromptManager:

    async def initialize(self):

-        loader_map = {
-            "normal": single.SingleSystemPromptLoader,
-            "full_scenario": scenario.ScenarioPromptLoader
-        }
+        mode_name = self.ap.provider_cfg.data['prompt-mode']

-        loader_cls = loader_map[self.ap.provider_cfg.data['prompt-mode']]
+        for loader_cls in loader.preregistered_loaders:
+            if loader_cls.name == mode_name:
+                loader_cls = loader_cls
+                break
+        else:
+            raise ValueError(f'未知的 Prompt 加载器: {mode_name}')

        self.loader_inst: loader.PromptLoader = loader_cls(self.ap)