autocoder-nano 0.1.36__py3-none-any.whl → 0.1.38__py3-none-any.whl

autocoder_nano/auto_coder_nano.py

@@ -11,13 +11,13 @@ import uuid
 from autocoder_nano.agent.agentic_edit import AgenticEdit
 from autocoder_nano.agent.agentic_edit_types import AgenticEditRequest
 from autocoder_nano.chat import stream_chat_display
-from autocoder_nano.edit import Dispacher
+from autocoder_nano.edit import coding
 from autocoder_nano.helper import show_help
-from autocoder_nano.index.entry import build_index_and_filter_files
-from autocoder_nano.index.index_manager import IndexManager
-from autocoder_nano.index.symbols_utils import extract_symbols
+from autocoder_nano.project import project_source
+from autocoder_nano.index import (index_export, index_import, index_build,
+                                  index_build_and_filter, extract_symbols)
+from autocoder_nano.rules import rules_from_active_files, rules_from_commit_changes, get_rules_context
 from autocoder_nano.llm_client import AutoLLM
-from autocoder_nano.rules.rules_learn import AutoRulesLearn
 from autocoder_nano.utils.completer_utils import CommandCompleter
 from autocoder_nano.version import __version__
 from autocoder_nano.llm_types import *
@@ -26,7 +26,6 @@ from autocoder_nano.templates import create_actions
 from autocoder_nano.git_utils import (repo_init, commit_changes, revert_changes,
                                       get_uncommitted_changes, generate_commit_message)
 from autocoder_nano.sys_utils import default_exclude_dirs, detect_env
-from autocoder_nano.project import PyProject, SuffixProject
 from autocoder_nano.utils.printer_utils import Printer
 
 import yaml
@@ -49,11 +48,12 @@ from rich.text import Text
 
 printer = Printer()
 console = printer.get_console()
-# console = Console()
+
+
 project_root = os.getcwd()
 base_persist_dir = os.path.join(project_root, ".auto-coder", "plugins", "chat-auto-coder")
