GetAIModelRequestCount rpc added to Wallet service

ECTLab · Mar 7, 2024 · b879dae · b879dae
1 parent 9e56050
commit b879dae
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/protos/platform_management/wallet_server.proto b/protos/platform_management/wallet_server.proto
@@ -15,6 +15,7 @@ service Wallet {
     rpc SetAIModelCostPerReq(SetAIModelCostPerReqRequest) returns (SetAIModelCostPerReqResponse);
     rpc GetAIModelsCostPerReq(GetAIModelsCostPerReqRequest) returns (GetAIModelsCostPerReqResponse);
     rpc BuyAIModelReqPackage(BuyAIModelReqPackageRequest) returns (BuyAIModelReqPackageResponse);
+    rpc GetAIModelRequestCount(GetAIModelRequestCountRequest) returns (GetAIModelRequestCountResponse);
 }
 
 
@@ -120,3 +121,13 @@ message BuyAIModelReqPackageRequest {
 
 message BuyAIModelReqPackageResponse { // empty message
 }
+
+
+message GetAIModelRequestCountRequest {
+    string username = 1;
+    string ai_model_name = 2;
+}
+
+message GetAIModelRequestCountResponse {
+    uint64 req_count = 1;
+}