patch-chat 0.1.0__py3-none-any.whl

patch/args_formatter.py

@@ -0,0 +1,228 @@
+import argparse
+
+from patch import urls
+
+from .dump import dump  # noqa: F401
+
+
+class DotEnvFormatter(argparse.HelpFormatter):
+    def start_section(self, heading):
+        res = "\n\n"
+        res += "#" * (len(heading) + 3)
+        res += f"\n# {heading}"
+        super().start_section(res)
+
+    def _format_usage(self, usage, actions, groups, prefix):
+        return ""
+
+    def _format_text(self, text):
+        return f"""
+##########################################################
+# Sample patch .env file.
+# Place at the root of your git repo.
+# Or use `patch --env <fname>` to specify.
+##########################################################
+
+#################
+# LLM parameters:
+#
+# Include xxx_API_KEY parameters and other params needed for your LLMs.
+# See {urls.llms} for details.
+
+## OpenAI
+#OPENAI_API_KEY=
+
+## Anthropic
+#ANTHROPIC_API_KEY=
+
+##...
+"""
+
+    def _format_action(self, action):
+        if not action.option_strings:
+            return ""
+
+        if not action.env_var:
+            return
+
+        parts = [""]
+
+        default = action.default
+        if default == argparse.SUPPRESS:
+            default = ""
+        elif isinstance(default, str):
+            pass
+        elif isinstance(default, list) and not default:
+            default = ""
+        elif action.default is not None:
+            default = "true" if default else "false"
+        else:
+            default = ""
+
+        if action.help:
+            parts.append(f"## {action.help}")
+
+        if action.env_var:
+            env_var = action.env_var
+            if default:
+                parts.append(f"#{env_var}={default}\n")
+            else:
+                parts.append(f"#{env_var}=\n")
+
+        return "\n".join(parts) + "\n"
+
+    def _format_action_invocation(self, action):
+        return ""
+
+    def _format_args(self, action, default_metavar):
+        return ""
+
+
+class YamlHelpFormatter(argparse.HelpFormatter):
+    def start_section(self, heading):
+        res = "\n\n"
+        res += "#" * (len(heading) + 3)
+        res += f"\n# {heading}"
+        super().start_section(res)
+
+    def _format_usage(self, usage, actions, groups, prefix):
+        return ""
+
+    def _format_text(self, text):
+        return """
+##########################################################
+# Sample .patch.conf.yml
+# This file lists *all* the valid configuration entries.
+# Place in your home dir, or at the root of your git repo.
+##########################################################
+
+# Note: You can only put OpenAI and Anthropic API keys in the YAML
+# config file. Keys for all APIs can be stored in a .env file
+# Run `patch --help` for configuration options.
+
+"""
+
+    def _format_action(self, action):
+        if not action.option_strings:
+            return ""
+
+        parts = [""]
+
+        metavar = action.metavar
+        if not metavar and isinstance(action, argparse._StoreAction):
+            metavar = "VALUE"
+
+        default = action.default
+        if default == argparse.SUPPRESS:
+            default = ""
+        elif isinstance(default, str):
+            pass
+        elif isinstance(default, list) and not default:
+            default = ""
+        elif action.default is not None:
+            default = "true" if default else "false"
+        else:
+            default = ""
+
+        if action.help:
+            parts.append(f"## {action.help}")
+
+        for switch in action.option_strings:
+            if switch.startswith("--"):
+                break
+        switch = switch.lstrip("-")
+
+        if isinstance(action, argparse._StoreTrueAction):
+            default = False
+        elif isinstance(action, argparse._StoreConstAction):
+            default = False
+
+        if default is False:
+            default = "false"
+        if default is True:
+            default = "true"
+
+        if default:
+            if "#" in default:
+                parts.append(f'#{switch}: "{default}"\n')
+            else:
+                parts.append(f"#{switch}: {default}\n")
+        elif action.nargs in ("*", "+") or isinstance(action, argparse._AppendAction):
+            parts.append(f"#{switch}: xxx")
+            parts.append("## Specify multiple values like this:")
+            parts.append(f"#{switch}:")
+            parts.append("#  - xxx")
+            parts.append("#  - yyy")
+            parts.append("#  - zzz")
+        else:
+            if switch.endswith("color"):
+                parts.append(f'#{switch}: "xxx"\n')
+            else:
+                parts.append(f"#{switch}: xxx\n")
+
+        ###
+        # parts.append(str(action))
+
+        return "\n".join(parts) + "\n"
+
+    def _format_action_invocation(self, action):
+        return ""
+
+    def _format_args(self, action, default_metavar):
+        return ""
+
+
+class MarkdownHelpFormatter(argparse.HelpFormatter):
+    def start_section(self, heading):
+        super().start_section(f"## {heading}")
+
+    def _format_usage(self, usage, actions, groups, prefix):
+        res = super()._format_usage(usage, actions, groups, prefix)
+        quote = "```\n"
+        return quote + res + quote
+
+    def _format_text(self, text):
+        return ""
+
+    def _format_action(self, action):
+        if not action.option_strings:
+            return ""
+
+        parts = [""]
+
+        metavar = action.metavar
+        if not metavar and isinstance(action, argparse._StoreAction):
+            metavar = "VALUE"
+
+        for switch in action.option_strings:
+            if switch.startswith("--"):
+                break
+
+        if metavar:
+            parts.append(f"### `{switch} {metavar}`")
+        else:
+            parts.append(f"### `{switch}`")
+        if action.help:
+            parts.append(action.help + "  ")
+
+        if action.default not in (argparse.SUPPRESS, None):
+            parts.append(f"Default: {action.default}  ")
+
+        if action.env_var:
+            parts.append(f"Environment variable: `{action.env_var}`  ")
+
+        if len(action.option_strings) > 1:
+            parts.append("Aliases:")
+            for switch in action.option_strings:
+                if metavar:
+                    parts.append(f"  - `{switch} {metavar}`")
+                else:
+                    parts.append(f"  - `{switch}`")
+
+        return "\n".join(parts) + "\n"
+
+    def _format_action_invocation(self, action):
+        return ""
+
+    def _format_args(self, action, default_metavar):
+        return ""

