auto-coder 0.1.245__py3-none-any.whl → 0.1.247__py3-none-any.whl

{auto_coder-0.1.245.dist-info → auto_coder-0.1.247.dist-info}/METADATA

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

{auto_coder-0.1.245.dist-info → auto_coder-0.1.247.dist-info}/RECORD

@@ -1,17 +1,17 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=832zmnXmZ7RlqsPnuEj-2IuhsJI0QkCGHJICH6QRCAY,63333
+autocoder/auto_coder.py,sha256=tElv9VAvv1hfb9qt8R9EFbXqhYs6pm89cy4mPIFFUQE,63127
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=illKgzP2bv-Tq50ujsofJnOHdI4pzr0ALtfR8NHHWdQ,22351
 autocoder/auto_coder_rag_client_mcp.py,sha256=WV7j5JUiQge0x4-B7Hp5-pSAFXLbvLpzQMcCovbauIM,6276
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=SLJQzXuQoj_2mcdbANG93ZM2-wIQsptA_h3VRv5xZAQ,105926
-autocoder/chat_auto_coder_lang.py,sha256=gbpjfMd1wYiIrOlLDc-G7eI497mMwjM_ud9GvO-wo9k,15261
+autocoder/chat_auto_coder.py,sha256=OUpoT-kUNoDS5VlO-Tui9zztd2ACyVa_kYQueBhb294,106724
+autocoder/chat_auto_coder_lang.py,sha256=C0QJkyIhW-9aksHznrcATf0JdI1uKqOr46zW4RZ2ljA,15759
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/models.py,sha256=7Z97Hzc_26dZG_wm6M2f9TL1ZxzzIN649U_Z0-m28EU,5342
-autocoder/version.py,sha256=sncGdxYQvG5ZX4oQL4xUEZ96_LS3u1YIOVwACJUeMF4,23
+autocoder/models.py,sha256=_9Kc8oS_tnnqKzRGnybgOfn1NOey7OXZ8y9qhBmgiB4,5517
+autocoder/version.py,sha256=rZZLAPZf903zmHSN4JzYQtRLpVsQe-LUkE0nelZEb6g,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_demand_organizer.py,sha256=NWSAEsEk94vT3lGjfo25kKLMwYdPcpy9e-i21txPasQ,6942
 autocoder/agent/auto_filegroup.py,sha256=CW7bqp0FW1GIEMnl-blyAc2UGT7O9Mom0q66ITz1ckM,6635
@@ -29,12 +29,12 @@ autocoder/common/__init__.py,sha256=2isE_u4VgfogwmcUCnFcussVFlzeNOLHDMFm5z_axbU,
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/anything2img.py,sha256=4TREa-sOA-iargieUy7MpyCYVUE-9Mmq0wJtwomPqnE,7662
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
-autocoder/common/auto_coder_lang.py,sha256=7sr3Dz43ASeWYLtMvkE6tMd8dWDPZBJOxNVQ8rC54Js,13963
+autocoder/common/auto_coder_lang.py,sha256=EBX8EVxoV6MxVFLYd_4kB7ZZyZN5akqBuG6A2jHeiX8,14912
 autocoder/common/buildin_tokenizer.py,sha256=L7d5t39ZFvUd6EoMPXUhYK1toD0FHlRH1jtjKRGokWU,1236
 autocoder/common/chunk_validation.py,sha256=BrR_ZWavW8IANuueEE7hS8NFAwEvm8TX34WnPx_1hs8,3030
 autocoder/common/cleaner.py,sha256=NU72i8C6o9m0vXExab7nao5bstBUsfJFcj11cXa9l4U,1089
 autocoder/common/code_auto_execute.py,sha256=4KXGmiGObr_B1d6tzV9dwS6MifCSc3Gm4j2d6ildBXQ,6867
-autocoder/common/code_auto_generate.py,sha256=5lEW9iudGYQIcd_QjrGyGaEmrWuyZ625PNVF6XX2rNs,10308
+autocoder/common/code_auto_generate.py,sha256=6rrtdEz8JhlKNutEqlWbgKUlACk90Lcm0UebtuAKEQ0,10320
 autocoder/common/code_auto_generate_diff.py,sha256=o5yeqpc3WXSRWlcLzhlwJNosKo7dcj0CeIsFh6Aibus,16248
 autocoder/common/code_auto_generate_editblock.py,sha256=QdUHUkGaervvQNCY8T2vQ_tfnQX_2kxxu4qq_QW_Nn8,17828
 autocoder/common/code_auto_generate_strict_diff.py,sha256=uteWDEHfIbrnVgwKgqC7qwrIeW0enJCXcHzZGa48yY8,14774
@@ -42,13 +42,13 @@ autocoder/common/code_auto_merge.py,sha256=8dtnz61l0B5gNbQmx26TZ4_jD825dsnnWtAFD
 autocoder/common/code_auto_merge_diff.py,sha256=yocfe8s3Pz6hTGDUl9wRIewY3NcTize_gEla64lsGT0,15331
 autocoder/common/code_auto_merge_editblock.py,sha256=sp7C0fZJMVcNgI8uWy43CKDk7gGXFTkMB9kbP2VdY8k,17485
 autocoder/common/code_auto_merge_strict_diff.py,sha256=9rm0NJ_n6M3LohEX7xl1Jym0xmm8UEYqj_ZTSO3oSlM,9519
