auto-coder 0.1.233__tar.gz → 0.1.235__tar.gz

{auto_coder-0.1.233 → auto_coder-0.1.235}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.233
+Version: 0.1.235
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -39,7 +39,6 @@ Requires-Dist: real_agent
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
-Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.233
+Version: 0.1.235
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -39,7 +39,6 @@ Requires-Dist: real_agent
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
-Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/auto_coder.egg-info/SOURCES.txt

@@ -9,7 +9,6 @@ src/auto_coder.egg-info/requires.txt
 src/auto_coder.egg-info/top_level.txt
 src/autocoder/__init__.py
 src/autocoder/auto_coder.py
-src/autocoder/auto_coder_lang.py
 src/autocoder/auto_coder_rag.py
 src/autocoder/auto_coder_rag_client_mcp.py
 src/autocoder/auto_coder_rag_mcp.py
@@ -37,6 +36,7 @@ src/autocoder/common/__init__.py
 src/autocoder/common/anything2images.py
 src/autocoder/common/anything2img.py
 src/autocoder/common/audio.py
+src/autocoder/common/auto_coder_lang.py
 src/autocoder/common/buildin_tokenizer.py
 src/autocoder/common/chunk_validation.py
 src/autocoder/common/cleaner.py
@@ -63,6 +63,7 @@ src/autocoder/common/mcp_hub.py
 src/autocoder/common/mcp_server.py
 src/autocoder/common/mcp_tools.py
 src/autocoder/common/memory_manager.py
+src/autocoder/common/printer.py
 src/autocoder/common/recall_validation.py
 src/autocoder/common/screenshots.py
 src/autocoder/common/search.py

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/auto_coder.egg-info/requires.txt

@@ -29,7 +29,6 @@ real_agent
 python-docx
 docx2txt
 pdf2image
-Spire.Doc
 docx2pdf
 pyperclip
 colorama

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/auto_coder.py

@@ -39,7 +39,7 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.markdown import Markdown
 from rich.live import Live
-from autocoder.auto_coder_lang import get_message
+from autocoder.common.auto_coder_lang import get_message
 from autocoder.common.memory_manager import save_to_memory_file
 from autocoder import models as models_module
 from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
@@ -69,8 +69,7 @@ def load_include_files(config, base_path, max_depth=10, current_depth=0):
             include_files = [include_files]
 
         for include_file in include_files:
-            abs_include_path = resolve_include_path(base_path, include_file)
-            logger.info(f"Loading include file: {abs_include_path}")
+            abs_include_path = resolve_include_path(base_path, include_file)            
             with open(abs_include_path, "r") as f:
                 include_config = yaml.safe_load(f)
                 if not include_config:
@@ -131,15 +130,15 @@ def main(input_args: Optional[List[str]] = None):
     if not os.path.isabs(args.source_dir):
         args.source_dir = os.path.abspath(args.source_dir)
 
-    if not args.silence:
-        print("Command Line Arguments:")
-        print("-" * 50)
-        for arg, value in vars(args).items():
-            if arg == "context" and value:
-                print(f"{arg:20}: {value[:30]}...")
-            else:
-                print(f"{arg:20}: {value}")
-        print("-" * 50)
+    # if not args.silence:
+    #     print("Command Line Arguments:")
+    #     print("-" * 50)
+    #     for arg, value in vars(args).items():
+    #         if arg == "context" and value:
+    #             print(f"{arg:20}: {value[:30]}...")
+    #         else:
+    #             print(f"{arg:20}: {value}")
+    #     print("-" * 50)
 
     # init store
     store = Store(os.path.join(args.source_dir, ".auto-coder", "metadata.db"))
@@ -244,10 +243,7 @@ def main(input_args: Optional[List[str]] = None):
                 new_file = os.path.join(
                     actions_dir, f"{new_seq}_{raw_args.name}.yml")
                 with open(new_file, "w") as f:
-                    f.write(content)
-
-        print(f"Successfully created new action file: {new_file}")
-
+                    f.write(content)        
         # open_yaml_file_in_editor(new_file)
         return
 