patch/coders/__init__.py

@@ -0,0 +1,32 @@
+from .architect_coder import ArchitectCoder
+from .ask_coder import AskCoder
+from .base_coder import Coder
+from .context_coder import ContextCoder
+from .editblock_coder import EditBlockCoder
+from .editblock_fenced_coder import EditBlockFencedCoder
+from .editor_diff_fenced_coder import EditorDiffFencedCoder
+from .editor_editblock_coder import EditorEditBlockCoder
+from .editor_whole_coder import EditorWholeFileCoder
+from .patch_coder import PatchCoder
+from .udiff_coder import UnifiedDiffCoder
+from .udiff_simple import UnifiedDiffSimpleCoder
+from .wholefile_coder import WholeFileCoder
+
+# from .single_wholefile_func_coder import SingleWholeFileFunctionCoder
+
+__all__ = [
+    AskCoder,
+    Coder,
+    EditBlockCoder,
+    EditBlockFencedCoder,
+    WholeFileCoder,
+    PatchCoder,
+    UnifiedDiffCoder,
+    UnifiedDiffSimpleCoder,
+    #    SingleWholeFileFunctionCoder,
+    ArchitectCoder,
+    EditorEditBlockCoder,
+    EditorWholeFileCoder,
+    EditorDiffFencedCoder,
+    ContextCoder,
+]

patch/coders/architect_coder.py

