auto-coder 0.1.196__py3-none-any.whl → 0.1.198__py3-none-any.whl

autocoder/agent/project_reader.py

@@ -187,7 +187,7 @@ def get_tools(args: AutoCoderArgs, llm: byzerllm.ByzerLLM):
         注意，尽量不要询问用户，除非你感受到你无法回答用户的问题。
         '''
 
-        if args.request_id and not args.silence:
+        if args.request_id and not args.silence and not args.skip_events:
             event_data = {
                 "question": question                
             }

autocoder/auto_coder.py

@@ -327,7 +327,7 @@ def main(input_args: Optional[List[str]] = None):
                         )
                     )
 
-                if args.request_id and not args.silence:
+                if args.request_id and not args.silence and not args.skip_events:
                     event_data = {
                         "instruction": final_ins,
                         "model": model,
@@ -747,6 +747,7 @@ def main(input_args: Optional[List[str]] = None):
                 pp = SuffixProject(args=args, llm=llm, file_filter=None)
             pp.run()
             sources = pp.sources
+
             s = build_index_and_filter_files(
                 llm=llm, args=args, sources=sources)
             if s:
@@ -762,7 +763,7 @@ def main(input_args: Optional[List[str]] = None):
 
             loaded_conversations = pre_conversations + \
                 chat_history["ask_conversation"]
-
+                        
             if args.human_as_model:
                 console = Console()
 
@@ -874,12 +875,12 @@ def main(input_args: Optional[List[str]] = None):
 
                 return {}
 
-            if args.enable_rag_search or args.enable_rag_context:
+            if args.enable_rag_search or args.enable_rag_context:                
                 rag = RAGFactory.get_rag(llm=chat_llm, args=args, path="")
                 response = rag.stream_chat_oai(
                     conversations=loaded_conversations)[0]
                 v = ([item, None] for item in response)
-            else:
+            else:                
                 v = chat_llm.stream_chat_oai(
                     conversations=loaded_conversations, delta_mode=True
                 )
@@ -895,13 +896,14 @@ def main(input_args: Optional[List[str]] = None):
                     for res in v:
                         markdown_content += res[0]
                         assistant_response += res[0]
-                        request_queue.add_request(
-                            args.request_id,
-                            RequestValue(
-                                value=StreamValue(value=[res[0]]),
-                                status=RequestOption.RUNNING,
-                            ),
-                        )
+                        if args.request_id:                            
+                            request_queue.add_request(
+                                args.request_id,
+                                RequestValue(
+                                    value=StreamValue(value=[res[0]]),
+                                    status=RequestOption.RUNNING,
+                                ),
+                            )
                         live.update(
                             Panel(
                                 Markdown(markdown_content),

autocoder/auto_coder_rag.py

@@ -184,7 +184,7 @@ def main(input_args: Optional[List[str]] = None):
     serve_parser.add_argument(
         "--rag_context_window_limit",
         type=int,
-        default=110000,
+        default=56000,
         help="The input context window limit for RAG",
     )
     serve_parser.add_argument(

autocoder/chat_auto_coder.py

@@ -395,7 +395,7 @@ def get_all_file_names_in_project() -> List[str]:
 
     file_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root):
+    for root, dirs, files in os.walk(project_root,followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         file_names.extend(files)
     return file_names
@@ -405,7 +405,7 @@ def get_all_file_in_project() -> List[str]:
 
     file_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root):
+    for root, dirs, files in os.walk(project_root,followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         for file in files:
             file_names.append(os.path.join(root, file))
@@ -415,7 +415,7 @@ def get_all_file_in_project() -> List[str]:
 def get_all_file_in_project_with_dot() -> List[str]:
     file_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root):
+    for root, dirs, files in os.walk(project_root,followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         for file in files:
             file_names.append(os.path.join(root, file).replace(project_root, "."))
@@ -425,7 +425,7 @@ def get_all_file_in_project_with_dot() -> List[str]:
 def get_all_dir_names_in_project() -> List[str]:
     dir_names = []
     final_exclude_dirs = defaut_exclude_dirs + memory.get("exclude_dirs", [])
-    for root, dirs, files in os.walk(project_root):
+    for root, dirs, files in os.walk(project_root,followlinks=True):
         dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
         for dir in dirs:
             dir_names.append(dir)
@@ -449,7 +449,7 @@ def find_files_in_project(patterns: List[str]) -> List[str]:
         else:
             is_added = False
             ## add files belongs to project
-            for root, dirs, files in os.walk(project_root):
+            for root, dirs, files in os.walk(project_root,followlinks=True):
                 dirs[:] = [d for d in dirs if d not in final_exclude_dirs]
                 if pattern in files:
                     matched_files.append(os.path.join(root, pattern))
@@ -1420,17 +1420,18 @@ def coding(query: str):
         yaml_config["query"] = v
 
         # Add context for active groups and their query prefixes
-        active_groups_context = "下面是对上面文件按分组给到的一些描述，当用户的需求正好匹配描述的时候，参考描述来做修改：\n"
-        for group in current_groups:
-            group_files = groups.get(group, [])
-            query_prefix = groups_info.get(group, {}).get("query_prefix", "")
-            active_groups_context += f"组名: {group}\n"
-            active_groups_context += f"文件列表:\n"
-            for file in group_files:
-                active_groups_context += f"- {file}\n"
-            active_groups_context += f"组描述: {query_prefix}\n\n"
-
-        yaml_config["context"] = active_groups_context + "\n"
+        if current_groups:
+            active_groups_context = "下面是对上面文件按分组给到的一些描述，当用户的需求正好匹配描述的时候，参考描述来做修改：\n"
+            for group in current_groups:
+                group_files = groups.get(group, [])
+                query_prefix = groups_info.get(group, {}).get("query_prefix", "")
+                active_groups_context += f"组名: {group}\n"
+                active_groups_context += f"文件列表:\n"
+                for file in group_files:
+                    active_groups_context += f"- {file}\n"
+                active_groups_context += f"组描述: {query_prefix}\n\n"
+
+            yaml_config["context"] = active_groups_context + "\n"
 
         if is_apply:
             memory_dir = os.path.join(".auto-coder", "memory")