auto-coder 0.1.305__py3-none-any.whl → 0.1.307__py3-none-any.whl

autocoder/commands/auto_command.py

@@ -213,6 +213,8 @@ class CommandAutoTuner:
     @byzerllm.prompt()
     def _analyze(self, request: AutoCommandRequest) -> str:
         """
+        你是 auto-coder.chat 软件，帮助用户完成编程方面的需求。我们的目标是根据用户输入和当前上下文，组合多个函数来完成用户的需求。
+
         ## 当前用户环境信息如下:
         <os_info>
         操作系统: {{ env_info.os_name }} {{ env_info.os_version }}
@@ -234,10 +236,11 @@ class CommandAutoTuner:
         {%- endif %}   
         </os_info>
 
-        我们的目标是根据用户输入和当前上下文，组合多个函数来完成用户的需求。
-
+        当前项目根目录：
+        {{ current_project }}
+    
         {% if current_files %}
-        ## 当前活跃区文件列表：
+        ## 当前用户手动添加关注的文件列表：
         <current_files>
         {% for file in current_files %}
         - {{ file }}
@@ -246,7 +249,7 @@ class CommandAutoTuner:
         {% endif %}
 
 
-        ## 当前用户的配置选项如下:
+        ## 这是用户对你的配置        
         <current_conf>
         {{ current_conf }}
         </current_conf>
@@ -260,6 +263,24 @@ class CommandAutoTuner:
         ## 函数组合说明：        
         {{ command_combination_readme }}
 
+        ## active-context 项目追踪文档系统
+        
+        在 {{ current_project }}/.auto-coder/active-context 下,我们提供了对该项目每个文件目录的追踪。
+        具体逻辑为：假设我们在当前项目有 ./src/package1/py1.py, 那么相应的在 .auto-coder/active-context 会有一个 ./src/package1 目录,
+        该目录下可能会有一个 active-context.md 文件，该文件记录了该目录下所有文件相关信息，可以帮你更好的理解这个目录下的文档，你可以通过 read_files 函数来读取
+        这个文件。注意，这个文件不一定存在。如果读取失败也是正常的。
+
+        ## 变更记录文档系统
+
+        在 {{ current_project }}/actions 目录下，会有格式类似 000000001201_chat_action.yml 的文件，该文件记录了最近10次对话，
+        你可以通过 read_files 函数来读取这些文件，从而更好的理解用户的需求。
+
+        下面是一些字段的简单介绍
+        - query: 用户需求
+        - urls： 用户提供的上下文文件列表
+        - dynamic_urls： auto-coder.chat 自动感知的一些文件列表
+        - add_updated_urls: 这次需求发生变更的文件列表        
+        
         {% if conversation_history %}
         ## 历史对话:
         <conversation_history>
@@ -314,7 +335,8 @@ class CommandAutoTuner:
             "conversation_safe_zone_tokens": self.args.conversation_prune_safe_zone_tokens,
             "os_distribution": shells.get_os_distribution(),
             "current_user": shells.get_current_username(),
-            "command_combination_readme": self._command_combination_readme.prompt()
+            "command_combination_readme": self._command_combination_readme.prompt(),
+            "current_project": os.path.abspath(self.args.source_dir)
         }
 
     @byzerllm.prompt()
@@ -337,8 +359,10 @@ class CommandAutoTuner:
         尽可通过了解项目后，多用 @文件和@@符号，这样 chat 函数可以更清晰的理解你关注的代码，文档和意图。
 
         ### 调用 coding 函数应该注意的事项
-        调用 coding 函数的时候，尽可能多的 @文件和@@符号，让需求更加清晰明了，建议多描述具体怎么完成对应的需求。
-        对于代码需求设计，尽可能使用 chat 函数。如果成功执行了 coding 函数， 最好再调用一次 chat("/review /commit")，方便总结这次代码变更。
+        调用 coding 函数的之前，你需要尽可能先了解用户需求，了解项目状态，包括读取 active_context 文件来了解项目。
+        然后清晰的描述自己的需求，完整的实现步骤，以及尽可能对@文件和@@符号需要参考以及修改的文件和符号。
+        对于比较复杂的需求，你还可以使用 chat 函数来进行讨论，从而获取一些有用的信息。
+        如果成功执行了 coding 函数， 最好再调用一次 chat("/review /commit")，方便总结这次代码变更。
         注意，review 完后，需要询问用户是否要做啥调整不，如果用户说不用，那么就停止。否则根据意图进行后续操作。
 
         ### 关于对话大小的问题

autocoder/common/__init__.py

@@ -412,6 +412,12 @@ class AutoCoderArgs(pydantic.BaseModel):
 
     event_file: Optional[str] = None
 
+    enable_active_context: Optional[bool] = False
+
+    generate_max_rounds: Optional[int] = 5
+
+    revert_commit_id: Optional[str] = None
+
     class Config:
         protected_namespaces = ()
 

autocoder/common/action_yml_file_manager.py

@@ -4,8 +4,10 @@ import hashlib
 import git
 from typing import List, Dict, Tuple, Optional, Union, Any
 from loguru import logger
+from autocoder.common.git_utils import get_repo
 from autocoder.common.printer import Printer
 import byzerllm
+from autocoder.common import git_utils
 
 class ActionYmlFileManager:
     """
