auto-coder-web 0.1.24__tar.gz → 0.1.25__tar.gz

{auto_coder_web-0.1.24 → auto_coder_web-0.1.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto_coder_web
-Version: 0.1.24
+Version: 0.1.25
 Summary: auto-coder.web: A Python Project
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.9

{auto_coder_web-0.1.24 → auto_coder_web-0.1.25}/src/auto_coder_web/auto_coder_runner_wrapper.py

@@ -46,6 +46,13 @@ class AutoCoderRunnerWrapper:
     
     def configure_wrapper(self,conf: str, skip_print=False ):
         return configure(conf, skip_print)
+
+
+    def coding_wapper(self,query):
+        return coding(query) 
+
+    def chat_wrapper(self,query):
+        return chat(query)          
         
     
     

auto_coder_web-0.1.25/src/auto_coder_web/common_router/auto_coder_conf_router.py

@@ -0,0 +1,42 @@
+from fastapi import APIRouter, Request, HTTPException, Depends
+
+router = APIRouter()
+
+async def get_auto_coder_runner(request: Request):
+    """获取AutoCoderRunner实例作为依赖"""
+    return request.app.state.auto_coder_runner
+
+@router.get("/api/conf")
+async def get_conf(
+    auto_coder_runner = Depends(get_auto_coder_runner)
+):
+    """获取配置信息"""
+    return {"conf": auto_coder_runner.get_config()}
+
+@router.post("/api/conf")
+async def config(
+    request: Request,
+    auto_coder_runner = Depends(get_auto_coder_runner)
+):
+    """更新配置信息"""
+    data = await request.json()
+    try:
+        for key, value in data.items():
+            auto_coder_runner.configure(key, str(value))
+        return {"status": "success"}
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+@router.delete("/api/conf/{key}")
+async def delete_config(
+    key: str,
+    auto_coder_runner = Depends(get_auto_coder_runner)
+):
+    """删除配置项"""
+    try:
+        result = auto_coder_runner.drop_config(key)
+        return result
+    except ValueError as e:
+        raise HTTPException(status_code=404, detail=str(e))
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e)) 

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