-autocoder/common/code_modification_ranker.py,sha256=l0OAR7ad0hTV3xdfn7rO0KqnY5Y5qSG4BmWcClZ9RUQ,6104
+autocoder/common/code_modification_ranker.py,sha256=TycYdtxfNfde2N9-9jEB-9IL2Q3PoHS-k-r9JqZYD6s,6545
 autocoder/common/command_completer.py,sha256=SSeb8MDH0JPvfdyW-S2uaHnui4VBDfSQvQPLbv3ORPA,9314
 autocoder/common/command_generator.py,sha256=v4LmU7sO-P7jEZIXCWHUC6P-vT7AvBi_x_PTwCqBAE8,1323
 autocoder/common/command_templates.py,sha256=mnB3n8i0yjH1mqzyClEg8Wpr9VbZV44kxky66Zu6OJY,8557
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
 autocoder/common/files.py,sha256=CguxG9digkWBJpRaILErZmL_G5ryPRahPmPFWGB7X18,1973
-autocoder/common/git_utils.py,sha256=btK45sxvfm4tX3fBRNUPRZoGQuZuOEQrWSAwLy1yoLw,23095
+autocoder/common/git_utils.py,sha256=zxgQt2PukabV_21podylAUzTY7Xk60bsQ7MQYw4s-Tg,23234
 autocoder/common/image_to_page.py,sha256=O0cNO_vHHUP-fP4GXiVojShmNqkPnZXeIyiY1MRLpKg,13936
 autocoder/common/interpreter.py,sha256=62-dIakOunYB4yjmX8SHC0Gdy2h8NtxdgbpdqRZJ5vk,2833
 autocoder/common/llm_rerank.py,sha256=FbvtCzaR661Mt2wn0qsuiEL1Y3puD6jeIJS4zg_e7Bs,3260
@@ -78,14 +78,14 @@ autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=ht_HWzZt84IEogoFMggnXI6aFFerrsuksVflAkcodfU,5545
 autocoder/dispacher/actions/plugins/action_translate.py,sha256=nVAtRSQpdGNmZxg1R_9zXG3AuTv3CHf2v7ODgj8u65c,7727
 autocoder/index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/index/entry.py,sha256=oFdSJW4ypc3_mhE3fk2O9UB34XMcq3a1Sp2MeTlfQ2o,11820
+autocoder/index/entry.py,sha256=hXSOi-jPgRBKQB55eqWkT95vxvWrbVHDuIMtDkqjNNw,12232
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
-autocoder/index/index.py,sha256=VjfcBYHywU4tjQTA7mpHfzRM8nBPhPHrUnkuBbsj6do,20409
+autocoder/index/index.py,sha256=xwh22nY0TtEJMJwYjOUd6xdRZozYXBt47YSwheZSP-4,20679
 autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6uo-0,2037
 autocoder/index/types.py,sha256=a2s_KV5FJlq7jqA2ELSo9E1sjuLwDB-JJYMhSpzBAhU,596
 autocoder/index/filter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/index/filter/normal_filter.py,sha256=V0MAUKgEG9vVTwZK5lMfpZCjU57S6cBioeHDjog0kLs,7992
-autocoder/index/filter/quick_filter.py,sha256=Omvsz9O1xQEH4xP-wNuCZhxn69P7Y59SiLPUIDuGFiA,3851
+autocoder/index/filter/normal_filter.py,sha256=APu34iSvWhtlLtWgkj8N3Vo4oW1TegtZQq2bwDX_cs4,8031
+autocoder/index/filter/quick_filter.py,sha256=k0c3liMC6M7tP2rGorZU1DJu2mqQlc_URf3BAsSWqcY,5490
 autocoder/pyproject/__init__.py,sha256=dQ2_7YZ7guybT9BhfxSGn43eLQJGQN2zgeKa6--JlaQ,14403
 autocoder/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7288
@@ -121,7 +121,7 @@ autocoder/rag/stream_event/types.py,sha256=rtLwOE8rShmi1dJdxyBpAV5ZjLBGG9vptMiSz
 autocoder/regex_project/__init__.py,sha256=EBZeCL5ORyD_9_5u_UuG4s7XtpXOu0y1sWDmxWFtufE,6781
 autocoder/regexproject/__init__.py,sha256=cEr-ZOaQjLD5sx7T7F2DhD5ips03HcJ02rded9EpSXc,9693
 autocoder/suffixproject/__init__.py,sha256=VcXjUbGf3uQrpoqVCItDvGG9DoeHJ_qEmghKwrVNw9w,11058
-autocoder/tsproject/__init__.py,sha256=yloVzkGLnbTd4Hcj9fMO-rcjNTTx4wI3Ga41LWOSYrY,11747
+autocoder/tsproject/__init__.py,sha256=boNuRCHi94xI_y4tvL5LKgSZ4gYxcPqUUQTw9MU_STI,11751
 autocoder/utils/__init__.py,sha256=KtcGElFNBgZPF7dEL8zF9JpXkCAjoyDrzaREJBhJrcs,994
 autocoder/utils/_markitdown.py,sha256=RU88qn4eZfYIy0GDrPxlI8oYXIypbi63VRJjdlnE0VU,47431
 autocoder/utils/coder.py,sha256=rK8e0svQBe0NOP26dIGToUXgha_hUDgxlWoC_p_r7oc,5698