@@ -281,6 +277,22 @@ def main(input_args: Optional[List[str]] = None):
                 byzerllm.connect_cluster(address=args.ray_address)
 
             llm = byzerllm.ByzerLLM(verbose=args.print_request)
+
+            code_model = byzerllm.ByzerLLM()
+            code_model.setup_default_model_name("deepseek_chat")
+            llm.setup_sub_client("code_model", code_model)
+
+            index_filter_model = byzerllm.ByzerLLM()
+            index_filter_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("index_filter_model", index_filter_model)
+
+            generate_rerank_model = byzerllm.ByzerLLM()
+            generate_rerank_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("generate_rerank_model", generate_rerank_model)
+
+            chat_model = byzerllm.ByzerLLM()
+            chat_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("chat_model", chat_model)
         
         if args.product_mode == "lite":
             llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
@@ -344,9 +356,23 @@ def main(input_args: Optional[List[str]] = None):
                 }
             )
 
+            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)
 
         if args.product_mode == "lite":                                    
             # Set up default models based on configuration
@@ -444,7 +470,7 @@ def main(input_args: Optional[List[str]] = None):
                         "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
-                llm.setup_sub_client("inference_model", inference_model) 
+                llm.setup_sub_client("inference_model", inference_model)                 
 
             if args.index_filter_model:
                 model_info = models_module.get_model_by_name(args.index_filter_model)
@@ -1114,20 +1140,25 @@ def main(input_args: Optional[List[str]] = None):
                 chat_llm = llm
 
             source_count = 0
-            pre_conversations = []
-            if args.context:
-                context = json.loads(args.context)
-                if "file_content" in context:
-                    file_content = context["file_content"]
-                    pre_conversations.append(
-                        {
-                            "role": "user",
-                            "content": f"请阅读下面的代码和文档：\n\n <files>\n{file_content}\n</files>",
-                        },
-                    )
-                    pre_conversations.append(
-                        {"role": "assistant", "content": "read"})
-                    source_count += 1
+            pre_conversations = []            
+            context_content = args.context if args.context else ""
+            if args.context:                
+                try:
+                    context = json.loads(args.context)
+                    if "file_content" in context:
+                        context_content = context["file_content"]
+                except:
+                    pass                     
+
+                pre_conversations.append(
+                    {
+                        "role": "user",
+                        "content": f"请阅读下面的代码和文档：\n\n <files>\n{context_content}\n</files>",
+                    },
+                )
+                pre_conversations.append(
+                    {"role": "assistant", "content": "read"})
+                source_count += 1
 
             from autocoder.index.index import IndexManager
             from autocoder.index.entry import build_index_and_filter_files
@@ -1142,10 +1173,11 @@ def main(input_args: Optional[List[str]] = None):
             else:
                 pp = SuffixProject(args=args, llm=llm, file_filter=None)
             pp.run()
