extensions/python/system_prompt/_13_secrets_prompt.py

from typing import Any

from helpers.extension import Extension, extensible
from agent import Agent, LoopData


class SecretsPrompt(Extension):

    async def execute(
        self,
        system_prompt: list[str] = [],
        loop_data: LoopData = LoopData(),
        **kwargs: Any,
    ):
        if not self.agent:
            return
        prompt = await build_prompt(self.agent)
        if prompt:
            system_prompt.append(prompt)


@extensible
async def build_prompt(agent: Agent) -> str:
    try:
        from helpers.secrets import get_secrets_manager
        from helpers.settings import get_settings_for_prompt

        secrets_manager = get_secrets_manager(agent.context)
        secrets = secrets_manager.get_secrets_for_prompt()
        variables = get_settings_for_prompt()["variables"]
        return agent.read_prompt(
            "agent.system.secrets.md", secrets=secrets, vars=variables
        )
    except Exception:
        return ""