@@ -139,11 +139,11 @@ autocoder/utils/rest.py,sha256=hLBhr78y-WVnV0oQf9Rxc22EwqF78KINkScvYa1MuYA,6435
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
 autocoder/utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=8OV1VOdbj8O7JXDrBUhXVPmMkoCd9n-hvshtR2XXYxk,9112
+autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=5MCtQwQDttvr0pRcYlYERFQpQAdi1Yvyf5Hm6F-_jvQ,9433
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.245.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.245.dist-info/METADATA,sha256=OtK6YNCP5q11TSBcEWYhfDeljzTN7OqNYW4ze_mggmk,2616
-auto_coder-0.1.245.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.245.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.245.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.245.dist-info/RECORD,,
+auto_coder-0.1.247.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.247.dist-info/METADATA,sha256=hS2KU58cmjBGHdqYM4e--IBprziX3RJUeLDBZnZEySA,2616
+auto_coder-0.1.247.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.247.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.247.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.247.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -276,7 +276,7 @@ def main(input_args: Optional[List[str]] = None):
                 )
                 byzerllm.connect_cluster(address=args.ray_address)
 
-            llm = byzerllm.ByzerLLM(verbose=args.print_request)
+            llm = byzerllm.ByzerLLM(verbose=args.print_request)            
 
             # code_model,index_filter_model,generate_rerank_model,chat_model
             # 这四个模型如果用户没有设置，就会使用默认的
@@ -300,86 +300,66 @@ def main(input_args: Optional[List[str]] = None):
             llm.setup_sub_client("chat_model", chat_model)
         
         if args.product_mode == "lite":
-            default_model = args.model
+            default_model = args.model        
+            model_info = models_module.get_model_by_name(default_model)                
             llm = byzerllm.SimpleByzerLLM(default_model_name=default_model)
-            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):                
-                raise Exception(f"API key file not found: {api_key_file}")
-            
-            with open(api_key_file, "r") as f:
-                api_key = f.read()
-            
             llm.deploy(
                 model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name=default_model,
+                pretrained_model_type=model_info["model_type"],
+                udf_name=args.model,
                 infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-chat",
-                    "saas.is_reasoning": False
+                    "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"]
                 }
-            )
-            
-            code_llm = byzerllm.SimpleByzerLLM(default_model)
-            code_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name=default_model,
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-chat",
-                    "saas.is_reasoning": False
-                }
-            )
-
-            chat_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-            chat_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_r1_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner",
-                    "saas.is_reasoning": True
-                }
-            )
+            )                
+                        
+            if models_module.check_model_exists("deepseek_r1_chat"):
+                r1_model_info = models_module.get_model_by_name("deepseek_r1_chat")  
+                api_key = r1_model_info["api_key"]
+                chat_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+                chat_llm.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name="deepseek_r1_chat",
+                    infer_params={
+                        "saas.base_url": "https://api.deepseek.com/v1",
+                        "saas.api_key": api_key,
+                        "saas.model": "deepseek-reasoner",
+                        "saas.is_reasoning": True
+                    }
+                )
 
-            generate_rerank_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-            generate_rerank_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_r1_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner",
-                    "saas.is_reasoning": True
-                }
-            )
+                generate_rerank_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+                generate_rerank_llm.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name="deepseek_r1_chat",
+                    infer_params={
+                        "saas.base_url": "https://api.deepseek.com/v1",
+                        "saas.api_key": api_key,
+                        "saas.model": "deepseek-reasoner",
+                        "saas.is_reasoning": True
+                    }
+                )
 
-            index_filter_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-            index_filter_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_r1_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner",
-                    "saas.is_reasoning": True
-                }
-            )
-            
-            # 这四个模型如果用户没有设置，就会使用默认的
-            llm.setup_sub_client("code_model", code_llm)
-            llm.setup_sub_client("chat_model", chat_llm)
-            llm.setup_sub_client("generate_rerank_model", generate_rerank_llm)
-            llm.setup_sub_client("index_filter_model", index_filter_llm)
+                index_filter_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+                index_filter_llm.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name="deepseek_r1_chat",
+                    infer_params={
+                        "saas.base_url": "https://api.deepseek.com/v1",
+                        "saas.api_key": api_key,
+                        "saas.model": "deepseek-reasoner",
+                        "saas.is_reasoning": True
+                    }
+                )
+                                                
+                llm.setup_sub_client("chat_model", chat_llm)
+                llm.setup_sub_client("generate_rerank_model", generate_rerank_llm)
+                llm.setup_sub_client("index_filter_model", index_filter_llm)
 
         if args.product_mode == "lite":                                    
             # Set up default models based on configuration
@@ -1359,7 +1339,7 @@ def main(input_args: Optional[List[str]] = None):
             elif "review_commit" in args.action:
                 from autocoder.agent.auto_review_commit import AutoReviewCommit
                 reviewer = AutoReviewCommit(llm=chat_llm, args=args)
