gimlet-api 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {gimlet_api-0.0.12.dist-info → gimlet_api-0.0.14.dist-info}/METADATA +3 -2
- {gimlet_api-0.0.12.dist-info → gimlet_api-0.0.14.dist-info}/RECORD +27 -24
- gml/client.py +71 -26
- gml/compile.py +16 -12
- gml/device.py +2 -2
- gml/hf.py +100 -63
- gml/model_utils.py +10 -1
- gml/pipelines.py +22 -6
- gml/proto/opentelemetry/proto/common/v1/common_pb2.py +3 -1
- gml/proto/opentelemetry/proto/resource/v1/resource_pb2.py +2 -2
- gml/proto/src/api/corepb/v1/controlplane_pb2.py +50 -39
- gml/proto/src/api/corepb/v1/cp_dp_pb2.py +88 -0
- gml/proto/src/api/corepb/v1/cp_edge_pb2.py +116 -69
- gml/proto/src/api/corepb/v1/dataplane_pb2.py +33 -0
- gml/proto/src/api/corepb/v1/deployed_pipeline_pb2.py +31 -7
- gml/proto/src/api/corepb/v1/device_info_pb2.py +19 -17
- gml/proto/src/api/corepb/v1/mediastream_pb2.py +15 -3
- gml/proto/src/api/corepb/v1/model_exec_pb2.py +123 -103
- gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py +10 -8
- gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2.py +50 -50
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py +31 -24
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py +33 -0
- gml/proto/src/controlplane/model/mpb/v1/mpb_pb2.py +17 -11
- gml/proto/src/controlplane/model/mpb/v1/mpb_pb2_grpc.py +33 -0
- gml/tensor.py +7 -0
- gml/version_utils.py +43 -0
- {gimlet_api-0.0.12.dist-info → gimlet_api-0.0.14.dist-info}/WHEEL +0 -0
@@ -20,7 +20,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
|
|
20
20
|
from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
21
21
|
|
22
22
|
|
23
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*src/controlplane/compiler/cpb/v1/cpb.proto\x12%gml.internal.controlplane.compiler.v1\x1a\"src/api/corepb/v1/model_exec.proto\x1a)src/api/corepb/v1/compiled_pipeline.proto\x1a$src/api/corepb/v1/controlplane.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\"src/api/corepb/v1/gem_config.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x14gogoproto/gogo.proto\"\
|
23
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*src/controlplane/compiler/cpb/v1/cpb.proto\x12%gml.internal.controlplane.compiler.v1\x1a\"src/api/corepb/v1/model_exec.proto\x1a)src/api/corepb/v1/compiled_pipeline.proto\x1a$src/api/corepb/v1/controlplane.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\"src/api/corepb/v1/gem_config.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x14gogoproto/gogo.proto\"\xe9\x04\n\x0e\x43ompileRequest\x12V\n\x13logical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12T\n\x10logical_pipeline\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12\x61\n\x13\x64\x65vice_capabilities\x18\x03 \x01(\x0b\x32,.gml.internal.api.core.v1.DeviceCapabilitiesB\x02\x18\x01R\x12\x64\x65viceCapabilities\x12\x37\n\x06models\x18\x04 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ModelR\x06models\x12\x46\n\ngem_config\x18\x05 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigB\x02\x18\x01R\tgemConfig\x12:\n\x07\x64\x65vices\x18\x06 \x03(\x0b\x32 .gml.internal.api.core.v1.DeviceR\x07\x64\x65vices\x12G\n\x0c\x64\x65vice_links\x18\x07 \x03(\x0b\x32$.gml.internal.api.core.v1.DeviceLinkR\x0b\x64\x65viceLinks\x12@\n\x0b\x65ndpoint_id\x18\x08 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0e\xe2\xde\x1f\nEndpointIDR\nendpointId\"\x92\x03\n\x0f\x43ompileResponse\x12H\n\texec_spec\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecB\x02\x18\x01R\x08\x65xecSpec\x12h\n\nexec_specs\x18\x02 \x03(\x0b\x32\x45.gml.internal.controlplane.compiler.v1.CompileResponse.ExecSpecsEntryB\x02\x18\x01R\texecSpecs\x12\x64\n\x16\x63ompiled_pipeline_spec\x18\x03 \x01(\x0b\x32..gml.internal.api.core.v1.CompiledPipelineSpecR\x14\x63ompiledPipelineSpec\x1a\x65\n\x0e\x45xecSpecsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12=\n\x05value\x18\x02 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05value:\x02\x38\x01\x32\x8b\x01\n\x0f\x43ompilerService\x12x\n\x07\x43ompile\x12\x35.gml.internal.controlplane.compiler.v1.CompileRequest\x1a\x36.gml.internal.controlplane.compiler.v1.CompileResponseB;Z9gimletlabs.ai/gimlet/src/controlplane/compiler/cpb/v1;cpbb\x06proto3')
|
24
24
|
|
25
25
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
26
26
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.compiler.cpb.v1.cpb_pb2', globals())
|
@@ -34,6 +34,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
34
34
|
_COMPILEREQUEST.fields_by_name['device_capabilities']._serialized_options = b'\030\001'
|
35
35
|
_COMPILEREQUEST.fields_by_name['gem_config']._options = None
|
36
36
|
_COMPILEREQUEST.fields_by_name['gem_config']._serialized_options = b'\030\001'
|
37
|
+
_COMPILEREQUEST.fields_by_name['endpoint_id']._options = None
|
38
|
+
_COMPILEREQUEST.fields_by_name['endpoint_id']._serialized_options = b'\342\336\037\nEndpointID'
|
37
39
|
_COMPILERESPONSE_EXECSPECSENTRY._options = None
|
38
40
|
_COMPILERESPONSE_EXECSPECSENTRY._serialized_options = b'8\001'
|
39
41
|
_COMPILERESPONSE.fields_by_name['exec_spec']._options = None
|
@@ -41,11 +43,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
41
43
|
_COMPILERESPONSE.fields_by_name['exec_specs']._options = None
|
42
44
|
_COMPILERESPONSE.fields_by_name['exec_specs']._serialized_options = b'\030\001'
|
43
45
|
_COMPILEREQUEST._serialized_start=325
|
44
|
-
_COMPILEREQUEST._serialized_end=
|
45
|
-
_COMPILERESPONSE._serialized_start=
|
46
|
-
_COMPILERESPONSE._serialized_end=
|
47
|
-
_COMPILERESPONSE_EXECSPECSENTRY._serialized_start=
|
48
|
-
_COMPILERESPONSE_EXECSPECSENTRY._serialized_end=
|
49
|
-
_COMPILERSERVICE._serialized_start=
|
50
|
-
_COMPILERSERVICE._serialized_end=
|
46
|
+
_COMPILEREQUEST._serialized_end=942
|
47
|
+
_COMPILERESPONSE._serialized_start=945
|
48
|
+
_COMPILERESPONSE._serialized_end=1347
|
49
|
+
_COMPILERESPONSE_EXECSPECSENTRY._serialized_start=1246
|
50
|
+
_COMPILERESPONSE_EXECSPECSENTRY._serialized_end=1347
|
51
|
+
_COMPILERSERVICE._serialized_start=1350
|
52
|
+
_COMPILERSERVICE._serialized_end=1489
|
51
53
|
# @@protoc_insertion_point(module_scope)
|
@@ -16,7 +16,7 @@ from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb
|
|
16
16
|
from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_dot_uuid__pb2
|
17
17
|
|
18
18
|
|
19
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9src/controlplane/directory/directorypb/v1/directory.proto\x12&gml.internal.controlplane.directory.v1\x1a\x14gogoproto/gogo.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1dsrc/common/typespb/uuid.proto\"\xc0\x02\n\x08UserInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x14\n\x05\x65mail\x18\x03 \x01(\tR\x05\x65mail\x12\x41\n\x0f\x64isplay_picture\x18\x04 \x01(\tB\x18\xf2\xde\x1f\x14\x64\x62:\"display_picture\"R\x0e\x64isplayPicture\x12G\n\x11identity_provider\x18\x05 \x01(\tB\x1a\xf2\xde\x1f\x16\x64\x62:\"identity_provider\"R\x10identityProvider\x12U\n\x10\x61uth_provider_id\x18\x06 \x01(\tB+\xe2\xde\x1f\x0e\x41uthProviderID\xf2\xde\x1f\x15\x64\x62:\"auth_provider_id\"R\x0e\x61uthProviderId\"b\n\x11UpsertUserRequest\x12M\n\tuser_info\x18\x01 \x01(\x0b\x32\x30.gml.internal.controlplane.directory.v1.UserInfoR\x08userInfo\"}\n\x12UpsertUserResponse\x12M\n\tuser_info\x18\x01 \x01(\x0b\x32\x30.gml.internal.controlplane.directory.v1.UserInfoR\x08userInfo\x12\x18\n\x07\x63reated\x18\x02 \x01(\x08R\x07\x63reated\"
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9src/controlplane/directory/directorypb/v1/directory.proto\x12&gml.internal.controlplane.directory.v1\x1a\x14gogoproto/gogo.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1dsrc/common/typespb/uuid.proto\"\xc0\x02\n\x08UserInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x14\n\x05\x65mail\x18\x03 \x01(\tR\x05\x65mail\x12\x41\n\x0f\x64isplay_picture\x18\x04 \x01(\tB\x18\xf2\xde\x1f\x14\x64\x62:\"display_picture\"R\x0e\x64isplayPicture\x12G\n\x11identity_provider\x18\x05 \x01(\tB\x1a\xf2\xde\x1f\x16\x64\x62:\"identity_provider\"R\x10identityProvider\x12U\n\x10\x61uth_provider_id\x18\x06 \x01(\tB+\xe2\xde\x1f\x0e\x41uthProviderID\xf2\xde\x1f\x15\x64\x62:\"auth_provider_id\"R\x0e\x61uthProviderId\"b\n\x11UpsertUserRequest\x12M\n\tuser_info\x18\x01 \x01(\x0b\x32\x30.gml.internal.controlplane.directory.v1.UserInfoR\x08userInfo\"}\n\x12UpsertUserResponse\x12M\n\tuser_info\x18\x01 \x01(\x0b\x32\x30.gml.internal.controlplane.directory.v1.UserInfoR\x08userInfo\x12\x18\n\x07\x63reated\x18\x02 \x01(\x08R\x07\x63reated\"O\n\x0eGetUserRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x14\n\x05\x65mail\x18\x02 \x01(\tR\x05\x65mail\"`\n\x0fGetUserResponse\x12M\n\tuser_info\x18\x01 \x01(\x0b\x32\x30.gml.internal.controlplane.directory.v1.UserInfoR\x08userInfo\"\x83\x01\n\x11UpdateUserRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x45\n\x0f\x64isplay_picture\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.StringValueR\x0e\x64isplayPicture\"\x14\n\x12UpdateUserResponse\"<\n\x11\x44\x65leteUserRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x14\n\x12\x44\x65leteUserResponse\"\\\n\x07OrgInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12(\n\x08org_name\x18\x02 \x01(\tB\r\xf2\xde\x1f\tdb:\"name\"R\x07orgName\"\x9d\x01\n\x10\x43reateOrgRequest\x12J\n\x08org_info\x18\x01 \x01(\x0b\x32/.gml.internal.controlplane.directory.v1.OrgInfoR\x07orgInfo\x12=\n\ncreator_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\r\xe2\xde\x1f\tCreatorIDR\tcreatorId\"<\n\x11\x43reateOrgResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"S\n\rGetOrgRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x19\n\x08org_name\x18\x02 \x01(\tR\x07orgName\"\\\n\x0eGetOrgResponse\x12J\n\x08org_info\x18\x01 \x01(\x0b\x32/.gml.internal.controlplane.directory.v1.OrgInfoR\x07orgInfo\";\n\x10\x44\x65leteOrgRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x13\n\x11\x44\x65leteOrgResponse\"\x18\n\x16GrantUserScopesRequest\"\x19\n\x17GrantUserScopesResponse\"\x19\n\x17RevokeUserScopesRequest\"\x1a\n\x18RevokeUserScopesResponse\"G\n\x0fListOrgsRequest\x12\x34\n\x07user_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06UserIDR\x06userId\"H\n\x10ListOrgsResponse\x12\x34\n\x07org_ids\x18\x01 \x03(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06OrgIDsR\x06orgIds\"z\n\x0cUserRoleInfo\x12M\n\tuser_info\x18\x01 \x01(\x0b\x32\x30.gml.internal.controlplane.directory.v1.UserInfoR\x08userInfo\x12\x1b\n\trole_name\x18\x02 \x01(\tR\x08roleName\"D\n\x0fGetUsersRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"g\n\x10GetUsersResponse\x12S\n\nuser_roles\x18\x01 \x03(\x0b\x32\x34.gml.internal.controlplane.directory.v1.UserRoleInfoR\tuserRoles2\xa4\x04\n\x14UserDirectoryService\x12z\n\x07GetUser\x12\x36.gml.internal.controlplane.directory.v1.GetUserRequest\x1a\x37.gml.internal.controlplane.directory.v1.GetUserResponse\x12\x83\x01\n\nUpdateUser\x12\x39.gml.internal.controlplane.directory.v1.UpdateUserRequest\x1a:.gml.internal.controlplane.directory.v1.UpdateUserResponse\x12\x83\x01\n\nDeleteUser\x12\x39.gml.internal.controlplane.directory.v1.DeleteUserRequest\x1a:.gml.internal.controlplane.directory.v1.DeleteUserResponse\x12\x83\x01\n\nUpsertUser\x12\x39.gml.internal.controlplane.directory.v1.UpsertUserRequest\x1a:.gml.internal.controlplane.directory.v1.UpsertUserResponse2\x94\x03\n\x13OrgDirectoryService\x12\x80\x01\n\tCreateOrg\x12\x38.gml.internal.controlplane.directory.v1.CreateOrgRequest\x1a\x39.gml.internal.controlplane.directory.v1.CreateOrgResponse\x12w\n\x06GetOrg\x12\x35.gml.internal.controlplane.directory.v1.GetOrgRequest\x1a\x36.gml.internal.controlplane.directory.v1.GetOrgResponse\x12\x80\x01\n\tDeleteOrg\x12\x38.gml.internal.controlplane.directory.v1.DeleteOrgRequest\x1a\x39.gml.internal.controlplane.directory.v1.DeleteOrgResponse2\xc5\x04\n\x18OrgUserManagementService\x12\x92\x01\n\x0fGrantUserScopes\x12>.gml.internal.controlplane.directory.v1.GrantUserScopesRequest\x1a?.gml.internal.controlplane.directory.v1.GrantUserScopesResponse\x12\x95\x01\n\x10RevokeUserScopes\x12?.gml.internal.controlplane.directory.v1.RevokeUserScopesRequest\x1a@.gml.internal.controlplane.directory.v1.RevokeUserScopesResponse\x12}\n\x08ListOrgs\x12\x37.gml.internal.controlplane.directory.v1.ListOrgsRequest\x1a\x38.gml.internal.controlplane.directory.v1.ListOrgsResponse\x12}\n\x08GetUsers\x12\x37.gml.internal.controlplane.directory.v1.GetUsersRequest\x1a\x38.gml.internal.controlplane.directory.v1.GetUsersResponseBLZJgimletlabs.ai/gimlet/src/controlplane/directory/directorypb/v1;directorypbb\x06proto3')
|
20
20
|
|
21
21
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
22
22
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.directory.directorypb.v1.directory_pb2', globals())
|
@@ -63,53 +63,53 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
63
63
|
_UPSERTUSERRESPONSE._serialized_start=609
|
64
64
|
_UPSERTUSERRESPONSE._serialized_end=734
|
65
65
|
_GETUSERREQUEST._serialized_start=736
|
66
|
-
_GETUSERREQUEST._serialized_end=
|
67
|
-
_GETUSERRESPONSE._serialized_start=
|
68
|
-
_GETUSERRESPONSE._serialized_end=
|
69
|
-
_UPDATEUSERREQUEST._serialized_start=
|
70
|
-
_UPDATEUSERREQUEST._serialized_end=
|
71
|
-
_UPDATEUSERRESPONSE._serialized_start=
|
72
|
-
_UPDATEUSERRESPONSE._serialized_end=
|
73
|
-
_DELETEUSERREQUEST._serialized_start=
|
74
|
-
_DELETEUSERREQUEST._serialized_end=
|
75
|
-
_DELETEUSERRESPONSE._serialized_start=
|
76
|
-
_DELETEUSERRESPONSE._serialized_end=
|
77
|
-
_ORGINFO._serialized_start=
|
78
|
-
_ORGINFO._serialized_end=
|
79
|
-
_CREATEORGREQUEST._serialized_start=
|
80
|
-
_CREATEORGREQUEST._serialized_end=
|
81
|
-
_CREATEORGRESPONSE._serialized_start=
|
82
|
-
_CREATEORGRESPONSE._serialized_end=
|
83
|
-
_GETORGREQUEST._serialized_start=
|
84
|
-
_GETORGREQUEST._serialized_end=
|
85
|
-
_GETORGRESPONSE._serialized_start=
|
86
|
-
_GETORGRESPONSE._serialized_end=
|
87
|
-
_DELETEORGREQUEST._serialized_start=
|
88
|
-
_DELETEORGREQUEST._serialized_end=
|
89
|
-
_DELETEORGRESPONSE._serialized_start=
|
90
|
-
_DELETEORGRESPONSE._serialized_end=
|
91
|
-
_GRANTUSERSCOPESREQUEST._serialized_start=
|
92
|
-
_GRANTUSERSCOPESREQUEST._serialized_end=
|
93
|
-
_GRANTUSERSCOPESRESPONSE._serialized_start=
|
94
|
-
_GRANTUSERSCOPESRESPONSE._serialized_end=
|
95
|
-
_REVOKEUSERSCOPESREQUEST._serialized_start=
|
96
|
-
_REVOKEUSERSCOPESREQUEST._serialized_end=
|
97
|
-
_REVOKEUSERSCOPESRESPONSE._serialized_start=
|
98
|
-
_REVOKEUSERSCOPESRESPONSE._serialized_end=
|
99
|
-
_LISTORGSREQUEST._serialized_start=
|
100
|
-
_LISTORGSREQUEST._serialized_end=
|
101
|
-
_LISTORGSRESPONSE._serialized_start=
|
102
|
-
_LISTORGSRESPONSE._serialized_end=
|
103
|
-
_USERROLEINFO._serialized_start=
|
104
|
-
_USERROLEINFO._serialized_end=
|
105
|
-
_GETUSERSREQUEST._serialized_start=
|
106
|
-
_GETUSERSREQUEST._serialized_end=
|
107
|
-
_GETUSERSRESPONSE._serialized_start=
|
108
|
-
_GETUSERSRESPONSE._serialized_end=
|
109
|
-
_USERDIRECTORYSERVICE._serialized_start=
|
110
|
-
_USERDIRECTORYSERVICE._serialized_end=
|
111
|
-
_ORGDIRECTORYSERVICE._serialized_start=
|
112
|
-
_ORGDIRECTORYSERVICE._serialized_end=
|
113
|
-
_ORGUSERMANAGEMENTSERVICE._serialized_start=
|
114
|
-
_ORGUSERMANAGEMENTSERVICE._serialized_end=
|
66
|
+
_GETUSERREQUEST._serialized_end=815
|
67
|
+
_GETUSERRESPONSE._serialized_start=817
|
68
|
+
_GETUSERRESPONSE._serialized_end=913
|
69
|
+
_UPDATEUSERREQUEST._serialized_start=916
|
70
|
+
_UPDATEUSERREQUEST._serialized_end=1047
|
71
|
+
_UPDATEUSERRESPONSE._serialized_start=1049
|
72
|
+
_UPDATEUSERRESPONSE._serialized_end=1069
|
73
|
+
_DELETEUSERREQUEST._serialized_start=1071
|
74
|
+
_DELETEUSERREQUEST._serialized_end=1131
|
75
|
+
_DELETEUSERRESPONSE._serialized_start=1133
|
76
|
+
_DELETEUSERRESPONSE._serialized_end=1153
|
77
|
+
_ORGINFO._serialized_start=1155
|
78
|
+
_ORGINFO._serialized_end=1247
|
79
|
+
_CREATEORGREQUEST._serialized_start=1250
|
80
|
+
_CREATEORGREQUEST._serialized_end=1407
|
81
|
+
_CREATEORGRESPONSE._serialized_start=1409
|
82
|
+
_CREATEORGRESPONSE._serialized_end=1469
|
83
|
+
_GETORGREQUEST._serialized_start=1471
|
84
|
+
_GETORGREQUEST._serialized_end=1554
|
85
|
+
_GETORGRESPONSE._serialized_start=1556
|
86
|
+
_GETORGRESPONSE._serialized_end=1648
|
87
|
+
_DELETEORGREQUEST._serialized_start=1650
|
88
|
+
_DELETEORGREQUEST._serialized_end=1709
|
89
|
+
_DELETEORGRESPONSE._serialized_start=1711
|
90
|
+
_DELETEORGRESPONSE._serialized_end=1730
|
91
|
+
_GRANTUSERSCOPESREQUEST._serialized_start=1732
|
92
|
+
_GRANTUSERSCOPESREQUEST._serialized_end=1756
|
93
|
+
_GRANTUSERSCOPESRESPONSE._serialized_start=1758
|
94
|
+
_GRANTUSERSCOPESRESPONSE._serialized_end=1783
|
95
|
+
_REVOKEUSERSCOPESREQUEST._serialized_start=1785
|
96
|
+
_REVOKEUSERSCOPESREQUEST._serialized_end=1810
|
97
|
+
_REVOKEUSERSCOPESRESPONSE._serialized_start=1812
|
98
|
+
_REVOKEUSERSCOPESRESPONSE._serialized_end=1838
|
99
|
+
_LISTORGSREQUEST._serialized_start=1840
|
100
|
+
_LISTORGSREQUEST._serialized_end=1911
|
101
|
+
_LISTORGSRESPONSE._serialized_start=1913
|
102
|
+
_LISTORGSRESPONSE._serialized_end=1985
|
103
|
+
_USERROLEINFO._serialized_start=1987
|
104
|
+
_USERROLEINFO._serialized_end=2109
|
105
|
+
_GETUSERSREQUEST._serialized_start=2111
|
106
|
+
_GETUSERSREQUEST._serialized_end=2179
|
107
|
+
_GETUSERSRESPONSE._serialized_start=2181
|
108
|
+
_GETUSERSRESPONSE._serialized_end=2284
|
109
|
+
_USERDIRECTORYSERVICE._serialized_start=2287
|
110
|
+
_USERDIRECTORYSERVICE._serialized_end=2835
|
111
|
+
_ORGDIRECTORYSERVICE._serialized_start=2838
|
112
|
+
_ORGDIRECTORYSERVICE._serialized_end=3242
|
113
|
+
_ORGUSERMANAGEMENTSERVICE._serialized_start=3245
|
114
|
+
_ORGUSERMANAGEMENTSERVICE._serialized_end=3826
|
115
115
|
# @@protoc_insertion_point(module_scope)
|
@@ -14,9 +14,10 @@ _sym_db = _symbol_database.Default()
|
|
14
14
|
from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
15
15
|
from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_dot_uuid__pb2
|
16
16
|
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
17
|
+
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
17
18
|
|
18
19
|
|
19
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/controlplane/logicalpipeline/lppb/v1/lppb.proto\x12,gml.internal.controlplane.logicalpipeline.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\
|
20
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/controlplane/logicalpipeline/lppb/v1/lppb.proto\x12,gml.internal.controlplane.logicalpipeline.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xf7\x02\n\x13LogicalPipelineInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04yaml\x18\x04 \x01(\tR\x04yaml\x12N\n\rpipeline_spec\x18\x05 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0cpipelineSpec\x12/\n\x0b\x65ncoded_dag\x18\x06 \x01(\tB\x0e\xe2\xde\x1f\nEncodedDAGR\nencodedDag\x12 \n\x0b\x64\x65scription\x18\x07 \x01(\tR\x0b\x64\x65scription\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\"c\n\x1bListLogicalPipelinesRequest\x12\x44\n\rorg_id_filter\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0f\xe2\xde\x1f\x0bOrgIDFilterR\x0borgIdFilter\"\x7f\n\x1cListLogicalPipelinesResponse\x12_\n\tpipelines\x18\x01 \x03(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\tpipelines\"\xa8\x01\n\x19GetLogicalPipelineRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x33\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\x18\x01\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x19\n\x08org_name\x18\x04 \x01(\tR\x07orgName\"\x9c\x01\n\x1aGetLogicalPipelineResponse\x12\x66\n\rpipeline_info\x18\x02 \x01(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\x0cpipelineInfoJ\x04\x08\x01\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06\"\xa1\x01\n\x1c\x43reateLogicalPipelineRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x12\n\x04yaml\x18\x04 \x01(\tR\x04yaml\x12 \n\x0b\x64\x65scription\x18\x05 \x01(\tR\x0b\x64\x65scriptionJ\x04\x08\x03\x10\x04\"H\n\x1d\x43reateLogicalPipelineResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"5\n\x1fParseLogicalPipelineYAMLRequest\x12\x12\n\x04yaml\x18\x01 \x01(\tR\x04yaml\"x\n ParseLogicalPipelineYAMLResponse\x12T\n\x10logical_pipeline\x18\x01 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\"\x9e\x01\n\x1bShareLogicalPipelineRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x1a\n\x08relation\x18\x02 \x01(\tR\x08relation\x12\x1b\n\tall_users\x18\x03 \x01(\x08R\x08\x61llUsers\x12\x1d\n\nuser_email\x18\x04 \x01(\tR\tuserEmail\"\x1e\n\x1cShareLogicalPipelineResponse2\x91\x07\n\x16LogicalPipelineService\x12\xad\x01\n\x14ListLogicalPipelines\x12I.gml.internal.controlplane.logicalpipeline.v1.ListLogicalPipelinesRequest\x1aJ.gml.internal.controlplane.logicalpipeline.v1.ListLogicalPipelinesResponse\x12\xa7\x01\n\x12GetLogicalPipeline\x12G.gml.internal.controlplane.logicalpipeline.v1.GetLogicalPipelineRequest\x1aH.gml.internal.controlplane.logicalpipeline.v1.GetLogicalPipelineResponse\x12\xb0\x01\n\x15\x43reateLogicalPipeline\x12J.gml.internal.controlplane.logicalpipeline.v1.CreateLogicalPipelineRequest\x1aK.gml.internal.controlplane.logicalpipeline.v1.CreateLogicalPipelineResponse\x12\xb9\x01\n\x18ParseLogicalPipelineYAML\x12M.gml.internal.controlplane.logicalpipeline.v1.ParseLogicalPipelineYAMLRequest\x1aN.gml.internal.controlplane.logicalpipeline.v1.ParseLogicalPipelineYAMLResponse\x12\xad\x01\n\x14ShareLogicalPipeline\x12I.gml.internal.controlplane.logicalpipeline.v1.ShareLogicalPipelineRequest\x1aJ.gml.internal.controlplane.logicalpipeline.v1.ShareLogicalPipelineResponseBDZBgimletlabs.ai/gimlet/src/controlplane/logicalpipeline/lppb/v1;lppbb\x06proto3')
|
20
21
|
|
21
22
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
22
23
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.logicalpipeline.lppb.v1.lppb_pb2', globals())
|
@@ -30,34 +31,40 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
30
31
|
_LOGICALPIPELINEINFO.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
31
32
|
_LOGICALPIPELINEINFO.fields_by_name['encoded_dag']._options = None
|
32
33
|
_LOGICALPIPELINEINFO.fields_by_name['encoded_dag']._serialized_options = b'\342\336\037\nEncodedDAG'
|
33
|
-
_LISTLOGICALPIPELINESREQUEST.fields_by_name['
|
34
|
-
_LISTLOGICALPIPELINESREQUEST.fields_by_name['
|
34
|
+
_LISTLOGICALPIPELINESREQUEST.fields_by_name['org_id_filter']._options = None
|
35
|
+
_LISTLOGICALPIPELINESREQUEST.fields_by_name['org_id_filter']._serialized_options = b'\342\336\037\013OrgIDFilter'
|
35
36
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['id']._options = None
|
36
37
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
37
38
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['org_id']._options = None
|
38
|
-
_GETLOGICALPIPELINEREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
39
|
+
_GETLOGICALPIPELINEREQUEST.fields_by_name['org_id']._serialized_options = b'\030\001\342\336\037\005OrgID'
|
39
40
|
_CREATELOGICALPIPELINEREQUEST.fields_by_name['org_id']._options = None
|
40
41
|
_CREATELOGICALPIPELINEREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
41
42
|
_CREATELOGICALPIPELINERESPONSE.fields_by_name['id']._options = None
|
42
43
|
_CREATELOGICALPIPELINERESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
43
|
-
|
44
|
-
|
45
|
-
|
46
|
-
|
47
|
-
|
48
|
-
|
49
|
-
|
50
|
-
|
51
|
-
|
52
|
-
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
|
59
|
-
|
60
|
-
|
61
|
-
|
62
|
-
|
44
|
+
_SHARELOGICALPIPELINEREQUEST.fields_by_name['id']._options = None
|
45
|
+
_SHARELOGICALPIPELINEREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
46
|
+
_LOGICALPIPELINEINFO._serialized_start=224
|
47
|
+
_LOGICALPIPELINEINFO._serialized_end=599
|
48
|
+
_LISTLOGICALPIPELINESREQUEST._serialized_start=601
|
49
|
+
_LISTLOGICALPIPELINESREQUEST._serialized_end=700
|
50
|
+
_LISTLOGICALPIPELINESRESPONSE._serialized_start=702
|
51
|
+
_LISTLOGICALPIPELINESRESPONSE._serialized_end=829
|
52
|
+
_GETLOGICALPIPELINEREQUEST._serialized_start=832
|
53
|
+
_GETLOGICALPIPELINEREQUEST._serialized_end=1000
|
54
|
+
_GETLOGICALPIPELINERESPONSE._serialized_start=1003
|
55
|
+
_GETLOGICALPIPELINERESPONSE._serialized_end=1159
|
56
|
+
_CREATELOGICALPIPELINEREQUEST._serialized_start=1162
|
57
|
+
_CREATELOGICALPIPELINEREQUEST._serialized_end=1323
|
58
|
+
_CREATELOGICALPIPELINERESPONSE._serialized_start=1325
|
59
|
+
_CREATELOGICALPIPELINERESPONSE._serialized_end=1397
|
60
|
+
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_start=1399
|
61
|
+
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_end=1452
|
62
|
+
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_start=1454
|
63
|
+
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_end=1574
|
64
|
+
_SHARELOGICALPIPELINEREQUEST._serialized_start=1577
|
65
|
+
_SHARELOGICALPIPELINEREQUEST._serialized_end=1735
|
66
|
+
_SHARELOGICALPIPELINERESPONSE._serialized_start=1737
|
67
|
+
_SHARELOGICALPIPELINERESPONSE._serialized_end=1767
|
68
|
+
_LOGICALPIPELINESERVICE._serialized_start=1770
|
69
|
+
_LOGICALPIPELINESERVICE._serialized_end=2683
|
63
70
|
# @@protoc_insertion_point(module_scope)
|
@@ -34,6 +34,11 @@ class LogicalPipelineServiceStub(object):
|
|
34
34
|
request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.SerializeToString,
|
35
35
|
response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.FromString,
|
36
36
|
)
|
37
|
+
self.ShareLogicalPipeline = channel.unary_unary(
|
38
|
+
'/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ShareLogicalPipeline',
|
39
|
+
request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineRequest.SerializeToString,
|
40
|
+
response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineResponse.FromString,
|
41
|
+
)
|
37
42
|
|
38
43
|
|
39
44
|
class LogicalPipelineServiceServicer(object):
|
@@ -63,6 +68,12 @@ class LogicalPipelineServiceServicer(object):
|
|
63
68
|
context.set_details('Method not implemented!')
|
64
69
|
raise NotImplementedError('Method not implemented!')
|
65
70
|
|
71
|
+
def ShareLogicalPipeline(self, request, context):
|
72
|
+
"""Missing associated documentation comment in .proto file."""
|
73
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
74
|
+
context.set_details('Method not implemented!')
|
75
|
+
raise NotImplementedError('Method not implemented!')
|
76
|
+
|
66
77
|
|
67
78
|
def add_LogicalPipelineServiceServicer_to_server(servicer, server):
|
68
79
|
rpc_method_handlers = {
|
@@ -86,6 +97,11 @@ def add_LogicalPipelineServiceServicer_to_server(servicer, server):
|
|
86
97
|
request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.FromString,
|
87
98
|
response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.SerializeToString,
|
88
99
|
),
|
100
|
+
'ShareLogicalPipeline': grpc.unary_unary_rpc_method_handler(
|
101
|
+
servicer.ShareLogicalPipeline,
|
102
|
+
request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineRequest.FromString,
|
103
|
+
response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineResponse.SerializeToString,
|
104
|
+
),
|
89
105
|
}
|
90
106
|
generic_handler = grpc.method_handlers_generic_handler(
|
91
107
|
'gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService', rpc_method_handlers)
|
@@ -163,3 +179,20 @@ class LogicalPipelineService(object):
|
|
163
179
|
src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.FromString,
|
164
180
|
options, channel_credentials,
|
165
181
|
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
182
|
+
|
183
|
+
@staticmethod
|
184
|
+
def ShareLogicalPipeline(request,
|
185
|
+
target,
|
186
|
+
options=(),
|
187
|
+
channel_credentials=None,
|
188
|
+
call_credentials=None,
|
189
|
+
insecure=False,
|
190
|
+
compression=None,
|
191
|
+
wait_for_ready=None,
|
192
|
+
timeout=None,
|
193
|
+
metadata=None):
|
194
|
+
return grpc.experimental.unary_unary(request, target, '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ShareLogicalPipeline',
|
195
|
+
src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineRequest.SerializeToString,
|
196
|
+
src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineResponse.FromString,
|
197
|
+
options, channel_credentials,
|
198
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
@@ -16,7 +16,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
|
|
16
16
|
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
17
17
|
|
18
18
|
|
19
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'src/controlplane/model/mpb/v1/mpb.proto\x12\"gml.internal.controlplane.model.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'src/controlplane/model/mpb/v1/mpb.proto\x12\"gml.internal.controlplane.model.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\x9e\x01\n\x0fGetModelRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x33\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\x18\x01\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x19\n\x08org_name\x18\x04 \x01(\tR\x07orgName\"\x7f\n\x10GetModelResponse\x12\x42\n\nmodel_info\x18\x01 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\tmodelInfo\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x9f\x01\n\x12\x43reateModelRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x42\n\nmodel_info\x18\x03 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\tmodelInfo\">\n\x13\x43reateModelResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x94\x01\n\x11ShareModelRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x1a\n\x08relation\x18\x02 \x01(\tR\x08relation\x12\x1b\n\tall_users\x18\x03 \x01(\x08R\x08\x61llUsers\x12\x1d\n\nuser_email\x18\x04 \x01(\tR\tuserEmail\"\x14\n\x12ShareModelResponse2\x82\x03\n\x0cModelService\x12u\n\x08GetModel\x12\x33.gml.internal.controlplane.model.v1.GetModelRequest\x1a\x34.gml.internal.controlplane.model.v1.GetModelResponse\x12~\n\x0b\x43reateModel\x12\x36.gml.internal.controlplane.model.v1.CreateModelRequest\x1a\x37.gml.internal.controlplane.model.v1.CreateModelResponse\x12{\n\nShareModel\x12\x35.gml.internal.controlplane.model.v1.ShareModelRequest\x1a\x36.gml.internal.controlplane.model.v1.ShareModelResponseB8Z6gimletlabs.ai/gimlet/src/controlplane/model/mpb/v1;mpbb\x06proto3')
|
20
20
|
|
21
21
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
22
22
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.model.mpb.v1.mpb_pb2', globals())
|
@@ -27,21 +27,27 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
27
27
|
_GETMODELREQUEST.fields_by_name['id']._options = None
|
28
28
|
_GETMODELREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
29
29
|
_GETMODELREQUEST.fields_by_name['org_id']._options = None
|
30
|
-
_GETMODELREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
30
|
+
_GETMODELREQUEST.fields_by_name['org_id']._serialized_options = b'\030\001\342\336\037\005OrgID'
|
31
31
|
_GETMODELRESPONSE.fields_by_name['id']._options = None
|
32
32
|
_GETMODELRESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
33
33
|
_CREATEMODELREQUEST.fields_by_name['org_id']._options = None
|
34
34
|
_CREATEMODELREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
35
35
|
_CREATEMODELRESPONSE.fields_by_name['id']._options = None
|
36
36
|
_CREATEMODELRESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
37
|
+
_SHAREMODELREQUEST.fields_by_name['id']._options = None
|
38
|
+
_SHAREMODELREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
37
39
|
_GETMODELREQUEST._serialized_start=169
|
38
|
-
_GETMODELREQUEST._serialized_end=
|
39
|
-
_GETMODELRESPONSE._serialized_start=
|
40
|
-
_GETMODELRESPONSE._serialized_end=
|
41
|
-
_CREATEMODELREQUEST._serialized_start=
|
42
|
-
_CREATEMODELREQUEST._serialized_end=
|
43
|
-
_CREATEMODELRESPONSE._serialized_start=
|
44
|
-
_CREATEMODELRESPONSE._serialized_end=
|
45
|
-
|
46
|
-
|
40
|
+
_GETMODELREQUEST._serialized_end=327
|
41
|
+
_GETMODELRESPONSE._serialized_start=329
|
42
|
+
_GETMODELRESPONSE._serialized_end=456
|
43
|
+
_CREATEMODELREQUEST._serialized_start=459
|
44
|
+
_CREATEMODELREQUEST._serialized_end=618
|
45
|
+
_CREATEMODELRESPONSE._serialized_start=620
|
46
|
+
_CREATEMODELRESPONSE._serialized_end=682
|
47
|
+
_SHAREMODELREQUEST._serialized_start=685
|
48
|
+
_SHAREMODELREQUEST._serialized_end=833
|
49
|
+
_SHAREMODELRESPONSE._serialized_start=835
|
50
|
+
_SHAREMODELRESPONSE._serialized_end=855
|
51
|
+
_MODELSERVICE._serialized_start=858
|
52
|
+
_MODELSERVICE._serialized_end=1244
|
47
53
|
# @@protoc_insertion_point(module_scope)
|
@@ -24,6 +24,11 @@ class ModelServiceStub(object):
|
|
24
24
|
request_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelRequest.SerializeToString,
|
25
25
|
response_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelResponse.FromString,
|
26
26
|
)
|
27
|
+
self.ShareModel = channel.unary_unary(
|
28
|
+
'/gml.internal.controlplane.model.v1.ModelService/ShareModel',
|
29
|
+
request_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelRequest.SerializeToString,
|
30
|
+
response_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelResponse.FromString,
|
31
|
+
)
|
27
32
|
|
28
33
|
|
29
34
|
class ModelServiceServicer(object):
|
@@ -41,6 +46,12 @@ class ModelServiceServicer(object):
|
|
41
46
|
context.set_details('Method not implemented!')
|
42
47
|
raise NotImplementedError('Method not implemented!')
|
43
48
|
|
49
|
+
def ShareModel(self, request, context):
|
50
|
+
"""Missing associated documentation comment in .proto file."""
|
51
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
52
|
+
context.set_details('Method not implemented!')
|
53
|
+
raise NotImplementedError('Method not implemented!')
|
54
|
+
|
44
55
|
|
45
56
|
def add_ModelServiceServicer_to_server(servicer, server):
|
46
57
|
rpc_method_handlers = {
|
@@ -54,6 +65,11 @@ def add_ModelServiceServicer_to_server(servicer, server):
|
|
54
65
|
request_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelRequest.FromString,
|
55
66
|
response_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelResponse.SerializeToString,
|
56
67
|
),
|
68
|
+
'ShareModel': grpc.unary_unary_rpc_method_handler(
|
69
|
+
servicer.ShareModel,
|
70
|
+
request_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelRequest.FromString,
|
71
|
+
response_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelResponse.SerializeToString,
|
72
|
+
),
|
57
73
|
}
|
58
74
|
generic_handler = grpc.method_handlers_generic_handler(
|
59
75
|
'gml.internal.controlplane.model.v1.ModelService', rpc_method_handlers)
|
@@ -97,3 +113,20 @@ class ModelService(object):
|
|
97
113
|
src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelResponse.FromString,
|
98
114
|
options, channel_credentials,
|
99
115
|
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
116
|
+
|
117
|
+
@staticmethod
|
118
|
+
def ShareModel(request,
|
119
|
+
target,
|
120
|
+
options=(),
|
121
|
+
channel_credentials=None,
|
122
|
+
call_credentials=None,
|
123
|
+
insecure=False,
|
124
|
+
compression=None,
|
125
|
+
wait_for_ready=None,
|
126
|
+
timeout=None,
|
127
|
+
metadata=None):
|
128
|
+
return grpc.experimental.unary_unary(request, target, '/gml.internal.controlplane.model.v1.ModelService/ShareModel',
|
129
|
+
src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelRequest.SerializeToString,
|
130
|
+
src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelResponse.FromString,
|
131
|
+
options, channel_credentials,
|
132
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
gml/tensor.py
CHANGED
@@ -278,6 +278,13 @@ class EmbeddingDimension(DimensionSemantics):
|
|
278
278
|
)
|
279
279
|
|
280
280
|
|
281
|
+
class PositionIDsDimension(DimensionSemantics):
|
282
|
+
def to_proto(self) -> modelexecpb.DimensionSemantics:
|
283
|
+
return modelexecpb.DimensionSemantics(
|
284
|
+
kind=modelexecpb.DimensionSemantics.DIMENSION_SEMANTICS_KIND_POSITION_IDS,
|
285
|
+
)
|
286
|
+
|
287
|
+
|
281
288
|
class TensorSemantics:
|
282
289
|
def __init__(
|
283
290
|
self,
|
gml/version_utils.py
ADDED
@@ -0,0 +1,43 @@
|
|
1
|
+
# Copyright 2023- Gimlet Labs, Inc.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
#
|
15
|
+
# SPDX-License-Identifier: Apache-2.0
|
16
|
+
|
17
|
+
import functools
|
18
|
+
import importlib.metadata
|
19
|
+
|
20
|
+
from packaging import version
|
21
|
+
|
22
|
+
|
23
|
+
def _is_version_greater_or_equal(pkg: str, target_version: str) -> bool:
|
24
|
+
"""
|
25
|
+
Check if the installed package version is greater than or equal to the specified version.
|
26
|
+
Args:
|
27
|
+
pkg (str): The package name to check, e.g., 'torch', 'transformers'.
|
28
|
+
target_version (str): The version to compare against, in the format 'major.minor.patch'.
|
29
|
+
Returns:
|
30
|
+
bool: True if the installed package version is greater than or equal to `target_version`, False otherwise.
|
31
|
+
"""
|
32
|
+
pkg_version_w_dev = importlib.metadata.version(pkg)
|
33
|
+
# ignore dev suffixes.
|
34
|
+
pkg_version = version.parse(version.parse(pkg_version_w_dev).base_version)
|
35
|
+
return pkg_version >= version.parse(target_version)
|
36
|
+
|
37
|
+
|
38
|
+
is_torch_version_greater_or_equal = functools.partial(
|
39
|
+
_is_version_greater_or_equal, "torch"
|
40
|
+
)
|
41
|
+
is_transformers_version_greater_or_equal = functools.partial(
|
42
|
+
_is_version_greater_or_equal, "transformers"
|
43
|
+
)
|
File without changes
|