mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-22 09:28:42 +00:00
27 lines
925 B
Python
27 lines
925 B
Python
from pilot.configs.config import Config
|
|
from pilot.prompts.generator import PromptGenerator
|
|
from typing import Any, Optional, Type
|
|
from pilot.prompts.prompt import build_default_prompt_generator
|
|
|
|
|
|
class CommandsLoad:
|
|
"""
|
|
Load Plugins Commands Info , help build system prompt!
|
|
"""
|
|
|
|
def __init__(self)->None:
|
|
self.command_registry = None
|
|
|
|
|
|
def getCommandInfos(self, prompt_generator: Optional[PromptGenerator] = None)-> str:
|
|
cfg = Config()
|
|
if prompt_generator is None:
|
|
prompt_generator = build_default_prompt_generator()
|
|
for plugin in cfg.plugins:
|
|
if not plugin.can_handle_post_prompt():
|
|
continue
|
|
prompt_generator = plugin.post_prompt(prompt_generator)
|
|
self.prompt_generator = prompt_generator
|
|
command_infos = ""
|
|
command_infos += f"\n\n{prompt_generator.commands()}"
|
|
return command_infos |