-                v = reviewer.review_commit(args.query)
+                v = reviewer.review_commit(query=args.query,conversations=loaded_conversations)
             else:                
                 v = stream_chat_with_continue(
                     llm=chat_llm,
@@ -1367,12 +1347,16 @@ def main(input_args: Optional[List[str]] = None):
                     llm_config={}
                 )              
 
-                
+
+            model_name = getattr(chat_llm, 'default_model_name', None)
+            if not model_name:
+                model_name = "unknown(without default model name)"    
             
             assistant_response, last_meta = stream_out(
                     v, 
                     request_id=args.request_id,                    
-                    console=console
+                    console=console,
+                    model_name=model_name
                 )
                                                                        
             # 打印耗时和token统计            

autocoder/chat_auto_coder.py

@@ -1712,14 +1712,24 @@ def commit(query: str):
             finally:
                 if os.path.exists(temp_yaml):
                     os.remove(temp_yaml)
-
-            llm = get_single_llm(args.code_model or args.model, product_mode)
-            uncommitted_changes = git_utils.get_uncommitted_changes(".")
-            commit_message = git_utils.generate_commit_message.with_llm(llm).run(
-                uncommitted_changes
-            )
-            memory["conversation"].append(
-                {"role": "user", "content": commit_message})
+            
+            target_model = args.code_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)
+            commit_message = ""
+            
+            try:
+                uncommitted_changes = git_utils.get_uncommitted_changes(".")
+                commit_message = git_utils.generate_commit_message.with_llm(llm).run(
+                    uncommitted_changes
+                )                
+                memory["conversation"].append(
+                    {"role": "user", "content": commit_message})
+            except Exception as e:
+                printer.print_in_terminal("commit_failed", style="red", error=str(e), model_name=target_model)
+                return
+            
             yaml_config["query"] = commit_message
             yaml_content = convert_yaml_config_to_str(yaml_config=yaml_config)
             with open(os.path.join(execute_file), "w") as f:
@@ -1732,6 +1742,8 @@ def commit(query: str):
                 ".", f"auto_coder_{file_name}_{md5}"
             )
             git_utils.print_commit_info(commit_result=commit_result)
+            if commit_message:
+                printer.print_in_terminal("commit_message", style="green", model_name=target_model, message=commit_message)                
         except Exception as e:
             import traceback
             traceback.print_exc()            
