cometapi-cli 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

cometapi_cli/__init__.py

@@ -1,3 +1,3 @@
 """CometAPI CLI — professional terminal interface for CometAPI."""
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"

cometapi_cli/commands/balance.py

@@ -7,28 +7,50 @@ from typing import Annotated
 import typer
 
 from ..config import get_client
+from ..console import err_console
 from ..errors import handle_errors
 from ..formatters import OutputFormat, output, resolve_format
 
+BALANCE_SOURCE_ALIASES = {
+    "account": "account",
+    "key": "key",
+    "token": "key",
+}
+
+
+def _normalize_balance_source(source: str | None) -> str | None:
+    if source is None:
+        return None
+
+    normalized = BALANCE_SOURCE_ALIASES.get(source.lower())
+    if normalized is not None:
+        return normalized
+
+    err_console.print(f"[red]Invalid source:[/] {source}. Valid sources: account, key")
+    raise typer.Exit(code=2)
+
 
 @handle_errors
 def balance(
     ctx: typer.Context,
     source: Annotated[
         str | None,
-        typer.Option("--source", "-s", help="Data source: 'account' (full account) or 'token' (current API key)."),
+        typer.Option("--source", "-s", help="Data source: 'account' (full account) or 'key' (current API key)."),
+    ] = None,
+    output_format: Annotated[
+        OutputFormat | None,
+        typer.Option("--format", "-f", help="Output format (table, json, yaml, csv, markdown)."),
     ] = None,
-    output_format: Annotated[OutputFormat | None, typer.Option("--format", "-f", help="Output format (table, json, yaml, csv, markdown).")] = None,
     json_output: Annotated[bool, typer.Option("--json", help="Output as JSON.")] = False,
 ) -> None:
     """Show your CometAPI account balance.
 
     By default shows account-level balance (requires access token), falling back
-    to per-token billing stats. Use --source to force a specific view.
+    to per-key billing stats. Use --source to force a specific view.
     """
     fmt = resolve_format(ctx, json_output, output_format)
     client = get_client()
-    raw = client.get_balance(source=source)
+    raw = client.get_balance(source=_normalize_balance_source(source))
 
     # For JSON output, pass raw data for maximum information
     if fmt == OutputFormat.JSON:

{cometapi_cli-0.2.1.dist-info → cometapi_cli-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cometapi-cli
-Version: 0.2.1
+Version: 0.2.2
 Summary: CometAPI CLI — official command-line interface for the CometAPI AI gateway
 Project-URL: Homepage, https://github.com/cometapi-dev/cometapi-cli
 Project-URL: Repository, https://github.com/cometapi-dev/cometapi-cli

{cometapi_cli-0.2.1.dist-info → cometapi_cli-0.2.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-cometapi_cli/__init__.py,sha256=PO4EJzXB-LMrMP41BGu4Cp0YrCXOG2oWfHl8uVdCw6Q,92
+cometapi_cli/__init__.py,sha256=YoHkL8RaBOgm8zi4mJO3F2ywNNSkk26kuGB8cjLLRrE,92
 cometapi_cli/app.py,sha256=tFrkyfVVBYkjOA9zXP72bZ1csHI9HBnGF-Z4_4EkXkE,2784
 cometapi_cli/client.py,sha256=4JkMSrjWwAQV8tmlwplZQ75gzt3zHno8ASKLgMCpYkU,9658
 cometapi_cli/config.py,sha256=oJXQidKCOsKNYPnE8OfLLoOfsv0MSZEDICB6VShJRSA,3307
@@ -9,7 +9,7 @@ cometapi_cli/formatters.py,sha256=QrXpsQHRYiJ49Rwz_7dOMoKxCcnRPumZMrxH21ZQaMw,45
 cometapi_cli/main.py,sha256=4NmO8MAAc_H30hahT6VNrdGdlaFlQ99AixKcRV5iDXE,168
 cometapi_cli/commands/__init__.py,sha256=mihzLmD4ADkH9WnCBP2p1fn95lU3GyzTliAJ1XL5Ngg,29
 cometapi_cli/commands/account.py,sha256=yGIY5iQeB4j1bISMUOqnvjcuD9HTd7VcWSvSFB6fAfo,1257
-cometapi_cli/commands/balance.py,sha256=GUOdbfDV0u9h23XG3pqRf_tl3mUxWGl781b9eep4Rxc,1833
+cometapi_cli/commands/balance.py,sha256=BT96WTt2I7dCLBlAkZMX4KYliqC5C7QCXKNWCbXwldw,2353
 cometapi_cli/commands/chat.py,sha256=5VuNOF7O3vjD0f8eD0faZ3xL2IFnSa2BNadxencoPJk,3937
 cometapi_cli/commands/chat_repl.py,sha256=b9lkYnbbaOb0AlSpRcq3wWHmnhtTXvaqeJiUImsMEBY,8176
 cometapi_cli/commands/config_cmd.py,sha256=JU4ZS8ur2Sq1MjMZKaGLaI1TdM3CAUDE45qsKDOxD_8,8528
@@ -20,8 +20,8 @@ cometapi_cli/commands/repl.py,sha256=b5z1jmEXOsCrb6fwEUyv-IKot929NIPQQAbA1uas1D4
 cometapi_cli/commands/stats.py,sha256=6ULBb7rX5Q0yWuIxtYLataNPoJ_SkU3Y4RovgdVyROs,1388
 cometapi_cli/commands/tasks.py,sha256=TW_Jgli7iyfIoHNyufbViAUMIieMNfsD8DMHt5mQA8Q,4625
 cometapi_cli/commands/tokens.py,sha256=9KHer9MQOZvm_Ybxnhjb5_TtGgKyv3MtI5lqsHy-Qxg,3019
-cometapi_cli-0.2.1.dist-info/METADATA,sha256=S_Gkmso14ZIEFTdRT4Hb57UVZxTh69fintCJ0ZQDKwE,8624
-cometapi_cli-0.2.1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-cometapi_cli-0.2.1.dist-info/entry_points.txt,sha256=xoiE2ZVNNWXTq0JRBtzC8hJ2JkdKuaBNz_fEd8OJpLs,50
-cometapi_cli-0.2.1.dist-info/licenses/LICENSE,sha256=-rBwHQzkmLbty07abmGvQvsRrvDeEQUkPDhNJfTcjdE,1065
-cometapi_cli-0.2.1.dist-info/RECORD,,
+cometapi_cli-0.2.2.dist-info/METADATA,sha256=Aveuc9r_GBg70ue7jZL4j1P6hkyFKTKruBQXa3kCk48,8624
+cometapi_cli-0.2.2.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+cometapi_cli-0.2.2.dist-info/entry_points.txt,sha256=xoiE2ZVNNWXTq0JRBtzC8hJ2JkdKuaBNz_fEd8OJpLs,50
+cometapi_cli-0.2.2.dist-info/licenses/LICENSE,sha256=-rBwHQzkmLbty07abmGvQvsRrvDeEQUkPDhNJfTcjdE,1065
+cometapi_cli-0.2.2.dist-info/RECORD,,