@@ -0,0 +1,75 @@
+import os
+import json
+from fastapi import APIRouter, HTTPException
+import aiofiles
+from auto_coder_web.types import ChatList
+
+router = APIRouter()
+
+@router.post("/api/chat-lists/save")
+async def save_chat_list(chat_list: ChatList):
+    try:
+        chat_lists_dir = os.path.join(
+            ".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))
+        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():
+    try:
+        chat_lists_dir = os.path.join(
+            ".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):
+    try:
+        file_path = os.path.join(
+            ".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):
+    try:
+        file_path = os.path.join(
+            ".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)) 

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

@@ -0,0 +1,53 @@
+import os
+from fastapi import APIRouter, Query, Request, Depends
+from auto_coder_web.types import CompletionItem, CompletionResponse
+
+router = APIRouter()
+
+async def get_auto_coder_runner(request: Request):
+    """获取AutoCoderRunner实例作为依赖"""
+    return request.app.state.auto_coder_runner
+
+@router.get("/api/completions/files")
+async def get_file_completions(
+    name: str = Query(...),
+    auto_coder_runner = Depends(get_auto_coder_runner)
+):
+    """获取文件名补全"""
+    matches = auto_coder_runner.find_files_in_project([name])
+    completions = []
+    project_root = auto_coder_runner.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(...),
+    auto_coder_runner = Depends(get_auto_coder_runner)
+):
+    """获取符号补全"""
+    symbols = auto_coder_runner.get_symbol_list()
+    matches = []
+
+    for symbol in symbols:
+        if name.lower() in symbol.symbol_name.lower():
+            relative_path = os.path.relpath(
+                symbol.file_name, auto_coder_runner.project_path)
+            matches.append(CompletionItem(
+                name=symbol.symbol_name,
+                path=f"{symbol.symbol_name} ({relative_path}/{symbol.symbol_type.value})",
+                display=f"{symbol.symbol_name}(location: {relative_path})"
+            ))
+    return CompletionResponse(completions=matches) 

auto_coder_web-0.1.25/src/auto_coder_web/common_router/file_group_router.py

@@ -0,0 +1,120 @@
+from fastapi import APIRouter, Request, HTTPException, Depends
+from autocoder.agent.auto_filegroup import AutoFileGroup
+from autocoder.utils import operate_config_api
+
+router = APIRouter()
+
+async def get_file_group_manager(request: Request):
+    """获取FileGroupManager实例作为依赖"""
+    return request.app.state.file_group_manager
+
+async def get_project_path(request: Request):
+    """获取项目路径作为依赖"""
+    return request.app.state.project_path
+
+async def get_auto_coder_runner(request: Request):
+    """获取AutoCoderRunner实例作为依赖"""
+    return request.app.state.auto_coder_runner
+
+@router.post("/api/file-groups")
+async def create_file_group(
+    request: Request,
+    file_group_manager = Depends(get_file_group_manager)
+):
+    data = await request.json()
+    name = data.get("name")
+    description = data.get("description", "")
+    group = await file_group_manager.create_group(name, description)
+    return group
+
+@router.post("/api/file-groups/auto")
+async def auto_create_groups(
+    request: Request,
+    file_group_manager = Depends(get_file_group_manager),
+    project_path: str = Depends(get_project_path),
+    auto_coder_runner = Depends(get_auto_coder_runner)
+):            
+    try:
+        data = await request.json()
+        file_size_limit = data.get("file_size_limit", 100)
+        skip_diff = data.get("skip_diff", False)
+        group_num_limit = data.get("group_num_limit", 10)
+
+        # Create AutoFileGroup instance
+        auto_grouper = AutoFileGroup(
+            operate_config_api.get_llm(auto_coder_runner.memory),
+            project_path,
+            skip_diff=skip_diff,
+            file_size_limit=file_size_limit,
+            group_num_limit=group_num_limit
+        )
+
+        # Get groups
+        groups = auto_grouper.group_files()
+
+        # Create groups using file_group_manager
+        for group in groups:
+            await file_group_manager.create_group(
+                name=group.name,
+                description=group.description
+            )
+            # Add files to the group
+            await file_group_manager.add_files_to_group(
+                group.name,
+                group.urls
+            )
+
+        return {"status": "success", "message": f"Created {len(groups)} groups"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.post("/api/file-groups/switch")
+async def switch_file_groups(
+    request: Request,
+    file_group_manager = Depends(get_file_group_manager)
+):
+    data = await request.json()
+    group_names = data.get("group_names", [])
+    result = await file_group_manager.switch_groups(group_names)
+    return result
+
+@router.delete("/api/file-groups/{name}")
+async def delete_file_group(
+    name: str,
+    file_group_manager = Depends(get_file_group_manager)
+):
+    await file_group_manager.delete_group(name)
+    return {"status": "success"}
+
+@router.post("/api/file-groups/{name}/files")
+async def add_files_to_group(
+    name: str, 
+    request: Request,
+    file_group_manager = Depends(get_file_group_manager)
+):
+    data = await request.json()
+    files = data.get("files", [])
+    description = data.get("description")
+    if description is not None:
+        group = await file_group_manager.update_group_description(name, description)
+    else:
+        group = await file_group_manager.add_files_to_group(name, files)
+    return group
+
+@router.delete("/api/file-groups/{name}/files")
+async def remove_files_from_group(
+    name: str, 
+    request: Request,
+    file_group_manager = Depends(get_file_group_manager)
+):
+    data = await request.json()
+    files = data.get("files", [])
+    group = await file_group_manager.remove_files_from_group(name, files)
+    return group
+
+@router.get("/api/file-groups")
+async def get_file_groups(
+    file_group_manager = Depends(get_file_group_manager)
+):
+    groups = await file_group_manager.get_groups()
+    return {"groups": groups} 

auto_coder_web-0.1.25/src/auto_coder_web/common_router/file_router.py

@@ -0,0 +1,78 @@
+import os
+import shutil
+from fastapi import APIRouter, Request, HTTPException, Depends
+from auto_coder_web.file_manager import get_directory_tree, read_file_content
+
+router = APIRouter()
+
+async def get_project_path(request: Request) -> str:
+    """获取项目路径作为依赖"""
+    return request.app.state.project_path
+
+async def get_auto_coder_runner(request: Request):
+    """获取AutoCoderRunner实例作为依赖"""
+    return request.app.state.auto_coder_runner
+
+@router.delete("/api/files/{path:path}")
+async def delete_file(
+    path: str,
+    project_path: str = Depends(get_project_path)
+):
+    try:
+        full_path = os.path.join(project_path, path)
+        if os.path.exists(full_path):
+            if os.path.isdir(full_path):
+                shutil.rmtree(full_path)
+            else:
+                os.remove(full_path)
+            return {"message": f"Successfully deleted {path}"}
+        else:
+            raise HTTPException(
+                status_code=404, detail="File not found")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/api/files")
+async def get_files(
+    project_path: str = Depends(get_project_path)
+):
+    tree = get_directory_tree(project_path)
+    return {"tree": tree}
+
+@router.put("/api/file/{path:path}")
+async def update_file(
+    path: str, 
+    request: Request,
+    project_path: str = Depends(get_project_path)
+):
+    try:
+        data = await request.json()
+        content = data.get("content")
+        if content is None:
+            raise HTTPException(
+                status_code=400, detail="Content is required")
+
+        full_path = os.path.join(project_path, path)
+
+        # Ensure the directory exists
+        os.makedirs(os.path.dirname(full_path), exist_ok=True)
+
+        # Write the file content
+        with open(full_path, 'w', encoding='utf-8') as f:
+            f.write(content)
+
+        return {"message": f"Successfully updated {path}"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/api/file/{path:path}")
+async def get_file_content(
+    path: str,
+    project_path: str = Depends(get_project_path)
+):
+    content = read_file_content(project_path, path)
+    if content is None:
+        raise HTTPException(
+            status_code=404, detail="File not found or cannot be read")
+
+    return {"content": content}