Moved PromptTemplate to text.py.

This commit is contained in:
Timothy Eakin
2023-06-19 15:10:20 -04:00
parent 906fec459e
commit 513a5118cd
-25
View File
@@ -315,31 +315,6 @@ class PreviewImageTest(SaveImage):
class PromptTemplate:
"""replaces the text in the given text string with a given other text at some key positions"""
@classmethod
def INPUT_TYPES(s):
return {"required":
{
"text": ("STRING", {"multiline": True}),
"replacement": ("STRING", {"multiline": True}),
"key1": ("STRING", {"multiline": False}),
}
}
CATEGORY = "text"
RETURN_TYPES = ("STRING",)
FUNCTION = "prompt_template_handler"
def prompt_template_handler(self, text: str, replacement, key1):
"""
>>> PromptTemplate().prompt_template_handler("hello world", "universe", "world")
'hello universe'
"""
return (text.replace(key1, replacement),)
class LoadImage:
@classmethod
def INPUT_TYPES(s):