auto-coder 0.1.252__tar.gz → 0.1.255__tar.gz

{auto_coder-0.1.252 → auto_coder-0.1.255}/PKG-INFO

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

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/auto_coder.egg-info/PKG-INFO

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

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/auto_coder.egg-info/SOURCES.txt

@@ -65,6 +65,7 @@ src/autocoder/common/mcp_hub.py
 src/autocoder/common/mcp_server.py
 src/autocoder/common/mcp_tools.py
 src/autocoder/common/memory_manager.py
+src/autocoder/common/model_speed_test.py
 src/autocoder/common/printer.py
 src/autocoder/common/recall_validation.py
 src/autocoder/common/screenshots.py
@@ -77,6 +78,7 @@ src/autocoder/common/types.py
 src/autocoder/common/utils_code_auto_generate.py
 src/autocoder/common/mcp_servers/__init__.py
 src/autocoder/common/mcp_servers/mcp_server_perplexity.py
+src/autocoder/data/byzerllm.md
 src/autocoder/data/tokenizer.json
 src/autocoder/db/__init__.py
 src/autocoder/db/store.py

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/auto_coder.py

@@ -256,11 +256,13 @@ def main(input_args: Optional[List[str]] = None):
             libs_dir = os.path.join(auto_coder_dir, "storage", "libs")
             code_search_path = None
             if os.path.exists(libs_dir):
-                retrieval_libs_dir = os.path.join(
-                    libs_dir, get_latest_byzer_retrieval_lib(libs_dir)
-                )
-                if os.path.exists(retrieval_libs_dir):
-                    code_search_path = [retrieval_libs_dir]
+                latest_retrieval_lib_dir = get_latest_byzer_retrieval_lib(libs_dir)
+                if latest_retrieval_lib_dir :
+                    retrieval_libs_dir = os.path.join(
+                        libs_dir, latest_retrieval_lib_dir
+                    )
+                    if os.path.exists(retrieval_libs_dir):
+                        code_search_path = [retrieval_libs_dir]
 
             try:
                 init_options = {}
@@ -1109,8 +1111,7 @@ def main(input_args: Optional[List[str]] = None):
             return
 
         elif raw_args.agent_command == "chat":
-            from autocoder.rag.rag_entry import RAGFactory
-
+            
             memory_dir = os.path.join(args.source_dir, ".auto-coder", "memory")
             os.makedirs(memory_dir, exist_ok=True)
             memory_file = os.path.join(memory_dir, "chat_history.json")
@@ -1335,6 +1336,7 @@ def main(input_args: Optional[List[str]] = None):
             start_time = time.time()    
                         
             if "rag" in args.action:
+                from autocoder.rag.rag_entry import RAGFactory
                 args.enable_rag_search = True
                 args.enable_rag_context = False
                 rag = RAGFactory.get_rag(llm=chat_llm, args=args, path="")
@@ -1348,7 +1350,8 @@ def main(input_args: Optional[List[str]] = None):
                 response = mcp_server.send_request(
                     McpRequest(
                         query=args.query,
-                        model=args.inference_model or args.model
+                        model=args.inference_model or args.model,
+                        product_mode=args.product_mode
                     )
                 )
                 v = [[response.result,None]]

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/chat_auto_coder.py

@@ -49,7 +49,7 @@ from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRe
 import byzerllm
 from byzerllm.utils import format_str_jinja2
 from autocoder.common.memory_manager import get_global_memory_file_paths 
-from autocoder import models
+from autocoder import models as models_module
 import shlex
 from autocoder.utils.llms import get_single_llm
 import pkg_resources
@@ -1539,11 +1539,12 @@ def mcp(query: str):
         if os.path.exists(temp_yaml):
             os.remove(temp_yaml)
 
-    mcp_server = get_mcp_server()
+    mcp_server = get_mcp_server()    
     response = mcp_server.send_request(
         McpRequest(
             query=query,
-            model=args.inference_model or args.model
+            model=args.inference_model or args.model,
+            product_mode=args.product_mode
         )
     )
 
@@ -1564,19 +1565,19 @@ def mcp(query: str):
         file_path = os.path.join(mcp_dir, f"{timestamp}.md")
         
         # Format response as markdown
-        markdown_content = f"# {printer.get_message_from_key('mcp_response_title')}\n\n{response.result}"
+        markdown_content = response.result
         
         # Save to file
         with open(file_path, "w", encoding="utf-8") as f:
             f.write(markdown_content)
             
