auto-coder-web 0.1.38__py3-none-any.whl → 0.1.40__py3-none-any.whl

auto_coder_web/auto_coder_runner_wrapper.py

@@ -33,13 +33,18 @@ from autocoder.auto_coder_runner import (
     completer,
     summon,
     get_memory,
+    get_all_extensions
 )
 
 class AutoCoderRunnerWrapper:
     def __init__(self, project_path: str, product_mode: str = "lite"):
+        self.project_path = project_path
+        self.product_mode = product_mode
         load_memory()
         load_tokenizer()
-
+    
+    def get_all_extensions_wrapper(self):
+        return get_all_extensions(self.project_path)
 
     def auto_command_wrapper(self, command: str, params: Dict[str, Any]) -> Dict[str, str]:
         return auto_command(command,params)

auto_coder_web/common_router/model_router.py

@@ -1,10 +1,22 @@
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Request
 from typing import List, Dict, Optional
+import json
+import os
 from pydantic import BaseModel
 from autocoder import models as model_utils
 
 router = APIRouter()
 
+async def get_project_path(request: Request):
+    """获取项目路径作为依赖"""
+    return request.app.state.project_path
+
+# Path for providers JSON file
+PROVIDERS_FILE = os.path.expanduser("~/.auto-coder/keys/models_providers.json")
+
+# Ensure directory exists
+os.makedirs(os.path.dirname(PROVIDERS_FILE), exist_ok=True)
+
 class Model(BaseModel):
     name: str
     description: str = ""
@@ -154,3 +166,147 @@ async def update_model_speed(model_name: str, speed: float):
             raise HTTPException(status_code=404, detail="Model not found")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+