-# defaut_exclude_dirs = [".git", ".svn", "node_modules", "dist", "build", "__pycache__", ".auto-coder", "actions",
-#                        ".vscode", ".idea", ".hg"]
+
+
 commands = [
     "/add_files", "/remove_files", "/list_files", "/conf", "/coding", "/chat", "/revert", "/index/query",
     "/index/build", "/exclude_dirs", "/exclude_files", "/help", "/shell", "/exit", "/mode", "/models", "/commit",
@@ -65,8 +65,6 @@ memory = {
     "current_files": {"files": [], "groups": {}},
     "conf": {
         "auto_merge": "editblock",
-        # "current_chat_model": "",
-        # "current_code_model": "",
         "chat_model": "",
         "code_model": "",
     },
@@ -317,119 +315,13 @@ def exclude_files(query: str):
 
 def index_command(llm):
     args = get_final_config(query="", delete_execute_file=True)
-
-    source_dir = os.path.abspath(args.source_dir)
-    printer.print_text(f"开始对目录 {source_dir} 中的源代码进行索引", style="green")
-    if args.project_type == "py":
-        pp = PyProject(llm=llm, args=args)
-    else:
-        pp = SuffixProject(llm=llm, args=args)
-    pp.run()
-    _sources = pp.sources
-    index_manager = IndexManager(args=args, source_codes=_sources, llm=llm)
-    index_manager.build_index()
-
-
-def index_export(export_path: str) -> bool:
-    try:
-        index_path = os.path.join(project_root, ".auto-coder", "index.json")
-        if not os.path.exists(index_path):
-            printer.print_text(Text(f"索引文件不存在. ", style="bold red"))
-            return False
-
-        with open(index_path, "r", encoding="utf-8") as f:
-            index_data = json.load(f)
-
-        converted_data = {}
-        for abs_path, data in index_data.items():
-            try:
-                rel_path = os.path.relpath(abs_path, project_root)
-                data["module_name"] = rel_path
-                converted_data[rel_path] = data
-            except ValueError:
-                printer.print_text(Text(f"索引转换路径失败. ", style="dim yellow"))
-                converted_data[abs_path] = data
-
-        export_file = os.path.join(export_path, "index.json")
-        with open(export_file, "w", encoding="utf-8") as f:
-            json.dump(converted_data, f, indent=2)
-        printer.print_text(Text(f"索引文件导出成功. ", style="bold green"))
-        return True
-    except Exception as err:
-        printer.print_text(Text(f"索引文件导出失败: {err}", style="bold red"))
-        return False
-
-
-def index_import(import_path: str):
-    try:
-        import_file = os.path.join(import_path, "index.json")
-        if not os.path.exists(import_file):
-            printer.print_text(Text(f"导入索引文件不存在", style="bold red"))
-            return False
-        with open(import_file, "r", encoding="utf-8") as f:
-            index_data = json.load(f)
-        converted_data = {}
-        for rel_path, data in index_data.items():
-            try:
-                abs_path = os.path.join(project_root, rel_path)
-                data["module_name"] = abs_path
-                converted_data[abs_path] = data
-            except Exception as err:
-                printer.print_text(Text(f"{rel_path} 索引转换路径失败: {err}", style="dim yellow"))
-                converted_data[rel_path] = data
-        # Backup existing index
-        index_path = os.path.join(project_root, ".auto-coder", "index.json")
-        if os.path.exists(index_path):
-            printer.print_text(Text(f"原索引文件不存在", style="bold yellow"))
-            backup_path = index_path + ".bak"
-            shutil.copy2(index_path, backup_path)
-
-        # Write new index
-        with open(index_path, "w", encoding="utf-8") as f:
-            json.dump(converted_data, f, indent=2)
-        return True
-    except Exception as err:
-        printer.print_text(Text(f"索引文件导入失败: {err}", style="bold red"))
-        return False
+    index_build(llm=llm, args=args, sources_codes=project_source(source_llm=llm, args=args))
+    return
 
 
 def index_query_command(query: str, llm: AutoLLM):
     args = get_final_config(query=query, delete_execute_file=True)
-
-    # args.query = query
-    if args.project_type == "py":
-        pp = PyProject(llm=llm, args=args)
-    else:
-        pp = SuffixProject(llm=llm, args=args)
-    pp.run()
-    _sources = pp.sources
-
-    final_files = []
-    index_manager = IndexManager(args=args, source_codes=_sources, llm=llm)
-    target_files = index_manager.get_target_files_by_query(query)
-
-    if target_files:
-        final_files.extend(target_files.file_list)
-
-    if target_files and args.index_filter_level >= 2:
-
-        related_fiels = index_manager.get_related_files([file.file_path for file in target_files.file_list])
-
-        if related_fiels is not None:
-            final_files.extend(related_fiels.file_list)
-
-    all_results = list({file.file_path: file for file in final_files}.values())
-    printer.print_key_value(
-        {"索引过滤级别": f"{args.index_filter_level}", "查询条件": f"{args.query}", "过滤后的文件数": f"{len(all_results)}"},
-        panel=True
-    )
-
-    printer.print_table_compact(
-        headers=["文件路径", "原因"],
-        data=[[_target_file.file_path, _target_file.reason] for _target_file in all_results],
-        title="Index Query 结果",
-        show_lines=True,
-    )
+    index_build_and_filter(llm=llm, args=args, sources_codes=project_source(source_llm=llm, args=args))
     return
 
 
@@ -539,10 +431,8 @@ def chat(query: str, llm: AutoLLM):
                 old_chat_history = json.load(f)
             if "conversation_history" not in old_chat_history:
                 old_chat_history["conversation_history"] = []
-            old_chat_history["conversation_history"].append(
-                old_chat_history.get("ask_conversation", []))
-            chat_history = {"ask_conversation": [
-            ], "conversation_history": old_chat_history["conversation_history"]}
+            old_chat_history["conversation_history"].append(old_chat_history.get("ask_conversation", []))
+            chat_history = {"ask_conversation": [], "conversation_history": old_chat_history["conversation_history"]}
         else:
             chat_history = {"ask_conversation": [],
                             "conversation_history": []}
@@ -569,8 +459,6 @@ def chat(query: str, llm: AutoLLM):
 
     if is_history:
         show_chat = []
-        # if "conversation_history" in chat_history:
-        #     show_chat.extend(chat_history["conversation_history"])
         if "ask_conversation" in chat_history:
             show_chat.extend(chat_history["ask_conversation"])
         print_chat_history(show_chat)
@@ -583,13 +471,15 @@ def chat(query: str, llm: AutoLLM):
     chat_llm = llm
     pre_conversations = []
 
-    if args.project_type == "py":
-        pp = PyProject(llm=llm, args=args)
-    else:
-        pp = SuffixProject(llm=llm, args=args)
-    pp.run()
-    _sources = pp.sources
-    s = build_index_and_filter_files(args=args, llm=llm, sources=_sources)
+    # if args.project_type == "py":
+    #     pp = PyProject(llm=llm, args=args)
+    # else:
+    #     pp = SuffixProject(llm=llm, args=args)
+    # pp.run()
+    # _sources = pp.sources
+    # _sources = project_source(source_llm=llm, args=args)
+    # s = build_index_and_filter_files(args=args, llm=llm, sources=_sources)
+    s = index_build_and_filter(llm=llm, args=args, sources_codes=project_source(source_llm=llm, args=args))
     if s:
         pre_conversations.append(
             {
@@ -606,51 +496,6 @@ def chat(query: str, llm: AutoLLM):
 
     assistant_response = stream_chat_display(chat_llm=llm, args=args, conversations=loaded_conversations)
 
-    # v = chat_llm.stream_chat_ai(conversations=loaded_conversations, model=args.chat_model)
-    #
-    # MAX_HISTORY_LINES = 15  # 最大保留历史行数
-    # lines_buffer = []
-    # current_line = ""
-    # assistant_response = ""
-    #
-    # try:
-    #     with Live(Panel("", title="Response", style="cyan"), refresh_per_second=12) as live:
-    #         for chunk in v:
-    #             if chunk.choices and chunk.choices[0].delta.content:
-    #                 content = chunk.choices[0].delta.content
-    #                 assistant_response += content
-    #
-    #                 # 处理换行符分割
-    #                 parts = (current_line + content).split('\n')
-    #
-    #                 # 最后一部分是未完成的新行
-    #                 if len(parts) > 1:
-    #                     # 将完整行加入缓冲区
-    #                     lines_buffer.extend(parts[:-1])
-    #                     # 保留最近N行历史
-    #                     if len(lines_buffer) > MAX_HISTORY_LINES:
-    #                         del lines_buffer[0: len(lines_buffer) - MAX_HISTORY_LINES]
-    #                 # 更新当前行（最后未完成的部分）
-    #                 current_line = parts[-1]
-    #                 # 构建显示内容 = 历史行 + 当前行
-    #                 display_content = '\n'.join(lines_buffer[-MAX_HISTORY_LINES:] + [current_line])
-    #
-    #                 live.update(
-    #                     Panel(Markdown(display_content), title="模型返回", border_style="cyan",
-    #                           height=min(25, live.console.height - 4))
-    #                 )
-    #
-    #         # 处理最后未换行的内容
-    #         if current_line:
-    #             lines_buffer.append(current_line)
-    #
-    #         # 最终完整渲染
-    #         live.update(
-    #             Panel(Markdown(assistant_response), title="模型返回", border_style="dim blue")
-    #         )
-    # except Exception as e:
-    #     printer.print_panel(Text(f"{str(e)}", style="red"), title="模型返回", center=True)
-
     chat_history["ask_conversation"].append({"role": "assistant", "content": assistant_response})
 
     with open(memory_file, "w") as fp:
@@ -772,14 +617,46 @@ def prepare_chat_yaml():
     return
 
 
-def coding(query: str, llm: AutoLLM):
+def get_conversation_history() -> str:
+    memory_dir = os.path.join(project_root, ".auto-coder", "memory")
+    os.makedirs(memory_dir, exist_ok=True)
+    memory_file = os.path.join(memory_dir, "chat_history.json")
+
+    def error_message():
+        printer.print_panel(Text("未找到可应用聊天记录.", style="yellow"), title="Chat History", center=True)
+
+    if not os.path.exists(memory_file):
+        error_message()
+        return ""
+
+    with open(memory_file, "r") as f:
+        chat_history = json.load(f)
+
+    if not chat_history["ask_conversation"]:
+        error_message()
+        return ""
+
+    conversations = chat_history["ask_conversation"]
+
+    context = ""
+    context += f"下面是我们的历史对话，参考我们的历史对话从而更好的理解需求和修改代码。\n\n<history>\n"
+    for conv in conversations:
+        if conv["role"] == "user":
+            context += f"用户: {conv['content']}\n"
+        elif conv["role"] == "assistant":
+            context += f"你: {conv['content']}\n"
+    context += "</history>\n"
+    return context
+
+
+def coding_command(query: str, llm: AutoLLM):
     is_apply = query.strip().startswith("/apply")
     if is_apply:
         query = query.replace("/apply", "", 1).strip()
+    is_rules = False
 
     memory["conversation"].append({"role": "user", "content": query})
     conf = memory.get("conf", {})
-    current_files = memory["current_files"]["files"]
 
     prepare_chat_yaml()  # 复制上一个序号的 yaml 文件, 生成一个新的聊天 yaml 文件
     latest_yaml_file = get_last_yaml_file(os.path.join(project_root, "actions"))
@@ -800,52 +677,14 @@ def coding(query: str, llm: AutoLLM):
             if converted_value is not None:
                 yaml_config[key] = converted_value
 
-        yaml_config["urls"] = current_files
+        yaml_config["urls"] = memory["current_files"]["files"]
         yaml_config["query"] = query
 
         if is_apply:
-            memory_dir = os.path.join(project_root, ".auto-coder", "memory")
-            os.makedirs(memory_dir, exist_ok=True)
-            memory_file = os.path.join(memory_dir, "chat_history.json")
-
-            def error_message():
-                printer.print_panel(
-                    Text("No chat history found to apply.", style="yellow"), title="Chat History", center=True
-                )
+            yaml_config["context"] += get_conversation_history()
 
-            if not os.path.exists(memory_file):
-                error_message()
-                return
-
-            with open(memory_file, "r") as f:
-                chat_history = json.load(f)
-
-            if not chat_history["ask_conversation"]:
-                error_message()
-                return
-            conversations = chat_history["ask_conversation"]
-
-            yaml_config["context"] += f"下面是我们的历史对话，参考我们的历史对话从而更好的理解需求和修改代码。\n\n<history>\n"
-            for conv in conversations:
-                if conv["role"] == "user":
-                    yaml_config["context"] += f"用户: {conv['content']}\n"
-                elif conv["role"] == "assistant":
-                    yaml_config["context"] += f"你: {conv['content']}\n"
-            yaml_config["context"] += "</history>\n"
-
-        # todo:暂时注释,后续通过一个 is_rules 的参数来控制
-        # if args.enable_rules:
-        #     rules_dir_path = os.path.join(project_root, ".auto-coder", "autocoderrules")
-        #     printer.print_text("已开启 Rules 模式", style="green")
-        #     yaml_config["context"] += f"下面是我们对代码进行深入分析,提取具有通用价值的功能模式和设计模式,可在其他需求中复用的Rules\n"
-        #     yaml_config["context"] += "你在编写代码时可以参考以下Rules\n"
-        #     yaml_config["context"] += "<rules>\n"
-        #     for rules_name in os.listdir(rules_dir_path):
-        #         printer.print_text(f"正在加载 Rules:{rules_name}", style="green")
-        #         rules_file_path = os.path.join(rules_dir_path, rules_name)
-        #         with open(rules_file_path, "r") as fp:
-        #             yaml_config["context"] += f"{fp.read()}\n"
-        #     yaml_config["context"] += "</rules>\n"
+        if is_rules:
+            yaml_config["context"] += get_rules_context()
 
         yaml_config["file"] = latest_yaml_file
         yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)
@@ -854,8 +693,7 @@ def coding(query: str, llm: AutoLLM):
             f.write(yaml_content)
         args = convert_yaml_to_config(execute_file)
 
-        dispacher = Dispacher(args=args, llm=llm)
-        dispacher.dispach()
+        coding(llm=llm, args=args)
     else:
         printer.print_text(f"创建新的 YAML 文件失败.", style="yellow")
 
@@ -1688,43 +1526,15 @@ def rules(query_args: List[str], llm: AutoLLM):
 
     if query_args[0] == "/commit":
         commit_id = query_args[1].strip()
-        auto_learn = AutoRulesLearn(llm=llm, args=args)
-
-        try:
-            result = auto_learn.analyze_commit_changes(commit_id=commit_id, conversations=[])
-            rules_file = os.path.join(rules_dir_path, f"rules-commit-{uuid.uuid4()}.md")
-            with open(rules_file, "w", encoding="utf-8") as f:
-                f.write(result)
-            printer.print_text(f"代码变更[{commit_id}]生成 Rules 成功", style="green")
-        except Exception as e:
-            printer.print_text(f"代码变更[{commit_id}]生成 Rules 失败: {e}", style="red")
+        rules_from_commit_changes(commit_id=commit_id, llm=llm, args=args)
 
     if query_args[0] == "/analyze":
-        auto_learn = AutoRulesLearn(llm=llm, args=args)
-
         files = memory.get("current_files", {}).get("files", [])
         if not files:
             printer.print_text("当前无活跃文件用于生成 Rules", style="yellow")
             return
 
-        sources = SourceCodeList([])
-        for file in files:
-            try:
-                with open(file, "r", encoding="utf-8") as f:
-                    source_code = f.read()
-                    sources.sources.append(SourceCode(module_name=file, source_code=source_code))
-            except Exception as e:
-                printer.print_text(f"读取文件生成 Rules 失败: {e}", style="yellow")
-                continue
-
-        try:
-            result = auto_learn.analyze_modules(sources=sources, conversations=[])
-            rules_file = os.path.join(rules_dir_path, f"rules-modules-{uuid.uuid4()}.md")
-            with open(rules_file, "w", encoding="utf-8") as f:
-                f.write(result)
-            printer.print_text(f"活跃文件[Files:{len(files)}]生成 Rules 成功", style="green")
-        except Exception as e:
-            printer.print_text(f"活跃文件生成 Rules 失败: {e}", style="red")
+        rules_from_active_files(files=files, llm=llm, args=args)
 
     completer.refresh_files()
 
@@ -1881,10 +1691,10 @@ def main():
                 index_query_command(query=query, llm=auto_llm)
             elif user_input.startswith("/index/export"):
                 export_path = user_input[len("/index/export"):].strip()
-                index_export(export_path)
+                index_export(project_root, export_path)
             elif user_input.startswith("/index/import"):
                 import_path = user_input[len("/index/import"):].strip()
-                index_import(import_path)
+                index_import(project_root, import_path)
             elif user_input.startswith("/list_files"):
                 list_files()
             elif user_input.startswith("/conf"):
@@ -1915,7 +1725,7 @@ def main():
                 if not query:
                     printer.print_text("Please enter your request.", style="yellow")
                     continue
-                coding(query=query, llm=auto_llm)
+                coding_command(query=query, llm=auto_llm)
             elif user_input.startswith("/auto"):
                 query = user_input[len("/auto"):].strip()
                 if not query:

autocoder_nano/edit/__init__.py

@@ -17,4 +17,12 @@ class Dispacher:
         ]
         for action in dispacher_actions:
             if action.run():
-                return
+                return
+
+
+def coding(llm: AutoLLM, args: AutoCoderArgs):
+    dispacher = Dispacher(args=args, llm=llm)
+    dispacher.dispach()
+
+
+__all__ = ["coding"]

autocoder_nano/index/__init__.py

@@ -1,4 +1,93 @@
+import json
+import os
+import shutil
+
 from autocoder_nano.index.entry import build_index_and_filter_files
+from autocoder_nano.index.index_manager import IndexManager
+from autocoder_nano.index.symbols_utils import extract_symbols
+from autocoder_nano.llm_client import AutoLLM
+from autocoder_nano.llm_types import AutoCoderArgs, SourceCode
+from autocoder_nano.project import project_source
+from autocoder_nano.utils.printer_utils import Printer
+
+
+printer = Printer()
+
+
+def index_build(llm: AutoLLM, args: AutoCoderArgs, sources_codes: list[SourceCode] = None):
+    if not sources_codes:
+        sources_codes = project_source(source_llm=llm, args=args)
+    index = IndexManager(args=args, source_codes=sources_codes, llm=llm)
+    index.build_index()
+
+
+def index_build_and_filter(llm: AutoLLM, args: AutoCoderArgs, sources_codes: list[SourceCode] = None) -> str:
+    if not sources_codes:
+        sources_codes = project_source(source_llm=llm, args=args)
+    return build_index_and_filter_files(args=args, llm=llm, sources=sources_codes)
+
+
+def index_export(project_root: str, export_path: str) -> bool:
+    try:
+        index_path = os.path.join(project_root, ".auto-coder", "index.json")
+        if not os.path.exists(index_path):
+            printer.print_text(f"索引文件不存在. ", style="red")
+            return False
+
+        with open(index_path, "r", encoding="utf-8") as f:
+            index_data = json.load(f)
+
+        converted_data = {}
+        for abs_path, data in index_data.items():
+            try:
+                rel_path = os.path.relpath(abs_path, project_root)
+                data["module_name"] = rel_path
+                converted_data[rel_path] = data
+            except ValueError:
+                printer.print_text(f"索引转换路径失败. ", style="yellow")
+                converted_data[abs_path] = data
+
+        export_file = os.path.join(export_path, "index.json")
+        with open(export_file, "w", encoding="utf-8") as f:
+            json.dump(converted_data, f, indent=2)
+        printer.print_text(f"索引文件导出成功.", style="green")
+        return True
+    except Exception as err:
+        printer.print_text(f"索引文件导出失败: {err}", style="red")
+        return False
+
+
+def index_import(project_root: str, import_path: str):
+    try:
+        import_file = os.path.join(import_path, "index.json")
+        if not os.path.exists(import_file):
+            printer.print_text(f"导入索引文件不存在. ", style="red")
+            return False
+        with open(import_file, "r", encoding="utf-8") as f:
+            index_data = json.load(f)
+        converted_data = {}
+        for rel_path, data in index_data.items():
+            try:
+                abs_path = os.path.join(project_root, rel_path)
+                data["module_name"] = abs_path
+                converted_data[abs_path] = data
+            except Exception as err:
+                printer.print_text(f"{rel_path} 索引转换路径失败: {err}", style="yellow")
+                converted_data[rel_path] = data
+        # Backup existing index
+        index_path = os.path.join(project_root, ".auto-coder", "index.json")
+        if os.path.exists(index_path):
+            printer.print_text(f"原索引文件不存在", style="yellow")
+            backup_path = index_path + ".bak"
+            shutil.copy2(index_path, backup_path)
+
+        # Write new index
+        with open(index_path, "w", encoding="utf-8") as f:
+            json.dump(converted_data, f, indent=2)
+        return True
+    except Exception as err:
+        printer.print_text(f"索引文件导入失败: {err}", style="red")
+        return False
 
 
-__all__ = ["build_index_and_filter_files"]
+__all__ = ["index_build", "index_export", "index_import", "index_build_and_filter", "extract_symbols", "IndexManager"]

autocoder_nano/project/__init__.py

@@ -1,5 +1,19 @@
+from autocoder_nano.llm_client import AutoLLM
+from autocoder_nano.llm_types import AutoCoderArgs, SourceCode
 from autocoder_nano.project.pyproject import PyProject
 from autocoder_nano.project.suffixproject import SuffixProject
 from autocoder_nano.project.tsproject import TSProject
 
-__all__ = ["PyProject", "SuffixProject", "TSProject"]
+
+def project_source(source_llm: AutoLLM, args: AutoCoderArgs) -> list[SourceCode]:
+    if args.project_type == "py":
+        pp = PyProject(llm=source_llm, args=args)
+    elif args.project_type == "ts":
+        pp = TSProject(llm=source_llm, args=args)
+    else:
+        pp = SuffixProject(llm=source_llm, args=args)
+    pp.run()
+    return pp.sources
+
+
+__all__ = ["PyProject", "SuffixProject", "TSProject", "project_source"]

autocoder_nano/rules/__init__.py

@@ -0,0 +1,67 @@
+import os
+import uuid
+
+from autocoder_nano.llm_client import AutoLLM
+from autocoder_nano.llm_types import AutoCoderArgs, SourceCodeList, SourceCode
+from autocoder_nano.rules.rules_learn import AutoRulesLearn
+from autocoder_nano.utils.printer_utils import Printer
+
+
+printer = Printer()
+
+
+def rules_from_commit_changes(commit_id: str, llm: AutoLLM, args: AutoCoderArgs):
+    rules_dir_path = os.path.join(args.source_dir, ".auto-coder", "autocoderrules")
+    auto_learn = AutoRulesLearn(llm=llm, args=args)
+
+    try:
+        result = auto_learn.analyze_commit_changes(commit_id=commit_id, conversations=[])
+        rules_file = os.path.join(rules_dir_path, f"rules-commit-{uuid.uuid4()}.md")
+        with open(rules_file, "w", encoding="utf-8") as f:
+            f.write(result)
+        printer.print_text(f"代码变更[{commit_id}]生成 Rules 成功", style="green")
+    except Exception as e:
+        printer.print_text(f"代码变更[{commit_id}]生成 Rules 失败: {e}", style="red")
+
+
+def rules_from_active_files(files: list[str], llm: AutoLLM, args: AutoCoderArgs):
+    rules_dir_path = os.path.join(args.source_dir, ".auto-coder", "autocoderrules")
+    auto_learn = AutoRulesLearn(llm=llm, args=args)
+
+    sources = SourceCodeList([])
+    for file in files:
+        try:
+            with open(file, "r", encoding="utf-8") as f:
+                source_code = f.read()
+                sources.sources.append(SourceCode(module_name=file, source_code=source_code))
+        except Exception as e:
+            printer.print_text(f"读取文件生成 Rules 失败: {e}", style="yellow")
+            continue
+
+    try:
+        result = auto_learn.analyze_modules(sources=sources, conversations=[])
+        rules_file = os.path.join(rules_dir_path, f"rules-modules-{uuid.uuid4()}.md")
+        with open(rules_file, "w", encoding="utf-8") as f:
+            f.write(result)
+        printer.print_text(f"活跃文件[Files:{len(files)}]生成 Rules 成功", style="green")
+    except Exception as e:
+        printer.print_text(f"活跃文件生成 Rules 失败: {e}", style="red")
+
+
+def get_rules_context(project_root):
+    rules_dir_path = os.path.join(project_root, ".auto-coder", "autocoderrules")
+    printer.print_text("已开启 Rules 模式", style="green")
+    context = ""
+    context += f"下面是我们对代码进行深入分析,提取具有通用价值的功能模式和设计模式,可在其他需求中复用的Rules\n"
+    context += "你在编写代码时可以参考以下Rules\n"
+    context += "<rules>\n"
+    for rules_name in os.listdir(rules_dir_path):
+        printer.print_text(f"正在加载 Rules:{rules_name}", style="green")
+        rules_file_path = os.path.join(rules_dir_path, rules_name)
+        with open(rules_file_path, "r") as fp:
+            context += f"{fp.read()}\n"
+    context += "</rules>\n"
+    return context
+
+
+__all__ = ["get_rules_context", "rules_from_commit_changes", "rules_from_active_files"]

autocoder_nano/version.py

@@ -1,3 +1,3 @@
-__version__ = "0.1.36"
+__version__ = "0.1.38"
 __author__ = "moofs"
 __license__ = "Apache License 2.0"

{autocoder_nano-0.1.36.dist-info → autocoder_nano-0.1.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: autocoder_nano
-Version: 0.1.36
+Version: 0.1.38
 Summary: AutoCoder Nano
 Author: moofs
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{autocoder_nano-0.1.36.dist-info → autocoder_nano-0.1.38.dist-info}/RECORD

@@ -1,5 +1,5 @@
 autocoder_nano/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder_nano/auto_coder_nano.py,sha256=75BNVyuktY2gdH8dXKBGMa-DVePWORf_5kgN5OvBODc,80571
+autocoder_nano/auto_coder_nano.py,sha256=rTcAq-GXnGxxmN5pIQfh7siYQAFjyIlZiPLjsaUXI9U,71685
 autocoder_nano/auto_coder_nano_rag.py,sha256=9BtNZ6nC5D5SPTIuziXZOfouCBLOMNzvJMTdDPQEgO8,10436
 autocoder_nano/auto_coder_nano_ui.py,sha256=ZBskcIJMeTJY7_JipGJaee58G9fUJaOv3LV4hptLc6c,12669
 autocoder_nano/file_utils.py,sha256=iGbkbQ191nKL4aNufdexYYYQSDM1XrDC9Uxp_PIbawY,661
@@ -10,7 +10,7 @@ autocoder_nano/llm_prompt.py,sha256=ViWUfCZp0gDESAAPHBhZc2WhHiFUHIxK6a2xbFu0sjU,
 autocoder_nano/llm_types.py,sha256=T0ugeWdwejy6BJaQrAlk8Pk5qweW2xbggxzHaSpTBOg,11588
 autocoder_nano/sys_utils.py,sha256=Sn6kr5diaEkVWbYDBrtenr9zw32jVIWvsAReY7_uEd0,1638
 autocoder_nano/templates.py,sha256=fqlRtnx6HvPE4CbdnPcnLBB-flPwufwcGRpsFD3aW2c,4271
-autocoder_nano/version.py,sha256=w6lOdNu-e-YHiAO2e5DN0oFTe4239b-ik57CnftvY_w,79
+autocoder_nano/version.py,sha256=n-gEsFbSQhkVooPpwNmF2t1Qt1jwBdkbwO_3nOzXhPA,79
 autocoder_nano/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder_nano/agent/agent_base.py,sha256=O5Hq6VnoqrXnBE_oXZHXlbmSRdOEe28H65bJ1WhAQjg,16377
 autocoder_nano/agent/agentic_edit.py,sha256=I1HjRhMabDmtfxcCOKawUJV0wU1GNzKtof19_GNgAjU,88749
@@ -50,18 +50,18 @@ autocoder_nano/app/templates/partials/input.html,sha256=8CY3JcHaA4nPZ2Vu4ragdYZz
 autocoder_nano/app/templates/partials/message.html,sha256=HWEh_j_yJAbP7zFs6jt88BDzkP7dG6VgPUbS2MT5Ax4,1548
 autocoder_nano/chat/__init__.py,sha256=FuXp0tcnegngct9Jp8HbgwFkwnhxMirwNFHtoa_vACw,2441
 autocoder_nano/data/tokenizer.json,sha256=7Lb5_DaYlDRvBRH0B0ynXO5c1fOwbQLxujX805-OEh0,7847602
-autocoder_nano/edit/__init__.py,sha256=QPMuW7tBTUe0Q00gUPJEmdxWqvunqko9_dsim0ncr7c,623
+autocoder_nano/edit/__init__.py,sha256=9_6am33w8U6nh1oMjWF-KmE1VmXe6UTTGd2CxSRaf5A,765
 autocoder_nano/edit/actions.py,sha256=N4qzSIE7Ifm7r6Sk-HbgWmbDqMP6jfrpByfpV7rbEo8,6480
 autocoder_nano/edit/text.py,sha256=nC7VkQYHCDdT3ULpy0DIjkFwB9suhjIxK39AEzXqbno,1150
 autocoder_nano/edit/code/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder_nano/edit/code/generate_editblock.py,sha256=s-VTZK0G1OhjEyZXqyjj4sY48fOo02EvHhaxTIw4ytY,13110
 autocoder_nano/edit/code/merge_editblock.py,sha256=Vk-FOVvaEzKcSRDMyMyR_M77kqj-s5-zejChn4QwLAY,17557
 autocoder_nano/edit/code/modification_ranker.py,sha256=hnF1acqAzPYKm9hEFxobJHfGGDdM-GclZLxvtt83lGA,3431
-autocoder_nano/index/__init__.py,sha256=r8HvwfyKJxnXMtAuxXIdbU0CWBGwJNlxIB4VZaUlBbo,112
+autocoder_nano/index/__init__.py,sha256=fYrXsjRMrL2cjHjH37Bcl51Uqr17_aqlTH-c_2WQcok,3767
 autocoder_nano/index/entry.py,sha256=S71dfnYC201eQLXwqNCo_Y83ImI1ZxuJ0_m2hz5nCJc,7729
 autocoder_nano/index/index_manager.py,sha256=ek7AqU8M-Snl5qZYhO_U0SEK3-y1u5OOxD9z-LdDesE,15619
 autocoder_nano/index/symbols_utils.py,sha256=z_16X6BozTfmric1uU-r2GqzDabJ5ChfAOB4lo7i-_8,1450
-autocoder_nano/project/__init__.py,sha256=8R90zhCcRTHWScAOYw20lkcHI4IhSm-ywCLcfezn0Oc,227
+autocoder_nano/project/__init__.py,sha256=KfSsvUVi-MCUJ-AITl8jQpOTPMIW55VmwPzSUP6LnlI,708
 autocoder_nano/project/pyproject.py,sha256=UZqHBrUmsCW73YkG8shjeFSEYGB_zFDH1ezoPP_f33Q,4478
 autocoder_nano/project/suffixproject.py,sha256=190GCS25qYi4kPav0Hpk2qgSFalkvJk7VM_pchnfurY,4717
 autocoder_nano/project/tsproject.py,sha256=3gBS-2aup2W5ehSbhD7Bdr-9v9uL7MgY_7TkLHShh9I,5565
@@ -79,7 +79,7 @@ autocoder_nano/rag/doc_storage.py,sha256=sEfX-2PsRTOk7MiwqtY3u0lzqMBwTsPXeIknkAY
 autocoder_nano/rag/long_context_rag.py,sha256=jpTfJg38pfVgL0FGFwe6nmBgQAMEAYtBw9ZLQQcFfpM,21862
 autocoder_nano/rag/token_counter.py,sha256=tHuht9huViEnpqiUVN_xUcgBOHJ0T-6dmk4IWzc_Cd0,1979
 autocoder_nano/research/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder_nano/rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autocoder_nano/rules/__init__.py,sha256=cNglrWEQ4d-xXhbuS6CCFY6DZHRRWANmUayCNnhYOJY,2941
 autocoder_nano/rules/rules_learn.py,sha256=P65_wwZ1sBxcQPTwdb7roqmkW_bj1SJQaZMLpUkglL4,7297
 autocoder_nano/ss/__init__.py,sha256=jp9Az7c0uafZcC6qfxjyZnSnVLtgA_4UdakSOcp8osE,34
 autocoder_nano/ss/search_engine.py,sha256=4_RcxF1almJX5XlLWB7d9UXM92YDK2bOqoCrkuGg5Mc,3720
@@ -91,9 +91,9 @@ autocoder_nano/utils/config_utils.py,sha256=r5n0De4mz5sL_nj-CeT_F5TxtgWQIN5vv0Z5
 autocoder_nano/utils/formatted_log_utils.py,sha256=1d3xvZ1Bo3-I1wQOMdXpwsMX5cl2FWkmpgHGHvTPEvI,5457
 autocoder_nano/utils/printer_utils.py,sha256=6rGHihCh8DDESWs6qWqwsf3B6qaeM_CNx6crzkl9UCk,15303
 autocoder_nano/utils/shell_utils.py,sha256=llVTrOrmS1RH2ws7W69tofVtf53Kq04uh-sURphejrU,2477
-autocoder_nano-0.1.36.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-autocoder_nano-0.1.36.dist-info/METADATA,sha256=7-k89fLfFGXd67c6xAaIvmZV2Q1vix4B5YecTSn6Qcw,13591
-autocoder_nano-0.1.36.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-autocoder_nano-0.1.36.dist-info/entry_points.txt,sha256=Dj8gGZ_AgLy8ANqr2do_DJjpsR3JMh-ztsrUXo4Vn5Q,194
-autocoder_nano-0.1.36.dist-info/top_level.txt,sha256=D7s34cwIs1F4EAjRRDvO_zTHtUz1Z7UVccFUNlJn7HI,15
-autocoder_nano-0.1.36.dist-info/RECORD,,
+autocoder_nano-0.1.38.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+autocoder_nano-0.1.38.dist-info/METADATA,sha256=TMDwPdAc0EKiR8VAMbV6z8Xjs6yBnuGpzHR9Q09AmeQ,13591
+autocoder_nano-0.1.38.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+autocoder_nano-0.1.38.dist-info/entry_points.txt,sha256=Dj8gGZ_AgLy8ANqr2do_DJjpsR3JMh-ztsrUXo4Vn5Q,194
+autocoder_nano-0.1.38.dist-info/top_level.txt,sha256=D7s34cwIs1F4EAjRRDvO_zTHtUz1Z7UVccFUNlJn7HI,15
+autocoder_nano-0.1.38.dist-info/RECORD,,