auto-coder 0.1.209__py3-none-any.whl → 0.1.211__py3-none-any.whl

autocoder/chat_auto_coder.py

@@ -60,7 +60,8 @@ from byzerllm.utils import format_str_jinja2
 from autocoder.chat_auto_coder_lang import get_message
 from autocoder.utils import operate_config_api
 from autocoder.agent.auto_guess_query import AutoGuessQuery
-
+from autocoder.common.mcp_server import get_mcp_server, McpRequest
+import asyncio
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -72,8 +73,7 @@ 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",
@@ -125,6 +125,7 @@ commands = [
     "/mode",
     "/lib",
     "/design",
+    "/mcp",
 ]
 
 
@@ -155,10 +156,8 @@ 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/commit\033[0m - \033[92m{get_message('commit_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"
     )
@@ -171,8 +170,7 @@ 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"
     )
@@ -182,11 +180,9 @@ 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()
 
 
@@ -211,12 +207,10 @@ 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"))
@@ -262,8 +256,7 @@ 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(
@@ -280,8 +273,7 @@ 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()
@@ -291,8 +283,7 @@ 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:
@@ -435,8 +426,7 @@ def get_all_file_in_project_with_dot() -> List[str]:
     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
 
 
@@ -564,10 +554,8 @@ 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/commit\033[0m - \033[92m{get_message('commit_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"
     )
@@ -580,8 +568,7 @@ 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"
     )
@@ -591,11 +578,9 @@ 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()
 
 
@@ -659,13 +644,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()
@@ -696,21 +681,18 @@ 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 = (
@@ -740,7 +722,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()
@@ -767,8 +749,7 @@ 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),
@@ -785,8 +766,7 @@ 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,
@@ -802,8 +782,7 @@ 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),
@@ -821,8 +800,7 @@ 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),
@@ -857,8 +835,7 @@ 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),
@@ -870,8 +847,7 @@ 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),
@@ -881,7 +857,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 ""
@@ -908,7 +884,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:
@@ -916,7 +892,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()
@@ -932,7 +908,7 @@ class CommandCompleter(Completer):
                                 lib_name, start_position=-len(current_word)
                             )
             elif words[0] == "/coding":
-                new_text = text[len("/coding"):]
+                new_text = text[len("/coding") :]
                 parser = CommandTextParser(new_text, words[0])
                 parser.lib()
                 current_word = parser.current_word()
@@ -941,7 +917,7 @@ class CommandCompleter(Completer):
                         yield Completion(command, start_position=-len(current_word))
 
             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 ""
@@ -960,8 +936,7 @@ 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()
@@ -969,8 +944,7 @@ 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()
@@ -1064,8 +1038,7 @@ 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
 
@@ -1073,8 +1046,7 @@ 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(
@@ -1099,8 +1071,7 @@ 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),
@@ -1132,8 +1103,7 @@ 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}'.",
@@ -1306,8 +1276,7 @@ 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 = []
@@ -1423,6 +1392,7 @@ def get_llm_friendly_package_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)
@@ -1437,6 +1407,46 @@ def convert_yaml_to_config(yaml_file: str):
     return args
 
 
+def mcp(query: str):
+    conf = memory.get("conf", {})
+    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
+
+    temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
+    try:
+        with open(temp_yaml, "w") as f:
+            f.write(convert_yaml_config_to_str(yaml_config=yaml_config))
+        args = convert_yaml_to_config(temp_yaml)
+    finally:
+        if os.path.exists(temp_yaml):
+            os.remove(temp_yaml)    
+    
+    mcp_server = get_mcp_server()
+    response = mcp_server.send_request(
+        McpRequest(
+            query=query,
+            model=args.inference_model or args.model
+        )
+    )
+    
+    if response.error:
+        print(f"Error from MCP server: {response.error}")
+    else:
+        print(response.result)
+
+
 def code_next(query: str):
     conf = memory.get("conf", {})
     yaml_config = {
@@ -1463,17 +1473,13 @@ def code_next(query: str):
         if os.path.exists(temp_yaml):
             os.remove(temp_yaml)
 
-    llm = byzerllm.ByzerLLM.from_default_model(
-        args.inference_model or args.model)
+    llm = byzerllm.ByzerLLM.from_default_model(args.inference_model or args.model)
 
-    auto_guesser = AutoGuessQuery(
-        llm=llm,
-        project_dir=os.getcwd(),
-        skip_diff=True
-    )
+    auto_guesser = AutoGuessQuery(llm=llm, project_dir=os.getcwd(), skip_diff=True)
 
     predicted_tasks = auto_guesser.predict_next_tasks(
-        5, is_human_as_model=args.human_as_model)
+        5, is_human_as_model=args.human_as_model
+    )
 
     if not predicted_tasks:
         console = Console()
@@ -1483,39 +1489,34 @@ def code_next(query: str):
     console = Console()
 
     # Create main panel for all predicted tasks
-    table = Table(show_header=True,
-                  header_style="bold magenta", show_lines=True)
+    table = Table(show_header=True, header_style="bold magenta", show_lines=True)
     table.add_column("Priority", style="cyan", width=8)
-    table.add_column("Task Description", style="green",
-                     width=40, overflow="fold")
+    table.add_column("Task Description", style="green", width=40, overflow="fold")
     table.add_column("Files", style="yellow", width=30, overflow="fold")
     table.add_column("Reason", style="blue", width=30, overflow="fold")
