blob: e37eeb5fcd5591fa24e72d1c81aef0bcf5694817 (
about) (
plain)
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
27
28
29
30
31
|
"""Abstraction for a prompt that can be formatted with inputs."""
from typing import Any
from pydantic import BaseModel
class Prompt(BaseModel):
"""A prompt that can be formatted with inputs."""
name: str
template: str
input_types: dict[str, str]
def format_prompt(self, inputs: dict[str, Any]) -> str:
self._validate_inputs(inputs)
return self.template.format(**inputs)
def _validate_inputs(self, inputs: dict[str, Any]) -> None:
for var, expected_type_name in self.input_types.items():
expected_type = self._convert_type(expected_type_name)
if var not in inputs:
raise ValueError(f"Missing input: {var}")
if not isinstance(inputs[var], expected_type):
raise TypeError(
f"Input '{var}' must be of type {expected_type.__name__}, got {type(inputs[var]).__name__} instead."
)
def _convert_type(self, type_name: str) -> type:
type_mapping = {"int": int, "str": str}
return type_mapping.get(type_name, str)
|