@@ -1875,8 +1887,8 @@ def coding(query: str):
 @byzerllm.prompt()
 def code_review(query: str) -> str:
     """
-    对代码进行review，参考如下检查点。
-    1. 有没有调用不符合方法，类的签名的调用
+    掐面提供了上下文，对代码进行review，参考如下检查点。
+    1. 有没有调用不符合方法，类的签名的调用，包括对第三方类，模块，方法的检查（如果上下文提供了这些信息）
     2. 有没有未声明直接使用的变量，方法，类
     3. 有没有明显的语法错误
     4. 如果是python代码，检查有没有缩进方面的错误
@@ -2706,9 +2718,10 @@ def main():
                 memory["mode"] = "normal"
 
             # 处理 user_input 的空格
-            temp_user_input = user_input.lstrip()  # 去掉左侧空格
-            if temp_user_input.startswith('/'):
-                user_input = temp_user_input
+            if user_input:
+                temp_user_input = user_input.lstrip()  # 去掉左侧空格
+                if temp_user_input.startswith('/'):
+                    user_input = temp_user_input
 
             if (
                 memory["mode"] == "auto_detect"

autocoder/chat_auto_coder_lang.py

@@ -113,9 +113,15 @@ MESSAGES = {
         "remove_files_none": "No files were removed.",
         "files_removed": "Files Removed",
         "models_api_key_empty": "Warning : {{name}} API key is empty. Please set a valid API key.",
+        "commit_generating": "{{ model_name }} Generating commit message...",
+        "commit_message": "{{ model_name }} Generated commit message: {{ message }}",
+        "commit_failed": "{{ model_name }} Failed to generate commit message: {{ error }}",
     },
     "zh": {
-        "mcp_remove_error": "移除 MCP 服务器时出错：{error}",
+        "commit_generating": "{{ model_name }} 正在生成提交信息...", 
+        "commit_message": "{{ model_name }} 生成的提交信息: {{ message }}",
+        "commit_failed": "{{ model_name }} 生成提交信息失败: {{ error }}",
+        "mcp_remove_error": "移除 MCP 服务器时出错:{error}",
         "mcp_remove_success": "成功移除 MCP 服务器：{result}",
         "mcp_list_running_error": "列出运行中的 MCP 服务器时出错：{error}",
         "mcp_list_running_title": "正在运行的 MCP 服务器：",

autocoder/common/auto_coder_lang.py

@@ -14,8 +14,8 @@ MESSAGES = {
         "no_latest_commit": "Unable to get latest commit information",
         "code_review_error": "Code review process error: {{error}}",
         "index_file_too_large": "⚠️ File {{ file_path }} is too large ({{ file_size }} > {{ max_length }}), splitting into chunks...",
-        "index_update_success": "✅ Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s",
-        "index_build_error": "❌ Error building index for {{ file_path }}: {{ error }}",
+        "index_update_success": "✅ {{ model_name }} Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s",
+        "index_build_error": "❌ {{ model_name }} Error building index for {{ file_path }}: {{ error }}",
         "index_build_summary": "📊 Total Files: {{ total_files }}, Need to Build Index: {{ num_files }}",
         "building_index_progress": "⏳ Building Index: {{ counter }}/{{ num_files }}...",
         "index_source_dir_mismatch": "⚠️ Source directory mismatch (file_path: {{ file_path }}, source_dir: {{ source_dir }})",
@@ -53,8 +53,8 @@ MESSAGES = {
         "code_generation_complete": "Code generation completed in {{ duration }} seconds, input_tokens_count: {{ input_tokens }}, generated_tokens_count: {{ output_tokens }}",
         "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": "Starting filter context(quick_filter)...",
-        "normal_filter_start": "Starting filter context(normal_filter)...",
+        "quick_filter_start": "{{ model_name }} Starting filter context(quick_filter)...",
+        "normal_filter_start": "{{ model_name }} Starting filter context(normal_filter)...",
         "pylint_check_failed": "⚠️ Pylint check failed: {{ error_message }}",
         "pylint_error": "❌ Error running pylint: {{ error_message }}",
         "unmerged_blocks_warning": "⚠️ Found {{ num_blocks }} unmerged blocks, the changes will not be applied. Please review them manually then try again.",
@@ -65,7 +65,7 @@ MESSAGES = {
         "merge_failed": "❌ Merge file {{ path }} failed: {{ error }}",
         "files_merged_total": "✅ Merged {{ total }} files into the project.",
         "ranking_skip": "Only 1 candidate, skip ranking",
-        "ranking_start": "Start ranking {{ count }} candidates",
+        "ranking_start": "Start ranking {{ count }} candidates using model {{ model_name }}",
         "ranking_failed_request": "Ranking request failed: {{ error }}",
         "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 }}",
@@ -73,13 +73,17 @@ MESSAGES = {
         "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 }}",
+        "quick_filter_stats": "快速过滤器完成，耗时 {{ elapsed_time }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}",
         "upsert_file": "✅ Updated file: {{ file_path }}",  
-        "unmerged_blocks_title": "Unmerged Blocks",                     
+        "unmerged_blocks_title": "Unmerged Blocks",
+        "quick_filter_title": "{{ model_name }} is analyzing how to filter context...",
+        "quick_filter_failed": "❌ Quick filter failed: {{ error }}. ",
         "unmerged_file_path": "File: {{file_path}}",
         "unmerged_search_block": "Search Block({{similarity}}):",
         "unmerged_replace_block": "Replace Block:",
         "unmerged_blocks_total": "Total unmerged blocks: {{num_blocks}}",
-        "git_init_required": "⚠️ auto_merge only applies to git repositories.\n\nPlease try using git init in the source directory:\n\n```shell\ncd {{ source_dir }}\ngit init.\n```\n\nThen run auto - coder again.\nError: {{ error }}"
+        "git_init_required": "⚠️ auto_merge only applies to git repositories.\n\nPlease try using git init in the source directory:\n\n```shell\ncd {{ source_dir }}\ngit init.\n```\n\nThen run auto - coder again.\nError: {{ error }}",
+        "quick_filter_reason": "Auto get(quick_filter mode)"
     },
     "zh": {
         "model_not_found": "未找到模型: {{model_name}}",
@@ -93,8 +97,8 @@ MESSAGES = {
         "no_latest_commit": "无法获取最新的提交信息",
         "code_review_error": "代码审查过程出错: {{error}}",
         "index_file_too_large": "⚠️ 文件 {{ file_path }} 过大 ({{ file_size }} > {{ max_length }}), 正在分块处理...",
-        "index_update_success": "✅ 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒",
-        "index_build_error": "❌ 构建 {{ file_path }} 索引时出错: {{ error }}",
+        "index_update_success": "✅ {{ model_name }} 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒",
+        "index_build_error": "❌ {{ model_name }} 构建 {{ file_path }} 索引时出错: {{ error }}",
         "index_build_summary": "📊 总文件数: {{ total_files }}, 需要构建索引: {{ num_files }}",
         "building_index_progress": "⏳ 正在构建索引: {{ counter }}/{{ num_files }}...",
         "index_source_dir_mismatch": "⚠️ 源目录不匹配 (文件路径: {{ file_path }}, 源目录: {{ source_dir }})",
@@ -132,8 +136,8 @@ MESSAGES = {
         "code_generation_complete": "代码生成完成，耗时 {{ duration }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}",
         "code_merge_start": "正在自动合并代码...",
         "code_execution_warning": "发送给模型的内容长度为 {{ content_length }} tokens（您可能收集了太多文件），超过了最大输入长度 {{ max_length }}",
-        "quick_filter_start": "开始查找上下文(quick_filter)...",
-        "normal_filter_start": "开始查找上下文(normal_filter)...",
+        "quick_filter_start": "{{ model_name }} 开始查找上下文(quick_filter)...",
+        "normal_filter_start": "{{ model_name }} 开始查找上下文(normal_filter)...",
         "pylint_check_failed": "⚠️ Pylint 检查失败: {{ error_message }}",
         "pylint_error": "❌ 运行 Pylint 时出错: {{ error_message }}",
         "begin_index_source_code": "🚀 开始为 {{ source_dir }} 中的源代码建立索引",
@@ -146,19 +150,23 @@ MESSAGES = {
         "unmerged_search_block": "Search Block({{similarity}}):",
         "unmerged_replace_block": "Replace Block:",
         "unmerged_blocks_total": "未合并代码块数量: {{num_blocks}}",
-        "git_init_required": "⚠️ auto_merge 仅适用于 git 仓库。\n\n请尝试在源目录中使用 git init：\n\n```shell\ncd {{ source_dir }}\ngit init.\n```\n\n然后再次运行 auto-coder。\n错误: {{ error }}",
+        "git_init_required": "⚠️ auto_merge 仅适用于 git 仓库。\n\n请尝试在源目录中使用 git init:\n\n```shell\ncd {{ source_dir }}\ngit init.\n```\n\n然后再次运行 auto-coder。\n错误: {{ error }}",
+        "quick_filter_reason": "自动获取(quick_filter模式)",
         "upsert_file": "✅ 更新文件: {{ file_path }}",
         "files_merged": "✅ 成功合并了 {{ total }} 个文件到项目中。",
         "merge_failed": "❌ 合并文件 {{ path }} 失败: {{ error }}",
         "files_merged_total": "✅ 合并了 {{ total }} 个文件到项目中。",
         "ranking_skip": "只有1个候选项，跳过排序",
-        "ranking_start": "开始对 {{ count }} 个候选项进行排序",
+        "ranking_start": "开始对 {{ count }} 个候选项进行排序,使用模型 {{ model_name }} 打分",
         "ranking_failed_request": "排序请求失败: {{ error }}",
         "ranking_all_failed": "所有排序请求都失败",
         "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数: {{ input_tokens }}, 输出token数: {{ output_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 }}. ",
     },
 }
 

autocoder/common/code_auto_generate.py

@@ -7,6 +7,7 @@ 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
+import json
 
 
 class CodeAutoGenerate:

autocoder/common/code_modification_ranker.py

@@ -77,8 +77,15 @@ class CodeModificationRanker:
             with ThreadPoolExecutor(max_workers=total_tasks) as executor:
                 # Submit tasks for each model and generate_times
                 futures = []
-                for llm in self.llms:
+                for llm in self.llms:                    
+                    model_name = getattr(llm, 'default_model_name', None)                    
+                    if not model_name:
+                        model_name = "unknown(without default model name)"
+                    self.printer.print_in_terminal(
+                        "ranking_start", style="blue", count=len(generate_result.contents), model_name=model_name)
+                    
                     for _ in range(rank_times):
+                        
                         futures.append(
                             executor.submit(
                                 chat_with_continue,

autocoder/common/git_utils.py

@@ -262,7 +262,9 @@ def get_uncommitted_changes(repo_path: str) -> str:
 def generate_commit_message(changes_report: str) -> str:
     '''
     我是一个Git提交信息生成助手。我们的目标是通过一些变更报告，倒推用户的需求，将需求作为commit message。
-    commit message 需要简洁，不要超过100个字符。
+    commit message 需要简洁,包含两部分：
+    1. 这个commit 背后的需求是什么
+    2. 为了完成这个需求做了哪些事情
 
     下面是一些示例：
     <examples>
@@ -613,7 +615,7 @@ def print_commit_info(commit_result: CommitResult):
 
     table.add_row("Commit Hash", commit_result.commit_hash)
     table.add_row("Commit Message", commit_result.commit_message)
-    table.add_row("Changed Files", "\n".join(commit_result.changed_files))
+    table.add_row("Changed Files", "\n".join(commit_result.changed_files) if commit_result.changed_files else "No files changed")
 
     console.print(
         Panel(table, expand=False, border_style="green", title="Git Commit Summary")

autocoder/index/entry.py

@@ -101,13 +101,19 @@ def build_index_and_filter_files(
                 )
             )
 
-        if not args.skip_filter_index and args.index_filter_model:       
-            printer.print_in_terminal("quick_filter_start", style="blue")
+        if not args.skip_filter_index and args.index_filter_model:
+            model_name = getattr(index_manager.index_filter_llm, 'default_model_name', None)
+            if not model_name:
+                model_name = "unknown(without default model name)"    
+            printer.print_in_terminal("quick_filter_start", style="blue", model_name=model_name)
             quick_filter = QuickFilter(index_manager,stats,sources)        
             final_files = quick_filter.filter(index_manager.read_index(),args.query)
         
         if not args.skip_filter_index and not args.index_filter_model:
-            printer.print_in_terminal("normal_filter_start", style="blue")
+            model_name = getattr(index_manager.llm, 'default_model_name', None)
+            if not model_name:
+                model_name = "unknown(without default model name)" 
+            printer.print_in_terminal("normal_filter_start", style="blue",model_name=model_name)
             normal_filter = NormalFilter(index_manager,stats,sources)
             final_files = normal_filter.filter(index_manager.read_index(),args.query)
         

autocoder/index/filter/normal_filter.py

@@ -60,7 +60,7 @@ class NormalFilter():
             phase_end = time.monotonic()
             self.stats["timings"]["normal_filter"]["level1_filter"] = phase_end - phase_start
 
-            # Phase 4: Level 2 filtering - Related files
+            # Phase 4: Level 2 filtering - Related files                        
             if target_files is not None and self.args.index_filter_level >= 2:
                 logger.info(
                     "Phase 4: Performing Level 2 filtering (related files)...")
@@ -84,13 +84,14 @@ class NormalFilter():
                 phase_end = time.monotonic()
                 self.stats["timings"]["normal_filter"]["level2_filter"] = phase_end - phase_start
 
-            if not final_files:
-                logger.warning("No related files found, using all files")
-                for source in self.sources:
-                    final_files[get_file_path(source.module_name)] = TargetFile(
-                        file_path=source.module_name,
-                        reason="No related files found, use all files",
-                    )
+            # if not final_files:
+            #     logger.warning("No related files found, using all files")
+            #     for source in self.sources:
+            #         final_files[get_file_path(source.module_name)] = TargetFile(
+            #             file_path=source.module_name,
+            #             reason="No related files found, use all files",
+            #         )
+
 
             # Phase 5: Relevance verification
             logger.info("Phase 5: Performing relevance verification...")

autocoder/index/filter/quick_filter.py

@@ -1,4 +1,7 @@
-from typing import List, Union,Dict,Any
+from typing import List, Union, Dict, Any
+from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
+from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
+from byzerllm.utils.str2model import to_model
 from autocoder.index.types import IndexItem
 from autocoder.common import AutoCoderArgs,SourceCode
 import byzerllm
@@ -11,6 +14,7 @@ from autocoder.index.types import (
 )
 from autocoder.rag.token_counter import count_tokens
 from loguru import logger
+from autocoder.common.printer import Printer
 
 
 def get_file_path(file_path):
@@ -25,6 +29,7 @@ class QuickFilter():
         self.args = index_manager.args
         self.stats = stats
         self.sources = sources
+        self.printer = Printer()
 
     @byzerllm.prompt()
     def quick_filter_files(self,file_meta_list:List[IndexItem],query:str) -> str:
@@ -69,6 +74,7 @@ class QuickFilter():
 
     def filter(self, index_items: List[IndexItem], query: str) -> Dict[str, TargetFile]:
         final_files: Dict[str, TargetFile] = {}
+        
         if not self.args.skip_filter_index and self.args.index_filter_model:
             start_time = time.monotonic()
             index_items = self.index_manager.read_index()
@@ -82,17 +88,47 @@ class QuickFilter():
                 return final_files
             
             try:
-                file_number_list = self.quick_filter_files.with_llm(
-                    self.index_manager.index_filter_llm).with_return_type(FileNumberList).run(index_items, self.args.query)
+                model_name = getattr(self.index_manager.index_filter_llm, 'default_model_name', None)
+                if not model_name:
+                    model_name = "unknown(without default model name)"
+                
+                # 渲染 Prompt 模板
+                query = self.quick_filter_files.prompt(index_items, self.args.query)
+                
+                # 使用流式输出处理
+                stream_generator = stream_chat_with_continue(
+                    self.index_manager.index_filter_llm,
+                    [{"role": "user", "content": query}],
+                    {}
+                )
+                
+                # 获取完整响应
+                full_response, last_meta = stream_out(
+                    stream_generator,
+                    model_name=model_name,
+                    title=self.printer.get_message_from_key_with_format("quick_filter_title", model_name=model_name)
+                )                
+                # 解析结果
+                file_number_list = to_model(full_response, FileNumberList)
+
+                # 打印 token 统计信息
+                self.printer.print_in_terminal(
+                    "quick_filter_stats", 
+                    style="blue",
+                    elapsed_time=f"{end_time - start_time:.2f}",
+                    input_tokens=last_meta.input_tokens_count,
+                    output_tokens=last_meta.generated_tokens_count
+                )
+                
             except Exception as e:
-                logger.error(f"Quick filter failed, error: {str(e)} fallback to normal filter")
+                self.printer.print_error(self.printer.get_message_from_key_with_format("quick_filter_failed", error=str(e)))
                 return final_files
             
             if file_number_list:
                 for file_number in file_number_list.file_list:
                     final_files[get_file_path(index_items[file_number].module_name)] = TargetFile(
                         file_path=index_items[file_number].module_name,
-                        reason="Quick Filter"
+                        reason=self.printer.get_message_from_key("quick_filter_reason")
                     )
             end_time = time.monotonic()            
             self.stats["timings"]["quick_filter"] = end_time - start_time            

autocoder/index/index.py

@@ -195,7 +195,7 @@ class IndexManager:
             return True
         return False
 
-    def build_index_for_single_source(self, source: SourceCode):
+    def build_index_for_single_source(self, source: SourceCode):        
         file_path = source.module_name
         if not os.path.exists(file_path):
             return None
@@ -205,6 +205,10 @@ class IndexManager:
 
         md5 = hashlib.md5(source.source_code.encode("utf-8")).hexdigest()
 
+        model_name = getattr(self.index_llm, 'default_model_name', None)
+        if not model_name:
+            model_name = "unknown(without default model name)"
+
         try:
             start_time = time.monotonic()
             source_code = source.source_code
@@ -230,13 +234,14 @@ class IndexManager:
                 symbols = self.get_all_file_symbols.with_llm(
                     self.index_llm).run(source.module_name, source_code)
                 time.sleep(self.anti_quota_limit)
-
+            
             self.printer.print_in_terminal(
                 "index_update_success",
                 style="green",
                 file_path=file_path,
                 md5=md5,
-                duration=time.monotonic() - start_time
+                duration=time.monotonic() - start_time,
+                model_name=model_name
             )
 
         except Exception as e:
@@ -246,7 +251,8 @@ class IndexManager:
                 "index_build_error",
                 style="red",
                 file_path=file_path,
-                error=str(e)
+                error=str(e),
+                model_name=model_name
             )
             return None
 
@@ -464,7 +470,7 @@ class IndexManager:
             {file.file_path: file for file in all_results}.values())
         return FileList(file_list=all_results)
 
-    def _query_index_with_thread(self, query, func):
+    def _query_index_with_thread(self, query, func):        
         all_results = []
         lock = threading.Lock()
         completed_threads = 0

autocoder/models.py

@@ -113,6 +113,14 @@ def get_model_by_name(name: str) -> Dict:
         raise Exception(get_message_with_format("model_not_found", model_name=name))
     return v[0]
 
+
+def check_model_exists(name: str) -> bool:
+    """
+    检查模型是否存在
+    """
+    models = load_models()
+    return any(m["name"] == name.strip() for m in models)
+
 def update_model_with_api_key(name: str, api_key: str) -> Dict:
     """
     根据模型名称查找并更新模型的 api_key_path。

