autocoder-nano 0.1.28__py3-none-any.whl → 0.1.29__py3-none-any.whl

autocoder_nano/edit/actions.py

@@ -1,6 +1,6 @@
 import time
 
-from loguru import logger
+# from loguru import logger
 
 from autocoder_nano.edit.code.generate_editblock import CodeAutoGenerateEditBlock
 from autocoder_nano.edit.code.merge_editblock import CodeAutoMergeEditBlock
@@ -8,6 +8,10 @@ from autocoder_nano.index.entry import build_index_and_filter_files
 from autocoder_nano.llm_client import AutoLLM
 from autocoder_nano.llm_types import AutoCoderArgs
 from autocoder_nano.project import PyProject, SuffixProject
+from autocoder_nano.utils.printer_utils import Printer
+
+
+printer = Printer()
 
 
 class BaseAction:
@@ -39,13 +43,14 @@ class ActionPyProject(BaseAction):
         if self.args.execute and self.llm:
             content_length = self._get_content_length(content)
             if content_length > self.args.model_max_input_length:
-                logger.warning(
-                    f"发送给模型的内容长度为 {content_length} 个 token（可能收集了过多文件），"
-                    f"已超过最大输入长度限制 {self.args.model_max_input_length}。"
+                printer.print_text(
+                    f"发送给模型的内容长度为 {content_length} 个 token(可能收集了过多文件),"
+                    f"已超过最大输入长度限制 {self.args.model_max_input_length}.",
+                    style="yellow"
                 )
 
         if self.args.execute:
-            logger.info("正在自动生成代码...")
+            printer.print_text("正在自动生成代码...", style="green")
             start_time = time.time()
             # diff, strict_diff, editblock 是代码自动生成或合并的不同策略, 通常用于处理代码的变更或生成
             # diff 模式,基于差异生成代码,生成最小的变更集,适用于局部优化,代码重构
@@ -60,10 +65,10 @@ class ActionPyProject(BaseAction):
                 generate_result = generate.multi_round_run(query=self.args.query, source_content=content)
             else:
                 generate_result = generate.single_round_run(query=self.args.query, source_content=content)
-            logger.info(f"代码生成完成，耗时 {time.time() - start_time:.2f} 秒")
+            printer.print_text(f"代码生成完毕，耗时 {time.time() - start_time:.2f} 秒", style="green")
 
             if self.args.auto_merge:
-                logger.info("正在自动合并代码...")
+                printer.print_text("正在自动合并代码...", style="green")
                 if self.args.auto_merge == "editblock":
                     code_merge = CodeAutoMergeEditBlock(args=self.args, llm=self.llm)
                     merge_result = code_merge.merge_code(generate_result=generate_result)
@@ -97,13 +102,14 @@ class ActionSuffixProject(BaseAction):
         if self.args.execute and self.llm:
             content_length = self._get_content_length(content)
             if content_length > self.args.model_max_input_length:
-                logger.warning(
-                    f"发送给模型的内容长度为 {content_length} 个 token（可能收集了过多文件），"
-                    f"已超过最大输入长度限制 {self.args.model_max_input_length}。"
+                printer.print_text(
+                    f"发送给模型的内容长度为 {content_length} 个 token(可能收集了过多文件),"
+                    f"已超过最大输入长度限制 {self.args.model_max_input_length}.",
+                    style="yellow"
                 )
 
         if self.args.execute:
-            logger.info("正在自动生成代码...")
+            printer.print_text("正在自动生成代码...", style="green")
             start_time = time.time()
             # diff, strict_diff, editblock 是代码自动生成或合并的不同策略, 通常用于处理代码的变更或生成
             # diff 模式,基于差异生成代码,生成最小的变更集,适用于局部优化,代码重构
@@ -118,10 +124,10 @@ class ActionSuffixProject(BaseAction):
                 generate_result = generate.multi_round_run(query=self.args.query, source_content=content)
             else:
                 generate_result = generate.single_round_run(query=self.args.query, source_content=content)
-            logger.info(f"代码生成完成，耗时 {time.time() - start_time:.2f} 秒")
+            printer.print_text(f"代码生成完毕，耗时 {time.time() - start_time:.2f} 秒", style="green")
 
             if self.args.auto_merge:
-                logger.info("正在自动合并代码...")
+                printer.print_text("正在自动合并代码...", style="green")
                 if self.args.auto_merge == "editblock":
                     code_merge = CodeAutoMergeEditBlock(args=self.args, llm=self.llm)
                     merge_result = code_merge.merge_code(generate_result=generate_result)

autocoder_nano/edit/code/merge_editblock.py

@@ -17,33 +17,29 @@ from autocoder_nano.llm_client import AutoLLM
 from autocoder_nano.llm_prompt import prompt
 from autocoder_nano.llm_types import AutoCoderArgs, PathAndCode, MergeCodeWithoutEffect, CodeGenerateResult, \
     CommitResult
+from autocoder_nano.utils.printer_utils import Printer
 
 
-console = Console()
+printer = Printer()
+console = printer.get_console()
+# console = Console()
 
 
 def git_print_commit_info(commit_result: CommitResult):
-    table = Table(
-        title="Commit Information (Use /revert to revert this commit)", show_header=True, header_style="bold magenta"
-    )
-    table.add_column("Attribute", style="cyan", no_wrap=True)
-    table.add_column("Value", style="green")
-
-    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))
-
-    console.print(
-        Panel(table, expand=False, border_style="green", title="Git Commit Summary")
+    printer.print_table_compact(
+        data=[
+            ["Commit Hash", commit_result.commit_hash],
+            ["Commit Message", commit_result.commit_message],
+            ["Changed Files", "\n".join(commit_result.changed_files)]
+        ],
+        title="Commit 信息", headers=["Attribute", "Value"], caption="(Use /revert to revert this commit)"
     )
 
     if commit_result.diffs:
         for file, diff in commit_result.diffs.items():
-            console.print(f"\n[bold blue]File: {file}[/bold blue]")
+            printer.print_text(f"File: {file}", style="green")
             syntax = Syntax(diff, "diff", theme="monokai", line_numbers=True)
-            console.print(
-                Panel(syntax, expand=False, border_style="yellow", title="File Diff")
-            )
+            printer.print_panel(syntax, title="File Diff", center=True)
 
 
 class CodeAutoMergeEditBlock:
@@ -79,12 +75,12 @@ class CodeAutoMergeEditBlock:
             os.unlink(temp_file_path)
             if result.returncode != 0:
                 error_message = result.stdout.strip() or result.stderr.strip()
-                logger.warning(f"Pylint 检查代码失败: {error_message}")
+                printer.print_text(f"Pylint 检查代码失败: {error_message}", style="yellow")
                 return False, error_message
             return True, ""
         except subprocess.CalledProcessError as e:
             error_message = f"运行 Pylint 时发生错误: {str(e)}"
-            logger.error(error_message)
+            printer.print_text(error_message, style="red")
             os.unlink(temp_file_path)
             return False, error_message
 
@@ -257,7 +253,7 @@ class CodeAutoMergeEditBlock:
     def choose_best_choice(self, generate_result: CodeGenerateResult) -> CodeGenerateResult:
         """ 选择最佳代码 """
         if len(generate_result.contents) == 1:  # 仅一份代码立即返回
-            logger.info("仅有一个候选结果，跳过排序")
+            printer.print_text("仅有一个候选结果，跳过排序", style="green")
             return generate_result
 
         ranker = CodeModificationRanker(args=self.args, llm=self.llm)
@@ -339,7 +335,8 @@ class CodeAutoMergeEditBlock:
                         }
                     )
                 return
