gimlet-api 0.0.10__py3-none-any.whl → 0.0.11__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.
Files changed (5) hide show
  1. {gimlet_api-0.0.10.dist-info → gimlet_api-0.0.11.dist-info}/METADATA +1 -1
  2. {gimlet_api-0.0.10.dist-info → gimlet_api-0.0.11.dist-info}/RECORD +5 -5
  3. gml/client.py +3 -1
  4. gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py +20 -20
  5. {gimlet_api-0.0.10.dist-info → gimlet_api-0.0.11.dist-info}/WHEEL +0 -0
{gimlet_api-0.0.10.dist-info → gimlet_api-0.0.11.dist-info}/METADATA RENAMED
@@ -16,6 +16,6 @@ Requires-Dist: rich
16
16
  Requires-Dist: transformers>=4.43.3
17
17
  Requires-Dist: tokenizers>=0.21.0
18
18
  Requires-Dist: safetensors-mlir
19
- Version: 0.0.10
19
+ Version: 0.0.11
20
20
 
21
21
  UNKNOWN
{gimlet_api-0.0.10.dist-info → gimlet_api-0.0.11.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  gml/__init__.py,sha256=H3WQZ_RaN7VNeb__qeHEbKLEwkaG7gpL5FQ8s1IotUA,773
2
2
  gml/_utils.py,sha256=mSCWHhCdzcUvHqmJIB2FS215K1LMgJCWcZ6e6FWK3hQ,1184
3
3
  gml/asset_manager.py,sha256=VnbqUZHPOgPrAh6ri9C0EuNhS8tAHIrbUyJPAJuD9po,2053
4
- gml/client.py,sha256=AcnG5mniHOfq-He-uCph2-xQ39cZwmXZePaUEed87b8,14378
4
+ gml/client.py,sha256=oLc-r8GxsR0iYss4dU_qKmSvEOTYetgF4wYuYCzCzRM,14446
5
5
  gml/compile.py,sha256=3L5fpD8DK45RLiywj1b5NuDlbsxpzRxI87k1GahlMpc,9851
6
6
  gml/device.py,sha256=Iw71NnuLcgjY32ZMXHlnlPkosTuHEmL9E98utmNChlM,2650
7
7
  gml/hf.py,sha256=Kv2yffy8omTRQDPnoIZocG2EOyfhr7UvLFIvTmRxw0g,36170
@@ -42,12 +42,12 @@ gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2.py,sha256=KgoU
42
42
  gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2_grpc.py,sha256=p3OpT8-hfNHu4-29qr-ZahRwO-LoCYM9Q4jomAHTXGA,24572
43
43
  gml/proto/src/controlplane/filetransfer/ftpb/v1/ftpb_pb2.py,sha256=r8mbJNTq45_c0amPnTr8OFZasCk7XWu2YS_eu7GfWJg,7050
44
44
  gml/proto/src/controlplane/filetransfer/ftpb/v1/ftpb_pb2_grpc.py,sha256=XlE4R2PJaOmzQocx7y6SKJvuqt8tYBGzBuhajvzG0cc,12919
45
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py,sha256=2s2p6dURKJLboaR965m2-rGTo_63Bi1cXsA90Hz9u-M,6632
45
+ gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py,sha256=aUHPoOzIGbOfElsSKDepdTQC-qwJ4ecJrzLUl7h1Xk0,6770
46
46
  gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py,sha256=-snjW7n6JveUzJVPFcm25XlL19kowPSKgd61l_jPnHA,9541
47
47
  gml/proto/src/controlplane/model/mpb/v1/mpb_pb2.py,sha256=RVedXkNYu2iF5OHiXoYyRw9AGRCUWG7qNyY-5QY71Go,3762
48
48
  gml/proto/src/controlplane/model/mpb/v1/mpb_pb2_grpc.py,sha256=KSdb6V04qUHDsb1R2o3wixwTyZgrhwnPYobjnRgWX4I,4735
49
49
  gml/register_submodules.py,sha256=U8IwjVygX2vxNi_aK6ljHOD4mmrOhbyVczvy4wwulqU,5027
50
50
  gml/tensor.py,sha256=aPLm3I3qkYNDcJmntaUycqqN5rsZmcj8ql0EkupJudY,14977
51
- gimlet_api-0.0.10.dist-info/WHEEL,sha256=sobxWSyDDkdg_rinUth-jxhXHqoNqlmNMJY3aTZn2Us,91
52
- gimlet_api-0.0.10.dist-info/METADATA,sha256=i3n2dnjznNFL6XFsj1bL0T544E0FmMVQySLgiBkUW04,586
53
- gimlet_api-0.0.10.dist-info/RECORD,,
51
+ gimlet_api-0.0.11.dist-info/WHEEL,sha256=sobxWSyDDkdg_rinUth-jxhXHqoNqlmNMJY3aTZn2Us,91
52
+ gimlet_api-0.0.11.dist-info/METADATA,sha256=gCI6KBvTBDa6g7YB7oEAQA64M9b1eXNyFT3E0uP0y_Q,586
53
+ gimlet_api-0.0.11.dist-info/RECORD,,
gml/client.py CHANGED
@@ -319,6 +319,7 @@ class Client:
319
319
  models: List[Model],
320
320
  pipeline_file: Optional[Path] = None,
321
321
  pipeline: Optional[Union[str, Pipeline]] = None,
322
+ description: Optional[str] = None,
322
323
  ) -> uuidpb.UUID:
323
324
  if pipeline_file is not None:
324
325
  with open(pipeline_file, "r") as f:
@@ -343,12 +344,13 @@ class Client:
343
344
  org_id=self._org_id(),
344
345
  name=name,
345
346
  yaml=yaml,
347
+ description=description,
346
348
  )
347
349
  resp: lppb.CreateLogicalPipelineResponse = stub.CreateLogicalPipeline(
348
350
  req, metadata=self._get_request_metadata(idempotent=True)
349
351
  )
350
352
 
351
- url = f"https://{os.getenv('GML_CONTROLPLANE_ADDR')}/orgs/{quote(self._org_name)}/pipelines/{quote(name)}"
353
+ url = f"https://{self._controlplane_addr}/orgs/{quote(self._org_name)}/pipeline/{quote(name)}"
352
354
  console.print(
353
355
  f"[green]Pipeline upload complete![/green]\nView your pipeline at: [cyan]{url}[/cyan]"
354
356
  )
gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py CHANGED
@@ -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'\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\"\x9a\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\"P\n\x1bListLogicalPipelinesRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x7f\n\x1cListLogicalPipelinesResponse\x12_\n\tpipelines\x18\x01 \x03(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\tpipelines\"\x8b\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\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\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\"\x7f\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\x04yamlJ\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\x0flogicalPipeline2\xe1\x05\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.ParseLogicalPipelineYAMLResponseBDZBgimletlabs.ai/gimlet/src/controlplane/logicalpipeline/lppb/v1;lppbb\x06proto3')
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\"\xbc\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\"P\n\x1bListLogicalPipelinesRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x7f\n\x1cListLogicalPipelinesResponse\x12_\n\tpipelines\x18\x01 \x03(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\tpipelines\"\x8b\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\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\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\x0flogicalPipeline2\xe1\x05\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.ParseLogicalPipelineYAMLResponseBDZBgimletlabs.ai/gimlet/src/controlplane/logicalpipeline/lppb/v1;lppbb\x06proto3')
20
20
 
21
21
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
22
22
  _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.logicalpipeline.lppb.v1.lppb_pb2', globals())
@@ -41,23 +41,23 @@ if _descriptor._USE_C_DESCRIPTORS == False:
41
41
  _CREATELOGICALPIPELINERESPONSE.fields_by_name['id']._options = None
42
42
  _CREATELOGICALPIPELINERESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
43
43
  _LOGICALPIPELINEINFO._serialized_start=191
44
- _LOGICALPIPELINEINFO._serialized_end=473
45
- _LISTLOGICALPIPELINESREQUEST._serialized_start=475
46
- _LISTLOGICALPIPELINESREQUEST._serialized_end=555
47
- _LISTLOGICALPIPELINESRESPONSE._serialized_start=557
48
- _LISTLOGICALPIPELINESRESPONSE._serialized_end=684
49
- _GETLOGICALPIPELINEREQUEST._serialized_start=687
50
- _GETLOGICALPIPELINEREQUEST._serialized_end=826
51
- _GETLOGICALPIPELINERESPONSE._serialized_start=829
52
- _GETLOGICALPIPELINERESPONSE._serialized_end=985
53
- _CREATELOGICALPIPELINEREQUEST._serialized_start=987
54
- _CREATELOGICALPIPELINEREQUEST._serialized_end=1114
55
- _CREATELOGICALPIPELINERESPONSE._serialized_start=1116
56
- _CREATELOGICALPIPELINERESPONSE._serialized_end=1188
57
- _PARSELOGICALPIPELINEYAMLREQUEST._serialized_start=1190
58
- _PARSELOGICALPIPELINEYAMLREQUEST._serialized_end=1243
59
- _PARSELOGICALPIPELINEYAMLRESPONSE._serialized_start=1245
60
- _PARSELOGICALPIPELINEYAMLRESPONSE._serialized_end=1365
61
- _LOGICALPIPELINESERVICE._serialized_start=1368
62
- _LOGICALPIPELINESERVICE._serialized_end=2105
44
+ _LOGICALPIPELINEINFO._serialized_end=507
45
+ _LISTLOGICALPIPELINESREQUEST._serialized_start=509
46
+ _LISTLOGICALPIPELINESREQUEST._serialized_end=589
47
+ _LISTLOGICALPIPELINESRESPONSE._serialized_start=591
48
+ _LISTLOGICALPIPELINESRESPONSE._serialized_end=718
49
+ _GETLOGICALPIPELINEREQUEST._serialized_start=721
50
+ _GETLOGICALPIPELINEREQUEST._serialized_end=860
51
+ _GETLOGICALPIPELINERESPONSE._serialized_start=863
52
+ _GETLOGICALPIPELINERESPONSE._serialized_end=1019
53
+ _CREATELOGICALPIPELINEREQUEST._serialized_start=1022
54
+ _CREATELOGICALPIPELINEREQUEST._serialized_end=1183
55
+ _CREATELOGICALPIPELINERESPONSE._serialized_start=1185
56
+ _CREATELOGICALPIPELINERESPONSE._serialized_end=1257
57
+ _PARSELOGICALPIPELINEYAMLREQUEST._serialized_start=1259
58
+ _PARSELOGICALPIPELINEYAMLREQUEST._serialized_end=1312
59
+ _PARSELOGICALPIPELINEYAMLRESPONSE._serialized_start=1314
60
+ _PARSELOGICALPIPELINEYAMLRESPONSE._serialized_end=1434
61
+ _LOGICALPIPELINESERVICE._serialized_start=1437
62
+ _LOGICALPIPELINESERVICE._serialized_end=2174
63
63
  # @@protoc_insertion_point(module_scope)