auto-coder-web 0.1.26__tar.gz → 0.1.29__tar.gz

{auto_coder_web-0.1.26 → auto_coder_web-0.1.29}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.1
 Name: auto_coder_web
-Version: 0.1.26
+Version: 0.1.29
 Summary: auto-coder.web: A Python Project
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Description-Content-Type: text/markdown
-Requires-Dist: auto-coder>=0.1.307
+Requires-Dist: auto-coder>=0.1.313
 Requires-Dist: aiofiles
 Requires-Dist: psutil
 Requires-Dist: sse-starlette

{auto_coder_web-0.1.26 → auto_coder_web-0.1.29}/src/auto_coder_web/auto_coder_runner_wrapper.py

@@ -45,7 +45,11 @@ class AutoCoderRunnerWrapper:
         return auto_command(command,params)
     
     def configure_wrapper(self,conf: str, skip_print=False ):
-        return configure(conf, skip_print)
+        return configure(conf, skip_print)    
+
+    def get_conf_wrapper(self):
+        memory = get_memory()
+        return memory["conf"]
 
 
     def coding_wapper(self,query):

auto_coder_web-0.1.29/src/auto_coder_web/common_router/chat_list_router.py

@@ -0,0 +1,143 @@
+import os
+import json
+from fastapi import APIRouter, HTTPException, Request, Depends
+import aiofiles
+from auto_coder_web.types import ChatList
+from pydantic import BaseModel
+
+
+class SessionNameRequest(BaseModel):
+    session_name: str
+
+
+async def get_project_path(request: Request) -> str:
+    """
+    从FastAPI请求上下文中获取项目路径
+    """
+    return request.app.state.project_path
+
+router = APIRouter()
+
+
+@router.post("/api/chat-lists/save")
+async def save_chat_list(chat_list: ChatList, project_path: str = Depends(get_project_path)):
+    try:
+        chat_lists_dir = os.path.join(project_path,
+                                      ".auto-coder", "auto-coder.web", "chat-lists")
+        os.makedirs(chat_lists_dir, exist_ok=True)
+
+        file_path = os.path.join(chat_lists_dir, f"{chat_list.name}.json")
+        async with aiofiles.open(file_path, 'w') as f:
+            await f.write(json.dumps({"messages": chat_list.messages}, indent=2, ensure_ascii=False))
+        return {"status": "success", "message": f"Chat list {chat_list.name} saved successfully"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.get("/api/chat-lists")
+async def get_chat_lists(project_path: str = Depends(get_project_path)):
+    try:
+        chat_lists_dir = os.path.join(
+            project_path, ".auto-coder", "auto-coder.web", "chat-lists")
+        os.makedirs(chat_lists_dir, exist_ok=True)
+
+        # Get files with their modification times
+        chat_lists = []
+        for file in os.listdir(chat_lists_dir):
+            if file.endswith('.json'):
+                file_path = os.path.join(chat_lists_dir, file)
+                mod_time = os.path.getmtime(file_path)
+                # Store tuple of (name, mod_time)
+                chat_lists.append((file[:-5], mod_time))
+
+        # Sort by modification time (newest first)
+        chat_lists.sort(key=lambda x: x[1], reverse=True)
+
+        # Return only the chat list names
+        return {"chat_lists": [name for name, _ in chat_lists]}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.get("/api/chat-lists/{name}")
+async def get_chat_list(name: str, project_path: str = Depends(get_project_path)):
+    try:
+        file_path = os.path.join(
+            project_path, ".auto-coder", "auto-coder.web", "chat-lists", f"{name}.json")
+        if not os.path.exists(file_path):
+            raise HTTPException(
+                status_code=404, detail=f"Chat list {name} not found")
+
+        async with aiofiles.open(file_path, 'r') as f:
+            content = await f.read()
+            return json.loads(content)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.delete("/api/chat-lists/{name}")
+async def delete_chat_list(name: str, project_path: str = Depends(get_project_path)):
+    try:
+        file_path = os.path.join(
+            project_path, ".auto-coder", "auto-coder.web", "chat-lists", f"{name}.json")
+        if not os.path.exists(file_path):
+            raise HTTPException(
+                status_code=404, detail=f"Chat list {name} not found")
+
+        os.remove(file_path)
+        return {"status": "success", "message": f"Chat list {name} deleted successfully"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.get("/api/chat-session/name")
+async def get_current_session_name(project_path: str = Depends(get_project_path)):
+    """
+    获取当前会话名称
+    """
+    try:
+        # 创建存储会话信息的目录
+        session_dir = os.path.join(project_path, ".auto-coder", "auto-coder.web")
+        os.makedirs(session_dir, exist_ok=True)
+        
+        # 会话信息文件路径
+        session_file = os.path.join(session_dir, "current-session.json")
+        
+        # 如果文件不存在，返回空会话名称
+        if not os.path.exists(session_file):
+            return {"session_name": ""}
+        
+        # 读取当前会话信息
+        async with aiofiles.open(session_file, 'r') as f:
+            content = await f.read()
+            session_data = json.loads(content)
+            return {"session_name": session_data.get("session_name", "")}
+            
+    except Exception as e:
+        # 如果发生错误，记录错误但返回空会话名
+        print(f"Error getting current session name: {str(e)}")
+        return {"session_name": ""}
+
+
+@router.post("/api/chat-session/name")
+async def set_current_session_name(request: SessionNameRequest, project_path: str = Depends(get_project_path)):
+    """
+    设置当前会话名称
+    """
+    try:
+        # 创建存储会话信息的目录
+        session_dir = os.path.join(project_path, ".auto-coder", "auto-coder.web")
+        os.makedirs(session_dir, exist_ok=True)
+        
+        # 会话信息文件路径
+        session_file = os.path.join(session_dir, "current-session.json")
+        
+        # 保存当前会话信息
+        session_data = {"session_name": request.session_name}
+        async with aiofiles.open(session_file, 'w') as f:
+            await f.write(json.dumps(session_data, indent=2, ensure_ascii=False))
+            
+        return {"status": "success", "message": "Current session name updated"}
+            
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to set current session name: {str(e)}")

auto_coder_web-0.1.29/src/auto_coder_web/common_router/completions_router.py

@@ -0,0 +1,166 @@
+import os
+import glob
+import json
+from typing import List
+from pydantic import BaseModel
+from fastapi import APIRouter, Query, Request, Depends
+from auto_coder_web.types import CompletionItem, CompletionResponse
+from autocoder.index.symbols_utils import (
+    extract_symbols,
+    symbols_info_to_str,
+    SymbolsInfo,
+    SymbolType,
+)
+
+from autocoder.auto_coder_runner import get_memory
+import json
+import asyncio
+
+
+router = APIRouter()
+
+class SymbolItem(BaseModel):
+    symbol_name: str
+    symbol_type: SymbolType
+    file_name: str
+
+async def get_auto_coder_runner(request: Request):
+    """获取AutoCoderRunner实例作为依赖"""
+    return request.app.state.auto_coder_runner
+
+
+async def get_project_path(request: Request):
+    """获取项目路径作为依赖"""
+    return request.app.state.project_path    
+
+def find_files_in_project(patterns: List[str], project_path: str) -> List[str]:
+    memory = get_memory()
+    default_exclude_dirs = [".git", "node_modules", "dist", "build", "__pycache__",".venv"]
+    active_file_list = memory["current_files"]["files"]
+
+    project_root = project_path
+    matched_files = []
+
+    if len(patterns) == 1 and patterns[0] == "":
+        return active_file_list
+
+    for pattern in patterns:
+        for file_path in active_file_list:
+            if pattern in os.path.basename(file_path):
+                matched_files.append(file_path)
+
+    final_exclude_dirs = default_exclude_dirs + \
+        memory.get("exclude_dirs", [])
+
+    for pattern in patterns:
+        if "*" in pattern or "?" in pattern:
+            for file_path in glob.glob(pattern, recursive=True):
+                if os.path.isfile(file_path):
+                    abs_path = os.path.abspath(file_path)
+                    if not any(
+                        exclude_dir in abs_path.split(os.sep)
+                        for exclude_dir in final_exclude_dirs
+                    ):
+                        matched_files.append(abs_path)
+        else:
+            is_added = False
+            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))
+                    is_added = True
+                else:
+                    for file in files:
+                        if pattern in os.path.join(root, file):
+                            matched_files.append(os.path.join(root, file))
+                            is_added = True
+            if not is_added:
+                matched_files.append(pattern)
+
+    return list(set(matched_files))
+
+def get_symbol_list(project_path: str) -> List[SymbolItem]:
+    list_of_symbols = []
+    index_file = os.path.join(
+        project_path, ".auto-coder", "index.json")
+
+    if os.path.exists(index_file):
+        with open(index_file, "r") as file:
+            index_data = json.load(file)
+    else:
+        index_data = {}
+
+    for item in index_data.values():
+        symbols_str = item["symbols"]
+        module_name = item["module_name"]
+        info1 = extract_symbols(symbols_str)
+        for name in info1.classes:
+            list_of_symbols.append(
+                SymbolItem(
+                    symbol_name=name,
+                    symbol_type=SymbolType.CLASSES,
+                    file_name=module_name,
+                )
+            )
+        for name in info1.functions:
+            list_of_symbols.append(
+                SymbolItem(
+                    symbol_name=name,
+                    symbol_type=SymbolType.FUNCTIONS,
+                    file_name=module_name,
+                )
+            )
+        for name in info1.variables:
+            list_of_symbols.append(
+                SymbolItem(
+                    symbol_name=name,
+                    symbol_type=SymbolType.VARIABLES,
+                    file_name=module_name,
+                )
+            )
+    return list_of_symbols
+
+@router.get("/api/completions/files")
+async def get_file_completions(
+    name: str = Query(...),
+    project_path: str = Depends(get_project_path)
+):
+    """获取文件名补全"""
+    patterns = [name]
+    matches = await asyncio.to_thread(find_files_in_project, patterns,project_path)
+    completions = []
+    project_root = project_path
+    for file_name in matches:
+        path_parts = file_name.split(os.sep)
+        # 只显示最后三层路径，让显示更简洁
+        display_name = os.sep.join(
+            path_parts[-3:]) if len(path_parts) > 3 else file_name
+        relative_path = os.path.relpath(file_name, project_root)
+
+        completions.append(CompletionItem(
+            name=relative_path,  # 给补全项一个唯一标识
+            path=relative_path,  # 实际用于替换的路径
+            display=display_name,  # 显示的简短路径
+            location=relative_path  # 完整的相对路径信息
+        ))
+    return CompletionResponse(completions=completions)
+
+@router.get("/api/completions/symbols")
+async def get_symbol_completions(
+    name: str = Query(...),
+    project_path: str = Depends(get_project_path)
+):
+    """获取符号补全"""
+    symbols = await asyncio.to_thread(get_symbol_list, project_path)
+    matches = []
+
+    for symbol in symbols:
+        if name.lower() in symbol.symbol_name.lower():
+            relative_path = os.path.relpath(
+                symbol.file_name, project_path)
+            matches.append(CompletionItem(
+                name=symbol.symbol_name,
+                path=relative_path,
+                display=f"{symbol.symbol_name}(location: {relative_path})"
+            ))
+    return CompletionResponse(completions=matches) 

