auto-coder-web 0.1.37__py3-none-any.whl → 0.1.39__py3-none-any.whl

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 = ""
@@ -17,7 +29,7 @@ class Model(BaseModel):
     output_price: float = 0.0
     average_speed: float = 0.0
 
-@router.get("/models", response_model=List[Model])
+@router.get("/api/models", response_model=List[Model])
 async def get_models():
     """
     Get all available models
@@ -28,7 +40,7 @@ async def get_models():
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/models/{model_name}", response_model=Model)
+@router.get("/api/models/{model_name}", response_model=Model)
 async def get_model(model_name: str):
     """
     Get a specific model by name
@@ -39,13 +51,13 @@ async def get_model(model_name: str):
     except Exception as e:
         raise HTTPException(status_code=404, detail=str(e))
 
-@router.post("/models", response_model=Model)
+@router.post("/api/models", response_model=Model)
 async def add_model(model: Model):
     """
     Add a new model
     """
     try:
-        existing_models = models.load_models()
+        existing_models = model_utils.load_models()
         if any(m["name"] == model.name for m in existing_models):
             raise HTTPException(status_code=400, detail="Model with this name already exists")
         
@@ -55,7 +67,7 @@ async def add_model(model: Model):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.put("/models/{model_name}", response_model=Model)
+@router.put("/api/models/{model_name}", response_model=Model)
 async def update_model(model_name: str, model: Model):
     """
     Update an existing model
@@ -78,7 +90,7 @@ async def update_model(model_name: str, model: Model):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.delete("/models/{model_name}")
+@router.delete("/api/models/{model_name}")
 async def delete_model(model_name: str):
     """
     Delete a model by name
@@ -95,7 +107,7 @@ async def delete_model(model_name: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.put("/models/{model_name}/api_key")
+@router.put("/api/models/{model_name}/api_key")
 async def update_model_api_key(model_name: str, api_key: str):
     """
     Update the API key for a specific model
@@ -109,7 +121,7 @@ async def update_model_api_key(model_name: str, api_key: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.put("/models/{model_name}/input_price")
+@router.put("/api/models/{model_name}/input_price")
 async def update_model_input_price(model_name: str, price: float):
     """
     Update the input price for a specific model
@@ -125,7 +137,7 @@ async def update_model_input_price(model_name: str, price: float):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.put("/models/{model_name}/output_price")
+@router.put("/api/models/{model_name}/output_price")
 async def update_model_output_price(model_name: str, price: float):
     """
     Update the output price for a specific model
@@ -141,7 +153,7 @@ async def update_model_output_price(model_name: str, price: float):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.put("/models/{model_name}/speed")
+@router.put("/api/models/{model_name}/speed")
 async def update_model_speed(model_name: str, speed: float):
     """
     Update the average speed for a specific model
@@ -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/version.py

@@ -1 +1 @@
-__version__ = "0.1.37"
+__version__ = "0.1.39"

auto_coder_web/web/asset-manifest.json

@@ -1,15 +1,15 @@
 {
   "files": {
-    "main.css": "/static/css/main.e70fd4a5.css",
-    "main.js": "/static/js/main.066e08e9.js",
+    "main.css": "/static/css/main.0ef9afe0.css",
+    "main.js": "/static/js/main.c3e68e66.js",
     "static/js/453.d855a71b.chunk.js": "/static/js/453.d855a71b.chunk.js",
     "index.html": "/index.html",
-    "main.e70fd4a5.css.map": "/static/css/main.e70fd4a5.css.map",
-    "main.066e08e9.js.map": "/static/js/main.066e08e9.js.map",
+    "main.0ef9afe0.css.map": "/static/css/main.0ef9afe0.css.map",
+    "main.c3e68e66.js.map": "/static/js/main.c3e68e66.js.map",
     "453.d855a71b.chunk.js.map": "/static/js/453.d855a71b.chunk.js.map"
   },
   "entrypoints": [
-    "static/css/main.e70fd4a5.css",
-    "static/js/main.066e08e9.js"
+    "static/css/main.0ef9afe0.css",
+    "static/js/main.c3e68e66.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.066e08e9.js"></script><link href="/static/css/main.e70fd4a5.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.c3e68e66.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>