Class Prompt

finagent/prompt/custom.py:12–121

@PROMPT.register_module(force=True)
class Prompt():
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def _replace_keys(self, text: str, params: Dict):
        keys = params.keys()
        for key in keys:
            if key in text:
                if isinstance(params[key], list) or isinstance(params[key], dict):
                    text = text.replace(f'<${key}$>', json.dumps(params[key], indent=4))
                else:
                    text = text.replace(f'<${key}$>', str(params[key]))
        return text

    def convert_state_info_to_parmas(self, *args, **kwargs):
        raise NotImplementedError

    def to_message(self, *args,
                   params: Dict = None,
                   template: Any = None,
                   **kwargs):

        assert params is not None, "params is None"
        assert template is not None, "template is None"

        html = generate_prompt_html(params, template)

        system_message = None
        system_div_tag = html.find("div", class_="message", role="system")
        if system_div_tag is not None:
            p_tag = system_div_tag.find("p")
            content = p_tag.text

            content = content_replace(content)

            message = {
                "role": "system",
                "content": [
                    {
                        "type": "text",
                        "text": content
                    }
                ]
            }
            system_message = message

        assert system_message is not None, "system_message is None"

        user_messages = []
        user_div_tags = html.find_all("div", class_="message", role="user")

        for user_div_tag in user_div_tags:
            tags = user_div_tag.find_all(["p", "img"])

            message = {
                "role": "user",
                "content": []
            }