auto-coder 0.1.245__tar.gz → 0.1.246__tar.gz

{auto_coder-0.1.245 → auto_coder-0.1.246}/PKG-INFO

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

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/auto_coder.egg-info/PKG-INFO

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

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/autocoder/auto_coder.py

@@ -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
@@ -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统计            

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/autocoder/chat_auto_coder.py

@@ -2706,9 +2706,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"

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/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 }})",
@@ -54,7 +54,7 @@ MESSAGES = {
         "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)...",
+        "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.",
@@ -93,8 +93,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 }})",
@@ -133,7 +133,7 @@ MESSAGES = {
         "code_merge_start": "正在自动合并代码...",
         "code_execution_warning": "发送给模型的内容长度为 {{ content_length }} tokens（您可能收集了太多文件），超过了最大输入长度 {{ max_length }}",
         "quick_filter_start": "开始查找上下文(quick_filter)...",
-        "normal_filter_start": "开始查找上下文(normal_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 }} 中的源代码建立索引",

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/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)
         

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/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...")

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/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

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/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。

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/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)

{auto_coder-0.1.245 → auto_coder-0.1.246}/src/autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -141,7 +141,8 @@ 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
+    console: Optional[Console] = None,
+    model_name: Optional[str] = None
 ) -> Tuple[str, Optional[Dict[str, Any]]]:
     """
     处理流式输出事件并在终端中展示
@@ -164,7 +165,7 @@ def stream_out(
     
     try:
         with Live(
-            Panel("", title="Response", border_style="green"),
+            Panel("", title=f"Response[ {model_name} ]", border_style="green"),
             refresh_per_second=4,
             console=console
         ) as live:
@@ -209,7 +210,7 @@ def stream_out(
                 live.update(
                     Panel(
                         Markdown(display_content),
-                        title="Response",
+                        title=f"Response[ {model_name} ]",
                         border_style="green",
                         height=min(50, live.console.height - 4)
                     )
@@ -223,7 +224,7 @@ def stream_out(
             live.update(
                 Panel(
                     Markdown(assistant_response),
-                    title="Final Response",
+                    title=f"Final Response[ {model_name} ]",
                     border_style="blue"
                 )
             )
@@ -231,7 +232,7 @@ def stream_out(
     except Exception as e:
         console.print(Panel(
             f"Error: {str(e)}",  
-            title="Error",
+            title=f"Error[ {model_name} ]",
             border_style="red"
         ))
         # import traceback

auto_coder-0.1.246/src/autocoder/version.py

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

auto_coder-0.1.245/src/autocoder/version.py

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