auto-coder-web 0.1.109__py3-none-any.whl → 0.1.111__py3-none-any.whl

auto_coder_web/auto_coder_runner_wrapper.py

@@ -61,12 +61,16 @@ class AutoCoderRunnerWrapper:
         memory = get_memory()
         return memory["conf"]
 
+    def get_memory_wrapper(self):
+        return get_memory()
 
     def coding_wapper(self,query):
         return coding(query) 
 
     def chat_wrapper(self,query):
-        return chat(query)          
-        
-    
+        return chat(query)       
+
+    def lib_wrapper(self,query):
+        return lib_command(query)   
+            
     

auto_coder_web/common_router/lib_router.py

@@ -0,0 +1,325 @@
+from fastapi import APIRouter, Depends, Request, HTTPException, Body
+from pydantic import BaseModel
+from typing import Dict, List, Optional
+import os
+import git
+import json
+from pathlib import Path
+from autocoder.auto_coder_runner import get_memory, save_memory_with_new_memory
+
+router = APIRouter()
+
+# 定义依赖项函数
+async def get_project_path(request: Request) -> str:
+    """获取项目路径作为依赖"""
+    return request.app.state.project_path
+
+
+# 模型定义
+class LibAddRequest(BaseModel):
+    lib_name: str
+
+class LibRemoveRequest(BaseModel):
+    lib_name: str
+
+class LibProxyRequest(BaseModel):
+    proxy_url: Optional[str] = None
+
+class LibGetRequest(BaseModel):
+    package_name: str
+
+class LibResponse(BaseModel):
+    success: bool
+    message: str
+    data: Optional[Dict] = None
+
+class LibItem(BaseModel):
+    domain: str
+    username: str
+    lib_name: str
+    full_path: str
+    is_added: bool
+
+@router.get("/api/lib/list", response_model=LibResponse)
+async def list_libs(
+    request: Request,
+    memory: Dict = Depends(get_memory)
+):
+    """列出所有已添加的库"""
+    libs = memory.get("libs", {})
+    
+    return LibResponse(
+        success=True,
+        message="Libraries retrieved successfully",
+        data={"libraries": list(libs.keys())}
+    )
+
+@router.post("/api/lib/add", response_model=LibResponse)
+async def add_lib(
+    request: Request,
+    lib_request: LibAddRequest,
+    memory: Dict = Depends(get_memory),
+    project_path: str = Depends(get_project_path)
+):
+    """添加一个库"""
+    lib_name = lib_request.lib_name.strip()
+    
+    # 初始化目录
+    lib_dir = os.path.join(project_path, ".auto-coder", "libs")
+    llm_friendly_packages_dir = os.path.join(lib_dir, "llm_friendly_packages")
+    
+    if not os.path.exists(lib_dir):
+        os.makedirs(lib_dir, exist_ok=True)
+        
+    if "libs" not in memory:
+        memory["libs"] = {}
+        
+    # 已存在的库直接返回
+    if lib_name in memory["libs"]:
+        return LibResponse(
+            success=True,
+            message=f"Library {lib_name} is already added"
+        )
+        
+    # 克隆仓库(如果不存在)
+    if not os.path.exists(llm_friendly_packages_dir):
+        try:
+            proxy_url = memory.get(
+                "lib-proxy", "https://github.com/allwefantasy/llm_friendly_packages"
+            )
+            git.Repo.clone_from(
+                proxy_url,
+                llm_friendly_packages_dir,
+            )
+        except git.exc.GitCommandError as e:
+            return LibResponse(
+                success=False,
+                message=f"Error cloning repository: {str(e)}"
+            )
+    
+    # 添加库到内存
+    memory["libs"][lib_name] = {}
+    await save_memory_with_new_memory(memory)
+    
+    return LibResponse(
+        success=True,
+        message=f"Added library: {lib_name}"
+    )
+
+@router.post("/api/lib/remove", response_model=LibResponse)
+async def remove_lib(
+    request: Request,
+    lib_request: LibRemoveRequest,
+    memory: Dict = Depends(get_memory)
+):
+    """移除一个库"""
+    lib_name = lib_request.lib_name.strip()
+    
+    if "libs" not in memory:
+        memory["libs"] = {}
+        
+    if lib_name in memory["libs"]:
+        del memory["libs"][lib_name]
+        await save_memory_with_new_memory(memory)
+        return LibResponse(
+            success=True,
+            message=f"Removed library: {lib_name}"
+        )
+    else:
+        return LibResponse(
+            success=False,
+            message=f"Library {lib_name} is not in the list"
+        )
+
+@router.post("/api/lib/set-proxy", response_model=LibResponse)
+async def set_proxy(
+    request: Request,
+    proxy_request: LibProxyRequest = Body(None),
+    memory: Dict = Depends(get_memory)
+):
+    """设置代理URL"""
+    if proxy_request is None or proxy_request.proxy_url is None:
+        # 获取当前代理
+        current_proxy = memory.get("lib-proxy", "No proxy set")
+        return LibResponse(
+            success=True,
+            message=f"Current proxy: {current_proxy}",
+            data={"proxy": current_proxy}
+        )
+    else:
+        # 设置代理
+        memory["lib-proxy"] = proxy_request.proxy_url
+        await save_memory_with_new_memory(memory)
+        return LibResponse(
+            success=True,
+            message=f"Set proxy to: {proxy_request.proxy_url}"
+        )
+
+@router.post("/api/lib/refresh", response_model=LibResponse)
+async def refresh_lib(
+    request: Request,
+    memory: Dict = Depends(get_memory),
+    project_path: str = Depends(get_project_path)
+):
+    """刷新llm_friendly_packages仓库"""
+    llm_friendly_packages_dir = os.path.join(project_path, ".auto-coder", "libs", "llm_friendly_packages")
+    
+    if not os.path.exists(llm_friendly_packages_dir):
+        return LibResponse(
+            success=False,
+            message="llm_friendly_packages repository does not exist. Please add a library first to clone it."
+        )
+        
+    try:
+        repo = git.Repo(llm_friendly_packages_dir)
+        origin = repo.remotes.origin
+        proxy_url = memory.get("lib-proxy")
+        
+        current_url = origin.url
+        
+        if proxy_url and proxy_url != current_url:
+            new_url = proxy_url
+            origin.set_url(new_url)
+        
+        origin.pull()
+        return LibResponse(
+            success=True,
+            message="Successfully updated llm_friendly_packages repository"
+        )
+    except git.exc.GitCommandError as e:
+        return LibResponse(
+            success=False,
+            message=f"Error updating repository: {str(e)}"
+        )
+
+@router.post("/api/lib/get", response_model=LibResponse)
+async def get_lib_docs(
+    request: Request,
+    get_request: LibGetRequest,
+    project_path: str = Depends(get_project_path),
+    memory: Dict = Depends(get_memory)
+):
+    """获取特定包的文档"""
+    package_name = get_request.package_name.strip()
+    
+    # 使用与auto_coder_runner中相似的逻辑
+    lib_dir = os.path.join(project_path, ".auto-coder", "libs")
+    llm_friendly_packages_dir = os.path.join(lib_dir, "llm_friendly_packages")
+    docs = []
+    
+    if not os.path.exists(llm_friendly_packages_dir):
+        return LibResponse(
+            success=False,
+            message="llm_friendly_packages repository does not exist"
+        )
+        
+    libs = list(memory.get("libs", {}).keys())
+    
+    for domain in os.listdir(llm_friendly_packages_dir):
+        domain_path = os.path.join(llm_friendly_packages_dir, domain)
+        if os.path.isdir(domain_path):
+            for username in os.listdir(domain_path):
+                username_path = os.path.join(domain_path, username)
+                if os.path.isdir(username_path):
+                    for lib_name in os.listdir(username_path):
+                        lib_path = os.path.join(username_path, lib_name)
+                        if (
+                            os.path.isdir(lib_path)
+                            and (
+                                package_name is None
+                                or lib_name == package_name
+                                or package_name == os.path.join(username, lib_name)
+                            )
+                            and lib_name in libs
+                        ):
+                            for root, _, files in os.walk(lib_path):
+                                for file in files:
+                                    if file.endswith(".md"):
+                                        file_path = os.path.join(root, file)
+                                        docs.append(str(Path(file_path).relative_to(project_path)))
+    
+    if docs:
+        return LibResponse(
+            success=True,
+            message=f"Found {len(docs)} markdown files for package: {package_name}",
+            data={"docs": docs}
+        )
+    else:
+        return LibResponse(
+            success=False,
+            message=f"No markdown files found for package: {package_name}"
+        )
+
+@router.get("/api/lib/list-all", response_model=LibResponse)
+async def list_all_libs(
+    request: Request,
+    memory: Dict = Depends(get_memory),
+    project_path: str = Depends(get_project_path)
+):
+    """列出所有可用的库，包括未添加的库"""
+    lib_dir = os.path.join(project_path, ".auto-coder", "libs")
+    llm_friendly_packages_dir = os.path.join(lib_dir, "llm_friendly_packages")
+    available_libs = []
+    
+    # 如果仓库不存在，自动下载
+    if not os.path.exists(llm_friendly_packages_dir):
+        if not os.path.exists(lib_dir):
+            os.makedirs(lib_dir, exist_ok=True)
+            
+        try:
+            # 获取代理URL（如果已设置）
+            proxy_url = memory.get(
+                "lib-proxy", "https://github.com/allwefantasy/llm_friendly_packages"
+            )
+            # 克隆仓库
+            git.Repo.clone_from(
+                proxy_url,
+                llm_friendly_packages_dir,
+            )
+        except git.exc.GitCommandError as e:
+            return LibResponse(
+                success=False,
+                message=f"Error cloning repository: {str(e)}",
+                data={"libraries": []}
+            )
+    
+    # 获取已添加的库列表
+    added_libs = set(memory.get("libs", {}).keys())
+    
+    # 遍历所有domain目录
+    for domain in os.listdir(llm_friendly_packages_dir):
+        domain_path = os.path.join(llm_friendly_packages_dir, domain)
+        if os.path.isdir(domain_path):
+            # 遍历所有username目录
+            for username in os.listdir(domain_path):
+                username_path = os.path.join(domain_path, username)
+                if os.path.isdir(username_path):
+                    # 遍历所有lib_name目录
+                    for lib_name in os.listdir(username_path):
+                        lib_path = os.path.join(username_path, lib_name)
+                        if os.path.isdir(lib_path):
+                            # 检查是否有Markdown文件
+                            has_md_files = False
+                            for root, _, files in os.walk(lib_path):
+                                if any(file.endswith('.md') for file in files):
+                                    has_md_files = True
+                                    break
+                            
+                            if has_md_files:
+                                lib_item = LibItem(
+                                    domain=domain,
+                                    username=username,
+                                    lib_name=lib_name,
+                                    full_path=f"{username}/{lib_name}",
+                                    is_added=lib_name in added_libs
+                                )
+                                available_libs.append(lib_item.dict())
+    
+    # 按domain和username分组排序
+    available_libs.sort(key=lambda x: (x["domain"], x["username"], x["lib_name"]))
+    
+    return LibResponse(
+        success=True,
+        message=f"Found {len(available_libs)} available libraries",
+        data={"libraries": available_libs}
+    ) 

