apishare-client 1.2.0__tar.gz

apishare_client-1.2.0/PKG-INFO

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: apishare-client
+Version: 1.2.0
+Summary: APIShare Mining Client - Contribute GPU/CPU compute power, run AI models, earn credits via tunnel
+Author-email: APIShare Team <apishare@proton.me>
+License: MIT
+Project-URL: Homepage, https://apishare.samai.cc
+Project-URL: Repository, https://github.com/ctz168/apishare
+Project-URL: Documentation, https://apishare.samai.cc/mining
+Project-URL: Bug Tracker, https://github.com/ctz168/apishare/issues
+Project-URL: PyPI, https://pypi.org/project/apishare-client/
+Keywords: apishare,mining,ai,gpu,compute,tunnel,websockets,decentralized,blockchain
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: System :: Distributed Computing
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: websockets>=12.0
+Requires-Dist: psutil>=5.9.0
+Requires-Dist: requests>=2.31.0

apishare_client-1.2.0/apishare_client/__init__.py

@@ -0,0 +1,17 @@
+"""APIShare Mining Client - 挖矿客户端
+
+矿工通过此客户端连接 APIShare 平台，贡献GPU/CPU算力赚取AIC积分。
+
+用法:
+    apishare login --email you@email.com
+    apishare register --name my-node --type gpu --model qwen2.5-0.5b
+    apishare serve --model qwen2.5-0.5b --port 8000
+    apishare tunnel --node NODE_ID --port 8000
+    apishare report --node NODE_ID
+    apishare status
+    apishare balance
+    apishare models
+"""
+
+__version__ = "1.2.0"
+__author__ = "APIShare Team"

apishare_client-1.2.0/apishare_client/cli.py

