auto-coder 0.1.239__tar.gz → 0.1.242__tar.gz

{auto_coder-0.1.239 → auto_coder-0.1.242}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.239
+Version: 0.1.242
 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.155
+Requires-Dist: byzerllm[saas]>=0.1.158
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.239
+Version: 0.1.242
 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.155
+Requires-Dist: byzerllm[saas]>=0.1.158
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/auto_coder.egg-info/requires.txt

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

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/autocoder/auto_coder.py

@@ -14,7 +14,7 @@ from autocoder.utils.queue_communicate import (
 
 import yaml
 import os
-import uuid
+import time
 from byzerllm.utils.client import EventCallbackResult, EventName
 from prompt_toolkit import prompt
 from prompt_toolkit.formatted_text import FormattedText
@@ -1310,6 +1310,9 @@ def main(input_args: Optional[List[str]] = None):
                     printer = Printer()
                     printer.print_in_terminal("memory_save_success")
                 return {}
+
+            # 计算耗时            
+            start_time = time.time()    
                         
             if "rag" in args.action:
                 args.enable_rag_search = True
@@ -1335,12 +1338,23 @@ def main(input_args: Optional[List[str]] = None):
                     conversations=loaded_conversations,
                     llm_config={}
                 )              
+
+                
             
             assistant_response, last_meta = stream_out(
                     v, 
                     request_id=args.request_id,                    
                     console=console
                 )
+                                                                       
+            # 打印耗时和token统计            
+            if last_meta:
+                elapsed_time = time.time() - start_time
+                printer = Printer()
+                printer.print_in_terminal("stream_out_stats", 
+                                    elapsed_time=elapsed_time,
+                                    input_tokens=last_meta.input_tokens_count,
+                                    output_tokens=last_meta.generated_tokens_count)
             
             chat_history["ask_conversation"].append(
                 {"role": "assistant", "content": assistant_response}
@@ -1361,7 +1375,8 @@ def main(input_args: Optional[List[str]] = None):
                 save_to_memory_file(ask_conversation=chat_history["ask_conversation"],
                                     query=args.query,
                                     response=assistant_response)  
-                print("Saved to your memory")                      
+                printer = Printer()
+                printer.print_in_terminal("memory_save_success")                      
             return
 
         else:

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/autocoder/common/auto_coder_lang.py

@@ -60,7 +60,10 @@ MESSAGES = {
         "ranking_all_failed": "All ranking requests failed",
         "ranking_complete": "Ranking completed in {{ elapsed }}s, total voters: {{ total_tasks }}, best candidate index: {{ best_candidate }}, scores: {{ scores }}, input_tokens: {{ input_tokens }}, output_tokens: {{ output_tokens }}",
         "ranking_process_failed": "Ranking process failed: {{ error }}",
-        "ranking_failed": "Ranking failed in {{ elapsed }}s, using original order"
+        "ranking_failed": "Ranking failed in {{ elapsed }}s, using original order",
+        "begin_index_source_code": "🚀 Begin to index source code in {{ source_dir }}",
+        "stream_out_stats": "Elapsed time {{ elapsed_time }} seconds, input tokens: {{ input_tokens }}, output tokens: {{ output_tokens }}",
+        "upsert_file": "✅ Updated file: {{ file_path }}",        
     },
     "zh": {
         "memory_save_success": "✅ 已保存到您的记忆中",
@@ -108,6 +111,7 @@ MESSAGES = {
         "normal_filter_start": "开始查找上下文(normal_filter)...",
         "pylint_check_failed": "⚠️ Pylint 检查失败: {{ error_message }}",
         "pylint_error": "❌ 运行 Pylint 时出错: {{ error_message }}",
+        "begin_index_source_code": "🚀 开始为 {{ source_dir }} 中的源代码建立索引",
         "unmerged_blocks_warning": "⚠️ 发现 {{ num_blocks }} 个未合并的代码块，更改将不会被应用。请手动检查后重试。",
         "pylint_file_check_failed": "⚠️ {{ file_path }} 的 Pylint 检查失败。更改未应用。错误: {{ error_message }}",
         "merge_success": "✅ 成功合并了 {{ num_files }} 个文件中的更改 {{ num_changes }}/{{ total_blocks }} 个代码块。",
@@ -128,8 +132,9 @@ MESSAGES = {
         "ranking_all_failed": "所有排序请求都失败",
         "ranking_complete": "排序完成，耗时 {{ elapsed }} 秒，总投票数: {{ total_tasks }}，最佳候选索引: {{ best_candidate }}，得分: {{ scores }}，输入token数: {{ input_tokens }}，输出token数: {{ output_tokens }}",
         "ranking_process_failed": "排序过程失败: {{ error }}",
-        "ranking_failed": "排序失败，耗时 {{ elapsed }} 秒，使用原始顺序"
-    }    
+        "ranking_failed": "排序失败，耗时 {{ elapsed }} 秒，使用原始顺序",
+        "stream_out_stats": "耗时 {{ elapsed_time }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}"
+    },    
 }
 
 

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/autocoder/common/code_modification_ranker.py

@@ -99,9 +99,7 @@ class CodeModificationRanker:
                         results.append(v.rank_result)
                     except Exception as e:
                         self.printer.print_in_terminal(
-                            "ranking_failed_request", style="yellow", error=str(e))
-                        if self.args.debug:
-                            print(traceback.format_exc())
+                            "ranking_failed_request", style="yellow", error=str(e))                        
                         continue
 
                 if not results:

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/autocoder/common/printer.py

@@ -5,6 +5,7 @@ from typing import Optional,Dict,Any
 from byzerllm.utils import format_str_jinja2
 from autocoder.common.auto_coder_lang import get_message
 from autocoder.chat_auto_coder_lang import get_message as get_chat_message
+
 class Printer:
     def __init__(self,console:Optional[Console]=None):
         if console is None:

{auto_coder-0.1.239 → auto_coder-0.1.242}/src/autocoder/index/for_command.py

@@ -5,7 +5,7 @@ from autocoder.tsproject import TSProject
 from autocoder.pyproject import PyProject
 import tabulate
 import textwrap
-from loguru import logger
+from autocoder.common.printer import Printer
 import os
 from autocoder.utils.request_queue import (
     request_queue,
@@ -35,7 +35,8 @@ def wrap_text_in_table(data, max_width=60):
 def index_command(args, llm):
     source_dir = os.path.abspath(args.source_dir)
     args.source_dir = source_dir
-    logger.info(f"Beging to index source code in {source_dir}")
+    printer = Printer()
+    printer.print_in_terminal("begin_index_source_code", style="bold green", source_dir=source_dir)
     if args.project_type == "ts":
         pp = TSProject(args=args, llm=llm)
     elif args.project_type == "py":

auto_coder-0.1.242/src/autocoder/version.py

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

auto_coder-0.1.239/src/autocoder/version.py

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