auto-coder-web 0.1.81__py3-none-any.whl → 0.1.83__py3-none-any.whl

auto_coder_web/auto_coder_runner_wrapper.py

@@ -4,8 +4,7 @@ from autocoder.auto_coder_runner import (
     load_memory,
     save_memory,
     configure,
-    manage_models,
-    print_conf,
+    manage_models,    
     exclude_dirs,
     exclude_files,
     ask,

auto_coder_web/file_manager.py

@@ -40,10 +40,11 @@ def get_directory_tree(root_path: str, path: str = None, lazy: bool = False) ->
 
     def should_ignore(name: str) -> bool:
         """Check if a file or directory should be ignored"""
-        # Ignore hidden files/directories
-        if name.startswith('.'):
+        allowed_hidden_files = {'.autocoderrules', '.gitignore', '.autocoderignore'}
+        # Ignore hidden files/directories, unless they are explicitly allowed
+        if name.startswith('.') and name not in allowed_hidden_files:
             return True
-        # Ignore exact matches and pattern matches
+        # Ignore exact matches and pattern matches from IGNORE_PATTERNS
         return name in IGNORE_PATTERNS
 
     def build_tree(current_path: str) -> List[Dict[str, Any]]:
@@ -146,11 +147,12 @@ async def get_directory_tree_async(root_path: str, path: str = None, lazy: bool
 
     def should_ignore(name: str) -> bool:
         """Check if a file or directory should be ignored"""
-        # Ignore hidden files/directories (starting with '.')
-        ## and name != ".auto-coder":
-        if name.startswith('.'):
+        allowed_hidden_files = {'.autocoderrules', '.gitignore', '.autocoderignore'}
+        # Ignore hidden files/directories (starting with '.'), unless explicitly allowed
+        ## and name != ".auto-coder": # Original comment kept for context if needed
+        if name.startswith('.') and name not in allowed_hidden_files:
             return True
-        # Ignore exact matches
+        # Ignore exact matches from IGNORE_PATTERNS
         return name in IGNORE_PATTERNS
 
     async def build_tree(current_path: str) -> List[Dict[str, Any]]:

auto_coder_web/init_project.py

@@ -20,6 +20,15 @@ def init_project(project_path: str):
         f.write("\n/actions/")
         f.write("\n/output.txt")
 
+
+    # 生成 .autocoderignore 文件，采用 .gitignore 格式
+    autocoderignore_path = os.path.join(source_dir, ".autocoderignore")
+    autocoderignore_content = (
+        "target\n"        
+    )
+    with open(autocoderignore_path, "w", encoding="utf-8") as f:
+        f.write(autocoderignore_content)    
+
     print(
         f"""Successfully initialized auto-coder project in {os.path.abspath(project_path)}."""
     )

auto_coder_web/proxy.py

@@ -19,7 +19,7 @@ import sys
 from auto_coder_web.terminal import terminal_manager
 from autocoder.common import AutoCoderArgs
 from auto_coder_web.auto_coder_runner_wrapper import AutoCoderRunnerWrapper
-from auto_coder_web.routers import todo_router, settings_router, auto_router, commit_router, chat_router, coding_router, index_router, config_router, upload_router, rag_router, editable_preview_router,mcp_router
+from auto_coder_web.routers import todo_router, settings_router, auto_router, commit_router, chat_router, coding_router, index_router, config_router, upload_router, rag_router, editable_preview_router, mcp_router, direct_chat_router
 from auto_coder_web.expert_routers import history_router
 from auto_coder_web.common_router import completions_router, file_router, auto_coder_conf_router, chat_list_router, file_group_router, model_router, compiler_router
 from auto_coder_web.common_router import active_context_router
@@ -106,7 +106,7 @@ class ProxyServer:
         self.app.include_router(editable_preview_router.router)
         self.app.include_router(mcp_router.router)
         self.app.include_router(active_context_router.router)
-        # self.app.include_router(rag_router.router)
+        self.app.include_router(direct_chat_router.router)
 
         @self.app.on_event("shutdown")
         async def shutdown_event():

auto_coder_web/routers/auto_router.py

@@ -96,7 +96,8 @@ async def auto_command(request: AutoCommandRequest, project_path: str = Depends(
         try:
             # 创建AutoCoderRunnerWrapper实例，使用从应用上下文获取的项目路径
             wrapper = AutoCoderRunnerWrapper(project_path)
-            wrapper.configure_wrapper(f"event_file:{event_file}")
+            wrapper.configure_wrapper(f"event_file:{event_file}")   
+            global_cancel.register_token(event_file)         
             prompt_text = request.command
 
             if request.include_conversation_history:
@@ -374,12 +375,10 @@ async def cancel_task(request: CancelTaskRequest, project_path: str = Depends(ge
     """
     # 定义在线程中执行的取消任务函数
     def cancel_task_thread(event_file_id: str, project_path: str):
-        try:
-            # 设置全局取消标志
-            global_cancel.set()
-            
+        try:                        
             # 获取事件文件路径和事件管理器
             event_file = get_event_file_path(file_id=event_file_id, project_path=project_path)
+            global_cancel.set(token=event_file)
             event_manager = get_event_manager(event_file)
             
             # 向事件流写入取消事件

auto_coder_web/routers/chat_router.py

@@ -68,7 +68,8 @@ async def chat_command(request: ChatCommandRequest, project_path: str = Depends(
         try:
             # 创建AutoCoderRunnerWrapper实例，使用从应用上下文获取的项目路径
             wrapper = AutoCoderRunnerWrapper(project_path)
-            wrapper.configure_wrapper(f"event_file:{event_file}")
+            wrapper.configure_wrapper(f"event_file:{event_file}")    
+            global_cancel.register_token(event_file)
 
             # 调用chat方法
             result = wrapper.chat_wrapper(request.command)            
@@ -304,9 +305,8 @@ async def cancel_task(request: CancelTaskRequest, project_path: str = Depends(ge
         event_file = get_event_file_path(file_id=request.event_file_id, project_path=project_path)
         
         def cancel_in_thread():
-            try:
-                # 设置全局取消标志
-                global_cancel.set_cancel(request.event_file_id)
+            try:                                
+                global_cancel.set(token=event_file)
                 
                 # 获取事件管理器
                 event_manager = get_event_manager(event_file)

auto_coder_web/routers/coding_router.py

@@ -89,6 +89,7 @@ async def coding_command(request: CodingCommandRequest, project_path: str = Depe
             # 创建AutoCoderRunnerWrapper实例，使用从应用上下文获取的项目路径
             wrapper = AutoCoderRunnerWrapper(project_path)
             wrapper.configure_wrapper(f"event_file:{event_file}")
+            global_cancel.register_token(event_file)
 
             # 获取当前会话名称
             current_session_file = os.path.join(project_path, ".auto-coder", "auto-coder.web", "current-session.json")
@@ -362,9 +363,8 @@ async def cancel_task(request: CancelTaskRequest, project_path: str = Depends(ge
         event_file = get_event_file_path(file_id=request.event_file_id, project_path=project_path)
         
         def cancel_in_thread():
-            try:
-                # 设置全局取消标志
-                global_cancel.set()
+            try:                                
+                global_cancel.set(token=event_file)
                 
                 # 获取事件管理器
                 event_manager = get_event_manager(event_file)

auto_coder_web/routers/direct_chat_router.py

@@ -0,0 +1,42 @@
+import traceback
+from fastapi import APIRouter, Request
+from pydantic import BaseModel
+from typing import Any, Dict
+from loguru import logger
+import byzerllm
+
+from autocoder.utils.llms import get_single_llm
+
+router = APIRouter()
+
+class DirectChatRequest(BaseModel):
+    model: str
+    content: str
+    product_mode: str = "lite"
+    options: Dict[str, Any] = {}
+
+class DirectChatResponse(BaseModel):
+    success: bool
+    result: Any = None
+    error: str = None
+
+@router.post("/api/direct_chat", response_model=DirectChatResponse)
+async def direct_chat(req: DirectChatRequest, request: Request):
+    """
+    简单直聊API，指定模型和内容，返回模型回复。
+    """
+    try:
+        # 获取模型
+        llm = get_single_llm(req.model, product_mode=req.product_mode)
+        
+        @byzerllm.prompt()
+        def chat_func(content: str) -> str:
+            """
+            {{ content }}
+            """
+        # 支持自定义llm_config等参数
+        result = chat_func.with_llm(llm).run(req.content)        
+        return DirectChatResponse(success=True, result=result)
+    except Exception as e:
+        logger.error(f"direct_chat error: {e}\n{traceback.format_exc()}")
+        return DirectChatResponse(success=False, error=f"{str(e)}")