auto-coder 0.1.300__tar.gz → 0.1.301__tar.gz

{auto_coder-0.1.300 → auto_coder-0.1.301}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.300
+Version: 0.1.301
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.300
+Version: 0.1.301
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/commands/auto_command.py

@@ -25,7 +25,7 @@ from autocoder.common.auto_configure import config_readme
 from autocoder.utils.auto_project_type import ProjectTypeAnalyzer
 from rich.text import Text
 from autocoder.common.mcp_server import get_mcp_server,McpServerInfoRequest
-
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
 class CommandMessage(BaseModel):
     role: str
     content: str
@@ -41,14 +41,17 @@ class CommandConversation(BaseModel):
     current_conversation: List[CommandMessage]
 
 
-def load_memory_file() -> CommandConversation:
-    """Load command conversations from memory file"""
+def load_memory_file(args: AutoCoderArgs) -> CommandConversation:
+    """Load command conversations from memory file"""    
+    
     memory_dir = os.path.join(".auto-coder", "memory")
     file_path = os.path.join(memory_dir, "command_chat_history.json")
     if os.path.exists(file_path):
         with open(file_path, "r", encoding="utf-8") as f:
             try:
-                return CommandConversation.model_validate_json(f.read())
+                conversation = CommandConversation.model_validate_json(f.read())
+                conversation.current_conversation = conversation.current_conversation
+                return conversation
             except Exception:
                 return CommandConversation(history={}, current_conversation=[])
     return CommandConversation(history={}, current_conversation=[])
@@ -174,8 +177,8 @@ class CommandAutoTuner:
     
     def get_conversations(self) -> List[CommandMessage]:
         """Get conversation history from memory file"""
-        conversation = load_memory_file()
-        return [extended_msg for extended_msg in conversation.current_conversation]
+        conversation = load_memory_file(args=self.args)
+        return [command_message.message for command_message in conversation.current_conversation]
 
     @byzerllm.prompt()
     def _analyze(self, request: AutoCommandRequest) -> str:
@@ -284,7 +287,7 @@ class CommandAutoTuner:
             "command_combination_readme": self._command_combination_readme.prompt()
         }
     
-
+    @byzerllm.prompt()
     def _command_combination_readme(self) -> str:
         """
         <function_combination_readme>
@@ -350,8 +353,16 @@ class CommandAutoTuner:
         # 获取 prompt 内容
         prompt = self._analyze.prompt(request)
         
+        # 获取对当前项目变更的最近8条历史人物
+        action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
+        history_tasks = action_yml_file_manager.to_tasks_prompt(limit=8)
+        new_messages = []
+        if self.args.enable_task_history:
+            new_messages.append({"role": "user", "content": history_tasks})
+            new_messages.append({"role": "assistant", "content": "好的，我知道最近的任务对项目的变更了，我会参考这些来更好的理解你的需求。"})
+        
         # 构造对话上下文
-        conversations = [{"role": "user", "content": prompt}]
+        conversations = new_messages + [{"role": "user", "content": prompt}]
         
         # 使用 stream_out 进行输出
         printer = Printer()

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/common/__init__.py

@@ -406,6 +406,10 @@ class AutoCoderArgs(pydantic.BaseModel):
 
     dynamic_urls: List[str] = []
 
+    add_updated_urls: List[str] = []
+
+    enable_task_history: Optional[bool] = False
+
     class Config:
         protected_namespaces = ()
 

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/common/action_yml_file_manager.py

@@ -5,7 +5,7 @@ import git
 from typing import List, Dict, Tuple, Optional, Union, Any
 from loguru import logger
 from autocoder.common.printer import Printer
-
+import byzerllm
 
 class ActionYmlFileManager:
     """