-            logger.warning(f"发现 {len(unmerged_blocks)} 个未合并的代码块，更改将不会应用，请手动检查这些代码块后重试。")
+            printer.print_text(f"发现 {len(unmerged_blocks)} 个未合并的代码块，更改将不会应用，请手动检查这些代码块后重试.",
+                               style="yellow")
             self._print_unmerged_blocks(unmerged_blocks)
             return
 
@@ -348,16 +345,15 @@ class CodeAutoMergeEditBlock:
             if file_path.endswith(".py"):
                 pylint_passed, error_message = self.run_pylint(new_content)
                 if not pylint_passed:
-                    logger.warning(f"代码文件 {file_path} 的 Pylint 检查未通过，本次更改未应用。错误信息: {error_message}")
+                    printer.print_text(
+                        f"代码文件 {file_path} 的 Pylint 检查未通过，本次更改未应用。错误信息: {error_message}",
+                        style="yellow")
 
         if changes_made and not force_skip_git and not self.args.skip_commit:
             try:
                 commit_changes(self.args.source_dir, f"auto_coder_pre_{file_name}_{md5}")
             except Exception as e:
-                logger.error(
-                    self.git_require_msg(
-                        source_dir=self.args.source_dir, error=str(e))
-                )
+                printer.print_text(self.git_require_msg(source_dir=self.args.source_dir, error=str(e)), style="red")
                 return
         # Now, apply the changes
         for file_path, new_content in file_content_mapping.items():