+# Provider management endpoints
+class ModelInfo(BaseModel):
+    id: str
+    name: str
+    input_price: float
+    output_price: float
+    is_reasoning: bool
+
+class ProviderConfig(BaseModel):
+    name: str
+    base_url: str
+    models: List[ModelInfo]
+
+def load_providers() -> List[Dict]:
+    """Load providers from JSON file"""
+    # Default providers if file doesn't exist
+    default_providers = [
+        {
+            "name": "volcanoEngine",
+            "base_url": "https://ark.cn-beijing.volces.com/api/v3",
+            "models": [
+                {
+                    "id": "deepseek-v3-241226",
+                    "name": "Deepseek V3",
+                    "input_price": 1.0,
+                    "output_price": 4.0,
+                    "is_reasoning": False
+                },
+                {
+                    "id": "deepseek-r1-250120",
+                    "name": "Deepseek R1",
+                    "input_price": 2.0,
+                    "output_price": 8.0,
+                    "is_reasoning": True
+                }
+            ]
+        },
+        {
+            "name": "openrouter",
+            "base_url": "https://openrouter.ai/api/v1",
+            "models": [
+                {
+                    "id": "anthropic/claude-3.7-sonnet:thinking",
+                    "name": "Claude 3.7 Sonnet Thinking",
+                    "input_price": 22.0,
+                    "output_price": 111.0,
+                    "is_reasoning": True
+                },
+                {
+                    "id": "anthropic/claude-3.7-sonnet",
+                    "name": "Claude 3.7 Sonnet",
+                    "input_price": 22.0,
+                    "output_price": 111.0,
+                    "is_reasoning": False
+                }
+            ]
+        }
+    ]
+    
+    if not os.path.exists(PROVIDERS_FILE):
+        return default_providers
+    try:
+        with open(PROVIDERS_FILE, 'r',encoding='utf-8') as f:
+            # 根据名字去重，优先保留文件中的提供商配置
+            loaded_providers = json.load(f)
+            providers_map = {provider["name"]: provider for provider in loaded_providers}
+            
+            # 只添加名字不重复的默认提供商
+            for default_provider in default_providers:
+                if default_provider["name"] not in providers_map:
+                    providers_map[default_provider["name"]] = default_provider
+            
+            return list(providers_map.values())
+    except Exception as e:
+        print(f"Error loading providers: {e}")
+        return default_providers
+
+def save_providers(providers: List[Dict]) -> None:
+    """Save providers to JSON file"""
+    with open(PROVIDERS_FILE, 'w',encoding='utf-8') as f:
+        # 根据名字去重，然后再统一保存        
+        json.dump(providers, f, indent=2,ensure_ascii=False)
+
+@router.get("/api/providers", response_model=List[ProviderConfig])
+async def get_providers():
+    """Get all available providers"""
+    try:
+        providers = load_providers()
+        return providers
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.post("/api/providers", response_model=ProviderConfig)
+async def add_provider(provider: ProviderConfig):
+    """Add a new provider"""
+    try:
+        providers = load_providers()
+        
+        # Check if provider with same name already exists
+        if any(p["name"] == provider.name for p in providers):
+            raise HTTPException(status_code=400, detail="Provider with this name already exists")
+        
+        providers.append(provider.model_dump())
+        save_providers(providers)
+        return provider
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.put("/api/providers/{provider_name}", response_model=ProviderConfig)
+async def update_provider(provider_name: str, provider: ProviderConfig):
+    """Update an existing provider"""
+    try:
+        providers = load_providers()
+        updated = False
+        
+        for p in providers:
+            if p["name"] == provider_name:
+                p.update(provider.model_dump())
+                updated = True
+                break
+        
+        if not updated:
+            raise HTTPException(status_code=404, detail="Provider not found")
+            
+        save_providers(providers)
+        return provider
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@router.delete("/api/providers/{provider_name}")
+async def delete_provider(provider_name: str):
+    """Delete a provider by name"""
+    try:
+        providers = load_providers()
+        providers_list = [p for p in providers if p["name"] != provider_name]
+        
+        if len(providers) == len(providers_list):
+            raise HTTPException(status_code=404, detail="Provider not found")
+            
+        save_providers(providers_list)
+        return {"message": f"Provider {provider_name} deleted successfully"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

auto_coder_web/init_project.py

@@ -0,0 +1,26 @@
+import os
+from autocoder.common import git_utils
+def init_project(project_path: str):
+    os.makedirs(os.path.join(project_path, "actions"), exist_ok=True)
+    os.makedirs(os.path.join(project_path,
+                ".auto-coder"), exist_ok=True)
+
+    from autocoder.common.command_templates import create_actions
+
+    source_dir = os.path.abspath(project_path)
+    create_actions(
+        source_dir=source_dir,
+        params={"project_type": "python",
+                "source_dir": source_dir},
+    )
+    git_utils.init(os.path.abspath(project_path))
+
+    with open(os.path.join(source_dir, ".gitignore"), "a") as f:
+        f.write("\n.auto-coder/")
+        f.write("\n/actions/")
+        f.write("\n/output.txt")
+
+    print(
+        f"""Successfully initialized auto-coder project in {os.path.abspath(project_path)}."""
+    )
+    return

auto_coder_web/proxy.py

@@ -31,26 +31,31 @@ from loguru import logger
 from auto_coder_web.lang import get_message
 
 class ProxyServer:
-    def __init__(self, project_path: str, quick: bool = False, product_mode: str = "pro"):
+    def __init__(self, project_path: str, quick: bool = False, product_mode: str = "pro"):    
         self.app = FastAPI()                        
-        self.setup_middleware()
+        self.setup_middleware()        
 
         self.setup_static_files()
         self.project_path = project_path
-        
+        self.product_mode = product_mode
+        self.auto_coder_runner = None                
         # Check if project is initialized
         self.is_initialized = self.check_project_initialization()
-        if not self.is_initialized:
+        if not self.is_initialized and product_mode == "pro":
             logger.warning(get_message("project_not_initialized"))
             logger.warning(get_message("run_auto_coder_chat"))
-            sys.exit(1)            
-
-        self.auto_coder_runner = AutoCoderRunnerWrapper(project_path, product_mode=product_mode)        
-         
+            sys.exit(1) 
 
+        if self.is_initialized:
+            self._initialize()                    
+        
         self.setup_routes()        
+
+    def _initialize(self):
+        self.auto_coder_runner = AutoCoderRunnerWrapper(self.project_path, product_mode=self.product_mode)        
         self.client = httpx.AsyncClient()
 
+
     def setup_middleware(self):
         self.app.add_middleware(
             CORSMiddleware,
@@ -127,10 +132,7 @@ class ProxyServer:
 
         @self.app.get("/api/project-path")
         async def get_project_path():
-            return {"project_path": self.project_path}
-
-        def get_project_runner(project_path: str) -> AutoCoderRunner:
-            return self.projects[project_path]
+            return {"project_path": self.project_path}        
 
         @self.app.get("/api/os")
         async def get_os():
@@ -162,65 +164,33 @@ class ProxyServer:
                     "default": field.default
                 })
             return {"keys": keys}