auto_coder_web/common_router/model_router.py

@@ -264,6 +264,22 @@ def load_providers() -> List[Dict]:
                     "output_price": 0.0,
                     "is_reasoning": False,
                     "max_output_tokens": 8096*2
+                },
+                {
+                    "id": "anthropic/claude-sonnet-4",
+                    "name": "claude-sonnet-4",
+                    "input_price": 22.0,
+                    "output_price": 111.0,
+                    "is_reasoning": False,
+                    "max_output_tokens": 8096*4
+                },
+                {
+                    "id": "anthropic/claude-opus-4",
+                    "name": "claude-opus-4",
+                    "input_price": 111.0,
+                    "output_price": 555.0,
+                    "is_reasoning": False,
+                    "max_output_tokens": 8096*4
                 }
             ]
         },        

auto_coder_web/proxy.py

@@ -21,7 +21,7 @@ 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, direct_chat_router, rules_router, chat_panels_router, code_editor_tabs_router, file_command_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 completions_router, file_router, auto_coder_conf_router, chat_list_router, file_group_router, model_router, compiler_router, lib_router
 from auto_coder_web.common_router import active_context_router
 from rich.console import Console
 from loguru import logger
@@ -86,6 +86,9 @@ class ProxyServer:
         # self.app.state.file_cacher = FileCacher(self.project_path)
         # Store initialization status
         self.app.state.is_initialized = self.is_initialized
+        # Store memory for lib_router
+        if self.auto_coder_runner:
+            self.app.state.memory = self.auto_coder_runner.get_memory_wrapper()
 
         self.app.include_router(todo_router.router)
         self.app.include_router(settings_router.router)
@@ -113,6 +116,7 @@ class ProxyServer:
         self.app.include_router(chat_panels_router.router)
         self.app.include_router(code_editor_tabs_router.router)
         self.app.include_router(file_command_router.router)
+        self.app.include_router(lib_router.router)
 
         @self.app.on_event("shutdown")
         async def shutdown_event():

auto_coder_web/version.py

@@ -1 +1 @@
-__version__ = "0.1.109"
+__version__ = "0.1.111"