@@ -385,17 +381,14 @@ class CodeAutoMergeEditBlock:
                     commit_result = commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
                     git_print_commit_info(commit_result=commit_result)
                 except Exception as e:
-                    logger.error(
-                        self.git_require_msg(
-                            source_dir=self.args.source_dir, error=str(e)
-                        )
-                    )
-            logger.info(
-                f"已在 {len(file_content_mapping.keys())} 个文件中合并更改，"
-                f"完成 {len(changes_to_make)}/{len(codes)} 个代码块。"
+                    printer.print_text(self.git_require_msg(source_dir=self.args.source_dir, error=str(e)),
+                                       style="red")
+            printer.print_text(
+                f"已在 {len(file_content_mapping.keys())} 个文件中合并更改,完成 {len(changes_to_make)}/{len(codes)} 个代码块.",
+                style="green"
             )
         else:
-            logger.warning("未对任何文件进行更改。")
+            printer.print_text("未对任何文件进行更改.", style="yellow")
 
     def merge_code(self, generate_result: CodeGenerateResult, force_skip_git: bool = False):
         result = self.choose_best_choice(generate_result)
@@ -404,15 +397,14 @@ class CodeAutoMergeEditBlock:
 
     @staticmethod
     def _print_unmerged_blocks(unmerged_blocks: List[tuple]):
-        console.print(f"\n[bold red]未合并的代码块:[/bold red]")
+        printer.print_text("未合并的代码块:", style="yellow")
         for file_path, head, update, similarity in unmerged_blocks:
-            console.print(f"\n[bold blue]文件:[/bold blue] {file_path}")
-            console.print(
-                f"\n[bold green]搜索代码块（相似度：{similarity}）:[/bold green]")
+            printer.print_text(f"文件: {file_path}", style="yellow")
+            printer.print_text(f"搜索代码块(相似度：{similarity}):", style="yellow")
             syntax = Syntax(head, "python", theme="monokai", line_numbers=True)
-            console.print(Panel(syntax, expand=False))
-            console.print("\n[bold yellow]替换代码块:[/bold yellow]")
+            printer.print_panel(syntax)
+            printer.print_text(f"替换代码块:", style="yellow")
             syntax = Syntax(update, "python", theme="monokai",
                             line_numbers=True)
-            console.print(Panel(syntax, expand=False))
-        console.print(f"\n[bold red]未合并的代码块总数: {len(unmerged_blocks)}[/bold red]")
+            printer.print_panel(syntax)
+        printer.print_text(f"未合并的代码块总数: {len(unmerged_blocks)}", style="yellow")

autocoder_nano/git_utils.py

@@ -2,9 +2,13 @@ import os
 
 from autocoder_nano.llm_prompt import prompt
 from git import Repo, GitCommandError
-from loguru import logger
+# from loguru import logger
 
 from autocoder_nano.llm_types import CommitResult
+from autocoder_nano.utils.printer_utils import Printer
+
+
+printer = Printer()
 
 
 def repo_init(repo_path: str) -> bool:
@@ -12,14 +16,14 @@ def repo_init(repo_path: str) -> bool:
         os.makedirs(repo_path)
 
     if os.path.exists(os.path.join(repo_path, ".git")):
-        logger.warning(f"目录 {repo_path} 已是一个 Git 仓库，跳过初始化操作。")
+        printer.print_text(f"目录 {repo_path} 已是一个 Git 仓库，跳过初始化操作.", style="yellow")
         return False
     try:
         Repo.init(repo_path)
-        logger.info(f"已在 {repo_path} 初始化新的 Git 仓库")
+        printer.print_text(f"已在 {repo_path} 初始化新的 Git 仓库.", style="green")
         return True
     except GitCommandError as e:
-        logger.error(f"Git 初始化过程中发生错误: {e}")
+        printer.print_text(f"Git 初始化过程中发生错误: {e}.", style="red")
         return False
 
 
@@ -72,19 +76,19 @@ def commit_changes(repo_path: str, message: str) -> CommitResult:
 def revert_changes(repo_path: str, message: str) -> bool:
     repo = get_repo(repo_path)
     if repo is None:
-        logger.error("仓库未初始化。")
+        printer.print_text("仓库未初始化.", style="red")
         return False
 
     try:
         # 检查当前工作目录是否有未提交的更改
         if repo.is_dirty():
-            logger.warning("工作目录有未提交的更改，请在回退前提交或暂存您的修改。")
+            printer.print_text("工作目录有未提交的更改，请在回退前提交或暂存您的修改.", style="yellow")
             return False
 
         # 通过message定位到commit_hash
         commit = repo.git.log("--all", f"--grep={message}", "--format=%H", "-n", "1")
         if not commit:
-            logger.warning(f"未找到提交信息包含 '{message}' 的提交记录。")
+            printer.print_text(f"未找到提交信息包含 '{message}' 的提交记录.", style="yellow")
             return False
 
         commit_hash = commit
@@ -94,25 +98,25 @@ def revert_changes(repo_path: str, message: str) -> bool:
 
         if not commits:
             repo.git.revert(commit, no_edit=True)
-            logger.info(f"已回退单条提交记录: {commit}")
+            printer.print_text(f"已回退单条提交记录: {commit}", style="green")
         else:
             # 从最新的提交开始，逐个回滚
             for commit in reversed(commits):
                 try:
                     repo.git.revert(commit.hexsha, no_commit=True)
-                    logger.info(f"已回退提交 {commit.hexsha} 的更改")
+                    printer.print_text(f"已回退提交 {commit.hexsha} 的更改", style="green")
                 except GitCommandError as e:
-                    logger.error(f"回退提交 {commit.hexsha} 时发生错误: {e}")
+                    printer.print_text(f"回退提交 {commit.hexsha} 时发生错误: {e}", style="red")
                     repo.git.revert("--abort")
                     return False
             # 提交所有的回滚更改
             repo.git.commit(message=f"Reverted all changes up to {commit_hash}")
-        logger.info(f"已成功回退到提交 {commit_hash} 的状态")
+        printer.print_text(f"已成功回退到提交 {commit_hash} 的状态", style="green")
         # this is a mark, chat_auto_coder.py need this
         print(f"Successfully reverted changes", flush=True)
         return True
     except GitCommandError as e:
-        logger.error(f"回退操作过程中发生错误: {e}")
+        printer.print_text(f"回退操作过程中发生错误: {e}", style="red")
         return False
 
 
@@ -154,7 +158,7 @@ def get_uncommitted_changes(repo_path: str) -> str:
                     content = f.read()
                 changes['new'].append((file_path, f'+++ {file_path}\n{content}'))
             except Exception as e:
-                logger.error(f"Error reading file {file_path}: {e}")
+                printer.print_text(f"Error reading file {file_path}: {e}", style="red")
         # 生成markdown报告
         report = ["# Git Changes Report\n"]
         # 新增文件
@@ -187,7 +191,7 @@ def get_uncommitted_changes(repo_path: str) -> str:
 
         return "\n".join(report)
     except GitCommandError as e:
-        logger.error(f"Error getting uncommitted changes: {e}")
+        printer.print_text(f"Error getting uncommitted changes: {e}", style="red")
         return f"Error: {str(e)}"
 
 

autocoder_nano/index/entry.py

@@ -2,15 +2,17 @@ import os
 import time
 from typing import List, Dict
 
-from loguru import logger
-from rich.console import Console
-from rich.table import Table
+# from loguru import logger
+# from rich.console import Console
+# from rich.table import Table
 
 from autocoder_nano.index.index_manager import IndexManager
 from autocoder_nano.llm_types import SourceCode, TargetFile, VerifyFileRelevance, AutoCoderArgs
 from autocoder_nano.llm_client import AutoLLM
+from autocoder_nano.utils.printer_utils import Printer
 
-console = Console()
+
+printer = Printer()
 
 
 def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: List[SourceCode]) -> str:
