corvic-engine 0.3.0rc52__cp38-abi3-win_amd64.whl → 0.3.0rc54__cp38-abi3-win_amd64.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.
- corvic/engine/_native.pyd +0 -0
- corvic/model/_pipeline.py +0 -30
- corvic/model/_proto_orm_convert.py +31 -23
- corvic/model/_resource.py +11 -0
- corvic/orm/__init__.py +0 -6
- corvic/pa_scalar/_to_value.py +11 -16
- {corvic_engine-0.3.0rc52.dist-info → corvic_engine-0.3.0rc54.dist-info}/METADATA +1 -1
- {corvic_engine-0.3.0rc52.dist-info → corvic_engine-0.3.0rc54.dist-info}/RECORD +12 -12
- corvic_generated/model/v1alpha/models_pb2.py +18 -22
- corvic_generated/model/v1alpha/models_pb2.pyi +6 -13
- {corvic_engine-0.3.0rc52.dist-info → corvic_engine-0.3.0rc54.dist-info}/WHEEL +0 -0
- {corvic_engine-0.3.0rc52.dist-info → corvic_engine-0.3.0rc54.dist-info}/licenses/LICENSE +0 -0
corvic/engine/_native.pyd
CHANGED
Binary file
|
corvic/model/_pipeline.py
CHANGED
@@ -22,7 +22,6 @@ from corvic.model._proto_orm_convert import (
|
|
22
22
|
pipeline_orm_to_proto,
|
23
23
|
pipeline_proto_to_orm,
|
24
24
|
)
|
25
|
-
from corvic.model._resource import Resource, ResourceID
|
26
25
|
from corvic.model._source import Source
|
27
26
|
from corvic.result import InvalidArgumentError, NotFoundError, Ok
|
28
27
|
from corvic_generated.model.v1alpha import models_pb2
|
@@ -122,9 +121,6 @@ class Pipeline(BaseModel[PipelineID, models_pb2.Pipeline, orm.Pipeline]):
|
|
122
121
|
@classmethod
|
123
122
|
def orm_load_options(cls) -> list[LoaderOption]:
|
124
123
|
return [
|
125
|
-
sa_orm.selectinload(orm.Pipeline.inputs)
|
126
|
-
.selectinload(orm.PipelineInput.resource)
|
127
|
-
.selectinload(orm.Resource.pipeline_ref),
|
128
124
|
sa_orm.selectinload(orm.Pipeline.outputs)
|
129
125
|
.selectinload(orm.PipelineOutput.source)
|
130
126
|
.selectinload(orm.Source.pipeline_ref),
|
@@ -168,13 +164,6 @@ class Pipeline(BaseModel[PipelineID, models_pb2.Pipeline, orm.Pipeline]):
|
|
168
164
|
def description(self):
|
169
165
|
return self.proto_self.description
|
170
166
|
|
171
|
-
@functools.cached_property
|
172
|
-
def inputs(self) -> Mapping[str, Resource]:
|
173
|
-
return {
|
174
|
-
name: Resource(self.client, proto_resource)
|
175
|
-
for name, proto_resource in self.proto_self.resource_inputs.items()
|
176
|
-
}
|
177
|
-
|
178
167
|
@functools.cached_property
|
179
168
|
def outputs(self) -> Mapping[str, Source]:
|
180
169
|
return {
|
@@ -187,25 +176,6 @@ class Pipeline(BaseModel[PipelineID, models_pb2.Pipeline, orm.Pipeline]):
|
|
187
176
|
new_proto.name = name
|
188
177
|
return self.__class__(self.client, proto_self=new_proto)
|
189
178
|
|
190
|
-
def with_input(
|
191
|
-
self, resource: Resource | ResourceID
|
192
|
-
) -> Ok[Self] | NotFoundError | InvalidArgumentError:
|
193
|
-
if isinstance(resource, ResourceID):
|
194
|
-
match Resource.from_id(resource, self.client):
|
195
|
-
case NotFoundError() as err:
|
196
|
-
return err
|
197
|
-
case Ok(obj):
|
198
|
-
resource = obj
|
199
|
-
|
200
|
-
if resource.room_id != self.room_id:
|
201
|
-
return InvalidArgumentError("cannot add inputs from other rooms")
|
202
|
-
|
203
|
-
input_name = f"output-{uuid.uuid4()}"
|
204
|
-
new_proto = copy.deepcopy(self.proto_self)
|
205
|
-
new_proto.resource_inputs[input_name].CopyFrom(resource.proto_self)
|
206
|
-
|
207
|
-
return Ok(self.__class__(self.client, proto_self=new_proto))
|
208
|
-
|
209
179
|
|
210
180
|
class UnknownTransformationPipeline(Pipeline):
|
211
181
|
"""A pipeline that this version of the code doesn't know what to do with."""
|
@@ -103,6 +103,11 @@ def timestamp_orm_to_proto(
|
|
103
103
|
|
104
104
|
|
105
105
|
def resource_orm_to_proto(resource_orm: orm.Resource) -> models_pb2.Resource:
|
106
|
+
pipeline_input_name = ""
|
107
|
+
pipeline_id = ""
|
108
|
+
if resource_orm.pipeline_ref:
|
109
|
+
pipeline_input_name = resource_orm.pipeline_ref.name
|
110
|
+
pipeline_id = str(resource_orm.pipeline_ref.pipeline_id)
|
106
111
|
return models_pb2.Resource(
|
107
112
|
id=str(resource_orm.id),
|
108
113
|
name=resource_orm.name,
|
@@ -115,9 +120,8 @@ def resource_orm_to_proto(resource_orm: orm.Resource) -> models_pb2.Resource:
|
|
115
120
|
room_id=str(resource_orm.room_id),
|
116
121
|
org_id=str(resource_orm.org_id),
|
117
122
|
recent_events=[resource_orm.latest_event] if resource_orm.latest_event else [],
|
118
|
-
pipeline_id=
|
119
|
-
|
120
|
-
else "",
|
123
|
+
pipeline_id=pipeline_id,
|
124
|
+
pipeline_input_name=pipeline_input_name,
|
121
125
|
created_at=timestamp_orm_to_proto(resource_orm.created_at),
|
122
126
|
)
|
123
127
|
|
@@ -192,10 +196,6 @@ def pipeline_orm_to_proto(
|
|
192
196
|
id=str(pipeline_orm.id),
|
193
197
|
name=pipeline_orm.name,
|
194
198
|
room_id=str(pipeline_orm.room_id),
|
195
|
-
resource_inputs={
|
196
|
-
input_obj.name: resource_orm_to_proto(input_obj.resource)
|
197
|
-
for input_obj in pipeline_orm.inputs
|
198
|
-
},
|
199
199
|
source_outputs={
|
200
200
|
output_obj.name: source_orm_to_proto(output_obj.source)
|
201
201
|
for output_obj in pipeline_orm.outputs
|
@@ -279,7 +279,30 @@ def resource_proto_to_orm(
|
|
279
279
|
latest_event=proto_obj.recent_events[-1] if proto_obj.recent_events else None,
|
280
280
|
room_id=room_id,
|
281
281
|
)
|
282
|
-
|
282
|
+
_add_orm_to_session(orm_obj, proto_obj.org_id, session)
|
283
|
+
|
284
|
+
if proto_obj.pipeline_id:
|
285
|
+
match _translate_orm_id(proto_obj.pipeline_id, orm.PipelineID):
|
286
|
+
case orm.InvalidORMIdentifierError() as err:
|
287
|
+
return err
|
288
|
+
case Ok(pipeline_id):
|
289
|
+
pass
|
290
|
+
if not pipeline_id:
|
291
|
+
return InvalidArgumentError("resource's pipeline cannot be anonymous")
|
292
|
+
session.flush()
|
293
|
+
if not orm_obj.id:
|
294
|
+
raise InternalError("internal assertion did not hold")
|
295
|
+
pipeline_input = orm.PipelineInput(
|
296
|
+
resource_id=orm_obj.id,
|
297
|
+
name=proto_obj.pipeline_input_name,
|
298
|
+
pipeline_id=pipeline_id,
|
299
|
+
room_id=room_id,
|
300
|
+
)
|
301
|
+
if orm_obj.org_id:
|
302
|
+
pipeline_input.org_id = orm_obj.org_id
|
303
|
+
orm_obj.pipeline_ref = session.merge(pipeline_input)
|
304
|
+
|
305
|
+
return Ok(orm_obj)
|
283
306
|
|
284
307
|
|
285
308
|
def _ensure_id(
|
@@ -329,21 +352,6 @@ def pipeline_proto_to_orm( # noqa: C901
|
|
329
352
|
if not orm_obj.id:
|
330
353
|
raise InternalError("internal assertion did not hold")
|
331
354
|
|
332
|
-
for name, val in proto_obj.resource_inputs.items():
|
333
|
-
match _ensure_id(val, resource_proto_to_orm, orm.ResourceID, session):
|
334
|
-
case orm.InvalidORMIdentifierError() | InvalidArgumentError() as err:
|
335
|
-
return err
|
336
|
-
case Ok(resource_id):
|
337
|
-
pass
|
338
|
-
inputs.append(
|
339
|
-
orm.PipelineInput(
|
340
|
-
resource_id=resource_id,
|
341
|
-
name=name,
|
342
|
-
pipeline_id=orm_obj.id,
|
343
|
-
room_id=room_id,
|
344
|
-
)
|
345
|
-
)
|
346
|
-
|
347
355
|
outputs = list[orm.PipelineOutput]()
|
348
356
|
for name, val in proto_obj.source_outputs.items():
|
349
357
|
match _ensure_id(val, source_proto_to_orm, orm.SourceID, session):
|
corvic/model/_resource.py
CHANGED
@@ -236,3 +236,14 @@ class Resource(BaseModel[ResourceID, models_pb2.Resource, orm.Resource]):
|
|
236
236
|
blob.content_type = "application/octet-stream"
|
237
237
|
blob.patch()
|
238
238
|
return cls.from_blob(blob.url, blob, client, room_id=room_id)
|
239
|
+
|
240
|
+
def as_input_to(self, pipeline_id: orm.PipelineID) -> Self:
|
241
|
+
new_proto = copy.deepcopy(self.proto_self)
|
242
|
+
new_proto.pipeline_id = str(pipeline_id)
|
243
|
+
new_proto.pipeline_input_name = f"output-{uuid.uuid4()}"
|
244
|
+
|
245
|
+
return self.__class__(self.client, proto_self=new_proto)
|
246
|
+
|
247
|
+
@property
|
248
|
+
def pipeline_input_name(self) -> str:
|
249
|
+
return self.proto_self.pipeline_input_name
|
corvic/orm/__init__.py
CHANGED
@@ -163,12 +163,6 @@ class Pipeline(BelongsToOrgMixin, BelongsToRoomMixin, Base):
|
|
163
163
|
description: sa_orm.Mapped[str | None] = sa_orm.mapped_column()
|
164
164
|
id: sa_orm.Mapped[PipelineID | None] = primary_key_identity_column()
|
165
165
|
|
166
|
-
inputs: sa_orm.Mapped[list[PipelineInput]] = sa_orm.relationship(
|
167
|
-
viewonly=True,
|
168
|
-
init=False,
|
169
|
-
default_factory=list,
|
170
|
-
)
|
171
|
-
|
172
166
|
outputs: sa_orm.Mapped[list[PipelineOutput]] = sa_orm.relationship(
|
173
167
|
viewonly=True,
|
174
168
|
init=False,
|
corvic/pa_scalar/_to_value.py
CHANGED
@@ -414,28 +414,23 @@ def _visit(
|
|
414
414
|
|
415
415
|
|
416
416
|
def to_value(
|
417
|
-
scalar: Scalar |
|
417
|
+
scalar: Scalar | _PyValue | struct_pb2.Value,
|
418
418
|
) -> struct_pb2.Value:
|
419
419
|
"""Convert Scalar (or convenient Scalar equivalents) to struct_pb2.Value."""
|
420
420
|
match scalar:
|
421
|
-
# Order matters: bools are ints but ints are not bools
|
422
|
-
case bool():
|
423
|
-
return struct_pb2.Value(bool_value=scalar)
|
424
|
-
case int():
|
425
|
-
return struct_pb2.Value(string_value=str(scalar))
|
426
|
-
case float():
|
427
|
-
if _is_invalid_float(scalar):
|
428
|
-
return to_value(None)
|
429
|
-
return struct_pb2.Value(number_value=scalar)
|
430
|
-
case str():
|
431
|
-
return struct_pb2.Value(string_value=scalar)
|
432
|
-
case None:
|
433
|
-
return struct_pb2.Value(null_value=struct_pb2.NULL_VALUE)
|
434
421
|
case struct_pb2.Value():
|
435
422
|
return scalar
|
436
|
-
case
|
423
|
+
case pa.Scalar():
|
437
424
|
value = cast(_PyValue, scalar.as_py())
|
438
|
-
return _visit(value, scalar.type)
|
425
|
+
return _visit(value, cast(Scalar, scalar).type)
|
426
|
+
case _:
|
427
|
+
try:
|
428
|
+
# Attempt to convert the python value into a pyarrow scalar and
|
429
|
+
# then go through our visitor
|
430
|
+
pa_value = cast(Scalar, pa.scalar(cast(Any, scalar)))
|
431
|
+
return to_value(pa_value)
|
432
|
+
except Exception:
|
433
|
+
return struct_pb2.Value(null_value=struct_pb2.NULL_VALUE)
|
439
434
|
|
440
435
|
|
441
436
|
def _to_py(scalar: Scalar) -> Any:
|
@@ -1,6 +1,6 @@
|
|
1
|
-
corvic_engine-0.3.
|
2
|
-
corvic_engine-0.3.
|
3
|
-
corvic_engine-0.3.
|
1
|
+
corvic_engine-0.3.0rc54.dist-info/METADATA,sha256=dIswsV76aebo40RElZBVxIi8a_5eXaWzbfg1013YXEE,1876
|
2
|
+
corvic_engine-0.3.0rc54.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
|
3
|
+
corvic_engine-0.3.0rc54.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
|
4
4
|
corvic/context/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
5
5
|
corvic/context/__init__.py,sha256=zBnPiP-tStGSVMG_0-G_0ay6-yIX2aerW_oYRzAex74,1702
|
6
6
|
corvic/embed/node2vec.py,sha256=JnYb8f2g4XhF6LL2TjpMxLfKhn_Yp1AzptsWwrKQWgc,11146
|
@@ -20,9 +20,9 @@ corvic/model/_defaults.py,sha256=yoKPPSmYJCE5YAD5jLTEmT4XNf_zXoggNK-uyG8MfVs,152
|
|
20
20
|
corvic/model/_errors.py,sha256=Ctlq04SDwHzJPvLaL1rzqzwVqf2b50EILfW3cH4vnh8,261
|
21
21
|
corvic/model/_feature_type.py,sha256=Y-_-wa9fv7XaCAkxfjjoCLxxK2Ftfba-PMefD7bNXzs,917
|
22
22
|
corvic/model/_feature_view.py,sha256=YThcU0T4pK_W6IOJ8uQUUsc3NP7JMBWqYjU_37UjN2o,49757
|
23
|
-
corvic/model/_pipeline.py,sha256=
|
24
|
-
corvic/model/_proto_orm_convert.py,sha256
|
25
|
-
corvic/model/_resource.py,sha256=
|
23
|
+
corvic/model/_pipeline.py,sha256=LtF6uq3VkXGEDTotNEpiPyWJLQMJSrJ8ttXIAHsj3RU,16350
|
24
|
+
corvic/model/_proto_orm_convert.py,sha256=-sOk72dKH91CWdnoTMIjv4RNE3SCUGH-3Bjhc7pM1LU,26942
|
25
|
+
corvic/model/_resource.py,sha256=qR9iQQgxRO7c_6VGMH8gXbmA-6PCigne2ycNIfEskZg,7854
|
26
26
|
corvic/model/_room.py,sha256=57MiBfj8hZcmUfq2PeECrOWDpBZAOSjnVqNUIGXOy2Q,2898
|
27
27
|
corvic/model/_source.py,sha256=JBCk1I6u_rUKPiB4Fvtl7uVm0Jx0LF1oWNd1-Wn_sbI,9412
|
28
28
|
corvic/model/_space.py,sha256=_qXYefPwwL6jGY3zUBYWW9X3ZE4FEuiOksPoCuG_O1Q,33928
|
@@ -50,12 +50,12 @@ corvic/orm/keys.py,sha256=Ag6Xbpvxev-VByT1KJ8ChUn9vKVEzkkMXxrjvtADCtY,2182
|
|
50
50
|
corvic/orm/mixins.py,sha256=HfmzJ7LblHtddbbkDmv7nNWURL87Bnj8NeOnNbfmSN4,17794
|
51
51
|
corvic/orm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
52
52
|
corvic/orm/_proto_columns.py,sha256=tcOu92UjFJFYZLasS6sWJQBDRK26yrnmpTii_LDY4iw,913
|
53
|
-
corvic/orm/__init__.py,sha256=
|
53
|
+
corvic/orm/__init__.py,sha256=_MOX6qoIL6tK64ibSgQkQzd9Is2bH1oR2rGhqI6xe8I,14293
|
54
54
|
corvic/pa_scalar/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
55
55
|
corvic/pa_scalar/_const.py,sha256=1nk6w3Y7crd3J5jSCq7DRVa1lcGk4H1RUr1l4NjnlzE,868
|
56
56
|
corvic/pa_scalar/_from_value.py,sha256=fS3TNPcPI3jAKGmcUIhn8rdqdQEAwgTLEneVxFUeK6M,27531
|
57
57
|
corvic/pa_scalar/_temporal.py,sha256=HPL1z5k4frIiGWM469YGj9j0hCPXNHtI_X1jk1y1c1Q,7842
|
58
|
-
corvic/pa_scalar/_to_value.py,sha256=
|
58
|
+
corvic/pa_scalar/_to_value.py,sha256=U8DoWDMSl58Yi35yd3JS5Gj773e5IOPa2UDfuZqTkjk,13307
|
59
59
|
corvic/pa_scalar/_types.py,sha256=shbytO0ji-H2rBOX_1fooVOshb22wwkVU1W99VBKz1A,1131
|
60
60
|
corvic/pa_scalar/__init__.py,sha256=1nfc0MFGpw78RQEI13VE5hpHuyw_DoE7sJbmzqx5pws,1063
|
61
61
|
corvic/proto_wrapper/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
@@ -156,7 +156,7 @@ corvic_generated/ingest/v2/table_pb2.py,sha256=aTJHaliZm5DMtp7gslNxyn9uDagz-2-_e
|
|
156
156
|
corvic_generated/ingest/v2/table_pb2_grpc.py,sha256=tVs7wMWyAfvHcCQEiUOHLwaptKxgMFG6E7Ki9vNmmvQ,8151
|
157
157
|
corvic_generated/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
158
158
|
corvic_generated/model/v1alpha/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
159
|
-
corvic_generated/model/v1alpha/models_pb2.py,sha256=
|
159
|
+
corvic_generated/model/v1alpha/models_pb2.py,sha256=7lin-DfTxDO5Nc15ENUfKJ738FoOiIYCgmoPEqm_zKU,8396
|
160
160
|
corvic_generated/model/v1alpha/models_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
|
161
161
|
corvic_generated/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
162
162
|
corvic_generated/orm/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
@@ -220,7 +220,7 @@ corvic_generated/ingest/v2/source_pb2.pyi,sha256=k7FdbgurQLk0JA1WiTUerznzxLv8b50
|
|
220
220
|
corvic_generated/ingest/v2/source_pb2_grpc.pyi,sha256=VG5gpql2SREHgqMC_ycT-QJBVpPeSYKOYS2COgGrZa4,6195
|
221
221
|
corvic_generated/ingest/v2/table_pb2.pyi,sha256=p22F8kv0HfM-9OzGP88bLofxmUtxfLR5eVN0HOxXiEo,4382
|
222
222
|
corvic_generated/ingest/v2/table_pb2_grpc.pyi,sha256=AEXYNtrU4xyENumcCrkD2FmFV7T1UVidxxeZ5pyE4Qc,4554
|
223
|
-
corvic_generated/model/v1alpha/models_pb2.pyi,sha256=
|
223
|
+
corvic_generated/model/v1alpha/models_pb2.pyi,sha256=SvRogPuA09--C6uSOU5TPTFWvj0eqcome_JGt_EK5es,11087
|
224
224
|
corvic_generated/model/v1alpha/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
|
225
225
|
corvic_generated/orm/v1/agent_pb2.pyi,sha256=AxcZC0AJqiOyu_5quSMR-E0MjVhDY7b5ym4uZa7WFug,4670
|
226
226
|
corvic_generated/orm/v1/agent_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
|
@@ -244,5 +244,5 @@ corvic_generated/status/v1/event_pb2.pyi,sha256=eU-ibrYpvEAJSIDlSa62-bC96AQU1ykF
|
|
244
244
|
corvic_generated/status/v1/event_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
|
245
245
|
corvic_generated/status/v1/service_pb2.pyi,sha256=iXLR2FOKQJpBgvBzpD2kVwcYOCksP2aRwK4JYaI9CBw,558
|
246
246
|
corvic_generated/status/v1/service_pb2_grpc.pyi,sha256=OoAnaZ64FD0UTzPoRhYvQU8ecoilhHj3ySjSfHbVDaU,1501
|
247
|
-
corvic/engine/_native.pyd,sha256=
|
248
|
-
corvic_engine-0.3.
|
247
|
+
corvic/engine/_native.pyd,sha256=I1MFZZQaMhCn6DLZyvp-gCPp22KegJanmob1Do2OH7Y,438272
|
248
|
+
corvic_engine-0.3.0rc54.dist-info/RECORD,,
|
@@ -22,37 +22,33 @@ from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot
|
|
22
22
|
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
23
23
|
|
24
24
|
|
25
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/model/v1alpha/models.proto\x12\x14\x63orvic.model.v1alpha\x1a\x19\x63orvic/orm/v1/agent.proto\x1a$corvic/orm/v1/completion_model.proto\x1a corvic/orm/v1/feature_view.proto\x1a\x1c\x63orvic/orm/v1/pipeline.proto\x1a\x19\x63orvic/orm/v1/space.proto\x1a\x19\x63orvic/orm/v1/table.proto\x1a\x1c\x63orvic/status/v1/event.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x90\x01\n\x04Room\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x15\n\x06org_id\x18\x03 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\
|
25
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/model/v1alpha/models.proto\x12\x14\x63orvic.model.v1alpha\x1a\x19\x63orvic/orm/v1/agent.proto\x1a$corvic/orm/v1/completion_model.proto\x1a corvic/orm/v1/feature_view.proto\x1a\x1c\x63orvic/orm/v1/pipeline.proto\x1a\x19\x63orvic/orm/v1/space.proto\x1a\x19\x63orvic/orm/v1/table.proto\x1a\x1c\x63orvic/status/v1/event.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x90\x01\n\x04Room\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x15\n\x06org_id\x18\x03 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xd8\x03\n\x08Resource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x1b\n\tmime_type\x18\x04 \x01(\tR\x08mimeType\x12\x10\n\x03url\x18\x05 \x01(\tR\x03url\x12\x12\n\x04size\x18\x06 \x01(\x04R\x04size\x12\x10\n\x03md5\x18\x07 \x01(\tR\x03md5\x12#\n\roriginal_path\x18\x08 \x01(\tR\x0coriginalPath\x12\x17\n\x07room_id\x18\t \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x0b \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\r \x01(\tR\npipelineId\x12.\n\x13pipeline_input_name\x18\x0f \x01(\tR\x11pipelineInputName\x12<\n\rrecent_events\x18\x0e \x03(\x0b\x32\x17.corvic.status.v1.EventR\x0crecentEvents\x12>\n\ncreated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x91\x02\n\x06Source\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x06 \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\x08 \x01(\tR\npipelineId\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xe9\x03\n\x08Pipeline\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\t \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12X\n\x0esource_outputs\x18\x05 \x03(\x0b\x32\x31.corvic.model.v1alpha.Pipeline.SourceOutputsEntryR\rsourceOutputs\x12^\n\x17pipeline_transformation\x18\x06 \x01(\x0b\x32%.corvic.orm.v1.PipelineTransformationR\x16pipelineTransformation\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x1a^\n\x12SourceOutputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x05value:\x02\x38\x01\x42\r\n\x0b_created_at\"\xca\x02\n\x11\x46\x65\x61tureViewSource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x34\n\x06source\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x06source\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12+\n\x11\x64rop_disconnected\x18\x04 \x01(\x08R\x10\x64ropDisconnected\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x07 \x01(\tR\x06roomIdB\r\n\x0b_created_at\"\x9c\x03\n\x0b\x46\x65\x61tureView\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12P\n\x13\x66\x65\x61ture_view_output\x18\x05 \x01(\x0b\x32 .corvic.orm.v1.FeatureViewOutputR\x11\x66\x65\x61tureViewOutput\x12Y\n\x14\x66\x65\x61ture_view_sources\x18\x06 \x03(\x0b\x32\'.corvic.model.v1alpha.FeatureViewSourceR\x12\x66\x65\x61tureViewSources\x12\x1b\n\tspace_ids\x18\x07 \x03(\tR\x08spaceIds\x12\x15\n\x06org_id\x18\x08 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xfa\x02\n\x05Space\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12I\n\x10space_parameters\x18\x05 \x01(\x0b\x32\x1e.corvic.orm.v1.SpaceParametersR\x0fspaceParameters\x12\x44\n\x0c\x66\x65\x61ture_view\x18\x06 \x01(\x0b\x32!.corvic.model.v1alpha.FeatureViewR\x0b\x66\x65\x61tureView\x12\x1b\n\tauto_sync\x18\t \x01(\x08R\x08\x61utoSync\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x85\x02\n\x05\x41gent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12I\n\x10\x61gent_parameters\x18\x04 \x01(\x0b\x32\x1e.corvic.orm.v1.AgentParametersR\x0f\x61gentParameters\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atJ\x04\x08\x06\x10\x07R\x08messages\"\xad\x02\n\x0f\x43ompletionModel\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x15\n\x06org_id\x18\x04 \x01(\tR\x05orgId\x12H\n\nparameters\x18\x05 \x01(\x0b\x32(.corvic.orm.v1.CompletionModelParametersR\nparameters\x12$\n\x0esecret_api_key\x18\x06 \x01(\tR\x0csecretApiKey\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atb\x06proto3')
|
26
26
|
|
27
27
|
_globals = globals()
|
28
28
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
29
29
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.model.v1alpha.models_pb2', _globals)
|
30
30
|
if _descriptor._USE_C_DESCRIPTORS == False:
|
31
31
|
DESCRIPTOR._options = None
|
32
|
-
_globals['_PIPELINE_RESOURCEINPUTSENTRY']._options = None
|
33
|
-
_globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_options = b'8\001'
|
34
32
|
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
|
35
33
|
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
|
36
34
|
_globals['_ROOM']._serialized_start=306
|
37
35
|
_globals['_ROOM']._serialized_end=450
|
38
36
|
_globals['_RESOURCE']._serialized_start=453
|
39
|
-
_globals['_RESOURCE']._serialized_end=
|
40
|
-
_globals['_SOURCE']._serialized_start=
|
41
|
-
_globals['_SOURCE']._serialized_end=
|
42
|
-
_globals['_PIPELINE']._serialized_start=
|
43
|
-
_globals['_PIPELINE']._serialized_end=
|
44
|
-
_globals['
|
45
|
-
_globals['
|
46
|
-
_globals['
|
47
|
-
_globals['
|
48
|
-
_globals['
|
49
|
-
_globals['
|
50
|
-
_globals['
|
51
|
-
_globals['
|
52
|
-
_globals['
|
53
|
-
_globals['
|
54
|
-
_globals['
|
55
|
-
_globals['
|
56
|
-
_globals['_COMPLETIONMODEL']._serialized_start=3233
|
57
|
-
_globals['_COMPLETIONMODEL']._serialized_end=3534
|
37
|
+
_globals['_RESOURCE']._serialized_end=925
|
38
|
+
_globals['_SOURCE']._serialized_start=928
|
39
|
+
_globals['_SOURCE']._serialized_end=1201
|
40
|
+
_globals['_PIPELINE']._serialized_start=1204
|
41
|
+
_globals['_PIPELINE']._serialized_end=1693
|
42
|
+
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1584
|
43
|
+
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1678
|
44
|
+
_globals['_FEATUREVIEWSOURCE']._serialized_start=1696
|
45
|
+
_globals['_FEATUREVIEWSOURCE']._serialized_end=2026
|
46
|
+
_globals['_FEATUREVIEW']._serialized_start=2029
|
47
|
+
_globals['_FEATUREVIEW']._serialized_end=2441
|
48
|
+
_globals['_SPACE']._serialized_start=2444
|
49
|
+
_globals['_SPACE']._serialized_end=2822
|
50
|
+
_globals['_AGENT']._serialized_start=2825
|
51
|
+
_globals['_AGENT']._serialized_end=3086
|
52
|
+
_globals['_COMPLETIONMODEL']._serialized_start=3089
|
53
|
+
_globals['_COMPLETIONMODEL']._serialized_end=3390
|
58
54
|
# @@protoc_insertion_point(module_scope)
|
@@ -26,7 +26,7 @@ class Room(_message.Message):
|
|
26
26
|
def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
|
27
27
|
|
28
28
|
class Resource(_message.Message):
|
29
|
-
__slots__ = ("id", "name", "description", "mime_type", "url", "size", "md5", "original_path", "room_id", "org_id", "pipeline_id", "recent_events", "created_at")
|
29
|
+
__slots__ = ("id", "name", "description", "mime_type", "url", "size", "md5", "original_path", "room_id", "org_id", "pipeline_id", "pipeline_input_name", "recent_events", "created_at")
|
30
30
|
ID_FIELD_NUMBER: _ClassVar[int]
|
31
31
|
NAME_FIELD_NUMBER: _ClassVar[int]
|
32
32
|
DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
|
@@ -38,6 +38,7 @@ class Resource(_message.Message):
|
|
38
38
|
ROOM_ID_FIELD_NUMBER: _ClassVar[int]
|
39
39
|
ORG_ID_FIELD_NUMBER: _ClassVar[int]
|
40
40
|
PIPELINE_ID_FIELD_NUMBER: _ClassVar[int]
|
41
|
+
PIPELINE_INPUT_NAME_FIELD_NUMBER: _ClassVar[int]
|
41
42
|
RECENT_EVENTS_FIELD_NUMBER: _ClassVar[int]
|
42
43
|
CREATED_AT_FIELD_NUMBER: _ClassVar[int]
|
43
44
|
id: str
|
@@ -51,9 +52,10 @@ class Resource(_message.Message):
|
|
51
52
|
room_id: str
|
52
53
|
org_id: str
|
53
54
|
pipeline_id: str
|
55
|
+
pipeline_input_name: str
|
54
56
|
recent_events: _containers.RepeatedCompositeFieldContainer[_event_pb2.Event]
|
55
57
|
created_at: _timestamp_pb2.Timestamp
|
56
|
-
def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., mime_type: _Optional[str] = ..., url: _Optional[str] = ..., size: _Optional[int] = ..., md5: _Optional[str] = ..., original_path: _Optional[str] = ..., room_id: _Optional[str] = ..., org_id: _Optional[str] = ..., pipeline_id: _Optional[str] = ..., recent_events: _Optional[_Iterable[_Union[_event_pb2.Event, _Mapping]]] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
|
58
|
+
def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., mime_type: _Optional[str] = ..., url: _Optional[str] = ..., size: _Optional[int] = ..., md5: _Optional[str] = ..., original_path: _Optional[str] = ..., room_id: _Optional[str] = ..., org_id: _Optional[str] = ..., pipeline_id: _Optional[str] = ..., pipeline_input_name: _Optional[str] = ..., recent_events: _Optional[_Iterable[_Union[_event_pb2.Event, _Mapping]]] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
|
57
59
|
|
58
60
|
class Source(_message.Message):
|
59
61
|
__slots__ = ("id", "name", "table_op_graph", "room_id", "org_id", "pipeline_id", "created_at")
|
@@ -74,14 +76,7 @@ class Source(_message.Message):
|
|
74
76
|
def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., table_op_graph: _Optional[_Union[_table_pb2.TableComputeOp, _Mapping]] = ..., room_id: _Optional[str] = ..., org_id: _Optional[str] = ..., pipeline_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
|
75
77
|
|
76
78
|
class Pipeline(_message.Message):
|
77
|
-
__slots__ = ("id", "name", "description", "room_id", "
|
78
|
-
class ResourceInputsEntry(_message.Message):
|
79
|
-
__slots__ = ("key", "value")
|
80
|
-
KEY_FIELD_NUMBER: _ClassVar[int]
|
81
|
-
VALUE_FIELD_NUMBER: _ClassVar[int]
|
82
|
-
key: str
|
83
|
-
value: Resource
|
84
|
-
def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Resource, _Mapping]] = ...) -> None: ...
|
79
|
+
__slots__ = ("id", "name", "description", "room_id", "source_outputs", "pipeline_transformation", "org_id", "created_at")
|
85
80
|
class SourceOutputsEntry(_message.Message):
|
86
81
|
__slots__ = ("key", "value")
|
87
82
|
KEY_FIELD_NUMBER: _ClassVar[int]
|
@@ -93,7 +88,6 @@ class Pipeline(_message.Message):
|
|
93
88
|
NAME_FIELD_NUMBER: _ClassVar[int]
|
94
89
|
DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
|
95
90
|
ROOM_ID_FIELD_NUMBER: _ClassVar[int]
|
96
|
-
RESOURCE_INPUTS_FIELD_NUMBER: _ClassVar[int]
|
97
91
|
SOURCE_OUTPUTS_FIELD_NUMBER: _ClassVar[int]
|
98
92
|
PIPELINE_TRANSFORMATION_FIELD_NUMBER: _ClassVar[int]
|
99
93
|
ORG_ID_FIELD_NUMBER: _ClassVar[int]
|
@@ -102,12 +96,11 @@ class Pipeline(_message.Message):
|
|
102
96
|
name: str
|
103
97
|
description: str
|
104
98
|
room_id: str
|
105
|
-
resource_inputs: _containers.MessageMap[str, Resource]
|
106
99
|
source_outputs: _containers.MessageMap[str, Source]
|
107
100
|
pipeline_transformation: _pipeline_pb2.PipelineTransformation
|
108
101
|
org_id: str
|
109
102
|
created_at: _timestamp_pb2.Timestamp
|
110
|
-
def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., room_id: _Optional[str] = ...,
|
103
|
+
def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., room_id: _Optional[str] = ..., source_outputs: _Optional[_Mapping[str, Source]] = ..., pipeline_transformation: _Optional[_Union[_pipeline_pb2.PipelineTransformation, _Mapping]] = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
|
111
104
|
|
112
105
|
class FeatureViewSource(_message.Message):
|
113
106
|
__slots__ = ("id", "source", "table_op_graph", "drop_disconnected", "org_id", "created_at", "room_id")
|
File without changes
|
File without changes
|