check-balance 0.1.0__tar.gz

check_balance-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,57 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+    tags: ['v*']
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Run tests
+        run: pytest -v
+
+  publish:
+    needs: test
+    runs-on: ubuntu-latest
+    if: startsWith(github.ref, 'refs/tags/v')
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install build tools
+        run: pip install build
+
+      - name: Build package
+        run: python -m build
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          password: ${{ secrets.PYPI_API_TOKEN }}
+        continue-on-error: true

check_balance-0.1.0/.gitignore

@@ -0,0 +1,37 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Distribution / packaging
+build/
+dist/
+*.egg-info/
+*.egg
+
+# Virtual environments
+.venv/
+venv/
+env/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Environment variables
+.env
+.env.*
+
+# Local config
+opencode.json

check_balance-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Han_new_new
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

check_balance-0.1.0/PKG-INFO

@@ -0,0 +1,202 @@
+Metadata-Version: 2.4
+Name: check-balance
+Version: 0.1.0
+Summary: 一键查询当前 AI 服务的账户余额，支持 14 个主流 AI 提供商
+Project-URL: Homepage, https://github.com/hanmumuHL/check_balance
+Project-URL: Repository, https://github.com/hanmumuHL/check_balance
+Project-URL: Issues, https://github.com/hanmumuHL/check_balance/issues
+Author-email: Han_new_new <hanmumuHL@users.noreply.github.com>
+License: MIT
+License-File: LICENSE
+Keywords: ai,api,balance,claude,deepseek,mcp,openai,opencode
+Classifier: Development Status :: 3 - Alpha
+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: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10
+Requires-Dist: mcp>=1.0
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# check-balance
+
+[中文文档](README_zh.md)
+
+One-click balance checker for your AI API services. Supports 14 mainstream AI providers with automatic detection. Works as an MCP server or standalone CLI.
+
+## Features
+
+- **MCP Server** — Integrates with opencode, Claude Code, Cursor, Claude Desktop
+- **CLI Tool** — Run `check-balance` in any terminal
+- **Auto-detection** — Reads opencode config and environment variables
+- **14 Providers** — China (￥) and International ($) regions
+- **Cross-platform** — Linux, macOS, Windows (WSL supported)
+- **Zero config** — Just set your API keys and go
+
+## Supported Providers
+
+### China Region (￥)
+
+| Provider | API Key Env | Balance API | Status |
+|----------|-------------|-------------|--------|
+| DeepSeek | `DEEPSEEK_API_KEY` | `GET /user/balance` | ✅ Direct query |
+| Moonshot / Kimi | `MOONSHOT_API_KEY` | `GET /v1/users/me/balance` | ✅ Direct query |
+| MiniMax | `MINIMAX_API_KEY` | `GET /v1/token_plan/remains` | ✅ Direct query |
+| StepFun | `STEP_API_KEY` | `GET /v1/accounts` | ✅ Direct query |
+| Zhipu / GLM | `ZHIPU_API_KEY` | Quota API | ~ Quota info |
+| DashScope / Qwen | `DASHSCOPE_API_KEY` | — | ⚠️ Console only |
+| Baichuan | `BAICHUAN_API_KEY` | — | ⚠️ Console only |
+
+### International Region ($)
+
+| Provider | API Key Env | Balance API | Status |
+|----------|-------------|-------------|--------|
+| OpenAI | `OPENAI_API_KEY` | Usage/Cost API | ~ Usage + cost |
+| Anthropic / Claude | `ANTHROPIC_API_KEY` | Admin Key Cost API | ~ Usage + cost |
+| xAI / Grok | `XAI_API_KEY` | Balance API | ✅ Direct query |
+| Together AI | `TOGETHER_API_KEY` | Usage API | ~ Usage + cost |
+| Mistral | `MISTRAL_API_KEY` | — | ⚠️ Console only |
+| Groq | `GROQ_API_KEY` | — | ⚠️ Console only |
+| Cohere | `COHERE_API_KEY` | — | ⚠️ Console only |
+
+> **Note:** Cursor subscription balance is not supported as Cursor uses its own proxy and does not expose API keys.
+
+## Installation
+
+```bash
+pip install check-balance
+```
+
+## Usage
+
+### As CLI
+
+```bash
+check-balance
+# or
+python -m check_balance
+```
+
+### As MCP Server
+
+#### opencode
+
+Add to your `opencode.json`:
+
+```json
+{
+  "mcp": {
+    "check-balance": {
+      "type": "local",
+      "command": ["python", "-m", "check_balance.mcp_server"],
+      "enabled": true
+    }
+  }
+}
+```
+
+Then type `/balance` in opencode or ask "check my AI balance".
+
+#### Claude Code
+
+Create `.mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+Or add to `~/.claude/mcp.json` for global access.
+
+#### Cursor
+
+Create `.cursor/mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+Or add to `~/.cursor/mcp.json` for global access.
+
+#### Claude Desktop
+
+Edit `claude_desktop_config.json`:
+
+| OS | Path |
+|----|------|
+| macOS | `~/Library/Application Support/Claude/claude_desktop_config.json` |
+| Windows | `%APPDATA%\Claude\claude_desktop_config.json` |
+| Linux | `~/.config/Claude/claude_desktop_config.json` |
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+## How It Works
+
+1. **Auto-detection** — Reads `opencode.json` to identify your current model/provider, then scans environment variables for all configured API keys
+2. **Balance query** — Calls each provider's balance/usage API
+3. **Formatted output** — Shows results with ￥ (China) or $ (International) currency
+
+## Output Example
+
+```
+AI 服务余额查询
+==================================================
+ * DeepSeek (中国)
+     余额: ￥ 142.50
+
+   Moonshot (中国)
+     余额: ￥ 3.00
+
+   OpenAI (海外)
+     OpenAI 未开放余额查询 API，近 30 天用量: $ 18.32
+     控制台: https://platform.openai.com/usage
+
+   Mistral (海外)
+     未开放余额查询 API
+     控制台: https://console.mistral.ai/usage
+
+* 标记为当前使用的服务
+```
+
+## Development
+
+```bash
+git clone https://github.com/hanmumuHL/check_balance.git
+cd check_balance
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+MIT

check_balance-0.1.0/README.md

@@ -0,0 +1,174 @@
+# check-balance
+
+[中文文档](README_zh.md)
+
+One-click balance checker for your AI API services. Supports 14 mainstream AI providers with automatic detection. Works as an MCP server or standalone CLI.
+
+## Features
+
+- **MCP Server** — Integrates with opencode, Claude Code, Cursor, Claude Desktop
+- **CLI Tool** — Run `check-balance` in any terminal
+- **Auto-detection** — Reads opencode config and environment variables
+- **14 Providers** — China (￥) and International ($) regions
+- **Cross-platform** — Linux, macOS, Windows (WSL supported)
+- **Zero config** — Just set your API keys and go
+
+## Supported Providers
+
+### China Region (￥)
+
+| Provider | API Key Env | Balance API | Status |
+|----------|-------------|-------------|--------|
+| DeepSeek | `DEEPSEEK_API_KEY` | `GET /user/balance` | ✅ Direct query |
+| Moonshot / Kimi | `MOONSHOT_API_KEY` | `GET /v1/users/me/balance` | ✅ Direct query |
+| MiniMax | `MINIMAX_API_KEY` | `GET /v1/token_plan/remains` | ✅ Direct query |
+| StepFun | `STEP_API_KEY` | `GET /v1/accounts` | ✅ Direct query |
+| Zhipu / GLM | `ZHIPU_API_KEY` | Quota API | ~ Quota info |
+| DashScope / Qwen | `DASHSCOPE_API_KEY` | — | ⚠️ Console only |
+| Baichuan | `BAICHUAN_API_KEY` | — | ⚠️ Console only |
+
+### International Region ($)
+
+| Provider | API Key Env | Balance API | Status |
+|----------|-------------|-------------|--------|
+| OpenAI | `OPENAI_API_KEY` | Usage/Cost API | ~ Usage + cost |
+| Anthropic / Claude | `ANTHROPIC_API_KEY` | Admin Key Cost API | ~ Usage + cost |
+| xAI / Grok | `XAI_API_KEY` | Balance API | ✅ Direct query |
+| Together AI | `TOGETHER_API_KEY` | Usage API | ~ Usage + cost |
+| Mistral | `MISTRAL_API_KEY` | — | ⚠️ Console only |
+| Groq | `GROQ_API_KEY` | — | ⚠️ Console only |
+| Cohere | `COHERE_API_KEY` | — | ⚠️ Console only |
+
+> **Note:** Cursor subscription balance is not supported as Cursor uses its own proxy and does not expose API keys.
+
+## Installation
+
+```bash
+pip install check-balance
+```
+
+## Usage
+
+### As CLI
+
+```bash
+check-balance
+# or
+python -m check_balance
+```
+
+### As MCP Server
+
+#### opencode
+
+Add to your `opencode.json`:
+
+```json
+{
+  "mcp": {
+    "check-balance": {
+      "type": "local",
+      "command": ["python", "-m", "check_balance.mcp_server"],
+      "enabled": true
+    }
+  }
+}
+```
+
+Then type `/balance` in opencode or ask "check my AI balance".
+
+#### Claude Code
+
+Create `.mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+Or add to `~/.claude/mcp.json` for global access.
+
+#### Cursor
+
+Create `.cursor/mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+Or add to `~/.cursor/mcp.json` for global access.
+
+#### Claude Desktop
+
+Edit `claude_desktop_config.json`:
+
+| OS | Path |
+|----|------|
+| macOS | `~/Library/Application Support/Claude/claude_desktop_config.json` |
+| Windows | `%APPDATA%\Claude\claude_desktop_config.json` |
+| Linux | `~/.config/Claude/claude_desktop_config.json` |
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+## How It Works
+
+1. **Auto-detection** — Reads `opencode.json` to identify your current model/provider, then scans environment variables for all configured API keys
+2. **Balance query** — Calls each provider's balance/usage API
+3. **Formatted output** — Shows results with ￥ (China) or $ (International) currency
+
+## Output Example
+
+```
+AI 服务余额查询
+==================================================
+ * DeepSeek (中国)
+     余额: ￥ 142.50
+
+   Moonshot (中国)
+     余额: ￥ 3.00
+
+   OpenAI (海外)
+     OpenAI 未开放余额查询 API，近 30 天用量: $ 18.32
+     控制台: https://platform.openai.com/usage
+
+   Mistral (海外)
+     未开放余额查询 API
+     控制台: https://console.mistral.ai/usage
+
+* 标记为当前使用的服务
+```
+
+## Development
+
+```bash
+git clone https://github.com/hanmumuHL/check_balance.git
+cd check_balance
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+MIT