+        
+        
+        @self.app.post("/api/initialization-project")
+        async def initialization_project():
+            """Get the project initialization status"""
+            from auto_coder_web.init_project import init_project
+            init_project(self.project_path)
+            base_persist_dir = os.path.join(self.project_path,".auto-coder", "plugins", "chat-auto-coder")
+            os.makedirs(base_persist_dir, exist_ok=True)
+            self.is_initialized = True
+            self._initialize()
+            return {"success": True}
+        
+        @self.app.get("/api/guess/project_type")
+        async def get_project_type():
+            v = self.auto_coder_runner.get_all_extensions_wrapper()
+            return {
+                "project_type":v
+            }
+        
+        @self.app.put("/api/congigure/project_type")
+        async def configure_project_type(project_type:str):
+            self.auto_coder_runner.configure_wrapper(f"project_type:{project_type}")
+            return {
+                "succcess": True
+            }
 
-        @self.app.post("/api/revert")
-        async def revert():
-            try:
-                result = self.auto_coder_runner.revert()
-                return result
-            except Exception as e:
-                raise HTTPException(status_code=500, detail=str(e))
-
-        @self.app.get("/api/active-files")
-        async def get_active_files():
-            """获取当前活动文件列表"""
-            active_files = self.auto_coder_runner.get_active_files()
-            return active_files
-
-        @self.app.post("/api/commit")
-        async def commit():
-            try:
-                result = self.auto_coder_runner.commit()
-                return result
-            except Exception as e:
-                raise HTTPException(status_code=500, detail=str(e))        
-
-        @self.app.get("/api/last-yaml")
-        async def get_last_yaml():
-            """Get information about the last YAML file"""
-            return JSONResponse(content=self.auto_coder_runner.get_last_yaml_info())
-
-        @self.app.get("/api/history/file-content/{file_number}", response_model=FileContentResponse)
-        async def get_file_content(file_number: int):
-            """获取指定编号文件的完整内容"""
-            auto_coder_dir = "actions"
-            file_name = f"{file_number}_chat_action.yml"
-            file_path = ""
-            
-            # 搜索文件
-            for root, _, files in os.walk(auto_coder_dir):
-                if file_name in files:
-                    file_path = os.path.join(root, file_name)
-                    break
-                    
-            if not file_path:
-                return FileContentResponse(
-                    success=False,
-                    message=f"找不到文件: {file_name}"
-                )
-                
-            try:
-                with open(file_path, 'r', encoding='utf-8') as f:
-                    content = f.read()
-                return FileContentResponse(
-                    success=True,
-                    content=content
-                )
-            except Exception as e:
-                return FileContentResponse(
-                    success=False, 
-                    message=f"读取文件出错: {str(e)}"
-                )
 
         @self.app.get("/api/initialization-status")
         async def get_initialization_status():

auto_coder_web/version.py

@@ -1 +1 @@
-__version__ = "0.1.38"
+__version__ = "0.1.40"

auto_coder_web/web/asset-manifest.json

@@ -1,15 +1,15 @@
 {
   "files": {
-    "main.css": "/static/css/main.0ef9afe0.css",
-    "main.js": "/static/js/main.2922014a.js",
+    "main.css": "/static/css/main.d58ae48e.css",
+    "main.js": "/static/js/main.8d7d4613.js",
     "static/js/453.d855a71b.chunk.js": "/static/js/453.d855a71b.chunk.js",
     "index.html": "/index.html",
-    "main.0ef9afe0.css.map": "/static/css/main.0ef9afe0.css.map",
-    "main.2922014a.js.map": "/static/js/main.2922014a.js.map",
+    "main.d58ae48e.css.map": "/static/css/main.d58ae48e.css.map",
+    "main.8d7d4613.js.map": "/static/js/main.8d7d4613.js.map",
     "453.d855a71b.chunk.js.map": "/static/js/453.d855a71b.chunk.js.map"
   },
   "entrypoints": [
-    "static/css/main.0ef9afe0.css",
-    "static/js/main.2922014a.js"
+    "static/css/main.d58ae48e.css",
+    "static/js/main.8d7d4613.js"
   ]
 }

auto_coder_web/web/index.html

@@ -1 +1 @@
-<!doctype html><html lang="en"><head><meta charset="utf-8"/><link rel="icon" href="/favicon.ico"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><meta name="description" content="Web site created using create-react-app"/><link rel="apple-touch-icon" href="/logo192.png"/><link rel="manifest" href="/manifest.json"/><title>React App</title><script defer="defer" src="/static/js/main.2922014a.js"></script><link href="/static/css/main.0ef9afe0.css" rel="stylesheet"></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>
+<!doctype html><html lang="en"><head><meta charset="utf-8"/><link rel="icon" href="/favicon.ico"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><meta name="description" content="Web site created using create-react-app"/><link rel="apple-touch-icon" href="/logo192.png"/><link rel="manifest" href="/manifest.json"/><title>React App</title><script defer="defer" src="/static/js/main.8d7d4613.js"></script><link href="/static/css/main.d58ae48e.css" rel="stylesheet"></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>