auto-coder 0.1.263__py3-none-any.whl → 0.1.265__py3-none-any.whl

autocoder/chat_auto_coder.py

@@ -134,6 +134,7 @@ commands = [
     "/index/export",
     "/index/import",
     "/exclude_dirs",
+    "/exclude_files",
     "/help",
     "/shell",
     "/voice_input",
@@ -145,6 +146,8 @@ commands = [
     "/mcp",
     "/models",
     "/auto",
+    "/conf/export",
+    "/conf/import",
 ]
 
 
@@ -268,7 +271,7 @@ def initialize_system(args):
     from autocoder.utils.model_provider_selector import ModelProviderSelector
     from autocoder import models as models_module
     print(f"\n\033[1;34m{get_message('initializing')}\033[0m")
-    
+
     first_time = [False]
     configure_success = [False]
 
@@ -391,7 +394,7 @@ def initialize_system(args):
             except subprocess.CalledProcessError:
                 print_status(get_message("deploy_fail"), "error")
                 return
-            
+
 
             deploy_cmd = [
                 "byzerllm",
@@ -604,7 +607,7 @@ def get_symbol_list() -> List[SymbolItem]:
     index_file = os.path.join(".auto-coder", "index.json")
 
     if os.path.exists(index_file):
-        with open(index_file, "r") as file:
+        with open(index_file, "r",encoding="utf-8") as file:
             index_data = json.load(file)
     else:
         index_data = {}
@@ -641,7 +644,7 @@ def get_symbol_list() -> List[SymbolItem]:
 
 
 def save_memory():
-    with open(os.path.join(base_persist_dir, "memory.json"), "w") as f:
+    with open(os.path.join(base_persist_dir, "memory.json"), "w",encoding="utf-8") as f:
         json.dump(memory, f, indent=2, ensure_ascii=False)
     load_memory()
 
@@ -650,7 +653,7 @@ def load_memory():
     global memory
     memory_path = os.path.join(base_persist_dir, "memory.json")
     if os.path.exists(memory_path):
-        with open(memory_path, "r") as f:
+        with open(memory_path, "r", encoding="utf-8") as f:
             memory = json.load(f)
     completer.update_current_files(memory["current_files"]["files"])
 
@@ -672,12 +675,12 @@ completer = CommandCompleter(commands,
 
 def print_conf(content:Dict[str,Any]):        
         """Display configuration dictionary in a Rich table format with enhanced visual styling.
-        
+
         Args:
             conf (Dict[str, Any]): Configuration dictionary to display
         """
         console = Console()
-        
+
         # Create a styled table with rounded borders
         table = Table(
             show_header=True,
@@ -687,11 +690,11 @@ def print_conf(content:Dict[str,Any]):
             border_style="blue",
             show_lines=True            
         )
-        
+
         # Add columns with explicit width and alignment
         table.add_column(get_message("conf_key"), style="cyan", justify="right", width=30, no_wrap=False)
         table.add_column(get_message("conf_value"), style="green", justify="left", width=50, no_wrap=False)
-        
+
         # Sort keys for consistent display
         for key in sorted(content.keys()):
             value = content[key]
@@ -704,9 +707,9 @@ def print_conf(content:Dict[str,Any]):
                 formatted_value = Text(str(value), style="bright_cyan")
             else:
                 formatted_value = Text(str(value), style="green")
-                
+
             table.add_row(str(key), formatted_value)
-        
+
         # Add padding and print with a panel
         console.print(Panel(
             table,
@@ -742,7 +745,7 @@ def revert():
 
 
 def add_files(args: List[str]):
-    
+
     result_manager = ResultManager()
     if "groups" not in memory["current_files"]:
         memory["current_files"]["groups"] = {}
@@ -837,7 +840,7 @@ def add_files(args: List[str]):
             )
             result_manager.append(content=f"Added group '{group_name}' with current files.", 
                               meta={"action": "add_files","success":True, "input":{ "args": args}})
-            
+
         elif len(args) >= 3 and args[1] == "/drop":
             group_name = args[2]
             if group_name in groups:
@@ -1093,7 +1096,7 @@ def ask(query: str):
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(os.path.join(execute_file), "w") as f:
+    with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
         f.write(yaml_content)
 
     def execute_ask():
@@ -1141,7 +1144,7 @@ def get_llm_friendly_package_docs(
                                         if return_paths:
                                             docs.append(file_path)
                                         else:
-                                            with open(file_path, "r") as f:
+                                            with open(file_path, "r",encoding="utf-8") as f:
                                                 docs.append(f.read())
 
     return docs
@@ -1151,7 +1154,7 @@ 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:
+    with open(yaml_file, "r",encoding="utf-8") as f:
         config = yaml.safe_load(f)
         config = load_include_files(config, yaml_file)
         for key, value in config.items():
@@ -1241,7 +1244,7 @@ def mcp(query: str):
 
     temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
     try:
-        with open(temp_yaml, "w") as f:
+        with open(temp_yaml, "w",encoding="utf-8") as f:
             f.write(convert_yaml_config_to_str(yaml_config=yaml_config))
         args = convert_yaml_to_config(temp_yaml)
     finally:
@@ -1272,14 +1275,14 @@ def mcp(query: str):
         os.makedirs(mcp_dir, exist_ok=True)
         timestamp = str(int(time.time()))
         file_path = os.path.join(mcp_dir, f"{timestamp}.md")
-        
+
         # Format response as markdown
         markdown_content = response.result
-        
+
         # Save to file
         with open(file_path, "w", encoding="utf-8") as f:
             f.write(markdown_content)
-            
+
         console = Console()
         console.print(
             Panel(
@@ -1310,7 +1313,7 @@ def code_next(query: str):
 
     temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
     try:
-        with open(temp_yaml, "w") as f:
+        with open(temp_yaml, "w",encoding="utf-8") as f:
             f.write(convert_yaml_config_to_str(yaml_config=yaml_config))
         args = convert_yaml_to_config(temp_yaml)
     finally:
@@ -1417,20 +1420,20 @@ def commit(query: str):
             # 临时保存yaml文件，然后读取yaml文件，转换为args
             temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
             try:
-                with open(temp_yaml, "w") as f:
+                with open(temp_yaml, "w",encoding="utf-8") 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)
-            
+
             target_model = args.commit_model or args.model
             llm = get_single_llm(target_model, product_mode)
             printer = Printer()
             printer.print_in_terminal("commit_generating", style="yellow", model_name=target_model)
             commit_message = ""
-            
+
             try:
                 uncommitted_changes = git_utils.get_uncommitted_changes(".")
                 commit_message = git_utils.generate_commit_message.with_llm(llm).run(
@@ -1441,10 +1444,10 @@ def commit(query: str):
             except Exception as e:
                 printer.print_in_terminal("commit_failed", style="red", error=str(e), model_name=target_model)
                 return
-            
+
             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:
+            with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
                 f.write(yaml_content)
 
             file_content = open(execute_file).read()
@@ -1504,6 +1507,7 @@ def coding(query: str):
             "silence": conf.get("silence", "true") == "true",
             "include_project_structure": conf.get("include_project_structure", "true")
             == "true",
+            "exclude_files": conf.get("exclude_files", []),
         }
 
         yaml_config["context"] = ""        
@@ -1513,14 +1517,14 @@ def coding(query: str):
             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
         )
 
         if conf.get("enable_global_memory", "true") in ["true", "True",True]:
             yaml_config["urls"] += get_global_memory_file_paths()
-        
+
         # handle image
         v = Image.convert_image_paths_from(query)
         yaml_config["query"] = v
@@ -1559,7 +1563,7 @@ def coding(query: str):
                 error_message()
                 return
 
-            with open(memory_file, "r") as f:
+            with open(memory_file, "r",encoding="utf-8") as f:
                 chat_history = json.load(f)
 
             if not chat_history["ask_conversation"]:
@@ -1583,7 +1587,7 @@ def coding(query: str):
         md5 = hashlib.md5(yaml_content.encode("utf-8")).hexdigest()
 
         execute_file = os.path.join("actions", latest_yaml_file)
-        with open(os.path.join(execute_file), "w") as f:
+        with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
             f.write(yaml_content)
 
         def execute_chat():
@@ -1632,6 +1636,7 @@ def chat(query: str):
         "skip_build_index": conf.get("skip_build_index", "true") == "true",
         "skip_confirm": conf.get("skip_confirm", "true") == "true",
         "silence": conf.get("silence", "true") == "true",
+        "exclude_files": conf.get("exclude_files", []),
     }
 
     current_files = memory["current_files"]["files"] + get_llm_friendly_package_docs(
@@ -1667,7 +1672,7 @@ def chat(query: str):
     if "/save" in query:
         yaml_config["action"].append("save")
         query = query.replace("/save", "", 1).strip()        
-    
+
     if "/review" in query and "/commit" in query:
         yaml_config["action"].append("review_commit")
         query = query.replace("/review", "", 1).replace("/commit", "", 1).strip()
@@ -1698,7 +1703,7 @@ def chat(query: str):
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(os.path.join(execute_file), "w") as f:
+    with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
         f.write(yaml_content)
 
     def execute_ask():
@@ -1722,7 +1727,7 @@ def summon(query: str):
     for file in current_files:
         if os.path.exists(file):
             try:
-                with open(file, "r") as f:
+                with open(file, "r",encoding="utf-8") as f:
                     content = f.read()
                     s = f"##File: {file}\n{content}\n\n"
                     file_contents.append(s)
@@ -1758,7 +1763,7 @@ def summon(query: str):
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(os.path.join(execute_file), "w") as f:
+    with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
         f.write(yaml_content)
 
     def execute_summon():
@@ -1805,7 +1810,7 @@ def design(query: str):
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(os.path.join(execute_file), "w") as f:
+    with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
         f.write(yaml_content)
 
     def execute_design():
@@ -1834,7 +1839,7 @@ def voice_input():
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(os.path.join(execute_file), "w") as f:
+    with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
         f.write(yaml_content)
 
     def execute_voice2text_command():
@@ -1842,7 +1847,7 @@ def voice_input():
 
     try:
         execute_voice2text_command()
-        with open(os.path.join(".auto-coder", "exchange.txt"), "r") as f:
+        with open(os.path.join(".auto-coder", "exchange.txt"), "r",encoding="utf-8") as f:
             return f.read()
     finally:
         os.remove(execute_file)
@@ -1864,12 +1869,12 @@ def generate_shell_command(input_text):
 
     execute_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(os.path.join(execute_file), "w") as f:
+    with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
         f.write(yaml_content)
 
     try:
         auto_coder_main(["agent", "generate_command", "--file", execute_file])
-        with open(os.path.join(".auto-coder", "exchange.txt"), "r") as f:
+        with open(os.path.join(".auto-coder", "exchange.txt"), "r",encoding="utf-8") as f:
             shell_script = f.read()
         result_manager = ResultManager()
         result_manager.add_result(content=shell_script,meta={
@@ -1892,22 +1897,22 @@ def manage_models(query: str):
     """
     printer = Printer()
     console = Console()
-    
+
     product_mode = memory.get("product_mode", "lite")
     if product_mode != "lite":
         printer.print_in_terminal("models_lite_only", style="red")
         return
-        
+
     models_data = models_module.load_models()
     subcmd = ""
     if "/list" in query:
         subcmd = "/list"
         query = query.replace("/list", "", 1).strip()
-    
+
     if "/add_model" in query:
         subcmd = "/add_model"
         query = query.replace("/add_model", "", 1).strip()
-    
+
     if "/add" in query:
         subcmd = "/add"
         query = query.replace("/add", "", 1).strip()
@@ -1916,7 +1921,7 @@ def manage_models(query: str):
     if "/activate" in query:
         subcmd = "/add"
         query = query.replace("/activate", "", 1).strip()    
-    
+
     if "/remove" in query:
         subcmd = "/remove"
         query = query.replace("/remove", "", 1).strip()
@@ -1936,23 +1941,23 @@ def manage_models(query: str):
     if "output_price" in query:
         subcmd = "/output_price"
         query = query.replace("/output_price", "", 1).strip()        
-            
+
     if "/speed" in query:
         subcmd = "/speed"
         query = query.replace("/speed", "", 1).strip()
-    
-        
+
+
 
     if not subcmd:
         printer.print_in_terminal("models_usage")        
-    
+
     result_manager = ResultManager()
     if subcmd == "/list":                    
         if models_data:
             # Sort models by speed (average_speed)
             sorted_models = sorted(models_data, key=lambda x: float(x.get('average_speed', 0)))
             sorted_models.reverse()
-            
+
             table = Table(
                 title=printer.get_message_from_key("models_title"),
                 expand=True,
@@ -1973,7 +1978,7 @@ def manage_models(query: str):
                     if not api_key:                                                
                         printer.print_in_terminal("models_api_key_empty", style="yellow", name=name)                                           
                     name = f"{name} *"
-                
+
                 table.add_row(
                     name,                    
                     m.get("model_name", ""),                    
@@ -1989,7 +1994,7 @@ def manage_models(query: str):
                     "query": query
                 }
             })
-            
+
         else:
             printer.print_in_terminal("models_no_models", style="yellow")
             result_manager.add_result(content="No models found",meta={
@@ -2037,7 +2042,7 @@ def manage_models(query: str):
                 }
             })
             printer.print_in_terminal("models_input_price_usage", style="red")
-            
+
     elif subcmd == "/output_price":
         args = query.strip().split()
         if len(args) >= 2:
@@ -2115,11 +2120,11 @@ def manage_models(query: str):
                 }
             })
             printer.print_in_terminal("models_speed_usage", style="red")
-            
+
     elif subcmd == "/speed-test":
         from autocoder.common.model_speed_test import render_speed_test_in_terminal
         test_rounds = 1  # 默认测试轮数
-        
+
         enable_long_context = False
         if "/long_context" in query:
             enable_long_context = True
@@ -2133,7 +2138,7 @@ def manage_models(query: str):
         args = query.strip().split()
         if args and args[0].isdigit():
             test_rounds = int(args[0])
-            
+
         render_speed_test_in_terminal(product_mode, test_rounds,enable_long_context=enable_long_context)
         ## 等待优化，获取明细数据
         result_manager.add_result(content="models test success",meta={
@@ -2142,7 +2147,7 @@ def manage_models(query: str):
                 "query": query
             }
         })
-    
+
     elif subcmd == "/add":
         # Support both simplified and legacy formats
         args = query.strip().split(" ")        
@@ -2265,18 +2270,61 @@ def manage_models(query: str):
 
 def exclude_dirs(dir_names: List[str]):
     new_dirs = dir_names
-    existing_dirs = memory.get("exclude_dirs", [])
+    existing_dirs = memory.get("exclude_dirs", [])    
     dirs_to_add = [d for d in new_dirs if d not in existing_dirs]
+    
     if dirs_to_add:
         existing_dirs.extend(dirs_to_add)
         if "exclude_dirs" not in memory:
             memory["exclude_dirs"] = existing_dirs
         print(f"Added exclude dirs: {dirs_to_add}")
+        exclude_files([f"regex://.*/{d}/*." for d in dirs_to_add])            
     else:
         print("All specified dirs are already in the exclude list.")
     save_memory()
     completer.refresh_files()
 
+def exclude_files(file_patterns: List[str]):
+    result_manager = ResultManager()
+    printer = Printer()
+    new_file_patterns = file_patterns
+    existing_file_patterns = memory.get("exclude_files", [])    
+    file_patterns_to_add = [f for f in new_file_patterns if f not in existing_file_patterns]
+
+    for file_pattern in file_patterns_to_add:
+        if not file_pattern.startswith("regex://"): 
+            result_manager.add_result(content=printer.get_message_from_key_with_format("invalid_file_pattern", file_pattern=file_pattern),meta={
+                "action": "exclude_files",
+                "input": {
+                    "query": file_pattern
+                }
+            })
+            raise ValueError(printer.get_message_from_key_with_format("invalid_file_pattern", file_pattern=file_pattern))
+
+    if file_patterns_to_add:
+        existing_file_patterns.extend(file_patterns_to_add)
+        if "exclude_files" not in memory:
+            memory["exclude_files"] = existing_file_patterns
+        
+        result_manager.add_result(content=f"Added exclude files: {file_patterns_to_add}",meta={
+            "action": "exclude_files",
+            "input": {
+                "query": file_patterns_to_add
+            }
+        })
+        save_memory()
+        print(f"Added exclude files: {file_patterns_to_add}")
+    else:
+        result_manager.add_result(content=f"All specified files are already in the exclude list.",meta={
+            "action": "exclude_files",
+            "input": {
+                "query": file_patterns_to_add
+            }
+        })
+        print("All specified files are already in the exclude list.")
+    
+    
+
 @run_in_raw_thread()
 def index_build():
     conf = memory.get("conf", {})
@@ -2292,7 +2340,7 @@ def index_build():
     yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)
     yaml_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(yaml_file, "w") as f:
+    with open(yaml_file, "w",encoding="utf-8") as f:
         f.write(yaml_content)
     try:
         with redirect_stdout() as output:
@@ -2322,7 +2370,7 @@ def get_final_config()->AutoCoderArgs:
 
     temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
     try:
-        with open(temp_yaml, "w") as f:
+        with open(temp_yaml, "w",encoding="utf-8") as f:
             f.write(convert_yaml_config_to_str(yaml_config=yaml_config))
         args = convert_yaml_to_config(temp_yaml)
     finally:
@@ -2339,26 +2387,26 @@ def help(query: str):
     auto_config_tuner.tune(AutoConfigRequest(query=query))    
 
 @run_in_raw_thread()
-def index_export(export_path: str):
+def index_export(path: str):
     from autocoder.common.index_import_export import export_index
     from autocoder.common.printer import Printer
     printer = Printer()
     project_root = os.getcwd()
-    if export_index(project_root, export_path):
-        printer.print_in_terminal("index_export_success", path=export_path)
+    if export_index(project_root, path):
+        printer.print_in_terminal("index_export_success", path=path)
     else:
-        printer.print_in_terminal("index_export_fail", path=export_path)
+        printer.print_in_terminal("index_export_fail", path=path)
 
 @run_in_raw_thread()
-def index_import(import_path: str):
+def index_import(path: str):
     from autocoder.common.index_import_export import import_index
     from autocoder.common.printer import Printer
     printer = Printer()
     project_root = os.getcwd()
-    if import_index(project_root, import_path):
-        printer.print_in_terminal("index_import_success", path=import_path)
+    if import_index(project_root, path):
+        printer.print_in_terminal("index_import_success", path=path)
     else:
-        printer.print_in_terminal("index_import_fail", path=import_path)
+        printer.print_in_terminal("index_import_fail", path=path)
 
 @run_in_raw_thread()
 def index_query(query: str):
@@ -2377,7 +2425,7 @@ def index_query(query: str):
     yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)
     yaml_file = os.path.join("actions", f"{uuid.uuid4()}.yml")
 
-    with open(yaml_file, "w") as f:
+    with open(yaml_file, "w",encoding="utf-8") as f:
         f.write(yaml_content)
     try:
         with redirect_stdout() as output:
@@ -2409,10 +2457,30 @@ def list_files():
             )
         )
 
-@run_in_raw_thread()
-def execute_shell_command(command: str):
-    from autocoder.common.shells import execute_shell_command as shell_exec
-    shell_exec(command)
+
+def gen_and_exec_shell_command(query: str):
+    from rich.prompt import Confirm    
+    from autocoder.common.printer import Printer 
+    from rich.console import Console
+    
+    printer = Printer()
+    console = Console()
+    # Generate the shell script
+    shell_script = generate_shell_command(query)    
+
+    # Ask for confirmation using rich
+    if Confirm.ask(
+        printer.get_message_from_key("confirm_execute_shell_script"),
+        default=False
+    ):
+        execute_shell_command(shell_script)
+    else:
+        console.print(
+            Panel(
+                printer.get_message_from_key("shell_script_not_executed"),
+                border_style="yellow"
+            )
+        )
 
 
 def lib_command(args: List[str]):
@@ -2542,18 +2610,32 @@ def lib_command(args: List[str]):
     else:
         console.print(f"Unknown subcommand: {subcommand}")
 
+
+def execute_shell_command(command: str):
+    from autocoder.common.shells import execute_shell_command as shell_exec
+    shell_exec(command)
+
+
+def conf_export(path: str):
+    from autocoder.common.conf_import_export import export_conf
+    export_conf(os.getcwd(), path)
+
+def conf_import(path: str):
+    from autocoder.common.conf_import_export import import_conf
+    import_conf(os.getcwd(), path)
+
 @run_in_raw_thread()
 def auto_command(params,query: str):
     """处理/auto指令"""
     from autocoder.commands.auto_command import CommandAutoTuner, AutoCommandRequest, CommandConfig, MemoryConfig
     args = get_final_config()
     # help(query)
-    
+
     # 准备请求参数
     request = AutoCommandRequest(
         user_input=query        
     )
-    
+
     # 初始化调优器
     llm = get_single_llm(args.chat_model or args.model,product_mode=args.product_mode)    
     tuner = CommandAutoTuner(llm, 
@@ -2568,6 +2650,7 @@ def auto_command(params,query: str):
                                  commit=commit,
                                  help=help,
                                  exclude_dirs=exclude_dirs,
+                                 exclude_files=exclude_files,
                                  ask=ask,
                                  chat=chat,
                                  coding=coding,
@@ -2579,9 +2662,13 @@ def auto_command(params,query: str):
                                  index_build=index_build,
                                  index_query=index_query,  
                                  execute_shell_command=execute_shell_command,  
-                                 generate_shell_command=generate_shell_command                                                                                       
+                                 generate_shell_command=generate_shell_command,
+                                 conf_export=conf_export,
+                                 conf_import=conf_import,
+                                 index_export=index_export,
+                                 index_import=index_import                                                                                       
                              ))
-    
+
     # 生成建议
     response = tuner.analyze(request)
     printer = Printer()
@@ -2593,7 +2680,7 @@ def auto_command(params,query: str):
         border_style="blue",
         padding=(1, 2)
     ))    
-      
+
 
 def main():
     from autocoder.rag.variable_holder import VariableHolder
@@ -2606,20 +2693,20 @@ def main():
         VariableHolder.TOKENIZER_MODEL = Tokenizer.from_file(tokenizer_path)
     except FileNotFoundError:
         tokenizer_path = None
-        
+
     ARGS = parse_arguments()
-    
+
     if ARGS.lite:
         ARGS.product_mode = "lite"
-    
+
     if ARGS.pro:
         ARGS.product_mode = "pro" 
 
     if not ARGS.quick:
         initialize_system(ARGS)    
-    
+
     load_memory()
-    
+
     configure(f"product_mode:{ARGS.product_mode}")
 
     MODES = {
@@ -2680,7 +2767,11 @@ def main():
         human_as_model = memory["conf"].get("human_as_model", "false")
         if mode not in MODES:
             mode = "auto_detect"
-        return f" Mode: {MODES[mode]} (ctl+k) | Human as Model: {human_as_model} (ctl+n or /conf human_as_model:true/false)"
+        pwd = os.getcwd()    
+        pwd_parts = pwd.split(os.sep)
+        if len(pwd_parts) > 3:
+            pwd = os.sep.join(pwd_parts[-3:])
+        return f"Current Dir: {pwd} \nMode: {MODES[mode]} | Human as Model: {human_as_model} "
 
     session = PromptSession(
         history=InMemoryHistory(),
@@ -2773,20 +2864,14 @@ def main():
 
             elif user_input.startswith("/index/build"):
                 index_build()
-                
+
             elif user_input.startswith("/index/export"):
                 export_path = user_input[len("/index/export"):].strip()
-                if not export_path:
-                    print("Please specify the export path")
-                else:
-                    index_export(export_path)
-                    
+                index_export(export_path)                    
+
             elif user_input.startswith("/index/import"):
-                import_path = user_input[len("/index/import"):].strip()
-                if not import_path:
-                    print("Please specify the import path")
-                else:
-                    index_import(import_path)
+                import_path  = user_input[len("/index/import"):].strip()
+                index_import(import_path)                    
 
             elif user_input.startswith("/list_files"):
                 list_files()
@@ -2804,7 +2889,17 @@ def main():
                     print(memory["mode"])
                 else:
                     memory["mode"] = conf                    
-
+            
+            elif user_input.startswith("/conf/export"):
+                from autocoder.common.conf_import_export import export_conf
+                export_path = user_input[len("/conf/export"):].strip()
+                export_conf(os.getcwd(), export_path)                    
+            
+            elif user_input.startswith("/conf/import"):
+                from autocoder.common.conf_import_export import import_conf
+                import_path = user_input[len("/conf/import"):].strip()
+                import_conf(os.getcwd(), import_path)                    
+                    
             elif user_input.startswith("/conf"):
                 conf = user_input[len("/conf"):].strip()
                 if not conf:
@@ -2822,11 +2917,17 @@ def main():
                     show_help()
                 else:
                     help(query)
-                    
+
             elif user_input.startswith("/exclude_dirs"):
                 dir_names = user_input[len(
                     "/exclude_dirs"):].strip().split(",")
                 exclude_dirs(dir_names)
+
+            elif user_input.startswith("/exclude_files"):
+                file_patterns = user_input[len(
+                    "/exclude_files"):].strip().split(",")
+                exclude_files(file_patterns)
+
             elif user_input.startswith("/ask"):
                 query = user_input[len("/ask"):].strip()
                 if not query:
@@ -2889,12 +2990,16 @@ def main():
             # elif user_input.startswith("/shell"):
             else:
                 command = user_input
-                if user_input.startswith("/shell"):
+                if user_input.startswith("/shell"):                    
                     command = user_input[len("/shell"):].strip()
                     if not command:
                         print("Please enter a shell command to execute.")
                     else:
-                        execute_shell_command(command)
+                        if command.startswith("/chat"):
+                            command = command[len("/chat"):].strip()
+                            gen_and_exec_shell_command(command)
+                        else:
+                            execute_shell_command(command)
 
         except KeyboardInterrupt:
             continue