langchain_prompty.renderers 的源代码

from langchain_core.utils import mustache
from pydantic import BaseModel

from .core import Invoker, Prompty, SimpleModel



[docs]
class MustacheRenderer(Invoker):
    """Render a mustache template."""


[docs]
    def __init__(self, prompty: Prompty) -> None:
        self.prompty = prompty



[docs]
    def invoke(self, data: BaseModel) -> BaseModel:
        assert isinstance(data, SimpleModel)
        generated = mustache.render(self.prompty.content, data.item)
        return SimpleModel[str](item=generated)