auto-coder 0.1.265__tar.gz → 0.1.266__tar.gz

{auto_coder-0.1.265 → auto_coder-0.1.266}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.265
+Version: 0.1.266
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter_client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas]>=0.1.167
+Requires-Dist: byzerllm[saas]>=0.1.168
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.265
+Version: 0.1.266
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter_client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas]>=0.1.167
+Requires-Dist: byzerllm[saas]>=0.1.168
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/auto_coder.egg-info/requires.txt

@@ -16,7 +16,7 @@ tabulate
 jupyter_client
 prompt-toolkit
 tokenizers
-byzerllm[saas]>=0.1.167
+byzerllm[saas]>=0.1.168
 patch
 diff_match_patch
 GitPython

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/commands/auto_command.py

@@ -641,9 +641,10 @@ class CommandAutoTuner:
         ## editblock_similarity: editblock相似度阈值
         - editblock相似度阈值，取值范围为0-1，默认值为0.9。如果设置的太低，虽然能合并进去，但是会引入错误。推荐不要修改该值。
 
-        ## generate_times_same_model: 相同模型生成次数
+        ## generate_times_same_model: 相同模型生成次数,也叫采样数
         当进行生成代码时，大模型会对同一个需求生成多份代码，然后会使用 generate_rerank_model 模型对多份代码进行重排序，
         然后选择得分最高的代码。一般次数越多，最终得到正确的代码概率越高。默认值为1，推荐设置为3。但是设置值越多，可能速度就越慢，消耗的token也越多。
+        当用户提到，帮我采样数设置为3， 那么你就设置该参数即可。
 
         ## skip_filter_index: 是否跳过索引过滤
         是否跳过根据用户的query 自动查找上下文。推荐设置为 false

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/common/auto_coder_lang.py

@@ -117,6 +117,7 @@ MESSAGES = {
         "quick_filter_stats": "{{ model_names }} Quick filter completed in {{ elapsed_time }} seconds, input tokens: {{ input_tokens }}, output tokens: {{ output_tokens }}, input cost: {{ input_cost }}, output cost: {{ output_cost }} speed: {{ speed }} tokens/s",
         "upsert_file": "✅ Updated file: {{ file_path }}",
         "unmerged_blocks_title": "Unmerged Blocks",
+        "merged_blocks_title": "Merged Changes",
         "quick_filter_title": "{{ model_name }} is analyzing how to filter context...",
         "quick_filter_failed": "❌ Quick filter failed: {{ error }}. ",
         "unmerged_file_path": "File: {{file_path}}",
@@ -157,6 +158,8 @@ MESSAGES = {
         "conf_not_found": "Configuration file not found: {{path}}",
         "index_export_success": "Index exported successfully: {{path}}",
         "index_import_success": "Index imported successfully: {{path}}",
+        "edits_title": "edits",
+        "diff_blocks_title":"diff blocks"
     },
     "zh": {
         "invalid_file_pattern": "无效的文件模式: {{file_pattern}}. 例如: regex://.*/package-lock\\.json",
@@ -273,6 +276,7 @@ MESSAGES = {
         "merge_success": "✅ 成功合并了 {{ num_files }} 个文件中的更改 {{ num_changes }}/{{ total_blocks }} 个代码块。",
         "no_changes_made": "⚠️ 未对任何文件进行更改。这个原因可能是因为coding函数生成的文本块格式有问题，导致无法合并进项目",
         "unmerged_blocks_title": "未合并代码块",
+        "merged_blocks_title": "合并的更改",
         "unmerged_file_path": "文件: {{file_path}}",
         "unmerged_search_block": "Search Block({{similarity}}):",
         "unmerged_replace_block": "Replace Block:",
@@ -311,6 +315,8 @@ MESSAGES = {
         "only_one_merge_result_success": "✅ 只有一个合并结果成功，返回该候选",
         "index_export_success": "索引导出成功: {{path}}",
         "index_import_success": "索引导入成功: {{path}}",
+        "edits_title": "编辑块",
+        "diff_blocks_title": "差异块",
     }}
 
 

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/common/code_auto_merge_diff.py

@@ -528,6 +528,37 @@ class CodeAutoMergeDiff:
             failed_blocks=failed_blocks
         )
 
+    def print_edits(self, edits: List[Tuple[str, List[str]]]):
+        """Print diffs for user review using rich library"""
+        from rich.syntax import Syntax
+        from rich.panel import Panel
+
+        # Group edits by file path
+        file_edits = {}
+        for path, hunk in edits:
+            if path not in file_edits:
+                file_edits[path] = []
+            file_edits[path].append(hunk)
+
+        # Generate formatted text for each file
+        formatted_text = ""
+        for path, hunks in file_edits.items():
+            formatted_text += f"##File: {path}\n"
+            for hunk in hunks:
+                formatted_text += "".join(hunk)
+            formatted_text += "\n"
+
+        # Print with rich panel
+        self.printer.print_in_terminal("edits_title", style="bold green")
+        self.printer.console.print(
+            Panel(
+                Syntax(formatted_text, "diff", theme="monokai"),
+                title="Edits",
+                border_style="green",
+                expand=False
+            )
+        )
+
     def _merge_code(self, content: str,force_skip_git:bool=False):        
         total = 0
         
@@ -544,9 +575,12 @@ class CodeAutoMergeDiff:
                 return            
        
         edits = self.get_edits(content)        
-        self.apply_edits(edits)
+        self.apply_edits(edits)        
 
         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"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)
+        else:
+            # Print edits for review
+            self.print_edits(edits)    

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/common/code_auto_merge_editblock.py

@@ -287,6 +287,7 @@ class CodeAutoMergeEditBlock:
                             for path, content in file_content_mapping.items()],
             failed_blocks=failed_blocks
         )
