@@ -338,6 +338,23 @@ class App(google___protobuf___message___Message):
338338 def ClearField (self , field_name : typing_extensions___Literal [u"created_at" ,b"created_at" ,u"default_language" ,b"default_language" ,u"default_workflow_id" ,b"default_workflow_id" ,u"description" ,b"description" ,u"id" ,b"id" ,u"legal_consent_status" ,b"legal_consent_status" ,u"metadata" ,b"metadata" ,u"name" ,b"name" ,u"sample_ms" ,b"sample_ms" ,u"user_id" ,b"user_id" ]) -> None : ...
339339type___App = App
340340
341+ class AppQuery (google___protobuf___message___Message ):
342+ DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
343+ name : typing___Text = ...
344+
345+ def __init__ (self ,
346+ * ,
347+ name : typing___Optional [typing___Text ] = None ,
348+ ) -> None : ...
349+ if sys .version_info >= (3 ,):
350+ @classmethod
351+ def FromString (cls , s : builtin___bytes ) -> AppQuery : ...
352+ else :
353+ @classmethod
354+ def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> AppQuery : ...
355+ def ClearField (self , field_name : typing_extensions___Literal [u"name" ,b"name" ]) -> None : ...
356+ type___AppQuery = AppQuery
357+
341358class Collaborator (google___protobuf___message___Message ):
342359 DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
343360 id : typing___Text = ...
@@ -1549,6 +1566,7 @@ class ModelTypeField(google___protobuf___message___Message):
15491566 COLLABORATORS = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 9 )
15501567 JSON = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 10 )
15511568 ARRAY_OF_NUMBERS = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 11 )
1569+ WORKFLOW_EMBED_MODELS = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 12 )
15521570 INVALID_MODEL_TYPE_FIELD_TYPE = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 0 )
15531571 BOOLEAN = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 1 )
15541572 STRING = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 2 )
@@ -1560,20 +1578,25 @@ class ModelTypeField(google___protobuf___message___Message):
15601578 COLLABORATORS = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 9 )
15611579 JSON = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 10 )
15621580 ARRAY_OF_NUMBERS = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 11 )
1581+ WORKFLOW_EMBED_MODELS = typing___cast (ModelTypeField .ModelTypeFieldTypeValue , 12 )
15631582 type___ModelTypeFieldType = ModelTypeFieldType
15641583
15651584 path : typing___Text = ...
15661585 field_type : type___ModelTypeField .ModelTypeFieldTypeValue = ...
15671586 description : typing___Text = ...
15681587 placeholder : typing___Text = ...
15691588 internal_only : builtin___bool = ...
1589+ required : builtin___bool = ...
15701590
15711591 @property
15721592 def default_value (self ) -> google___protobuf___struct_pb2___Value : ...
15731593
15741594 @property
15751595 def model_type_enum_options (self ) -> google___protobuf___internal___containers___RepeatedCompositeFieldContainer [type___ModelTypeEnumOption ]: ...
15761596
1597+ @property
1598+ def model_type_range_info (self ) -> type___ModelTypeRangeInfo : ...
1599+
15771600 def __init__ (self ,
15781601 * ,
15791602 path : typing___Optional [typing___Text ] = None ,
@@ -1583,17 +1606,40 @@ class ModelTypeField(google___protobuf___message___Message):
15831606 placeholder : typing___Optional [typing___Text ] = None ,
15841607 model_type_enum_options : typing___Optional [typing___Iterable [type___ModelTypeEnumOption ]] = None ,
15851608 internal_only : typing___Optional [builtin___bool ] = None ,
1609+ required : typing___Optional [builtin___bool ] = None ,
1610+ model_type_range_info : typing___Optional [type___ModelTypeRangeInfo ] = None ,
15861611 ) -> None : ...
15871612 if sys .version_info >= (3 ,):
15881613 @classmethod
15891614 def FromString (cls , s : builtin___bytes ) -> ModelTypeField : ...
15901615 else :
15911616 @classmethod
15921617 def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> ModelTypeField : ...
1593- def HasField (self , field_name : typing_extensions___Literal [u"default_value" ,b"default_value" ]) -> builtin___bool : ...
1594- def ClearField (self , field_name : typing_extensions___Literal [u"default_value" ,b"default_value" ,u"description" ,b"description" ,u"field_type" ,b"field_type" ,u"internal_only" ,b"internal_only" ,u"model_type_enum_options" ,b"model_type_enum_options" ,u"path" ,b"path" ,u"placeholder" ,b"placeholder" ]) -> None : ...
1618+ def HasField (self , field_name : typing_extensions___Literal [u"default_value" ,b"default_value" , u"model_type_range_info" , b"model_type_range_info" ]) -> builtin___bool : ...
1619+ def ClearField (self , field_name : typing_extensions___Literal [u"default_value" ,b"default_value" ,u"description" ,b"description" ,u"field_type" ,b"field_type" ,u"internal_only" ,b"internal_only" ,u"model_type_enum_options" ,b"model_type_enum_options" ,u"model_type_range_info" , b"model_type_range_info" , u" path" ,b"path" ,u"placeholder" ,b"placeholder" , u"required" , b"required " ]) -> None : ...
15951620type___ModelTypeField = ModelTypeField
15961621
1622+ class ModelTypeRangeInfo (google___protobuf___message___Message ):
1623+ DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
1624+ min : builtin___float = ...
1625+ max : builtin___float = ...
1626+ step : builtin___float = ...
1627+
1628+ def __init__ (self ,
1629+ * ,
1630+ min : typing___Optional [builtin___float ] = None ,
1631+ max : typing___Optional [builtin___float ] = None ,
1632+ step : typing___Optional [builtin___float ] = None ,
1633+ ) -> None : ...
1634+ if sys .version_info >= (3 ,):
1635+ @classmethod
1636+ def FromString (cls , s : builtin___bytes ) -> ModelTypeRangeInfo : ...
1637+ else :
1638+ @classmethod
1639+ def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> ModelTypeRangeInfo : ...
1640+ def ClearField (self , field_name : typing_extensions___Literal [u"max" ,b"max" ,u"min" ,b"min" ,u"step" ,b"step" ]) -> None : ...
1641+ type___ModelTypeRangeInfo = ModelTypeRangeInfo
1642+
15971643class ModelTypeEnumOption (google___protobuf___message___Message ):
15981644 DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
15991645 id : typing___Text = ...
@@ -1623,19 +1669,21 @@ class ModelQuery(google___protobuf___message___Message):
16231669 DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
16241670 name : typing___Text = ...
16251671 type : typing___Text = ...
1672+ model_type_id : typing___Text = ...
16261673
16271674 def __init__ (self ,
16281675 * ,
16291676 name : typing___Optional [typing___Text ] = None ,
16301677 type : typing___Optional [typing___Text ] = None ,
1678+ model_type_id : typing___Optional [typing___Text ] = None ,
16311679 ) -> None : ...
16321680 if sys .version_info >= (3 ,):
16331681 @classmethod
16341682 def FromString (cls , s : builtin___bytes ) -> ModelQuery : ...
16351683 else :
16361684 @classmethod
16371685 def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> ModelQuery : ...
1638- def ClearField (self , field_name : typing_extensions___Literal [u"name" ,b"name" ,u"type" ,b"type" ]) -> None : ...
1686+ def ClearField (self , field_name : typing_extensions___Literal [u"model_type_id" , b"model_type_id" , u" name" ,b"name" ,u"type" ,b"type" ]) -> None : ...
16391687type___ModelQuery = ModelQuery
16401688
16411689class ModelVersion (google___protobuf___message___Message ):
@@ -1682,6 +1730,7 @@ class PretrainedModelConfig(google___protobuf___message___Message):
16821730 type : typing___Text = ...
16831731 data_provider_params : typing___Text = ...
16841732 model_zip_url : typing___Text = ...
1733+ overwrite : builtin___bool = ...
16851734
16861735 @property
16871736 def input_fields_map (self ) -> google___protobuf___struct_pb2___Struct : ...
@@ -1697,6 +1746,7 @@ class PretrainedModelConfig(google___protobuf___message___Message):
16971746 output_fields_map : typing___Optional [google___protobuf___struct_pb2___Struct ] = None ,
16981747 data_provider_params : typing___Optional [typing___Text ] = None ,
16991748 model_zip_url : typing___Optional [typing___Text ] = None ,
1749+ overwrite : typing___Optional [builtin___bool ] = None ,
17001750 ) -> None : ...
17011751 if sys .version_info >= (3 ,):
17021752 @classmethod
@@ -1705,7 +1755,7 @@ class PretrainedModelConfig(google___protobuf___message___Message):
17051755 @classmethod
17061756 def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> PretrainedModelConfig : ...
17071757 def HasField (self , field_name : typing_extensions___Literal [u"input_fields_map" ,b"input_fields_map" ,u"output_fields_map" ,b"output_fields_map" ]) -> builtin___bool : ...
1708- def ClearField (self , field_name : typing_extensions___Literal [u"data_provider_params" ,b"data_provider_params" ,u"id" ,b"id" ,u"input_fields_map" ,b"input_fields_map" ,u"model_zip_url" ,b"model_zip_url" ,u"output_fields_map" ,b"output_fields_map" ,u"type" ,b"type" ]) -> None : ...
1758+ def ClearField (self , field_name : typing_extensions___Literal [u"data_provider_params" ,b"data_provider_params" ,u"id" ,b"id" ,u"input_fields_map" ,b"input_fields_map" ,u"model_zip_url" ,b"model_zip_url" ,u"output_fields_map" ,b"output_fields_map" ,u"overwrite" , b"overwrite" , u" type" ,b"type" ]) -> None : ...
17091759type___PretrainedModelConfig = PretrainedModelConfig
17101760
17111761class TrainStats (google___protobuf___message___Message ):
@@ -2880,28 +2930,82 @@ class TaskWorker(google___protobuf___message___Message):
28802930 @classmethod
28812931 def items (cls ) -> typing___List [typing___Tuple [builtin___str , TaskWorker .TaskWorkerStrategyValue ]]: ...
28822932 WORKER_STRATEGY_NOT_SET = typing___cast (TaskWorker .TaskWorkerStrategyValue , 0 )
2883- MANUAL = typing___cast (TaskWorker .TaskWorkerStrategyValue , 1 )
2933+ PARTITIONED = typing___cast (TaskWorker .TaskWorkerStrategyValue , 2 )
2934+ FULL = typing___cast (TaskWorker .TaskWorkerStrategyValue , 3 )
28842935 WORKER_STRATEGY_NOT_SET = typing___cast (TaskWorker .TaskWorkerStrategyValue , 0 )
2885- MANUAL = typing___cast (TaskWorker .TaskWorkerStrategyValue , 1 )
2936+ PARTITIONED = typing___cast (TaskWorker .TaskWorkerStrategyValue , 2 )
2937+ FULL = typing___cast (TaskWorker .TaskWorkerStrategyValue , 3 )
28862938 type___TaskWorkerStrategy = TaskWorkerStrategy
28872939
28882940 strategy : type___TaskWorker .TaskWorkerStrategyValue = ...
28892941 user_ids : google___protobuf___internal___containers___RepeatedScalarFieldContainer [typing___Text ] = ...
28902942
2943+ @property
2944+ def partitioned_strategy_info (self ) -> type___TaskWorkerPartitionedStrategyInfo : ...
2945+
28912946 def __init__ (self ,
28922947 * ,
28932948 strategy : typing___Optional [type___TaskWorker .TaskWorkerStrategyValue ] = None ,
28942949 user_ids : typing___Optional [typing___Iterable [typing___Text ]] = None ,
2950+ partitioned_strategy_info : typing___Optional [type___TaskWorkerPartitionedStrategyInfo ] = None ,
28952951 ) -> None : ...
28962952 if sys .version_info >= (3 ,):
28972953 @classmethod
28982954 def FromString (cls , s : builtin___bytes ) -> TaskWorker : ...
28992955 else :
29002956 @classmethod
29012957 def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> TaskWorker : ...
2902- def ClearField (self , field_name : typing_extensions___Literal [u"strategy" ,b"strategy" ,u"user_ids" ,b"user_ids" ]) -> None : ...
2958+ def HasField (self , field_name : typing_extensions___Literal [u"partitioned_strategy_info" ,b"partitioned_strategy_info" ,u"strategy_info" ,b"strategy_info" ]) -> builtin___bool : ...
2959+ def ClearField (self , field_name : typing_extensions___Literal [u"partitioned_strategy_info" ,b"partitioned_strategy_info" ,u"strategy" ,b"strategy" ,u"strategy_info" ,b"strategy_info" ,u"user_ids" ,b"user_ids" ]) -> None : ...
2960+ def WhichOneof (self , oneof_group : typing_extensions___Literal [u"strategy_info" ,b"strategy_info" ]) -> typing_extensions___Literal ["partitioned_strategy_info" ]: ...
29032961type___TaskWorker = TaskWorker
29042962
2963+ class TaskWorkerPartitionedStrategyInfo (google___protobuf___message___Message ):
2964+ DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
2965+ TaskWorkerPartitionedStrategyValue = typing___NewType ('TaskWorkerPartitionedStrategyValue' , builtin___int )
2966+ type___TaskWorkerPartitionedStrategyValue = TaskWorkerPartitionedStrategyValue
2967+ class TaskWorkerPartitionedStrategy (object ):
2968+ DESCRIPTOR : google___protobuf___descriptor___EnumDescriptor = ...
2969+ @classmethod
2970+ def Name (cls , number : builtin___int ) -> builtin___str : ...
2971+ @classmethod
2972+ def Value (cls , name : builtin___str ) -> TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue : ...
2973+ @classmethod
2974+ def keys (cls ) -> typing___List [builtin___str ]: ...
2975+ @classmethod
2976+ def values (cls ) -> typing___List [TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue ]: ...
2977+ @classmethod
2978+ def items (cls ) -> typing___List [typing___Tuple [builtin___str , TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue ]]: ...
2979+ PARTITIONED_WORKER_STRATEGY_NOT_SET = typing___cast (TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue , 0 )
2980+ EVENLY = typing___cast (TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue , 1 )
2981+ WEIGHTED = typing___cast (TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue , 2 )
2982+ PARTITIONED_WORKER_STRATEGY_NOT_SET = typing___cast (TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue , 0 )
2983+ EVENLY = typing___cast (TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue , 1 )
2984+ WEIGHTED = typing___cast (TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue , 2 )
2985+ type___TaskWorkerPartitionedStrategy = TaskWorkerPartitionedStrategy
2986+
2987+ type : type___TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue = ...
2988+ workers_per_input : builtin___int = ...
2989+
2990+ @property
2991+ def weights (self ) -> google___protobuf___struct_pb2___Struct : ...
2992+
2993+ def __init__ (self ,
2994+ * ,
2995+ type : typing___Optional [type___TaskWorkerPartitionedStrategyInfo .TaskWorkerPartitionedStrategyValue ] = None ,
2996+ workers_per_input : typing___Optional [builtin___int ] = None ,
2997+ weights : typing___Optional [google___protobuf___struct_pb2___Struct ] = None ,
2998+ ) -> None : ...
2999+ if sys .version_info >= (3 ,):
3000+ @classmethod
3001+ def FromString (cls , s : builtin___bytes ) -> TaskWorkerPartitionedStrategyInfo : ...
3002+ else :
3003+ @classmethod
3004+ def FromString (cls , s : typing___Union [builtin___bytes , builtin___buffer , builtin___unicode ]) -> TaskWorkerPartitionedStrategyInfo : ...
3005+ def HasField (self , field_name : typing_extensions___Literal [u"weights" ,b"weights" ]) -> builtin___bool : ...
3006+ def ClearField (self , field_name : typing_extensions___Literal [u"type" ,b"type" ,u"weights" ,b"weights" ,u"workers_per_input" ,b"workers_per_input" ]) -> None : ...
3007+ type___TaskWorkerPartitionedStrategyInfo = TaskWorkerPartitionedStrategyInfo
3008+
29053009class TaskInputSource (google___protobuf___message___Message ):
29063010 DESCRIPTOR : google___protobuf___descriptor___Descriptor = ...
29073011 TaskInputSourceTypeValue = typing___NewType ('TaskInputSourceTypeValue' , builtin___int )
0 commit comments