corvic-engine 0.3.0rc42__cp38-abi3-win_amd64.whl → 0.3.0rc44__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.
Files changed (26) hide show
  1. corvic/embedding_metric/embeddings.py +30 -6
  2. corvic/engine/_native.pyd +0 -0
  3. corvic/model/_completion_model.py +46 -16
  4. corvic/model/_feature_view.py +1 -0
  5. corvic/model/_proto_orm_convert.py +22 -9
  6. corvic/model/_space.py +56 -39
  7. corvic/orm/__init__.py +27 -44
  8. corvic/orm/base.py +4 -0
  9. corvic/system/_dimension_reduction.py +37 -5
  10. corvic/system/in_memory_executor.py +14 -7
  11. {corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/METADATA +19 -19
  12. {corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/RECORD +26 -22
  13. {corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/WHEEL +1 -1
  14. corvic_generated/model/v1alpha/models_pb2.py +32 -31
  15. corvic_generated/model/v1alpha/models_pb2.pyi +21 -14
  16. corvic_generated/orm/v1/agent_pb2.py +2 -2
  17. corvic_generated/orm/v1/agent_pb2.pyi +2 -0
  18. corvic_generated/orm/v1/completion_model_pb2.py +30 -0
  19. corvic_generated/orm/v1/completion_model_pb2.pyi +33 -0
  20. corvic_generated/orm/v1/completion_model_pb2_grpc.py +4 -0
  21. corvic_generated/orm/v1/completion_model_pb2_grpc.pyi +17 -0
  22. corvic_generated/platform/v1/platform_pb2.py +17 -5
  23. corvic_generated/platform/v1/platform_pb2.pyi +20 -0
  24. corvic_generated/platform/v1/platform_pb2_grpc.py +66 -0
  25. corvic_generated/platform/v1/platform_pb2_grpc.pyi +28 -0
  26. {corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/licenses/LICENSE +0 -0
corvic/system/_dimension_reduction.py CHANGED
@@ -3,11 +3,30 @@ from typing import Any, Protocol
3
3
  import numpy as np
4
4
  from numpy.typing import NDArray
5
5
 
6
+ from corvic.result import InvalidArgumentError, Ok
7
+
8
+
9
+ def _validate_embedding_array(
10
+ embeddings: NDArray[Any],
11
+ ) -> Ok[None] | InvalidArgumentError:
12
+ embeddings_ndim = 2
13
+ if embeddings.ndim != embeddings_ndim:
14
+ return InvalidArgumentError(
15
+ f"embeddings ndim must be {embeddings_ndim}",
16
+ ndim=embeddings.ndim,
17
+ )
18
+ if not np.issubdtype(embeddings.dtype, np.number):
19
+ return InvalidArgumentError(
20
+ "embeddings must have a numerical dtype",
21
+ dtype=str(embeddings.dtype),
22
+ )
23
+ return Ok(None)
24
+
6
25
 
7
26
  class DimensionReducer(Protocol):
8
27
  def reduce_dimensions(
9
28
  self, vectors: NDArray[Any], output_dimensions: int, metric: str
10
- ) -> NDArray[Any]: ...
29
+ ) -> Ok[NDArray[Any]] | InvalidArgumentError: ...
11
30
 
12
31
 
13
32
  class UmapDimensionReducer(DimensionReducer):
@@ -16,8 +35,16 @@ class UmapDimensionReducer(DimensionReducer):
16
35
  vectors: NDArray[Any],
17
36
  output_dimensions: int,
18
37
  metric: str,
19
- ):
38
+ ) -> Ok[NDArray[Any]] | InvalidArgumentError:
39
+ match _validate_embedding_array(vectors):
40
+ case InvalidArgumentError() as err:
41
+ return err
42
+ case Ok():
43
+ pass
44
+
20
45
  vectors = np.nan_to_num(vectors.astype(np.float32))
46
+ if vectors.shape[1] == output_dimensions:
47
+ return Ok(vectors)
21
48
  n_neighbors = 15
22
49
  init = "spectral"
23
50
  # y spectral initialization cannot be used when n_neighbors
@@ -45,11 +72,16 @@ class UmapDimensionReducer(DimensionReducer):
45
72
  low_memory=False,
46
73
  verbose=False,
47
74
  )
48
- return projector.fit_transform(vectors)
75
+ return Ok(projector.fit_transform(vectors))
49
76
 
50
77
 
51
78
  class TruncateDimensionReducer(DimensionReducer):
52
79
  def reduce_dimensions(
53
80
  self, vectors: NDArray[Any], output_dimensions: int, metric: str
54
- ) -> NDArray[Any]:
55
- return vectors[:, :output_dimensions]
81
+ ) -> Ok[NDArray[Any]] | InvalidArgumentError:
82
+ match _validate_embedding_array(vectors):
83
+ case InvalidArgumentError() as err:
84
+ return err
85
+ case Ok():
86
+ pass
87
+ return Ok(vectors[:, :output_dimensions])
corvic/system/in_memory_executor.py CHANGED
@@ -531,22 +531,24 @@ class InMemoryExecutor(OpGraphExecutor):
531
531
  case Ok(metric):
532
532
  metrics["ne_sum"] = metric
533
533
  case InvalidArgumentError() as err:
534
- return InternalError.from_(err)
534
+ _logger.warning("could not compute ne_sum", exc_info=str(err))
535
535
  match embedding_metric.condition_number(embedding, normalize=True):
536
536
  case Ok(metric):
537
537
  metrics["condition_number"] = metric
538
538
  case InvalidArgumentError() as err:
539
- return InternalError.from_(err)
539
+ _logger.warning("could not compute condition_number", exc_info=str(err))
540
540
  match embedding_metric.rcondition_number(embedding, normalize=True):
541
541
  case Ok(metric):
542
542
  metrics["rcondition_number"] = metric
543
543
  case InvalidArgumentError() as err:
544
- return InternalError.from_(err)
544
+ _logger.warning(
545
+ "could not compute rcondition_number", exc_info=str(err)
546
+ )
545
547
  match embedding_metric.stable_rank(embedding, normalize=True):
546
548
  case Ok(metric):
547
549
  metrics["stable_rank"] = metric
548
550
  case InvalidArgumentError() as err:
549
- return InternalError.from_(err)
551
+ _logger.warning("could not compute stable_rank", exc_info=str(err))
550
552
  return Ok(_SchemaAndBatches.from_dataframe(embedding_df, metrics=metrics))
551
553
 
