@@ -174,6 +174,21 @@ message CMsgSteamLearn_GetProject_Response {
174174 optional .CMsgSteamLearnProject project = 1 ;
175175}
176176
177+ message CMsgSteamLearn_GetSnapshotStatus_Request {
178+ optional uint32 project_id = 1 ;
179+ optional uint32 published_version = 2 ;
180+ optional uint32 snapshot_example_count = 3 ;
181+ }
182+
183+ message CMsgSteamLearn_GetSnapshotStatus_Response {
184+ optional int32 result = 1 [(.description ) = "enum" ];
185+ optional uint32 project_id = 2 ;
186+ optional uint32 published_version = 3 ;
187+ optional uint32 snapshot_count = 4 ;
188+ optional uint32 snapshot_object_count = 5 ;
189+ repeated .CMsgSteamLearnSnapshotStorage snapshot_examples = 6 ;
190+ }
191+
177192message CMsgSteamLearn_GetTrainLogs_Request {
178193 optional uint32 project_id = 1 ;
179194 optional uint32 fetch_id = 2 ;
@@ -231,6 +246,8 @@ message CMsgSteamLearn_GetTrainStatus_Response_Epoch {
231246 repeated .CMsgSteamLearn_GetTrainStatus_Response_Batch validate_batches = 7 ;
232247 optional uint32 start_time = 8 ;
233248 optional uint32 end_time = 9 ;
249+ repeated float epoch_validate_f1_score = 10 ;
250+ repeated float epoch_train_f1_score = 11 ;
234251}
235252
236253message CMsgSteamLearn_GetTrainStatusVersions_Request {
@@ -997,6 +1014,20 @@ message CMsgSteamLearnScheduledTrainConfig {
9971014 optional float auto_activate_accuracy_threshold = 6 ;
9981015}
9991016
1017+ message CMsgSteamLearnSnapshotStorage {
1018+ repeated .CMsgSteamLearnSnapshotStorage_MapStorageElementsEntry map_storage_elements = 1 ;
1019+ }
1020+
1021+ message CMsgSteamLearnSnapshotStorage_MapStorageElementsEntry {
1022+ optional uint32 key = 1 ;
1023+ optional .CMsgSteamLearnSnapshotStorage_StorageElement value = 2 ;
1024+ }
1025+
1026+ message CMsgSteamLearnSnapshotStorage_StorageElement {
1027+ optional float float_value = 1 ;
1028+ optional string string_value = 2 ;
1029+ }
1030+
10001031message CMsgSteamLearnTrainConfig {
10011032 optional uint32 fetch_workers = 1 ;
10021033 optional uint32 fetch_chunk_size = 2 ;
@@ -1036,6 +1067,7 @@ service SteamLearn {
10361067 rpc GetLogEvents (.CMsgSteamLearn_GetLogEvents_Request ) returns (.CMsgSteamLearn_GetLogEvents_Response );
10371068 rpc GetNearestEmbedding (.CMsgSteamLearn_GetNearestEmbedding_Request ) returns (.CMsgSteamLearn_GetNearestEmbedding_Response );
10381069 rpc GetProject (.CMsgSteamLearn_GetProject_Request ) returns (.CMsgSteamLearn_GetProject_Response );
1070+ rpc GetSnapshotStatus (.CMsgSteamLearn_GetSnapshotStatus_Request ) returns (.CMsgSteamLearn_GetSnapshotStatus_Response );
10391071 rpc GetTrainLogs (.CMsgSteamLearn_GetTrainLogs_Request ) returns (.CMsgSteamLearn_GetTrainLogs_Response );
10401072 rpc GetTrainStatus (.CMsgSteamLearn_GetTrainStatus_Request ) returns (.CMsgSteamLearn_GetTrainStatus_Response );
10411073 rpc GetTrainStatusVersions (.CMsgSteamLearn_GetTrainStatusVersions_Request ) returns (.CMsgSteamLearn_GetTrainStatusVersions_Response );
0 commit comments