cli-anything-cliproxyapi 1.0.0__py3-none-any.whl

cli_anything/cliproxyapi/core/models.py

@@ -0,0 +1,65 @@
+"""模型管理模块。"""
+
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from .client import ManagementClient
+
+
+class ModelManager:
+    """管理 CLIProxyAPI 的模型列表、别名和排除规则。"""
+
+    def __init__(self, client: ManagementClient):
+        self.client = client
+
+    def list_models(self, api_key: str) -> dict:
+        resp = self.client.proxy_get("/v1/models", api_key)
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_oauth_model_alias(self) -> dict:
+        resp = self.client.get("/oauth-model-alias")
+        resp.raise_for_status()
+        return resp.json()
+
+    def put_oauth_model_alias(self, aliases: dict) -> dict:
+        resp = self.client.put("/oauth-model-alias", json_data=aliases)
+        resp.raise_for_status()
+        return resp.json()
+
+    def patch_oauth_model_alias(self, channel: str, entry: dict) -> dict:
+        resp = self.client.patch("/oauth-model-alias", json_data={"channel": channel, **entry})
+        resp.raise_for_status()
+        return resp.json()
+
+    def delete_oauth_model_alias(self, channel: str, name: str = "") -> dict:
+        data: dict[str, Any] = {"channel": channel}
+        if name:
+            data["name"] = name
+        resp = self.client.delete("/oauth-model-alias", json_data=data)
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_oauth_excluded_models(self) -> dict:
+        resp = self.client.get("/oauth-excluded-models")
+        resp.raise_for_status()
+        return resp.json()
+
+    def put_oauth_excluded_models(self, exclusions: dict) -> dict:
+        resp = self.client.put("/oauth-excluded-models", json_data=exclusions)
+        resp.raise_for_status()
+        return resp.json()
+
+    def patch_oauth_excluded_models(self, channel: str, models: list) -> dict:
+        resp = self.client.patch("/oauth-excluded-models", json_data={"channel": channel, "models": models})
+        resp.raise_for_status()
+        return resp.json()
+
+    def delete_oauth_excluded_models(self, channel: str, model: str = "") -> dict:
+        data: dict[str, Any] = {"channel": channel}
+        if model:
+            data["model"] = model
+        resp = self.client.delete("/oauth-excluded-models", json_data=data)
+        resp.raise_for_status()
+        return resp.json()

cli_anything/cliproxyapi/core/oauth.py

@@ -0,0 +1,57 @@
+"""OAuth 登录流程模块。"""
+
+from typing import Optional
+
+from .client import ManagementClient
+
+
+class OAuthManager:
+    """管理 CLIProxyAPI 的 OAuth 登录流程。"""
+
+    PROVIDERS = {
+        "anthropic": {"path": "/anthropic-auth-url", "label": "Claude"},
+        "codex": {"path": "/codex-auth-url", "label": "Codex"},
+        "gemini": {"path": "/gemini-cli-auth-url", "label": "Gemini CLI"},
+        "antigravity": {"path": "/antigravity-auth-url", "label": "Antigravity"},
+        "qwen": {"path": "/qwen-auth-url", "label": "Qwen Code"},
+        "kimi": {"path": "/kimi-auth-url", "label": "Kimi"},
+        "iflow": {"path": "/iflow-auth-url", "label": "iFlow"},
+    }
+
+    def __init__(self, client: ManagementClient):
+        self.client = client
+
+    def request_auth_url(self, provider: str, no_browser: bool = False) -> dict:
+        if provider not in self.PROVIDERS:
+            raise ValueError(f"不支持提供商: {provider}，可选: {', '.join(self.PROVIDERS)}")
+
+        info = self.PROVIDERS[provider]
+        params = {}
+        if no_browser:
+            params["no_browser"] = "true"
+
+        if provider == "iflow":
+            resp = self.client.get(info["path"], params=params)
+        else:
+            resp = self.client.get(info["path"], params=params)
+        resp.raise_for_status()
+        return resp.json()
+
+    def request_iflow_cookie(self, cookie: str) -> dict:
+        resp = self.client.post("/iflow-auth-url", json_data={"cookie": cookie})
+        resp.raise_for_status()
+        return resp.json()
+
+    def post_oauth_callback(self, provider: str, code: str, state: str) -> dict:
+        resp = self.client.post("/oauth-callback", json_data={
+            "provider": provider,
+            "code": code,
+            "state": state,
+        })
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_auth_status(self, session_id: str) -> dict:
+        resp = self.client.get("/get-auth-status", params={"session_id": session_id})
+        resp.raise_for_status()
+        return resp.json()

cli_anything/cliproxyapi/core/proxy.py