552
554
  def _execute_embedding_coordinates(
@@ -583,9 +585,14 @@ class InMemoryExecutor(OpGraphExecutor):
583
585
  case InvalidArgumentError() as err:
584
586
  raise err
585
587
 
586
- coordinates = self._dimension_reducer.reduce_dimensions(
588
+ match self._dimension_reducer.reduce_dimensions(
587
589
  embedding, op.n_components, op.metric
588
- )
590
+ ):
591
+ case Ok(coordinates):
592
+ pass
593
+ case InvalidArgumentError() as err:
594
+ raise err
595
+
589
596
  coordinates_df = embedding_df.with_columns(
590
597
  pl.Series(
591
598
  name=embedding_column_name,
@@ -897,7 +904,7 @@ class InMemoryExecutor(OpGraphExecutor):
897
904
 
898
905
  encoder = MaxAbsScaler()
899
906
  encoded = encoder.fit_transform(
900
- to_encode.to_numpy().reshape(-1, 1)
907
+ np.nan_to_num(to_encode.to_numpy()).reshape(-1, 1)
901
908
  ).flatten()
902
909
 
903
910
  case op_graph.encoder.StandardScaler():
{corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: corvic-engine
3
- Version: 0.3.0rc42
3
+ Version: 0.3.0rc44
4
4
  Classifier: Environment :: Console
5
5
  Classifier: License :: Other/Proprietary License
6
6
  Classifier: Programming Language :: Python :: Implementation :: CPython
@@ -11,25 +11,25 @@ Classifier: Programming Language :: Python :: 3.12
11
11
  Classifier: Programming Language :: Python :: 3.13
12
12
  Classifier: Programming Language :: Rust
13
13
  Classifier: Topic :: Scientific/Engineering
14
- Requires-Dist: cachetools >=5
15
- Requires-Dist: duckdb >=1.0.0
16
- Requires-Dist: more-itertools >=10
17
- Requires-Dist: numpy >=1.26
18
- Requires-Dist: polars >=1.7.1
19
- Requires-Dist: protobuf >=4.25
20
- Requires-Dist: protovalidate >=0.3
21
- Requires-Dist: pyarrow >=17
22
- Requires-Dist: sqlalchemy >=2
23
- Requires-Dist: sqlglot >=25.6.0, <26
24
- Requires-Dist: structlog >=24
14
+ Requires-Dist: cachetools>=5
15
+ Requires-Dist: duckdb>=1.0.0
16
+ Requires-Dist: more-itertools>=10
17
+ Requires-Dist: numpy>=1.26
18
+ Requires-Dist: polars>=1.7.1
19
+ Requires-Dist: protobuf>=4.25
20
+ Requires-Dist: protovalidate>=0.3
21
+ Requires-Dist: pyarrow>=17
22
+ Requires-Dist: sqlalchemy>=2
23
+ Requires-Dist: sqlglot>=25.6.0,<26
24
+ Requires-Dist: structlog>=24
25
25
  Requires-Dist: tqdm
26
- Requires-Dist: typing-extensions >=4.9
27
- Requires-Dist: umap-learn >=0.5.5 ; extra == 'ml'
28
- Requires-Dist: pillow >=10.0.0 ; extra == 'ml'
29
- Requires-Dist: scikit-learn >=1.4.0 ; extra == 'ml'
30
- Requires-Dist: transformers[torch] >=4.45.0 ; extra == 'ml'
31
- Requires-Dist: opentelemetry-api >=1.20.0 ; extra == 'telemetry'
32
- Requires-Dist: opentelemetry-sdk >=1.20.0 ; extra == 'telemetry'
26
+ Requires-Dist: typing-extensions>=4.9
27
+ Requires-Dist: umap-learn>=0.5.5 ; extra == 'ml'
28
+ Requires-Dist: pillow>=10.0.0 ; extra == 'ml'
29
+ Requires-Dist: scikit-learn>=1.4.0 ; extra == 'ml'
30
+ Requires-Dist: transformers[torch]>=4.45.0 ; extra == 'ml'
31
+ Requires-Dist: opentelemetry-api>=1.20.0 ; extra == 'telemetry'
32
+ Requires-Dist: opentelemetry-sdk>=1.20.0 ; extra == 'telemetry'
33
33
  Provides-Extra: ml
34
34
  Provides-Extra: telemetry
35
35
  License-File: LICENSE
{corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/RECORD RENAMED
@@ -1,12 +1,12 @@
1
- corvic_engine-0.3.0rc42.dist-info/METADATA,sha256=NCpmQQXRbq8jopf7efVCe1CLLYKruLYxsGmO9cAiTlk,1895
2
- corvic_engine-0.3.0rc42.dist-info/WHEEL,sha256=mYDPFrpmO1_UhonyEoMo2i0zHOwqc92hj_Q3CYuEBdM,94
3
- corvic_engine-0.3.0rc42.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
1
+ corvic_engine-0.3.0rc44.dist-info/METADATA,sha256=HPsaBURKtnipM5xOeacGAdHHNWtIlTqrHcmcWG7aPYo,1876
2
+ corvic_engine-0.3.0rc44.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
3
+ corvic_engine-0.3.0rc44.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
7
7
  corvic/embed/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
8
  corvic/embed/__init__.py,sha256=cZZSrRXmezJuTafcQgrB1rbitqXZTVY1B5ryRzAlvgs,144
9
- corvic/embedding_metric/embeddings.py,sha256=jyqL8GthdFcC2iemtP-U-p8WiguiG_40Ld2iJJ1Tz_Y,13337
9
+ corvic/embedding_metric/embeddings.py,sha256=5jvSY0cg5P-Wg_KN7DsrcPo5AfJ_1-XKdErx_dNN5B8,14082
10
10
  corvic/embedding_metric/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  corvic/embedding_metric/__init__.py,sha256=8a-QKSQNbiksueHk5LdkugjZr6wasP4ff8A-Jsn5gUI,536
12
12
  corvic/engine/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -15,17 +15,17 @@ corvic/engine/__init__.py,sha256=XL4Vg7rNcBi29ccVelpeFizR9oJtGYXDn84W9zok9d4,975
15
15
  corvic/model/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
16
  corvic/model/_agent.py,sha256=WGdxu0oLHU7EEgBOMQ0UOnH2AIIUGfeH6VSQZgC0kEk,3621
17
17
  corvic/model/_base_model.py,sha256=r6NKFAlTkxQPDF-RcuWoZkoOdrRSZc8iL3B6W8JvmK0,8341
18
- corvic/model/_completion_model.py,sha256=J8dDy42I66EinqgxjdsGaQu40uKhYYRfPezwh-daNjw,5503
18
+ corvic/model/_completion_model.py,sha256=uoqF7hwxzGXXqSPZT_CIcNBSDmYhcxMotpGPucWH6Q0,6656
19
19
  corvic/model/_defaults.py,sha256=yoKPPSmYJCE5YAD5jLTEmT4XNf_zXoggNK-uyG8MfVs,1524
20
20
  corvic/model/_errors.py,sha256=Ctlq04SDwHzJPvLaL1rzqzwVqf2b50EILfW3cH4vnh8,261
21
21
  corvic/model/_feature_type.py,sha256=Y-_-wa9fv7XaCAkxfjjoCLxxK2Ftfba-PMefD7bNXzs,917
22
- corvic/model/_feature_view.py,sha256=SW-BOCWpW9GXkFrC5b_JfY8AUAKK_q-BpHYFpkkTXd4,49894
22
+ corvic/model/_feature_view.py,sha256=yEYRCudoNEQVCVh00Ax3LuCUd-dAOlclpPKpzXGJv5o,49936
23
23
  corvic/model/_pipeline.py,sha256=IEUaMX8XYDH-qx5336X5MM10F-h0nHoNNnM86b8sBqA,18193
24
- corvic/model/_proto_orm_convert.py,sha256=hHtWyYwFkhel65wtoVRve9zsdFvxc84ShaDQh98epzs,23783
24
+ corvic/model/_proto_orm_convert.py,sha256=2sqohdO1Y-e7yfVvIK852h7OqnTNjk9bV3G9sHLp548,24161
25
25
  corvic/model/_resource.py,sha256=UFff-ZGjSnIo810kELRZ9NiA4yXETIrsxjyMId9sEmc,7304
26
26
  corvic/model/_room.py,sha256=57MiBfj8hZcmUfq2PeECrOWDpBZAOSjnVqNUIGXOy2Q,2898
27
27
  corvic/model/_source.py,sha256=pv2vsjZNuc4KAqy39pCN0tK6kvtowODQhIeUfvS0ek0,9526
28
- corvic/model/_space.py,sha256=Vu4YEghLvf6O1k_wi87nfDo7WvAtII5DDmwj7VpifNQ,36224
28
+ corvic/model/_space.py,sha256=1tyCFz_AvBgYffA00v17ubJxXMwe8tU-sKTasCSg9eQ,36632
29
29
  corvic/model/__init__.py,sha256=_Hjo5INX0urDsUZPlUtnwvODNkZ7H0azMS0RXCS3MZI,2360
30
30
  corvic/op_graph/aggregation.py,sha256=8X6vqXD7dLHrhYJU0BqmhUsWGbzD1zSP5Db5VHdIru4,6187
31
31
  corvic/op_graph/encoders.py,sha256=EhEmAiwgnXNiJ8NU0xm4deC7EZm80UzuzBL5MV140LQ,9217
@@ -40,7 +40,7 @@ corvic/op_graph/sample_strategy.py,sha256=DrbtJ3ORkIRfyIE_FdlOh_UMnCW_K9jL1LeonV
40
40
  corvic/op_graph/_schema.py,sha256=STbxY5PIqIA6xkSDeK8k72Nutsxq5jGe7e_aT35aznI,5733
41
41
  corvic/op_graph/_transformations.py,sha256=L9Au_GcciPynww4ZXojMtNdPJ36Qboc9gn0bVzXLifU,9445
42
42
  corvic/op_graph/__init__.py,sha256=1DMrQfuuS3FkLa9DXYDjSDLurdxxpG5H1jB2ctaa9xo,1444
43
- corvic/orm/base.py,sha256=0GP9zwf7I1rZjGO2MdE3l14DbrbOlXWlp-qIevaNWJM,8636
43
+ corvic/orm/base.py,sha256=2LJh6gRZXpQwV50hquGIGkgqRsIq_pew1djgGd7umW0,8817
44
44
  corvic/orm/errors.py,sha256=uFhFXpVG6pby1lndJZHGHxv3Y0Fbt0RiaZ-CqDfuY1o,545
45
45
  corvic/orm/func/utc_func.py,sha256=-FC6w9wBWXejMv1AICT2Gg7tdkSo7gqL2dFT-YKPGQ4,4518
46
46
  corvic/orm/func/uuid_func.py,sha256=oXPjDGAl3mvlNtvcvBrLmRRHPJgtKffShIPbHm-EswA,1152
@@ -50,7 +50,7 @@ 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=s0X4PcgW82LETtlRAOX7dUl2ajZoCKPQMpo33mWEW1k,21961
53
+ corvic/orm/__init__.py,sha256=k8g7W5g0QoTusYBwVg3zqmmoqdVPV4b45ff3SWWF7nY,21533
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
@@ -68,12 +68,12 @@ corvic/sql/parse_ops.py,sha256=1ZXVlDzIzqwW_KP0mwMxaY91tLSXqpeaUHyrGJkh56o,29444
68
68
  corvic/sql/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
69
69
  corvic/sql/__init__.py,sha256=kZ1a39KVZ08P8Bg6XuXDLD_dTQX0k620u4nwxZF4SnY,303
70
70
  corvic/system/client.py,sha256=hGhZX8RtHrFEOlOmJNlUHktOZrutOwNYUY_a1htQSrg,821
71
- corvic/system/in_memory_executor.py,sha256=xmb2386RsExMuzCR_XfmJHDdfI2K64v5Z91dbDkhNCk,60803
71
+ corvic/system/in_memory_executor.py,sha256=dYgcxbA_O0mM1pI19t2OXs8q5B4TX-NFacR7TBIhWBk,61136
72
72
  corvic/system/op_graph_executor.py,sha256=gXFnVkemS5EwNegJdU-xVAfMLPULqMFPF7d3EG3AD_U,3482
73
73
  corvic/system/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
74
74
  corvic/system/staging.py,sha256=K5P5moiuAMfPx7lxK4mArxeURBwKoyB6x9HGu9JJ16E,1846
75
75
  corvic/system/storage.py,sha256=y7NKArf9104S6muHpxDXL8uNp7toFBXRQlikNAfdR4Q,5644
76
- corvic/system/_dimension_reduction.py,sha256=r-IlkZR7qnEweO9VjNeovtMmuY1gfXCPe50ChPnc7iw,1774
76
+ corvic/system/_dimension_reduction.py,sha256=vyD8wOs0vE-hlVnCrBTjStTowAPWYREqnQ_bVuGYvis,2907
77
77
  corvic/system/_embedder.py,sha256=0WO24IKi8VC8jsFdvNuzDsgNejyacQf6r9Q34jxhHc4,3844
78
78
  corvic/system/_image_embedder.py,sha256=iQc3KlLcqrhP6K84hncHutThAN8Qd6K7K5dceHyU1TU,8373
79
79
  corvic/system/_planner.py,sha256=HUf6UjCy1iHRrXfhU25w19TG4Ik3zVHhtzVcor0eTQY,7888
@@ -156,14 +156,16 @@ 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=gyINQ-NT_kl_ZMvVqkoZOJOMq_hV6DJOA4S9skrRmC8,10627
159
+ corvic_generated/model/v1alpha/models_pb2.py,sha256=_WkLvkNip8iE5Elo7ziHbh-rcjCvM_xjnIEuSj48E1k,10952
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
163
- corvic_generated/orm/v1/agent_pb2.py,sha256=CRPwvKVzExmz48fg24IzQv_joUiQm8aMd_H-JQW4DE8,3850
163
+ corvic_generated/orm/v1/agent_pb2.py,sha256=YDAzQEHtVlm3f8BW7jEzayHoSCjZjSx0Ar-oRBB0LyQ,3891
164
164
  corvic_generated/orm/v1/agent_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
165
165
  corvic_generated/orm/v1/common_pb2.py,sha256=pu1Rto4Pm8ntdB12Rgb_qnz3qOalxlOnrLuKE1fCETs,3490
166
166
  corvic_generated/orm/v1/common_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
167
+ corvic_generated/orm/v1/completion_model_pb2.py,sha256=aQOc_42UHONp-7O0H9ftiuxTbAzfNnjk-3jwtP9dKyA,2081
168
+ corvic_generated/orm/v1/completion_model_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
167
169
  corvic_generated/orm/v1/feature_view_pb2.py,sha256=EnPKwNSmaJxbEcqFk2vVzW3Rsr2eBpV49IC71HA7v2w,3012
168
170
  corvic_generated/orm/v1/feature_view_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
169
171
  corvic_generated/orm/v1/pipeline_pb2.py,sha256=_J1kHAe2CQT_epXows4eOZKdQFLTLqZgz_DsffZ0i68,2417
@@ -174,8 +176,8 @@ corvic_generated/orm/v1/table_pb2.py,sha256=LArWzoLr31IRO2-gKc0JBn_SFB-g9JcISN8e
174
176
  corvic_generated/orm/v1/table_pb2_grpc.py,sha256=ixBOrA7wwNxEQCRT1kO2N_LayeFYEdFJjVRkkhesWbY,4558
175
177
  corvic_generated/platform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
176
178
  corvic_generated/platform/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
177
- corvic_generated/platform/v1/platform_pb2.py,sha256=lNbPCY-tZYIfl_ZTuMaPgkL8m-nJyn-9K4ghUH529XM,9224
178
- corvic_generated/platform/v1/platform_pb2_grpc.py,sha256=jdPQAgEHaTrD7JTDyjUl8ND1ox1lnjNSEsYq6AwYehs,13407
179
+ corvic_generated/platform/v1/platform_pb2.py,sha256=Il1UYttRhPk54stIbednz8rnyrEqvp5d9cNXp2bzwag,10662
180
+ corvic_generated/platform/v1/platform_pb2_grpc.py,sha256=sx3PmrtfiqZqW2Mr9NyYyKXfv4dYimbBZfvFxhE197g,17052
179
181
  corvic_generated/query/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
180
182
  corvic_generated/query/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
181
183
  corvic_generated/query/v1/query_pb2.py,sha256=cRHixYUF9eihJ2GGL9r3Ny49cfsK_zWsBCnLv23JzYs,2831
@@ -218,12 +220,14 @@ corvic_generated/ingest/v2/source_pb2.pyi,sha256=k7FdbgurQLk0JA1WiTUerznzxLv8b50
218
220
  corvic_generated/ingest/v2/source_pb2_grpc.pyi,sha256=VG5gpql2SREHgqMC_ycT-QJBVpPeSYKOYS2COgGrZa4,6195
219
221
  corvic_generated/ingest/v2/table_pb2.pyi,sha256=p22F8kv0HfM-9OzGP88bLofxmUtxfLR5eVN0HOxXiEo,4382
220
222
  corvic_generated/ingest/v2/table_pb2_grpc.pyi,sha256=AEXYNtrU4xyENumcCrkD2FmFV7T1UVidxxeZ5pyE4Qc,4554
221
- corvic_generated/model/v1alpha/models_pb2.pyi,sha256=sXptdQj57MRdg280qszxpq8E2-eWT5UZwkE2k_3EH_w,14158
223
+ corvic_generated/model/v1alpha/models_pb2.pyi,sha256=4JXVvtxGNee1mgjd3ktwyX8boEngKYy_15KLeJTLX_0,14649
222
224
  corvic_generated/model/v1alpha/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
223
- corvic_generated/orm/v1/agent_pb2.pyi,sha256=2zqfoP9Gm1kKBc1FtqzmtZKUZYhFsCDtQ-Pgr0HfE1c,4496
225
+ corvic_generated/orm/v1/agent_pb2.pyi,sha256=9AExLKFRvOJ1fSaOdZc3Src015uvNl-le2lSPatZjSQ,4575
224
226
  corvic_generated/orm/v1/agent_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
225
227
  corvic_generated/orm/v1/common_pb2.pyi,sha256=PZi9KfnXWzlQC0bMuDC71IOmgIaYClY2jPghzKV-mAo,3557
226
228
  corvic_generated/orm/v1/common_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
229
+ corvic_generated/orm/v1/completion_model_pb2.pyi,sha256=wJagec60xt2cl1tqLu8ZKNliEh6lfbBPojzIOa38fd4,1718
230
+ corvic_generated/orm/v1/completion_model_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
227
231
  corvic_generated/orm/v1/feature_view_pb2.pyi,sha256=MDvBZysNPr6q8TfCVdPT2Bz-IVSf3Tcq2k957KHAtyA,2671
228
232
  corvic_generated/orm/v1/feature_view_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
229
233
  corvic_generated/orm/v1/pipeline_pb2.pyi,sha256=i2VWx5wqO5GR8OGGSUUr3w7n-TqSo_UPy0J-3TuriKA,2264
@@ -232,13 +236,13 @@ corvic_generated/orm/v1/space_pb2.pyi,sha256=qKMymobwu_qQAlFxayifiUkQBpjrK9tAgoQ
232
236
  corvic_generated/orm/v1/space_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
233
237
  corvic_generated/orm/v1/table_pb2.pyi,sha256=qUtOna4mJQXcf2Za0buDQHHJPdtvoevjAtfyesExIwE,49683
234
238
  corvic_generated/orm/v1/table_pb2_grpc.pyi,sha256=K4hyNndkiKpxt9PYxcn_98RTpb4yxET3Um2rDe3VJTI,2499
235
- corvic_generated/platform/v1/platform_pb2.pyi,sha256=pz29JG9WNwVuppIRSCWxWo7B8zoyAE4Qqph-uPLKEUE,5001
236
- corvic_generated/platform/v1/platform_pb2_grpc.pyi,sha256=6rmhbov-ZGD8BLFDYdBPLQoV40Z0sbtDB6eRgRJtato,6064
239
+ corvic_generated/platform/v1/platform_pb2.pyi,sha256=y6kR7rBuar5cFgn7vTaAVTETOsmki-fC4I-4Y1M8JrQ,5627
240
+ corvic_generated/platform/v1/platform_pb2_grpc.pyi,sha256=wlnUjgSOjBJzyG6ubpMv9H9XD_jJUQyUsgYIyx_hx20,7652
237
241
  corvic_generated/query/v1/query_pb2.pyi,sha256=scOTjQSNmG1mPrgx9cMN8anovQMpfJDSJnoMN0P8J1s,2972
238
242
  corvic_generated/query/v1/query_pb2_grpc.pyi,sha256=Uy_oO0HSK_QlJsjneYktgFi67gMcq9GtEsM5I1CoPy8,1494
239
243
  corvic_generated/status/v1/event_pb2.pyi,sha256=eU-ibrYpvEAJSIDlSa62-bC96AQU1ykFi3_sQ1RzBvo,2027
240
244
  corvic_generated/status/v1/event_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
241
245
  corvic_generated/status/v1/service_pb2.pyi,sha256=iXLR2FOKQJpBgvBzpD2kVwcYOCksP2aRwK4JYaI9CBw,558
242
246
  corvic_generated/status/v1/service_pb2_grpc.pyi,sha256=OoAnaZ64FD0UTzPoRhYvQU8ecoilhHj3ySjSfHbVDaU,1501
243
- corvic/engine/_native.pyd,sha256=i4tlkC-v-gjeqMp1z3zPjxGqCn8gsHYNor63JfCD5Xc,438272
244
- corvic_engine-0.3.0rc42.dist-info/RECORD,,
247
+ corvic/engine/_native.pyd,sha256=1e3JunS4dV33W_zozmeSUQmYx4X9CyLHgoi-A7BmCH8,438272
248
+ corvic_engine-0.3.0rc44.dist-info/RECORD,,
{corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc44.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: maturin (1.8.1)
2
+ Generator: maturin (1.8.2)
3
3
  Root-Is-Purelib: false
4
4
  Tag: cp38-abi3-win_amd64
corvic_generated/model/v1alpha/models_pb2.py CHANGED
@@ -14,6 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
 
15
15
  from corvic_generated.orm.v1 import agent_pb2 as corvic_dot_orm_dot_v1_dot_agent__pb2
16
16
  from corvic_generated.orm.v1 import common_pb2 as corvic_dot_orm_dot_v1_dot_common__pb2
17
+ from corvic_generated.orm.v1 import completion_model_pb2 as corvic_dot_orm_dot_v1_dot_completion__model__pb2
17
18
  from corvic_generated.orm.v1 import feature_view_pb2 as corvic_dot_orm_dot_v1_dot_feature__view__pb2
18
19
  from corvic_generated.orm.v1 import pipeline_pb2 as corvic_dot_orm_dot_v1_dot_pipeline__pb2
19
20
  from corvic_generated.orm.v1 import space_pb2 as corvic_dot_orm_dot_v1_dot_space__pb2
@@ -23,7 +24,7 @@ from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
23
24
  from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
24
25
 
25
26
 
26
- 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\x1a\x63orvic/orm/v1/common.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\x1cgoogle/protobuf/struct.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\"\xc7\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\x1d\n\nsource_ids\x18\n \x03(\tR\tsourceIds\x12\x15\n\x06org_id\x18\x0b \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\r \x01(\tR\npipelineId\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\"\xb2\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\x1f\n\x0bresource_id\x18\x05 \x01(\tR\nresourceId\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\"\xa9\x05\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\x12[\n\x0fresource_inputs\x18\x04 \x03(\x0b\x32\x32.corvic.model.v1alpha.Pipeline.ResourceInputsEntryR\x0eresourceInputs\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\x61\n\x13ResourceInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.corvic.model.v1alpha.ResourceR\x05value:\x02\x38\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\"\xb1\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\x42\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\"\xdc\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&\n\x0f\x66\x65\x61ture_view_id\x18\x06 \x01(\tR\rfeatureViewId\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\"7\n\x0bUserMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\"\xce\x02\n\x0c\x41gentMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12/\n\x06policy\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x06policy\x12\x18\n\x07\x63ontext\x18\x04 \x01(\tR\x07\x63ontext\x12P\n\x10message_reaction\x18\x05 \x01(\x0e\x32%.corvic.model.v1alpha.MessageReactionR\x0fmessageReaction\x12&\n\x0fuser_message_id\x18\x06 \x01(\tR\ruserMessageId\x12O\n\x12retrieved_entities\x18\x07 \x01(\x0b\x32 .corvic.orm.v1.RetrievedEntitiesR\x11retrievedEntities\"\x8b\x02\n\x0cMessageEntry\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x46\n\x0cuser_message\x18\x02 \x01(\x0b\x32!.corvic.model.v1alpha.UserMessageH\x00R\x0buserMessage\x12I\n\ragent_message\x18\x03 \x01(\x0b\x32\".corvic.model.v1alpha.AgentMessageH\x00R\x0c\x61gentMessage\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tcreatedAt\x88\x01\x01\x42\t\n\x07\x63ontentB\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\"\x9e\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\x12\x1d\n\nmodel_name\x18\x08 \x01(\tR\tmodelName\x12\x1a\n\x08\x65ndpoint\x18\x05 \x01(\tR\x08\x65ndpoint\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_at*u\n\x0fMessageReaction\x12 \n\x1cMESSAGE_REACTION_UNSPECIFIED\x10\x00\x12\x1e\n\x1aMESSAGE_REACTION_THUMBS_UP\x10\x01\x12 \n\x1cMESSAGE_REACTION_THUMBS_DOWN\x10\x02\x62\x06proto3')
27
+ 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\x1a\x63orvic/orm/v1/common.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\x1cgoogle/protobuf/struct.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\"\xc7\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\x1d\n\nsource_ids\x18\n \x03(\tR\tsourceIds\x12\x15\n\x06org_id\x18\x0b \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\r \x01(\tR\npipelineId\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\"\xb2\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\x1f\n\x0bresource_id\x18\x05 \x01(\tR\nresourceId\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\"\xa9\x05\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\x12[\n\x0fresource_inputs\x18\x04 \x03(\x0b\x32\x32.corvic.model.v1alpha.Pipeline.ResourceInputsEntryR\x0eresourceInputs\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\x61\n\x13ResourceInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.corvic.model.v1alpha.ResourceR\x05value:\x02\x38\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\"\xdc\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&\n\x0f\x66\x65\x61ture_view_id\x18\x06 \x01(\tR\rfeatureViewId\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\"P\n\x0bUserMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\"\xe7\x02\n\x0c\x41gentMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12/\n\x06policy\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x06policy\x12\x18\n\x07\x63ontext\x18\x04 \x01(\tR\x07\x63ontext\x12P\n\x10message_reaction\x18\x05 \x01(\x0e\x32%.corvic.model.v1alpha.MessageReactionR\x0fmessageReaction\x12&\n\x0fuser_message_id\x18\x06 \x01(\tR\ruserMessageId\x12O\n\x12retrieved_entities\x18\x07 \x01(\x0b\x32 .corvic.orm.v1.RetrievedEntitiesR\x11retrievedEntities\x12\x17\n\x07room_id\x18\x08 \x01(\tR\x06roomId\"\xa4\x02\n\x0cMessageEntry\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x46\n\x0cuser_message\x18\x02 \x01(\x0b\x32!.corvic.model.v1alpha.UserMessageH\x00R\x0buserMessage\x12I\n\ragent_message\x18\x03 \x01(\x0b\x32\".corvic.model.v1alpha.AgentMessageH\x00R\x0c\x61gentMessage\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomIdB\t\n\x07\x63ontentB\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_at*u\n\x0fMessageReaction\x12 \n\x1cMESSAGE_REACTION_UNSPECIFIED\x10\x00\x12\x1e\n\x1aMESSAGE_REACTION_THUMBS_UP\x10\x01\x12 \n\x1cMESSAGE_REACTION_THUMBS_DOWN\x10\x02\x62\x06proto3')
27
28
 
28
29
  _globals = globals()
29
30
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -34,34 +35,34 @@ if _descriptor._USE_C_DESCRIPTORS == False:
34
35
  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_options = b'8\001'
35
36
  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
36
37
  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
37
- _globals['_MESSAGEREACTION']._serialized_start=4214
38
- _globals['_MESSAGEREACTION']._serialized_end=4331
39
- _globals['_ROOM']._serialized_start=326
40
- _globals['_ROOM']._serialized_end=470
41
- _globals['_RESOURCE']._serialized_start=473
42
- _globals['_RESOURCE']._serialized_end=928
43
- _globals['_SOURCE']._serialized_start=931
44
- _globals['_SOURCE']._serialized_end=1237
45
- _globals['_PIPELINE']._serialized_start=1240
46
- _globals['_PIPELINE']._serialized_end=1921
47
- _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1713
48
- _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1810
49
- _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1812
50
- _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1906
51
- _globals['_FEATUREVIEWSOURCE']._serialized_start=1924
52
- _globals['_FEATUREVIEWSOURCE']._serialized_end=2229
53
- _globals['_FEATUREVIEW']._serialized_start=2232
54
- _globals['_FEATUREVIEW']._serialized_end=2644
55
- _globals['_SPACE']._serialized_start=2647
56
- _globals['_SPACE']._serialized_end=2995
57
- _globals['_USERMESSAGE']._serialized_start=2997
58
- _globals['_USERMESSAGE']._serialized_end=3052
59
- _globals['_AGENTMESSAGE']._serialized_start=3055
60
- _globals['_AGENTMESSAGE']._serialized_end=3389
61
- _globals['_MESSAGEENTRY']._serialized_start=3392
62
- _globals['_MESSAGEENTRY']._serialized_end=3659
63
- _globals['_AGENT']._serialized_start=3662
64
- _globals['_AGENT']._serialized_end=3923
65
- _globals['_COMPLETIONMODEL']._serialized_start=3926
66
- _globals['_COMPLETIONMODEL']._serialized_end=4212
38
+ _globals['_MESSAGEREACTION']._serialized_start=4367
39
+ _globals['_MESSAGEREACTION']._serialized_end=4484
40
+ _globals['_ROOM']._serialized_start=364
41
+ _globals['_ROOM']._serialized_end=508
42
+ _globals['_RESOURCE']._serialized_start=511
43
+ _globals['_RESOURCE']._serialized_end=966
44
+ _globals['_SOURCE']._serialized_start=969
45
+ _globals['_SOURCE']._serialized_end=1275
46
+ _globals['_PIPELINE']._serialized_start=1278
47
+ _globals['_PIPELINE']._serialized_end=1959
48
+ _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1751
49
+ _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1848
50
+ _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1850
51
+ _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1944
52
+ _globals['_FEATUREVIEWSOURCE']._serialized_start=1962
53
+ _globals['_FEATUREVIEWSOURCE']._serialized_end=2292
54
+ _globals['_FEATUREVIEW']._serialized_start=2295
55
+ _globals['_FEATUREVIEW']._serialized_end=2707
56
+ _globals['_SPACE']._serialized_start=2710
57
+ _globals['_SPACE']._serialized_end=3058
58
+ _globals['_USERMESSAGE']._serialized_start=3060
59
+ _globals['_USERMESSAGE']._serialized_end=3140
60
+ _globals['_AGENTMESSAGE']._serialized_start=3143
61
+ _globals['_AGENTMESSAGE']._serialized_end=3502
62
+ _globals['_MESSAGEENTRY']._serialized_start=3505
63
+ _globals['_MESSAGEENTRY']._serialized_end=3797
64
+ _globals['_AGENT']._serialized_start=3800
65
+ _globals['_AGENT']._serialized_end=4061
66
+ _globals['_COMPLETIONMODEL']._serialized_start=4064
67
+ _globals['_COMPLETIONMODEL']._serialized_end=4365
67
68
  # @@protoc_insertion_point(module_scope)
corvic_generated/model/v1alpha/models_pb2.pyi CHANGED
@@ -1,5 +1,6 @@
1
1
  from corvic_generated.orm.v1 import agent_pb2 as _agent_pb2
2
2
  from corvic_generated.orm.v1 import common_pb2 as _common_pb2
3
+ from corvic_generated.orm.v1 import completion_model_pb2 as _completion_model_pb2
3
4
  from corvic_generated.orm.v1 import feature_view_pb2 as _feature_view_pb2
4
5
  from corvic_generated.orm.v1 import pipeline_pb2 as _pipeline_pb2
5
6
  from corvic_generated.orm.v1 import space_pb2 as _space_pb2
@@ -125,20 +126,22 @@ class Pipeline(_message.Message):
125
126
  def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., room_id: _Optional[str] = ..., resource_inputs: _Optional[_Mapping[str, Resource]] = ..., 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: ...
126
127
 
127
128
  class FeatureViewSource(_message.Message):
128
- __slots__ = ("id", "source", "table_op_graph", "drop_disconnected", "org_id", "created_at")
129
+ __slots__ = ("id", "source", "table_op_graph", "drop_disconnected", "org_id", "created_at", "room_id")
129
130
  ID_FIELD_NUMBER: _ClassVar[int]
130
131
  SOURCE_FIELD_NUMBER: _ClassVar[int]
131
132
  TABLE_OP_GRAPH_FIELD_NUMBER: _ClassVar[int]
132
133
  DROP_DISCONNECTED_FIELD_NUMBER: _ClassVar[int]
133
134
  ORG_ID_FIELD_NUMBER: _ClassVar[int]
134
135
  CREATED_AT_FIELD_NUMBER: _ClassVar[int]
136
+ ROOM_ID_FIELD_NUMBER: _ClassVar[int]
135
137
  id: str
136
138
  source: Source
137
139
  table_op_graph: _table_pb2.TableComputeOp
138
140
  drop_disconnected: bool
139
141
  org_id: str
140
142
  created_at: _timestamp_pb2.Timestamp
141
- def __init__(self, id: _Optional[str] = ..., source: _Optional[_Union[Source, _Mapping]] = ..., table_op_graph: _Optional[_Union[_table_pb2.TableComputeOp, _Mapping]] = ..., drop_disconnected: bool = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
143
+ room_id: str
144
+ def __init__(self, id: _Optional[str] = ..., source: _Optional[_Union[Source, _Mapping]] = ..., table_op_graph: _Optional[_Union[_table_pb2.TableComputeOp, _Mapping]] = ..., drop_disconnected: bool = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., room_id: _Optional[str] = ...) -> None: ...
142
145
 
143
146
  class FeatureView(_message.Message):
144
147
  __slots__ = ("id", "name", "description", "room_id", "feature_view_output", "feature_view_sources", "space_ids", "org_id", "created_at")
@@ -185,15 +188,17 @@ class Space(_message.Message):
185
188
  def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., room_id: _Optional[str] = ..., space_parameters: _Optional[_Union[_space_pb2.SpaceParameters, _Mapping]] = ..., feature_view_id: _Optional[str] = ..., auto_sync: bool = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
186
189
 
187
190
  class UserMessage(_message.Message):
188
- __slots__ = ("id", "message")
191
+ __slots__ = ("id", "message", "room_id")
189
192
  ID_FIELD_NUMBER: _ClassVar[int]
190
193
  MESSAGE_FIELD_NUMBER: _ClassVar[int]
194
+ ROOM_ID_FIELD_NUMBER: _ClassVar[int]
191
195
  id: str
192
196
  message: str
193
- def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ...) -> None: ...
197
+ room_id: str
198
+ def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ..., room_id: _Optional[str] = ...) -> None: ...
194
199
 
195
200
  class AgentMessage(_message.Message):
196
- __slots__ = ("id", "message", "policy", "context", "message_reaction", "user_message_id", "retrieved_entities")
201
+ __slots__ = ("id", "message", "policy", "context", "message_reaction", "user_message_id", "retrieved_entities", "room_id")
197
202
  ID_FIELD_NUMBER: _ClassVar[int]
198
203
  MESSAGE_FIELD_NUMBER: _ClassVar[int]
199
204
  POLICY_FIELD_NUMBER: _ClassVar[int]
@@ -201,6 +206,7 @@ class AgentMessage(_message.Message):
201
206
  MESSAGE_REACTION_FIELD_NUMBER: _ClassVar[int]
202
207
  USER_MESSAGE_ID_FIELD_NUMBER: _ClassVar[int]
203
208
  RETRIEVED_ENTITIES_FIELD_NUMBER: _ClassVar[int]
209
+ ROOM_ID_FIELD_NUMBER: _ClassVar[int]
204
210
  id: str
205
211
  message: str
206
212
  policy: _struct_pb2.Struct
@@ -208,19 +214,22 @@ class AgentMessage(_message.Message):
208
214
  message_reaction: MessageReaction
209
215
  user_message_id: str
210
216
  retrieved_entities: _common_pb2.RetrievedEntities
211
- def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ..., policy: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., context: _Optional[str] = ..., message_reaction: _Optional[_Union[MessageReaction, str]] = ..., user_message_id: _Optional[str] = ..., retrieved_entities: _Optional[_Union[_common_pb2.RetrievedEntities, _Mapping]] = ...) -> None: ...
217
+ room_id: str
218
+ def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ..., policy: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., context: _Optional[str] = ..., message_reaction: _Optional[_Union[MessageReaction, str]] = ..., user_message_id: _Optional[str] = ..., retrieved_entities: _Optional[_Union[_common_pb2.RetrievedEntities, _Mapping]] = ..., room_id: _Optional[str] = ...) -> None: ...
212
219
 
213
220
  class MessageEntry(_message.Message):
214
- __slots__ = ("id", "user_message", "agent_message", "created_at")
221
+ __slots__ = ("id", "user_message", "agent_message", "created_at", "room_id")
215
222
  ID_FIELD_NUMBER: _ClassVar[int]
216
223
  USER_MESSAGE_FIELD_NUMBER: _ClassVar[int]
217
224
  AGENT_MESSAGE_FIELD_NUMBER: _ClassVar[int]
218
225
  CREATED_AT_FIELD_NUMBER: _ClassVar[int]
226
+ ROOM_ID_FIELD_NUMBER: _ClassVar[int]
219
227
  id: str
220
228
  user_message: UserMessage
221
229
  agent_message: AgentMessage
222
230
  created_at: _timestamp_pb2.Timestamp
223
- def __init__(self, id: _Optional[str] = ..., user_message: _Optional[_Union[UserMessage, _Mapping]] = ..., agent_message: _Optional[_Union[AgentMessage, _Mapping]] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
231
+ room_id: str
232
+ def __init__(self, id: _Optional[str] = ..., user_message: _Optional[_Union[UserMessage, _Mapping]] = ..., agent_message: _Optional[_Union[AgentMessage, _Mapping]] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., room_id: _Optional[str] = ...) -> None: ...
224
233
 
225
234
  class Agent(_message.Message):
226
235
  __slots__ = ("id", "name", "room_id", "agent_parameters", "org_id", "created_at")
@@ -239,21 +248,19 @@ class Agent(_message.Message):
239
248
  def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., room_id: _Optional[str] = ..., agent_parameters: _Optional[_Union[_agent_pb2.AgentParameters, _Mapping]] = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
240
249
 
241
250
  class CompletionModel(_message.Message):
242
- __slots__ = ("id", "name", "description", "org_id", "model_name", "endpoint", "secret_api_key", "created_at")
251
+ __slots__ = ("id", "name", "description", "org_id", "parameters", "secret_api_key", "created_at")
243
252
  ID_FIELD_NUMBER: _ClassVar[int]
244
253
  NAME_FIELD_NUMBER: _ClassVar[int]
245
254
  DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
246
255
  ORG_ID_FIELD_NUMBER: _ClassVar[int]
247
- MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
248
- ENDPOINT_FIELD_NUMBER: _ClassVar[int]
256
+ PARAMETERS_FIELD_NUMBER: _ClassVar[int]
249
257
  SECRET_API_KEY_FIELD_NUMBER: _ClassVar[int]
250
258
  CREATED_AT_FIELD_NUMBER: _ClassVar[int]
251
259
  id: str
252
260
  name: str
253
261
  description: str
254
262
  org_id: str
255
- model_name: str
256
- endpoint: str
263
+ parameters: _completion_model_pb2.CompletionModelParameters
257
264
  secret_api_key: str
258
265
  created_at: _timestamp_pb2.Timestamp
259
- def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., org_id: _Optional[str] = ..., model_name: _Optional[str] = ..., endpoint: _Optional[str] = ..., secret_api_key: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
266
+ def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., org_id: _Optional[str] = ..., parameters: _Optional[_Union[_completion_model_pb2.CompletionModelParameters, _Mapping]] = ..., secret_api_key: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
corvic_generated/orm/v1/agent_pb2.py CHANGED
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
 
15
15
 
16
16
 
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19\x63orvic/orm/v1/agent.proto\x12\rcorvic.orm.v1\"\xf3\x01\n\x12\x45xecutorParameters\x12L\n\x15\x63ompletion_model_type\x18\x01 \x01(\x0e\x32\x18.corvic.orm.v1.ModelTypeR\x13\x63ompletionModelType\x12\x42\n\x1d\x63ompletion_system_instruction\x18\x02 \x01(\tR\x1b\x63ompletionSystemInstruction\x12\x33\n\x13\x63ompletion_model_id\x18\x03 \x01(\tH\x00R\x11\x63ompletionModelId\x88\x01\x01\x42\x16\n\x14_completion_model_id\"V\n\x10SpaceInstruction\x12 \n\x0binstruction\x18\x01 \x01(\tR\x0binstruction\x12 \n\x0cspace_run_id\x18\x02 \x01(\tR\nspaceRunId\"\x89\x02\n\x16OrchestratorParameters\x12\x1b\n\tspace_ids\x18\x01 \x03(\tR\x08spaceIds\x12k\n\x12space_instructions\x18\x02 \x03(\x0b\x32<.corvic.orm.v1.OrchestratorParameters.SpaceInstructionsEntryR\x11spaceInstructions\x1a\x65\n\x16SpaceInstructionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.corvic.orm.v1.SpaceInstructionR\x05value:\x02\x38\x01\"\xc5\x01\n\x0f\x41gentParameters\x12R\n\x13\x65xecutor_parameters\x18\x01 \x01(\x0b\x32!.corvic.orm.v1.ExecutorParametersR\x12\x65xecutorParameters\x12^\n\x17orchestrator_parameters\x18\x02 \x01(\x0b\x32%.corvic.orm.v1.OrchestratorParametersR\x16orchestratorParameters*\xc7\x04\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12\x1d\n\x19MODEL_TYPE_GEMINI_1_5_PRO\x10\x01\x12%\n!MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW\x10\x02\x12\x1f\n\x1bMODEL_TYPE_GEMINI_1_5_FLASH\x10\x03\x12\'\n#MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW\x10\x04\x12\'\n#MODEL_TYPE_GEMINI_2_0_FLASH_PREVIEW\x10\r\x12\x30\n,MODEL_TYPE_GEMINI_2_0_FLASH_THINKING_PREVIEW\x10\x10\x12\x1b\n\x17MODEL_TYPE_GPT_4_O_MINI\x10\x05\x12\x16\n\x12MODEL_TYPE_GPT_4_O\x10\x06\x12\x1c\n\x18MODEL_TYPE_GPT_3_5_TURBO\x10\x07\x12\x12\n\x0eMODEL_TYPE_O_1\x10\x0e\x12\x17\n\x13MODEL_TYPE_O_1_MINI\x10\x0f\x12 \n\x1cMODEL_TYPE_CLAUDE_3_5_SONNET\x10\x08\x12\x1d\n\x19MODEL_TYPE_CLAUDE_3_HAIKU\x10\t\x12\x1c\n\x18MODEL_TYPE_MISTRAL_LARGE\x10\n\x12\x1b\n\x17MODEL_TYPE_MISTRAL_NEMO\x10\x0b\x12 \n\x1cMODEL_TYPE_MISTRAL_CODESTRAL\x10\x0c\x12\x15\n\x11MODEL_TYPE_CUSTOM\x10\x11\x62\x06proto3')
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19\x63orvic/orm/v1/agent.proto\x12\rcorvic.orm.v1\"\xf3\x01\n\x12\x45xecutorParameters\x12L\n\x15\x63ompletion_model_type\x18\x01 \x01(\x0e\x32\x18.corvic.orm.v1.ModelTypeR\x13\x63ompletionModelType\x12\x42\n\x1d\x63ompletion_system_instruction\x18\x02 \x01(\tR\x1b\x63ompletionSystemInstruction\x12\x33\n\x13\x63ompletion_model_id\x18\x03 \x01(\tH\x00R\x11\x63ompletionModelId\x88\x01\x01\x42\x16\n\x14_completion_model_id\"V\n\x10SpaceInstruction\x12 \n\x0binstruction\x18\x01 \x01(\tR\x0binstruction\x12 \n\x0cspace_run_id\x18\x02 \x01(\tR\nspaceRunId\"\x89\x02\n\x16OrchestratorParameters\x12\x1b\n\tspace_ids\x18\x01 \x03(\tR\x08spaceIds\x12k\n\x12space_instructions\x18\x02 \x03(\x0b\x32<.corvic.orm.v1.OrchestratorParameters.SpaceInstructionsEntryR\x11spaceInstructions\x1a\x65\n\x16SpaceInstructionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.corvic.orm.v1.SpaceInstructionR\x05value:\x02\x38\x01\"\xc5\x01\n\x0f\x41gentParameters\x12R\n\x13\x65xecutor_parameters\x18\x01 \x01(\x0b\x32!.corvic.orm.v1.ExecutorParametersR\x12\x65xecutorParameters\x12^\n\x17orchestrator_parameters\x18\x02 \x01(\x0b\x32%.corvic.orm.v1.OrchestratorParametersR\x16orchestratorParameters*\xe0\x04\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12\x1d\n\x19MODEL_TYPE_GEMINI_1_5_PRO\x10\x01\x12%\n!MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW\x10\x02\x12\x1f\n\x1bMODEL_TYPE_GEMINI_1_5_FLASH\x10\x03\x12\'\n#MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW\x10\x04\x12\'\n#MODEL_TYPE_GEMINI_2_0_FLASH_PREVIEW\x10\r\x12\x30\n,MODEL_TYPE_GEMINI_2_0_FLASH_THINKING_PREVIEW\x10\x10\x12\x1b\n\x17MODEL_TYPE_GPT_4_O_MINI\x10\x05\x12\x16\n\x12MODEL_TYPE_GPT_4_O\x10\x06\x12\x1c\n\x18MODEL_TYPE_GPT_3_5_TURBO\x10\x07\x12\x12\n\x0eMODEL_TYPE_O_1\x10\x0e\x12\x17\n\x13MODEL_TYPE_O_1_MINI\x10\x0f\x12\x17\n\x13MODEL_TYPE_O_3_MINI\x10\x12\x12 \n\x1cMODEL_TYPE_CLAUDE_3_5_SONNET\x10\x08\x12\x1d\n\x19MODEL_TYPE_CLAUDE_3_HAIKU\x10\t\x12\x1c\n\x18MODEL_TYPE_MISTRAL_LARGE\x10\n\x12\x1b\n\x17MODEL_TYPE_MISTRAL_NEMO\x10\x0b\x12 \n\x1cMODEL_TYPE_MISTRAL_CODESTRAL\x10\x0c\x12\x15\n\x11MODEL_TYPE_CUSTOM\x10\x11\x62\x06proto3')
18
18
 
19
19
  _globals = globals()
20
20
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -24,7 +24,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
24
24
  _globals['_ORCHESTRATORPARAMETERS_SPACEINSTRUCTIONSENTRY']._options = None
25
25
  _globals['_ORCHESTRATORPARAMETERS_SPACEINSTRUCTIONSENTRY']._serialized_options = b'8\001'
26
26
  _globals['_MODELTYPE']._serialized_start=847
27
- _globals['_MODELTYPE']._serialized_end=1430
27
+ _globals['_MODELTYPE']._serialized_end=1455
28
28
  _globals['_EXECUTORPARAMETERS']._serialized_start=45
29
29
  _globals['_EXECUTORPARAMETERS']._serialized_end=288
30
30
  _globals['_SPACEINSTRUCTION']._serialized_start=290
corvic_generated/orm/v1/agent_pb2.pyi CHANGED
@@ -20,6 +20,7 @@ class ModelType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
20
20
  MODEL_TYPE_GPT_3_5_TURBO: _ClassVar[ModelType]
21
21
  MODEL_TYPE_O_1: _ClassVar[ModelType]
22
22
  MODEL_TYPE_O_1_MINI: _ClassVar[ModelType]
23
+ MODEL_TYPE_O_3_MINI: _ClassVar[ModelType]
23
24
  MODEL_TYPE_CLAUDE_3_5_SONNET: _ClassVar[ModelType]
24
25
  MODEL_TYPE_CLAUDE_3_HAIKU: _ClassVar[ModelType]
25
26
  MODEL_TYPE_MISTRAL_LARGE: _ClassVar[ModelType]
@@ -38,6 +39,7 @@ MODEL_TYPE_GPT_4_O: ModelType
38
39
  MODEL_TYPE_GPT_3_5_TURBO: ModelType
39
40
  MODEL_TYPE_O_1: ModelType
40
41
  MODEL_TYPE_O_1_MINI: ModelType
42
+ MODEL_TYPE_O_3_MINI: ModelType
41
43
  MODEL_TYPE_CLAUDE_3_5_SONNET: ModelType
42
44
  MODEL_TYPE_CLAUDE_3_HAIKU: ModelType
43
45
  MODEL_TYPE_MISTRAL_LARGE: ModelType