mirror of
https://github.com/tiennm99/serena.git
synced 2026-07-14 20:22:36 +00:00
Michael Panchenko
26 lines
986 B
Python
26 lines
986 B
Python
# ruff: noqa
|
|
# black: skip
|
|
# mypy: ignore-errors
|
|
|
|
from .multilang_prompt import MultiLangContainer, MultiLangPromptTemplateCollection, PromptList
|
|
|
|
|
|
class PromptFactory:
|
|
# NOTE: This class is auto-generated by gen_prompt_factory.py
|
|
|
|
def __init__(self, lang_shortcode: str = "en", fallback_mode=MultiLangContainer.FallbackMode.EXCEPTION):
|
|
self.lang_shortcode = lang_shortcode
|
|
self.collection = MultiLangPromptTemplateCollection()
|
|
self.fallback_mode = fallback_mode
|
|
|
|
def _format_prompt(self, prompt_name: str, kwargs) -> str:
|
|
del kwargs["self"]
|
|
mpt = self.collection.get_multilang_prompt_template(prompt_name)
|
|
return mpt.get_item(self.lang_shortcode, self.fallback_mode).instantiate(**kwargs)
|
|
|
|
def _get_list(self, prompt_name: str) -> PromptList:
|
|
mpl = self.collection.get_multilang_prompt_list(prompt_name)
|
|
return mpl.get_item(self.lang_shortcode, self.fallback_mode)
|
|
|
|
# methods
|