autocoder/tsproject/__init__.py

@@ -1,5 +1,5 @@
 from autocoder.common import SourceCode, AutoCoderArgs
-from autocoder import common as FileUtils
+from autocoder import common as CommonUtils
 from autocoder.utils.rest import HttpDoc
 import os
 from typing import Optional, Generator, List, Dict, Any
@@ -164,7 +164,7 @@ class TSProject:
             logger.warning(f"Failed to read file: {file_path}. Error: {str(e)}")
             return None
 
-        if not FileUtils.has_sufficient_content(source_code, min_line_count=1):
+        if not CommonUtils.has_sufficient_content(source_code, min_line_count=1):
             return None
 
         return SourceCode(module_name=module_name, source_code=source_code)

autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -9,6 +9,7 @@ from typing import Generator, List, Dict, Any, Optional, Tuple, Literal
 from autocoder.utils.request_queue import RequestValue, RequestOption, StreamValue
 from autocoder.utils.request_queue import request_queue
 import time
+from byzerllm.utils.types import SingleOutputMeta
 
 MAX_HISTORY_LINES = 40  # 最大保留历史行数
 
@@ -141,8 +142,10 @@ def multi_stream_out(
 def stream_out(
     stream_generator: Generator[Tuple[str, Dict[str, Any]], None, None],
     request_id: Optional[str] = None,    
-    console: Optional[Console] = None
-) -> Tuple[str, Optional[Dict[str, Any]]]:
+    console: Optional[Console] = None,
+    model_name: Optional[str] = None,
+    title: Optional[str] = None
+) -> Tuple[str, Optional[SingleOutputMeta]]:
     """
     处理流式输出事件并在终端中展示
     
@@ -150,7 +153,8 @@ def stream_out(
         stream_generator: 生成流式输出的生成器
         request_id: 请求ID,用于更新请求队列        
         console: Rich Console对象
-        
+        model_name: 模型名称
+        title: 面板标题，如果没有提供则使用默认值
     Returns:
         Tuple[str, Dict[str, Any]]: 返回完整的响应内容和最后的元数据
     """
@@ -161,10 +165,10 @@ def stream_out(
     current_line = ""  # 当前行
     assistant_response = ""
     last_meta = None
-    
-    try:
+    panel_title = title if title is not None else f"Response[ {model_name} ]"
+    try:        
         with Live(
-            Panel("", title="Response", border_style="green"),
+            Panel("", title=panel_title, border_style="green"),
             refresh_per_second=4,
             console=console
         ) as live:
@@ -209,7 +213,7 @@ def stream_out(
                 live.update(
                     Panel(
                         Markdown(display_content),
-                        title="Response",
+                        title=panel_title,
                         border_style="green",
                         height=min(50, live.console.height - 4)
                     )
@@ -223,7 +227,7 @@ def stream_out(
             live.update(
                 Panel(
                     Markdown(assistant_response),
-                    title="Final Response",
+                    title=f"Final {panel_title}",
                     border_style="blue"
                 )
             )
@@ -231,7 +235,7 @@ def stream_out(
     except Exception as e:
         console.print(Panel(
             f"Error: {str(e)}",  
-            title="Error",
+            title=f"Error[ {panel_title} ]",
             border_style="red"
         ))
         # import traceback

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.245"
+__version__ = "0.1.247"