From 9280c2f27a03a9f2349ee847e3b229e1e635469d Mon Sep 17 00:00:00 2001
From: tzwang <wtz1003@sina.com>
Date: Thu, 25 Jul 2024 10:07:43 +0800
Subject: [PATCH] added deployinstance stat proto

Former-commit-id: df080e5be01aea84f11b4b9b28e2e3937f862806
---
 desc/ai/pcm-ai.api           |  5 +++++
 desc/inference/inference.api | 15 +++++++++++++++
 desc/pcm.api                 | 10 ++++++++++
 3 files changed, 30 insertions(+)

diff --git a/desc/ai/pcm-ai.api b/desc/ai/pcm-ai.api
index 8d2ccfe3..3981cf4d 100644
--- a/desc/ai/pcm-ai.api
+++ b/desc/ai/pcm-ai.api
@@ -1738,6 +1738,11 @@ PayloadCreateTrainJob{
 		Card string `json:"card,optional"`
 		TimeElapsed int32 `json:"elapsed,optional"`
 	}
+
+	TrainingTaskStatResp {
+		Running string `json:"running"`
+		Total string `json:"total"`
+	}
 )
 
 /******************create TrainIngJob end*************************/
diff --git a/desc/inference/inference.api b/desc/inference/inference.api
index b0cca535..36f3f167 100644
--- a/desc/inference/inference.api
+++ b/desc/inference/inference.api
@@ -103,4 +103,19 @@ type (
      StopDeployInstanceResp {
      }
 
+     DeployInstanceStatReq {
+
+     }
+
+     DeployInstanceStatResp {
+          Running string `json:"running"`
+          Total string `json:"total"`
+     }
+
+     InferenceTaskStatReq {}
+
+     InferenceTaskStatResp {
+          Running string `json:"running"`
+          Total string `json:"total"`
+     }
 )
diff --git a/desc/pcm.api b/desc/pcm.api
index 78b35b62..db00def6 100644
--- a/desc/pcm.api
+++ b/desc/pcm.api
@@ -244,6 +244,10 @@ service pcm {
 	group: ai
 )
 service pcm {
+	@doc "训练任务统计"
+	@handler trainingTaskStatHandler
+	get /ai/trainingTaskStat returns (TrainingTaskStatResp)
+
 	@doc "智算中心概览"
 	@handler getCenterOverviewHandler
 	get /ai/getCenterOverview returns (CenterOverviewResp)
@@ -949,6 +953,12 @@ service pcm {
 
 	@handler StopDeployInstanceHandler
 	post /inference/stopDeployInstance (StopDeployInstanceReq) returns (StopDeployInstanceResp)
+
+	@handler DeployInstanceStatHandler
+	get /inference/deployInstanceStat (DeployInstanceStatReq) returns (DeployInstanceStatResp)
+
+	@handler InferenceTaskStatHandler
+	get /inference/taskStat (InferenceTaskStatReq) returns (InferenceTaskStatResp)
 }
 
 @server(