@@ -44,7 +46,7 @@ class ActionYmlFileManager:
             except Exception as e:
                 logger.error(f"Failed to create actions directory: {e}")
                 return False
-        return True
+        return True    
     
     def get_action_files(self, filter_prefix: Optional[str] = None) -> List[str]:
         """
@@ -183,16 +185,21 @@ class ActionYmlFileManager:
         Returns:
             Dict: YAML 内容，如果加载失败返回空字典
         """
-        yaml_path = os.path.join(self.actions_dir, file_name)
-        
+        yaml_path = os.path.join(self.actions_dir, file_name)        
         try:
             with open(yaml_path, 'r', encoding='utf-8') as f:
                 content = yaml.safe_load(f) or {}
             return content
-        except Exception as e:
+        except Exception as e:            
             self.printer.print_in_terminal("yaml_load_error", style="red", 
                                           yaml_file=yaml_path, error=str(e))
             return {}
+        
+    def get_full_path_by_file_name(self, file_name: str) -> str:
+        """
+        根据文件名获取完整路径
+        """
+        return os.path.join(self.actions_dir, file_name)
     
     def save_yaml_content(self, file_name: str, content: Dict) -> bool:
         """
@@ -232,6 +239,26 @@ class ActionYmlFileManager:
         yaml_content = self.load_yaml_content(file_name)
         yaml_content[field] = value
         return self.save_yaml_content(file_name, yaml_content)
+
+    def get_all_commit_id_from_file(self,file_name:str):
+        '''
+        会包含 revert 信息
+        '''
+        repo = get_repo(self.source_dir)
+        if repo is None:
+            logger.error("Repository is not initialized.")
+            return []
+        
+        commit_hashes = []        
+        for commit in repo.iter_commits():
+            lines = commit.message.strip().split('\n')
+            last_line = lines[-1]            
+            if file_name in last_line or (commit.message.startswith("<revert>") and file_name in commit.message):
+                commit_hash = commit.hexsha
+                commit_hashes.append(commit_hash)
+                
+        return commit_hashes
+
     
     def get_commit_id_from_file(self, file_name: str) -> Optional[str]:
         """
