QChatGPT/pkg/provider/sysprompt/sysprompt.py

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