-        # Print with markdown formatting
-        printer.print_panel(
-            Markdown(markdown_content),
-            text_options={"justify": "left"},
-            panel_options={
-                "border_style": "green"
-            }
+        console = Console()
+        console.print(
+            Panel(
+                Markdown(markdown_content, justify="left"),
+                title=printer.get_message_from_key('mcp_response_title'),
+                border_style="green"
+            )
         )
 
 
@@ -1741,7 +1742,7 @@ def commit(query: str):
             md5 = hashlib.md5(file_content.encode("utf-8")).hexdigest()
             file_name = os.path.basename(execute_file)
             commit_result = git_utils.commit_changes(
-                ".", f"auto_coder_{file_name}_{md5}"
+                ".", f"auto_coder_{file_name}_{md5}\n{commit_message}"
             )
             git_utils.print_commit_info(commit_result=commit_result)
             if commit_message:
@@ -2173,7 +2174,7 @@ def manage_models(params, query: str):
         printer.print_in_terminal("models_lite_only", style="red")
         return
         
-    models_data = models.load_models()
+    models_data = models_module.load_models()
     subcmd = ""
     if "/list" in query:
         subcmd = "/list"
@@ -2196,24 +2197,49 @@ def manage_models(params, query: str):
         subcmd = "/remove"
         query = query.replace("/remove", "", 1).strip()
 
+    if "/speed-test" in query:
+        subcmd = "/speed-test"
+        query = query.replace("/speed-test", "", 1).strip()
+
+    if "/speed_test" in query:
+        subcmd = "/speed-test"
+        query = query.replace("/speed_test", "", 1).strip() 
+
+    if "input_price" in query:
+        subcmd = "/input_price"
+        query = query.replace("/input_price", "", 1).strip()
+
+    if "output_price" in query:
+        subcmd = "/output_price"
+        query = query.replace("/output_price", "", 1).strip()        
+            
+    if "/speed" in query:
+        subcmd = "/speed"
+        query = query.replace("/speed", "", 1).strip()
+    
+        
+
     if not subcmd:
-        printer.print_in_terminal("models_usage")
-        return    
+        printer.print_in_terminal("models_usage")        
 
     if subcmd == "/list":                    
         if models_data:
+            # Sort models by speed (average_speed)
+            sorted_models = sorted(models_data, key=lambda x: float(x.get('average_speed', 0)))
+            sorted_models.reverse()
+            
             table = Table(
                 title=printer.get_message_from_key("models_title"),
                 expand=True,
                 show_lines=True
             )
-            table.add_column("Name", style="cyan", width=40, no_wrap=False)
-            table.add_column("Model Name", style="magenta", width=30, overflow="fold")
-            table.add_column("Base URL", style="white", width=50, overflow="fold")
-            table.add_column("Input Price (M)", style="magenta", width=15)
-            table.add_column("Output Price (M)", style="magenta", width=15)
-            table.add_column("Speed (s/req)", style="blue", width=15)
-            for m in models_data:
+            table.add_column("Name", style="cyan", width=30, overflow="fold", no_wrap=False)
+            table.add_column("Model Name", style="magenta", width=30, overflow="fold", no_wrap=False)
+            table.add_column("Base URL", style="white", width=40, overflow="fold", no_wrap=False)
+            table.add_column("Input Price (M)", style="magenta", width=15, overflow="fold", no_wrap=False)
+            table.add_column("Output Price (M)", style="magenta", width=15, overflow="fold", no_wrap=False)
+            table.add_column("Speed (s/req)", style="blue", width=15, overflow="fold", no_wrap=False)
+            for m in sorted_models:
                 # Check if api_key_path exists and file exists
                 is_api_key_set = "api_key" in m  
                 name = m.get("name", "")              
@@ -2241,7 +2267,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 price = float(args[1])
-                if models.update_model_input_price(name, price):
+                if models_module.update_model_input_price(name, price):
                     printer.print_in_terminal("models_input_price_updated", style="green", name=name, price=price)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2256,7 +2282,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 price = float(args[1])
-                if models.update_model_output_price(name, price):
+                if models_module.update_model_output_price(name, price):
                     printer.print_in_terminal("models_output_price_updated", style="green", name=name, price=price)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2271,7 +2297,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 speed = float(args[1])
-                if models.update_model_speed(name, speed):
+                if models_module.update_model_speed(name, speed):
                     printer.print_in_terminal("models_speed_updated", style="green", name=name, speed=speed)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2280,13 +2306,33 @@ def manage_models(params, query: str):
         else:
             printer.print_in_terminal("models_speed_usage", style="red")
             
+    elif subcmd == "/speed-test":
+        from autocoder.common.model_speed_test import render_speed_test_in_terminal
+        test_rounds = 1  # 默认测试轮数
+        
+        enable_long_context = False
+        if "/long_context" in query:
+            enable_long_context = True
+            query = query.replace("/long_context", "", 1).strip()
+
+        if "/long-context" in query:
+            enable_long_context = True
+            query = query.replace("/long-context", "", 1).strip()
+
+        # 解析可选的测试轮数参数
+        args = query.strip().split()
+        if args and args[0].isdigit():
+            test_rounds = int(args[0])
+            
+        render_speed_test_in_terminal(params.product_mode, test_rounds,enable_long_context=enable_long_context)
+    
     elif subcmd == "/add":
         # Support both simplified and legacy formats
         args = query.strip().split(" ")        
         if len(args) == 2:
             # Simplified: /models /add <name> <api_key>
             name, api_key = args[0], args[1]            
-            result = models.update_model_with_api_key(name, api_key)
+            result = models_module.update_model_with_api_key(name, api_key)
             if result:
                 printer.print_in_terminal("models_added", style="green", name=name)
             else:
@@ -2328,7 +2374,7 @@ def manage_models(params, query: str):
         }
 
         models_data.append(final_model)
