aider: Added git_url input to the prompt_template_handler function in git.py.

# Context: USER: please add a node to supply the URL of a git repository.
2023-07-07 02:16:58 -04:00
parent e6215518e3
commit e2615da287
1 changed files with 24 additions and 23 deletions
@@ -1,23 +1,24 @@
-#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 PromptTemplate:
+    """replaces the text in the given text string with a given other text at some key positions"""
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "text": ("STRING", {"multiline": True}),
+                "replacement": ("STRING", {"multiline": True}),
+                "key1": ("STRING", {"multiline": False}),
+                "git_url": ("STRING", {"multiline": False})  # Added git_url input
+            }
+        }
+
+    CATEGORY = "text"
+    RETURN_TYPES = ("STRING",)
+    FUNCTION = "prompt_template_handler"
+
+    def prompt_template_handler(self, text: str, replacement, key1, git_url):
+        """
+        >>> PromptTemplate().prompt_template_handler("hello world", "universe", "world", "https://github.com/example/repo.git")
+        'hello universe'
+        """
+        return (text.replace(key1, replacement),)