forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""Utilities for formatting strings."""
|
|
from string import Formatter
|
|
from typing import Any, List, Mapping, Sequence, Union
|
|
|
|
|
|
class StrictFormatter(Formatter):
|
|
"""A subclass of formatter that checks for extra keys."""
|
|
|
|
def check_unused_args(
|
|
self,
|
|
used_args: Sequence[Union[int, str]],
|
|
args: Sequence,
|
|
kwargs: Mapping[str, Any],
|
|
) -> None:
|
|
"""Check to see if extra parameters are passed."""
|
|
extra = set(kwargs).difference(used_args)
|
|
if extra:
|
|
raise KeyError(extra)
|
|
|
|
def vformat(
|
|
self, format_string: str, args: Sequence, kwargs: Mapping[str, Any]
|
|
) -> str:
|
|
"""Check that no arguments are provided."""
|
|
if len(args) > 0:
|
|
raise ValueError(
|
|
"No arguments should be provided, "
|
|
"everything should be passed as keyword arguments."
|
|
)
|
|
return super().vformat(format_string, args, kwargs)
|
|
|
|
def validate_input_variables(
|
|
self, format_string: str, input_variables: List[str]
|
|
) -> None:
|
|
dummy_inputs = {input_variable: "foo" for input_variable in input_variables}
|
|
super().format(format_string, **dummy_inputs)
|
|
|
|
|
|
formatter = StrictFormatter()
|