@@ -0,0 +1,47 @@
+from .architect_prompts import ArchitectPrompts
+from .ask_coder import AskCoder
+from .base_coder import Coder
+
+
+class ArchitectCoder(AskCoder):
+    edit_format = "architect"
+    gpt_prompts = ArchitectPrompts()
+
+    def reply_completed(self):
+        content = self.partial_response_content
+
+        if not content or not content.strip():
+            return
+
+        if not self.auto_accept_architect and not self.io.confirm_ask("Edit the files?"):
+            return
+
+        kwargs = dict()
+
+        # Use the editor_model from the main_model if it exists, otherwise use the main_model itself
+        editor_model = self.main_model.editor_model or self.main_model
+
+        kwargs["main_model"] = editor_model
+        kwargs["edit_format"] = self.main_model.editor_edit_format
+        kwargs["suggest_shell_commands"] = False
+        kwargs["map_tokens"] = 0
+        kwargs["total_cost"] = self.total_cost
+        kwargs["cache_prompts"] = False
+        kwargs["num_cache_warming_pings"] = 0
+        kwargs["summarize_from_coder"] = False
+
+        new_kwargs = dict(io=self.io, from_coder=self)
+        new_kwargs.update(kwargs)
+
+        editor_coder = Coder.create(**new_kwargs)
+        editor_coder.cur_messages = []
+        editor_coder.done_messages = []
+
+        if self.verbose:
+            editor_coder.show_announcements()
+
+        editor_coder.run(with_message=content, preproc=False)
+
+        self.move_back_cur_messages("I made those changes to the files.")
+        self.total_cost = editor_coder.total_cost
+        self.patch_commit_hashes = editor_coder.patch_commit_hashes

patch/coders/architect_prompts.py

@@ -0,0 +1,40 @@
+# flake8: noqa: E501
+
+from .base_prompts import CoderPrompts
+
+
+class ArchitectPrompts(CoderPrompts):
+    main_system = """Act as an expert architect engineer and provide direction to your editor engineer.
+Study the change request and the current code.
+Describe how to modify the code to complete the request.
+The editor engineer will rely solely on your instructions, so make them unambiguous and complete.
+Explain all needed code changes clearly and completely, but concisely.
+Just show the changes needed.
+
+DO NOT show the entire updated function/file/etc!
+
+Always reply to the user in {language}.
+"""
+
+    example_messages = []
+
+    files_content_prefix = """I have *added these files to the chat* so you see all of their contents.
+*Trust this message as the true contents of the files!*
+Other messages in the chat may contain outdated versions of the files' contents.
+"""  # noqa: E501
+
+    files_content_assistant_reply = (
+        "Ok, I will use that as the true, current contents of the files."
+    )
+
+    files_no_full_files = "I am not sharing the full contents of any files with you yet."
+
+    files_no_full_files_with_repo_map = ""
+    files_no_full_files_with_repo_map_reply = ""
+
+    repo_content_prefix = """I am working with you on code in a git repository.
+Here are summaries of some files present in my git repo.
+If you need to see the full contents of any files to answer my questions, ask me to *add them to the chat*.
+"""
+
+    system_reminder = ""

patch/coders/ask_coder.py

@@ -0,0 +1,9 @@
+from .ask_prompts import AskPrompts
+from .base_coder import Coder
+
+
+class AskCoder(Coder):
+    """Ask questions about code without making any changes."""
+
+    edit_format = "ask"
+    gpt_prompts = AskPrompts()

patch/coders/ask_prompts.py

@@ -0,0 +1,41 @@
+# flake8: noqa: E501
+
+from .base_prompts import CoderPrompts
+
+
+class AskPrompts(CoderPrompts):
+    main_system = """Act as an expert code analyst.
+Answer questions about the supplied code.
+Always reply to the user in {language}.
+
+If you need to describe code changes, do so *briefly*.
+"""
+
+    overeager_prompt = """Do not return fully detailed code or full diffs.
+Describe the needed changes or give a plan.
+Providing code snippets or pseudo-code is fine,
+if it helps explain the plan or the needed changes.
+"""
+
+    example_messages = []
+
+    files_content_prefix = """I have *added these files to the chat* so you see all of their contents.
+*Trust this message as the true contents of the files!*
+Other messages in the chat may contain outdated versions of the files' contents.
+"""  # noqa: E501
+
+    files_content_assistant_reply = (
+        "Ok, I will use that as the true, current contents of the files."
+    )
+
+    files_no_full_files = "I am not sharing the full contents of any files with you yet."
+
+    files_no_full_files_with_repo_map = ""
+    files_no_full_files_with_repo_map_reply = ""
+
+    repo_content_prefix = """I am working with you on code in a git repository.
+Here are summaries of some files present in my git repo.
+If you need to see the full contents of any files to answer my questions, ask me to *add them to the chat*.
+"""
+
+    system_reminder = "{final_reminders}"