@@ -19,15 +19,15 @@ class ActionYmlFileManager:
     - 处理 commit 消息与 YAML 文件的关联
     """
     
-    def __init__(self, source_dir: str):
+    def __init__(self, source_dir: Optional[str] = None):
         """
         初始化 ActionYmlFileManager
         
         Args:
             source_dir: 项目根目录
         """
-        self.source_dir = source_dir
-        self.actions_dir = os.path.join(source_dir, "actions")
+        self.source_dir = source_dir or os.getcwd()
+        self.actions_dir = os.path.join(self.source_dir, "actions")
         self.printer = Printer()
         
     def ensure_actions_dir(self) -> bool:
@@ -364,4 +364,69 @@ class ActionYmlFileManager:
                 yaml_content['file_name'] = file_name
                 history_tasks.append(yaml_content)
         
-        return history_tasks 
+        return history_tasks 
+    
+    @byzerllm.prompt()
+    def _to_tasks_prompt(self, history_tasks: List[Dict]) -> str:
+        """       
+        <history_tasks>                
+        最近的任务历史记录，从最新到最旧排序：
+        
+        {% for task in history_tasks %}
+        ## 任务 {{ loop.index }}: {{ task.file_name }}
+        
+        {% if task.query %}
+        **用户需求**: {{ task.query }}
+        {% endif %}
+        
+        {% if task.urls %}
+        **用户提供的相关文件**:
+        {% if task.urls is string %}
+        - {{ task.urls }}
+        {% else %}
+        {% for url in task.urls %}
+        - {{ url }}
+        {% endfor %}
+        {% endif %}
+        {% endif %}
+        
+        {% if task.dynamic_urls %}
+        **系统提取的相关文件**:
+        {% if task.dynamic_urls is string %}
+        - {{ task.dynamic_urls }}
+        {% else %}
+        {% for url in task.dynamic_urls %}
+        - {{ url }}
+        {% endfor %}
+        {% endif %}
+        {% endif %}
+        
+        {% if task.add_updated_urls %}
+        **变更的文件**:
+        {% if task.add_updated_urls is string %}
+        - {{ task.add_updated_urls }}
+        {% else %}
+        {% for url in task.add_updated_urls %}
+        - {{ url }}
+        {% endfor %}
+        {% endif %}
+        {% endif %}
+        
+        {% if task.how_to_reproduce %}
+        **变更过程**:
+        ```
+        {{ task.how_to_reproduce }}
+        ```
+        {% endif %}
+        
+        {% if not loop.last %}
+        ---
+        {% endif %}
+        {% endfor %}
+        </history_tasks>
+        请注意上述历史任务记录，以便更好地理解当前用户需求的上下文和连续性。
+        """
+    
+    def to_tasks_prompt(self, limit: int = 5) -> str:
+        history_tasks = self.parse_history_tasks(limit)
+        return self._to_tasks_prompt.prompt(history_tasks)

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/common/code_auto_merge.py

@@ -10,6 +10,7 @@ import hashlib
 from autocoder.common import files as FileUtils
 from autocoder.common.printer import Printer
 from autocoder.common.auto_coder_lang import get_message
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
 
 class PathAndCode(pydantic.BaseModel):
     path: str
@@ -197,4 +198,17 @@ class CodeAutoMerge:
         self.printer.print_in_terminal("files_merged", total=total)
         if not force_skip_git and not self.args.skip_commit:
             commit_result = git_utils.commit_changes(self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}_{md5}")
+            
+            action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
+            action_file_name = os.path.basename(self.args.file)
+            add_updated_urls = []
+            commit_result.changed_files
+            for file in commit_result.changed_files:
+                add_updated_urls.append(os.path.join(self.args.source_dir, file))
+
+            self.args.add_updated_urls = add_updated_urls
+            update_yaml_success = action_yml_file_manager.update_yaml_field(action_file_name, "add_updated_urls", add_updated_urls)
+            if not update_yaml_success:                        
+                self.printer.print_in_terminal("yaml_save_error", style="red", yaml_file=action_file_name)  
+            
             git_utils.print_commit_info(commit_result=commit_result)

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/common/code_auto_merge_diff.py

@@ -4,6 +4,7 @@ from autocoder.common import AutoCoderArgs,git_utils
 from typing import List,Union,Tuple
 import pydantic
 import byzerllm
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
 from autocoder.common.printer import Printer
 import hashlib
 from pathlib import Path
@@ -580,6 +581,19 @@ class CodeAutoMergeDiff:
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git and not self.args.skip_commit:
             commit_result = git_utils.commit_changes(self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}_{md5}")
+
+            action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
+            action_file_name = os.path.basename(self.args.file)
+            add_updated_urls = []
+            commit_result.changed_files
+            for file in commit_result.changed_files:
+                add_updated_urls.append(os.path.join(self.args.source_dir, file))
+
+            self.args.add_updated_urls = add_updated_urls
+            update_yaml_success = action_yml_file_manager.update_yaml_field(action_file_name, "add_updated_urls", add_updated_urls)
+            if not update_yaml_success:                        
+                self.printer.print_in_terminal("yaml_save_error", style="red", yaml_file=action_file_name)    
+                          
             git_utils.print_commit_info(commit_result=commit_result)
         else:
             # Print edits for review

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/common/code_auto_merge_editblock.py

@@ -1,6 +1,7 @@
 import os
 from byzerllm.utils.client import code_utils
 from autocoder.common import AutoCoderArgs, git_utils
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
 from autocoder.common.text import TextSimilarity
 from autocoder.utils.queue_communicate import (
     queue_communicate,
@@ -428,6 +429,19 @@ class CodeAutoMergeEditBlock:
                         self.args.source_dir,
                         f"{self.args.query}\nauto_coder_{file_name}_{md5}",
                     )
+                    
+                    action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
+                    action_file_name = os.path.basename(self.args.file)
+                    add_updated_urls = []
+                    commit_result.changed_files
+                    for file in commit_result.changed_files:
+                        add_updated_urls.append(os.path.join(self.args.source_dir, file))
+
+                    self.args.add_updated_urls = add_updated_urls
+                    update_yaml_success = action_yml_file_manager.update_yaml_field(action_file_name, "add_updated_urls", add_updated_urls)
+                    if not update_yaml_success:                        
+                        self.printer.print_in_terminal("yaml_save_error", style="red", yaml_file=action_file_name)  
+
                     git_utils.print_commit_info(commit_result=commit_result)
                 except Exception as e:
                     self.printer.print_str_in_terminal(

{auto_coder-0.1.300 → auto_coder-0.1.301}/src/autocoder/common/code_auto_merge_strict_diff.py

@@ -5,6 +5,7 @@ from autocoder.common import AutoCoderArgs, git_utils
 from typing import List,Tuple
 import pydantic
 import byzerllm
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
 from autocoder.common.printer import Printer
 import hashlib
 from pathlib import Path
@@ -288,6 +289,18 @@ class CodeAutoMergeStrictDiff:
             commit_result = git_utils.commit_changes(
                 self.args.source_dir, f"{self.args.query}\nauto_coder_{file_name}_{md5}"
             )
+            action_yml_file_manager = ActionYmlFileManager(self.args.source_dir)
+            action_file_name = os.path.basename(self.args.file)
+            add_updated_urls = []
+            commit_result.changed_files
+            for file in commit_result.changed_files:
+                add_updated_urls.append(os.path.join(self.args.source_dir, file))
+
+            self.args.add_updated_urls = add_updated_urls
+            update_yaml_success = action_yml_file_manager.update_yaml_field(action_file_name, "add_updated_urls", add_updated_urls)
+            if not update_yaml_success:                        
+                self.printer.print_in_terminal("yaml_save_error", style="red", yaml_file=action_file_name)  
+            
             git_utils.print_commit_info(commit_result=commit_result)
         else:
             # Print diff blocks for review

auto_coder-0.1.301/src/autocoder/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.301"

auto_coder-0.1.300/src/autocoder/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.300"