auto-coder 0.1.206__py3-none-any.whl → 0.1.208__py3-none-any.whl

autocoder/common/code_modification_ranker.py

@@ -0,0 +1,100 @@
+import byzerllm
+from typing import List,Union
+from autocoder.common import AutoCoderArgs
+from autocoder.common.types import CodeGenerateResult
+from pydantic import BaseModel
+from loguru import logger
+from concurrent.futures import ThreadPoolExecutor, as_completed
+import traceback
+
+class RankResult(BaseModel):
+    rank_result:List[int]
+
+class CodeModificationRanker:
+    def __init__(self, llm: byzerllm.ByzerLLM, args: AutoCoderArgs):
+        self.llm = llm
+        self.args = args     
+        if self.llm.get_sub_client("generate_rerank_model"):
+            self.rerank_llm = self.llm.get_sub_client("generate_rerank_model")
+        else:
+            self.rerank_llm = self.llm
+    
+    @byzerllm.prompt()
+    def _rank_modifications(self, s:CodeGenerateResult) -> str:
+        '''
+        对一组代码修改进行质量评估并排序。
+
+        下面是修改需求：
+        
+        <edit_requirement>
+        {{ s.conversations[0][-2]["content"] }}
+        </edit_requirement>
+        
+        下面是相应的代码修改：
+        {% for content in s.contents %}
+        <edit_block id="{{ loop.index0 }}">
+        {{content}}
+        </edit_block>
+        {% endfor %}
+        
+        请输出如下格式的评估结果,只包含 JSON 数据:
+
+        ```json
+        {
+            "rank_result": [id1, id2, id3]  // id 为 edit_block 的 id,按质量从高到低排序
+        }
+        ```
+
+        注意，只输出前面要求的 Json 格式就好，不要输出其他内容，Json 需要使用 ```json ```包裹。
+        '''
+        
+
+    def rank_modifications(self, generate_result: CodeGenerateResult) -> CodeGenerateResult:
+        import time
+        start_time = time.time()
+        
+        # 如果只有一个候选，直接返回
+        if len(generate_result.contents) == 1:
+            logger.info("Only 1 candidate, skip ranking")
+            return generate_result
+            
+        logger.info(f"Start ranking {len(generate_result.contents)} candidates")
+        generate_times = self.args.generate_times_same_model
+        
+        try:
+            # Create a thread pool with generate_times workers
+            with ThreadPoolExecutor(max_workers=generate_times) as executor:
+                # Submit tasks
+                futures = [
+                    executor.submit(
+                        self._rank_modifications.with_llm(self.rerank_llm).with_return_type(RankResult).run,
+                        generate_result
+                    ) for _ in range(generate_times)
+                ]
+                
+                # Process results as they complete
+                for future in as_completed(futures):
+                    try:
+                        v = future.result()
+                        # If we get a valid result, use it and cancel other tasks
+                        for f in futures:
+                            f.cancel()
+                            
+                        elapsed = time.time() - start_time
+                        logger.info(f"Ranking completed in {elapsed:.2f}s, best candidate index: {v.rank_result[0]}")
+                        
+                        rerank_contents = [generate_result.contents[i] for i in v.rank_result]
+                        rerank_conversations = [generate_result.conversations[i] for i in v.rank_result]
+                        return CodeGenerateResult(contents=rerank_contents,conversations=rerank_conversations)
+                    except Exception as e:
+                        logger.warning(f"Ranking request failed: {str(e)}")
+                        logger.debug(traceback.format_exc())
+                        continue
+        except Exception as e:
+            logger.error(f"Ranking process failed: {str(e)}")
+            logger.debug(traceback.format_exc())
+        
+        # If all requests failed, use the original codes
+        elapsed = time.time() - start_time
+        logger.warning(f"All ranking requests failed in {elapsed:.2f}s, using original order")
+        return generate_result

autocoder/common/command_completer.py

@@ -12,7 +12,7 @@ COMMANDS = {
         "/svg": {},
         "/sd": {},
     },
