aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/azure/ai/inference/prompts/_renderers.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/inference/prompts/_renderers.py')
-rw-r--r--.venv/lib/python3.12/site-packages/azure/ai/inference/prompts/_renderers.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/inference/prompts/_renderers.py b/.venv/lib/python3.12/site-packages/azure/ai/inference/prompts/_renderers.py
new file mode 100644
index 00000000..0d682a7f
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/inference/prompts/_renderers.py
@@ -0,0 +1,30 @@
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------
+# mypy: disable-error-code="union-attr,assignment,arg-type"
+from pathlib import Path
+from ._core import Prompty
+from ._invoker import Invoker, InvokerFactory
+from ._mustache import render
+
+
+@InvokerFactory.register_renderer("mustache")
+class MustacheRenderer(Invoker):
+ """Render a mustache template."""
+
+ def __init__(self, prompty: Prompty) -> None:
+ super().__init__(prompty)
+ self.templates = {}
+ cur_prompt = self.prompty
+ while cur_prompt:
+ self.templates[Path(cur_prompt.file).name] = cur_prompt.content
+ cur_prompt = cur_prompt.basePrompty
+ self.name = Path(self.prompty.file).name
+
+ def invoke(self, data: str) -> str:
+ generated = render(self.prompty.content, data) # type: ignore
+ return generated
+
+ async def invoke_async(self, data: str) -> str:
+ return self.invoke(data)