-            sources = pp.sources
+            sources = pp.sources                       
 
             s = build_index_and_filter_files(
-                llm=llm, args=args, sources=sources)
+                llm=llm, args=args, sources=sources)                        
+            
             if s:
                 pre_conversations.append(
                     {

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/chat_auto_coder.py

@@ -370,6 +370,33 @@ def initialize_system(args):
             "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
+        
+
+        deploy_cmd = [
+            "byzerllm",
+            "deploy",
+            "--pretrained_model_type",
+            "saas/reasoning_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-reasoner",
+            "--model",
+            "deepseek_r1_chat",
+        ]
+
         try:
             subprocess.run(deploy_cmd, check=True)
             print_status(get_message("deploy_complete"), "success")
@@ -1715,7 +1742,8 @@ def coding(query: str):
             == "true",
         }
 
-        yaml_config["context"] = ""
+        yaml_config["context"] = ""        
+        yaml_config["in_code_apply"] = is_apply
 
         for key, value in conf.items():
             converted_value = convert_config_value(key, value)

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/common/__init__.py

@@ -355,5 +355,8 @@ class AutoCoderArgs(pydantic.BaseModel):
     enable_global_memory: Optional[bool] = True  
     product_mode: Optional[str] = "lite"
 
+    in_code_apply: bool = False
+
     class Config:
         protected_namespaces = ()
+

{auto_coder-0.1.233/src/autocoder → auto_coder-0.1.235/src/autocoder/common}/auto_coder_lang.py

@@ -2,6 +2,15 @@ import locale
 
 MESSAGES = {
     "en": {
+        "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_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 }})",
+        "index_related_files_fail": "⚠️ Failed to find related files for chunk {{ chunk_count }}",
+        "index_threads_completed": "✅ Completed {{ completed_threads }}/{{ total_threads }} threads",
+        "index_related_files_fail": "⚠️ Failed to find related files for chunk {{ chunk_count }}",
         "human_as_model_instructions": (
             "You are now in Human as Model mode. The content has been copied to your clipboard.\n"
             "The system is waiting for your input. When finished, enter 'EOF' on a new line to submit.\n"
@@ -16,6 +25,10 @@ MESSAGES = {
             "The system is waiting for your input. When finished, enter 'EOF' on a new line to submit.\n"
             "Use '/break' to exit this mode. If you have issues with copy-paste, use '/clear' to clean and paste again."
         ),
+        "phase1_processing_sources": "Phase 1: Processing REST/RAG/Search sources...",
+        "phase2_building_index": "Phase 2: Building index for all files...",
+        "phase6_file_selection": "Phase 6: Processing file selection and limits...",
+        "phase7_preparing_output": "Phase 7: Preparing final output...",
         "chat_human_as_model_instructions": (
             "Chat is now in Human as Model mode.\n"
             "The question has been copied to your clipboard.\n"
@@ -25,6 +38,15 @@ MESSAGES = {
         )
     },
     "zh": {
+        "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_build_summary": "📊 总文件数: {{ total_files }}, 需要构建索引: {{ num_files }}",
+        "building_index_progress": "⏳ 正在构建索引: {{ counter }}/{{ num_files }}...",
+        "index_source_dir_mismatch": "⚠️ 源目录不匹配 (文件路径: {{ file_path }}, 源目录: {{ source_dir }})",
+        "index_related_files_fail": "⚠️ 无法为块 {{ chunk_count }} 找到相关文件",
+        "index_threads_completed": "✅ 已完成 {{ completed_threads }}/{{ total_threads }} 个线程",
+        "index_related_files_fail": "⚠️ 无法为块 {{ chunk_count }} 找到相关文件",
         "human_as_model_instructions": (
             "您现在处于人类作为模型模式。内容已复制到您的剪贴板。\n"
             "系统正在等待您的输入。完成后，在新行输入'EOF'提交。\n"
@@ -39,22 +61,30 @@ MESSAGES = {
             "系统正在等待您的输入。完成后，在新行输入'EOF'提交。\n"
             "使用'/break'退出此模式。如果复制粘贴有问题，使用'/clear'清理并重新粘贴。"
         ),
+
+        "phase1_processing_sources": "阶段 1: 正在处理 REST/RAG/Search 源...",
+        "phase2_building_index": "阶段 2: 正在为所有文件构建索引...",
+        "phase6_file_selection": "阶段 6: 正在处理文件选择和限制...",
+        "phase7_preparing_output": "阶段 7: 正在准备最终输出...",
+
         "chat_human_as_model_instructions": (
             "\n============= Chat 处于 Human as Model 模式 =============\n"
             "问题已复制到剪贴板\n"
             "请使用Web版本模型获取答案\n"
-            "或者使用 /conf human_as_model:false 关闭该模式直接在终端获得答案。" 
-            "将获得答案黏贴到下面的输入框，换行后，使用 '/break' 退出，'/clear' 清屏，'/eof' 提交。"           
+            "或者使用 /conf human_as_model:false 关闭该模式直接在终端获得答案。"
+            "将获得答案黏贴到下面的输入框，换行后，使用 '/break' 退出，'/clear' 清屏，'/eof' 提交。"
         ),
     }
 }
 
+
 def get_system_language():
     try:
         return locale.getdefaultlocale()[0][:2]
     except:
         return 'en'
 
+
 def get_message(key):
     lang = get_system_language()
-    return MESSAGES.get(lang, MESSAGES['en']).get(key, MESSAGES['en'][key])
+    return MESSAGES.get(lang, MESSAGES['en']).get(key, MESSAGES['en'][key])

auto_coder-0.1.235/src/autocoder/common/printer.py

@@ -0,0 +1,36 @@
+from rich.console import Console
+from typing import Optional
+from byzerllm.utils import format_str_jinja2
+from autocoder.common.auto_coder_lang import get_message
+from autocoder.chat_auto_coder_lang import get_message as get_chat_message
+class Printer:
+    def __init__(self,console:Optional[Console]=None):
+        if console is None:
+            self.console = Console()
+        else:
+            self.console = console
+
+    def get_message_from_key(self, key: str):
+        try:
+            return get_message(key)
+        except Exception as e:
+            return get_chat_message(key)
+
+    def print_in_terminal(self, key: str, style: str = None,**kwargs):     
+        try:
+            if style:
+                self.console.print(format_str_jinja2(self.get_message_from_key(key),**kwargs), style=style)
+            else:
+                self.console.print(format_str_jinja2(self.get_message_from_key(key),**kwargs))
+        except Exception as e:
+            print(self.get_message_from_key(key))
+
+    
+    def print_str_in_terminal(self, content: str, style: str = None):     
+        try:
+            if style:
+                self.console.print(content, style=style)
+            else:
+                self.console.print(content)
+        except Exception as e:
+            print(content)        

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/dispacher/actions/action.py

@@ -54,9 +54,14 @@ class ActionTSProject(BaseAction):
 
         source_code = pp.output()
         if self.llm:
+            if args.in_code_apply:
+                old_query = args.query
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
 
         if args.image_file:
             if args.image_mode == "iterative":
@@ -256,11 +261,17 @@ class ActionPyProject(BaseAction):
         pp = PyProject(args=self.args, llm=self.llm)
         self.pp = pp
         pp.run(packages=args.py_packages.split(",") if args.py_packages else [])            
-        source_code = pp.output()
+        source_code = pp.output()                
+
         if self.llm:
+            old_query = args.query
+            if args.in_code_apply:
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
 
         self.process_content(source_code)
         return True
@@ -355,9 +366,14 @@ class ActionSuffixProject(BaseAction):
         pp.run()
         source_code = pp.output()
         if self.llm:
+            if args.in_code_apply:
+                old_query = args.query
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
         self.process_content(source_code)
 
     def process_content(self, content: str):

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/dispacher/actions/plugins/action_regex_project.py

@@ -36,9 +36,14 @@ class ActionRegexProject:
         pp.run()
         source_code = pp.output()
         if self.llm:
+            if args.in_code_apply:
+                old_query = args.query
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
         self.process_content(source_code)
 
     def process_content(self, content: str):

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/index/entry.py

@@ -9,7 +9,7 @@ from rich.console import Console
 from rich.table import Table
 from rich.panel import Panel
 
-from loguru import logger
+from autocoder.common.printer import Printer
 from autocoder.utils.queue_communicate import (
     queue_communicate,
     CommunicateEvent,
@@ -58,7 +58,8 @@ def build_index_and_filter_files(
     final_files: Dict[str, TargetFile] = {}
 
     # Phase 1: Process REST/RAG/Search sources
-    logger.info("Phase 1: Processing REST/RAG/Search sources...")
+    printer = Printer()
+    printer.print_in_terminal("phase1_processing_sources")
     phase_start = time.monotonic()
     for source in sources:
         if source.tag in ["REST", "RAG", "SEARCH"]:
@@ -79,7 +80,7 @@ def build_index_and_filter_files(
                 )
             )
 
-        logger.info("Phase 2: Building index for all files...")
+        printer.print_in_terminal("phase2_building_index")
         phase_start = time.monotonic()
         index_manager = IndexManager(llm=llm, sources=sources, args=args)
         index_data = index_manager.build_index()
@@ -133,19 +134,48 @@ def build_index_and_filter_files(
 
         return [file for file in result] if result else []
 
+    def shorten_path(path: str, keep_levels: int = 3) -> str:
+        """
+        优化长路径显示，保留最后指定层级
+        示例：/a/b/c/d/e/f.py -> .../c/d/e/f.py
+        """
+        parts = path.split(os.sep)
+        if len(parts) > keep_levels:
+            return ".../" + os.sep.join(parts[-keep_levels:])
+        return path
+
     def print_selected(data):
         console = Console()
+        
+        # 获取终端宽度
+        console_width = console.width
 
         table = Table(
             title="Files Used as Context",
             show_header=True,
             header_style="bold magenta",
+            # 设置表格最大宽度为终端宽度（留 10 字符边距）
+            width=min(console_width - 10, 120),
+            expand=True
         )
-        table.add_column("File Path", style="cyan", no_wrap=True)
-        table.add_column("Reason", style="green")
+        
+        # 优化列配置
+        table.add_column("File Path", 
+                        style="cyan", 
+                        width=int((console_width - 10) * 0.6),  # 分配 60% 宽度给文件路径
+                        overflow="fold",  # 自动折叠过长的路径
+                        no_wrap=False)  # 允许换行
+        
+        table.add_column("Reason", 
+                        style="green", 
+                        width=int((console_width - 10) * 0.4),  # 分配 40% 宽度给原因
+                        no_wrap=False)
 
+        # 添加处理过的文件路径
         for file, reason in data:
-            table.add_row(file, reason)
+            # 路径截取优化：保留最后 3 级路径
+            processed_path = shorten_path(file, keep_levels=3)
+            table.add_row(processed_path, reason)
 
         panel = Panel(
             table,
@@ -157,7 +187,7 @@ def build_index_and_filter_files(
         console.print(panel)
 
     # Phase 6: File selection and limitation
-    logger.info("Phase 6: Processing file selection and limits...")
+    printer.print_in_terminal("phase6_file_selection")
     phase_start = time.monotonic()
 
     if args.index_filter_file_num > 0:
@@ -188,7 +218,7 @@ def build_index_and_filter_files(
     stats["timings"]["file_selection"] = phase_end - phase_start
 
     # Phase 7: Display results and prepare output
-    logger.info("Phase 7: Preparing final output...")
+    printer.print_in_terminal("phase7_preparing_output")
     phase_start = time.monotonic()    
     try:
         print_selected(
@@ -269,7 +299,7 @@ def build_index_and_filter_files(
 • Total time: {total_time:.2f}s
 ====================================
 """
-    logger.info(summary)
+    printer.print_str_in_terminal(summary)
 
     if args.request_id and not args.skip_events:
         queue_communicate.send_event(

{auto_coder-0.1.233 → auto_coder-0.1.235}/src/autocoder/index/filter/quick_filter.py

@@ -55,7 +55,10 @@ class QuickFilter():
         }
         ```
 
-        特别注意,如果用户的query里 @文件 或者 @@符号，那么被@的文件或者@@的符号必须要返回，并且查看他们依赖的文件是否相关。        
+        特别注意
+        1. 如果用户的query里 @文件 或者 @@符号，那么被@的文件或者@@的符号必须要返回，并且查看他们依赖的文件是否相关。        
+        2. 如果 query 里是一段历史对话，那么对话里的内容提及的文件路径必须要返回。
+        3. json格式数据不允许有注释
         '''
         file_meta_str = "\n".join([f"##[{index}]{item.module_name}\n{item.symbols}" for index,item in enumerate(file_meta_list)])
         context = {
@@ -66,15 +69,13 @@ 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:
+        if not self.args.skip_filter_index and self.index_manager.llm.get_sub_client("index_filter_model"):
             start_time = time.monotonic()
             index_items = self.index_manager.read_index()
 
-            prompt_str = self.quick_filter_files.prompt(index_items,query)
-
-            print(prompt_str)
+            prompt_str = self.quick_filter_files.prompt(index_items,query)            
             
-            tokens_len = count_tokens(prompt_str)
+            tokens_len = count_tokens(prompt_str)            
             
             if tokens_len > 55*1024:
                 logger.warning(f"Quick filter prompt is too long, tokens_len: {tokens_len}/{55*1024} fallback to normal filter")