auto-coder 0.1.195__py3-none-any.whl → 0.1.197__py3-none-any.whl

autocoder/index/index.py

@@ -24,6 +24,11 @@ from rich.panel import Panel
 from rich.text import Text
 
 from loguru import logger
+from autocoder.utils.queue_communicate import (
+    queue_communicate,
+    CommunicateEvent,
+    CommunicateEventType,
+)
 
 
 class IndexItem(pydantic.BaseModel):
@@ -39,10 +44,12 @@ class TargetFile(pydantic.BaseModel):
         ..., description="The reason why the file is the target file"
     )
 
+
 class VerifyFileRelevance(pydantic.BaseModel):
     relevant_score: int
     reason: str
 
+
 class FileList(pydantic.BaseModel):
     file_list: List[TargetFile]
 
@@ -93,7 +100,7 @@ class IndexManager:
             "reason": "这是相关的原因..."
         }
         ```
-        """        
+        """
 
     @byzerllm.prompt()
     def _get_related_files(self, indices: str, file_paths: str) -> str:
@@ -127,7 +134,7 @@ class IndexManager:
         1. 找到的文件名必须出现在上面的文件列表中
         2. 原因控制在20字以内
         3. 如果没有相关的文件，输出如下 json 即可：
-        
+
         ```json
         {"file_list": []}
         ```
@@ -231,12 +238,14 @@ class IndexManager:
                 )
                 symbols = []
                 for chunk in chunks:
-                    chunk_symbols = self.get_all_file_symbols.with_llm(self.index_llm).run(source.module_name, chunk)
+                    chunk_symbols = self.get_all_file_symbols.with_llm(
+                        self.index_llm).run(source.module_name, chunk)
                     time.sleep(self.anti_quota_limit)
                     symbols.append(chunk_symbols)
                 symbols = "\n".join(symbols)
             else:
-                symbols = self.get_all_file_symbols.with_llm(self.index_llm).run(source.module_name, source_code)
+                symbols = self.get_all_file_symbols.with_llm(
+                    self.index_llm).run(source.module_name, source_code)
                 time.sleep(self.anti_quota_limit)
 
             logger.info(
@@ -286,7 +295,7 @@ class IndexManager:
                     v = source.source_code.splitlines()
                     new_v = []
                     for line in v:
-                        new_v.append(line[line.find(":") :])
+                        new_v.append(line[line.find(":"):])
                     source_code = "\n".join(new_v)
 
                 md5 = hashlib.md5(source_code.encode("utf-8")).hexdigest()
@@ -299,7 +308,8 @@ class IndexManager:
             counter = 0
             num_files = len(wait_to_build_files)
             total_files = len(self.sources)
-            logger.info(f"Total Files: {total_files}, Need to Build Index: {num_files}")
+            logger.info(
+                f"Total Files: {total_files}, Need to Build Index: {num_files}")
 
             futures = [
                 executor.submit(self.build_index_for_single_source, source)
@@ -366,13 +376,13 @@ class IndexManager:
                 item_str = f"##{item.module_name}\n{symbols_str}\n\n"
 
                 if skip_symbols:
-                    item_str = f"{item.module_name}\n"                
+                    item_str = f"{item.module_name}\n"
 
                 if len(current_chunk) > self.args.filter_batch_size:
                     yield "".join(current_chunk)
-                    current_chunk = [item_str]                    
+                    current_chunk = [item_str]
                 else:
-                    current_chunk.append(item_str)                    
+                    current_chunk.append(item_str)
 
             if current_chunk:
                 yield "".join(current_chunk)
@@ -405,7 +415,8 @@ class IndexManager:
         lock = threading.Lock()
 
         def process_chunk(chunk, chunk_count):
-            result = self._get_related_files.with_llm(self.llm).with_return_type(FileList).run(chunk, "\n".join(file_paths))
+            result = self._get_related_files.with_llm(self.llm).with_return_type(
+                FileList).run(chunk, "\n".join(file_paths))
             if result is not None:
                 with lock:
                     all_results.extend(result.file_list)
@@ -419,7 +430,7 @@ class IndexManager:
             futures = []
             chunk_count = 0
             for chunk in self._get_meta_str(
-                max_chunk_size= -1
+                max_chunk_size=-1
             ):
                 future = executor.submit(process_chunk, chunk, chunk_count)
                 futures.append(future)
@@ -428,7 +439,8 @@ class IndexManager:
             for future in as_completed(futures):
                 future.result()
 
-        all_results = list({file.file_path: file for file in all_results}.values())
+        all_results = list(
+            {file.file_path: file for file in all_results}.values())
         return FileList(file_list=all_results)
 
     def _query_index_with_thread(self, query, func):
@@ -439,7 +451,8 @@ class IndexManager:
 
         def process_chunk(chunk):
             nonlocal completed_threads
-            result = self._get_target_files_by_query.with_llm(self.llm).with_return_type(FileList).run(chunk, query)
+            result = self._get_target_files_by_query.with_llm(
+                self.llm).with_return_type(FileList).run(chunk, query)
             if result is not None:
                 with lock:
                     all_results.extend(result.file_list)
@@ -469,24 +482,27 @@ class IndexManager:
         def w():
             return self._get_meta_str(
                 skip_symbols=False,
-                max_chunk_size= -1,
+                max_chunk_size=-1,
                 includes=[SymbolType.USAGE],
             )
-        
-        temp_result, total_threads, completed_threads = self._query_index_with_thread(query, w)        
+
+        temp_result, total_threads, completed_threads = self._query_index_with_thread(
+            query, w)
         all_results.extend(temp_result)
 
         if self.args.index_filter_level >= 1:
 
             def w():
                 return self._get_meta_str(
-                    skip_symbols=False, max_chunk_size= -1
+                    skip_symbols=False, max_chunk_size=-1
                 )
 
-            temp_result, total_threads, completed_threads = self._query_index_with_thread(query, w)            
+            temp_result, total_threads, completed_threads = self._query_index_with_thread(
+                query, w)
             all_results.extend(temp_result)
 
-        all_results = list({file.file_path: file for file in all_results}.values())
+        all_results = list(
+            {file.file_path: file for file in all_results}.values())
         # Limit the number of files based on index_filter_file_num
         limited_results = all_results[: self.args.index_filter_file_num]
         return FileList(file_list=limited_results)
@@ -518,7 +534,7 @@ class IndexManager:
             ]
         }
         ```