{auto_coder_web-0.1.26 → auto_coder_web-0.1.29}/src/auto_coder_web/common_router/file_group_router.py

@@ -64,12 +64,20 @@ def _get_groups(project_path: str):
     ]
     return v
 
-def _switch_groups(group_names: List[str]):
+def _switch_groups(group_names: List[str], file_paths: List[str] = None):
     memory = get_memory()
     new_files = []
     for group_name in group_names:
         files = memory["current_files"]["groups"][group_name]
         new_files.extend(files)
+    
+    # Add individual file paths if provided
+    if file_paths:
+        for file_path in file_paths:
+            # Only add unique paths
+            if file_path not in new_files:
+                new_files.append(file_path)
+    
     memory["current_files"]["files"] = new_files
     memory["current_files"]["current_groups"] = group_names
     save_memory()
@@ -139,12 +147,21 @@ async def auto_create_groups(
 
 @router.post("/api/file-groups/switch")
 async def switch_file_groups(
-    request: Request
+    request: Request,
+    project_path: str = Depends(get_project_path)
 ):
     data = await request.json()
     group_names = data.get("group_names", [])
-    await asyncio.to_thread(_switch_groups, group_names)
-    return {"status": "success", "message": f"Switched to groups: {group_names}"}
+    file_paths = data.get("file_paths", [])
+    
+    # Convert relative file paths to absolute paths
+    absolute_file_paths = []
+    for file_path in file_paths:
+        absolute_path = os.path.join(project_path, file_path)
+        absolute_file_paths.append(absolute_path)
+    
+    await asyncio.to_thread(_switch_groups, group_names, absolute_file_paths)
+    return {"status": "success", "message": f"Switched to groups: {group_names} and additional files"}
 
 
 @router.delete("/api/file-groups/{name}")

auto_coder_web-0.1.29/src/auto_coder_web/lang.py

@@ -0,0 +1,35 @@
+import locale
+from byzerllm.utils import format_str_jinja2
+
+MESSAGES = {
+    "project_not_initialized": {
+        "en":"Warning: Project not initialized.",
+        "zh":"警告：项目未初始化。"
+    },
+    "run_auto_coder_chat": {
+        "en":"Please run 'auto-coder.chat' to initialize the project first.",
+        "zh":"请先运行 'auto-coder.chat' 来初始化项目。"
+    },
+    "human_as_model_warning": {
+        "en":"Warning: Project is configured to use human as model, auto-coder.web will not work, we will set human_as_model to false",
+        "zh":"警告：项目配置为使用人类作为模型，auto-coder.web 将无法工作，我们将设置 human_as_model 为 false"
+    }    
+}
+
+
+def get_system_language():
+    try:
+        return locale.getdefaultlocale()[0][:2]
+    except:
+        return 'en'
+
+
+def get_message(key):
+    lang = get_system_language()
+    if key in MESSAGES:
+        return MESSAGES[key].get(lang, MESSAGES[key].get("en", ""))
+    return ""
+
+
+def get_message_with_format(msg_key: str, **kwargs):
+    return format_str_jinja2(get_message(msg_key), **kwargs)