gimlet-api 0.0.14__py3-none-any.whl → 0.0.15__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.14.dist-info → gimlet_api-0.0.15.dist-info}/METADATA +1 -1
- {gimlet_api-0.0.14.dist-info → gimlet_api-0.0.15.dist-info}/RECORD +14 -13
- gml/hf.py +1 -0
- gml/pipelines.py +86 -20
- gml/proto/src/api/corepb/v1/actor_net_pb2.py +77 -0
- gml/proto/src/api/corepb/v1/compiled_pipeline_pb2.py +28 -28
- gml/proto/src/api/corepb/v1/cp_dp_pb2.py +32 -28
- gml/proto/src/api/corepb/v1/cp_edge_pb2.py +70 -66
- gml/proto/src/api/corepb/v1/mediastream_pb2.py +11 -9
- gml/proto/src/api/corepb/v1/model_exec_pb2.py +160 -153
- gml/proto/src/common/typespb/jwt_pb2.py +14 -10
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py +17 -11
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py +33 -0
- {gimlet_api-0.0.14.dist-info → gimlet_api-0.0.15.dist-info}/WHEEL +0 -0
@@ -4,10 +4,10 @@ gml/asset_manager.py,sha256=VnbqUZHPOgPrAh6ri9C0EuNhS8tAHIrbUyJPAJuD9po,2053
|
|
4
4
|
gml/client.py,sha256=3rqVTMSv7QUlSrMqSQDzXuXDvfm3WTF3G8enK8on9zU,15861
|
5
5
|
gml/compile.py,sha256=1zM0ihwbrptZ4FphyvtvfKJyVFuVVfyBnDnnEeJ12fg,10397
|
6
6
|
gml/device.py,sha256=9Z7dsBfpvTHShd1OWSi1Pvn85EFYDmn1dszWV8YHIJI,2648
|
7
|
-
gml/hf.py,sha256=
|
7
|
+
gml/hf.py,sha256=4tU2c3Th_mc__78Odetg2g3b16eZM4oSevUtMr27H_k,37811
|
8
8
|
gml/model.py,sha256=8fIYlLRduTsUZfYJr_YVPNxbEVIzr7_yaaTe4T-TZ2Y,8429
|
9
9
|
gml/model_utils.py,sha256=Kw08MIPmwIOocoQXfjlqjn78mVerCQ2uzleT0H_zcck,1821
|
10
|
-
gml/pipelines.py,sha256=
|
10
|
+
gml/pipelines.py,sha256=Nif9FNqXqjYM6iL3RZzkX-KBsZamSu7xh8HwjNqsB5A,10220
|
11
11
|
gml/preprocessing.py,sha256=YPcxwBOdx0h0ADzoloYbFw9qUGFbi167E8HA4Zwn7Pk,3928
|
12
12
|
gml/proto/gogoproto/gogo_pb2.py,sha256=WVMIAR8K--mCUkTPM7mEeeXGpQlRRtt_kco10iP3CZs,15728
|
13
13
|
gml/proto/mediapipe/framework/calculator_contract_test_pb2.py,sha256=hNjyZCBz3RYa6rN4xR3FOCZKA24gq_LsJ3EMegl5wK4,2031
|
@@ -25,17 +25,18 @@ gml/proto/mediapipe/framework/thread_pool_executor_pb2.py,sha256=9TJ66fqSo1BiJmE
|
|
25
25
|
gml/proto/opentelemetry/proto/common/v1/common_pb2.py,sha256=2l9c_xGfUvShLFkzofChHmbgpa7I0-u-FJ_J2Wv3lvs,3168
|
26
26
|
gml/proto/opentelemetry/proto/metrics/v1/metrics_pb2.py,sha256=k8oW5tmFlJK2574Ky6kDc0JmNNQCLroRwCCGyxDd7JA,9968
|
27
27
|
gml/proto/opentelemetry/proto/resource/v1/resource_pb2.py,sha256=08f2F5overFnGlyNyZHb5rUyv7-G9pC15c4xDCccIPY,1831
|
28
|
-
gml/proto/src/api/corepb/v1/
|
28
|
+
gml/proto/src/api/corepb/v1/actor_net_pb2.py,sha256=XV3UZbyxvHcXRG-kFdA5LImaTAnSVRw2scw8Cz5Mn6Q,11009
|
29
|
+
gml/proto/src/api/corepb/v1/compiled_pipeline_pb2.py,sha256=K2xhqSAxQ3w_3VFC43XpeqXm9_EDmVz6Fb8PrA_jxOY,7609
|
29
30
|
gml/proto/src/api/corepb/v1/controlplane_pb2.py,sha256=n6LOgoQCAJZ2LJivb9DffmzLpYY7e9K1EDT8pQsIbno,14705
|
30
|
-
gml/proto/src/api/corepb/v1/cp_dp_pb2.py,sha256=
|
31
|
-
gml/proto/src/api/corepb/v1/cp_edge_pb2.py,sha256=
|
31
|
+
gml/proto/src/api/corepb/v1/cp_dp_pb2.py,sha256=cHJkgfehbgnohipFXMtoHSzR5mGfVZ9oKwOVFjEMRc4,10358
|
32
|
+
gml/proto/src/api/corepb/v1/cp_edge_pb2.py,sha256=Ruw7_GcoElKzrhEoOMYkUezJDvrjwA29cVMtBhfV8I8,23294
|
32
33
|
gml/proto/src/api/corepb/v1/dataplane_pb2.py,sha256=D3nA4c8624Irh1cIWM8rbvUBUmr29CJ0lQlPko2BgMU,1966
|
33
34
|
gml/proto/src/api/corepb/v1/deployed_pipeline_pb2.py,sha256=XbppBI1fQ-FazD2in1o6Z9_BIPRBArCE5dVUF7iUn3Y,6649
|
34
35
|
gml/proto/src/api/corepb/v1/device_info_pb2.py,sha256=lXFF04AkL_Y3Tcg9mXAvgo-w3lmAMiuHQZT5yLpyO4s,7029
|
35
36
|
gml/proto/src/api/corepb/v1/gem_config_pb2.py,sha256=vC0g3k9hDv-LhiV6LwaYCly6x00Xx_YA0i2AZSwCo_I,5396
|
36
|
-
gml/proto/src/api/corepb/v1/mediastream_pb2.py,sha256=
|
37
|
-
gml/proto/src/api/corepb/v1/model_exec_pb2.py,sha256=
|
38
|
-
gml/proto/src/common/typespb/jwt_pb2.py,sha256=
|
37
|
+
gml/proto/src/api/corepb/v1/mediastream_pb2.py,sha256=fAB7s7w4soBtaWJXwni5OI--lWapnLM1LeqZzIBWnlo,10359
|
38
|
+
gml/proto/src/api/corepb/v1/model_exec_pb2.py,sha256=eh6-tQPq5GEhsuRM6IgJRc2PvKhMlQADu_Lj7FZN5O8,37754
|
39
|
+
gml/proto/src/common/typespb/jwt_pb2.py,sha256=JxBZr8JU1mBoo1PKPClXr3SdfjZynRYRlQ-JHZRjqhE,6134
|
39
40
|
gml/proto/src/common/typespb/status_pb2.py,sha256=IbBJnbsAlvsuTtyT285ZuW6k5VaPfl5kRSOnBxD_H8M,2109
|
40
41
|
gml/proto/src/common/typespb/uuid_pb2.py,sha256=5Fm3jYpCPX7sMrP6RhRYsF0SnuZNIBEQJk9f0jwZ2Rw,1188
|
41
42
|
gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py,sha256=MIizns1dezQjocpJjeNJ4Z7BFWNweKrRpJ070L9IaCk,5203
|
@@ -44,13 +45,13 @@ gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2.py,sha256=S3Oz
|
|
44
45
|
gml/proto/src/controlplane/directory/directorypb/v1/directory_pb2_grpc.py,sha256=p3OpT8-hfNHu4-29qr-ZahRwO-LoCYM9Q4jomAHTXGA,24572
|
45
46
|
gml/proto/src/controlplane/filetransfer/ftpb/v1/ftpb_pb2.py,sha256=r8mbJNTq45_c0amPnTr8OFZasCk7XWu2YS_eu7GfWJg,7050
|
46
47
|
gml/proto/src/controlplane/filetransfer/ftpb/v1/ftpb_pb2_grpc.py,sha256=XlE4R2PJaOmzQocx7y6SKJvuqt8tYBGzBuhajvzG0cc,12919
|
47
|
-
gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py,sha256=
|
48
|
-
gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py,sha256=
|
48
|
+
gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py,sha256=CaRhKtLOcz9AhIw9Rxws0ALqmFAktqkQyRlgmoA6OF0,8976
|
49
|
+
gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py,sha256=NA_Ud55lnbJfgOw729j97MqfKuylCopiimt6pryNJoU,13740
|
49
50
|
gml/proto/src/controlplane/model/mpb/v1/mpb_pb2.py,sha256=IryUZ-TlpQKvU52-XFRKvuAfiH-0EkXrTzwvmmK7Fmk,4591
|
50
51
|
gml/proto/src/controlplane/model/mpb/v1/mpb_pb2_grpc.py,sha256=ZABewZxEthfQx2pEfvBfLc4M8JoKazheQOH181CegjY,6586
|
51
52
|
gml/register_submodules.py,sha256=U8IwjVygX2vxNi_aK6ljHOD4mmrOhbyVczvy4wwulqU,5027
|
52
53
|
gml/tensor.py,sha256=ojRlfMEf5wsLyOHuAVl0wuZlcaqO0KF4EDYdtEju6hk,15229
|
53
54
|
gml/version_utils.py,sha256=ouCemolnoDm71NiQRcfpa5k5bETTLaFCH6lrEyivGNY,1626
|
54
|
-
gimlet_api-0.0.
|
55
|
-
gimlet_api-0.0.
|
56
|
-
gimlet_api-0.0.
|
55
|
+
gimlet_api-0.0.15.dist-info/WHEEL,sha256=sobxWSyDDkdg_rinUth-jxhXHqoNqlmNMJY3aTZn2Us,91
|
56
|
+
gimlet_api-0.0.15.dist-info/METADATA,sha256=E-gLFjOe2M6B_KiEyg41boQKRbEw7ej_FwV-pJaop1k,611
|
57
|
+
gimlet_api-0.0.15.dist-info/RECORD,,
|
gml/hf.py
CHANGED
gml/pipelines.py
CHANGED
@@ -100,12 +100,26 @@ nodes:
|
|
100
100
|
outputs:
|
101
101
|
- detections
|
102
102
|
{track_node}
|
103
|
-
- name:
|
104
|
-
kind:
|
103
|
+
- name: scale_video
|
104
|
+
kind: ScaleVideo
|
105
105
|
attributes:
|
106
|
-
|
106
|
+
target_width: 640
|
107
|
+
target_height: 480
|
108
|
+
preserve_aspect_ratio: true
|
107
109
|
inputs:
|
108
110
|
frame: .camera_source.frame
|
111
|
+
outputs:
|
112
|
+
- scaled_frame
|
113
|
+
- name: video_stream_sink
|
114
|
+
kind: EndpointSink
|
115
|
+
attributes:
|
116
|
+
response_output_info:
|
117
|
+
- name: frame
|
118
|
+
type: SEMANTIC_TYPE_VIDEO
|
119
|
+
- name: detections
|
120
|
+
type: SEMANTIC_TYPE_DETECTIONS
|
121
|
+
inputs:
|
122
|
+
frame: .scale_video.scaled_frame
|
109
123
|
detections: {video_stream_detections}
|
110
124
|
"""
|
111
125
|
|
@@ -134,12 +148,26 @@ nodes:
|
|
134
148
|
frame: .camera_source.frame
|
135
149
|
outputs:
|
136
150
|
- segmentation
|
137
|
-
- name:
|
138
|
-
kind:
|
151
|
+
- name: scale_video
|
152
|
+
kind: ScaleVideo
|
139
153
|
attributes:
|
140
|
-
|
154
|
+
target_width: 640
|
155
|
+
target_height: 480
|
156
|
+
preserve_aspect_ratio: true
|
141
157
|
inputs:
|
142
158
|
frame: .camera_source.frame
|
159
|
+
outputs:
|
160
|
+
- scaled_frame
|
161
|
+
- name: video_stream_sink
|
162
|
+
kind: EndpointSink
|
163
|
+
attributes:
|
164
|
+
response_output_info:
|
165
|
+
- name: frame
|
166
|
+
type: SEMANTIC_TYPE_VIDEO
|
167
|
+
- name: segmentation
|
168
|
+
type: SEMANTIC_TYPE_SEGMENTATION
|
169
|
+
inputs:
|
170
|
+
frame: .scale_video.scaled_frame
|
143
171
|
segmentation: .segment.segmentation
|
144
172
|
"""
|
145
173
|
|
@@ -165,12 +193,24 @@ nodes:
|
|
165
193
|
frame: .camera_source.frame
|
166
194
|
outputs:
|
167
195
|
- depth
|
168
|
-
- name:
|
169
|
-
kind:
|
196
|
+
- name: scale_video
|
197
|
+
kind: ScaleVideo
|
170
198
|
attributes:
|
171
|
-
|
199
|
+
target_width: 640
|
200
|
+
target_height: 480
|
201
|
+
preserve_aspect_ratio: true
|
172
202
|
inputs:
|
173
203
|
frame: .estimate_depth.depth
|
204
|
+
outputs:
|
205
|
+
- scaled_frame
|
206
|
+
- name: video_stream_sink
|
207
|
+
kind: EndpointSink
|
208
|
+
attributes:
|
209
|
+
response_output_info:
|
210
|
+
- name: frame
|
211
|
+
type: SEMANTIC_TYPE_VIDEO
|
212
|
+
inputs:
|
213
|
+
frame: .scale_video.scaled_frame
|
174
214
|
"""
|
175
215
|
|
176
216
|
|
@@ -217,8 +257,12 @@ class LiveChatPipeline(Pipeline):
|
|
217
257
|
)
|
218
258
|
return f"""---
|
219
259
|
nodes:
|
220
|
-
- name:
|
221
|
-
kind:
|
260
|
+
- name: endpoint_source
|
261
|
+
kind: EndpointSource
|
262
|
+
attributes:
|
263
|
+
request_input_info:
|
264
|
+
- name: prompt
|
265
|
+
type: SEMANTIC_TYPE_TEXT
|
222
266
|
outputs:
|
223
267
|
- prompt
|
224
268
|
- name: query_template
|
@@ -228,7 +272,7 @@ nodes:
|
|
228
272
|
message_template: {escape_prompt(message_template)}
|
229
273
|
preset_system_prompt: {escape_prompt(self.system_prompt)}
|
230
274
|
inputs:
|
231
|
-
query: .
|
275
|
+
query: .endpoint_source.prompt
|
232
276
|
outputs:
|
233
277
|
- chat_message
|
234
278
|
- name: tokenize
|
@@ -264,8 +308,12 @@ nodes:
|
|
264
308
|
tokens: .generate.generated_tokens
|
265
309
|
outputs:
|
266
310
|
- text
|
267
|
-
- name:
|
268
|
-
kind:
|
311
|
+
- name: endpoint_sink
|
312
|
+
kind: EndpointSink
|
313
|
+
attributes:
|
314
|
+
response_output_info:
|
315
|
+
- name: text_batch
|
316
|
+
type: SEMANTIC_TYPE_TEXT
|
269
317
|
inputs:
|
270
318
|
text_batch: .detokenize.text
|
271
319
|
"""
|
@@ -297,8 +345,12 @@ nodes:
|
|
297
345
|
kind: CameraSource
|
298
346
|
outputs:
|
299
347
|
- frame
|
300
|
-
- name:
|
301
|
-
kind:
|
348
|
+
- name: endpoint_source
|
349
|
+
kind: EndpointSource
|
350
|
+
attributes:
|
351
|
+
request_input_info:
|
352
|
+
- name: prompt
|
353
|
+
type: SEMANTIC_TYPE_TEXT
|
302
354
|
outputs:
|
303
355
|
- prompt
|
304
356
|
- name: detect
|
@@ -315,15 +367,29 @@ nodes:
|
|
315
367
|
conf_threshold: {self.conf_threshold}
|
316
368
|
inputs:
|
317
369
|
frame: .camera_source.frame
|
318
|
-
prompt: .
|
370
|
+
prompt: .endpoint_source.prompt
|
319
371
|
outputs:
|
320
372
|
- detections
|
321
|
-
- name:
|
322
|
-
kind:
|
373
|
+
- name: scale_video
|
374
|
+
kind: ScaleVideo
|
323
375
|
attributes:
|
324
|
-
|
376
|
+
target_width: 640
|
377
|
+
target_height: 480
|
378
|
+
preserve_aspect_ratio: true
|
325
379
|
inputs:
|
326
380
|
frame: .camera_source.frame
|
381
|
+
outputs:
|
382
|
+
- scaled_frame
|
383
|
+
- name: video_stream_sink
|
384
|
+
kind: EndpointSink
|
385
|
+
attributes:
|
386
|
+
response_output_info:
|
387
|
+
- name: frame
|
388
|
+
type: SEMANTIC_TYPE_VIDEO
|
389
|
+
- name: detections
|
390
|
+
type: SEMANTIC_TYPE_DETECTIONS
|
391
|
+
inputs:
|
392
|
+
frame: .scale_video.scaled_frame
|
327
393
|
detections: .detect.detections
|
328
394
|
"""
|
329
395
|
|
@@ -0,0 +1,77 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: src/api/corepb/v1/actor_net.proto
|
4
|
+
"""Generated protocol buffer code."""
|
5
|
+
from google.protobuf.internal import builder as _builder
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
# @@protoc_insertion_point(imports)
|
10
|
+
|
11
|
+
_sym_db = _symbol_database.Default()
|
12
|
+
|
13
|
+
|
14
|
+
from gml.proto.mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
|
15
|
+
try:
|
16
|
+
mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
|
17
|
+
except AttributeError:
|
18
|
+
mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
|
19
|
+
|
20
|
+
|
21
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!src/api/corepb/v1/actor_net.proto\x12\x18gml.internal.api.core.v1\x1a$mediapipe/framework/calculator.proto\"S\n\x0c\x41\x63torTaskRef\x12\x1d\n\nactor_name\x18\x01 \x01(\tR\tactorName\x12$\n\x0etask_atom_name\x18\x02 \x01(\tR\x0ctaskAtomName\")\n\x08\x41\x63torRef\x12\x1d\n\nactor_name\x18\x01 \x01(\tR\tactorName\"\x94\x06\n\rMPActorConfig\x12\x43\n\x0cgraph_config\x18\x01 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigR\x0bgraphConfig\x12Z\n\x12input_message_type\x18\x02 \x01(\x0e\x32,.gml.internal.api.core.v1.MPActorMessageTypeR\x10inputMessageType\x12k\n\x12input_stream_types\x18\x03 \x03(\x0b\x32=.gml.internal.api.core.v1.MPActorConfig.InputStreamTypesEntryR\x10inputStreamTypes\x12\\\n\x13output_message_type\x18\x04 \x01(\x0e\x32,.gml.internal.api.core.v1.MPActorMessageTypeR\x11outputMessageType\x12n\n\x13output_stream_types\x18\x05 \x03(\x0b\x32>.gml.internal.api.core.v1.MPActorConfig.OutputStreamTypesEntryR\x11outputStreamTypes\x12\x42\n\x08\x63onsumer\x18\x06 \x01(\x0b\x32&.gml.internal.api.core.v1.ActorTaskRefR\x08\x63onsumer\x1ap\n\x15InputStreamTypesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x41\n\x05value\x18\x02 \x01(\x0e\x32+.gml.internal.api.core.v1.MPActorPacketTypeR\x05value:\x02\x38\x01\x1aq\n\x16OutputStreamTypesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x41\n\x05value\x18\x02 \x01(\x0e\x32+.gml.internal.api.core.v1.MPActorPacketTypeR\x05value:\x02\x38\x01\"r\n\x19PagedKVCoordinatorOptions\x12&\n\x0ftokens_per_page\x18\x01 \x01(\x03R\rtokensPerPage\x12-\n\x13max_pages_in_memory\x18\x02 \x01(\x03R\x10maxPagesInMemory\"\xb1\x01\n\x14KVCoordinatorOptions\x12?\n\x04kind\x18\x01 \x01(\x0e\x32+.gml.internal.api.core.v1.KVCoordinatorKindR\x04kind\x12X\n\rpaged_options\x18\x02 \x01(\x0b\x32\x33.gml.internal.api.core.v1.PagedKVCoordinatorOptionsR\x0cpagedOptions\"\xfa\x02\n\x13LLMSchedulerOptions\x12\x45\n\x06policy\x18\x01 \x01(\x0e\x32-.gml.internal.api.core.v1.LLMSchedulingPolicyR\x06policy\x12$\n\x0emax_batch_size\x18\x02 \x01(\x03R\x0cmaxBatchSize\x12/\n\x14max_tokens_per_batch\x18\x03 \x01(\x03R\x11maxTokensPerBatch\x12\x32\n\x15max_inflight_requests\x18\x04 \x01(\x03R\x13maxInflightRequests\x12\x30\n\x14max_inflight_batches\x18\x05 \x01(\x03R\x12maxInflightBatches\x12\x1d\n\neos_tokens\x18\x06 \x03(\x05R\teosTokens\x12@\n\x1d\x64\x65\x66\x61ult_max_tokens_before_eos\x18\x07 \x01(\x03R\x19\x64\x65\x66\x61ultMaxTokensBeforeEos\"\xaf\x04\n\x17LLMSchedulerActorConfig\x12\x64\n\x16kv_coordinator_options\x18\x01 \x01(\x0b\x32..gml.internal.api.core.v1.KVCoordinatorOptionsR\x14kvCoordinatorOptions\x12Z\n\x11scheduler_options\x18\x02 \x01(\x0b\x32-.gml.internal.api.core.v1.LLMSchedulerOptionsR\x10schedulerOptions\x12`\n\x18generated_token_consumer\x18\x03 \x01(\x0b\x32&.gml.internal.api.core.v1.ActorTaskRefR\x16generatedTokenConsumer\x12W\n\x13inference_consumers\x18\x04 \x03(\x0b\x32&.gml.internal.api.core.v1.ActorTaskRefR\x12inferenceConsumers\x12G\n\rdecode_actors\x18\x05 \x03(\x0b\x32\".gml.internal.api.core.v1.ActorRefR\x0c\x64\x65\x63odeActors\x12N\n\x11kv_transfer_actor\x18\x06 \x01(\x0b\x32\".gml.internal.api.core.v1.ActorRefR\x0fkvTransferActor\"_\n\x11RemoteActorConfig\x12)\n\x10remote_namespace\x18\x01 \x01(\tR\x0fremoteNamespace\x12\x1f\n\x0bremote_name\x18\x02 \x01(\tR\nremoteName\"\x8a\x01\n\x12PagedKVCacheParams\x12\x1d\n\nnum_blocks\x18\x01 \x01(\x03R\tnumBlocks\x12(\n\x10tokens_per_block\x18\x02 \x01(\x03R\x0etokensPerBlock\x12+\n\x12max_blocks_per_seq\x18\x03 \x01(\x03R\x0fmaxBlocksPerSeq\"\xc7\x02\n\rKVCacheParams\x12\x1d\n\nnum_layers\x18\x01 \x01(\x03R\tnumLayers\x12\x1b\n\tnum_heads\x18\x02 \x01(\x03R\x08numHeads\x12\x19\n\x08head_dim\x18\x03 \x01(\x03R\x07headDim\x12O\n\x0cpaged_params\x18\x04 \x01(\x0b\x32,.gml.internal.api.core.v1.PagedKVCacheParamsR\x0bpagedParams\x12?\n\x06\x66ormat\x18\x05 \x01(\x0e\x32\'.gml.internal.api.core.v1.KVCacheFormatR\x06\x66ormat\x12,\n\x12max_context_length\x18\x06 \x01(\x03R\x10maxContextLength\x12\x1f\n\x0bplugin_name\x18\x07 \x01(\tR\npluginName\"\x8a\x02\n\x12KVCacheActorConfig\x12O\n\x0fkv_cache_params\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.KVCacheParamsR\rkvCacheParams\x12I\n\x0ckv_consumers\x18\x02 \x03(\x0b\x32&.gml.internal.api.core.v1.ActorTaskRefR\x0bkvConsumers\x12X\n\x16transfer_manager_actor\x18\x03 \x01(\x0b\x32\".gml.internal.api.core.v1.ActorRefR\x14transferManagerActor\"n\n\x1cKVTransferManagerActorConfig\x12N\n\x11kv_manager_actors\x18\x01 \x03(\x0b\x32\".gml.internal.api.core.v1.ActorRefR\x0fkvManagerActors\"\xc5\x04\n\x0f\x41\x63torDefinition\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x37\n\x04kind\x18\x02 \x01(\x0e\x32#.gml.internal.api.core.v1.ActorKindR\x04kind\x12\x46\n\tmp_config\x18\x03 \x01(\x0b\x32\'.gml.internal.api.core.v1.MPActorConfigH\x00R\x08mpConfig\x12\x65\n\x14llm_scheduler_config\x18\x04 \x01(\x0b\x32\x31.gml.internal.api.core.v1.LLMSchedulerActorConfigH\x00R\x12llmSchedulerConfig\x12R\n\rremote_config\x18\x05 \x01(\x0b\x32+.gml.internal.api.core.v1.RemoteActorConfigH\x00R\x0cremoteConfig\x12\x61\n\x15kv_cache_actor_config\x18\x06 \x01(\x0b\x32,.gml.internal.api.core.v1.KVCacheActorConfigH\x00R\x12kvCacheActorConfig\x12u\n\x1akv_transfer_manager_config\x18\x07 \x01(\x0b\x32\x36.gml.internal.api.core.v1.KVTransferManagerActorConfigH\x00R\x17kvTransferManagerConfigB\x08\n\x06\x63onfig\"n\n\x15\x41\x63torNetworkNamespace\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x06\x61\x63tors\x18\x02 \x03(\x0b\x32).gml.internal.api.core.v1.ActorDefinitionR\x06\x61\x63tors*\xb4\x01\n\tActorKind\x12\x16\n\x12\x41\x43TOR_KIND_UNKNOWN\x10\x00\x12\x17\n\x13\x41\x43TOR_KIND_MP_ACTOR\x10\x01\x12\x1c\n\x18\x41\x43TOR_KIND_LLM_SCHEDULER\x10\x02\x12\x1b\n\x17\x41\x43TOR_KIND_REMOTE_ACTOR\x10\x03\x12\x17\n\x13\x41\x43TOR_KIND_KV_CACHE\x10\x04\x12\"\n\x1e\x41\x43TOR_KIND_KV_TRANSFER_MANAGER\x10\x05*\x95\x02\n\x12MPActorMessageType\x12!\n\x1dMP_ACTOR_MESSAGE_TYPE_UNKNOWN\x10\x00\x12\x1e\n\x1aMP_ACTOR_MESSAGE_TYPE_NONE\x10\x01\x12\x30\n,MP_ACTOR_MESSAGE_TYPE_LLM_GENERATION_REQUEST\x10\x02\x12-\n)MP_ACTOR_MESSAGE_TYPE_LLM_SCHEDULED_BATCH\x10\x03\x12+\n\'MP_ACTOR_MESSAGE_TYPE_LLM_BATCH_RESULTS\x10\x04\x12.\n*MP_ACTOR_MESSAGE_TYPE_LLM_GENERATED_TOKENS\x10\x05*\xc6\x02\n\x11MPActorPacketType\x12 \n\x1cMP_ACTOR_PACKET_TYPE_UNKNOWN\x10\x00\x12#\n\x1fMP_ACTOR_PACKET_TYPE_REQUEST_ID\x10\x01\x12#\n\x1fMP_ACTOR_PACKET_TYPE_SESSION_ID\x10\x02\x12\x1f\n\x1bMP_ACTOR_PACKET_TYPE_TOKENS\x10\x03\x12%\n!MP_ACTOR_PACKET_TYPE_TOKENS_BATCH\x10\x04\x12-\n)MP_ACTOR_PACKET_TYPE_KV_ALLOC_INFOS_BATCH\x10\x05\x12\x30\n,MP_ACTOR_PACKET_TYPE_LLM_REQUEST_TYPES_BATCH\x10\x06\x12\x1c\n\x18MP_ACTOR_PACKET_TYPE_EOS\x10\x07*{\n\x11KVCoordinatorKind\x12\x1f\n\x1bKV_COORDINATOR_KIND_UNKNOWN\x10\x00\x12&\n\"KV_COORDINATOR_KIND_SINGLE_REQUEST\x10\x01\x12\x1d\n\x19KV_COORDINATOR_KIND_PAGED\x10\x02*\xad\x01\n\x13LLMSchedulingPolicy\x12!\n\x1dLLM_SCHEDULING_POLICY_UNKNOWN\x10\x00\x12!\n\x1dLLM_SCHEDULING_POLICY_DEFAULT\x10\x01\x12+\n\'LLM_SCHEDULING_POLICY_INFLIGHT_BATCHING\x10\x02\x12#\n\x1fLLM_SCHEDULING_POLICY_SPLITWISE\x10\x03*\x88\x02\n\rKVCacheFormat\x12\x1b\n\x17KV_CACHE_FORMAT_UNKNOWN\x10\x00\x12:\n6KV_CACHE_FORMAT_CONTIGUOUS_SEPARATE_KV_NO_OUTPUT_SPACE\x10\x01\x12<\n8KV_CACHE_FORMAT_CONTIGUOUS_COMBINED_KV_WITH_OUTPUT_SPACE\x10\x02\x12\x1e\n\x1aKV_CACHE_FORMAT_PAGED_TLLM\x10\x03\x12 \n\x1cKV_CACHE_FORMAT_PREALLOCATED\x10\x04\x12\x1e\n\x1aKV_CACHE_FORMAT_PAGED_VLLM\x10\x05\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
22
|
+
|
23
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
24
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.actor_net_pb2', globals())
|
25
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
26
|
+
|
27
|
+
DESCRIPTOR._options = None
|
28
|
+
DESCRIPTOR._serialized_options = b'Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepb'
|
29
|
+
_MPACTORCONFIG_INPUTSTREAMTYPESENTRY._options = None
|
30
|
+
_MPACTORCONFIG_INPUTSTREAMTYPESENTRY._serialized_options = b'8\001'
|
31
|
+
_MPACTORCONFIG_OUTPUTSTREAMTYPESENTRY._options = None
|
32
|
+
_MPACTORCONFIG_OUTPUTSTREAMTYPESENTRY._serialized_options = b'8\001'
|
33
|
+
_ACTORKIND._serialized_start=3905
|
34
|
+
_ACTORKIND._serialized_end=4085
|
35
|
+
_MPACTORMESSAGETYPE._serialized_start=4088
|
36
|
+
_MPACTORMESSAGETYPE._serialized_end=4365
|
37
|
+
_MPACTORPACKETTYPE._serialized_start=4368
|
38
|
+
_MPACTORPACKETTYPE._serialized_end=4694
|
39
|
+
_KVCOORDINATORKIND._serialized_start=4696
|
40
|
+
_KVCOORDINATORKIND._serialized_end=4819
|
41
|
+
_LLMSCHEDULINGPOLICY._serialized_start=4822
|
42
|
+
_LLMSCHEDULINGPOLICY._serialized_end=4995
|
43
|
+
_KVCACHEFORMAT._serialized_start=4998
|
44
|
+
_KVCACHEFORMAT._serialized_end=5262
|
45
|
+
_ACTORTASKREF._serialized_start=101
|
46
|
+
_ACTORTASKREF._serialized_end=184
|
47
|
+
_ACTORREF._serialized_start=186
|
48
|
+
_ACTORREF._serialized_end=227
|
49
|
+
_MPACTORCONFIG._serialized_start=230
|
50
|
+
_MPACTORCONFIG._serialized_end=1018
|
51
|
+
_MPACTORCONFIG_INPUTSTREAMTYPESENTRY._serialized_start=791
|
52
|
+
_MPACTORCONFIG_INPUTSTREAMTYPESENTRY._serialized_end=903
|
53
|
+
_MPACTORCONFIG_OUTPUTSTREAMTYPESENTRY._serialized_start=905
|
54
|
+
_MPACTORCONFIG_OUTPUTSTREAMTYPESENTRY._serialized_end=1018
|
55
|
+
_PAGEDKVCOORDINATOROPTIONS._serialized_start=1020
|
56
|
+
_PAGEDKVCOORDINATOROPTIONS._serialized_end=1134
|
57
|
+
_KVCOORDINATOROPTIONS._serialized_start=1137
|
58
|
+
_KVCOORDINATOROPTIONS._serialized_end=1314
|
59
|
+
_LLMSCHEDULEROPTIONS._serialized_start=1317
|
60
|
+
_LLMSCHEDULEROPTIONS._serialized_end=1695
|
61
|
+
_LLMSCHEDULERACTORCONFIG._serialized_start=1698
|
62
|
+
_LLMSCHEDULERACTORCONFIG._serialized_end=2257
|
63
|
+
_REMOTEACTORCONFIG._serialized_start=2259
|
64
|
+
_REMOTEACTORCONFIG._serialized_end=2354
|
65
|
+
_PAGEDKVCACHEPARAMS._serialized_start=2357
|
66
|
+
_PAGEDKVCACHEPARAMS._serialized_end=2495
|
67
|
+
_KVCACHEPARAMS._serialized_start=2498
|
68
|
+
_KVCACHEPARAMS._serialized_end=2825
|
69
|
+
_KVCACHEACTORCONFIG._serialized_start=2828
|
70
|
+
_KVCACHEACTORCONFIG._serialized_end=3094
|
71
|
+
_KVTRANSFERMANAGERACTORCONFIG._serialized_start=3096
|
72
|
+
_KVTRANSFERMANAGERACTORCONFIG._serialized_end=3206
|
73
|
+
_ACTORDEFINITION._serialized_start=3209
|
74
|
+
_ACTORDEFINITION._serialized_end=3790
|
75
|
+
_ACTORNETWORKNAMESPACE._serialized_start=3792
|
76
|
+
_ACTORNETWORKNAMESPACE._serialized_end=3902
|
77
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -15,7 +15,7 @@ from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
|
|
15
15
|
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
16
16
|
|
17
17
|
|
18
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)src/api/corepb/v1/compiled_pipeline.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\x8c\x01\n\x14\x43ompiledPipelineSpec\x12t\n\x17\x64istribution_candidates\x18\x01 \x03(\x0b\x32;.gml.internal.api.core.v1.PipelineDistributionCandidateSpecR\x16\x64istributionCandidates\"\xe4\x01\n!PipelineDistributionCandidateSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\\\n\x0f\x66ragment_groups\x18\x02 \x03(\x0b\x32\x33.gml.internal.api.core.v1.PipelineFragmentGroupSpecR\x0e\x66ragmentGroups\x12M\n\x05ports\x18\x03 \x03(\x0b\x32\x37.gml.internal.api.core.v1.PipelineFragmentGroupPortSpecR\x05ports\"\xc7\x01\n\x19PipelineFragmentGroupSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12H\n\x0b\x63omm_groups\x18\x02 \x03(\x0b\x32\'.gml.internal.api.core.v1.CommGroupSpecR\ncommGroups\x12L\n\tfragments\x18\x03 \x03(\x0b\x32..gml.internal.api.core.v1.PipelineFragmentSpecR\tfragments\"\x83\x01\n\x1dPipelineFragmentGroupPortSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12!\n\x0csource_group\x18\x02 \x01(\tR\x0bsourceGroup\x12+\n\x11\x64\x65stination_group\x18\x03 \x01(\tR\x10\x64\x65stinationGroup\"\
|
18
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)src/api/corepb/v1/compiled_pipeline.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\x8c\x01\n\x14\x43ompiledPipelineSpec\x12t\n\x17\x64istribution_candidates\x18\x01 \x03(\x0b\x32;.gml.internal.api.core.v1.PipelineDistributionCandidateSpecR\x16\x64istributionCandidates\"\xe4\x01\n!PipelineDistributionCandidateSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\\\n\x0f\x66ragment_groups\x18\x02 \x03(\x0b\x32\x33.gml.internal.api.core.v1.PipelineFragmentGroupSpecR\x0e\x66ragmentGroups\x12M\n\x05ports\x18\x03 \x03(\x0b\x32\x37.gml.internal.api.core.v1.PipelineFragmentGroupPortSpecR\x05ports\"\xc7\x01\n\x19PipelineFragmentGroupSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12H\n\x0b\x63omm_groups\x18\x02 \x03(\x0b\x32\'.gml.internal.api.core.v1.CommGroupSpecR\ncommGroups\x12L\n\tfragments\x18\x03 \x03(\x0b\x32..gml.internal.api.core.v1.PipelineFragmentSpecR\tfragments\"\x83\x01\n\x1dPipelineFragmentGroupPortSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12!\n\x0csource_group\x18\x02 \x01(\tR\x0bsourceGroup\x12+\n\x11\x64\x65stination_group\x18\x03 \x01(\tR\x10\x64\x65stinationGroup\"\xe5\x01\n\x14PipelineFragmentSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12i\n\x0e\x63onfigurations\x18\x02 \x03(\x0b\x32\x41.gml.internal.api.core.v1.PipelineFragmentConfigurationOptionSpecR\x0e\x63onfigurations\x12N\n\x0b\x63omm_groups\x18\x03 \x03(\x0b\x32-.gml.internal.api.core.v1.CommGroupMemberSpecR\ncommGroups\"\xd5\x04\n\'PipelineFragmentConfigurationOptionSpec\x12\x44\n\texec_spec\x18\x02 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x08\x65xecSpec\x12?\n\x06labels\x18\x03 \x03(\x0b\x32\'.gml.internal.api.core.v1.FragmentLabelR\x06labels\x12_\n\x08gem_spec\x18\x04 \x01(\x0b\x32\x37.gml.internal.api.core.v1.ScopedAffinityAndResourceSpecB\x0b\xe2\xde\x1f\x07GEMSpecR\x07gemSpec\x12\x64\n\x11\x61\x63\x63\x65lerator_specs\x18\x05 \x03(\x0b\x32\x37.gml.internal.api.core.v1.ScopedAffinityAndResourceSpecR\x10\x61\x63\x63\x65leratorSpecs\x12S\n\x11\x66ragment_affinity\x18\x06 \x01(\x0b\x32&.gml.internal.api.core.v1.AffinitySpecR\x10\x66ragmentAffinity\x12\\\n\x16\x66ragment_anti_affinity\x18\x07 \x01(\x0b\x32&.gml.internal.api.core.v1.AffinitySpecR\x14\x66ragmentAntiAffinity\x12#\n\rfragment_name\x18\x08 \x01(\tR\x0c\x66ragmentNameJ\x04\x08\x01\x10\x02\"A\n\x0fResourceRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x1a\n\x08quantity\x18\x02 \x01(\x03R\x08quantity\"a\n\x18ResourceRequirementsSpec\x12\x45\n\x08requests\x18\x01 \x03(\x0b\x32).gml.internal.api.core.v1.ResourceRequestR\x08requests\"\xc0\x01\n\x1dScopedAffinityAndResourceSpec\x12M\n\x0escope_affinity\x18\x01 \x01(\x0b\x32&.gml.internal.api.core.v1.AffinitySpecR\rscopeAffinity\x12P\n\tresources\x18\x04 \x01(\x0b\x32\x32.gml.internal.api.core.v1.ResourceRequirementsSpecR\tresources\"\x8a\x01\n\x14LabelMatchExpression\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x16\n\x06values\x18\x02 \x03(\tR\x06values\x12H\n\x08operator\x18\x03 \x01(\x0e\x32,.gml.internal.api.core.v1.LabelMatchOperatorR\x08operator\"\x8f\x01\n\rLabelSelector\x12[\n\x11match_expressions\x18\x01 \x03(\x0b\x32..gml.internal.api.core.v1.LabelMatchExpressionR\x10matchExpressions\x12!\n\x0ctopology_key\x18\x02 \x01(\tR\x0btopologyKey\"U\n\x0c\x41\x66\x66inityRule\x12\x45\n\tselectors\x18\x01 \x03(\x0b\x32\'.gml.internal.api.core.v1.LabelSelectorR\tselectors\"]\n\x0c\x41\x66\x66initySpec\x12M\n\x0erequired_rules\x18\x02 \x03(\x0b\x32&.gml.internal.api.core.v1.AffinityRuleR\rrequiredRules\"7\n\rFragmentLabel\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value\"\x87\x01\n\rCommGroupSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04size\x18\x02 \x01(\x03R\x04size\x12N\n\x0ctopology_key\x18\x03 \x01(\x0e\x32+.gml.internal.api.core.v1.FabricTopologyKeyR\x0btopologyKey\"=\n\x13\x43ommGroupMemberSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04rank\x18\x02 \x01(\x03R\x04rank*\xbe\x01\n\x12LabelMatchOperator\x12 \n\x1cLABEL_MATCH_OPERATOR_UNKNOWN\x10\x00\x12\x1b\n\x17LABEL_MATCH_OPERATOR_IN\x10\x01\x12\x1f\n\x1bLABEL_MATCH_OPERATOR_NOT_IN\x10\x02\x12\x1f\n\x1bLABEL_MATCH_OPERATOR_EXISTS\x10\x03\x12\'\n#LABEL_MATCH_OPERATOR_DOES_NOT_EXIST\x10\x04*\xa9\x01\n\x11\x46\x61\x62ricTopologyKey\x12\x1f\n\x1b\x46\x41\x42RIC_TOPOLOGY_KEY_UNKNOWN\x10\x00\x12\x1b\n\x17\x46\x41\x42RIC_TOPOLOGY_KEY_ANY\x10\x01\x12\x1c\n\x18\x46\x41\x42RIC_TOPOLOGY_KEY_NODE\x10\x02\x12\x1c\n\x18\x46\x41\x42RIC_TOPOLOGY_KEY_RACK\x10\x03\x12\x1a\n\x16\x46\x41\x42RIC_TOPOLOGY_KEY_DC\x10\x04\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
19
19
|
|
20
20
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
21
21
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.compiled_pipeline_pb2', globals())
|
@@ -25,10 +25,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
25
25
|
DESCRIPTOR._serialized_options = b'Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepb'
|
26
26
|
_PIPELINEFRAGMENTCONFIGURATIONOPTIONSPEC.fields_by_name['gem_spec']._options = None
|
27
27
|
_PIPELINEFRAGMENTCONFIGURATIONOPTIONSPEC.fields_by_name['gem_spec']._serialized_options = b'\342\336\037\007GEMSpec'
|
28
|
-
_LABELMATCHOPERATOR._serialized_start=
|
29
|
-
_LABELMATCHOPERATOR._serialized_end=
|
30
|
-
_FABRICTOPOLOGYKEY._serialized_start=
|
31
|
-
_FABRICTOPOLOGYKEY._serialized_end=
|
28
|
+
_LABELMATCHOPERATOR._serialized_start=2760
|
29
|
+
_LABELMATCHOPERATOR._serialized_end=2950
|
30
|
+
_FABRICTOPOLOGYKEY._serialized_start=2953
|
31
|
+
_FABRICTOPOLOGYKEY._serialized_end=3122
|
32
32
|
_COMPILEDPIPELINESPEC._serialized_start=130
|
33
33
|
_COMPILEDPIPELINESPEC._serialized_end=270
|
34
34
|
_PIPELINEDISTRIBUTIONCANDIDATESPEC._serialized_start=273
|
@@ -38,27 +38,27 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
38
38
|
_PIPELINEFRAGMENTGROUPPORTSPEC._serialized_start=706
|
39
39
|
_PIPELINEFRAGMENTGROUPPORTSPEC._serialized_end=837
|
40
40
|
_PIPELINEFRAGMENTSPEC._serialized_start=840
|
41
|
-
_PIPELINEFRAGMENTSPEC._serialized_end=
|
42
|
-
_PIPELINEFRAGMENTCONFIGURATIONOPTIONSPEC._serialized_start=
|
43
|
-
_PIPELINEFRAGMENTCONFIGURATIONOPTIONSPEC._serialized_end=
|
44
|
-
_RESOURCEREQUEST._serialized_start=
|
45
|
-
_RESOURCEREQUEST._serialized_end=
|
46
|
-
_RESOURCEREQUIREMENTSSPEC._serialized_start=
|
47
|
-
_RESOURCEREQUIREMENTSSPEC._serialized_end=
|
48
|
-
_SCOPEDAFFINITYANDRESOURCESPEC._serialized_start=
|
49
|
-
_SCOPEDAFFINITYANDRESOURCESPEC._serialized_end=
|
50
|
-
_LABELMATCHEXPRESSION._serialized_start=
|
51
|
-
_LABELMATCHEXPRESSION._serialized_end=
|
52
|
-
_LABELSELECTOR._serialized_start=
|
53
|
-
_LABELSELECTOR._serialized_end=
|
54
|
-
_AFFINITYRULE._serialized_start=
|
55
|
-
_AFFINITYRULE._serialized_end=
|
56
|
-
_AFFINITYSPEC._serialized_start=
|
57
|
-
_AFFINITYSPEC._serialized_end=
|
58
|
-
_FRAGMENTLABEL._serialized_start=
|
59
|
-
_FRAGMENTLABEL._serialized_end=
|
60
|
-
_COMMGROUPSPEC._serialized_start=
|
61
|
-
_COMMGROUPSPEC._serialized_end=
|
62
|
-
_COMMGROUPMEMBERSPEC._serialized_start=
|
63
|
-
_COMMGROUPMEMBERSPEC._serialized_end=
|
41
|
+
_PIPELINEFRAGMENTSPEC._serialized_end=1069
|
42
|
+
_PIPELINEFRAGMENTCONFIGURATIONOPTIONSPEC._serialized_start=1072
|
43
|
+
_PIPELINEFRAGMENTCONFIGURATIONOPTIONSPEC._serialized_end=1669
|
44
|
+
_RESOURCEREQUEST._serialized_start=1671
|
45
|
+
_RESOURCEREQUEST._serialized_end=1736
|
46
|
+
_RESOURCEREQUIREMENTSSPEC._serialized_start=1738
|
47
|
+
_RESOURCEREQUIREMENTSSPEC._serialized_end=1835
|
48
|
+
_SCOPEDAFFINITYANDRESOURCESPEC._serialized_start=1838
|
49
|
+
_SCOPEDAFFINITYANDRESOURCESPEC._serialized_end=2030
|
50
|
+
_LABELMATCHEXPRESSION._serialized_start=2033
|
51
|
+
_LABELMATCHEXPRESSION._serialized_end=2171
|
52
|
+
_LABELSELECTOR._serialized_start=2174
|
53
|
+
_LABELSELECTOR._serialized_end=2317
|
54
|
+
_AFFINITYRULE._serialized_start=2319
|
55
|
+
_AFFINITYRULE._serialized_end=2404
|
56
|
+
_AFFINITYSPEC._serialized_start=2406
|
57
|
+
_AFFINITYSPEC._serialized_end=2499
|
58
|
+
_FRAGMENTLABEL._serialized_start=2501
|
59
|
+
_FRAGMENTLABEL._serialized_end=2556
|
60
|
+
_COMMGROUPSPEC._serialized_start=2559
|
61
|
+
_COMMGROUPSPEC._serialized_end=2694
|
62
|
+
_COMMGROUPMEMBERSPEC._serialized_start=2696
|
63
|
+
_COMMGROUPMEMBERSPEC._serialized_end=2757
|
64
64
|
# @@protoc_insertion_point(module_scope)
|
@@ -19,7 +19,7 @@ from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb
|
|
19
19
|
from gml.proto.src.api.corepb.v1 import compiled_pipeline_pb2 as src_dot_api_dot_corepb_dot_v1_dot_compiled__pipeline__pb2
|
20
20
|
|
21
21
|
|
22
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dsrc/api/corepb/v1/cp_dp.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x19google/protobuf/any.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fsrc/common/typespb/status.proto\x1a\"src/api/corepb/v1/model_exec.proto\x1a)src/api/corepb/v1/compiled_pipeline.proto\"\xe2\x01\n\x19PipelineDeploymentRequest\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12\x64\n\x16\x63ompiled_pipeline_spec\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.CompiledPipelineSpecR\x14\x63ompiledPipelineSpec\"\
|
22
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dsrc/api/corepb/v1/cp_dp.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x19google/protobuf/any.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fsrc/common/typespb/status.proto\x1a\"src/api/corepb/v1/model_exec.proto\x1a)src/api/corepb/v1/compiled_pipeline.proto\"\xe2\x01\n\x19PipelineDeploymentRequest\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12\x64\n\x16\x63ompiled_pipeline_spec\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.CompiledPipelineSpecR\x14\x63ompiledPipelineSpec\"\xb1\x04\n\x1cPipelineDeploymentSpecUpdate\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12=\n\x05state\x18\x02 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12V\n\x13logical_pipeline_id\x18\x04 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12T\n\x10logical_pipeline\x18\x05 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12;\n\x06models\x18\x06 \x03(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\x06models\x12@\n\x0b\x65ndpoint_id\x18\x07 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0e\xe2\xde\x1f\nEndpointIDR\nendpointId\x12#\n\rendpoint_name\x18\x08 \x01(\tR\x0c\x65ndpointName\x12\x19\n\x08org_name\x18\t \x01(\tR\x07orgNameJ\x04\x08\x03\x10\x04\"\xaa\x01\n\x14WorkloadStatusUpdate\x12\x46\n\rdeployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x10\xe2\xde\x1f\x0c\x44\x65ploymentIDR\x0c\x64\x65ploymentId\x12J\n\x06status\x18\x02 \x01(\x0b\x32\x32.gml.internal.api.core.v1.PipelineDeploymentStatusR\x06status\"\x91\x03\n\x1cPipelineFragmentStatusUpdate\x12@\n\x0b\x66ragment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0e\xe2\xde\x1f\nFragmentIDR\nfragmentId\x12\x46\n\rdeployment_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x10\xe2\xde\x1f\x0c\x44\x65ploymentIDR\x0c\x64\x65ploymentId\x12:\n\tdevice_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12=\n\x05state\x18\x04 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x16\n\x06reason\x18\x05 \x01(\tR\x06reason\x12T\n\x18\x61ssigned_accelerator_ids\x18\x06 \x03(\tB\x1a\xe2\xde\x1f\x16\x41ssignedAcceleratorIDsR\x16\x61ssignedAcceleratorIds\"\x14\n\x12\x44\x61taplaneHeartbeat\"G\n\x0c\x44PCPMetadata\x12\x37\n\x08\x66leet_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\"z\n\x0b\x44PCPMessage\x12\x42\n\x08metadata\x18\x01 \x01(\x0b\x32&.gml.internal.api.core.v1.DPCPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"G\n\x0c\x43PDPMetadata\x12\x37\n\x08\x66leet_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\"z\n\x0b\x43PDPMessage\x12\x42\n\x08metadata\x18\x01 \x01(\x0b\x32&.gml.internal.api.core.v1.CPDPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"\xd0\x01\n\x0cStartSession\x12@\n\x0b\x65ndpoint_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0e\xe2\xde\x1f\nEndpointIDR\nendpointId\x12#\n\rendpoint_name\x18\x02 \x01(\tR\x0c\x65ndpointName\x12\x1d\n\nauth_token\x18\x03 \x01(\tR\tauthToken\x12:\n\tdevice_id\x18\x04 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\r\n\x0bStopSession\"D\n\x0e\x46orwardMessage\x12\x32\n\x0cjson_message\x18\x01 \x01(\tB\x0f\xe2\xde\x1f\x0bJSONMessageR\x0bjsonMessage\"\xdf\x02\n\x14\x45ndpointProxyRequest\x12M\n\x10proxy_session_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x12\xe2\xde\x1f\x0eProxySessionIDR\x0eproxySessionId\x12M\n\rstart_session\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.StartSessionH\x00R\x0cstartSession\x12J\n\x0cstop_session\x18\x03 \x01(\x0b\x32%.gml.internal.api.core.v1.StopSessionH\x00R\x0bstopSession\x12S\n\x0f\x66orward_message\x18\x04 \x01(\x0b\x32(.gml.internal.api.core.v1.ForwardMessageH\x00R\x0e\x66orwardMessageB\x08\n\x06\x61\x63tion\"\x0b\n\tHeartbeat\"\xb9\x02\n\x15\x45ndpointProxyResponse\x12M\n\x10proxy_session_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x12\xe2\xde\x1f\x0eProxySessionIDR\x0eproxySessionId\x12S\n\x0f\x66orward_message\x18\x02 \x01(\x0b\x32(.gml.internal.api.core.v1.ForwardMessageH\x00R\x0e\x66orwardMessage\x12+\n\x06status\x18\x03 \x01(\x0b\x32\x11.gml.types.StatusH\x00R\x06status\x12\x43\n\theartbeat\x18\x04 \x01(\x0b\x32#.gml.internal.api.core.v1.HeartbeatH\x00R\theartbeatB\n\n\x08responseB/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
|
23
23
|
|
24
24
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
25
25
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.cp_dp_pb2', globals())
|
@@ -43,12 +43,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
43
43
|
_PIPELINEFRAGMENTSTATUSUPDATE.fields_by_name['deployment_id']._serialized_options = b'\342\336\037\014DeploymentID'
|
44
44
|
_PIPELINEFRAGMENTSTATUSUPDATE.fields_by_name['device_id']._options = None
|
45
45
|
_PIPELINEFRAGMENTSTATUSUPDATE.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
|
46
|
+
_PIPELINEFRAGMENTSTATUSUPDATE.fields_by_name['assigned_accelerator_ids']._options = None
|
47
|
+
_PIPELINEFRAGMENTSTATUSUPDATE.fields_by_name['assigned_accelerator_ids']._serialized_options = b'\342\336\037\026AssignedAcceleratorIDs'
|
46
48
|
_DPCPMETADATA.fields_by_name['fleet_id']._options = None
|
47
49
|
_DPCPMETADATA.fields_by_name['fleet_id']._serialized_options = b'\342\336\037\007FleetID'
|
48
50
|
_CPDPMETADATA.fields_by_name['fleet_id']._options = None
|
49
51
|
_CPDPMETADATA.fields_by_name['fleet_id']._serialized_options = b'\342\336\037\007FleetID'
|
50
52
|
_STARTSESSION.fields_by_name['endpoint_id']._options = None
|
51
53
|
_STARTSESSION.fields_by_name['endpoint_id']._serialized_options = b'\342\336\037\nEndpointID'
|
54
|
+
_STARTSESSION.fields_by_name['device_id']._options = None
|
55
|
+
_STARTSESSION.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
|
52
56
|
_FORWARDMESSAGE.fields_by_name['json_message']._options = None
|
53
57
|
_FORWARDMESSAGE.fields_by_name['json_message']._serialized_options = b'\342\336\037\013JSONMessage'
|
54
58
|
_ENDPOINTPROXYREQUEST.fields_by_name['proxy_session_id']._options = None
|
@@ -58,31 +62,31 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
58
62
|
_PIPELINEDEPLOYMENTREQUEST._serialized_start=252
|
59
63
|
_PIPELINEDEPLOYMENTREQUEST._serialized_end=478
|
60
64
|
_PIPELINEDEPLOYMENTSPECUPDATE._serialized_start=481
|
61
|
-
_PIPELINEDEPLOYMENTSPECUPDATE._serialized_end=
|
62
|
-
_WORKLOADSTATUSUPDATE._serialized_start=
|
63
|
-
_WORKLOADSTATUSUPDATE._serialized_end=
|
64
|
-
_PIPELINEFRAGMENTSTATUSUPDATE._serialized_start=
|
65
|
-
_PIPELINEFRAGMENTSTATUSUPDATE._serialized_end=
|
66
|
-
_DATAPLANEHEARTBEAT._serialized_start=
|
67
|
-
_DATAPLANEHEARTBEAT._serialized_end=
|
68
|
-
_DPCPMETADATA._serialized_start=
|
69
|
-
_DPCPMETADATA._serialized_end=
|
70
|
-
_DPCPMESSAGE._serialized_start=
|
71
|
-
_DPCPMESSAGE._serialized_end=
|
72
|
-
_CPDPMETADATA._serialized_start=
|
73
|
-
_CPDPMETADATA._serialized_end=
|
74
|
-
_CPDPMESSAGE._serialized_start=
|
75
|
-
_CPDPMESSAGE._serialized_end=
|
76
|
-
_STARTSESSION._serialized_start=
|
77
|
-
_STARTSESSION._serialized_end=
|
78
|
-
_STOPSESSION._serialized_start=
|
79
|
-
_STOPSESSION._serialized_end=
|
80
|
-
_FORWARDMESSAGE._serialized_start=
|
81
|
-
_FORWARDMESSAGE._serialized_end=
|
82
|
-
_ENDPOINTPROXYREQUEST._serialized_start=
|
83
|
-
_ENDPOINTPROXYREQUEST._serialized_end=
|
84
|
-
_HEARTBEAT._serialized_start=
|
85
|
-
_HEARTBEAT._serialized_end=
|
86
|
-
_ENDPOINTPROXYRESPONSE._serialized_start=
|
87
|
-
_ENDPOINTPROXYRESPONSE._serialized_end=
|
65
|
+
_PIPELINEDEPLOYMENTSPECUPDATE._serialized_end=1042
|
66
|
+
_WORKLOADSTATUSUPDATE._serialized_start=1045
|
67
|
+
_WORKLOADSTATUSUPDATE._serialized_end=1215
|
68
|
+
_PIPELINEFRAGMENTSTATUSUPDATE._serialized_start=1218
|
69
|
+
_PIPELINEFRAGMENTSTATUSUPDATE._serialized_end=1619
|
70
|
+
_DATAPLANEHEARTBEAT._serialized_start=1621
|
71
|
+
_DATAPLANEHEARTBEAT._serialized_end=1641
|
72
|
+
_DPCPMETADATA._serialized_start=1643
|
73
|
+
_DPCPMETADATA._serialized_end=1714
|
74
|
+
_DPCPMESSAGE._serialized_start=1716
|
75
|
+
_DPCPMESSAGE._serialized_end=1838
|
76
|
+
_CPDPMETADATA._serialized_start=1840
|
77
|
+
_CPDPMETADATA._serialized_end=1911
|
78
|
+
_CPDPMESSAGE._serialized_start=1913
|
79
|
+
_CPDPMESSAGE._serialized_end=2035
|
80
|
+
_STARTSESSION._serialized_start=2038
|
81
|
+
_STARTSESSION._serialized_end=2246
|
82
|
+
_STOPSESSION._serialized_start=2248
|
83
|
+
_STOPSESSION._serialized_end=2261
|
84
|
+
_FORWARDMESSAGE._serialized_start=2263
|
85
|
+
_FORWARDMESSAGE._serialized_end=2331
|
86
|
+
_ENDPOINTPROXYREQUEST._serialized_start=2334
|
87
|
+
_ENDPOINTPROXYREQUEST._serialized_end=2685
|
88
|
+
_HEARTBEAT._serialized_start=2687
|
89
|
+
_HEARTBEAT._serialized_end=2698
|
90
|
+
_ENDPOINTPROXYRESPONSE._serialized_start=2701
|
91
|
+
_ENDPOINTPROXYRESPONSE._serialized_end=3014
|
88
92
|
# @@protoc_insertion_point(module_scope)
|