@@ -20,7 +22,7 @@ def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: Lis
         return _file_path
 
     final_files: Dict[str, TargetFile] = {}
-    logger.info("第一阶段：处理 REST/RAG/Search 资源...")
+    printer.print_text("第一阶段：处理 REST/RAG/Search 资源...", style="green")
     for source in sources:
         if source.tag in ["REST", "RAG", "SEARCH"]:
             final_files[get_file_path(source.module_name)] = TargetFile(
@@ -28,14 +30,14 @@ def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: Lis
             )
 
     if not args.skip_build_index and llm:
-        logger.info("第二阶段：为所有文件构建索引...")
+        printer.print_text("第二阶段：为所有文件构建索引...", style="green")
         index_manager = IndexManager(args=args, llm=llm, source_codes=sources)
         index_data = index_manager.build_index()
         indexed_files_count = len(index_data) if index_data else 0
-        logger.info(f"总索引文件数: {indexed_files_count}")
+        printer.print_text(f"总索引文件数: {indexed_files_count}", style="green")
 
         if not args.skip_filter_index and args.index_filter_level >= 1:
-            logger.info("第三阶段：执行 Level 1 过滤(基于查询) ...")
+            printer.print_text("第三阶段：执行 Level 1 过滤(基于查询) ...", style="green")
             target_files = index_manager.get_target_files_by_query(args.query)
             if target_files:
                 for file in target_files.file_list:
@@ -43,7 +45,7 @@ def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: Lis
                     final_files[get_file_path(file_path)] = file
 
             if target_files is not None and args.index_filter_level >= 2:
-                logger.info("第四阶段：执行 Level 2 过滤（基于相关文件）...")
+                printer.print_text("第四阶段：执行 Level 2 过滤（基于相关文件）...", style="green")
                 related_files = index_manager.get_related_files(
                     [file.file_path for file in target_files.file_list]
                 )
@@ -54,29 +56,28 @@ def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: Lis
 
             # 如果 Level 1 filtering 和 Level 2 filtering 都未获取路径，则使用全部文件
             if not final_files:
-                logger.warning("Level 1, Level 2 过滤未找到相关文件, 将使用所有文件 ...")
+                printer.print_text("Level 1, Level 2 过滤未找到相关文件, 将使用所有文件 ...", style="yellow")
                 for source in sources:
                     final_files[get_file_path(source.module_name)] = TargetFile(
                         file_path=source.module_name,
                         reason="No related files found, use all files",
                     )
 
-            logger.info("第五阶段：执行相关性验证 ...")
+            printer.print_text("第五阶段：执行相关性验证 ...", style="green")
             verified_files = {}
             temp_files = list(final_files.values())
             verification_results = []
 
             def _print_verification_results(results):
-                table = Table(title="文件相关性验证结果", expand=True, show_lines=True)
-                table.add_column("文件路径", style="cyan", no_wrap=True)
-                table.add_column("得分", justify="right", style="green")
-                table.add_column("状态", style="yellow")
-                table.add_column("原因/错误")
+                data_list = []
                 if result:
                     for _file_path, _score, _status, _reason in results:
-                        table.add_row(_file_path,
-                                      str(_score) if _score is not None else "N/A", _status, _reason)
-                console.print(table)
+                        data_list.append([_file_path, str(_score) if _score is not None else "N/A", _status, _reason])
+                printer.print_table_compact(
+                    data=data_list,
+                    title="文件相关性验证结果",
+                    headers=["文件路径", "得分", "状态", "原因/错误"]
+                )
 
             def _verify_single_file(single_file: TargetFile):
                 for _source in sources:
@@ -115,12 +116,12 @@ def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: Lis
             # Keep all files, not just verified ones
             final_files = verified_files
 
-    logger.info("第六阶段：筛选文件并应用限制条件 ...")
+    printer.print_text("第六阶段：筛选文件并应用限制条件 ...", style="green")
     if args.index_filter_file_num > 0:
-        logger.info(f"从 {len(final_files)} 个文件中获取前 {args.index_filter_file_num} 个文件(Limit)")
+        printer.print_text(f"> 从 {len(final_files)} 个文件中获取前 {args.index_filter_file_num} 个文件(Limit)", style="green")
     final_filenames = [file.file_path for file in final_files.values()]
     if not final_filenames:
