auto-coder 0.1.329__py3-none-any.whl → 0.1.331__py3-none-any.whl

autocoder/linters/python_linter.py

@@ -25,6 +25,7 @@ class PythonLinter(BaseLinter):
             verbose (bool): Whether to display verbose output.
         """
         super().__init__(verbose)
+        BaseLinter.tt()
     
     def get_supported_extensions(self) -> List[str]:
         """
@@ -152,6 +153,7 @@ class PythonLinter(BaseLinter):
                 sys.executable, 
                 "-m", 
                 "pylint", 
+                "--disable=import-error",
                 "--output-format=json", 
                 target
             ]

autocoder/pyproject/__init__.py

@@ -233,15 +233,7 @@ 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}")
@@ -250,16 +242,7 @@ class PyProject:
         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)
-                )
-            )
+            doc.tag = "RAG"                    
         else:
             console = Console()
             console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")

autocoder/rag/cache/simple_cache.py

@@ -45,7 +45,7 @@ def generate_content_md5(content: Union[str, bytes]) -> str:
 
 
 class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
-    def __init__(self, path: str, ignore_spec, required_exts: list):
+    def __init__(self, path: str, ignore_spec, required_exts: list, update_interval: int = 5):
         """
         初始化异步更新队列，用于管理代码文件的缓存。
         
@@ -53,6 +53,7 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
             path: 需要索引的代码库根目录
             ignore_spec: 指定哪些文件/目录应被忽略的规则
             required_exts: 需要处理的文件扩展名列表
+            update_interval: 自动触发更新的时间间隔（秒），默认为5秒
             
         缓存结构 (self.cache):
             self.cache 是一个字典，其结构如下:
@@ -90,13 +91,22 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
         self.path = path
         self.ignore_spec = ignore_spec
         self.required_exts = required_exts
+        self.update_interval = update_interval
         self.queue = []
         self.cache = {}  # 初始化为空字典，稍后通过 read_cache() 填充
         self.lock = threading.Lock()
         self.stop_event = threading.Event()
-        self.thread = threading.Thread(target=self._process_queue)
-        self.thread.daemon = True
-        self.thread.start()
+        
+        # 启动处理队列的线程
+        self.queue_thread = threading.Thread(target=self._process_queue)
+        self.queue_thread.daemon = True
+        self.queue_thread.start()
+        
+        # 启动定时触发更新的线程
+        self.update_thread = threading.Thread(target=self._periodic_update)
+        self.update_thread.daemon = True
+        self.update_thread.start()
+        
         self.cache = self.read_cache()
 
     def _process_queue(self):
@@ -106,10 +116,25 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
             except Exception as e:
                 logger.error(f"Error in process_queue: {e}")
             time.sleep(1)  # 避免过于频繁的检查
+            
+    def _periodic_update(self):
+        """定时触发文件更新检查"""
+        while not self.stop_event.is_set():            
+            try:
+                logger.debug(f"Periodic update triggered (every {self.update_interval}s)")
+                # 如果没有被初始化过，不会增量触发
+                if not self.cache:
+                    time.sleep(self.update_interval)
+                    continue
+                self.trigger_update()
+            except Exception as e:
+                logger.error(f"Error in periodic update: {e}")
+            time.sleep(self.update_interval)
 
     def stop(self):
         self.stop_event.set()
-        self.thread.join()
+        self.queue_thread.join()
+        self.update_thread.join()
 
     def fileinfo_to_tuple(self, file_info: FileInfo) -> Tuple[str, str, float, str]:
         return (file_info.file_path, file_info.relative_path, file_info.modify_time, file_info.file_md5)    
@@ -285,7 +310,7 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
 
     def get_cache(self, options: Optional[Dict[str, Any]] = None):
         self.load_first()
-        self.trigger_update()
+        # 不再在这里触发更新，因为已经有定时线程在处理
         return self.cache
 
     def get_all_files(self) -> List[Tuple[str, str, float]]:

autocoder/regexproject/__init__.py

@@ -121,17 +121,8 @@ class RegexProject:
         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}")
+        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, "")
@@ -139,17 +130,8 @@ class RegexProject:
         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]")
+        console = Console()
+        console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")
             
         return docs
 

autocoder/suffixproject/__init__.py

@@ -162,36 +162,18 @@ 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}")
+                
+        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]")
+ 
+        console = Console()
+        console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")
             
         return docs
 

autocoder/tsproject/__init__.py

@@ -198,17 +198,8 @@ class TSProject:
         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}")
+        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, "")
@@ -216,17 +207,9 @@ class TSProject:
         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]")
+
+        console = Console()
+        console.print(f"[bold green]Found {len(docs)} relevant documents[/bold green]")
             
         return docs
 

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.329"
+__version__ = "0.1.331"