@@ -0,0 +1,145 @@
+"""代理/服务器管理模块。"""
+
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from .client import ManagementClient
+
+
+class ProxyManager:
+    """管理 CLIProxyAPI 的代理设置和 Amp 集成。"""
+
+    def __init__(self, client: ManagementClient):
+        self.client = client
+
+    # ---- 服务器状态 ----
+
+    def health_check(self) -> dict:
+        resp = self.client.health_check()
+        resp.raise_for_status()
+        return resp.json()
+
+    # ---- Amp 集成 ----
+
+    def get_amp_config(self) -> dict:
+        resp = self.client.get("/ampcode")
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_amp_upstream_url(self) -> Optional[str]:
+        resp = self.client.get("/ampcode/upstream-url")
+        resp.raise_for_status()
+        return resp.json().get("value")
+
+    def set_amp_upstream_url(self, url: str) -> dict:
+        resp = self.client.put("/ampcode/upstream-url", json_data={"value": url})
+        resp.raise_for_status()
+        return resp.json()
+
+    def delete_amp_upstream_url(self) -> dict:
+        resp = self.client.delete("/ampcode/upstream-url")
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_amp_upstream_api_key(self) -> Optional[str]:
+        resp = self.client.get("/ampcode/upstream-api-key")
+        resp.raise_for_status()
+        return resp.json().get("value")
+
+    def set_amp_upstream_api_key(self, key: str) -> dict:
+        resp = self.client.put("/ampcode/upstream-api-key", json_data={"value": key})
+        resp.raise_for_status()
+        return resp.json()
+
+    def delete_amp_upstream_api_key(self) -> dict:
+        resp = self.client.delete("/ampcode/upstream-api-key")
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_amp_model_mappings(self) -> list:
+        resp = self.client.get("/ampcode/model-mappings")
+        resp.raise_for_status()
+        return resp.json().get("value", [])
+
+    def set_amp_model_mappings(self, mappings: list) -> dict:
+        resp = self.client.put("/ampcode/model-mappings", json_data={"value": mappings})
+        resp.raise_for_status()
+        return resp.json()
+
+    def patch_amp_model_mappings(self, mapping: dict) -> dict:
+        resp = self.client.patch("/ampcode/model-mappings", json_data=mapping)
+        resp.raise_for_status()
+        return resp.json()
+
+    def delete_amp_model_mappings(self, from_model: str = "") -> dict:
+        data: dict[str, Any] = {}
+        if from_model:
+            data["from"] = from_model
+        resp = self.client.delete("/ampcode/model-mappings", json_data=data)
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_amp_force_model_mappings(self) -> bool:
+        resp = self.client.get("/ampcode/force-model-mappings")
+        resp.raise_for_status()
+        return resp.json().get("value", False)
+
+    def set_amp_force_model_mappings(self, enabled: bool) -> dict:
+        resp = self.client.put("/ampcode/force-model-mappings", json_data={"value": enabled})
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_amp_restrict_management_to_localhost(self) -> bool:
+        resp = self.client.get("/ampcode/restrict-management-to-localhost")
+        resp.raise_for_status()
+        return resp.json().get("value", False)
+
+    def set_amp_restrict_management_to_localhost(self, enabled: bool) -> dict:
+        resp = self.client.put("/ampcode/restrict-management-to-localhost", json_data={"value": enabled})
+        resp.raise_for_status()
+        return resp.json()
+
+    def get_amp_upstream_api_keys(self) -> list:
+        resp = self.client.get("/ampcode/upstream-api-keys")
+        resp.raise_for_status()
+        return resp.json().get("value", [])
+
+    def set_amp_upstream_api_keys(self, keys: list) -> dict:
+        resp = self.client.put("/ampcode/upstream-api-keys", json_data={"value": keys})
+        resp.raise_for_status()
+        return resp.json()
+
+    def patch_amp_upstream_api_keys(self, entry: dict) -> dict:
+        resp = self.client.patch("/ampcode/upstream-api-keys", json_data=entry)
+        resp.raise_for_status()
+        return resp.json()
+
+    def delete_amp_upstream_api_keys(self, upstream_api_key: str = "") -> dict:
+        data: dict[str, Any] = {}
+        if upstream_api_key:
+            data["upstream_api_key"] = upstream_api_key
+        resp = self.client.delete("/ampcode/upstream-api-keys", json_data=data)
+        resp.raise_for_status()
+        return resp.json()
+
+    # ---- 通用 API 调用 ----
+
+    def api_call(
+        self,
+        method: str,
+        url: str,
+        headers: Optional[dict] = None,
+        data: Optional[str] = None,
+        auth_index: Optional[str] = None,
+    ) -> dict:
+        body: dict[str, Any] = {"method": method, "url": url}
+        if headers:
+            body["header"] = headers
+        if data:
+            body["data"] = data
+        if auth_index:
+            body["auth_index"] = auth_index
+        resp = self.client.post("/api-call", json_data=body)
+        resp.raise_for_status()
+        return resp.json()

cli_anything/cliproxyapi/core/usage.py

@@ -0,0 +1,25 @@
+"""使用统计模块。"""
+
+from .client import ManagementClient
+
+
+class UsageManager:
+    """管理 CLIProxyAPI 的使用统计。"""
+
+    def __init__(self, client: ManagementClient):
+        self.client = client
+
+    def get_stats(self) -> dict:
+        resp = self.client.get("/usage")
+        resp.raise_for_status()
+        return resp.json()
+
+    def export_stats(self) -> str:
+        resp = self.client.get("/usage/export")
+        resp.raise_for_status()
+        return resp.text
+
+    def import_stats(self, data: str) -> dict:
+        resp = self.client.post("/usage/import", json_data={"data": data})
+        resp.raise_for_status()
+        return resp.json()