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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.251
+Version: 0.1.252
 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.161
+Requires-Dist: byzerllm[saas]>=0.1.163
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.251
+Version: 0.1.252
 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.161
+Requires-Dist: byzerllm[saas]>=0.1.163
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

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

@@ -57,6 +57,7 @@ src/autocoder/common/command_templates.py
 src/autocoder/common/const.py
 src/autocoder/common/files.py
 src/autocoder/common/git_utils.py
+src/autocoder/common/global_cancel.py
 src/autocoder/common/image_to_page.py
 src/autocoder/common/interpreter.py
 src/autocoder/common/llm_rerank.py
@@ -144,6 +145,7 @@ src/autocoder/utils/request_event_queue.py
 src/autocoder/utils/request_queue.py
 src/autocoder/utils/rest.py
 src/autocoder/utils/tests.py
+src/autocoder/utils/thread_utils.py
 src/autocoder/utils/types.py
 src/autocoder/utils/auto_coder_utils/__init__.py
 src/autocoder/utils/auto_coder_utils/chat_stream_out.py

{auto_coder-0.1.251 → auto_coder-0.1.252}/src/auto_coder.egg-info/requires.txt

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

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

@@ -6,6 +6,7 @@ from autocoder.common import git_utils, code_auto_execute
 from autocoder.utils.llm_client_interceptors import token_counter_interceptor
 from autocoder.db.store import Store
 
+from autocoder.utils.llms import get_llm_names
 from autocoder.utils.queue_communicate import (
     queue_communicate,
     CommunicateEvent,
@@ -765,6 +766,23 @@ def main(input_args: Optional[List[str]] = None):
                 )
                 llm.setup_sub_client("planner_model", planner_model)
 
+            if args.commit_model:
+                model_name = args.commit_model.strip()
+                model_info = models_module.get_model_by_name(model_name)
+                commit_model = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+                commit_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=model_name,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
+                )
+                llm.setup_sub_client("commit_model", commit_model)    
+
             if args.designer_model:
                 model_name = args.designer_model.strip()
                 model_info = models_module.get_model_by_name(model_name)
@@ -875,6 +893,11 @@ def main(input_args: Optional[List[str]] = None):
                 designer_model.setup_default_model_name(args.designer_model)
                 llm.setup_sub_client("designer_model", designer_model)
 
+            if args.commit_model:
+                commit_model = byzerllm.ByzerLLM()
+                commit_model.setup_default_model_name(args.commit_model)
+                llm.setup_sub_client("commit_model", commit_model)
+
     else:
         llm = None
 
@@ -1348,9 +1371,7 @@ def main(input_args: Optional[List[str]] = None):
                 )              
 
 
-            model_name = getattr(chat_llm, 'default_model_name', None)
-            if not model_name:
-                model_name = "unknown(without default model name)"    
+            model_name = ",".join(get_llm_names(chat_llm))
             
             assistant_response, last_meta = stream_out(
                     v, 
@@ -1364,10 +1385,13 @@ def main(input_args: Optional[List[str]] = None):
             if last_meta:
                 elapsed_time = time.time() - start_time
                 printer = Printer()
+                speed = last_meta.generated_tokens_count / elapsed_time
                 printer.print_in_terminal("stream_out_stats", 
                                     elapsed_time=elapsed_time,
+                                    first_token_time=last_meta.first_token_time,
                                     input_tokens=last_meta.input_tokens_count,
-                                    output_tokens=last_meta.generated_tokens_count)
+                                    output_tokens=last_meta.generated_tokens_count,
+                                    speed=round(speed, 2))
             
             chat_history["ask_conversation"].append(
                 {"role": "assistant", "content": assistant_response}

{auto_coder-0.1.251 → auto_coder-0.1.252}/src/autocoder/auto_coder_rag.py

@@ -32,7 +32,10 @@ if platform.system() == "Windows":
     init()
 
 
-def initialize_system():
+def initialize_system(args):
+    if args.product_mode == "lite":
+        return
+
     print(f"\n\033[1;34m{get_message('initializing')}\033[0m")
 
     def print_status(message, status):
@@ -316,13 +319,30 @@ def main(input_args: Optional[List[str]] = None):
         action="store_true",
         help="Whether to return responses without contexts. only works when pro plugin is installed",
     )
-    serve_parser.add_argument(
-        "--data_cells_max_num",
+
+    serve_parser.add_argument("--data_cells_max_num",
         type=int,
         default=2000,
         help="Maximum number of data cells to process",
     )
 
+    serve_parser.add_argument(
+        "--product_mode",
+        type=str,
+        default="pro",
+        help="The mode of the auto-coder.rag, lite/pro default is pro",
+    )
+    serve_parser.add_argument(
+        "--lite",
+        action="store_true",
+        help="Run in lite mode (equivalent to --product_mode=lite)",
+    )
+    serve_parser.add_argument(
+        "--pro",
+        action="store_true",
+        help="Run in pro mode (equivalent to --product_mode=pro)",
+    )
+
     serve_parser.add_argument(
         "--recall_model",
         default="",
@@ -373,6 +393,22 @@ def main(input_args: Optional[List[str]] = None):
     # Tools command
     tools_parser = subparsers.add_parser("tools", help="Various tools")
     tools_subparsers = tools_parser.add_subparsers(dest="tool", help="Available tools")
+    tools_parser.add_argument(
+        "--product_mode",
+        type=str,
+        default="pro",
+        help="The mode of the auto-coder.rag, lite/pro default is pro",
+    )
+    tools_parser.add_argument(
+        "--lite",
+        action="store_true",
+        help="Run in lite mode (equivalent to --product_mode=lite)",
+    )
+    tools_parser.add_argument(
+        "--pro",
+        action="store_true",
+        help="Run in pro mode (equivalent to --product_mode=pro)",
+    )
 
     # Count tool
     count_parser = tools_subparsers.add_parser("count", help="Count tokens in a file")
@@ -431,8 +467,15 @@ def main(input_args: Optional[List[str]] = None):
             benchmark_byzerllm(args.model, args.parallel, args.rounds, args.query)
 
     elif args.command == "serve":
+         # Handle lite/pro flags
+        if args.lite:
+            args.product_mode = "lite"
+        elif args.pro:
+            args.product_mode = "pro"
+
         if not args.quick:
-            initialize_system()
+            initialize_system(args)
+       
         server_args = ServerArgs(
             **{
                 arg: getattr(args, arg)
@@ -448,7 +491,11 @@ def main(input_args: Optional[List[str]] = None):
             }
         )
 
-        if auto_coder_args.enable_hybrid_index:
+
+        if auto_coder_args.enable_hybrid_index and args.product_mode == "lite":
+            raise Exception("Hybrid index is not supported in lite mode")
+
+        if auto_coder_args.enable_hybrid_index and args.product_mode == "pro":
             # 尝试连接storage
             try:
                 from byzerllm.apps.byzer_storage.simple_api import ByzerStorage
@@ -460,36 +507,107 @@ def main(input_args: Optional[List[str]] = None):
                     "When enable_hybrid_index is true, ByzerStorage must be started"
                 )
                 logger.error("Please run 'byzerllm storage start' first")
-                return
-        else:
-            byzerllm.connect_cluster(address=args.ray_address)
-        llm = byzerllm.ByzerLLM()
-        llm.setup_default_model_name(args.model)
-
-        # Setup sub models if specified
-        if args.recall_model:
-            recall_model = byzerllm.ByzerLLM()
-            recall_model.setup_default_model_name(args.recall_model)
-            llm.setup_sub_client("recall_model", recall_model)
-
-        if args.chunk_model:
-            chunk_model = byzerllm.ByzerLLM()
-            chunk_model.setup_default_model_name(args.chunk_model)
-            llm.setup_sub_client("chunk_model", chunk_model)
+                return                        
+        
+        
 
-        if args.qa_model:
-            qa_model = byzerllm.ByzerLLM()
-            qa_model.setup_default_model_name(args.qa_model)
-            llm.setup_sub_client("qa_model", qa_model)
+        if args.product_mode == "pro":
+            byzerllm.connect_cluster(address=args.ray_address)
+            llm = byzerllm.ByzerLLM()
+            llm.skip_nontext_check = True
+            llm.setup_default_model_name(args.model)
+
+            # Setup sub models if specified
+            if args.recall_model:
+                recall_model = byzerllm.ByzerLLM()
+                recall_model.setup_default_model_name(args.recall_model)
+                recall_model.skip_nontext_check = True
+                llm.setup_sub_client("recall_model", recall_model)
+
+            if args.chunk_model:
+                chunk_model = byzerllm.ByzerLLM()
+                chunk_model.setup_default_model_name(args.chunk_model)
+                llm.setup_sub_client("chunk_model", chunk_model)
+
+            if args.qa_model:
+                qa_model = byzerllm.ByzerLLM()
+                qa_model.setup_default_model_name(args.qa_model)
+                qa_model.skip_nontext_check = True
+                llm.setup_sub_client("qa_model", qa_model)
+
+            # 当启用hybrid_index时,检查必要的组件
+            if auto_coder_args.enable_hybrid_index:
+                if not llm.is_model_exist("emb"):
+                    logger.error(
+                        "When enable_hybrid_index is true, an 'emb' model must be deployed"
+                    )
+                    return
+                llm.setup_default_emb_model_name("emb")
+
+        elif args.product_mode == "lite":
+            from autocoder import models as models_module
+            model_info = models_module.get_model_by_name(args.model)
+            llm = byzerllm.SimpleByzerLLM(default_model_name=args.model)
+            llm.deploy(
+                model_path="",
+                pretrained_model_type=model_info["model_type"],
+                udf_name=args.model,
+                infer_params={
+                    "saas.base_url": model_info["base_url"],
+                    "saas.api_key": model_info["api_key"],
+                    "saas.model": model_info["model_name"],
+                    "saas.is_reasoning": model_info["is_reasoning"]
+                }
+            )
 
-        # 当启用hybrid_index时,检查必要的组件
-        if auto_coder_args.enable_hybrid_index:
-            if not llm.is_model_exist("emb"):
-                logger.error(
-                    "When enable_hybrid_index is true, an 'emb' model must be deployed"
+            # Setup sub models if specified
+            if args.recall_model:
+                model_info = models_module.get_model_by_name(args.recall_model)
+                recall_model = byzerllm.SimpleByzerLLM(default_model_name=args.recall_model)
+                recall_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=args.recall_model,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
                 )
-                return
-            llm.setup_default_emb_model_name("emb")
+                llm.setup_sub_client("recall_model", recall_model)
+
+            if args.chunk_model:
+                model_info = models_module.get_model_by_name(args.chunk_model)
+                chunk_model = byzerllm.SimpleByzerLLM(default_model_name=args.chunk_model)
+                chunk_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=args.chunk_model,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
+                )
+                llm.setup_sub_client("chunk_model", chunk_model)
+
+            if args.qa_model:
+                model_info = models_module.get_model_by_name(args.qa_model)
+                qa_model = byzerllm.SimpleByzerLLM(default_model_name=args.qa_model)
+                qa_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=args.qa_model,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
+                )
+                llm.setup_sub_client("qa_model", qa_model)
 
         if server_args.doc_dir:
             auto_coder_args.rag_type = "simple"
@@ -577,16 +695,61 @@ def main(input_args: Optional[List[str]] = None):
             count_tokens(args.tokenizer_path, args.file)
         elif args.tool == "recall":
             from .common.recall_validation import validate_recall
-
-            llm = byzerllm.ByzerLLM.from_default_model(args.model)
+            from autocoder import models as models_module
+
+            # Handle lite/pro flags
+            if args.lite:
+                args.product_mode = "lite"
+            elif args.pro:
+                args.product_mode = "pro"
+
+            if args.product_mode == "pro":
+                llm = byzerllm.ByzerLLM.from_default_model(args.model)
+            else:  # lite mode
+                model_info = models_module.get_model_by_name(args.model)
+                llm = byzerllm.SimpleByzerLLM(default_model_name=args.model)
+                llm.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=args.model,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
+                )
 
             content = None if not args.content else [args.content]
             result = validate_recall(llm, content=content, query=args.query)
             print(f"Recall Validation Result:\n{result}")
+
         elif args.tool == "chunk":
             from .common.chunk_validation import validate_chunk
+            from autocoder import models as models_module
+            
+            if args.lite:
+                args.product_mode = "lite"
+            elif args.pro:
+                args.product_mode = "pro"
+
+            if args.product_mode == "pro":
+                llm = byzerllm.ByzerLLM.from_default_model(args.model)
+            else:  # lite mode
+                model_info = models_module.get_model_by_name(args.model)
+                llm = byzerllm.SimpleByzerLLM(default_model_name=args.model)
+                llm.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=args.model,
+                    infer_params={
+                        "saas.base_url": model_info["base_url"],
+                        "saas.api_key": model_info["api_key"],
+                        "saas.model": model_info["model_name"],
+                        "saas.is_reasoning": model_info["is_reasoning"]
+                    }
+                )
 
-            llm = byzerllm.ByzerLLM.from_default_model(args.model)
             content = None if not args.content else [args.content]
             result = validate_chunk(llm, content=content, query=args.query)
             print(f"Chunk Model Validation Result:\n{result}")

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

@@ -54,7 +54,7 @@ import shlex
 from autocoder.utils.llms import get_single_llm
 import pkg_resources
 from autocoder.common.printer import Printer
-from byzerllm.utils.langutil import run_in_thread,run_in_raw_thread
+from autocoder.utils.thread_utils import run_in_thread,run_in_raw_thread
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -136,6 +136,8 @@ commands = [
 
 
 def show_help():
+    print(f"\033[1m{get_message('official_doc')}\033[0m")
+    print()
     print(f"\033[1m{get_message('supported_commands')}\033[0m")
     print()
     print(
@@ -1713,7 +1715,7 @@ def commit(query: str):
                 if os.path.exists(temp_yaml):
                     os.remove(temp_yaml)
             
-            target_model = args.code_model or args.model
+            target_model = args.commit_model or args.model
             llm = get_single_llm(target_model, product_mode)
             printer = Printer()
             printer.print_in_terminal("commit_generating", style="yellow", model_name=target_model)
@@ -2208,6 +2210,9 @@ def manage_models(params, query: str):
             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:
                 # Check if api_key_path exists and file exists
                 is_api_key_set = "api_key" in m  
@@ -2221,12 +2226,60 @@ def manage_models(params, query: str):
                 table.add_row(
                     name,                    
                     m.get("model_name", ""),                    
-                    m.get("base_url", "")
+                    m.get("base_url", ""),
+                    f"{m.get('input_price', 0.0):.2f}",
+                    f"{m.get('output_price', 0.0):.2f}",
+                    f"{m.get('average_speed', 0.0):.3f}"
                 )
             console.print(table)
         else:
             printer.print_in_terminal("models_no_models", style="yellow")
 
+    elif subcmd == "/input_price":
+        args = query.strip().split()
+        if len(args) >= 2:
+            name = args[0]
+            try:
+                price = float(args[1])
+                if models.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)
+            except ValueError as e:
+                printer.print_in_terminal("models_invalid_price", style="red", error=str(e))
+        else:
+            printer.print_in_terminal("models_input_price_usage", style="red")
+            
+    elif subcmd == "/output_price":
+        args = query.strip().split()
+        if len(args) >= 2:
+            name = args[0]
+            try:
+                price = float(args[1])
+                if models.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)
+            except ValueError as e:
+                printer.print_in_terminal("models_invalid_price", style="red", error=str(e))
+        else:
+            printer.print_in_terminal("models_output_price_usage", style="red")
+
+    elif subcmd == "/speed":
+        args = query.strip().split()
+        if len(args) >= 2:
+            name = args[0]
+            try:
+                speed = float(args[1])
+                if models.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)
+            except ValueError as e:
+                printer.print_in_terminal("models_invalid_speed", style="red", error=str(e))
+        else:
+            printer.print_in_terminal("models_speed_usage", style="red")
+            
     elif subcmd == "/add":
         # Support both simplified and legacy formats
         args = query.strip().split(" ")        

{auto_coder-0.1.251 → auto_coder-0.1.252}/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 ...",
+        "models_usage": "Usage: /models /list|/add|/add_model|/remove|/price|/speed ...",
         "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]",
@@ -96,6 +96,14 @@ MESSAGES = {
         "models_add_model_remove": "Model '{{name}}' not found.",
         "models_add_model_removed": "Removed model: {{name}}",
         "models_unknown_subcmd": "Unknown subcommand: {{subcmd}}",
+        "models_input_price_updated": "Updated input price for model {{name}} to {{price}} M/token",
+        "models_output_price_updated": "Updated output price for model {{name}} to {{price}} M/token",
+        "models_invalid_price": "Invalid price value: {{error}}",
+        "models_input_price_usage": "Usage: /models /input_price <name> <value>",
+        "models_output_price_usage": "Usage: /models /output_price <name> <value>",
+        "models_speed_updated": "Updated speed for model {{name}} to {{speed}} s/request",
+        "models_invalid_speed": "Invalid speed value: {{error}}",
+        "models_speed_usage": "Usage: /models /speed <name> <value>",
         "models_title": "All Models (内置 + models.json)",
         "models_no_models": "No models found.",
         "models_lite_only": "The /models command is only available in lite mode",
@@ -117,6 +125,7 @@ MESSAGES = {
         "commit_message": "{{ model_name }} Generated commit message: {{ message }}",
         "commit_failed": "{{ model_name }} Failed to generate commit message: {{ error }}",
         "confirm_execute": "Do you want to execute this script?",
+        "official_doc": "Official Documentation: https://uelng8wukz.feishu.cn/wiki/NhPNwSRcWimKFIkQINIckloBncI",
     },
     "zh": {
         "commit_generating": "{{ model_name }} 正在生成提交信息...", 
@@ -204,7 +213,7 @@ MESSAGES = {
         "conf_value": "值",
         "conf_title": "配置设置",
         "conf_subtitle": "使用 /conf <key>:<value> 修改这些设置",
-        "models_usage": "用法: /models /list|/add|/add_model|/remove ...",
+        "models_usage": "用法: /models /list|/add|/add_model|/remove|/price|/speed ...",
         "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]",
@@ -215,6 +224,14 @@ MESSAGES = {
         "models_add_model_remove": "找不到模型 '{{name}}'。",
         "models_add_model_removed": "已移除模型: {{name}}",
         "models_unknown_subcmd": "未知的子命令: {{subcmd}}",
+        "models_input_price_updated": "已更新模型 {{name}} 的输入价格为 {{price}} M/token",
+        "models_output_price_updated": "已更新模型 {{name}} 的输出价格为 {{price}} M/token",
+        "models_invalid_price": "无效的价格值: {{error}}",
+        "models_input_price_usage": "用法: /models /input_price <name> <value>",
+        "models_output_price_usage": "用法: /models /output_price <name> <value>",
+        "models_speed_updated": "已更新模型 {{name}} 的速度为 {{speed}} 秒/请求",
+        "models_invalid_speed": "无效的速度值: {{error}}",
+        "models_speed_usage": "用法: /models /speed <name> <value>",
         "models_title": "所有模型 (内置 + models.json)",
         "models_no_models": "未找到任何模型。",
         "models_lite_only": "/models 命令仅在 lite 模式下可用",
@@ -232,7 +249,8 @@ MESSAGES = {
         "remove_files_none": "没有文件被移除。",
         "files_removed": "移除的文件",
         "models_api_key_empty": "警告:  {{name}}  API key 为空。请设置一个有效的 API key。",
-        "confirm_execute": "是否执行此脚本？",
+        "confirm_execute": "是否执行此脚本?",
+        "official_doc": "官方文档: https://uelng8wukz.feishu.cn/wiki/NhPNwSRcWimKFIkQINIckloBncI",
     }
 }
 

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

@@ -254,6 +254,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     planner_model: Optional[str] = ""
     voice2text_model: Optional[str] = ""
     text2voice_model: Optional[str] = ""
+    commit_model: Optional[str] = ""
     
     skip_build_index: Optional[bool] = False
     skip_filter_index: Optional[bool] = False

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

@@ -3,6 +3,7 @@ from byzerllm.utils import format_str_jinja2
 
 MESSAGES = {
     "en": {
+        "generation_cancelled": "[Interrupted] Generation cancelled",
         "model_not_found": "Model {{model_name}} not found",
         "generating_shell_script": "Generating Shell Script",
         "new_session_started": "New session started. Previous chat history has been archived.",    
@@ -51,7 +52,7 @@ MESSAGES = {
             "Paste the answer to the input box below, use '/break' to exit, '/clear' to clear the screen, '/eof' to submit."
         ),
         "code_generation_start": "Auto generate the code...",
-        "code_generation_complete": "Code generation completed in {{ duration }} seconds, input_tokens_count: {{ input_tokens }}, generated_tokens_count: {{ output_tokens }}",
+        "code_generation_complete": "Code generation completed in {{ duration }} seconds, input_tokens_count: {{ input_tokens }}, generated_tokens_count: {{ output_tokens }}, speed: {{ speed }} tokens/s",
         "code_merge_start": "Auto merge the code...",
         "code_execution_warning": "Content(send to model) is {{ content_length }} tokens (you may collect too much files), which is larger than the maximum input length {{ max_length }}",
         "quick_filter_start": "{{ model_name }} Starting filter context(quick_filter)...",
@@ -73,7 +74,7 @@ MESSAGES = {
         "ranking_process_failed": "Ranking process failed: {{ error }}",
         "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 }}",
+        "stream_out_stats": "Elapsed time {{ elapsed_time }} seconds, first token time: {{ first_token_time }} seconds, input tokens: {{ input_tokens }}, output tokens: {{ output_tokens }}, speed: {{ speed }} tokens/s",
         "quick_filter_stats": "快速过滤器完成，耗时 {{ elapsed_time }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}",
         "upsert_file": "✅ Updated file: {{ file_path }}",  
         "unmerged_blocks_title": "Unmerged Blocks",
@@ -91,6 +92,7 @@ MESSAGES = {
         "estimated_input_tokens_in_generate": "Estimated input tokens in generate ({{ generate_mode }}): {{ estimated_input_tokens }}",
     },
     "zh": {
+        "generation_cancelled": "[已中断] 生成已取消",
         "model_not_found": "未找到模型: {{model_name}}",
         "generating_shell_script": "正在生成 Shell 脚本",
         "new_session_started": "新会话已开始。之前的聊天历史已存档。",
@@ -139,7 +141,7 @@ MESSAGES = {
             "将获得答案黏贴到下面的输入框，换行后，使用 '/break' 退出，'/clear' 清屏，'/eof' 提交。"
         ),
         "code_generation_start": "正在自动生成代码...",
-        "code_generation_complete": "代码生成完成，耗时 {{ duration }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}",
+        "code_generation_complete": "代码生成完成，耗时 {{ duration }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}, 速度: {{ speed }} tokens/秒",
         "code_merge_start": "正在自动合并代码...",
         "code_execution_warning": "发送给模型的内容长度为 {{ content_length }} tokens（您可能收集了太多文件），超过了最大输入长度 {{ max_length }}",
         "quick_filter_start": "{{ model_name }} 开始查找上下文(quick_filter)...",
@@ -171,7 +173,7 @@ MESSAGES = {
         "ranking_complete": "排序完成，耗时 {{ elapsed }} 秒，总投票数: {{ total_tasks }}，最佳候选索引: {{ best_candidate }}，得分: {{ scores }}，输入token数: {{ input_tokens }}，输出token数: {{ output_tokens }}",
         "ranking_process_failed": "排序过程失败: {{ error }}",
         "ranking_failed": "排序失败，耗时 {{ elapsed }} 秒，使用原始顺序",
-        "stream_out_stats": "耗时 {{ elapsed_time }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}",
+        "stream_out_stats": "总耗时 {{ elapsed_time }} 秒，首token时间: {{ first_token_time }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}, 速度: {{ speed }} tokens/秒",
         "quick_filter_stats": "Quick filter completed in {{ elapsed_time }} seconds, input tokens: {{ input_tokens }}, output tokens: {{ output_tokens }}",
         "quick_filter_title": "{{ model_name }} 正在分析如何筛选上下文...",
         "quick_filter_failed": "❌ 快速过滤器失败: {{ error }}. ",