cecli-dev 0.95.5__py3-none-any.whl

cecli/coders/architect_coder.py

@@ -0,0 +1,63 @@
+import asyncio
+
+from ..commands import SwitchCoderSignal
+from .ask_coder import AskCoder
+from .base_coder import Coder
+
+
+class ArchitectCoder(AskCoder):
+    edit_format = "architect"
+    prompt_format = "architect"
+    auto_accept_architect = False
+
+    async def reply_completed(self):
+        content = self.partial_response_content
+
+        if not content or not content.strip():
+            return
+
+        tweak_responses = getattr(self.args, "tweak_responses", False)
+        confirmation = await self.io.confirm_ask("Edit the files?", allow_tweak=tweak_responses)
+
+        if not self.auto_accept_architect and not confirmation:
+            return
+
+        if confirmation == "tweak":
+            content = self.io.edit_in_editor(content)
+
+        await asyncio.sleep(0.1)
+
+        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["args"] = self.args
+        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 = await Coder.create(**new_kwargs)
+        editor_coder.cur_messages = []
+        editor_coder.done_messages = []
+
+        if self.verbose:
+            editor_coder.show_announcements()
+
+        try:
+            await editor_coder.generate(user_message=content, preproc=False)
+            self.move_back_cur_messages("I made those changes to the files.")
+            self.total_cost = editor_coder.total_cost
+            self.coder_commit_hashes = editor_coder.coder_commit_hashes
+        except Exception as e:
+            self.io.tool_error(e)
+
+        raise SwitchCoderSignal(main_model=self.main_model, edit_format="architect")

cecli/coders/ask_coder.py

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