-        models.save_models(models_data)
+        models_module.save_models(models_data)
         printer.print_in_terminal("models_add_model_success", style="green", name=data_dict["name"])
 
     elif subcmd == "/remove":
@@ -2341,7 +2387,7 @@ def manage_models(params, query: str):
         if len(filtered_models) == len(models_data):
             printer.print_in_terminal("models_add_model_remove", style="yellow", name=name)
             return
-        models.save_models(filtered_models)
+        models_module.save_models(filtered_models)
         printer.print_in_terminal("models_add_model_removed", style="green", name=name)
 
     else:

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/chat_auto_coder_lang.py

@@ -85,7 +85,7 @@ MESSAGES = {
         "design_desc": "Generate SVG image based on the provided description",
         "commit_desc": "Auto generate yaml file and commit changes based on user's manual changes",
         "models_desc": "Manage model configurations, only available in lite mode",
-        "models_usage": "Usage: /models /list|/add|/add_model|/remove|/price|/speed ...",
+        "models_usage": "Usage: /models <command>\nAvailable subcommands:\n  /list - List all models\n  /add <name> <api_key> - Add a built-in model\n  /add_model - Add a custom model\n  /remove <name> - Remove a model\n  /input_price <name> <value> - Set model input price\n  /output_price <name> <value> - Set model output price\n  /speed <name> <value> - Set model speed\n  /speed-test - Test models speed\n  /speed-test-long - Test models speed with long context",
         "models_added": "Added/Updated model '{{name}}' successfully.",
         "models_add_failed": "Failed to add model '{{name}}'. Model not found in defaults.",
         "models_add_usage": "Usage: /models /add <name> <api_key> or\n/models /add <name> <model_type> <model_name> <base_url> <api_key_path> [description]",
@@ -213,7 +213,7 @@ MESSAGES = {
         "conf_value": "值",
         "conf_title": "配置设置",
         "conf_subtitle": "使用 /conf <key>:<value> 修改这些设置",
-        "models_usage": "用法: /models /list|/add|/add_model|/remove|/price|/speed ...",
+        "models_usage": "用法: /models <命令>\n可用的子命令:\n  /list - 列出所有模型\n  /add <名称> <API密钥> - 添加内置模型\n  /add_model - 添加自定义模型\n  /remove <名称> - 移除模型\n  /input_price <名称> <价格> - 设置模型输入价格\n  /output_price <名称> <价格> - 设置模型输出价格\n  /speed <名称> <速度> - 设置模型速度\n  /speed-test - 测试模型速度\n  /speed-test-long - 使用长文本上下文测试模型速度",
         "models_added": "成功添加/更新模型 '{{name}}'。",
         "models_add_failed": "添加模型 '{{name}}' 失败。在默认模型中未找到该模型。",
         "models_add_usage": "用法: /models /add <name> <api_key> 或\n/models /add <name> <model_type> <model_name> <base_url> <api_key_path> [description]",

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/__init__.py

@@ -359,6 +359,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     product_mode: Optional[str] = "lite"
 
     keep_reasoning_content: Optional[bool] = False
+    keep_only_reasoning_content: Optional[bool] = False
 
     in_code_apply: bool = False
 

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/auto_coder_lang.py

@@ -3,6 +3,11 @@ from byzerllm.utils import format_str_jinja2
 
 MESSAGES = {
     "en": {
+        "models_no_active": "No active models found",
+        "models_speed_test_results": "Model Speed Test Results",
+        "models_testing": "Testing model: {{name}}...",
+        "models_testing_start": "Starting speed test for all active models...",
+        "models_testing_progress": "Testing progress: {{ completed }}/{{ total }} models",
         "generation_cancelled": "[Interrupted] Generation cancelled",
         "model_not_found": "Model {{model_name}} not found",
         "generating_shell_script": "Generating Shell Script",
@@ -92,6 +97,10 @@ MESSAGES = {
         "estimated_input_tokens_in_generate": "Estimated input tokens in generate ({{ generate_mode }}): {{ estimated_input_tokens }}",
     },
     "zh": {
+        "models_no_active": "未找到激活的模型",
+        "models_speed_test_results": "模型速度测试结果",
+        "models_testing": "正在测试模型: {{name}}...",
+        "models_testing_start": "开始对所有激活的模型进行速度测试...",
         "generation_cancelled": "[已中断] 生成已取消",
         "model_not_found": "未找到模型: {{model_name}}",
         "generating_shell_script": "正在生成 Shell 脚本",

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/code_auto_merge.py

@@ -177,5 +177,5 @@ class CodeAutoMerge:
 
         self.printer.print_in_terminal("files_merged", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            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)

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/code_auto_merge_diff.py

@@ -523,5 +523,5 @@ class CodeAutoMergeDiff:
 
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            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)

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/code_auto_merge_editblock.py

@@ -403,7 +403,7 @@ class CodeAutoMergeEditBlock:
             if not force_skip_git:
                 try:
                     commit_result = git_utils.commit_changes(
-                        self.args.source_dir, f"auto_coder_{file_name}_{md5}"
+                        self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}"
                     )
                     git_utils.print_commit_info(commit_result=commit_result)
                 except Exception as e:

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/code_auto_merge_strict_diff.py

@@ -233,7 +233,7 @@ class CodeAutoMergeStrictDiff:
                             
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            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)
 
     @byzerllm.prompt(render="jinja2")

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/command_completer.py

@@ -33,7 +33,11 @@ COMMANDS = {
         "/add": "",
         "/add_model": "",
         "/remove": "",
-        "/list": ""        
+        "/list": "",
+        "/speed": "",
+        "/speed-test": "",
+        "/input_price": "",
+        "/output_price": "",       
     }
 }
 

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/mcp_server.py

@@ -3,7 +3,6 @@ from asyncio import Queue as AsyncQueue
 import threading
 from typing import List, Dict, Any, Optional
 from dataclasses import dataclass
-import byzerllm
 from autocoder.common.mcp_hub import McpHub
 from autocoder.common.mcp_tools import McpExecutor
 from autocoder.common.mcp_hub import MCP_BUILD_IN_SERVERS
@@ -13,12 +12,13 @@ import time
 from pydantic import BaseModel
 import sys
 from loguru import logger
+from autocoder.utils.llms import get_single_llm
 
 @dataclass
 class McpRequest:
     query: str
     model: Optional[str] = None
-
+    product_mode: Optional[str] = None
 
 @dataclass
 class McpInstallRequest:
@@ -279,8 +279,7 @@ class McpServer:
                         await self._response_queue.put(McpResponse(result="", error=f"Failed to refresh MCP servers: {str(e)}"))
 
                 else:
-                    llm = byzerllm.ByzerLLM.from_default_model(
-                        model=request.model)
+                    llm = get_single_llm(request.model,product_mode=request.product_mode)
                     mcp_executor = McpExecutor(hub, llm)
                     conversations = [
                         {"role": "user", "content": request.query}]

{auto_coder-0.1.252 → auto_coder-0.1.255}/src/autocoder/common/mcp_servers/mcp_server_perplexity.py

@@ -55,7 +55,7 @@ async def handle_list_tools() -> list[types.Tool]:
                             # Commenting out larger models,which have higher risks of timing out,
                             # until Claude Desktop can handle long-running tasks effectively.
                             # "llama-3.1-sonar-large-128k-online",
-                            "llama-3.1-sonar-huge-128k-online",
+                            "sonar-reasoning-pro",
                         ],
                     },
                     "messages": {