-    table.add_column("Dependencies", style="magenta",
-                     width=30, overflow="fold")
+    table.add_column("Dependencies", style="magenta", width=30, overflow="fold")
 
     for task in predicted_tasks:
         # Format file paths to be more readable
-        file_list = "\n".join([os.path.relpath(f, os.getcwd())
-                              for f in task.urls])
+        file_list = "\n".join([os.path.relpath(f, os.getcwd()) for f in task.urls])
 
         # Format dependencies to be more readable
-        dependencies = "\n".join(
-            task.dependency_queries) if task.dependency_queries else "None"
+        dependencies = (
+            "\n".join(task.dependency_queries) if task.dependency_queries else "None"
+        )
 
         table.add_row(
-            str(task.priority),
-            task.query,
-            file_list,
-            task.reason,
-            dependencies
+            str(task.priority), task.query, file_list, task.reason, dependencies
         )
 
-    console.print(Panel(
-        table,
-        title="[bold]Predicted Next Tasks[/bold]",
-        border_style="blue",
-        padding=(1, 2)  # Add more horizontal padding
-    ))
+    console.print(
+        Panel(
+            table,
+            title="[bold]Predicted Next Tasks[/bold]",
+            border_style="blue",
+            padding=(1, 2),  # Add more horizontal padding
+        )
+    )
 
 
 def commit(query: str):
@@ -1560,30 +1561,29 @@ def commit(query: str):
             temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
             try:
                 with open(temp_yaml, "w") as f:
-                    f.write(convert_yaml_config_to_str(
-                        yaml_config=yaml_config))
+                    f.write(convert_yaml_config_to_str(yaml_config=yaml_config))
                 args = convert_yaml_to_config(temp_yaml)
             finally:
                 if os.path.exists(temp_yaml):
                     os.remove(temp_yaml)
 
-            llm = byzerllm.ByzerLLM.from_default_model(
-                args.code_model or args.model)
+            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})
+            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()
+            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}")
+                ".", 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}")
@@ -1650,8 +1650,7 @@ 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:
@@ -1718,18 +1717,18 @@ def coding(query: str):
 @byzerllm.prompt()
 def code_review(query: str) -> str:
     """
-    对代码进行review，参考如下检查点。        
+    对代码进行review，参考如下检查点。
     1. 有没有调用不符合方法，类的签名的调用
     2. 有没有未声明直接使用的变量，方法，类
     3. 有没有明显的语法错误
-    4. 如果是python代码，检查有没有缩进方面的错误  
+    4. 如果是python代码，检查有没有缩进方面的错误
     5. 如果是python代码，检查是否 try 后面缺少 except 或者 finally
     {% if query %}
     6. 用户的额外的检查需求：{{ query }}
     {% endif %}
 
     如果用户的需求包含了@一个文件名 或者 @@符号， 那么重点关注这些文件或者符号（函数，类）进行上述的review。
-    review 过程中严格遵循上述的检查点，不要遗漏，没有发现异常的点直接跳过，只对发现的异常点，给出具体的修改后的代码。    
+    review 过程中严格遵循上述的检查点，不要遗漏，没有发现异常的点直接跳过，只对发现的异常点，给出具体的修改后的代码。
     """
 
 
@@ -1738,7 +1737,8 @@ def chat(query: str):
 
     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",
@@ -2095,8 +2095,7 @@ 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(
@@ -2144,8 +2143,7 @@ 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}")
 
@@ -2206,8 +2204,7 @@ 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}")
@@ -2229,8 +2226,7 @@ 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}")
@@ -2300,9 +2296,7 @@ def main():
             memory["mode"] = "normal"
         mode = memory["mode"]
         human_as_model = memory["conf"].get("human_as_model", "false")
-        return (
-            f" Mode: {MODES[mode]} (ctl+k) | Human as Model: {human_as_model} (ctl+n or /conf human_as_model:true/false)"
-        )
+        return f" Mode: {MODES[mode]} (ctl+k) | Human as Model: {human_as_model} (ctl+n or /conf human_as_model:true/false)"
 
     session = PromptSession(
         history=InMemoryHistory(),
@@ -2354,8 +2348,7 @@ 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:
@@ -2380,14 +2373,13 @@ 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"):
@@ -2397,14 +2389,14 @@ 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:
@@ -2412,16 +2404,15 @@ def main():
             elif user_input.startswith("/revert"):
                 revert()
             elif user_input.startswith("/commit"):
-                query = user_input[len("/commit"):].strip()
+                query = user_input[len("/commit") :].strip()
                 commit(query)
             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:
@@ -2431,38 +2422,45 @@ 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("/mcp"):
+                query = user_input[len("/mcp") :].strip()
+                if not query:
+                    print("Please enter your query.")
+                else:
+                    mcp(query)
+
             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}")
@@ -2473,7 +2471,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:
@@ -2484,6 +2482,11 @@ def main():
         except EOFError:
             try:
                 save_memory()
+                try:
+                    if get_mcp_server():
+                        get_mcp_server().stop()                
+                except Exception as e:
+                    pass
             except Exception as e:
                 print(
                     f"\033[91mAn error occurred while saving memory:\033[0m \033[93m{type(e).__name__}\033[0m - {str(e)}"