auto-coder 0.1.305__py3-none-any.whl → 0.1.307__py3-none-any.whl

autocoder/auto_coder_runner.py

@@ -17,6 +17,7 @@ from contextlib import contextmanager
 from typing import List, Dict, Any, Optional
 from autocoder.common import AutoCoderArgs
 from pydantic import BaseModel
+from autocoder.common.action_yml_file_manager import ActionYmlFileManager
 from autocoder.common.result_manager import ResultManager
 from autocoder.version import __version__
 from autocoder.auto_coder import main as auto_coder_main
@@ -49,10 +50,12 @@ from autocoder.utils.llms import get_single_llm
 import pkg_resources
 from autocoder.common.printer import Printer
 from autocoder.utils.thread_utils import run_in_raw_thread
+from autocoder.memory.active_context_manager import ActiveContextManager
 from autocoder.common.command_completer import CommandCompleter,FileSystemModel as CCFileSystemModel,MemoryConfig as CCMemoryModel
 from autocoder.common.conf_validator import ConfigValidator
 from autocoder import command_parser as CommandParser
 from loguru import logger
+from autocoder.utils.project_structure import EnhancedFileAnalyzer
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -168,9 +171,10 @@ def configure_project_type():
     print_info(get_message("examples"))
 
     print_warning(f"{get_message('default_type')}\n")
-
+    
+    extensions = get_all_extensions(project_root) or "py"
     project_type = prompt(
-        get_message("enter_project_type"), default="py", style=style
+        get_message("enter_project_type"), default=extensions, style=style
     ).strip()
 
     if project_type:
@@ -186,6 +190,32 @@ def configure_project_type():
     return project_type
 
 
+def get_all_extensions(directory: str = ".") -> str:
+    """获取指定目录下所有文件的后缀名,多个按逗号分隔，并且带."""
+    args = AutoCoderArgs(
+        source_dir=directory,
+        # 其他必要参数设置为默认值
+        target_file="",
+        git_url="",
+        project_type="",
+        conversation_prune_safe_zone_tokens=0
+    )
+    
+    analyzer = EnhancedFileAnalyzer(
+        args=args,
+        llm=None,  # 如果只是获取后缀名，可以不需要LLM
+        config=None  # 使用默认配置
+    )
+    
+    # 获取分析结果
+    analysis_result = analyzer.analyze_extensions()
+    
+    # 合并 code 和 config 的后缀名
+    all_extensions = set(analysis_result["code"] + analysis_result["config"])
+    
+    # 转换为逗号分隔的字符串
+    return ",".join(sorted(all_extensions))
+
 def initialize_system(args:InitializeSystemRequest):
     from autocoder.utils.model_provider_selector import ModelProviderSelector
     from autocoder import models as models_module
@@ -666,9 +696,7 @@ def revert():
         
         if "Successfully reverted changes" in s:
             result_manager.append(content=s, meta={"action": "revert","success":False, "input":{                
-            }})
-                        
-            os.remove(file_path)
+            }})                                    
         else:
             result_manager.append(content=s, meta={"action": "revert","success":False, "input":{                
             }})
@@ -1399,12 +1427,33 @@ def commit(query: str):
             with open(os.path.join(execute_file), "w",encoding="utf-8") as f:
                 f.write(yaml_content)
 
-            file_content = open(execute_file).read()
-            md5 = hashlib.md5(file_content.encode("utf-8")).hexdigest()
+            args.file = execute_file             
+            
             file_name = os.path.basename(execute_file)
             commit_result = git_utils.commit_changes(
                 ".", f"{commit_message}\nauto_coder_{file_name}"
             )
+
+            printer = Printer()
+
+            action_yml_file_manager = ActionYmlFileManager(args.source_dir)
+            action_file_name = os.path.basename(args.file)
+            add_updated_urls = []
+            commit_result.changed_files
+            for file in commit_result.changed_files:
+                add_updated_urls.append(os.path.join(args.source_dir, file))
+
+            args.add_updated_urls = add_updated_urls
+            update_yaml_success = action_yml_file_manager.update_yaml_field(action_file_name, "add_updated_urls", add_updated_urls)
+            if not update_yaml_success:                        
+                printer.print_in_terminal("yaml_save_error", style="red", yaml_file=action_file_name)  
+                        
+            if args.enable_active_context:                
+                active_context_manager = ActiveContextManager(llm, args.source_dir)
+                task_id = active_context_manager.process_changes(args)
+                printer.print_in_terminal("active_context_background_task", 
+                                             style="blue",
+                                             task_id=task_id)
             git_utils.print_commit_info(commit_result=commit_result)
             if commit_message:
                 printer.print_in_terminal("commit_message", style="green", model_name=target_model, message=commit_message)                
@@ -1417,7 +1466,7 @@ def commit(query: str):
 
 
 @run_in_raw_thread()
-def coding(query: str):
+def coding(query: str):    
     console = Console()
     is_apply = query.strip().startswith("/apply")
     if is_apply:
@@ -1756,6 +1805,102 @@ def design(query: str):
         os.remove(execute_file)
 
 