-        
+
         如果没有找到，返回如下 json 即可：
 
         ```json
@@ -585,6 +601,15 @@ def build_index_and_filter_files(
 
     if not args.skip_build_index and llm:
         # Phase 2: Build index
+        if args.request_id and not args.skip_events:
+            queue_communicate.send_event(
+                request_id=args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_INDEX_BUILD_START.value,
+                    data=json.dumps({"total_files": len(sources)})
+                )
+            )
+
         logger.info("Phase 2: Building index for all files...")
         phase_start = time.monotonic()
         index_manager = IndexManager(llm=llm, sources=sources, args=args)
@@ -592,9 +617,31 @@ def build_index_and_filter_files(
         stats["indexed_files"] = len(index_data) if index_data else 0
         stats["timings"]["build_index"] = time.monotonic() - phase_start
 
+        if args.request_id and not args.skip_events:
+            queue_communicate.send_event(
+                request_id=args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_INDEX_BUILD_END.value,
+                    data=json.dumps({
+                        "indexed_files": stats["indexed_files"],
+                        "build_index_time": stats["timings"]["build_index"],
+                    })
+                )
+            )
+
         if not args.skip_filter_index:
+            if args.request_id and not args.skip_events:
+                queue_communicate.send_event(
+                    request_id=args.request_id,
+                    event=CommunicateEvent(
+                        event_type=CommunicateEventType.CODE_INDEX_FILTER_START.value,
+                        data=json.dumps({})
+                    )
+                )
             # Phase 3: Level 1 filtering - Query-based
-            logger.info("Phase 3: Performing Level 1 filtering (query-based)...")
+            logger.info(
+                "Phase 3: Performing Level 1 filtering (query-based)...")
+
             phase_start = time.monotonic()
             target_files = index_manager.get_target_files_by_query(args.query)
 
@@ -607,7 +654,16 @@ def build_index_and_filter_files(
 
             # Phase 4: Level 2 filtering - Related files
             if target_files is not None and args.index_filter_level >= 2:
-                logger.info("Phase 4: Performing Level 2 filtering (related files)...")
+                logger.info(
+                    "Phase 4: Performing Level 2 filtering (related files)...")
+                if args.request_id and not args.skip_events:
+                    queue_communicate.send_event(
+                        request_id=args.request_id,
+                        event=CommunicateEvent(
+                            event_type=CommunicateEventType.CODE_INDEX_FILTER_START.value,
+                            data=json.dumps({})
+                        )
+                    )
                 phase_start = time.monotonic()
                 related_files = index_manager.get_related_files(
                     [file.file_path for file in target_files.file_list]
@@ -617,7 +673,8 @@ def build_index_and_filter_files(
                         file_path = file.file_path.strip()
                         final_files[get_file_path(file_path)] = file
                     stats["level2_filtered"] = len(related_files.file_list)
-                stats["timings"]["level2_filter"] = time.monotonic() - phase_start
+                stats["timings"]["level2_filter"] = time.monotonic() - \
+                    phase_start
 
             if not final_files:
                 logger.warning("No related files found, using all files")
@@ -632,37 +689,40 @@ def build_index_and_filter_files(
             phase_start = time.monotonic()
             verified_files = {}
             temp_files = list(final_files.values())
-            
+
             def verify_single_file(file: TargetFile):
                 for source in sources:
                     if source.module_name == file.file_path:
                         file_content = source.source_code
                         try:
                             result = index_manager.verify_file_relevance.with_llm(llm).with_return_type(VerifyFileRelevance).run(
-                                file_content=file_content, 
+                                file_content=file_content,
                                 query=args.query
-                            )                            
+                            )
                             if result.relevant_score >= args.verify_file_relevance_score:
                                 return file.file_path, TargetFile(
                                     file_path=file.file_path,
                                     reason=f"Score:{result.relevant_score}, {result.reason}"
                                 )
                         except Exception as e:
-                            logger.warning(f"Failed to verify file {file.file_path}: {str(e)}")
+                            logger.warning(
+                                f"Failed to verify file {file.file_path}: {str(e)}")
                 return None
 
             with ThreadPoolExecutor(max_workers=args.index_filter_workers) as executor:
-                futures = [executor.submit(verify_single_file, file) for file in temp_files]
+                futures = [executor.submit(verify_single_file, file)
+                           for file in temp_files]
                 for future in as_completed(futures):
                     result = future.result()
                     if result:
                         file_path, target_file = result
                         verified_files[file_path] = target_file
                         time.sleep(args.anti_quota_limit)
-                        
+
             stats["verified_files"] = len(verified_files)
-            stats["timings"]["relevance_verification"] = time.monotonic() - phase_start
-            
+            stats["timings"]["relevance_verification"] = time.monotonic() - \
+                phase_start
+
             final_files = verified_files if verified_files else final_files
 
     def display_table_and_get_selections(data):
@@ -714,12 +774,13 @@ def build_index_and_filter_files(
 
         console.print(panel)
 
-    # Phase 6: File selection and limitation 
+    # Phase 6: File selection and limitation
     logger.info("Phase 6: Processing file selection and limits...")
     phase_start = time.monotonic()
-    
+
     if args.index_filter_file_num > 0:
-        logger.info(f"Limiting files from {len(final_files)} to {args.index_filter_file_num}")
+        logger.info(
+            f"Limiting files from {len(final_files)} to {args.index_filter_file_num}")
 
     if args.skip_confirm:
         final_filenames = [file.file_path for file in final_files.values()]
@@ -735,17 +796,17 @@ def build_index_and_filter_files(
             )
             final_filenames = []
         else:
-            final_filenames = display_table_and_get_selections(target_files_data)
-            
+            final_filenames = display_table_and_get_selections(
+                target_files_data)
+
         if args.index_filter_file_num > 0:
             final_filenames = final_filenames[: args.index_filter_file_num]
-    
+
     stats["timings"]["file_selection"] = time.monotonic() - phase_start
 
     # Phase 7: Display results and prepare output
     logger.info("Phase 7: Preparing final output...")
-    phase_start = time.monotonic()
-    
+    phase_start = time.monotonic()    
     try:
         print_selected(
             [
@@ -772,26 +833,52 @@ def build_index_and_filter_files(
             depulicated_sources.add(file.module_name)
             source_code += f"##File: {file.module_name}\n"
             source_code += f"{file.source_code}\n\n"
-    
+
+    if args.request_id and not args.skip_events:
+        queue_communicate.send_event(
+            request_id=args.request_id,
+            event=CommunicateEvent(
+                event_type=CommunicateEventType.CODE_INDEX_FILTER_FILE_SELECTED.value,
+                data=json.dumps([
+                    (file.file_path, file.reason)
+                    for file in final_files.values()
+                    if file.file_path in depulicated_sources
+                ])
+            )
+        )        
+
     stats["final_files"] = len(depulicated_sources)
     stats["timings"]["prepare_output"] = time.monotonic() - phase_start
-    
+
     # Calculate total time and print summary
     total_time = time.monotonic() - total_start_time
     stats["timings"]["total"] = total_time
-    
+
     # Print final statistics
     logger.info("\n=== Build Index and Filter Files Summary ===")
     logger.info(f"Total files in project: {stats['total_files']}")
     logger.info(f"Files indexed: {stats['indexed_files']}")
     logger.info(f"Files after Level 1 filter: {stats['level1_filtered']}")
     logger.info(f"Files after Level 2 filter: {stats['level2_filtered']}")
-    logger.info(f"Files after relevance verification: {stats.get('verified_files', 0)}")
+    logger.info(
+        f"Files after relevance verification: {stats.get('verified_files', 0)}")
     logger.info(f"Final files selected: {stats['final_files']}")
     logger.info("\nTime breakdown:")
     for phase, duration in stats["timings"].items():
         logger.info(f"  - {phase}: {duration:.2f}s")
     logger.info(f"Total execution time: {total_time:.2f}s")
     logger.info("==========================================\n")
-    
+
+    if args.request_id and not args.skip_events:
+        queue_communicate.send_event(
+            request_id=args.request_id,
+            event=CommunicateEvent(
+                event_type=CommunicateEventType.CODE_INDEX_FILTER_END.value,
+                data=json.dumps({
+                    "filtered_files": stats["final_files"],
+                    "filter_time": stats['level1_filtered'] + stats['level2_filtered'] + stats.get('verified_files', 0)
+                })
+            )
+        )
+
     return source_code

autocoder/lang.py

@@ -71,7 +71,12 @@ lang_desc = {
         "base_dir": "Alternative path for /~/.auto-coder to store or retrieve text embeddings. Using /~/.auto-coder/ if not specified",
         "editblock_similarity": "The similarity threshold of TextSimilarity when merging edit blocks. Default is 0.9",
         "include_project_structure": "Whether to include the project directory structure in the code generation prompt. Default is False",
-        "filter_batch_size": "The batch size used when filtering files. Default is 5"
+        "filter_batch_size": "The batch size used when filtering files. Default is 5",
+        "skip_events": "Skip sending events during execution. Default is False",
+         "rag_url": "The URL of the RAG service. Default is empty",
+        "rag_token": "The token for the RAG service. Default is empty",
+        "rag_type": "RAG type (simple/storage), default is storage",
+        "rag_params_max_tokens": "The maximum number of tokens for RAG parameters. Default is 4096",
     },
     "zh": {
         "request_id": "Request ID",
@@ -145,6 +150,11 @@ lang_desc = {
         "base_dir": "用于替代byzerllm中/~/.auto-coder的路径存放或读取向量化后的文本。不指定则使用默认路径",
         "editblock_similarity": "合并编辑块时TextSimilarity的相似度阈值。默认为0.9",
         "include_project_structure": "在生成代码的提示中是否包含项目目录结构。默认为False",
-        "filter_batch_size": "文件过滤时使用的批处理大小。默认为5"
+        "filter_batch_size": "文件过滤时使用的批处理大小。默认为5",
+        "skip_events": "在执行过程中跳过事件发送。默认为False",
+         "rag_url": "RAG服务的URL",
+        "rag_token": "RAG服务的令牌",
+        "rag_type": "RAG类型(simple/storage),默认是storage",
+        "rag_params_max_tokens": "RAG参数的最大token数。默认为4096",    
     }
 }

autocoder/pyproject/__init__.py

@@ -14,6 +14,9 @@ from autocoder.common.search import Search, SearchEngine
 from loguru import logger
 import re
 from pydantic import BaseModel, Field
+from rich.console import Console
+import json
+from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
 
 
 class RegPattern(BaseModel):
@@ -228,11 +231,37 @@ class PyProject:
     def get_rag_source_codes(self):
         if not self.args.enable_rag_search and not self.args.enable_rag_context:
             return []
+            
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_RAG_SEARCH_START.value,
+                    data=json.dumps({},ensure_ascii=False)
+                )
+            )
+        else:
+            console = Console()
+            console.print(f"\n[bold blue]Starting RAG search for:[/bold blue] {self.args.query}")
+            
         from autocoder.rag.rag_entry import RAGFactory
         rag = RAGFactory.get_rag(self.llm, self.args, "")
         docs = rag.search(self.args.query)
         for doc in docs:
             doc.tag = "RAG"
+            
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_RAG_SEARCH_END.value,
+                    data=json.dumps({},ensure_ascii=False)
+                )
+            )
+        else:
+            console = Console()
+            console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")
+            
         return docs
 
     def get_search_source_codes(self):
@@ -261,7 +290,7 @@ class PyProject:
         return temp + []
 
     def get_source_codes(self) -> Generator[SourceCode, None, None]:
-        for root, dirs, files in os.walk(self.directory):
+        for root, dirs, files in os.walk(self.directory,followlinks=True):
             dirs[:] = [d for d in dirs if d not in self.default_exclude_dirs]
             for file in files:
                 file_path = os.path.join(root, file)

autocoder/rag/cache/byzer_storage_cache.py

@@ -364,7 +364,7 @@ class ByzerStorageCache(BaseCacheManager):
 
     def get_all_files(self) -> List[Tuple[str, str, float]]:
         all_files = []
-        for root, dirs, files in os.walk(self.path):
+        for root, dirs, files in os.walk(self.path,followlinks=True):
             dirs[:] = [d for d in dirs if not d.startswith(".")]
 
             if self.ignore_spec:

autocoder/rag/cache/file_monitor_cache.py

@@ -117,7 +117,7 @@ class AutoCoderRAGDocListener(BaseCacheManager):
 
     def get_all_files(self) -> List[str]:
         all_files = []
-        for root, dirs, files in os.walk(self.path):
+        for root, dirs, files in os.walk(self.path,followlinks=True):
             dirs[:] = [d for d in dirs if not d.startswith(".")]
 
             if self.ignore_spec:

autocoder/rag/cache/simple_cache.py

@@ -174,7 +174,7 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
 
     def get_all_files(self) -> List[Tuple[str, str, float]]:
         all_files = []
-        for root, dirs, files in os.walk(self.path):
+        for root, dirs, files in os.walk(self.path,followlinks=True):
             dirs[:] = [d for d in dirs if not d.startswith(".")]
 
             if self.ignore_spec:
@@ -184,7 +184,7 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
                     for d in dirs
                     if not self.ignore_spec.match_file(os.path.join(relative_root, d))
                 ]
-                files = [
+                files[:] = [
                     f
                     for f in files
                     if not self.ignore_spec.match_file(os.path.join(relative_root, f))

autocoder/rag/long_context_rag.py

@@ -260,6 +260,7 @@ class LongContextRAG:
             response = self.client.chat.completions.create(
                 messages=[{"role": "user", "content": new_query}],
                 model=self.args.model,
+                max_tokens=self.args.rag_params_max_tokens,
             )
             v = response.choices[0].message.content
             if not only_contexts:

autocoder/regexproject/__init__.py

@@ -9,6 +9,9 @@ import byzerllm
 from autocoder.common.search import Search, SearchEngine
 from loguru import logger
 from pydantic import BaseModel, Field
+from rich.console import Console
+import json
+from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
 
 
 class RegPattern(BaseModel):
@@ -90,7 +93,7 @@ class RegexProject:
         return SourceCode(module_name=module_name, source_code=source_code)
 
     def get_source_codes(self) -> Generator[SourceCode, None, None]:
-        for root, dirs, files in os.walk(self.directory):
+        for root, dirs, files in os.walk(self.directory,followlinks=True):
             for file in files:
                 file_path = os.path.join(root, file)
                 if self.is_regex_match(file_path):
@@ -117,11 +120,37 @@ class RegexProject:
     def get_rag_source_codes(self):
         if not self.args.enable_rag_search and not self.args.enable_rag_context:
             return []
+            
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_RAG_SEARCH_START.value,
+                    data=json.dumps({},ensure_ascii=False)
+                )
+            )
+        else:
+            console = Console()
+            console.print(f"\n[bold blue]Starting RAG search for:[/bold blue] {self.args.query}")
+            
         from autocoder.rag.rag_entry import RAGFactory
         rag = RAGFactory.get_rag(self.llm, self.args, "")
         docs = rag.search(self.args.query)
         for doc in docs:
             doc.tag = "RAG"
+            
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_RAG_SEARCH_END.value,
+                    data=json.dumps({},ensure_ascii=False)
+                )
+            )
+        else:
+            console = Console()
+            console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")
+            
         return docs
 
     def get_search_source_codes(self):

autocoder/suffixproject/__init__.py

@@ -9,6 +9,9 @@ from autocoder.common.search import Search, SearchEngine
 from loguru import logger
 import re
 from pydantic import BaseModel, Field
+from rich.console import Console
+import json
+from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
 
 
 class RegPattern(BaseModel):
@@ -128,7 +131,7 @@ class SuffixProject:
         return SourceCode(module_name=module_name, source_code=source_code)
 
     def get_source_codes(self) -> Generator[SourceCode, None, None]:
-        for root, dirs, files in os.walk(self.directory):
+        for root, dirs, files in os.walk(self.directory,followlinks=True):
             dirs[:] = [d for d in dirs if d not in self.default_exclude_dirs]
             for file in files:
                 file_path = os.path.join(root, file)
@@ -157,11 +160,37 @@ class SuffixProject:
     def get_rag_source_codes(self):
         if not self.args.enable_rag_search and not self.args.enable_rag_context:
             return []
+            
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_RAG_SEARCH_START.value,
+                    data=json.dumps({},ensure_ascii=False)
+                )
+            )
+        else:
+            console = Console()
+            console.print(f"\n[bold blue]Starting RAG search for:[/bold blue] {self.args.query}")
+            
         from autocoder.rag.rag_entry import RAGFactory
         rag = RAGFactory.get_rag(self.llm, self.args, "")
         docs = rag.search(self.args.query)
         for doc in docs:
             doc.tag = "RAG"
+            
+        if self.args.request_id and not self.args.skip_events:
+            _ = queue_communicate.send_event(
+                request_id=self.args.request_id,
+                event=CommunicateEvent(
+                    event_type=CommunicateEventType.CODE_RAG_SEARCH_END.value,
+                    data=json.dumps({},ensure_ascii=False)
+                )
+            )
+        else:
+            console = Console()
+            console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")
+            
         return docs
 
     def get_search_source_codes(self):