File size: 689 Bytes
246d201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import os
from utils import load_file
PROMPT_DIR = os.path.dirname(__file__)
TEMPLATE_WITH_TOOL = load_file(os.path.join(PROMPT_DIR, 'template_with_tool.txt'))
class PromptTemplate:
"""A prompt template."""
def __init__(self, template: str):
self.template: str = template
def __call__(self, **kwargs) -> str:
return self.template.format(**kwargs)
class ToolPromptTemplate(PromptTemplate):
def __init__(self, use_tool: bool):
if use_tool:
template = TEMPLATE_WITH_TOOL
else:
raise NotImplementedError('Evaluation without tool is not supported yet.')
super().__init__(template)
|