+@run_in_raw_thread()
+def active_context(query: str):
+    """
+    管理活动上下文任务，支持列表、查询等操作
+    
+    Args:
+        query: 命令参数，例如 "list" 列出所有任务
+    """    
+    # 解析命令
+    commands_infos = CommandParser.parse_query(query)
+    command = "list"  # 默认命令是列出所有任务
+    
+    if len(commands_infos) > 0:
+        if "list" in commands_infos:
+            command = "list"
+        if "run" in commands_infos:
+            command = "run"
+    
+    args = get_final_config()
+    printer = Printer()
+    # 获取LLM实例    
+    llm = get_single_llm(args.model,product_mode=args.product_mode)
+    action_file_manager = ActionYmlFileManager(args.source_dir)
+    
+    # 获取配置和参数
+    
+    
+    active_context_manager = ActiveContextManager(llm, args.source_dir)
+    if command == "run":
+        file_name = commands_infos["run"]["args"][-1]
+        args.file = action_file_manager.get_full_path_by_file_name(file_name)        
+        ## 因为更新了args.file
+        active_context_manager = ActiveContextManager(llm, args.source_dir)
+        task_id = active_context_manager.process_changes(args)
+        printer.print_in_terminal("active_context_background_task", 
+                                        style="blue",
+                                        task_id=task_id)
+
+    # 处理不同的命令
+    if command == "list":
+        # 获取所有任务
+        all_tasks = active_context_manager.get_all_tasks()
+        
+        if not all_tasks:
+            console = Console()
+            console.print("[yellow]没有找到任何活动上下文任务[/yellow]")
+            return
+        
+        # 创建表格
+        table = Table(title="活动上下文任务列表")
+        table.add_column("任务ID", style="cyan")
+        table.add_column("状态", style="green")
+        table.add_column("开始时间", style="yellow")
+        table.add_column("完成时间", style="yellow")
+        table.add_column("文件", style="blue")
+        
+        # 添加任务数据
+        for task in all_tasks:
+            status = task.get("status", "未知")
+            status_display = status
+            
+            # 根据状态设置不同的显示样式
+            if status == "completed":
+                status_display = "[green]已完成[/green]"
+            elif status == "running":
+                status_display = "[blue]运行中[/blue]"
+            elif status == "queued":
+                position = task.get("queue_position", 0)
+                status_display = f"[yellow]排队中 (位置: {position})[/yellow]"
+            elif status == "failed":
+                status_display = "[red]失败[/red]"
+            
+            # 格式化时间
+            start_time = task.get("start_time", "") 
+            start_time_str = start_time.strftime("%Y-%m-%d %H:%M:%S") if start_time else "未知"
+            
+            completion_time = task.get("completion_time", "")
+            completion_time_str = completion_time.strftime("%Y-%m-%d %H:%M:%S") if completion_time else "-"
+            
+            # 获取文件名
+            file_name = task.get("file_name", "未知")
+            
+            # 添加到表格
+            table.add_row(
+                task.get("task_id", "未知"),
+                status_display,
+                start_time_str,
+                completion_time_str,
+                file_name
+            )
+        
+        # 显示表格
+        console = Console()
+        console.print(table)
+
+
 def voice_input():
     conf = memory.get("conf", {})
     yaml_config = {

autocoder/chat_auto_coder.py

@@ -49,6 +49,7 @@ from autocoder.auto_coder_runner import (
     completer,
     summon,
     get_memory,
+    active_context,
 )
 
 # Create a global plugin manager
@@ -63,6 +64,7 @@ original_functions = {
     "voice_input": voice_input,
     "auto_command": auto_command,
     "execute_shell_command": execute_shell_command,
+    "active_context": active_context,
 }
 
 
@@ -149,6 +151,7 @@ def show_help():
     print(f"  \033[94m/lib\033[0m - \033[92m{get_message('lib_desc')}\033[0m")
     print(f"  \033[94m/models\033[0m - \033[92m{get_message('models_desc')}\033[0m")
     print(f"  \033[94m/plugins\033[0m - \033[92m{get_message('plugins_desc')}\033[0m")
+    print(f"  \033[94m/active_context\033[0m - \033[92m{get_message('active_context_desc')}\033[0m")
     print(f"  \033[94m/exit\033[0m - \033[92m{get_message('exit_desc')}\033[0m")
     print()
 
@@ -423,13 +426,14 @@ def main():
         )
 
     # Replace original functions with wrapped versions
-    global ask, coding, chat, design, voice_input, auto_command, execute_shell_command
+    global ask, coding, chat, design, voice_input, auto_command, execute_shell_command, active_context
     ask = wrapped_functions.get("ask", ask)
     coding = wrapped_functions.get("coding", coding)
     chat = wrapped_functions.get("chat", chat)
     design = wrapped_functions.get("design", design)
     voice_input = wrapped_functions.get("voice_input", voice_input)
     auto_command = wrapped_functions.get("auto_command", auto_command)
+    active_context = wrapped_functions.get("active_context", active_context)
     execute_shell_command = wrapped_functions.get(
         "execute_shell_command", execute_shell_command
     )
@@ -620,6 +624,10 @@ def main():
                     print("Please enter your query.")
                 else:
                     mcp(query)
+                    
+            elif user_input.startswith("/active_context"):
+                query = user_input[len("/active_context") :].strip()
+                active_context(query)
 
             elif user_input.startswith("/auto"):
                 query = user_input[len("/auto") :].strip()