+    
 
     def _merge_code(self, content: str, force_skip_git: bool = False):
         file_content = FileUtils.read_file(self.args.file)
@@ -391,6 +392,7 @@ class CodeAutoMergeEditBlock:
                     style="red"
                 )
                 return
+            
         # Now, apply the changes
         for file_path, new_content in file_content_mapping.items():
             os.makedirs(os.path.dirname(file_path), exist_ok=True)
@@ -431,10 +433,14 @@ class CodeAutoMergeEditBlock:
                         self.git_require_msg(source_dir=self.args.source_dir, error=str(e)),
                         style="red"
                     )
+            else:
+                self.print_merged_blocks(merged_blocks)
+            
             self.printer.print_in_terminal("merge_success", 
                                          num_files=len(file_content_mapping.keys()),
                                          num_changes=len(changes_to_make),
                                          total_blocks=len(codes))
+            
         else:
             self.printer.print_in_terminal("no_changes_made")
 
@@ -455,3 +461,39 @@ class CodeAutoMergeEditBlock:
             syntax = Syntax(update, "python", theme="monokai", line_numbers=True)
             self.printer.console.print(Panel(syntax, expand=False))
         self.printer.print_in_terminal("unmerged_blocks_total", num_blocks=len(unmerged_blocks), style="bold red")
+
+
+    def print_merged_blocks(self, merged_blocks: List[tuple]):
+        """Print search/replace blocks for user review using rich library"""
+        from rich.syntax import Syntax
+        from rich.panel import Panel
+
+        # Group blocks by file path
+        file_blocks = {}
+        for file_path, head, update, similarity in merged_blocks:
+            if file_path not in file_blocks:
+                file_blocks[file_path] = []
+            file_blocks[file_path].append((head, update, similarity))
+
+        # Generate formatted text for each file
+        formatted_text = ""
+        for file_path, blocks in file_blocks.items():
+            formatted_text += f"##File: {file_path}\n"
+            for head, update, similarity in blocks:
+                formatted_text += "<<<<<<< SEARCH\n"
+                formatted_text += head + "\n"
+                formatted_text += "=======\n"
+                formatted_text += update + "\n"
+                formatted_text += ">>>>>>> REPLACE\n"
+            formatted_text += "\n"
+
+        # Print with rich panel
+        self.printer.print_in_terminal("merged_blocks_title", style="bold green")
+        self.printer.console.print(
+            Panel(
+                Syntax(formatted_text, "diff", theme="monokai"),
+                title="Merged Changes",
+                border_style="green",
+                expand=False
+            )
+        )

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/common/code_auto_merge_strict_diff.py