-    "/coding": {},
+    "/coding": {"/apply": {}, "/next": {}},
     "/chat": {"/new": {}, "/review": {}, "/no_context": {}},
     "/lib": {
         "/add": "",
@@ -147,7 +147,8 @@ class CommandTextParser:
                 current_word += v
         self.is_extracted = True
         self.current_word_end_pos = self.pos + 1
-        self.current_word_start_pos = self.current_word_end_pos - len(current_word)
+        self.current_word_start_pos = self.current_word_end_pos - \
+            len(current_word)
 
     def previous(self) -> str:
         if self.pos > 1:
@@ -227,10 +228,11 @@ class CommandTextParser:
             self.is_extracted = True
 
         self.current_word_end_pos = self.pos + 1
-        self.current_word_start_pos = self.current_word_end_pos - len(current_word)
+        self.current_word_start_pos = self.current_word_end_pos - \
+            len(current_word)
 
     def current_word(self) -> str:
-        return self.text[self.current_word_start_pos : self.current_word_end_pos]
+        return self.text[self.current_word_start_pos: self.current_word_end_pos]
 
     def get_current_word(self) -> str:
         return self.current_word()

autocoder/common/types.py

@@ -1,10 +1,18 @@
 from enum import Enum
 import pydantic
-
+from typing import List, Dict, Tuple,Any
 class Mode(Enum):
     MULTI_ROUND = "multi_round"
     SINGLE_ROUND = "single_round"
 
 class StepNum(pydantic.BaseModel):
     step_num:int= pydantic.Field(1,description="总共步骤数")
-    content:int= pydantic.Field(1,description="详细的执行步骤，每个步骤需要包含一个shell/python 代码块")    
+    content:int= pydantic.Field(1,description="详细的执行步骤，每个步骤需要包含一个shell/python 代码块")    
+
+class CodeGenerateResult(pydantic.BaseModel):
+    contents:List[str]
+    conversations:List[List[Dict[str, Any]]]
+
+class MergeCodeWithoutEffect(pydantic.BaseModel):
+    success_blocks: List[Tuple[str, str]]
+    failed_blocks: List[Any]

autocoder/dispacher/actions/action.py

@@ -97,38 +97,48 @@ class ActionTSProject:
             else:
                 generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
             if self.args.enable_multi_round_generate:
-                result, conversations = generate.multi_round_run(
+                generate_result = generate.multi_round_run(
                     query=args.query, source_content=content
                 )
             else:
-                result, conversations = generate.single_round_run(
+                generate_result = generate.single_round_run(
                     query=args.query, source_content=content
                 )
-            content = "\n\n".join(result)
-
-            store_code_model_conversation(
-                args=self.args,
-                instruction=self.args.query,
-                conversations=conversations,
-                model=self.llm.default_model_name,
-            )
-        with open(args.target_file, "w") as file:
-            file.write(content)
-
-        if args.execute and args.auto_merge:
-            logger.info("Auto merge the code...")
-            if args.auto_merge == "diff":
-                code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "strict_diff":
-                code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "editblock":
-                code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            else:
-                code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+            merge_result = None
+            if args.execute and args.auto_merge:
+                logger.info("Auto merge the code...")
+                if args.auto_merge == "diff":
+                    code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "strict_diff":
+                    code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "editblock":
+                    code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                else:
+                    code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+
+                if merge_result is not None:
+                    content = merge_result.contents[0]
+                    store_code_model_conversation(
+                        args=self.args,
+                        instruction=self.args.query,
+                        conversations=merge_result.conversations[0],
+                        model=self.llm.default_model_name,
+                    )
+                else:
+                    content = generate_result.contents[0]
+                    store_code_model_conversation(
+                        args=self.args,
+                        instruction=self.args.query,
+                        conversations=generate_result.conversations[0],
+                        model=self.llm.default_model_name,
+                    )
+
+                with open(args.target_file, "w") as file:
+                    file.write(content)
 
 
 class ActionPyScriptProject:
@@ -167,38 +177,49 @@ class ActionPyScriptProject:
             else:
                 generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
             if self.args.enable_multi_round_generate:
-                result, conversations = generate.multi_round_run(
+                generate_result = generate.multi_round_run(
                     query=args.query, source_content=content
                 )
             else:
-                result, conversations = generate.single_round_run(
+                generate_result = generate.single_round_run(
                     query=args.query, source_content=content
                 )
-            content = "\n\n".join(result)
+            merge_result = None
+            if args.execute and args.auto_merge:
+                logger.info("Auto merge the code...")
+                if args.auto_merge == "diff":
+                    code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "strict_diff":
+                    code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "editblock":
+                    code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                else:
+                    code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+
+                content = merge_result.contents[0]
+
+                store_code_model_conversation(
+                    args=self.args,
+                    instruction=self.args.query,
+                    conversations=merge_result.conversations[0],
+                    model=self.llm.default_model_name,
+                )
+            else:
+                content = generate_result.contents[0]
 
-            store_code_model_conversation(
-                args=self.args,
-                instruction=self.args.query,
-                conversations=conversations,
-                model=self.llm.default_model_name,
-            )
-        with open(self.args.target_file, "w") as file:
-            file.write(content)
+                store_code_model_conversation(
+                    args=self.args,
+                    instruction=self.args.query,
+                    conversations=generate_result.conversations[0],
+                    model=self.llm.default_model_name,
+                )
 
-        if args.execute and args.auto_merge:
-            logger.info("Auto merge the code...")
-            if args.auto_merge == "diff":
-                code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "strict_diff":
-                code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "editblock":
-                code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            else:
-                code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+            with open(self.args.target_file, "w") as file:
+                file.write(content)
 
 
 class ActionPyProject:
@@ -255,39 +276,50 @@ class ActionPyProject:
 
 
             if self.args.enable_multi_round_generate:
-                result, conversations = generate.multi_round_run(
+                generate_result = generate.multi_round_run(
                     query=args.query, source_content=content
                 )
             else:
-                result, conversations = generate.single_round_run(
+                generate_result = generate.single_round_run(
                     query=args.query, source_content=content
                 )
                
-            content = "\n\n".join(result)
+            merge_result = None
+            if args.execute and args.auto_merge:
+                logger.info("Auto merge the code...")
+                if args.auto_merge == "diff":
+                    code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "strict_diff":
+                    code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "editblock":
+                    code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                else:
+                    code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+
+                content = merge_result.contents[0]
+
+                store_code_model_conversation(
+                    args=self.args,
+                    instruction=self.args.query,
+                    conversations=merge_result.conversations[0],
+                    model=self.llm.default_model_name,
+                )
+            else:
+                content = generate_result.contents[0]
 
-            store_code_model_conversation(
-                args=self.args,
-                instruction=self.args.query,
-                conversations=conversations,
-                model=self.llm.default_model_name,
-            )
-        with open(args.target_file, "w") as file:
-            file.write(content)
+                store_code_model_conversation(
+                    args=self.args,
+                    instruction=self.args.query,
+                    conversations=generate_result.conversations[0],
+                    model=self.llm.default_model_name,
+                )
 
-        if args.execute and args.auto_merge:
-            logger.info("Auto merge the code...")
-            if args.auto_merge == "diff":
-                code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "strict_diff":
-                code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "editblock":
-                code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            else:
-                code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+            with open(args.target_file, "w") as file:
+                file.write(content)
 
 
 class ActionSuffixProject:
@@ -337,36 +369,51 @@ class ActionSuffixProject:
             else:
                 generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
             if self.args.enable_multi_round_generate:
-                result, conversations = generate.multi_round_run(
+                generate_result = generate.multi_round_run(
                     query=args.query, source_content=content
                 )
             else:
-                result, conversations = generate.single_round_run(
+                generate_result = generate.single_round_run(
                     query=args.query, source_content=content
                 )
-            content = "\n\n".join(result)
-
-            store_code_model_conversation(
-                args=self.args,
-                instruction=self.args.query,
-                conversations=conversations,
-                model=self.llm.default_model_name,
-            )
-
-        with open(args.target_file, "w") as file:
-            file.write(content)
+              
 
+        merge_result = None
         if args.execute and args.auto_merge:
             logger.info("Auto merge the code...")
             if args.auto_merge == "diff":
                 code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+                merge_result = code_merge.merge_code(generate_result=generate_result)
             elif args.auto_merge == "strict_diff":
                 code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+                merge_result = code_merge.merge_code(generate_result=generate_result)
             elif args.auto_merge == "editblock":
                 code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+                merge_result = code_merge.merge_code(generate_result=generate_result)
             else:
                 code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+                merge_result = code_merge.merge_code(generate_result=generate_result)
+
+        if merge_result is not None:
+            content = merge_result.contents[0]
+            store_code_model_conversation(
+                args=self.args,
+                instruction=self.args.query,
+                conversations=merge_result.conversations[0],
+                model=self.llm.default_model_name,
+            )
+            with open(args.target_file, "w") as file:
+                file.write(content)
+        else:
+            content = generate_result.contents[0]
+
+            store_code_model_conversation(
+                args=self.args,
+                instruction=self.args.query,
+                conversations=generate_result.conversations[0],
+                model=self.llm.default_model_name,
+            )
+
+            with open(args.target_file, "w") as file:
+                file.write(content)
+

autocoder/dispacher/actions/plugins/action_regex_project.py

@@ -68,35 +68,45 @@ class ActionRegexProject:
             else:
                 generate = CodeAutoGenerate(llm=self.llm, args=self.args, action=self)
             if self.args.enable_multi_round_generate:
-                result, conversations = generate.multi_round_run(
+                generate_result = generate.multi_round_run(
                     query=args.query, source_content=content
                 )
             else:
-                result, conversations = generate.single_round_run(
+                generate_result = generate.single_round_run(
                     query=args.query, source_content=content
                 )
-            content = "\n\n".join(result)
+            merge_result = None
+            if args.execute and args.auto_merge:
+                logger.info("Auto merge the code...")
+                if args.auto_merge == "diff":
+                    code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "strict_diff":
+                    code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                elif args.auto_merge == "editblock":
+                    code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
+                else:
+                    code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
+                    merge_result = code_merge.merge_code(generate_result=generate_result)
 
-            store_code_model_conversation(
-                args=self.args,
-                instruction=self.args.query,
-                conversations=conversations,
-                model=self.llm.default_model_name,
-            )
-        with open(args.target_file, "w") as file:
-            file.write(content)
-
-        if args.execute and args.auto_merge:
-            logger.info("Auto merge the code...")
-            if args.auto_merge == "diff":
-                code_merge = CodeAutoMergeDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "strict_diff":
-                code_merge = CodeAutoMergeStrictDiff(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
-            elif args.auto_merge == "editblock":
-                code_merge = CodeAutoMergeEditBlock(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+            if merge_result is not None:
+                content = merge_result.contents[0]
+                store_code_model_conversation(
+                    args=self.args,
+                    instruction=self.args.query,
+                    conversations=merge_result.conversations[0],
+                    model=self.llm.default_model_name,
+                )
             else:
-                code_merge = CodeAutoMerge(llm=self.llm, args=self.args)
-                code_merge.merge_code(content=content)
+                content = generate_result.contents[0]
+                store_code_model_conversation(
+                    args=self.args,
+                    instruction=self.args.query,
+                    conversations=generate_result.conversations[0],
+                    model=self.llm.default_model_name,
+                )
+
+            with open(args.target_file, "w") as file:
+                file.write(content)

autocoder/lang.py

@@ -21,6 +21,7 @@ lang_desc = {
         "cmd_args_title": "Command Line Arguments:",
         "py_packages": "The Python packages added to context, only works for py project type. Default is empty.",
         "human_as_model": "Use human as model or not. Default is False",
+        "human_model_num": "Number of human models to use. Default is 1",
         "urls": "The urls to crawl and extract text from, separated by comma",
         "search_engine": "The search engine to use. Supported engines: bing, google. Default is empty",
         "search_engine_token": "The token for the search engine API. Default is empty",
@@ -62,6 +63,8 @@ lang_desc = {
         "screenshot_url": "The URL of the webpage to capture",
         "screenshot_output": "The directory to save the screenshots",
         "code_model": "The name of the code model to use. Default is empty",
+        "generate_rerank_model": "The name of the generate rerank model to use. Default is empty",
+        "inference_model": "The name of the inference model to use. Default is empty",
         "system_prompt": "The system prompt for the model. Default is empty",
         "planner_model": "The name of the planner model to use. Default is empty",
         "designer_model": "The name of the designer model to use. Default is empty",
@@ -78,6 +81,7 @@ lang_desc = {
         "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",
+        "generate_times_same_model": "Number of times to generate using the same model. Default is 1",
     },
     "zh": {
         "request_id": "Request ID",
@@ -101,6 +105,7 @@ lang_desc = {
         "cmd_args_title": "命令行参数:", 
         "py_packages": "添加到上下文的Python包,仅适用于py项目类型。默认为空。",
         "human_as_model": "是否使用人工作为模型。默认为False",
+        "human_model_num": "使用的人工模型数量。默认为1",
         "urls": "要爬取并提取文本的URL,多个URL以逗号分隔",
         "search_engine": "要使用的搜索引擎。支持的引擎:bing、google。默认为空",
         "search_engine_token": "搜索引擎API的令牌。默认为空",
@@ -142,6 +147,8 @@ lang_desc = {
         "screenshot_url": "要捕获的网页的URL",
         "screenshot_output": "保存截图的目录",
         "code_model": "要使用的代码模型的名称。默认为空",
+        "generate_rerank_model": "要使用的生成重排序模型的名称。默认为空",
+        "inference_model": "要使用的推理模型的名称。默认为空",
         "system_prompt": "模型使用的系统提示词。默认为空",
         "next_desc": "基于上一个action文件创建一个新的action文件",
         "planner_model": "要使用的规划模型的名称。默认为空",
@@ -157,6 +164,7 @@ lang_desc = {
          "rag_url": "RAG服务的URL",
         "rag_token": "RAG服务的令牌",
         "rag_type": "RAG类型(simple/storage),默认是storage",
-        "rag_params_max_tokens": "RAG参数的最大token数。默认为4096",    
+        "rag_params_max_tokens": "RAG参数的最大token数。默认为4096",  
+        "generate_times_same_model": "使用相同模型生成的次数。默认为1",
     }
 }

autocoder/pyproject/__init__.py

@@ -114,6 +114,10 @@ class PyProject:
             "dist",
             "__pycache__",
             "node_modules",
+            ".auto-coder",
+            "actions",
+            ".vscode",
+            ".idea",
         ]
 
     @byzerllm.prompt()

autocoder/rag/long_context_rag.py

@@ -395,6 +395,7 @@ class LongContextRAG:
                     inference_deep_thought=self.args.inference_deep_thought,
                     inference_slow_without_deep_thought=self.args.inference_slow_without_deep_thought,                    
                     precision=self.args.inference_compute_precision,
+                    data_cells_max_num=self.args.data_cells_max_num,
                 )
                 conversations = conversations[:-1]
                 new_conversations = llm_compute_engine.process_conversation(
@@ -576,6 +577,7 @@ class LongContextRAG:
                     inference_enhance=not self.args.disable_inference_enhance,
                     inference_deep_thought=self.args.inference_deep_thought,
                     precision=self.args.inference_compute_precision,
+                    data_cells_max_num=self.args.data_cells_max_num,
                     debug=False,
                 )
                 new_conversations = llm_compute_engine.process_conversation(

autocoder/rag/rag_entry.py

@@ -1,4 +1,4 @@
-from typing import List, Dict, Any, Optional
+from typing import List, Dict, Any, Optional,Union
 from autocoder.common import AutoCoderArgs, SourceCode
 from byzerllm import ByzerLLM
 from .simple_rag import SimpleRAG
@@ -7,7 +7,7 @@ class RAGFactory:
     
     
     @staticmethod
-    def get_rag(llm: ByzerLLM, args: AutoCoderArgs, path: str,**kargs) -> SimpleRAG | LongContextRAG:
+    def get_rag(llm: ByzerLLM, args: AutoCoderArgs, path: str,**kargs) -> Union[SimpleRAG, LongContextRAG]:
         """
         Factory method to get the appropriate RAG implementation based on arguments.
         

autocoder/suffixproject/__init__.py

@@ -55,6 +55,8 @@ class SuffixProject:
             "node_modules",
             ".auto-coder",
             ".vscode",
+            "actions",
+            ".idea",
         ]
 
     @byzerllm.prompt()

autocoder/tsproject/__init__.py

@@ -43,6 +43,10 @@ class TSProject:
             "dist",
             "__pycache__",
             "node_modules",
+            ".auto-coder",
+            "actions",
+            ".vscode",
+            ".idea",
         ]
 
     @byzerllm.prompt()

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.206"
+__version__ = "0.1.208"