singnet#565

Grpc server impl of training.proto - Skeleton code checkin
anandrgitnirman · May 11, 2022 · 1013417 · 1013417
1 parent 9eadb2c
commit 1013417
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/config/config.go b/config/config.go
@@ -37,6 +37,7 @@ const (
 	LogKey                         = "log"
 	MaxMessageSizeInMB             = "max_message_size_in_mb"
 	MeteringEnabled                = "metering_enabled"
+	ModelTrainingEndpoint          = "model_training_endpoint"
 	OrganizationId                 = "organization_id"
 	ServiceId                      = "service_id"
 	PassthroughEnabledKey          = "passthrough_enabled"

diff --git a/training/storage.go b/training/storage.go
@@ -26,16 +26,19 @@ type ModelUserKey struct {
 	ServiceId      string
 	GroupID        string
 	ChannelId      *big.Int
+	ModelId        string
 }
 
 func (key *ModelUserKey) String() string {
-	return fmt.Sprintf("{ID:%v/%v/%v/%v}", key.OrganizationId,
-		key.ServiceId, key.GroupID, key.ChannelId)
+	return fmt.Sprintf("{ID:%v/%v/%v/%v/%v}", key.OrganizationId,
+		key.ServiceId, key.GroupID, key.ChannelId, key.ModelId)
 }
 
 type ModelUserData struct {
 	isPublic            bool
 	AuthorizedAddresses []string
+	Status              string
+	CreatedByAddress    string
 }
 
 func serializeModelKey(key interface{}) (serialized string, err error) {