Skip to content

Commit c84287c

Browse files
author
wangyudong
committed
fix: update premium model usage field name and test data
Update field name from premium_model_fast_request_usage to premium_model_fast_amount to match API changes. Also update test data to reflect current subscription model and usage patterns.
1 parent cf819c4 commit c84287c

3 files changed

Lines changed: 55 additions & 53 deletions

File tree

‎package.json‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
"name": "trae-usage-monitor",
33
"displayName": "Trae Usage",
44
"description": "Monitor Trae AI usage statistics in real-time",
5-
"version": "1.3.1",
5+
"version": "1.3.2",
66
"publisher": "whyuds",
77
"repository": {
88
"type": "git",

‎src/extension.ts‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ export class TraeUsageProvider {
286286
}
287287

288288
this.usageData.user_entitlement_pack_list.forEach(pack => {
289-
const usage = pack.usage.premium_model_fast_request_usage;
289+
const usage = pack.usage.premium_model_fast_amount;
290290
const limit = pack.entitlement_base_info.quota.premium_model_fast_request_limit;
291291

292292
if (limit > 0) {
@@ -345,7 +345,7 @@ export class TraeUsageProvider {
345345
const subscriptionType = TraeUsageProvider.getSubscriptionTypeLabel(pack);
346346

347347
// Premium Fast Request使用情况(带进度条)
348-
const fastUsed = usage.premium_model_fast_request_usage;
348+
const fastUsed = usage.premium_model_fast_amount;
349349
const fastLimit = quota.premium_model_fast_request_limit;
350350

351351
if (fastLimit > 0) {

‎src/test/data/multi_subscription.json‎

Lines changed: 52 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -1,100 +1,102 @@
11
{
2-
"mark": "PRO订阅接近过期，额外订阅100次请求，当前同时有两份订阅",
32
"is_pay_freshman": false,
4-
"user_entitlement_pack_list":
5-
[
3+
"user_entitlement_pack_list": [
64
{
7-
"entitlement_base_info":
8-
{
9-
"charge_amount": 0,
5+
"entitlement_base_info": {
6+
"charge_amount": 300,
107
"currency": 0,
11-
"end_time": 1758609428,
12-
"entitlement_id": "2977530115",
13-
"product_extra":
14-
{
15-
"subscription_extra":
16-
{
8+
"end_time": 1763038285,
9+
"entitlement_id": "12984810244",
10+
"product_extra": {
11+
"subscription_extra": {
1712
"period_type": 0,
18-
"quota":
19-
{
13+
"quota": {
2014
"advanced_model_request_limit": -1,
2115
"auto_completion_limit": -1,
16+
"enable_solo_builder": true,
17+
"enable_solo_builder_v1": false,
18+
"enable_solo_coder": true,
19+
"enable_super_model": false,
2220
"premium_model_fast_request_limit": 600,
2321
"premium_model_slow_request_limit": -1
2422
}
2523
}
2624
},
2725
"product_type": 1,
28-
"quota":
29-
{
26+
"quota": {
3027
"advanced_model_request_limit": -1,
3128
"auto_completion_limit": -1,
29+
"enable_solo_builder": true,
30+
"enable_solo_builder_v1": false,
31+
"enable_solo_coder": true,
32+
"enable_super_model": false,
3233
"premium_model_fast_request_limit": 600,
3334
"premium_model_slow_request_limit": -1
3435
},
35-
"start_time": 1756017428,
36-
"user_id": "7509828020950320136"
36+
"start_time": 1760446285,
37+
"user_id": "7553652665889539089"
3738
},
38-
"expire_time": 1758609428,
39+
"expire_time": 0,
3940
"is_last_period": false,
40-
"next_billing_time": 1758523028,
41-
"source_id": "2993965315",
42-
"status": 3,
43-
"usage":
44-
{
45-
"advanced_model_amount": 1,
46-
"advanced_model_request_usage": 1,
47-
"auto_completion_amount": 942,
48-
"auto_completion_usage": 942,
41+
"next_billing_time": 1762951885,
42+
"source_id": "12930744580",
43+
"status": 1,
44+
"usage": {
45+
"advanced_model_amount": 0,
46+
"advanced_model_request_usage": 0,
47+
"auto_completion_amount": 168,
48+
"auto_completion_usage": 0,
4949
"is_flash_consuming": true,
50-
"premium_model_fast_amount": 584,
51-
"premium_model_fast_request_usage": 584,
50+
"premium_model_fast_amount": 278.19,
51+
"premium_model_fast_request_usage": 0,
5252
"premium_model_slow_amount": 0,
5353
"premium_model_slow_request_usage": 0
5454
},
5555
"yearly_expire_time": 0
5656
},
5757
{
58-
"entitlement_base_info":
59-
{
58+
"entitlement_base_info": {
6059
"charge_amount": 0,
6160
"currency": 0,
62-
"end_time": 1760586110,
63-
"entitlement_id": "3498177027",
64-
"product_extra":
65-
{
66-
"package_extra":
67-
{
61+
"end_time": 1764814939,
62+
"entitlement_id": "15049367300",
63+
"product_extra": {
64+
"package_extra": {
6865
"duration": 1,
6966
"package_duration_type": 1,
70-
"package_source_type": 0,
71-
"quota":
72-
{
67+
"package_source_type": 2,
68+
"quota": {
7369
"advanced_model_request_limit": 0,
7470
"auto_completion_limit": 0,
75-
"premium_model_fast_request_limit": 100,
71+
"enable_solo_builder": false,
72+
"enable_solo_builder_v1": false,
73+
"enable_solo_coder": false,
74+
"enable_super_model": false,
75+
"premium_model_fast_request_limit": 300,
7676
"premium_model_slow_request_limit": 0
7777
}
7878
}
7979
},
8080
"product_type": 2,
81-
"quota":
82-
{
81+
"quota": {
8382
"advanced_model_request_limit": 0,
8483
"auto_completion_limit": 0,
85-
"premium_model_fast_request_limit": 100,
84+
"enable_solo_builder": false,
85+
"enable_solo_builder_v1": false,
86+
"enable_solo_coder": false,
87+
"enable_super_model": false,
88+
"premium_model_fast_request_limit": 300,
8689
"premium_model_slow_request_limit": 0
8790
},
88-
"start_time": 1757994110,
89-
"user_id": "7509828020950320136"
91+
"start_time": 1762222939,
92+
"user_id": "7553652665889539089"
9093
},
9194
"expire_time": 0,
9295
"is_last_period": false,
9396
"next_billing_time": 0,
94-
"source_id": "3496332547",
97+
"source_id": "12984810244",
9598
"status": 0,
96-
"usage":
97-
{
99+
"usage": {
98100
"advanced_model_amount": 0,
99101
"advanced_model_request_usage": 0,
100102
"auto_completion_amount": 0,

0 commit comments

Comments
 (0)