-        logger.warning("未找到目标文件，你可能需要重新编写查询并重试.")
+        printer.print_text("未找到目标文件，你可能需要重新编写查询并重试.", style="yellow")
     if args.index_filter_file_num > 0:
         final_filenames = final_filenames[: args.index_filter_file_num]
 
@@ -135,16 +136,13 @@ def build_index_and_filter_files(args: AutoCoderArgs, llm: AutoLLM, sources: Lis
         return path
 
     def _print_selected(data):
-        table = Table(title="代码上下文文件", expand=True, show_lines=True)
-        table.add_column("文件路径", style="cyan")
-        table.add_column("原因", style="cyan")
-        for _file, _reason in data:
-            # 路径截取优化：保留最后 3 级路径
-            _processed_path = _shorten_path(_file, keep_levels=3)
-            table.add_row(_processed_path, _reason)
-        console.print(table)
-
-    logger.info("第七阶段：准备最终输出 ...")
+        printer.print_table_compact(
+            data=[[_shorten_path(_file, keep_levels=3), _reason] for _file, _reason in data],
+            title="代码上下文文件",
+            headers=["文件路径", "原因"]
+        )
+
+    printer.print_text("第七阶段：准备最终输出 ...", style="green")
     _print_selected(
         [
             (file.file_path, file.reason)

autocoder_nano/index/index_manager.py

@@ -4,12 +4,16 @@ import os
 import time
 from typing import List, Optional
 
-from loguru import logger
+# from loguru import logger
 
 from autocoder_nano.index.symbols_utils import extract_symbols, symbols_info_to_str
 from autocoder_nano.llm_client import AutoLLM
 from autocoder_nano.llm_prompt import prompt
 from autocoder_nano.llm_types import SourceCode, AutoCoderArgs, IndexItem, SymbolType, FileList
+from autocoder_nano.utils.printer_utils import Printer
+
+
+printer = Printer()
 
 
 class IndexManager:
@@ -34,7 +38,7 @@ class IndexManager:
             with open(self.index_file, "r") as file:  # 读缓存
                 index_data = json.load(file)
         else:  # 首次 build index
-            logger.info("首次生成索引.")
+            printer.print_text("首次生成索引.", style="green")
             index_data = {}
 
         @prompt()
@@ -47,7 +51,7 @@ class IndexManager:
 
         for item in index_data.keys():
             if not item.startswith(self.source_dir):
-                logger.warning(error_message(source_dir=self.source_dir, file_path=item))
+                printer.print_text(error_message.prompt(source_dir=self.source_dir, file_path=item), style="yellow")
                 break
 
         updated_sources = []
@@ -60,13 +64,13 @@ class IndexManager:
         counter = 0
         num_files = len(wait_to_build_files)
         total_files = len(self.sources)
-        logger.info(f"总文件数: {total_files}, 需要索引文件数: {num_files}")
+        printer.print_text(f"总文件数: {total_files}, 需要索引文件数: {num_files}", style="green")
 
         for source in wait_to_build_files:
             build_result = self.build_index_for_single_source(source)
             if build_result is not None:
                 counter += 1
-                logger.info(f"正在构建索引:{counter}/{num_files}...")
+                printer.print_text(f"正在构建索引:{counter}/{num_files}...", style="green")
                 module_name = build_result["module_name"]
                 index_data[module_name] = build_result
                 updated_sources.append(module_name)
@@ -168,10 +172,13 @@ class IndexManager:
             start_time = time.monotonic()
             source_code = source.source_code
             if len(source.source_code) > self.max_input_length:
-                logger.warning(
-                    f"警告[构建索引]: 源代码({source.module_name})长度过长 "
-                    f"({len(source.source_code)}) > 模型最大输入长度({self.max_input_length})，"
-                    f"正在分割为多个块..."
+                printer.print_text(
+                    f"""
+                    警告[构建索引]: 源代码({source.module_name})长度过长,
+                    ({len(source.source_code)}) > 模型最大输入长度({self.max_input_length}),
+                    正在分割为多个块...
+                    """,
+                    style="yellow"
                 )
                 chunks = self.split_text_into_chunks(source_code)
                 symbols_list = []
@@ -185,9 +192,10 @@ class IndexManager:
                 symbols = single_symbols.output
                 time.sleep(self.anti_quota_limit)
 
-            logger.info(f"解析并更新索引：文件 {file_path}（MD5: {md5}），耗时 {time.monotonic() - start_time:.2f} 秒")
+            printer.print_text(f"解析并更新索引：{file_path}（MD5: {md5}），耗时 {time.monotonic() - start_time:.2f} 秒",
+                               style="green")
         except Exception as e:
-            logger.warning(f"源文件 {file_path} 处理失败: {e}")
+            printer.print_text(f"源文件 {file_path} 处理失败: {e}", style="yellow")
             return None
 
         return {
@@ -313,11 +321,11 @@ class IndexManager:
                 all_results.extend(result.file_list)
                 completed += 1
             else:
-                logger.warning(f"无法找到分块的目标文件。原因可能是模型响应未返回 JSON 格式数据，或返回的 JSON 为空。")
+                printer.print_text(f"无法找到分块的目标文件.原因可能是模型响应未返回格式错误.", style="yellow")
             total += 1
             time.sleep(self.anti_quota_limit)
 
-        logger.info(f"已完成 {completed}/{total} 个分块(基于查询条件)")
+        printer.print_text(f"已完成 {completed}/{total} 个分块(基于查询条件)", style="green")
         all_results = list({file.file_path: file for file in all_results}.values())
         if self.args.index_filter_file_num > 0:
             limited_results = all_results[: self.args.index_filter_file_num]
@@ -378,10 +386,10 @@ class IndexManager:
                 all_results.extend(result.file_list)
                 completed += 1
             else:
-                logger.warning(f"无法找到与分块相关的文件。原因可能是模型限制或查询条件与文件不匹配。")
+                printer.print_text(f"无法找到与分块相关的文件。原因可能是模型限制或查询条件与文件不匹配.", style="yellow")
             total += 1
             time.sleep(self.anti_quota_limit)
-        logger.info(f"已完成 {completed}/{total} 个分块(基于相关文件)")
+        printer.print_text(f"已完成 {completed}/{total} 个分块(基于相关文件)", style="green")
         all_results = list({file.file_path: file for file in all_results}.values())
         return FileList(file_list=all_results)
 

autocoder_nano/llm_client.py

@@ -1,10 +1,14 @@
 from typing import List
 
-from loguru import logger
+# from loguru import logger
 from openai import OpenAI, Stream
 from openai.types.chat import ChatCompletionChunk, ChatCompletion
 
 from autocoder_nano.llm_types import LLMRequest, LLMResponse
+from autocoder_nano.utils.printer_utils import Printer
+
+
+printer = Printer()
 
 
 class AutoLLM:
@@ -37,7 +41,11 @@ class AutoLLM:
             model = self.default_model_name
 
         model_name = self.sub_clients[model]["model_name"]
-        logger.info(f"正在使用 {model} 模型, 模型名称 {model_name}")
+        printer.print_card(
+            title="模型调用",
+            content=f"调用函数: stream_chat_ai\n使用模型: {model}\n模型名称: {model_name}",
+            width=60
+        )
         request = LLMRequest(
             model=model_name,
             messages=conversations
@@ -57,7 +65,11 @@ class AutoLLM:
             conversations = [{"role": "user", "content": conversations}]
 
         model_name = self.sub_clients[model]["model_name"]
-        logger.info(f"正在使用 {model} 模型, 模型名称 {model_name}")
+        printer.print_card(
+            title="模型调用",
+            content=f"调用函数: chat_ai\n使用模型: {model}\n模型名称: {model_name}",
+            width=60
+        )
         request = LLMRequest(
             model=model_name,
             messages=conversations
@@ -98,7 +110,11 @@ class AutoLLM:
             model = self.default_model_name
 
         model_name = self.sub_clients[model]["model_name"]
-        # logger.info(f"正在使用 {model} 模型, 模型名称 {model_name}")
+        printer.print_card(
+            title="模型调用",
+            content=f"调用函数: embedding\n使用模型: {model}\n模型名称: {model_name}",
+            width=60
+        )
 
         res = self.sub_clients[model]["client"].embeddings.create(
             model=model_name,