@@ -221,6 +221,37 @@ class CodeAutoMergeStrictDiff:
             failed_blocks=failed_blocks
         )
 
+    def print_diff_blocks(self, diff_blocks: List[PathAndCode]):
+        """Print diff blocks for user review using rich library"""
+        from rich.syntax import Syntax
+        from rich.panel import Panel
+
+        # Group blocks by file path
+        file_blocks = {}
+        for block in diff_blocks:
+            if block.path not in file_blocks:
+                file_blocks[block.path] = []
+            file_blocks[block.path].append(block.content)
+
+        # Generate formatted text for each file
+        formatted_text = ""
+        for path, contents in file_blocks.items():
+            formatted_text += f"##File: {path}\n"
+            for content in contents:
+                formatted_text += content + "\n"
+            formatted_text += "\n"
+
+        # Print with rich panel
+        self.printer.print_in_terminal("diff_blocks_title", style="bold green")
+        self.printer.console.print(
+            Panel(
+                Syntax(formatted_text, "diff", theme="monokai"),
+                title="Diff Blocks",
+                border_style="green",
+                expand=False
+            )
+        )
+
     def _merge_code(self, content: str, force_skip_git: bool = False):        
         total = 0
         
@@ -236,7 +267,8 @@ class CodeAutoMergeStrictDiff:
                 self.printer.print_in_terminal("git_init_required", style="red", source_dir=self.args.source_dir, error=str(e))
                 return            
        
-        diff_blocks = self.parse_diff_block(content)
+        diff_blocks = self.parse_diff_block(content)        
+
         for diff_blocks in diff_blocks:
             path = diff_blocks.path
             content = diff_blocks.content          
@@ -255,6 +287,9 @@ class CodeAutoMergeStrictDiff:
         if not force_skip_git and not self.args.skip_commit:
             commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)
+        else:
+            # Print diff blocks for review
+            self.print_diff_blocks(diff_blocks)    
 
     @byzerllm.prompt(render="jinja2")
     def git_require_msg(self, source_dir: str, error: str) -> str:

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/index/entry.py

@@ -110,8 +110,8 @@ def build_index_and_filter_files(
             printer.print_in_terminal("quick_filter_start", style="blue", model_name=model_name)
             quick_filter = QuickFilter(index_manager,stats,sources)        
             quick_filter_result = quick_filter.filter(index_manager.read_index(),args.query)
-            if quick_filter_result.has_error:                
-                raise KeyboardInterrupt(printer.get_message_from_key_with_format("quick_filter_failed",error=quick_filter_result.error_message))
+            # if quick_filter_result.has_error:                
+            #     raise KeyboardInterrupt(printer.get_message_from_key_with_format("quick_filter_failed",error=quick_filter_result.error_message))
 
             # Merge quick filter results into final_files
             if args.context_prune:

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/index/filter/normal_filter.py

@@ -31,7 +31,7 @@ class NormalFilterResult(BaseModel):
     files: Dict[str, TargetFile]
     has_error: bool
     error_message: Optional[str] = None
-    file_positions: Optional[Dict[str, int]]
+    file_positions: Optional[Dict[str, int]] = {}
 
 class NormalFilter():
     def __init__(self, index_manager: IndexManager,stats:Dict[str,Any],sources:List[SourceCode]):

{auto_coder-0.1.265 → auto_coder-0.1.266}/src/autocoder/index/filter/quick_filter.py

@@ -154,7 +154,7 @@ class QuickFilter():
                 else:
                     # 其他chunks直接使用with_llm
                     meta_holder = MetaHolder()
-                    start_time = time.monotonic()
+                    start_time = time.monotonic()                    
                     file_number_list = self.quick_filter_files.with_llm(self.index_manager.index_filter_llm).with_meta(
                         meta_holder).with_return_type(FileNumberList).run(chunk, self.args.query)
                     end_time = time.monotonic()

auto_coder-0.1.266/src/autocoder/version.py

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

auto_coder-0.1.265/src/autocoder/version.py

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