@@ -243,16 +270,19 @@ class ActionYmlFileManager:
         Returns:
             Optional[str]: commit ID，如果计算失败返回 None
         """
-        yaml_path = os.path.join(self.actions_dir, file_name)
-        
-        try:
-            with open(yaml_path, 'r', encoding='utf-8') as f:
-                yaml_content = f.read()
-            file_md5 = hashlib.md5(yaml_content.encode("utf-8")).hexdigest()
-            return f"auto_coder_{file_name}"
-        except Exception as e:
-            logger.error(f"Failed to calculate commit ID: {e}")
+        repo = get_repo(self.source_dir)
+        if repo is None:
+            logger.error("Repository is not initialized.")
             return None
+        
+        commit_hash = None
+        # 这里遍历从最新的commit 开始遍历
+        for commit in repo.iter_commits():
+            last_line = commit.message.strip().split('\n')[-1]
+            if file_name in last_line and not commit.message.startswith("<revert>"):
+                commit_hash = commit.hexsha
+                break
+        return commit_hash
     
     def get_file_name_from_commit_id(self, commit_id: str) -> Optional[str]:
         """
@@ -298,7 +328,7 @@ class ActionYmlFileManager:
             return self.get_file_name_from_commit_id(last_line)
         except Exception as e:
             logger.error(f"Failed to extract file name from commit message: {e}")
-            return None
+            return None            
     
     def get_commit_changes(self, file_name: Optional[str] = None) -> List[Tuple[str, List[str], Dict[str, Tuple[str, str]]]]:
         """
@@ -327,31 +357,30 @@ class ActionYmlFileManager:
         changes = {}
         try:
             repo = git.Repo(self.source_dir)
-            for commit in repo.iter_commits():
-                if commit_id in commit.message:
-                    if commit.parents:
-                        parent = commit.parents[0]
-                        # 获取所有文件的前后内容
-                        for diff_item in parent.diff(commit):
-                            file_path = diff_item.a_path if diff_item.a_path else diff_item.b_path
-                            
-                            # 获取变更前内容
-                            before_content = None
-                            try:
-                                if diff_item.a_blob:
-                                    before_content = repo.git.show(f"{parent.hexsha}:{file_path}")
-                            except git.exc.GitCommandError:
-                                pass  # 文件可能是新增的
+            commit =repo.commit(commit_id)            
+            if commit.parents:
+                parent = commit.parents[0]
+                # 获取所有文件的前后内容
+                for diff_item in parent.diff(commit):
+                    file_path = diff_item.a_path if diff_item.a_path else diff_item.b_path
+                    
+                    # 获取变更前内容
+                    before_content = None
+                    try:
+                        if diff_item.a_blob:
+                            before_content = repo.git.show(f"{parent.hexsha}:{file_path}")
+                    except git.exc.GitCommandError:
+                        pass  # 文件可能是新增的
 
-                            # 获取变更后内容
-                            after_content = None
-                            try:
-                                if diff_item.b_blob:
-                                    after_content = repo.git.show(f"{commit.hexsha}:{file_path}")
-                            except git.exc.GitCommandError:
-                                pass  # 文件可能被删除
+                    # 获取变更后内容
+                    after_content = None
+                    try:
+                        if diff_item.b_blob:
+                            after_content = repo.git.show(f"{commit.hexsha}:{file_path}")
+                    except git.exc.GitCommandError:
+                        pass  # 文件可能被删除
 
-                            changes[file_path] = (before_content, after_content)
+                    changes[file_path] = (before_content, after_content)
                     break
         except git.exc.GitCommandError as e:
             self.printer.print_in_terminal("git_command_error", style="red", error=str(e))
@@ -360,6 +389,15 @@ class ActionYmlFileManager:
         
         return [(query, urls, changes)]
     
+    def revert_file(self, file_name: str) -> bool:
+        revert_result = git_utils.revert_changes(
+            self.source_dir, f"auto_coder_{file_name}"
+        )
+        if revert_result:
+            self.update_yaml_field(file_name, "revert_commit_id", revert_result.get("new_commit_hash"))
+            return True
+        return False
+    
     def parse_history_tasks(self, limit: int = 5) -> List[Dict]:
         """
         解析历史任务信息