Add balance information when boot the kimi CLI

JacksonTian · Apr 24, 2024 · 3e76859 · 3e76859
1 parent 41f0c74
commit 3e76859
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -120,6 +120,7 @@ const response = await client.models();
 - `getFileContent()`
 - `deleteFile()`
 - `estimateTokenCount()`
+- `getBalance()`
 
 The detail of parameters can be found at <https://platform.moonshot.cn/docs/api-reference> or [`test/kimi.test.js`](./test/kimi.test.js).
 

diff --git a/bin/kimi.js b/bin/kimi.js
@@ -118,6 +118,10 @@ const pkg = loadJSONSync(path.join(__dirname, '../package.json'));
 console.log(`Welcome to KIMI CLI(v${pkg.version}), type ${chalk.bgGray('.help')} for more information.`);
 console.log(`Current model is ${chalk.bgGreen(config.model)}.`);
 
+const balance = await kimi.getBalance();
+const {available_balance, cash_balance, voucher_balance } = balance.data;
+console.log(`Current balance: ￥${available_balance}(Cash: ￥${cash_balance}, Voucher: ￥${voucher_balance}}).`);
+
 // eslint-disable-next-line no-constant-condition
 while (true) {
   const answer = await rl.question('What is your query: ');