@@ -0,0 +1,234 @@
+#!/usr/bin/env python3
+"""
+APIShare Mining Client - 挖矿客户端 CLI
+======================================
+
+矿工通过此客户端连接 APIShare 平台，贡献GPU/CPU算力赚取AIC积分。
+
+用法:
+    apishare login --email you@email.com         # 登录平台
+    apishare register --name my-node --type gpu   # 注册计算节点
+    apishare serve --model qwen2.5-0.5b --port 8000  # 启动本地推理服务
+    apishare tunnel --node NODE_ID --port 8000   # 建立WebSocket隧道
+    apishare report --node NODE_ID               # 上报算力数据
+    apishare status                              # 查看挖矿状态
+    apishare balance                             # 查看积分余额
+    apishare models                              # 查看可用模型
+    apishare logout                              # 登出
+
+完整挖矿流程:
+    1. 登录:     apishare login --email you@email.com
+    2. 注册节点: apishare register --name my-gpu --type gpu --model qwen2.5-0.5b
+    3. 启动模型: apishare serve --model qwen2.5-0.5b --port 8000
+    4. 建隧道:   apishare tunnel --node NODE_ID --model qwen2.5-0.5b --port 8000
+    5. 查状态:   apishare status
+
+Docker一键模式:
+    docker run -d --gpus all ctz168/apishare-client:latest \\
+      tunnel --node NODE_ID --model qwen2.5-0.5b --port 8000 --token YOUR_TOKEN
+"""
+
+import argparse
+import sys
+import json
+import urllib.request
+import urllib.error
+
+from apishare_client import __version__
+from apishare_client.config import (
+    get_token, get_platform_url, get_email, get_node_id,
+    logout as config_logout, get_supported_models, DEFAULT_PLATFORM_URL,
+)
+from apishare_client.commands.login import login as cmd_login
+from apishare_client.commands.register import register_node as cmd_register
+from apishare_client.commands.serve import start_serve as cmd_serve
+from apishare_client.commands.tunnel import start_tunnel as cmd_tunnel
+from apishare_client.commands.report import report_mining as cmd_report
+from apishare_client.commands.status import show_status as cmd_status
+
+
+def cmd_balance(args):
+    """查看积分余额"""
+    token = get_token()
+    if not token:
+        print("❌ 请先登录: apishare login --email you@email.com")
+        return
+
+    base_url = get_platform_url()
+    try:
+        url = f"{base_url}/api/auth/me"
+        req = urllib.request.Request(url, headers={"Authorization": f"Bearer {token}"})
+        with urllib.request.urlopen(req, timeout=10) as resp:
+            data = json.loads(resp.read().decode("utf-8"))
+        print(f"💰 APIShare 积分余额: {data.get('balance', 0):.2f}")
+        print(f"   邮箱: {data.get('email', '-')}")
+        print(f"   角色: {data.get('role', '-')}")
+    except urllib.error.HTTPError as e:
+        if e.code == 401:
+            print("❌ Token已过期，请重新登录")
+        else:
+            print(f"❌ 查询失败: HTTP {e.code}")
+    except Exception as e:
+        print(f"❌ 查询错误: {e}")
+
+
+def cmd_models(args):
+    """查看可用模型"""
+    models = get_supported_models()
+    print()
+    print("=" * 55)
+    print("  APIShare 支持的模型")
+    print("=" * 55)
+    print()
+    print("  🖥️  GPU 模型:")
+    for m in models["gpu"]:
+        print(f"     - {m}")
+    print()
+    print("  💻 CPU 模型:")
+    for m in models["cpu"]:
+        print(f"     - {m}")
+    print()
+    print("  💡 也可以使用 --model 指定任何模型名称")
+    print()
+
+
+def cmd_logout(args):
+    """登出"""
+    config_logout()
+    print("✅ 已登出")
+
+
+def main():
+    """CLI入口"""
+    parser = argparse.ArgumentParser(
+        prog="apishare",
+        description="APIShare 挖矿客户端 - 贡献GPU/CPU算力赚取AIC积分",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+完整挖矿流程:
+  1. 登录平台:     apishare login --email you@email.com
+  2. 注册计算节点: apishare register --name my-gpu --type gpu --model qwen2.5-0.5b
+  3. 启动本地模型: apishare serve --model qwen2.5-0.5b --port 8000
+  4. 建立隧道连接: apishare tunnel --node NODE_ID --model qwen2.5-0.5b --port 8000
+  5. 查看挖矿状态: apishare status
+
+更多帮助:
+  https://apishare.samai.cc/mining
+        """,
+    )
+
+    parser.add_argument("--version", action="version", version=f"apishare {__version__}")
+    parser.add_argument("--platform", type=str, help="平台URL (默认: https://apishare.samai.cc)")
+    parser.add_argument("--ws-url", type=str, help="WebSocket隧道URL")
+
+    subparsers = parser.add_subparsers(dest="command", help="可用命令")
+
+    # login
+    p_login = subparsers.add_parser("login", help="登录 APIShare 平台")
+    p_login.add_argument("--email", required=True, help="登录邮箱")
+    p_login.add_argument("--password", type=str, help="密码 (不填则交互输入)")
+
+    # register
+    p_reg = subparsers.add_parser("register", help="注册计算节点")
+    p_reg.add_argument("--name", required=True, help="节点名称")
+    p_reg.add_argument("--type", default="gpu", choices=["gpu", "cpu"], help="计算类型")
+    p_reg.add_argument("--model", default="qwen2.5-0.5b", help="服务模型名称")
+    p_reg.add_argument("--gpu-info", default="", help="GPU信息")
+    p_reg.add_argument("--memory", type=int, default=0, help="内存(GB)")
+
+    # serve
+    p_serve = subparsers.add_parser("serve", help="启动本地推理服务")
+    p_serve.add_argument("--model", required=True, help="模型名称")
+    p_serve.add_argument("--port", type=int, default=8000, help="模型服务端口")
+    p_serve.add_argument("--backend", default="auto", choices=["auto", "vllm", "ollama", "openai-compatible"], help="推理后端")
+    p_serve.add_argument("--proxy-port", type=int, default=8001, help="代理服务端口")
+    p_serve.add_argument("--host", default="0.0.0.0", help="监听地址")
+
+    # tunnel
+    p_tunnel = subparsers.add_parser("tunnel", help="建立WebSocket隧道")
+    p_tunnel.add_argument("--node", required=True, help="节点ID")
+    p_tunnel.add_argument("--model", default="qwen2.5-0.5b", help="模型名称")
+    p_tunnel.add_argument("--port", type=int, default=8000, help="本地模型端口")
+    p_tunnel.add_argument("--token", type=str, help="JWT token (覆盖配置文件)")
+
+    # report
+    p_report = subparsers.add_parser("report", help="上报算力数据")
+    p_report.add_argument("--node", required=True, help="节点ID")
+    p_report.add_argument("--model", default="qwen2.5-0.5b", help="模型名称")
+    p_report.add_argument("--tokens-in", type=int, default=0, help="输入token数")
+    p_report.add_argument("--tokens-out", type=int, default=0, help="输出token数")
+    p_report.add_argument("--gpu-usage", type=float, default=0, help="GPU使用率")
+    p_report.add_argument("--auto", action="store_true", help="自动采集系统信息")
+
+    # status
+    p_status = subparsers.add_parser("status", help="查看挖矿状态")
+
+    # balance
+    p_balance = subparsers.add_parser("balance", help="查看积分余额")
+
+    # models
+    p_models = subparsers.add_parser("models", help="查看可用模型")
+
+    # logout
+    p_logout = subparsers.add_parser("logout", help="登出")
+
+    args = parser.parse_args()
+
+    if not args.command:
+        parser.print_help()
+        return
+
+    # Override platform URL if specified
+    if args.platform:
+        from apishare_client.config import save_config, load_config
+        config = load_config()
+        config["platform_url"] = args.platform
+        save_config(config)
+
+    # Dispatch commands
+    if args.command == "login":
+        cmd_login(email=args.email, password=args.password)
+    elif args.command == "register":
+        cmd_register(
+            name=args.name,
+            compute_type=args.type,
+            model=args.model,
+            gpu_info=args.gpu_info,
+            memory_gb=args.memory,
+        )
+    elif args.command == "serve":
+        cmd_serve(
+            model=args.model,
+            port=args.port,
+            backend=args.backend,
+            host=args.host,
+            proxy_port=args.proxy_port,
+        )
+    elif args.command == "tunnel":
+        cmd_tunnel(
+            node_id=args.node,
+            model=args.model,
+            port=args.port,
+            token=args.token,
+        )
+    elif args.command == "report":
+        cmd_report(
+            node_id=args.node,
+            model=args.model,
+            tokens_in=args.tokens_in,
+            tokens_out=args.tokens_out,
+            gpu_usage=args.gpu_usage,
+            auto=args.auto,
+        )
+    elif args.command == "status":
+        cmd_status()
+    elif args.command == "balance":
+        cmd_balance(args)
+    elif args.command == "models":
+        cmd_models(args)
+    elif args.command == "logout":
+        cmd_logout(args)
+
+
+if __name__ == "__main__":
+    main()

apishare_client-1.2.0/apishare_client/commands/__init__.py

@@ -0,0 +1 @@
+"""APIShare Mining Client - 命令模块"""

apishare_client-1.2.0/apishare_client/commands/login.py

@@ -0,0 +1,77 @@
+"""APIShare Mining Client - 登录命令
+
+apishare login --email you@email.com
+apishare login --email you@email.com --platform https://apishare.samai.cc
+"""
+
+import getpass
+import sys
+import urllib.request
+import urllib.error
+import json
+
+from apishare_client.config import set_login, get_platform_url, DEFAULT_PLATFORM_URL
+
+
+def login(email: str, platform_url: str = None, password: str = None) -> bool:
+    """登录 APIShare 平台，获取 JWT token"""
+    base_url = platform_url or get_platform_url()
+
+    if not password:
+        password = getpass.getpass("请输入密码: ")
+
+    if not email or not password:
+        print("❌ 邮箱和密码不能为空")
+        return False
+
+    print(f"正在登录 {base_url} ...")
+
+    # 调用平台登录API
+    url = f"{base_url}/api/auth/login"
+    payload = json.dumps({"email": email, "password": password}).encode("utf-8")
+    req = urllib.request.Request(
+        url,
+        data=payload,
+        headers={"Content-Type": "application/json"},
+        method="POST",
+    )
+
+    try:
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            data = json.loads(resp.read().decode("utf-8"))
+
+        token = data.get("token") or data.get("access_token")
+        user = data.get("user", {})
+
+        if token:
+            ws_url = base_url.replace("https://", "wss://").replace("http://", "ws://") + "/api/tunnel/connect"
+            set_login(token, email, base_url, ws_url)
+            balance = user.get("balance", 0)
+            role = user.get("role", "user")
+            print(f"✅ 登录成功！")
+            print(f"   邮箱: {email}")
+            print(f"   角色: {role}")
+            print(f"   积分余额: {balance:.2f} AIC")
+            print(f"   平台: {base_url}")
+            return True
+        else:
+            print(f"❌ 登录失败: 服务器未返回有效token")
+            print(f"   响应: {json.dumps(data, ensure_ascii=False)[:200]}")
+            return False
+
+    except urllib.error.HTTPError as e:
+        body = e.read().decode("utf-8", errors="replace")[:200]
+        try:
+            err_data = json.loads(body)
+            msg = err_data.get("detail", body)
+        except json.JSONDecodeError:
+            msg = body
+        print(f"❌ 登录失败: {msg}")
+        return False
+    except urllib.error.URLError as e:
+        print(f"❌ 网络错误: {e.reason}")
+        print(f"   请检查网络连接或平台地址是否正确")
+        return False
+    except Exception as e:
+        print(f"❌ 登录异常: {e}")
+        return False

apishare_client-1.2.0/apishare_client/commands/register.py

@@ -0,0 +1,93 @@
+"""APIShare Mining Client - 注册计算节点
+
+apishare register --name my-gpu-node --type gpu --model qwen2.5-0.5b
+"""
+
+import json
+import sys
+import urllib.request
+import urllib.error
+
+from apishare_client.config import get_token, get_platform_url, get_email, set_node_id
+
+
+def register_node(
+    name: str,
+    compute_type: str = "gpu",
+    model: str = "qwen2.5-0.5b",
+    gpu_info: str = "",
+    cpu_info: str = "",
+    memory_gb: int = 0,
+) -> bool:
+    """在平台上注册一个计算节点"""
+
+    token = get_token()
+    if not token:
+        print("❌ 未登录，请先运行: apishare login --email you@email.com")
+        return False
+
+    email = get_email() or "unknown"
+    base_url = get_platform_url()
+
+    print(f"正在注册计算节点...")
+    print(f"  名称: {name}")
+    print(f"  类型: {compute_type.upper()}")
+    print(f"  模型: {model}")
+
+    # 调用平台注册API
+    url = f"{base_url}/api/compute/register"
+    payload = json.dumps({
+        "nodeName": name,
+        "computeType": compute_type,
+        "servingModel": model,
+        "gpuInfo": gpu_info or f"GPU for {model}",
+        "cpuInfo": cpu_info or ("CPU for " + model if compute_type == "cpu" else ""),
+        "memoryGB": memory_gb or 8,
+    }).encode("utf-8")
+
+    req = urllib.request.Request(
+        url,
+        data=payload,
+        headers={
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {token}",
+        },
+        method="POST",
+    )
+
+    try:
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            data = json.loads(resp.read().decode("utf-8"))
+
+        node_id = data.get("id") or data.get("nodeId")
+        if node_id:
+            set_node_id(node_id)
+            print(f"\n✅ 节点注册成功！")
+            print(f"   节点ID: {node_id}")
+            print(f"   节点名称: {name}")
+            print(f"\n下一步:")
+            print(f"  1. 启动本地模型服务:")
+            print(f"     apishare serve --model {model} --port 8000")
+            print(f"  2. 建立隧道连接（让平台能调用你的模型）:")
+            print(f"     apishare tunnel --node {node_id} --model {model} --port 8000")
+            return True
+        else:
+            print(f"❌ 注册失败: 服务器未返回节点ID")
+            print(f"   响应: {json.dumps(data, ensure_ascii=False)[:200]}")
+            return False
+
+    except urllib.error.HTTPError as e:
+        body = e.read().decode("utf-8", errors="replace")[:200]
+        try:
+            err_data = json.loads(body)
+            msg = err_data.get("detail", body)
+        except json.JSONDecodeError:
+            msg = body
+        print(f"❌ 注册失败: {msg}")
+        return False
+    except urllib.error.URLError as e:
+        print(f"❌ 网络错误: {e.reason}")
+        return False
+    except Exception as e:
+        print(f"❌ 注册异常: {e}")
+        return False

apishare_client-1.2.0/apishare_client/commands/report.py

@@ -0,0 +1,119 @@
+"""APIShare Mining Client - 算力上报
+
+apishare report --node NODE_ID
+apishare report --node NODE_ID --auto  (自动采集系统信息)
+"""
+
+import json
+import time
+import urllib.request
+import urllib.error
+from typing import Optional
+
+from apishare_client.config import get_token, get_platform_url
+
+
+def report_mining(
+    node_id: str,
+    tokens_in: int = 0,
+    tokens_out: int = 0,
+    inference_time: float = 0.0,
+    gpu_usage: float = 0.0,
+    model: str = "qwen2.5-0.5b",
+    compute_type: str = "gpu",
+    auto: bool = False,
+) -> bool:
+    """上报挖矿算力数据到平台"""
+
+    token = get_token()
+    if not token:
+        print("❌ 未登录，请先运行: apishare login")
+        return False
+
+    base_url = get_platform_url()
+
+    if auto:
+        sys_info = _collect_system_info()
+        gpu_usage = gpu_usage or sys_info.get("gpu_usage", 0)
+        compute_type = compute_type or sys_info.get("compute_type", "gpu")
+
+    # 构建上报数据
+    report_data = {
+        "nodeId": node_id,
+        "tokensIn": tokens_in,
+        "tokensOut": tokens_out,
+        "inferenceTime": inference_time,
+        "gpuUsage": gpu_usage,
+        "modelName": model,
+        "computeType": compute_type,
+        "timestamp": time.time(),
+    }
+
+    url = f"{base_url}/api/compute/report"
+    payload = json.dumps(report_data).encode("utf-8")
+    req = urllib.request.Request(
+        url,
+        data=payload,
+        headers={
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {token}",
+        },
+        method="POST",
+    )
+
+    try:
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            data = json.loads(resp.read().decode("utf-8"))
+
+        reward = data.get("reward") or data.get("miningReward", 0)
+        score = data.get("computeScore") or data.get("score", 0)
+
+        print(f"✅ 算力上报成功")
+        print(f"   节点: {node_id[:16]}...")
+        if score:
+            print(f"   算力分: {score:.2f}")
+        if reward:
+            print(f"   获得奖励: {reward}")
+        return True
+
+    except urllib.error.HTTPError as e:
+        body = e.read().decode("utf-8", errors="replace")[:200]
+        print(f"❌ 上报失败 (HTTP {e.code}): {body}")
+        return False
+    except urllib.error.URLError as e:
+        print(f"❌ 网络错误: {e.reason}")
+        return False
+    except Exception as e:
+        print(f"❌ 上报异常: {e}")
+        return False
+
+
+def _collect_system_info() -> dict:
+    """采集系统信息"""
+    info = {"gpu_usage": 0, "compute_type": "cpu", "cpu_count": 0, "memory_gb": 0}
+
+    try:
+        import psutil
+        info["cpu_count"] = psutil.cpu_count()
+        info["memory_gb"] = round(psutil.virtual_memory().total / (1024**3), 1)
+    except ImportError:
+        pass
+
+    # Try to detect GPU
+    try:
+        import subprocess
+        result = subprocess.run(
+            ["nvidia-smi", "--query-gpu=utilization.gpu,memory.used,memory.total",
+             "--format=csv,noheader,nounits"],
+            capture_output=True, text=True, timeout=5
+        )
+        if result.returncode == 0:
+            info["compute_type"] = "gpu"
+            lines = result.stdout.strip().split("\n")
+            if lines:
+                parts = lines[0].split(",")
+                info["gpu_usage"] = float(parts[0].strip())
+    except Exception:
+        pass
+
+    return info