auto-coder 0.1.201__py3-none-any.whl → 0.1.203__py3-none-any.whl

autocoder/chat_auto_coder.py

@@ -1,3 +1,6 @@
+from prompt_toolkit.formatted_text import HTML
+from prompt_toolkit.shortcuts import radiolist_dialog
+from prompt_toolkit import prompt
 import os
 import yaml
 import json
@@ -6,6 +9,7 @@ import io
 import uuid
 import glob
 import time
+import hashlib
 from contextlib import contextmanager
 from typing import List, Dict, Any, Optional
 from prompt_toolkit import PromptSession
@@ -41,6 +45,7 @@ from rich.live import Live
 from byzerllm.utils.nontext import Image
 import re
 import git
+from autocoder.common import git_utils
 from autocoder.utils.request_queue import (
     request_queue,
     RequestValue,
@@ -65,7 +70,8 @@ def parse_arguments():
     import argparse
 
     parser = argparse.ArgumentParser(description="Chat Auto Coder")
-    parser.add_argument("--debug", action="store_true", help="Enable debug mode")
+    parser.add_argument("--debug", action="store_true",
+                        help="Enable debug mode")
     parser.add_argument(
         "--quick",
         action="store_true",
@@ -81,9 +87,6 @@ if platform.system() == "Windows":
 
     init()
 
-from prompt_toolkit import prompt
-from prompt_toolkit.shortcuts import radiolist_dialog
-from prompt_toolkit.formatted_text import HTML
 
 memory = {
     "conversation": [],
@@ -107,6 +110,7 @@ commands = [
     "/coding",
     "/chat",
     "/ask",
+    "/commit",
     "/revert",
     "/index/query",
     "/index/build",
@@ -149,7 +153,10 @@ def show_help():
     print(
         f"  \033[94m/summon\033[0m \033[93m<query>\033[0m - \033[92m{get_message('summon_desc')}\033[0m"
     )
-    print(f"  \033[94m/revert\033[0m - \033[92m{get_message('revert_desc')}\033[0m")
+    print(
+        f"  \033[94m/revert\033[0m - \033[92m{get_message('revert_desc')}\033[0m")
+    print(
+        f"  \033[94m/commit\033[0m - \033[92m{get_message('commit_desc')}\033[0m")
     print(
         f"  \033[94m/conf\033[0m \033[93m<key>:<value>\033[0m  - \033[92m{get_message('conf_desc')}\033[0m"
     )
@@ -162,7 +169,8 @@ def show_help():
     print(
         f"  \033[94m/list_files\033[0m - \033[92m{get_message('list_files_desc')}\033[0m"
     )
-    print(f"  \033[94m/help\033[0m - \033[92m{get_message('help_desc')}\033[0m")
+    print(
+        f"  \033[94m/help\033[0m - \033[92m{get_message('help_desc')}\033[0m")
     print(
         f"  \033[94m/exclude_dirs\033[0m \033[93m<dir1>,<dir2> ...\033[0m - \033[92m{get_message('exclude_dirs_desc')}\033[0m"
     )
@@ -172,9 +180,11 @@ def show_help():
     print(
         f"  \033[94m/voice_input\033[0m - \033[92m{get_message('voice_input_desc')}\033[0m"
     )
-    print(f"  \033[94m/mode\033[0m - \033[92m{get_message('mode_desc')}\033[0m")
+    print(
+        f"  \033[94m/mode\033[0m - \033[92m{get_message('mode_desc')}\033[0m")
     print(f"  \033[94m/lib\033[0m - \033[92m{get_message('lib_desc')}\033[0m")
-    print(f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
+    print(
+        f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
     print()
 
 
@@ -199,10 +209,12 @@ def configure_project_type():
         print_formatted_text(HTML(f"<info>{escape(text)}</info>"), style=style)
 
     def print_warning(text):
-        print_formatted_text(HTML(f"<warning>{escape(text)}</warning>"), style=style)
+        print_formatted_text(
+            HTML(f"<warning>{escape(text)}</warning>"), style=style)
 
     def print_header(text):
-        print_formatted_text(HTML(f"<header>{escape(text)}</header>"), style=style)
+        print_formatted_text(
+            HTML(f"<header>{escape(text)}</header>"), style=style)
 
     print_header(f"\n=== {get_message('project_type_config')} ===\n")
     print_info(get_message("project_type_supports"))
@@ -248,7 +260,8 @@ def initialize_system():
         if not os.path.exists(".auto-coder"):
             first_time = True
             print_status(get_message("not_initialized"), "warning")
-            init_choice = input(f"  {get_message('init_prompt')}").strip().lower()
+            init_choice = input(
+                f"  {get_message('init_prompt')}").strip().lower()
             if init_choice == "y":
                 try:
                     subprocess.run(
@@ -265,7 +278,8 @@ def initialize_system():
 
         if not os.path.exists(base_persist_dir):
             os.makedirs(base_persist_dir, exist_ok=True)
-            print_status(get_message("created_dir").format(base_persist_dir), "success")
+            print_status(get_message("created_dir").format(
+                base_persist_dir), "success")
 
         if first_time:
             configure_project_type()
@@ -275,7 +289,8 @@ def initialize_system():
     init_project()
     # Check if Ray is running
     print_status(get_message("checking_ray"), "")
-    ray_status = subprocess.run(["ray", "status"], capture_output=True, text=True)
+    ray_status = subprocess.run(
+        ["ray", "status"], capture_output=True, text=True)
     if ray_status.returncode != 0:
         print_status(get_message("ray_not_running"), "warning")
         try:
@@ -395,7 +410,7 @@ def get_all_file_names_in_project() -> List[str]:
 
     file_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root,followlinks=True):
+    for root, dirs, files in os.walk(project_root, followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         file_names.extend(files)
     return file_names
@@ -405,7 +420,7 @@ def get_all_file_in_project() -> List[str]:
 
     file_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root,followlinks=True):
+    for root, dirs, files in os.walk(project_root, followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         for file in files:
             file_names.append(os.path.join(root, file))
@@ -415,17 +430,18 @@ def get_all_file_in_project() -> List[str]:
 def get_all_file_in_project_with_dot() -> List[str]:
     file_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root,followlinks=True):
+    for root, dirs, files in os.walk(project_root, followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         for file in files:
-            file_names.append(os.path.join(root, file).replace(project_root, "."))
+            file_names.append(os.path.join(
+                root, file).replace(project_root, "."))
     return file_names
 
 
 def get_all_dir_names_in_project() -> List[str]:
     dir_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root,followlinks=True):
+    for root, dirs, files in os.walk(project_root, followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         for dir in dirs:
             dir_names.append(dir)
@@ -448,8 +464,8 @@ def find_files_in_project(patterns: List[str]) -> List[str]:
                         matched_files.append(abs_path)
         else:
             is_added = False
-            ## add files belongs to project
-            for root, dirs, files in os.walk(project_root,followlinks=True):
+            # add files belongs to project
+            for root, dirs, files in os.walk(project_root, followlinks=True):
                 dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
                 if pattern in files:
                     matched_files.append(os.path.join(root, pattern))
@@ -460,7 +476,7 @@ def find_files_in_project(patterns: List[str]) -> List[str]:
                         if _pattern in os.path.join(root, file):
                             matched_files.append(os.path.join(root, file))
                             is_added = True
-            ## add files not belongs to project
+            # add files not belongs to project
             if not is_added:
                 matched_files.append(pattern)
 
@@ -546,7 +562,10 @@ def show_help():
     print(
         f"  \033[94m/summon\033[0m \033[93m<query>\033[0m - \033[92m{get_message('summon_desc')}\033[0m"
     )
-    print(f"  \033[94m/revert\033[0m - \033[92m{get_message('revert_desc')}\033[0m")
+    print(
+        f"  \033[94m/revert\033[0m - \033[92m{get_message('revert_desc')}\033[0m")
+    print(
+        f"  \033[94m/commit\033[0m - \033[92m{get_message('commit_desc')}\033[0m")
     print(
         f"  \033[94m/conf\033[0m \033[93m<key>:<value>\033[0m  - \033[92m{get_message('conf_desc')}\033[0m"
     )
@@ -559,7 +578,8 @@ def show_help():
     print(
         f"  \033[94m/list_files\033[0m - \033[92m{get_message('list_files_desc')}\033[0m"
     )
-    print(f"  \033[94m/help\033[0m - \033[92m{get_message('help_desc')}\033[0m")
+    print(
+        f"  \033[94m/help\033[0m - \033[92m{get_message('help_desc')}\033[0m")
     print(
         f"  \033[94m/exclude_dirs\033[0m \033[93m<dir1>,<dir2> ...\033[0m - \033[92m{get_message('exclude_dirs_desc')}\033[0m"
     )
@@ -569,9 +589,11 @@ def show_help():
     print(
         f"  \033[94m/voice_input\033[0m - \033[92m{get_message('voice_input_desc')}\033[0m"
     )
-    print(f"  \033[94m/mode\033[0m - \033[92m{get_message('mode_desc')}\033[0m")
+    print(
+        f"  \033[94m/mode\033[0m - \033[92m{get_message('mode_desc')}\033[0m")
     print(f"  \033[94m/lib\033[0m - \033[92m{get_message('lib_desc')}\033[0m")
-    print(f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
+    print(
+        f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
     print()
 
 
@@ -635,13 +657,13 @@ class CommandCompleter(Completer):
 
         if len(words) > 0:
             if words[0] == "/mode":
-                left_word = text[len("/mode") :]
+                left_word = text[len("/mode"):]
                 for mode in ["normal", "auto_detect", "voice_input"]:
                     if mode.startswith(left_word.strip()):
                         yield Completion(mode, start_position=-len(left_word.strip()))
 
             if words[0] == "/add_files":
-                new_text = text[len("/add_files") :]
+                new_text = text[len("/add_files"):]
                 parser = CommandTextParser(new_text, words[0])
                 parser.add_files()
                 current_word = parser.current_word()
@@ -672,18 +694,21 @@ class CommandCompleter(Completer):
                         for file_name in self.all_files_with_dot:
                             if file_name.startswith(current_word):
                                 yield Completion(
-                                    file_name, start_position=-len(current_word)
+                                    file_name, start_position=-
+                                    len(current_word)
                                 )
                     else:
                         for file_name in self.all_file_names:
                             if file_name.startswith(current_word):
                                 yield Completion(
-                                    file_name, start_position=-len(current_word)
+                                    file_name, start_position=-
+                                    len(current_word)
                                 )
                         for file_name in self.all_files:
                             if current_word and current_word in file_name:
                                 yield Completion(
-                                    file_name, start_position=-len(current_word)
+                                    file_name, start_position=-
+                                    len(current_word)
                                 )
             elif words[0] in ["/chat", "/coding"]:
                 image_extensions = (
@@ -713,7 +738,7 @@ class CommandCompleter(Completer):
                     ".psd",
                     ".xcf",
                 )
-                new_text = text[len(words[0]) :]
+                new_text = text[len(words[0]):]
                 parser = CommandTextParser(new_text, words[0])
 
                 parser.coding()
@@ -740,7 +765,8 @@ class CommandCompleter(Completer):
                                 if len(path_parts) > 3
                                 else file_name
                             )
-                            relative_path = os.path.relpath(file_name, project_root)
+                            relative_path = os.path.relpath(
+                                file_name, project_root)
                             yield Completion(
                                 relative_path,
                                 start_position=-len(name),
@@ -757,7 +783,8 @@ class CommandCompleter(Completer):
                                 if len(path_parts) > 3
                                 else file_name
                             )
-                            relative_path = os.path.relpath(file_name, project_root)
+                            relative_path = os.path.relpath(
+                                file_name, project_root)
 
                             yield Completion(
                                 relative_path,
@@ -773,7 +800,8 @@ class CommandCompleter(Completer):
                                 if len(path_parts) > 3
                                 else file_name
                             )
-                            relative_path = os.path.relpath(file_name, project_root)
+                            relative_path = os.path.relpath(
+                                file_name, project_root)
                             yield Completion(
                                 relative_path,
                                 start_position=-len(name),
@@ -791,7 +819,8 @@ class CommandCompleter(Completer):
                                 if len(path_parts) > 3
                                 else symbol.symbol_name
                             )
-                            relative_path = os.path.relpath(file_name, project_root)
+                            relative_path = os.path.relpath(
+                                file_name, project_root)
                             yield Completion(
                                 f"{symbol.symbol_name}(location: {relative_path})",
                                 start_position=-len(name),
@@ -826,7 +855,8 @@ class CommandCompleter(Completer):
                         for dir in dirs:
                             full_path = os.path.join(root, dir)
                             if full_path.startswith(file_name):
-                                relative_path = os.path.relpath(full_path, search_dir)
+                                relative_path = os.path.relpath(
+                                    full_path, search_dir)
                                 yield Completion(
                                     relative_path,
                                     start_position=-len(file_basename),
@@ -838,7 +868,8 @@ class CommandCompleter(Completer):
                                 image_extensions
                             ) and file.startswith(file_basename):
                                 full_path = os.path.join(root, file)
-                                relative_path = os.path.relpath(full_path, search_dir)
+                                relative_path = os.path.relpath(
+                                    full_path, search_dir)
                                 yield Completion(
                                     relative_path,
                                     start_position=-len(file_basename),
@@ -848,7 +879,7 @@ class CommandCompleter(Completer):
                         break
 
             elif words[0] == "/remove_files":
-                new_words = text[len("/remove_files") :].strip().split(",")
+                new_words = text[len("/remove_files"):].strip().split(",")
 
                 is_at_space = text[-1] == " "
                 last_word = new_words[-2] if len(new_words) > 1 else ""
@@ -875,7 +906,7 @@ class CommandCompleter(Completer):
                                 file_name, start_position=-len(current_word)
                             )
             elif words[0] == "/exclude_dirs":
-                new_words = text[len("/exclude_dirs") :].strip().split(",")
+                new_words = text[len("/exclude_dirs"):].strip().split(",")
                 current_word = new_words[-1]
 
                 for file_name in self.all_dir_names:
@@ -883,7 +914,7 @@ class CommandCompleter(Completer):
                         yield Completion(file_name, start_position=-len(current_word))
 
             elif words[0] == "/lib":
-                new_text = text[len("/lib") :]
+                new_text = text[len("/lib"):]
                 parser = CommandTextParser(new_text, words[0])
                 parser.lib()
                 current_word = parser.current_word()
@@ -900,7 +931,7 @@ class CommandCompleter(Completer):
                             )
 
             elif words[0] == "/conf":
-                new_words = text[len("/conf") :].strip().split()
+                new_words = text[len("/conf"):].strip().split()
                 is_at_space = text[-1] == " "
                 last_word = new_words[-2] if len(new_words) > 1 else ""
                 current_word = new_words[-1] if new_words else ""
@@ -919,7 +950,8 @@ class CommandCompleter(Completer):
                     ]
                 # /conf [curosr]
                 elif not last_word and not current_word:
-                    completions = ["/drop"] if "/drop".startswith(current_word) else []
+                    completions = [
+                        "/drop"] if "/drop".startswith(current_word) else []
                     completions += [
                         field_name + ":"
                         for field_name in AutoCoderArgs.model_fields.keys()
@@ -927,7 +959,8 @@ class CommandCompleter(Completer):
                     ]
                 # /conf p[cursor]
                 elif not last_word and current_word:
-                    completions = ["/drop"] if "/drop".startswith(current_word) else []
+                    completions = [
+                        "/drop"] if "/drop".startswith(current_word) else []
                     completions += [
                         field_name + ":"
                         for field_name in AutoCoderArgs.model_fields.keys()
@@ -955,7 +988,7 @@ class CommandCompleter(Completer):
         self.all_files = get_all_file_in_project()
         self.all_dir_names = get_all_dir_names_in_project()
         self.all_files_with_dot = get_all_file_in_project_with_dot()
-        self.symbol_list = get_symbol_list()        
+        self.symbol_list = get_symbol_list()
 
 
 completer = CommandCompleter(commands)
@@ -1021,7 +1054,8 @@ def add_files(args: List[str]):
         completer.refresh_files()
         load_memory()
         console.print(
-            Panel("Refreshed file list.", title="Files Refreshed", border_style="green")
+            Panel("Refreshed file list.",
+                  title="Files Refreshed", border_style="green")
         )
         return
 
@@ -1029,7 +1063,8 @@ def add_files(args: List[str]):
         if len(args) == 1 or (len(args) == 2 and args[1] == "list"):
             if not groups:
                 console.print(
-                    Panel("No groups defined.", title="Groups", border_style="yellow")
+                    Panel("No groups defined.", title="Groups",
+                          border_style="yellow")
                 )
             else:
                 table = Table(
@@ -1054,7 +1089,8 @@ def add_files(args: List[str]):
                     )
                     table.add_row(
                         group_name,
-                        "\n".join([os.path.relpath(f, project_root) for f in files]),
+                        "\n".join([os.path.relpath(f, project_root)
+                                  for f in files]),
                         query_prefix,
                         is_active,
                         end_section=(i == len(groups) - 1),
@@ -1086,7 +1122,8 @@ def add_files(args: List[str]):
                 if group_name in groups_info:
                     del memory["current_files"]["groups_info"][group_name]
                 if group_name in memory["current_files"]["current_groups"]:
-                    memory["current_files"]["current_groups"].remove(group_name)
+                    memory["current_files"]["current_groups"].remove(
+                        group_name)
                 console.print(
                     Panel(
                         f"Dropped group '{group_name}'.",
@@ -1259,7 +1296,8 @@ def remove_files(file_names: List[str]):
         memory["current_files"]["files"] = []
         memory["current_files"]["current_groups"] = []
         console.print(
-            Panel("Removed all files.", title="Files Removed", border_style="green")
+            Panel("Removed all files.",
+                  title="Files Removed", border_style="green")
         )
     else:
         removed_files = []
@@ -1373,6 +1411,76 @@ def get_llm_friendly_package_docs(
     return docs
 
 
+def convert_yaml_to_config(yaml_file: str):
+    from autocoder.auto_coder import AutoCoderArgs, load_include_files, Template
+    args = AutoCoderArgs()
+    with open(yaml_file, "r") as f:
+        config = yaml.safe_load(f)
+        config = load_include_files(config, yaml_file)
+        for key, value in config.items():
+            if key != "file":  # 排除 --file 参数本身
+                # key: ENV {{VARIABLE_NAME}}
+                if isinstance(value, str) and value.startswith("ENV"):
+                    template = Template(value.removeprefix("ENV").strip())
+                    value = template.render(os.environ)
+                setattr(args, key, value)
+    return args
+
+def commit():
+    def prepare_commit_yaml():
+        auto_coder_main(["next", "chat_action"])
+
+    prepare_commit_yaml()
+
+    latest_yaml_file = get_last_yaml_file("actions")
+    
+    conf = memory.get("conf", {})
+    current_files = memory["current_files"]["files"]
+    execute_file = None
+    
+    if latest_yaml_file:        
+        try:
+            execute_file = os.path.join("actions", latest_yaml_file)
+            yaml_config = {
+                "include_file": ["./base/base.yml"],
+                "auto_merge": conf.get("auto_merge", "editblock"),
+                "human_as_model": conf.get("human_as_model", "false") == "true",
+                "skip_build_index": conf.get("skip_build_index", "true") == "true",
+                "skip_confirm": conf.get("skip_confirm", "true") == "true",
+                "silence": conf.get("silence", "true") == "true",
+                "include_project_structure": conf.get("include_project_structure", "true")
+                == "true",
+            }
+            for key, value in conf.items():
+                converted_value = convert_config_value(key, value)
+                if converted_value is not None:
+                    yaml_config[key] = converted_value
+
+            yaml_config["urls"] = current_files + get_llm_friendly_package_docs(
+                return_paths=True
+            )
+            args = convert_yaml_to_config(execute_file)
+            llm = byzerllm.ByzerLLM.from_default_model(args.code_model or args.model)
+            uncommitted_changes = git_utils.get_uncommitted_changes(".")            
+            commit_message = git_utils.generate_commit_message.with_llm(
+                llm).run(uncommitted_changes)
+            memory["conversation"].append({"role": "user", "content": commit_message})
+            yaml_config["query"] = commit_message
+            yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)            
+            with open(os.path.join(execute_file), "w") as f:
+                f.write(yaml_content) 
+                        
+            file_content = open(execute_file).read()
+            md5 = hashlib.md5(file_content.encode('utf-8')).hexdigest()            
+            file_name = os.path.basename(execute_file)             
+            commit_result = git_utils.commit_changes(".", f"auto_coder_{file_name}_{md5}")            
+            git_utils.print_commit_info(commit_result=commit_result)
+        except Exception as e:
+            print(f"Failed to commit: {e}")
+            if execute_file:
+                os.remove(execute_file)
+
+
 def coding(query: str):
     console = Console()
     is_apply = query.strip().startswith("/apply")
@@ -1415,7 +1523,7 @@ def coding(query: str):
             return_paths=True
         )
 
-        ## handle image
+        # handle image
         v = Image.convert_image_paths_from(query)
         yaml_config["query"] = v
 
@@ -1424,7 +1532,8 @@ def coding(query: str):
             active_groups_context = "下面是对上面文件按分组给到的一些描述，当用户的需求正好匹配描述的时候，参考描述来做修改：\n"
             for group in current_groups:
                 group_files = groups.get(group, [])
-                query_prefix = groups_info.get(group, {}).get("query_prefix", "")
+                query_prefix = groups_info.get(
+                    group, {}).get("query_prefix", "")
                 active_groups_context += f"组名: {group}\n"
                 active_groups_context += f"文件列表:\n"
                 for file in group_files:
@@ -1508,10 +1617,10 @@ def code_review(query: str) -> str:
 
 def chat(query: str):
     conf = memory.get("conf", {})
-    
+
     yaml_config = {
         "include_file": ["./base/base.yml"],
-        "include_project_structure": conf.get("include_project_structure", "true") in ["true","True"],
+        "include_project_structure": conf.get("include_project_structure", "true") in ["true", "True"],
         "human_as_model": conf.get("human_as_model", "false") == "true",
         "skip_build_index": conf.get("skip_build_index", "true") == "true",
         "skip_confirm": conf.get("skip_confirm", "true") == "true",
@@ -1868,7 +1977,8 @@ def execute_shell_command(command: str):
                 f"[bold red]Command failed with return code {process.returncode}[/bold red]"
             )
         else:
-            console.print("[bold green]Command completed successfully[/bold green]")
+            console.print(
+                "[bold green]Command completed successfully[/bold green]")
 
     except FileNotFoundError:
         console.print(
@@ -1916,7 +2026,8 @@ def lib_command(args: List[str]):
                     proxy_url,
                     llm_friendly_packages_dir,
                 )
-                console.print("Successfully cloned llm_friendly_packages repository")
+                console.print(
+                    "Successfully cloned llm_friendly_packages repository")
             except git.exc.GitCommandError as e:
                 console.print(f"Error cloning repository: {e}")
 
@@ -1977,7 +2088,8 @@ def lib_command(args: List[str]):
                     console.print(f"Updated remote URL to: {new_url}")
 
                 origin.pull()
-                console.print("Successfully updated llm_friendly_packages repository")
+                console.print(
+                    "Successfully updated llm_friendly_packages repository")
 
             except git.exc.GitCommandError as e:
                 console.print(f"Error updating repository: {e}")
@@ -1999,7 +2111,8 @@ def lib_command(args: List[str]):
                 table.add_row(doc)
             console.print(table)
         else:
-            console.print(f"No markdown files found for package: {package_name}")
+            console.print(
+                f"No markdown files found for package: {package_name}")
 
     else:
         console.print(f"Unknown subcommand: {subcommand}")
@@ -2123,7 +2236,8 @@ def main():
                     FormattedText(prompt_message), default=new_prompt, style=style
                 )
             else:
-                user_input = session.prompt(FormattedText(prompt_message), style=style)
+                user_input = session.prompt(
+                    FormattedText(prompt_message), style=style)
             new_prompt = ""
 
             if "mode" not in memory:
@@ -2148,13 +2262,14 @@ def main():
                 new_prompt = "/coding " + text
 
             elif user_input.startswith("/add_files"):
-                args = user_input[len("/add_files") :].strip().split()
+                args = user_input[len("/add_files"):].strip().split()
                 add_files(args)
             elif user_input.startswith("/remove_files"):
-                file_names = user_input[len("/remove_files") :].strip().split(",")
+                file_names = user_input[len(
+                    "/remove_files"):].strip().split(",")
                 remove_files(file_names)
             elif user_input.startswith("/index/query"):
-                query = user_input[len("/index/query") :].strip()
+                query = user_input[len("/index/query"):].strip()
                 index_query(query)
 
             elif user_input.startswith("/index/build"):
@@ -2164,27 +2279,30 @@ def main():
                 list_files()
 
             elif user_input.startswith("/mode"):
-                conf = user_input[len("/mode") :].strip()
+                conf = user_input[len("/mode"):].strip()
                 if not conf:
                     print(memory["mode"])
                 else:
                     memory["mode"] = conf
 
             elif user_input.startswith("/conf"):
-                conf = user_input[len("/conf") :].strip()
+                conf = user_input[len("/conf"):].strip()
                 if not conf:
                     print(memory["conf"])
                 else:
                     configure(conf)
             elif user_input.startswith("/revert"):
                 revert()
+            elif user_input.startswith("/commit"):
+                commit()
             elif user_input.startswith("/help"):
                 show_help()
             elif user_input.startswith("/exclude_dirs"):
-                dir_names = user_input[len("/exclude_dirs") :].strip().split(",")
+                dir_names = user_input[len(
+                    "/exclude_dirs"):].strip().split(",")
                 exclude_dirs(dir_names)
             elif user_input.startswith("/ask"):
-                query = user_input[len("/ask") :].strip()
+                query = user_input[len("/ask"):].strip()
                 if not query:
                     print("Please enter your question.")
                 else:
@@ -2194,38 +2312,38 @@ def main():
                 raise EOFError()
 
             elif user_input.startswith("/coding"):
-                query = user_input[len("/coding") :].strip()
+                query = user_input[len("/coding"):].strip()
                 if not query:
                     print("\033[91mPlease enter your request.\033[0m")
                     continue
                 coding(query)
             elif user_input.startswith("/chat"):
-                query = user_input[len("/chat") :].strip()
+                query = user_input[len("/chat"):].strip()
                 if not query:
                     print("\033[91mPlease enter your request.\033[0m")
                 else:
                     chat(query)
 
             elif user_input.startswith("/design"):
-                query = user_input[len("/design") :].strip()
+                query = user_input[len("/design"):].strip()
                 if not query:
                     print("\033[91mPlease enter your design request.\033[0m")
                 else:
                     design(query)
 
             elif user_input.startswith("/summon"):
-                query = user_input[len("/summon") :].strip()
+                query = user_input[len("/summon"):].strip()
                 if not query:
                     print("\033[91mPlease enter your request.\033[0m")
                 else:
                     summon(query)
 
             elif user_input.startswith("/lib"):
-                args = user_input[len("/lib") :].strip().split()
+                args = user_input[len("/lib"):].strip().split()
                 lib_command(args)
 
             elif user_input.startswith("/debug"):
-                code = user_input[len("/debug") :].strip()
+                code = user_input[len("/debug"):].strip()
                 try:
                     result = eval(code)
                     print(f"Debug result: {result}")
@@ -2236,7 +2354,7 @@ def main():
             else:
                 command = user_input
                 if user_input.startswith("/shell"):
-                    command = user_input[len("/shell") :].strip()
+                    command = user_input[len("/shell"):].strip()
                 if not command:
                     print("Please enter a shell command to execute.")
                 else: