auto-coder 0.1.226__py3-none-any.whl → 0.1.227__py3-none-any.whl

autocoder/chat_auto_coder.py

@@ -49,6 +49,8 @@ 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
+import shlex
 
 
 class SymbolItem(BaseModel):
@@ -68,6 +70,17 @@ def parse_arguments():
         action="store_true",
         help="Enter the auto-coder.chat without initializing the system",
     )
+
+    parser.add_argument(
+        "--product_mode",
+        type=str,
+        default="pro",
+        help="The mode of the auto-coder.chat, lite/pro default is pro",
+    )
+
+    parser.add_argument("--lite", action="store_true", help="Lite mode")
+    parser.add_argument("--pro", action="store_true", help="Pro mode")
+
     return parser.parse_args()
 
 
@@ -115,6 +128,7 @@ commands = [
     "/lib",
     "/design",
     "/mcp",
+    "/models",
 ]
 
 
@@ -136,9 +150,6 @@ def show_help():
     print(
         f"  \033[94m/coding\033[0m \033[93m<query>\033[0m - \033[92m{get_message('coding_desc')}\033[0m"
     )
-    print(
-        f"  \033[94m/design\033[0m \033[93m<query>\033[0m - \033[92m{get_message('design_desc')}\033[0m"
-    )
     print(
         f"  \033[94m/ask\033[0m \033[93m<query>\033[0m - \033[92m{get_message('ask_desc')}\033[0m"
     )
@@ -175,6 +186,7 @@ def show_help():
     print(
         f"  \033[94m/mode\033[0m - \033[92m{get_message('mode_desc')}\033[0m")
     print(f"  \033[94m/lib\033[0m - \033[92m{get_message('lib_desc')}\033[0m")
+    print(f"  \033[94m/models\033[0m - \033[92m{get_message('models_desc')}\033[0m")
     print(
         f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
     print()
@@ -234,7 +246,7 @@ def configure_project_type():
     return project_type
 
 
-def initialize_system():
+def initialize_system(args):
     print(f"\n\033[1;34m{get_message('initializing')}\033[0m")
 
     def print_status(message, status):
@@ -279,88 +291,108 @@ def initialize_system():
         print_status(get_message("init_complete"), "success")
 
     init_project()
-    # Check if Ray is running
-    print_status(get_message("checking_ray"), "")
-    ray_status = subprocess.run(
-        ["ray", "status"], capture_output=True, text=True)
-    if ray_status.returncode != 0:
-        print_status(get_message("ray_not_running"), "warning")
+
+    if args.product_mode == "lite":
+        # Setup deepseek api key
+        api_key_dir = os.path.expanduser("~/.auto-coder/keys")
+        api_key_file = os.path.join(api_key_dir, "api.deepseek.com")
+        
+        if not os.path.exists(api_key_file):
+            print_status(get_message("model_not_available"), "warning")
+            api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
+            
+            # Create directory if it doesn't exist
+            os.makedirs(api_key_dir, exist_ok=True)
+            
+            # Save the API key
+            with open(api_key_file, "w") as f:
+                f.write(api_key)
+            
+            print_status(f"API key saved successfully: {api_key_file}", "success")                    
+
+    if args.product_mode == "pro":
+        # Check if Ray is running
+        print_status(get_message("checking_ray"), "")
+        ray_status = subprocess.run(
+            ["ray", "status"], capture_output=True, text=True)
+        if ray_status.returncode != 0:
+            print_status(get_message("ray_not_running"), "warning")
+            try:
+                subprocess.run(["ray", "start", "--head"], check=True)
+                print_status(get_message("ray_start_success"), "success")
+            except subprocess.CalledProcessError:
+                print_status(get_message("ray_start_fail"), "error")
+                return
+        else:
+            print_status(get_message("ray_running"), "success")
+
+        # Check if deepseek_chat model is available
+        print_status(get_message("checking_model"), "")
         try:
-            subprocess.run(["ray", "start", "--head"], check=True)
-            print_status(get_message("ray_start_success"), "success")
+            result = subprocess.run(
+                ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
+                capture_output=True,
+                text=True,
+                timeout=30,
+            )
+            if result.returncode == 0:
+                print_status(get_message("model_available"), "success")
+                init_project()
+                print_status(get_message("init_complete_final"), "success")
+                return
+        except subprocess.TimeoutExpired:
+            print_status(get_message("model_timeout"), "error")
         except subprocess.CalledProcessError:
-            print_status(get_message("ray_start_fail"), "error")
-            return
-    else:
-        print_status(get_message("ray_running"), "success")
+            print_status(get_message("model_error"), "error")
+
+        # If deepseek_chat is not available
+        print_status(get_message("model_not_available"), "warning")
+        api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
+
+        print_status(get_message("deploying_model").format("Deepseek官方"), "")
+        deploy_cmd = [
+            "byzerllm",
+            "deploy",
+            "--pretrained_model_type",
+            "saas/openai",
+            "--cpus_per_worker",
+            "0.001",
+            "--gpus_per_worker",
+            "0",
+            "--worker_concurrency",
+            "1000",
+            "--num_workers",
+            "1",
+            "--infer_params",
+            f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-chat",
+            "--model",
+            "deepseek_chat",
+        ]
 
-    # Check if deepseek_chat model is available
-    print_status(get_message("checking_model"), "")
-    try:
-        result = subprocess.run(
-            ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
-            capture_output=True,
-            text=True,
-            timeout=30,
-        )
-        if result.returncode == 0:
-            print_status(get_message("model_available"), "success")
-            init_project()
-            print_status(get_message("init_complete_final"), "success")
+        try:
+            subprocess.run(deploy_cmd, check=True)
+            print_status(get_message("deploy_complete"), "success")
+        except subprocess.CalledProcessError:
+            print_status(get_message("deploy_fail"), "error")
             return
-    except subprocess.TimeoutExpired:
-        print_status(get_message("model_timeout"), "error")
-    except subprocess.CalledProcessError:
-        print_status(get_message("model_error"), "error")
-
-    # If deepseek_chat is not available
-    print_status(get_message("model_not_available"), "warning")
-    api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
-
-    print_status(get_message("deploying_model").format("Deepseek官方"), "")
-    deploy_cmd = [
-        "byzerllm",
-        "deploy",
-        "--pretrained_model_type",
-        "saas/openai",
-        "--cpus_per_worker",
-        "0.001",
-        "--gpus_per_worker",
-        "0",
-        "--worker_concurrency",
-        "1000",
-        "--num_workers",
-        "1",
-        "--infer_params",
-        f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-chat",
-        "--model",
-        "deepseek_chat",
-    ]
-
-    try:
-        subprocess.run(deploy_cmd, check=True)
-        print_status(get_message("deploy_complete"), "success")
-    except subprocess.CalledProcessError:
-        print_status(get_message("deploy_fail"), "error")
-        return
 
-    # Validate the deployment
-    print_status(get_message("validating_deploy"), "")
-    try:
-        validation_result = subprocess.run(
-            ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
-            capture_output=True,
-            text=True,
-            timeout=30,
-            check=True,
-        )
-        print_status(get_message("validation_success"), "success")
-    except (subprocess.TimeoutExpired, subprocess.CalledProcessError):
-        print_status(get_message("validation_fail"), "error")
-        print_status(get_message("manual_start"), "warning")
-        print_status("easy-byzerllm chat deepseek_chat 你好", "")
+        # Validate the deployment
+        print_status(get_message("validating_deploy"), "")
+        try:
+            validation_result = subprocess.run(
+                ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
+                capture_output=True,
+                text=True,
+                timeout=30,
+                check=True,
+            )
+            print_status(get_message("validation_success"), "success")
+        except (subprocess.TimeoutExpired, subprocess.CalledProcessError):
+            print_status(get_message("validation_fail"), "error")
+            print_status(get_message("manual_start"), "warning")
+            print_status("easy-byzerllm chat deepseek_chat 你好", "")
 
-    print_status(get_message("init_complete_final"), "success")
+        print_status(get_message("init_complete_final"), "success")
 
 
 def convert_yaml_config_to_str(yaml_config):
@@ -504,66 +536,6 @@ def configure(conf: str, skip_print=False):
         if not skip_print:
             print(f"\033[92mSet {key} to {value}\033[0m")
 
-
-def show_help():
-    print(f"\033[1m{get_message('supported_commands')}\033[0m")
-    print()
-    print(
-        f"  \033[94m{get_message('commands')}\033[0m - \033[93m{get_message('description')}\033[0m"
-    )
-    print(
-        f"  \033[94m/add_files\033[0m \033[93m<file1> <file2> ...\033[0m - \033[92m{get_message('add_files_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/remove_files\033[0m \033[93m<file1>,<file2> ...\033[0m - \033[92m{get_message('remove_files_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/chat\033[0m \033[93m<query>\033[0m - \033[92m{get_message('chat_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/coding\033[0m \033[93m<query>\033[0m - \033[92m{get_message('coding_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/ask\033[0m \033[93m<query>\033[0m - \033[92m{get_message('ask_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/summon\033[0m \033[93m<query>\033[0m - \033[92m{get_message('summon_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/revert\033[0m - \033[92m{get_message('revert_desc')}\033[0m")
-    print(
-        f"  \033[94m/commit\033[0m - \033[92m{get_message('commit_desc')}\033[0m")
-    print(
-        f"  \033[94m/conf\033[0m \033[93m<key>:<value>\033[0m  - \033[92m{get_message('conf_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/index/query\033[0m \033[93m<args>\033[0m - \033[92m{get_message('index_query_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/index/build\033[0m - \033[92m{get_message('index_build_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/list_files\033[0m - \033[92m{get_message('list_files_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/help\033[0m - \033[92m{get_message('help_desc')}\033[0m")
-    print(
-        f"  \033[94m/exclude_dirs\033[0m \033[93m<dir1>,<dir2> ...\033[0m - \033[92m{get_message('exclude_dirs_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/shell\033[0m \033[93m<command>\033[0m - \033[92m{get_message('shell_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/voice_input\033[0m - \033[92m{get_message('voice_input_desc')}\033[0m"
-    )
-    print(
-        f"  \033[94m/mode\033[0m - \033[92m{get_message('mode_desc')}\033[0m")
-    print(f"  \033[94m/lib\033[0m - \033[92m{get_message('lib_desc')}\033[0m")
-    print(
-        f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
-    print()
-
-
 # word_completer = WordCompleter(commands)
 
 
@@ -904,6 +876,14 @@ class CommandCompleter(Completer):
                 for command in parser.get_sub_commands():
                     if command.startswith(current_word):
                         yield Completion(command, start_position=-len(current_word))
+            elif words[0] == "/models":
+                new_text = text[len("/models"):]
+                parser = CommandTextParser(new_text, words[0])
+                parser.lib()
+                current_word = parser.current_word()
+                for command in parser.get_sub_commands():
+                    if command.startswith(current_word):
+                        yield Completion(command, start_position=-len(current_word))
 
             elif words[0] == "/coding":
                 new_text = text[len("/coding"):]
@@ -2088,6 +2068,133 @@ def generate_shell_command(input_text):
     finally:
         os.remove(execute_file)
 
+def manage_models(params, query: str):
+    """    
+    Handle /models subcommands:
+      /models /list - List all models (default + custom)
+      /models /add <name> <api_key> - Add model with simplified params
+      /models /add_model name=xxx base_url=xxx ... - Add model with custom params
+      /models /remove <name> - Remove model by name
+    """
+    print("manage_models", params, query)
+    console = Console()
+    
+    if params.product_mode != "lite":
+        console.print(f"[red]{get_message('models_lite_only')}[/red]")
+        return
+        
+    models_data = models.load_models()
+    subcmd = ""
+    if "/list" in query:
+        subcmd = "/list"
+        query = query.replace("/list", "", 1).strip()
+    
+    if "/add_model" in query:
+        subcmd = "/add_model"
+        query = query.replace("/add_model", "", 1).strip()
+    
+    if "/add" in query:
+        subcmd = "/add"
+        query = query.replace("/add", "", 1).strip()
+    
+    if "/remove" in query:
+        subcmd = "/remove"
+        query = query.replace("/remove", "", 1).strip()
+    if not subcmd:
+        console.print(get_message("models_usage"))
+        return    
+
+    if subcmd == "/list":                    
+        if models_data:
+            table = Table(title=get_message("models_title"))
+            table.add_column("Name", style="cyan")
+            table.add_column("Model Type", style="green")
+            table.add_column("Model Name", style="magenta")             
+            table.add_column("Description", style="white")
+            for m in models_data:
+                # Check if api_key_path exists and file exists
+                api_key_path = m.get("api_key_path", "")
+                name = m.get("name", "")
+                if api_key_path:
+                    api_key_file = os.path.expanduser(f"~/.auto-coder/keys/{api_key_path}")
+                    if os.path.exists(api_key_file):
+                        name = f"{name}*"
+                
+                table.add_row(
+                    name,
+                    m.get("model_type", ""),
+                    m.get("model_name", ""),                    
+                    m.get("description", "")
+                )
+            console.print(table)
+        else:
+            console.print(f"[yellow]{get_message('models_no_models')}[/yellow]")
+
+    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)
+            if result:
+                console.print(f"[green]{get_message('models_added').format(name=name)}[/green]")
+            else:
+                console.print(f"[red]{get_message('models_add_failed').format(name=name)}[/red]")
+        else:
+            console.print(f"[red]{get_message('models_add_usage')}[/red]")
+
+    elif subcmd == "/add_model":
+        # Parse key=value pairs: /models /add_model name=abc base_url=http://xx ...       
+        # Collect key=value pairs
+        kv_pairs = shlex.split(query)
+        data_dict = {}
+        for pair in kv_pairs:
+            if '=' not in pair:
+                console.print(f"[red]Invalid parameter: {pair}, should be key=value[/red]")
+                continue
+            k, v = pair.split('=', 1)
+            data_dict[k.strip()] = v.strip()
+
+        # Name is required
+        if "name" not in data_dict:
+            console.print(f"[red]{get_message('models_add_model_name_required')}[/red]")
+            return
+
+        # Check duplication
+        if any(m["name"] == data_dict["name"] for m in models_data):
+            console.print(f"[yellow]{get_message('models_add_model_exists').format(name=data_dict['name'])}[/yellow]")
+            return
+
+        # Create model with defaults
+        final_model = {
+            "name": data_dict["name"],
+            "model_type": data_dict.get("model_type", "saas/openai"),
+            "model_name": data_dict.get("model_name", data_dict["name"]),
+            "base_url": data_dict.get("base_url", "https://api.openai.com/v1"),
+            "api_key_path": data_dict.get("api_key_path", "api.openai.com"),
+            "description": data_dict.get("description", "")
+        }
+
+        models_data.append(final_model)
+        models.save_models(models_data)
+        console.print(f"[green]{get_message('models_add_model_success').format(name=data_dict['name'])}[/green]")
+
+    elif subcmd == "/remove":
+        args = query.strip().split(" ")
+        if len(args) < 1:
+            console.print(f"[red]{get_message('models_add_usage')}[/red]")
+            return
+        name = args[0]
+        filtered_models = [m for m in models_data if m["name"] != name]
+        if len(filtered_models) == len(models_data):
+            console.print(f"[yellow]{get_message('models_add_model_remove').format(name=name)}[/yellow]")
+            return
+        models.save_models(filtered_models)
+        console.print(f"[green]{get_message('models_add_model_removed').format(name=name)}[/green]")
+
+    else:
+        console.print(f"[yellow]{get_message('models_unknown_subcmd').format(subcmd=subcmd)}[/yellow]")
 
 def exclude_dirs(dir_names: List[str]):
     new_dirs = dir_names
@@ -2372,11 +2479,19 @@ def lib_command(args: List[str]):
 
 def main():
     ARGS = parse_arguments()
+    
+    if ARGS.lite:
+        ARGS.product_mode = "lite"
+    
+    if ARGS.pro:
+        ARGS.product_mode = "pro" 
 
     if not ARGS.quick:
-        initialize_system()
-
+        initialize_system(ARGS)    
+    
     load_memory()
+    
+    configure(f"product_mode:{ARGS.product_mode}")
 
     MODES = {
         "normal": "normal",
@@ -2434,6 +2549,8 @@ def main():
             memory["mode"] = "normal"
         mode = memory["mode"]
         human_as_model = memory["conf"].get("human_as_model", "false")
+        if mode not in MODES:
+            mode = "normal"
         return f" Mode: {MODES[mode]} (ctl+k) | Human as Model: {human_as_model} (ctl+n or /conf human_as_model:true/false)"
 
     session = PromptSession(
@@ -2528,6 +2645,13 @@ def main():
             elif user_input.startswith("/list_files"):
                 list_files()
 
+            elif user_input.startswith("/models"):
+                query = user_input[len("/models"):].strip()
+                if not query:
+                    print("Please enter your query.")
+                else:
+                    manage_models(ARGS,query)    
+
             elif user_input.startswith("/mode"):
                 conf = user_input[len("/mode"):].strip()
                 if not conf:
@@ -2606,7 +2730,7 @@ def main():
                     result = eval(code)
                     print(f"Debug result: {result}")
                 except Exception as e:
-                    print(f"Debug error: {str(e)}")
+                    print(f"Debug error: {str(e)}")            
 
             # elif user_input.startswith("/shell"):
             else:

autocoder/chat_auto_coder_lang.py

@@ -66,6 +66,21 @@ MESSAGES = {
         "exit_desc": "Exit the program",
         "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_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]",
+        "models_add_model_params": "Please provide parameters in key=value format",
+        "models_add_model_name_required": "'name' parameter is required",
+        "models_add_model_exists": "Model '{name}' already exists.",
+        "models_add_model_success": "Successfully added custom model: {name}",
+        "models_add_model_remove": "Model '{name}' not found.",
+        "models_add_model_removed": "Removed model: {name}",
+        "models_unknown_subcmd": "Unknown subcommand: {subcmd}",
+        "models_title": "All Models (内置 + models.json)",
+        "models_no_models": "No models found.",
+        "models_lite_only": "The /models command is only available in lite mode"
     },
     "zh": {
         "initializing": "🚀 正在初始化系统...",
@@ -132,7 +147,21 @@ MESSAGES = {
         "exit_desc": "退出程序",
         "design_desc": "根据需求设计SVG图片",
         "commit_desc": "根据用户人工修改的代码自动生成yaml文件并提交更改",
-
+        "models_desc": "管理模型配置,仅在lite模式下可用",
+        "models_usage": "用法: /models /list|/add|/add_model|/remove ...",
+        "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]",
+        "models_add_model_params": "请提供 key=value 格式的参数",
+        "models_add_model_name_required": "缺少必需的 'name' 参数",
+        "models_add_model_exists": "模型 '{name}' 已存在。",
+        "models_add_model_success": "成功添加自定义模型: {name}",
+        "models_add_model_remove": "找不到模型 '{name}'。",
+        "models_add_model_removed": "已移除模型: {name}",
+        "models_unknown_subcmd": "未知的子命令: {subcmd}",
+        "models_title": "所有模型 (内置 + models.json)",
+        "models_no_models": "未找到任何模型。",
+        "models_lite_only": "/models 命令仅在 lite 模式下可用"
     }
 }
 

autocoder/common/__init__.py

@@ -351,7 +351,8 @@ class AutoCoderArgs(pydantic.BaseModel):
     generate_times_same_model: Optional[int] = 1
 
     action: List[str] = []
-    enable_global_memory: Optional[bool] = True    
+    enable_global_memory: Optional[bool] = True  
+    product_mode: Optional[str] = "lite"
 
     class Config:
         protected_namespaces = ()

autocoder/common/code_auto_generate.py

@@ -6,6 +6,7 @@ from autocoder.utils.queue_communicate import queue_communicate, CommunicateEven
 from autocoder.common import sys_prompt
 from concurrent.futures import ThreadPoolExecutor
 from autocoder.common.types import CodeGenerateResult
+from autocoder.common.utils_code_auto_generate import chat_with_continue
 
 
 class CodeAutoGenerate:
@@ -187,34 +188,47 @@ class CodeAutoGenerate:
 
         conversations_list = []
         results = []
+        input_tokens_count = 0
+        generated_tokens_count = 0
         if not self.args.human_as_model:
             with ThreadPoolExecutor(max_workers=len(self.llms) * self.generate_times_same_model) as executor:
                 futures = []
                 for llm in self.llms:
                     for _ in range(self.generate_times_same_model):
                         futures.append(executor.submit(
-                            llm.chat_oai, conversations=conversations, llm_config=llm_config))
-                results = [future.result()[0].output for future in futures]
+                            chat_with_continue, llm=llm, conversations=conversations, llm_config=llm_config))
+                temp_results = [future.result() for future in futures]
+                for result in temp_results:
+                    results.append(result.content)
+                    input_tokens_count += result.input_tokens_count
+                    generated_tokens_count += result.generated_tokens_count
+            
             for result in results:
                 conversations_list.append(
                     conversations + [{"role": "assistant", "content": result}])
         else:            
             for _ in range(self.args.human_model_num):
-                v = self.llms[0].chat_oai(
-                    conversations=conversations, llm_config=llm_config)
-                results.append(v[0].output)
-                conversations_list.append(conversations + [{"role": "assistant", "content": v[0].output}])
-
+                single_result = chat_with_continue(llm=self.llms[0], conversations=conversations, llm_config=llm_config)                
+                results.append(single_result.content)
+                input_tokens_count += single_result.input_tokens_count
+                generated_tokens_count += single_result.generated_tokens_count
+                conversations_list.append(conversations + [{"role": "assistant", "content": single_result.content}])
+        
+        statistics = {
+            "input_tokens_count": input_tokens_count,
+            "generated_tokens_count": generated_tokens_count
+        }        
+    
         if self.args.request_id and not self.args.skip_events:
             queue_communicate.send_event_no_wait(
                 request_id=self.args.request_id,
                 event=CommunicateEvent(
                     event_type=CommunicateEventType.CODE_GENERATE_END.value,
-                    data="",
+                    data=json.dumps(statistics, ensure_ascii=False),
                 ),
             )
 
-        return CodeGenerateResult(contents=results, conversations=conversations_list)
+        return CodeGenerateResult(contents=results, conversations=conversations_list, metadata=statistics)
 
     def multi_round_run(
         self, query: str, source_content: str, max_steps: int = 10