mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-25 03:19:41 +00:00
feat(core): Support higher-order operators (#1984)
Co-authored-by: 谨欣 <echo.cmy@antgroup.com>
This commit is contained in:
@@ -254,6 +254,18 @@ class ChatPromptTemplate(BasePromptTemplate):
|
||||
values["input_variables"] = sorted(input_variables)
|
||||
return values
|
||||
|
||||
def get_placeholders(self) -> List[str]:
|
||||
"""Get all placeholders in the prompt template.
|
||||
|
||||
Returns:
|
||||
List[str]: The placeholders.
|
||||
"""
|
||||
placeholders = set()
|
||||
for message in self.messages:
|
||||
if isinstance(message, MessagesPlaceholder):
|
||||
placeholders.add(message.variable_name)
|
||||
return sorted(placeholders)
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class PromptTemplateIdentifier(ResourceIdentifier):
|
||||
|
||||
Reference in New Issue
Block a user