auto-coder 0.1.369__tar.gz → 0.1.370__tar.gz

{auto_coder-0.1.369 → auto_coder-0.1.370}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.369
+Version: 0.1.370
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.10

{auto_coder-0.1.369 → auto_coder-0.1.370}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.369
+Version: 0.1.370
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.10

auto_coder-0.1.370/src/autocoder/agent/auto_learn.py

@@ -0,0 +1,375 @@
+from typing import Generator, List, Dict, Union, Tuple, Optional
+import os
+import byzerllm
+import pydantic
+import git
+from rich.console import Console
+from autocoder.common.printer import Printer
+from autocoder.common import AutoCoderArgs
+from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
+from autocoder.common import SourceCode, SourceCodeList
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
+
+
+class AutoLearn:
+    def __init__(self, llm: Union[byzerllm.ByzerLLM, byzerllm.SimpleByzerLLM],
+                 args: AutoCoderArgs,
+                 console: Optional[Console] = None):
+        """
+        初始化 AutoLearn
+
+        Args:
+            llm: ByzerLLM 实例，用于代码分析和学习
+            args: AutoCoderArgs 实例，包含配置信息
+            console: Rich Console 实例，用于输出
+        """
+        self.llm = llm
+        self.args = args
+        self.console = console or Console()
+        self.printer = Printer()
+
+    @byzerllm.prompt()
+    def analyze_commit(self,
+                       querie_with_urls_and_changes: List[Tuple[str, List[str], Dict[str, Tuple[str, str]]]],                       
+                       new_query:str,            
+                       ) -> str:
+        """        
+        下面是用户一次提交的代码变更：
+        <changes>
+        {% for query,urls,changes in querie_with_urls_and_changes %}
+        ## 原始的任务需求
+        {{ query }}
+
+        修改的文件:
+        {% for url in urls %}
+        - {{ url }}
+        {% endfor %}
+
+        代码变更:
+        {% for file_path, (before, after) in changes.items() %}
+        ##File: {{ file_path }}
+        ##修改前:
+
+        {{ before or "New file" }}        
+
+        ##File: {{ file_path }}
+        ##修改后:
+
+        {{ after or "File deleted" }}
+
+        {% endfor %}
+        {% endfor %}
+        </changes>
+        
+
+        请对根据上面的代码变更进行深入分析，提取具有通用价值的功能模式和设计模式，转化为可在其他项目中复用的代码规则（rules）。
+        
+        - 识别代码变更中具有普遍应用价值的功能点和模式
+        - 将这些功能点提炼为结构化规则，便于在其他项目中快速复用
+        - 生成清晰的使用示例，包含完整依赖和调用方式
+        - 这些规则将被存储在项目的 {{ rules_path }} 目录，供后续自动化代码生成使用   
+        - 更新 {{rules_index_path}} 文件,确保改文件有新增文件的描述信息
+        
+        当前 {{rules_index_path}} 文件内容:
+        {% if index_file_content %}                                
+        {{ index_file_content }}      
+        {% else %}
+        <为空>
+        {% endif %}
+        - {{ new_query }} 
+                                
+
+        最后，新生成的文件格式要是这种形态的：
+    
+       <example_rules>
+        ---
+        description: [简明描述规则的功能，20字以内]
+        globs: [匹配应用此规则的文件路径，如"src/services/*.py"]
+        alwaysApply: [是否总是应用，通常为false]
+        ---
+
+        # [规则主标题]
+
+        ## 简要说明
+        [该规则的功能、适用场景和价值，100字以内]
+
+        ## 典型用法
+        ```python
+        # 完整的代码示例，包含:
+        # 1. 必要的import语句
+        # 2. 类/函数定义
+        # 3. 参数说明
+        # 4. 调用方式
+        # 5. 关键注释        
+        ```
+
+        ## 依赖说明
+        - [必要的依赖库及版本]
+        - [环境要求]
+        - [初始化流程(如有)]
+
+        ## 学习来源
+        [从哪个提交变更的哪部分代码中提取的该功能点]    
+        </example_rules>                
+        """
+        return {
+            "project_root": os.path.abspath(self.args.source_dir),
+            "index_file_content": self._get_index_file_content(),
+            "rules_path": os.path.join(os.path.abspath(self.args.source_dir), ".autocoderrules"),
+            "rules_index_path": os.path.join(os.path.abspath(self.args.source_dir), ".autocoderrules", "index.md")
+        }
+
+    @byzerllm.prompt()
+    def analyze_modules(self, sources: SourceCodeList, query: str) -> str:
+        """
+        下面是用户提供的需要抽取规则的代码：
+        <files>
+        {% for source in sources.sources %}
+        ##File: {{ source.module_name }}        
+        {{ source.source_code }}        
+        {% endfor %}
+        </files>
+
+        请对对上面的代码进行深入分析，提取具有通用价值的功能模式和设计模式，转化为可在其他项目中复用的代码规则（rules）。
+        
+        - 识别代码变更中具有普遍应用价值的功能点和模式
+        - 将这些功能点提炼为结构化规则，便于在其他项目中快速复用
+        - 生成清晰的使用示例，包含完整依赖和调用方式
+        - 这些规则将被存储在项目的 {{ rules_path }} 目录，供后续自动化代码生成使用   
+        - 更新 {{rules_index_path}} 文件,确保改文件有新增文件的描述信息
+        
+        当前 {{rules_index_path}} 文件内容:
+        {% if index_file_content %}                                
+        {{ index_file_content }}      
+        {% else %}
+        <为空>
+        {% endif %}
+        - {{ query }} 
+                                
+        最后，新生成的文件格式要是这种形态的：
+    
+        <example_rules>
+        ---
+        description: [简明描述规则的功能，20字以内]
+        globs: [匹配应用此规则的文件路径，如"src/services/*.py"]
+        alwaysApply: [是否总是应用，通常为false]
+        ---
+
+        # [规则主标题]
+
+        ## 简要说明
+        [该规则的功能、适用场景和价值，100字以内]
+
+        ## 典型用法
+        ```python
+        # 完整的代码示例，包含:
+        # 1. 必要的import语句
+        # 2. 类/函数定义
+        # 3. 参数说明
+        # 4. 调用方式
+        # 5. 关键注释        
+        ```
+
+        ## 依赖说明
+        - [必要的依赖库及版本]
+        - [环境要求]
+        - [初始化流程(如有)]
+
+        ## 学习来源
+        [从哪个提交变更的哪部分代码中提取的该功能点]    
+        </example_rules>   
+        """
+
+        # 获取索引文件内容
+        index_file_content = self._get_index_file_content()
+
+        return {
+            "project_root": os.path.abspath(self.args.source_dir),
+            "index_file_content": index_file_content,
+            "rules_path": os.path.join(os.path.abspath(self.args.source_dir), ".autocoderrules"),
+            "rules_index_path": os.path.join(os.path.abspath(self.args.source_dir), ".autocoderrules", "index.md")
+        }
+
+    def get_commit_changes(self, commit_id: str) -> Tuple[List[Tuple[str, List[str], Dict[str, Tuple[str, str]]]], Optional[str]]:
+        """
+        直接从Git仓库获取指定commit的变更
+
+        Args:
+            commit_id: Git commit的ID
+
+        Returns:
+            List[Tuple[str, List[str], Dict[str, Tuple[str, str]]]]: 包含查询、URL和变更信息的列表
+        """
+        printer = Printer()
+        querie_with_urls_and_changes = []
+        try:
+            repo = git.Repo(self.args.source_dir)
+            commit = repo.commit(commit_id)
+            modified_files = []
+            changes = {}
+
+            # 检查是否是首次提交（没有父提交）
+            if not commit.parents:
+                # 首次提交，获取所有文件
+                for item in commit.tree.traverse():
+                    if item.type == 'blob':  # 只处理文件，不处理目录
+                        file_path = item.path
+                        modified_files.append(file_path)
+                        # 首次提交前没有内容
+                        before_content = None
+                        # 获取提交后的内容
+                        after_content = repo.git.show(
+                            f"{commit.hexsha}:{file_path}")
+                        changes[file_path] = (before_content, after_content)
+            else:
+                # 获取parent commit
+                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
+                    modified_files.append(file_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  # 文件可能被删除
+
+                    changes[file_path] = (before_content, after_content)
+
+            # 使用commit消息作为查询内容
+            query = commit.message
+            querie_with_urls_and_changes.append(
+                (query, modified_files, changes))
+
+        except git.exc.GitCommandError as e:
+            printer.print_in_terminal(
+                "git_command_error", style="red", error=str(e))
+        except Exception as e:
+            printer.print_in_terminal(
+                "get_commit_changes_error", style="red", error=str(e))
+
+        return querie_with_urls_and_changes, None
+
+    def analyze_commit_changes(self, query: str, commit_id: str, conversations: List[Dict] = []) -> Optional[Generator[str, None, None]]:
+        """
+        分析指定commit的代码变更
+
+        Args:
+            query: 用户的查询/要求
+            commit_id: 指定的commit ID
+            conversations: 之前的对话历史 (可选)
+
+        Returns:
+            Optional[Generator]: 分析结果生成器，如果出错则返回None
+        """
+        printer = Printer()
+
+        # 获取commit的变更信息
+        changes, _ = self.get_commit_changes(commit_id)
+
+        if not changes:
+            printer.print_in_terminal("no_commit_changes", style="red")
+            return None
+
+        # 调用LLM进行代码分析
+        try:
+            # 获取prompt内容
+            prompt_content = self.analyze_commit.prompt(
+                querie_with_urls_and_changes=changes,
+                new_query=query
+            )
+
+            # 准备对话历史
+            if conversations:
+                new_conversations = conversations[:-1]
+            else:
+                new_conversations = []
+            new_conversations.append(
+                {"role": "user", "content": prompt_content})
+
+            # 调用LLM
+            v = stream_chat_with_continue(
+                llm=self.llm,
+                conversations=new_conversations,
+                llm_config={},
+                args=self.args
+            )
+            return v
+        except Exception as e:
+            printer.print_in_terminal(
+                "commit_analysis_error", style="red", error=str(e))
+            return None
+
+    def _get_index_file_content(self) -> str:
+        """获取索引文件内容"""
+        index_file_path = os.path.join(os.path.abspath(
+            self.args.source_dir), ".autocoderrules", "index.md")
+        index_file_content = ""
+
+        try:
+            if os.path.exists(index_file_path):
+                with open(index_file_path, 'r', encoding='utf-8') as f:
+                    index_file_content = f.read()
+        except Exception as e:
+            self.printer.print_str_in_terminal(
+                f"读取索引文件时出错: {str(e)}", style="yellow")
+
+        return index_file_content
+
+    def analyze(self, sources: SourceCodeList, query: str, conversations: List[Dict] = []) -> Optional[Generator[str, None, None]]:
+        """
+        分析给定的模块文件，根据用户需求生成可复用功能点的总结。
+
+        Args:
+            sources: 包含模块路径和内容的 SourceCodeList 对象。
+            query: 用户的具体分析要求。
+            conversations: 之前的对话历史 (可选)。
+
+        Returns:
+            Optional[Generator]: LLM 返回的分析结果生成器，如果出错则返回 None。
+        """
+        if not sources or not sources.sources:
+            self.printer.print_str_in_terminal("没有提供有效的模块文件进行分析。", style="red")
+            return None
+
+        try:
+            # 准备 Prompt
+            prompt_content = self.analyze_modules.prompt(
+                sources=sources,
+                query=query
+            )
+
+            # 准备对话历史
+            # 如果提供了 conversations，我们假设最后一个是用户的原始查询，替换它
+            if conversations:
+                new_conversations = conversations[:-1]
+            else:
+                new_conversations = []
+            new_conversations.append(
+                {"role": "user", "content": prompt_content})
+
+            # 调用 LLM
+            v = stream_chat_with_continue(
+                llm=self.llm,
+                conversations=new_conversations,
+                llm_config={},
+                args=self.args
+            )
+            return v
+        except Exception as e:
+            self.printer.print_in_terminal(
+                "代码分析时出错", style="red", error=str(e))
+            return None

auto_coder-0.1.370/src/autocoder/version.py

@@ -0,0 +1,2 @@
+
+__version__ = "0.1.370"