mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from ...core import app
|
|
from . import loader
|
|
from .loaders import single, scenario
|
|
|
|
|
|
class PromptManager:
|
|
"""Prompt管理器
|
|
"""
|
|
|
|
ap: app.Application
|
|
|
|
loader_inst: loader.PromptLoader
|
|
|
|
default_prompt: str = 'default'
|
|
|
|
def __init__(self, ap: app.Application):
|
|
self.ap = ap
|
|
|
|
async def initialize(self):
|
|
|
|
mode_name = self.ap.provider_cfg.data['prompt-mode']
|
|
|
|
loader_class = None
|
|
|
|
for loader_cls in loader.preregistered_loaders:
|
|
if loader_cls.name == mode_name:
|
|
loader_class = loader_cls
|
|
break
|
|
else:
|
|
raise ValueError(f'未知的 Prompt 加载器: {mode_name}')
|
|
|
|
self.loader_inst: loader.PromptLoader = loader_class(self.ap)
|
|
|
|
await self.loader_inst.initialize()
|
|
await self.loader_inst.load()
|
|
|
|
def get_all_prompts(self) -> list[loader.entities.Prompt]:
|
|
"""获取所有Prompt
|
|
"""
|
|
return self.loader_inst.get_prompts()
|
|
|
|
async def get_prompt(self, name: str) -> loader.entities.Prompt:
|
|
"""获取Prompt
|
|
"""
|
|
for prompt in self.get_all_prompts():
|
|
if prompt.name == name:
|
|
return prompt
|
|
|
|
async def get_prompt_by_prefix(self, prefix: str) -> loader.entities.Prompt:
|
|
"""通过前缀获取Prompt
|
|
"""
|
|
for prompt in self.get_all_prompts():
|
|
if prompt.name.startswith(prefix):
|
|
return prompt
|