check_balance-0.1.0/README_zh.md

@@ -0,0 +1,174 @@
+# check-balance
+
+[English](README.md)
+
+一键查询当前 AI 服务的账户余额。支持 14 个主流 AI 提供商，自动检测，可作为 MCP Server 或独立 CLI 使用。
+
+## 特性
+
+- **MCP Server** — 集成 opencode、Claude Code、Cursor、Claude Desktop
+- **CLI 工具** — 终端直接运行 `check-balance`
+- **自动检测** — 读取 opencode 配置和环境变量
+- **14 个提供商** — 中国区（￥）和海外区（$）
+- **跨平台** — Linux、macOS、Windows（含 WSL）
+- **零配置** — 设置 API Key 即可使用
+
+## 支持的提供商
+
+### 中国区（￥）
+
+| 提供商 | 环境变量 | 余额 API | 状态 |
+|--------|---------|---------|------|
+| DeepSeek | `DEEPSEEK_API_KEY` | `GET /user/balance` | ✅ 直接查询 |
+| Moonshot / Kimi | `MOONSHOT_API_KEY` | `GET /v1/users/me/balance` | ✅ 直接查询 |
+| MiniMax | `MINIMAX_API_KEY` | `GET /v1/token_plan/remains` | ✅ 直接查询 |
+| StepFun / 阶跃星辰 | `STEP_API_KEY` | `GET /v1/accounts` | ✅ 直接查询 |
+| 智谱 / GLM | `ZHIPU_API_KEY` | 配额 API | ~ 配额信息 |
+| 通义千问 / DashScope | `DASHSCOPE_API_KEY` | — | ⚠️ 仅控制台 |
+| 百川 | `BAICHUAN_API_KEY` | — | ⚠️ 仅控制台 |
+
+### 海外区（$）
+
+| 提供商 | 环境变量 | 余额 API | 状态 |
+|--------|---------|---------|------|
+| OpenAI | `OPENAI_API_KEY` | 用量/费用 API | ~ 用量+费用 |
+| Anthropic / Claude | `ANTHROPIC_API_KEY` | Admin Key 用量 API | ~ 用量+费用 |
+| xAI / Grok | `XAI_API_KEY` | 余额 API | ✅ 直接查询 |
+| Together AI | `TOGETHER_API_KEY` | 用量 API | ~ 用量+费用 |
+| Mistral | `MISTRAL_API_KEY` | — | ⚠️ 仅控制台 |
+| Groq | `GROQ_API_KEY` | — | ⚠️ 仅控制台 |
+| Cohere | `COHERE_API_KEY` | — | ⚠️ 仅控制台 |
+
+> **注意：** Cursor 订阅余额不支持查询，因为 Cursor 使用自己的代理，不暴露 API Key。
+
+## 安装
+
+```bash
+pip install check-balance
+```
+
+## 使用方法
+
+### CLI 模式
+
+```bash
+check-balance
+# 或
+python -m check_balance
+```
+
+### MCP Server 模式
+
+#### opencode
+
+在 `opencode.json` 中添加：
+
+```json
+{
+  "mcp": {
+    "check-balance": {
+      "type": "local",
+      "command": ["python", "-m", "check_balance.mcp_server"],
+      "enabled": true
+    }
+  }
+}
+```
+
+然后在 opencode 中输入 `/balance` 或说 "查一下余额"。
+
+#### Claude Code
+
+在项目根目录创建 `.mcp.json`：
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+或添加到 `~/.claude/mcp.json` 全局使用。
+
+#### Cursor
+
+在项目根目录创建 `.cursor/mcp.json`：
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+或添加到 `~/.cursor/mcp.json` 全局使用。
+
+#### Claude Desktop
+
+编辑 `claude_desktop_config.json`：
+
+| 系统 | 路径 |
+|------|------|
+| macOS | `~/Library/Application Support/Claude/claude_desktop_config.json` |
+| Windows | `%APPDATA%\Claude\claude_desktop_config.json` |
+| Linux | `~/.config/Claude/claude_desktop_config.json` |
+
+```json
+{
+  "mcpServers": {
+    "check-balance": {
+      "command": "python",
+      "args": ["-m", "check_balance.mcp_server"]
+    }
+  }
+}
+```
+
+## 工作原理
+
+1. **自动检测** — 读取 `opencode.json` 识别当前使用的模型/提供商，扫描环境变量获取所有已配置的 API Key
+2. **余额查询** — 调用各提供商的余额/用量 API
+3. **格式化输出** — 中国区显示 ￥，海外区显示 $
+
+## 输出示例
+
+```
+AI 服务余额查询
+==================================================
+ * DeepSeek (中国)
+     余额: ￥ 142.50
+
+   Moonshot (中国)
+     余额: ￥ 3.00
+
+   OpenAI (海外)
+     OpenAI 未开放余额查询 API，近 30 天用量: $ 18.32
+     控制台: https://platform.openai.com/usage
+
+   Mistral (海外)
+     未开放余额查询 API
+     控制台: https://console.mistral.ai/usage
+
+* 标记为当前使用的服务
+```
+
+## 开发
+
+```bash
+git clone https://github.com/hanmumuHL/check_balance.git
+cd check_balance
+pip install -e ".[dev]"
+pytest
+```
+
+## 许可
+
+MIT

check_balance-0.1.0/check_balance/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

check_balance-0.1.0/check_balance/__main__.py

@@ -0,0 +1,4 @@
+from check_balance.main import main
+
+if __name__ == "__main__":
+    main()