corvic-engine 0.3.0rc42__cp38-abi3-win_amd64.whl → 0.3.0rc43__cp38-abi3-win_amd64.whl

corvic/model/_completion_model.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import copy
 import datetime
 from collections.abc import Iterable, Sequence
-from typing import TypeAlias
+from typing import Literal, TypeAlias
 
 from sqlalchemy import orm as sa_orm
 
@@ -19,6 +19,7 @@ from corvic.model._proto_orm_convert import (
 )
 from corvic.result import InvalidArgumentError, NotFoundError, Ok
 from corvic_generated.model.v1alpha import models_pb2
+from corvic_generated.orm.v1 import completion_model_pb2
 
 CompletionModelID: TypeAlias = orm.CompletionModelID
 OrgID: TypeAlias = orm.OrgID
@@ -62,12 +63,46 @@ class CompletionModel(
         return OrgID(self.proto_self.org_id)
 
     @property
-    def model_name(self) -> str:
-        return self.proto_self.model_name
+    def provider(self) -> Literal["openai-generic", "azure-openai"] | None:
+        match self.proto_self.parameters.WhichOneof("params"):
+            case "azure_openai_parameters":
+                return "azure-openai"
+            case "generic_openai_parameters":
+                return "openai-generic"
+            case _:
+                return None
 
     @property
-    def endpoint(self) -> str:
-        return self.proto_self.endpoint
+    def parameters(
+        self,
+    ) -> (
+        completion_model_pb2.AzureOpenAIParameters
+        | completion_model_pb2.GenericOpenAIParameters
+        | None
+    ):
+        match self.provider:
+            case "azure-openai":
+                return self.azure_openai_parameters
+            case "openai-generic":
+                return self.generic_openai_parameters
+            case None:
+                return None
+
+    @property
+    def azure_openai_parameters(
+        self,
+    ) -> completion_model_pb2.AzureOpenAIParameters | None:
+        if self.proto_self.parameters.HasField("azure_openai_parameters"):
+            return self.proto_self.parameters.azure_openai_parameters
+        return None
+
+    @property
+    def generic_openai_parameters(
+        self,
+    ) -> completion_model_pb2.GenericOpenAIParameters | None:
+        if self.proto_self.parameters.HasField("generic_openai_parameters"):
+            return self.proto_self.parameters.generic_openai_parameters
+        return None
 
     @property
     def secret_api_key(self) -> str:
@@ -83,8 +118,7 @@ class CompletionModel(
         *,
         name: str,
         description: str,
-        model_name: str,
-        endpoint: str,
+        parameters: completion_model_pb2.CompletionModelParameters,
         secret_api_key: str,
         client: system.Client | None = None,
     ):
@@ -94,8 +128,7 @@ class CompletionModel(
             models_pb2.CompletionModel(
                 name=name,
                 description=description,
-                model_name=model_name,
-                endpoint=endpoint,
+                parameters=parameters,
                 secret_api_key=secret_api_key,
             ),
         )
@@ -153,14 +186,11 @@ class CompletionModel(
         proto_self.description = description
         return CompletionModel(self.client, proto_self)
 
-    def with_model_name(self, model_name: str) -> CompletionModel:
-        proto_self = copy.deepcopy(self.proto_self)
-        proto_self.model_name = model_name
-        return CompletionModel(self.client, proto_self)
-
-    def with_endpoint(self, endpoint: str) -> CompletionModel:
+    def with_parameters(
+        self, parameters: completion_model_pb2.CompletionModelParameters
+    ) -> CompletionModel:
         proto_self = copy.deepcopy(self.proto_self)
-        proto_self.endpoint = endpoint
+        proto_self.parameters.CopyFrom(parameters)
         return CompletionModel(self.client, proto_self)
 
     def with_secret_api_key(self, secret_api_key: str) -> CompletionModel:

corvic/model/_proto_orm_convert.py

@@ -225,8 +225,7 @@ def completion_model_orm_to_proto(
         name=completion_model_orm.name,
         description=completion_model_orm.description,
         org_id=str(completion_model_orm.org_id),
-        model_name=completion_model_orm.model_name,
-        endpoint=completion_model_orm.endpoint,
+        parameters=completion_model_orm.parameters,
         secret_api_key=completion_model_orm.secret_api_key,
         created_at=timestamp_orm_to_proto(completion_model_orm.created_at),
     )
@@ -543,8 +542,7 @@ def completion_model_proto_to_orm(
         id=ids.obj_id,
         name=proto_obj.name,
         description=proto_obj.description,
-        model_name=proto_obj.model_name,
-        endpoint=proto_obj.endpoint,
+        parameters=proto_obj.parameters,
         secret_api_key=proto_obj.secret_api_key,
     )
 

corvic/model/_space.py

@@ -782,50 +782,67 @@ class TabularSpace(Space):
 
             embedding_column_tmp_name = f"__embed-{uuid.uuid4()}"
 
-            op = (
-                op.concat_list(
-                    column_names=embedding_column_tmp_names,
-                    concat_list_column_name=embedding_column_tmp_name,
-                )
-                .and_then(
-                    lambda t,
-                    embedding_column_name=embedding_column_tmp_name: op_graph.op.coordinates_from_embedding(  # noqa: E501
-                        table=t,
-                        embedding_column_name=embedding_column_name,
-                        output_dims=parameters.ndim,
-                    )
+            # Avoid 0 padding for spaces with small numbers of columns
+            target_list_length = min(parameters.ndim, len(embedding_column_tmp_names))
+
+            def reduce_dimension(
+                op: op_graph.Op,
+                embedding_column_tmp_name=embedding_column_tmp_name,
+                target_list_length=target_list_length,
+            ):
+                return op.truncate_list(
+                    list_column_name=embedding_column_tmp_name,
+                    target_list_length=target_list_length,
+                    padding_value=0,
                 )
-                .and_then(
-                    lambda t,
-                    pk_field=pk_field,
-                    embedding_column_tmp_name=embedding_column_tmp_name,
-                    output_source=output_source: t.select_columns(
-                        [pk_field.name, embedding_column_tmp_name]
-                    )
+
+            def select_columns(
+                op: op_graph.Op,
+                pk_field=pk_field,
+                embedding_column_tmp_name=embedding_column_tmp_name,
+            ):
+                return op.select_columns([pk_field.name, embedding_column_tmp_name])
+
+            def update_feature_types(
+                op: op_graph.Op,
+                embedding_column_tmp_name=embedding_column_tmp_name,
+            ):
+                return op.update_feature_types(
+                    {embedding_column_tmp_name: op_graph.feature_type.embedding()}
                 )
-                .and_then(
-                    lambda t,
-                    embedding_column_tmp_name=embedding_column_tmp_name: t.update_feature_types(  # noqa: E501
-                        {embedding_column_tmp_name: op_graph.feature_type.embedding()}
-                    )
+
+            def rename_columns(
+                op: op_graph.Op,
+                pk_field=pk_field,
+                embedding_column_tmp_name=embedding_column_tmp_name,
+            ):
+                return op.rename_columns(
+                    {
+                        pk_field.name: "entity_id",
+                        embedding_column_tmp_name: "embedding",
+                    }
                 )
-                .and_then(
-                    lambda t,
-                    pk_field=pk_field,
-                    embedding_column_tmp_name=embedding_column_tmp_name: t.rename_columns(  # noqa: E501
-                        {
-                            pk_field.name: "entity_id",
-                            embedding_column_tmp_name: "embedding",
-                        }
-                    )
+
+            def add_literal_column(
+                op: op_graph.Op,
+                output_source=output_source,
+            ):
+                return op.add_literal_column(
+                    "source_id",
+                    str(output_source.id),
+                    pa.string(),
                 )
-                .and_then(
-                    lambda t, output_source=output_source: t.add_literal_column(
-                        "source_id",
-                        str(output_source.id),
-                        pa.string(),
-                    )
+
+            op = (
+                op.concat_list(
+                    column_names=embedding_column_tmp_names,
+                    concat_list_column_name=embedding_column_tmp_name,
                 )
+                .and_then(reduce_dimension)
+                .and_then(select_columns)
+                .and_then(update_feature_types)
+                .and_then(rename_columns)
+                .and_then(add_literal_column)
             )
 
             match op:

corvic/orm/__init__.py

@@ -48,6 +48,7 @@ from corvic.orm.mixins import (
 from corvic_generated.orm.v1 import (
     agent_pb2,
     common_pb2,
+    completion_model_pb2,
     feature_view_pb2,
     pipeline_pb2,
     space_pb2,
@@ -580,8 +581,9 @@ class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base):
     id: sa_orm.Mapped[CompletionModelID | None] = primary_key_identity_column()
     name: sa_orm.Mapped[str] = sa_orm.mapped_column(sa.Text, default=None)
     description: sa_orm.Mapped[str] = sa_orm.mapped_column(sa.Text, default=None)
-    model_name: sa_orm.Mapped[str] = sa_orm.mapped_column(sa.Text, default=None)
-    endpoint: sa_orm.Mapped[str] = sa_orm.mapped_column(sa.Text, default=None)
+    parameters: sa_orm.Mapped[completion_model_pb2.CompletionModelParameters | None] = (
+        sa_orm.mapped_column(default=None)
+    )
     secret_api_key: sa_orm.Mapped[str] = sa_orm.mapped_column(sa.Text, default=None)
 
     @property

corvic/orm/base.py

@@ -41,6 +41,7 @@ from corvic.orm.keys import (
 from corvic_generated.orm.v1 import (
     agent_pb2,
     common_pb2,
+    completion_model_pb2,
     feature_view_pb2,
     pipeline_pb2,
     space_pb2,
@@ -112,6 +113,9 @@ class Base(sa_orm.MappedAsDataclass, sa_orm.DeclarativeBase):
             pipeline_pb2.PipelineTransformation()
         ),
         event_pb2.Event: ProtoMessageDecorator(event_pb2.Event()),
+        completion_model_pb2.CompletionModelParameters: ProtoMessageDecorator(
+            completion_model_pb2.CompletionModelParameters()
+        ),
         # ID types
         OrgID: StrIDDecorator(OrgID()),
         RoomID: IntIDDecorator(RoomID()),

corvic/system/_dimension_reduction.py

@@ -3,11 +3,30 @@ from typing import Any, Protocol
 import numpy as np
 from numpy.typing import NDArray
 
+from corvic.result import InvalidArgumentError, Ok
+
+
+def _validate_embedding_array(
+    embeddings: NDArray[Any],
+) -> Ok[None] | InvalidArgumentError:
+    embeddings_ndim = 2
+    if embeddings.ndim != embeddings_ndim:
+        return InvalidArgumentError(
+            f"embeddings ndim must be {embeddings_ndim}",
+            ndim=embeddings.ndim,
+        )
+    if not np.issubdtype(embeddings.dtype, np.number):
+        return InvalidArgumentError(
+            "embeddings must have a numerical dtype",
+            dtype=str(embeddings.dtype),
+        )
+    return Ok(None)
+
 
 class DimensionReducer(Protocol):
     def reduce_dimensions(
         self, vectors: NDArray[Any], output_dimensions: int, metric: str
-    ) -> NDArray[Any]: ...
+    ) -> Ok[NDArray[Any]] | InvalidArgumentError: ...
 
 
 class UmapDimensionReducer(DimensionReducer):
@@ -16,8 +35,16 @@ class UmapDimensionReducer(DimensionReducer):
         vectors: NDArray[Any],
         output_dimensions: int,
         metric: str,
-    ):
+    ) -> Ok[NDArray[Any]] | InvalidArgumentError:
+        match _validate_embedding_array(vectors):
+            case InvalidArgumentError() as err:
+                return err
+            case Ok():
+                pass
+
         vectors = np.nan_to_num(vectors.astype(np.float32))
+        if vectors.shape[1] == output_dimensions:
+            return Ok(vectors)
         n_neighbors = 15
         init = "spectral"
         # y spectral initialization cannot be used when n_neighbors
@@ -45,11 +72,16 @@ class UmapDimensionReducer(DimensionReducer):
             low_memory=False,
             verbose=False,
         )
-        return projector.fit_transform(vectors)
+        return Ok(projector.fit_transform(vectors))
 
 
 class TruncateDimensionReducer(DimensionReducer):
     def reduce_dimensions(
         self, vectors: NDArray[Any], output_dimensions: int, metric: str
-    ) -> NDArray[Any]:
-        return vectors[:, :output_dimensions]
+    ) -> Ok[NDArray[Any]] | InvalidArgumentError:
+        match _validate_embedding_array(vectors):
+            case InvalidArgumentError() as err:
+                return err
+            case Ok():
+                pass
+        return Ok(vectors[:, :output_dimensions])

corvic/system/in_memory_executor.py

@@ -583,9 +583,14 @@ class InMemoryExecutor(OpGraphExecutor):
             case InvalidArgumentError() as err:
                 raise err
 
-        coordinates = self._dimension_reducer.reduce_dimensions(
+        match self._dimension_reducer.reduce_dimensions(
             embedding, op.n_components, op.metric
-        )
+        ):
+            case Ok(coordinates):
+                pass
+            case InvalidArgumentError() as err:
+                raise err
+
         coordinates_df = embedding_df.with_columns(
             pl.Series(
                 name=embedding_column_name,
@@ -897,7 +902,7 @@ class InMemoryExecutor(OpGraphExecutor):
 
                     encoder = MaxAbsScaler()
                     encoded = encoder.fit_transform(
-                        to_encode.to_numpy().reshape(-1, 1)
+                        np.nan_to_num(to_encode.to_numpy()).reshape(-1, 1)
                     ).flatten()
 
                 case op_graph.encoder.StandardScaler():

{corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc43.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corvic-engine
-Version: 0.3.0rc42
+Version: 0.3.0rc43
 Classifier: Environment :: Console
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: Implementation :: CPython

{corvic_engine-0.3.0rc42.dist-info → corvic_engine-0.3.0rc43.dist-info}/RECORD

@@ -1,6 +1,6 @@
-corvic_engine-0.3.0rc42.dist-info/METADATA,sha256=NCpmQQXRbq8jopf7efVCe1CLLYKruLYxsGmO9cAiTlk,1895
-corvic_engine-0.3.0rc42.dist-info/WHEEL,sha256=mYDPFrpmO1_UhonyEoMo2i0zHOwqc92hj_Q3CYuEBdM,94
-corvic_engine-0.3.0rc42.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
+corvic_engine-0.3.0rc43.dist-info/METADATA,sha256=dFcSElUKY_9aoHfg-8CN6ppfT6tnIl_5y_FeeL5Np3M,1895
+corvic_engine-0.3.0rc43.dist-info/WHEEL,sha256=mYDPFrpmO1_UhonyEoMo2i0zHOwqc92hj_Q3CYuEBdM,94
+corvic_engine-0.3.0rc43.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
 corvic/context/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/context/__init__.py,sha256=zBnPiP-tStGSVMG_0-G_0ay6-yIX2aerW_oYRzAex74,1702
 corvic/embed/node2vec.py,sha256=JnYb8f2g4XhF6LL2TjpMxLfKhn_Yp1AzptsWwrKQWgc,11146
@@ -15,17 +15,17 @@ corvic/engine/__init__.py,sha256=XL4Vg7rNcBi29ccVelpeFizR9oJtGYXDn84W9zok9d4,975
 corvic/model/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/model/_agent.py,sha256=WGdxu0oLHU7EEgBOMQ0UOnH2AIIUGfeH6VSQZgC0kEk,3621
 corvic/model/_base_model.py,sha256=r6NKFAlTkxQPDF-RcuWoZkoOdrRSZc8iL3B6W8JvmK0,8341
-corvic/model/_completion_model.py,sha256=J8dDy42I66EinqgxjdsGaQu40uKhYYRfPezwh-daNjw,5503
+corvic/model/_completion_model.py,sha256=uoqF7hwxzGXXqSPZT_CIcNBSDmYhcxMotpGPucWH6Q0,6656
 corvic/model/_defaults.py,sha256=yoKPPSmYJCE5YAD5jLTEmT4XNf_zXoggNK-uyG8MfVs,1524
 corvic/model/_errors.py,sha256=Ctlq04SDwHzJPvLaL1rzqzwVqf2b50EILfW3cH4vnh8,261
 corvic/model/_feature_type.py,sha256=Y-_-wa9fv7XaCAkxfjjoCLxxK2Ftfba-PMefD7bNXzs,917
 corvic/model/_feature_view.py,sha256=SW-BOCWpW9GXkFrC5b_JfY8AUAKK_q-BpHYFpkkTXd4,49894
 corvic/model/_pipeline.py,sha256=IEUaMX8XYDH-qx5336X5MM10F-h0nHoNNnM86b8sBqA,18193
-corvic/model/_proto_orm_convert.py,sha256=hHtWyYwFkhel65wtoVRve9zsdFvxc84ShaDQh98epzs,23783
+corvic/model/_proto_orm_convert.py,sha256=dU292sUqBOdb_nBgxtQXk1VcHc4ttWUoHSwLrjjtL8A,23696
 corvic/model/_resource.py,sha256=UFff-ZGjSnIo810kELRZ9NiA4yXETIrsxjyMId9sEmc,7304
 corvic/model/_room.py,sha256=57MiBfj8hZcmUfq2PeECrOWDpBZAOSjnVqNUIGXOy2Q,2898
 corvic/model/_source.py,sha256=pv2vsjZNuc4KAqy39pCN0tK6kvtowODQhIeUfvS0ek0,9526
-corvic/model/_space.py,sha256=Vu4YEghLvf6O1k_wi87nfDo7WvAtII5DDmwj7VpifNQ,36224
+corvic/model/_space.py,sha256=1tyCFz_AvBgYffA00v17ubJxXMwe8tU-sKTasCSg9eQ,36632
 corvic/model/__init__.py,sha256=_Hjo5INX0urDsUZPlUtnwvODNkZ7H0azMS0RXCS3MZI,2360
 corvic/op_graph/aggregation.py,sha256=8X6vqXD7dLHrhYJU0BqmhUsWGbzD1zSP5Db5VHdIru4,6187
 corvic/op_graph/encoders.py,sha256=EhEmAiwgnXNiJ8NU0xm4deC7EZm80UzuzBL5MV140LQ,9217
@@ -40,7 +40,7 @@ corvic/op_graph/sample_strategy.py,sha256=DrbtJ3ORkIRfyIE_FdlOh_UMnCW_K9jL1LeonV
 corvic/op_graph/_schema.py,sha256=STbxY5PIqIA6xkSDeK8k72Nutsxq5jGe7e_aT35aznI,5733
 corvic/op_graph/_transformations.py,sha256=L9Au_GcciPynww4ZXojMtNdPJ36Qboc9gn0bVzXLifU,9445
 corvic/op_graph/__init__.py,sha256=1DMrQfuuS3FkLa9DXYDjSDLurdxxpG5H1jB2ctaa9xo,1444
-corvic/orm/base.py,sha256=0GP9zwf7I1rZjGO2MdE3l14DbrbOlXWlp-qIevaNWJM,8636
+corvic/orm/base.py,sha256=2LJh6gRZXpQwV50hquGIGkgqRsIq_pew1djgGd7umW0,8817
 corvic/orm/errors.py,sha256=uFhFXpVG6pby1lndJZHGHxv3Y0Fbt0RiaZ-CqDfuY1o,545
 corvic/orm/func/utc_func.py,sha256=-FC6w9wBWXejMv1AICT2Gg7tdkSo7gqL2dFT-YKPGQ4,4518
 corvic/orm/func/uuid_func.py,sha256=oXPjDGAl3mvlNtvcvBrLmRRHPJgtKffShIPbHm-EswA,1152
@@ -50,7 +50,7 @@ corvic/orm/keys.py,sha256=Ag6Xbpvxev-VByT1KJ8ChUn9vKVEzkkMXxrjvtADCtY,2182
 corvic/orm/mixins.py,sha256=HfmzJ7LblHtddbbkDmv7nNWURL87Bnj8NeOnNbfmSN4,17794
 corvic/orm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/orm/_proto_columns.py,sha256=tcOu92UjFJFYZLasS6sWJQBDRK26yrnmpTii_LDY4iw,913
-corvic/orm/__init__.py,sha256=s0X4PcgW82LETtlRAOX7dUl2ajZoCKPQMpo33mWEW1k,21961
+corvic/orm/__init__.py,sha256=VYyTMal77Gp7IcXkk5yCmb5333DavrwP1qokuQKqOeM,21967
 corvic/pa_scalar/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/pa_scalar/_const.py,sha256=1nk6w3Y7crd3J5jSCq7DRVa1lcGk4H1RUr1l4NjnlzE,868
 corvic/pa_scalar/_from_value.py,sha256=fS3TNPcPI3jAKGmcUIhn8rdqdQEAwgTLEneVxFUeK6M,27531
@@ -68,12 +68,12 @@ corvic/sql/parse_ops.py,sha256=1ZXVlDzIzqwW_KP0mwMxaY91tLSXqpeaUHyrGJkh56o,29444
 corvic/sql/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/sql/__init__.py,sha256=kZ1a39KVZ08P8Bg6XuXDLD_dTQX0k620u4nwxZF4SnY,303
 corvic/system/client.py,sha256=hGhZX8RtHrFEOlOmJNlUHktOZrutOwNYUY_a1htQSrg,821
-corvic/system/in_memory_executor.py,sha256=xmb2386RsExMuzCR_XfmJHDdfI2K64v5Z91dbDkhNCk,60803
+corvic/system/in_memory_executor.py,sha256=vX4aGkNHW9gNpH3teMwOxC2Vjhoxc2YBtEXeEU2n4h0,60946
 corvic/system/op_graph_executor.py,sha256=gXFnVkemS5EwNegJdU-xVAfMLPULqMFPF7d3EG3AD_U,3482
 corvic/system/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/system/staging.py,sha256=K5P5moiuAMfPx7lxK4mArxeURBwKoyB6x9HGu9JJ16E,1846
 corvic/system/storage.py,sha256=y7NKArf9104S6muHpxDXL8uNp7toFBXRQlikNAfdR4Q,5644
-corvic/system/_dimension_reduction.py,sha256=r-IlkZR7qnEweO9VjNeovtMmuY1gfXCPe50ChPnc7iw,1774
+corvic/system/_dimension_reduction.py,sha256=vyD8wOs0vE-hlVnCrBTjStTowAPWYREqnQ_bVuGYvis,2907
 corvic/system/_embedder.py,sha256=0WO24IKi8VC8jsFdvNuzDsgNejyacQf6r9Q34jxhHc4,3844
 corvic/system/_image_embedder.py,sha256=iQc3KlLcqrhP6K84hncHutThAN8Qd6K7K5dceHyU1TU,8373
 corvic/system/_planner.py,sha256=HUf6UjCy1iHRrXfhU25w19TG4Ik3zVHhtzVcor0eTQY,7888
@@ -156,14 +156,16 @@ corvic_generated/ingest/v2/table_pb2.py,sha256=aTJHaliZm5DMtp7gslNxyn9uDagz-2-_e
 corvic_generated/ingest/v2/table_pb2_grpc.py,sha256=tVs7wMWyAfvHcCQEiUOHLwaptKxgMFG6E7Ki9vNmmvQ,8151
 corvic_generated/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/model/v1alpha/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-corvic_generated/model/v1alpha/models_pb2.py,sha256=gyINQ-NT_kl_ZMvVqkoZOJOMq_hV6DJOA4S9skrRmC8,10627
+corvic_generated/model/v1alpha/models_pb2.py,sha256=3cBEUrlygqnfeIBsB-hzT7OTxhFd3cb5sTJmWnpotnA,10766
 corvic_generated/model/v1alpha/models_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
 corvic_generated/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/orm/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-corvic_generated/orm/v1/agent_pb2.py,sha256=CRPwvKVzExmz48fg24IzQv_joUiQm8aMd_H-JQW4DE8,3850
+corvic_generated/orm/v1/agent_pb2.py,sha256=YDAzQEHtVlm3f8BW7jEzayHoSCjZjSx0Ar-oRBB0LyQ,3891
 corvic_generated/orm/v1/agent_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
 corvic_generated/orm/v1/common_pb2.py,sha256=pu1Rto4Pm8ntdB12Rgb_qnz3qOalxlOnrLuKE1fCETs,3490
 corvic_generated/orm/v1/common_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
+corvic_generated/orm/v1/completion_model_pb2.py,sha256=aQOc_42UHONp-7O0H9ftiuxTbAzfNnjk-3jwtP9dKyA,2081
+corvic_generated/orm/v1/completion_model_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
 corvic_generated/orm/v1/feature_view_pb2.py,sha256=EnPKwNSmaJxbEcqFk2vVzW3Rsr2eBpV49IC71HA7v2w,3012
 corvic_generated/orm/v1/feature_view_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
 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
 corvic_generated/orm/v1/table_pb2_grpc.py,sha256=ixBOrA7wwNxEQCRT1kO2N_LayeFYEdFJjVRkkhesWbY,4558
 corvic_generated/platform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/platform/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-corvic_generated/platform/v1/platform_pb2.py,sha256=lNbPCY-tZYIfl_ZTuMaPgkL8m-nJyn-9K4ghUH529XM,9224
-corvic_generated/platform/v1/platform_pb2_grpc.py,sha256=jdPQAgEHaTrD7JTDyjUl8ND1ox1lnjNSEsYq6AwYehs,13407
+corvic_generated/platform/v1/platform_pb2.py,sha256=w19YYczRJWAhx34JLQRqDUnP2MEWmuh8VsItGubn2A0,9844
+corvic_generated/platform/v1/platform_pb2_grpc.py,sha256=lpPzShz0YAgQ2GSUwIZqM6_BEPkx5cHc25qxpEXlX0s,15145
 corvic_generated/query/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/query/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic_generated/query/v1/query_pb2.py,sha256=cRHixYUF9eihJ2GGL9r3Ny49cfsK_zWsBCnLv23JzYs,2831
@@ -218,12 +220,14 @@ corvic_generated/ingest/v2/source_pb2.pyi,sha256=k7FdbgurQLk0JA1WiTUerznzxLv8b50
 corvic_generated/ingest/v2/source_pb2_grpc.pyi,sha256=VG5gpql2SREHgqMC_ycT-QJBVpPeSYKOYS2COgGrZa4,6195
 corvic_generated/ingest/v2/table_pb2.pyi,sha256=p22F8kv0HfM-9OzGP88bLofxmUtxfLR5eVN0HOxXiEo,4382
 corvic_generated/ingest/v2/table_pb2_grpc.pyi,sha256=AEXYNtrU4xyENumcCrkD2FmFV7T1UVidxxeZ5pyE4Qc,4554
-corvic_generated/model/v1alpha/models_pb2.pyi,sha256=sXptdQj57MRdg280qszxpq8E2-eWT5UZwkE2k_3EH_w,14158
+corvic_generated/model/v1alpha/models_pb2.pyi,sha256=e2OJo0zeB4lL-tUQlbpAXwSw9BAl__2fk9f63o178k0,14241
 corvic_generated/model/v1alpha/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
-corvic_generated/orm/v1/agent_pb2.pyi,sha256=2zqfoP9Gm1kKBc1FtqzmtZKUZYhFsCDtQ-Pgr0HfE1c,4496
+corvic_generated/orm/v1/agent_pb2.pyi,sha256=9AExLKFRvOJ1fSaOdZc3Src015uvNl-le2lSPatZjSQ,4575
 corvic_generated/orm/v1/agent_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/orm/v1/common_pb2.pyi,sha256=PZi9KfnXWzlQC0bMuDC71IOmgIaYClY2jPghzKV-mAo,3557
 corvic_generated/orm/v1/common_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
+corvic_generated/orm/v1/completion_model_pb2.pyi,sha256=wJagec60xt2cl1tqLu8ZKNliEh6lfbBPojzIOa38fd4,1718
+corvic_generated/orm/v1/completion_model_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/orm/v1/feature_view_pb2.pyi,sha256=MDvBZysNPr6q8TfCVdPT2Bz-IVSf3Tcq2k957KHAtyA,2671
 corvic_generated/orm/v1/feature_view_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 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
 corvic_generated/orm/v1/space_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/orm/v1/table_pb2.pyi,sha256=qUtOna4mJQXcf2Za0buDQHHJPdtvoevjAtfyesExIwE,49683
 corvic_generated/orm/v1/table_pb2_grpc.pyi,sha256=K4hyNndkiKpxt9PYxcn_98RTpb4yxET3Um2rDe3VJTI,2499
-corvic_generated/platform/v1/platform_pb2.pyi,sha256=pz29JG9WNwVuppIRSCWxWo7B8zoyAE4Qqph-uPLKEUE,5001
-corvic_generated/platform/v1/platform_pb2_grpc.pyi,sha256=6rmhbov-ZGD8BLFDYdBPLQoV40Z0sbtDB6eRgRJtato,6064
+corvic_generated/platform/v1/platform_pb2.pyi,sha256=MBcHInUjm1uytkGkQs6edEgcqFnbyT928nYQMauRJsQ,5293
+corvic_generated/platform/v1/platform_pb2_grpc.pyi,sha256=XSkcFXnMIdK2lcBu8axH95BOTM9j19P8qVJUg5KRjqE,6793
 corvic_generated/query/v1/query_pb2.pyi,sha256=scOTjQSNmG1mPrgx9cMN8anovQMpfJDSJnoMN0P8J1s,2972
 corvic_generated/query/v1/query_pb2_grpc.pyi,sha256=Uy_oO0HSK_QlJsjneYktgFi67gMcq9GtEsM5I1CoPy8,1494
 corvic_generated/status/v1/event_pb2.pyi,sha256=eU-ibrYpvEAJSIDlSa62-bC96AQU1ykFi3_sQ1RzBvo,2027
 corvic_generated/status/v1/event_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/status/v1/service_pb2.pyi,sha256=iXLR2FOKQJpBgvBzpD2kVwcYOCksP2aRwK4JYaI9CBw,558
 corvic_generated/status/v1/service_pb2_grpc.pyi,sha256=OoAnaZ64FD0UTzPoRhYvQU8ecoilhHj3ySjSfHbVDaU,1501
-corvic/engine/_native.pyd,sha256=i4tlkC-v-gjeqMp1z3zPjxGqCn8gsHYNor63JfCD5Xc,438272
-corvic_engine-0.3.0rc42.dist-info/RECORD,,
+corvic/engine/_native.pyd,sha256=JQnE7t3U1UEeA9I7Li0fb6Tz5f07Fw5yZuJwe6gdevY,438272
+corvic_engine-0.3.0rc43.dist-info/RECORD,,

corvic_generated/model/v1alpha/models_pb2.py

@@ -14,6 +14,7 @@ _sym_db = _symbol_database.Default()
 
 from corvic_generated.orm.v1 import agent_pb2 as corvic_dot_orm_dot_v1_dot_agent__pb2
 from corvic_generated.orm.v1 import common_pb2 as corvic_dot_orm_dot_v1_dot_common__pb2
+from corvic_generated.orm.v1 import completion_model_pb2 as corvic_dot_orm_dot_v1_dot_completion__model__pb2
 from corvic_generated.orm.v1 import feature_view_pb2 as corvic_dot_orm_dot_v1_dot_feature__view__pb2
 from corvic_generated.orm.v1 import pipeline_pb2 as corvic_dot_orm_dot_v1_dot_pipeline__pb2
 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
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-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')
+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\"\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\"\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')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -34,34 +35,34 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_options = b'8\001'
   _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
   _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
-  _globals['_MESSAGEREACTION']._serialized_start=4214
-  _globals['_MESSAGEREACTION']._serialized_end=4331
-  _globals['_ROOM']._serialized_start=326
-  _globals['_ROOM']._serialized_end=470
-  _globals['_RESOURCE']._serialized_start=473
-  _globals['_RESOURCE']._serialized_end=928
-  _globals['_SOURCE']._serialized_start=931
-  _globals['_SOURCE']._serialized_end=1237
-  _globals['_PIPELINE']._serialized_start=1240
-  _globals['_PIPELINE']._serialized_end=1921
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1713
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1810
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1812
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1906
-  _globals['_FEATUREVIEWSOURCE']._serialized_start=1924
-  _globals['_FEATUREVIEWSOURCE']._serialized_end=2229
-  _globals['_FEATUREVIEW']._serialized_start=2232
-  _globals['_FEATUREVIEW']._serialized_end=2644
-  _globals['_SPACE']._serialized_start=2647
-  _globals['_SPACE']._serialized_end=2995
-  _globals['_USERMESSAGE']._serialized_start=2997
-  _globals['_USERMESSAGE']._serialized_end=3052
-  _globals['_AGENTMESSAGE']._serialized_start=3055
-  _globals['_AGENTMESSAGE']._serialized_end=3389
-  _globals['_MESSAGEENTRY']._serialized_start=3392
-  _globals['_MESSAGEENTRY']._serialized_end=3659
-  _globals['_AGENT']._serialized_start=3662
-  _globals['_AGENT']._serialized_end=3923
-  _globals['_COMPLETIONMODEL']._serialized_start=3926
-  _globals['_COMPLETIONMODEL']._serialized_end=4212
+  _globals['_MESSAGEREACTION']._serialized_start=4267
+  _globals['_MESSAGEREACTION']._serialized_end=4384
+  _globals['_ROOM']._serialized_start=364
+  _globals['_ROOM']._serialized_end=508
+  _globals['_RESOURCE']._serialized_start=511
+  _globals['_RESOURCE']._serialized_end=966
+  _globals['_SOURCE']._serialized_start=969
+  _globals['_SOURCE']._serialized_end=1275
+  _globals['_PIPELINE']._serialized_start=1278
+  _globals['_PIPELINE']._serialized_end=1959
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1751
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1848
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1850
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1944
+  _globals['_FEATUREVIEWSOURCE']._serialized_start=1962
+  _globals['_FEATUREVIEWSOURCE']._serialized_end=2267
+  _globals['_FEATUREVIEW']._serialized_start=2270
+  _globals['_FEATUREVIEW']._serialized_end=2682
+  _globals['_SPACE']._serialized_start=2685
+  _globals['_SPACE']._serialized_end=3033
+  _globals['_USERMESSAGE']._serialized_start=3035
+  _globals['_USERMESSAGE']._serialized_end=3090
+  _globals['_AGENTMESSAGE']._serialized_start=3093
+  _globals['_AGENTMESSAGE']._serialized_end=3427
+  _globals['_MESSAGEENTRY']._serialized_start=3430
+  _globals['_MESSAGEENTRY']._serialized_end=3697
+  _globals['_AGENT']._serialized_start=3700
+  _globals['_AGENT']._serialized_end=3961
+  _globals['_COMPLETIONMODEL']._serialized_start=3964
+  _globals['_COMPLETIONMODEL']._serialized_end=4265
 # @@protoc_insertion_point(module_scope)

corvic_generated/model/v1alpha/models_pb2.pyi

@@ -1,5 +1,6 @@
 from corvic_generated.orm.v1 import agent_pb2 as _agent_pb2
 from corvic_generated.orm.v1 import common_pb2 as _common_pb2
+from corvic_generated.orm.v1 import completion_model_pb2 as _completion_model_pb2
 from corvic_generated.orm.v1 import feature_view_pb2 as _feature_view_pb2
 from corvic_generated.orm.v1 import pipeline_pb2 as _pipeline_pb2
 from corvic_generated.orm.v1 import space_pb2 as _space_pb2
@@ -239,21 +240,19 @@ class Agent(_message.Message):
     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: ...
 
 class CompletionModel(_message.Message):
-    __slots__ = ("id", "name", "description", "org_id", "model_name", "endpoint", "secret_api_key", "created_at")
+    __slots__ = ("id", "name", "description", "org_id", "parameters", "secret_api_key", "created_at")
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
     ORG_ID_FIELD_NUMBER: _ClassVar[int]
-    MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
-    ENDPOINT_FIELD_NUMBER: _ClassVar[int]
+    PARAMETERS_FIELD_NUMBER: _ClassVar[int]
     SECRET_API_KEY_FIELD_NUMBER: _ClassVar[int]
     CREATED_AT_FIELD_NUMBER: _ClassVar[int]
     id: str
     name: str
     description: str
     org_id: str
-    model_name: str
-    endpoint: str
+    parameters: _completion_model_pb2.CompletionModelParameters
     secret_api_key: str
     created_at: _timestamp_pb2.Timestamp
-    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: ...
+    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

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-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')
+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')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -24,7 +24,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_ORCHESTRATORPARAMETERS_SPACEINSTRUCTIONSENTRY']._options = None
   _globals['_ORCHESTRATORPARAMETERS_SPACEINSTRUCTIONSENTRY']._serialized_options = b'8\001'
   _globals['_MODELTYPE']._serialized_start=847
-  _globals['_MODELTYPE']._serialized_end=1430
+  _globals['_MODELTYPE']._serialized_end=1455
   _globals['_EXECUTORPARAMETERS']._serialized_start=45
   _globals['_EXECUTORPARAMETERS']._serialized_end=288
   _globals['_SPACEINSTRUCTION']._serialized_start=290

corvic_generated/orm/v1/agent_pb2.pyi

@@ -20,6 +20,7 @@ class ModelType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     MODEL_TYPE_GPT_3_5_TURBO: _ClassVar[ModelType]
     MODEL_TYPE_O_1: _ClassVar[ModelType]
     MODEL_TYPE_O_1_MINI: _ClassVar[ModelType]
+    MODEL_TYPE_O_3_MINI: _ClassVar[ModelType]
     MODEL_TYPE_CLAUDE_3_5_SONNET: _ClassVar[ModelType]
     MODEL_TYPE_CLAUDE_3_HAIKU: _ClassVar[ModelType]
     MODEL_TYPE_MISTRAL_LARGE: _ClassVar[ModelType]
@@ -38,6 +39,7 @@ MODEL_TYPE_GPT_4_O: ModelType
 MODEL_TYPE_GPT_3_5_TURBO: ModelType
 MODEL_TYPE_O_1: ModelType
 MODEL_TYPE_O_1_MINI: ModelType
+MODEL_TYPE_O_3_MINI: ModelType
 MODEL_TYPE_CLAUDE_3_5_SONNET: ModelType
 MODEL_TYPE_CLAUDE_3_HAIKU: ModelType
 MODEL_TYPE_MISTRAL_LARGE: ModelType

corvic_generated/orm/v1/completion_model_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: corvic/orm/v1/completion_model.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$corvic/orm/v1/completion_model.proto\x12\rcorvic.orm.v1\"S\n\x17GenericOpenAIParameters\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x19\n\x08\x62\x61se_url\x18\x02 \x01(\tR\x07\x62\x61seUrl\"\x9d\x01\n\x15\x41zureOpenAIParameters\x12\x19\n\x08\x62\x61se_url\x18\x01 \x01(\tR\x07\x62\x61seUrl\x12#\n\rresource_name\x18\x02 \x01(\tR\x0cresourceName\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId\x12\x1f\n\x0b\x61pi_version\x18\x04 \x01(\tR\napiVersion\"\xeb\x01\n\x19\x43ompletionModelParameters\x12\x64\n\x19generic_openai_parameters\x18\x01 \x01(\x0b\x32&.corvic.orm.v1.GenericOpenAIParametersH\x00R\x17genericOpenaiParameters\x12^\n\x17\x61zure_openai_parameters\x18\x02 \x01(\x0b\x32$.corvic.orm.v1.AzureOpenAIParametersH\x00R\x15\x61zureOpenaiParametersB\x08\n\x06paramsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.orm.v1.completion_model_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_GENERICOPENAIPARAMETERS']._serialized_start=55
+  _globals['_GENERICOPENAIPARAMETERS']._serialized_end=138
+  _globals['_AZUREOPENAIPARAMETERS']._serialized_start=141
+  _globals['_AZUREOPENAIPARAMETERS']._serialized_end=298
+  _globals['_COMPLETIONMODELPARAMETERS']._serialized_start=301
+  _globals['_COMPLETIONMODELPARAMETERS']._serialized_end=536
+# @@protoc_insertion_point(module_scope)

corvic_generated/orm/v1/completion_model_pb2.pyi

@@ -0,0 +1,33 @@
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class GenericOpenAIParameters(_message.Message):
+    __slots__ = ("model_name", "base_url")
+    MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
+    BASE_URL_FIELD_NUMBER: _ClassVar[int]
+    model_name: str
+    base_url: str
+    def __init__(self, model_name: _Optional[str] = ..., base_url: _Optional[str] = ...) -> None: ...
+
+class AzureOpenAIParameters(_message.Message):
+    __slots__ = ("base_url", "resource_name", "deployment_id", "api_version")
+    BASE_URL_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_NAME_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    API_VERSION_FIELD_NUMBER: _ClassVar[int]
+    base_url: str
+    resource_name: str
+    deployment_id: str
+    api_version: str
+    def __init__(self, base_url: _Optional[str] = ..., resource_name: _Optional[str] = ..., deployment_id: _Optional[str] = ..., api_version: _Optional[str] = ...) -> None: ...
+
+class CompletionModelParameters(_message.Message):
+    __slots__ = ("generic_openai_parameters", "azure_openai_parameters")
+    GENERIC_OPENAI_PARAMETERS_FIELD_NUMBER: _ClassVar[int]
+    AZURE_OPENAI_PARAMETERS_FIELD_NUMBER: _ClassVar[int]
+    generic_openai_parameters: GenericOpenAIParameters
+    azure_openai_parameters: AzureOpenAIParameters
+    def __init__(self, generic_openai_parameters: _Optional[_Union[GenericOpenAIParameters, _Mapping]] = ..., azure_openai_parameters: _Optional[_Union[AzureOpenAIParameters, _Mapping]] = ...) -> None: ...

corvic_generated/orm/v1/completion_model_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

corvic_generated/orm/v1/completion_model_pb2_grpc.pyi

@@ -0,0 +1,17 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import abc
+import collections.abc
+import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar('_T')
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta):
+    ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore
+    ...

corvic_generated/platform/v1/platform_pb2.py

@@ -17,7 +17,7 @@ from google.protobuf import descriptor_pb2 as google_dot_protobuf_dot_descriptor
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/platform/v1/platform.proto\x12\x12\x63orvic.platform.v1\x1a\x1b\x62uf/validate/validate.proto\x1a google/protobuf/descriptor.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x8c\x02\n\x03Org\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\x12p\n\x02id\x18\x02 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\x12\x39\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12;\n\x0bmodified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nmodifiedAt\"\xc8\x01\n\x04User\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\x12\"\n\x08org_name\x18\x02 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\x12\x14\n\x05\x65mail\x18\x03 \x01(\tR\x05\x65mail\x12\x14\n\x05roles\x18\x04 \x03(\tR\x05roles\"/\n\x10\x43reateOrgRequest\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\">\n\x11\x43reateOrgResponse\x12)\n\x03org\x18\x01 \x01(\x0b\x32\x17.corvic.platform.v1.OrgR\x03org\"#\n\rGetOrgRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\";\n\x0eGetOrgResponse\x12)\n\x03org\x18\x01 \x01(\x0b\x32\x17.corvic.platform.v1.OrgR\x03org\"\x11\n\x0fListOrgsRequest\"?\n\x10ListOrgsResponse\x12+\n\x04orgs\x18\x01 \x03(\x0b\x32\x17.corvic.platform.v1.OrgR\x04orgs\"9\n\x13ListOrgUsersRequest\x12\"\n\x08org_name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\"F\n\x14ListOrgUsersResponse\x12.\n\x05users\x18\x01 \x03(\x0b\x32\x18.corvic.platform.v1.UserR\x05users\"}\n\x11\x41\x64\x64OrgUserRequest\x12\x14\n\x05\x65mail\x18\x01 \x01(\tR\x05\x65mail\x12\"\n\x08org_name\x18\x02 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\x12.\n\x05roles\x18\x03 \x03(\x0e\x32\x18.corvic.platform.v1.RoleR\x05roles\"B\n\x12\x41\x64\x64OrgUserResponse\x12,\n\x04user\x18\x01 \x01(\x0b\x32\x18.corvic.platform.v1.UserR\x04user\"\x85\x01\n\x11GetOrgUserRequest\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\"B\n\x12GetOrgUserResponse\x12,\n\x04user\x18\x01 \x01(\x0b\x32\x18.corvic.platform.v1.UserR\x04user\"\x8c\x01\n\x18\x44\x65\x61\x63tivateOrgUserRequest\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\"\x1b\n\x19\x44\x65\x61\x63tivateOrgUserResponse*C\n\x04Role\x12\x14\n\x10ROLE_UNSPECIFIED\x10\x00\x12\x12\n\x0eROLE_ORG_ADMIN\x10\x01\x12\x11\n\rROLE_ORG_USER\x10\x02\x32\xb7\x05\n\nOrgService\x12T\n\x06GetOrg\x12!.corvic.platform.v1.GetOrgRequest\x1a\".corvic.platform.v1.GetOrgResponse\"\x03\x90\x02\x01\x12Z\n\tCreateOrg\x12$.corvic.platform.v1.CreateOrgRequest\x1a%.corvic.platform.v1.CreateOrgResponse\"\x00\x12Z\n\x08ListOrgs\x12#.corvic.platform.v1.ListOrgsRequest\x1a$.corvic.platform.v1.ListOrgsResponse\"\x03\x90\x02\x01\x12`\n\nGetOrgUser\x12%.corvic.platform.v1.GetOrgUserRequest\x1a&.corvic.platform.v1.GetOrgUserResponse\"\x03\x90\x02\x01\x12]\n\nAddOrgUser\x12%.corvic.platform.v1.AddOrgUserRequest\x1a&.corvic.platform.v1.AddOrgUserResponse\"\x00\x12\x66\n\x0cListOrgUsers\x12\'.corvic.platform.v1.ListOrgUsersRequest\x1a(.corvic.platform.v1.ListOrgUsersResponse\"\x03\x90\x02\x01\x12r\n\x11\x44\x65\x61\x63tivateOrgUser\x12,.corvic.platform.v1.DeactivateOrgUserRequest\x1a-.corvic.platform.v1.DeactivateOrgUserResponse\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/platform/v1/platform.proto\x12\x12\x63orvic.platform.v1\x1a\x1b\x62uf/validate/validate.proto\x1a google/protobuf/descriptor.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x8c\x02\n\x03Org\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\x12p\n\x02id\x18\x02 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\x12\x39\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12;\n\x0bmodified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nmodifiedAt\"\xc8\x01\n\x04User\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\x12\"\n\x08org_name\x18\x02 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\x12\x14\n\x05\x65mail\x18\x03 \x01(\tR\x05\x65mail\x12\x14\n\x05roles\x18\x04 \x03(\tR\x05roles\"/\n\x10\x43reateOrgRequest\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\">\n\x11\x43reateOrgResponse\x12)\n\x03org\x18\x01 \x01(\x0b\x32\x17.corvic.platform.v1.OrgR\x03org\"#\n\rGetOrgRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\";\n\x0eGetOrgResponse\x12)\n\x03org\x18\x01 \x01(\x0b\x32\x17.corvic.platform.v1.OrgR\x03org\"\x11\n\x0fListOrgsRequest\"?\n\x10ListOrgsResponse\x12+\n\x04orgs\x18\x01 \x03(\x0b\x32\x17.corvic.platform.v1.OrgR\x04orgs\"9\n\x13ListOrgUsersRequest\x12\"\n\x08org_name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\"F\n\x14ListOrgUsersResponse\x12.\n\x05users\x18\x01 \x03(\x0b\x32\x18.corvic.platform.v1.UserR\x05users\"}\n\x11\x41\x64\x64OrgUserRequest\x12\x14\n\x05\x65mail\x18\x01 \x01(\tR\x05\x65mail\x12\"\n\x08org_name\x18\x02 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\x12.\n\x05roles\x18\x03 \x03(\x0e\x32\x18.corvic.platform.v1.RoleR\x05roles\"B\n\x12\x41\x64\x64OrgUserResponse\x12,\n\x04user\x18\x01 \x01(\x0b\x32\x18.corvic.platform.v1.UserR\x04user\"\x85\x01\n\x11GetOrgUserRequest\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\"B\n\x12GetOrgUserResponse\x12,\n\x04user\x18\x01 \x01(\x0b\x32\x18.corvic.platform.v1.UserR\x04user\"\x8c\x01\n\x18\x44\x65\x61\x63tivateOrgUserRequest\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\"\x1b\n\x19\x44\x65\x61\x63tivateOrgUserResponse\".\n\x0fPurgeOrgRequest\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\"\x12\n\x10PurgeOrgResponse*C\n\x04Role\x12\x14\n\x10ROLE_UNSPECIFIED\x10\x00\x12\x12\n\x0eROLE_ORG_ADMIN\x10\x01\x12\x11\n\rROLE_ORG_USER\x10\x02\x32\x90\x06\n\nOrgService\x12T\n\x06GetOrg\x12!.corvic.platform.v1.GetOrgRequest\x1a\".corvic.platform.v1.GetOrgResponse\"\x03\x90\x02\x01\x12Z\n\tCreateOrg\x12$.corvic.platform.v1.CreateOrgRequest\x1a%.corvic.platform.v1.CreateOrgResponse\"\x00\x12Z\n\x08ListOrgs\x12#.corvic.platform.v1.ListOrgsRequest\x1a$.corvic.platform.v1.ListOrgsResponse\"\x03\x90\x02\x01\x12`\n\nGetOrgUser\x12%.corvic.platform.v1.GetOrgUserRequest\x1a&.corvic.platform.v1.GetOrgUserResponse\"\x03\x90\x02\x01\x12]\n\nAddOrgUser\x12%.corvic.platform.v1.AddOrgUserRequest\x1a&.corvic.platform.v1.AddOrgUserResponse\"\x00\x12\x66\n\x0cListOrgUsers\x12\'.corvic.platform.v1.ListOrgUsersRequest\x1a(.corvic.platform.v1.ListOrgUsersResponse\"\x03\x90\x02\x01\x12r\n\x11\x44\x65\x61\x63tivateOrgUser\x12,.corvic.platform.v1.DeactivateOrgUserRequest\x1a-.corvic.platform.v1.DeactivateOrgUserResponse\"\x00\x12W\n\x08PurgeOrg\x12#.corvic.platform.v1.PurgeOrgRequest\x1a$.corvic.platform.v1.PurgeOrgResponse\"\x00\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -42,6 +42,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_GETORGUSERREQUEST'].fields_by_name['id']._serialized_options = b'\272H]\272\001Z\n\016string.pattern\022\026value must be a number\0320this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')'
   _globals['_DEACTIVATEORGUSERREQUEST'].fields_by_name['id']._options = None
   _globals['_DEACTIVATEORGUSERREQUEST'].fields_by_name['id']._serialized_options = b'\272H]\272\001Z\n\016string.pattern\022\026value must be a number\0320this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')'
+  _globals['_PURGEORGREQUEST'].fields_by_name['name']._options = None
+  _globals['_PURGEORGREQUEST'].fields_by_name['name']._serialized_options = b'\272H\004r\002\020\001'
   _globals['_ORGSERVICE'].methods_by_name['GetOrg']._options = None
   _globals['_ORGSERVICE'].methods_by_name['GetOrg']._serialized_options = b'\220\002\001'
   _globals['_ORGSERVICE'].methods_by_name['ListOrgs']._options = None
@@ -50,8 +52,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_ORGSERVICE'].methods_by_name['GetOrgUser']._serialized_options = b'\220\002\001'
   _globals['_ORGSERVICE'].methods_by_name['ListOrgUsers']._options = None
   _globals['_ORGSERVICE'].methods_by_name['ListOrgUsers']._serialized_options = b'\220\002\001'
-  _globals['_ROLE']._serialized_start=1624
-  _globals['_ROLE']._serialized_end=1691
+  _globals['_ROLE']._serialized_start=1692
+  _globals['_ROLE']._serialized_end=1759
   _globals['_ORG']._serialized_start=154
   _globals['_ORG']._serialized_end=422
   _globals['_USER']._serialized_start=425
@@ -84,6 +86,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_DEACTIVATEORGUSERREQUEST']._serialized_end=1593
   _globals['_DEACTIVATEORGUSERRESPONSE']._serialized_start=1595
   _globals['_DEACTIVATEORGUSERRESPONSE']._serialized_end=1622
-  _globals['_ORGSERVICE']._serialized_start=1694
-  _globals['_ORGSERVICE']._serialized_end=2389
+  _globals['_PURGEORGREQUEST']._serialized_start=1624
+  _globals['_PURGEORGREQUEST']._serialized_end=1670
+  _globals['_PURGEORGRESPONSE']._serialized_start=1672
+  _globals['_PURGEORGRESPONSE']._serialized_end=1690
+  _globals['_ORGSERVICE']._serialized_start=1762
+  _globals['_ORGSERVICE']._serialized_end=2546
 # @@protoc_insertion_point(module_scope)

corvic_generated/platform/v1/platform_pb2.pyi

@@ -125,3 +125,13 @@ class DeactivateOrgUserRequest(_message.Message):
 class DeactivateOrgUserResponse(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
+
+class PurgeOrgRequest(_message.Message):
+    __slots__ = ("name",)
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    def __init__(self, name: _Optional[str] = ...) -> None: ...
+
+class PurgeOrgResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...

corvic_generated/platform/v1/platform_pb2_grpc.py

@@ -49,6 +49,11 @@ class OrgServiceStub(object):
                 request_serializer=corvic_dot_platform_dot_v1_dot_platform__pb2.DeactivateOrgUserRequest.SerializeToString,
                 response_deserializer=corvic_dot_platform_dot_v1_dot_platform__pb2.DeactivateOrgUserResponse.FromString,
                 )
+        self.PurgeOrg = channel.unary_unary(
+                '/corvic.platform.v1.OrgService/PurgeOrg',
+                request_serializer=corvic_dot_platform_dot_v1_dot_platform__pb2.PurgeOrgRequest.SerializeToString,
+                response_deserializer=corvic_dot_platform_dot_v1_dot_platform__pb2.PurgeOrgResponse.FromString,
+                )
 
 
 class OrgServiceServicer(object):
@@ -96,6 +101,12 @@ class OrgServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def PurgeOrg(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_OrgServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -134,6 +145,11 @@ def add_OrgServiceServicer_to_server(servicer, server):
                     request_deserializer=corvic_dot_platform_dot_v1_dot_platform__pb2.DeactivateOrgUserRequest.FromString,
                     response_serializer=corvic_dot_platform_dot_v1_dot_platform__pb2.DeactivateOrgUserResponse.SerializeToString,
             ),
+            'PurgeOrg': grpc.unary_unary_rpc_method_handler(
+                    servicer.PurgeOrg,
+                    request_deserializer=corvic_dot_platform_dot_v1_dot_platform__pb2.PurgeOrgRequest.FromString,
+                    response_serializer=corvic_dot_platform_dot_v1_dot_platform__pb2.PurgeOrgResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'corvic.platform.v1.OrgService', rpc_method_handlers)
@@ -262,3 +278,20 @@ class OrgService(object):
             corvic_dot_platform_dot_v1_dot_platform__pb2.DeactivateOrgUserResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PurgeOrg(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/corvic.platform.v1.OrgService/PurgeOrg',
+            corvic_dot_platform_dot_v1_dot_platform__pb2.PurgeOrgRequest.SerializeToString,
+            corvic_dot_platform_dot_v1_dot_platform__pb2.PurgeOrgResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

corvic_generated/platform/v1/platform_pb2_grpc.pyi

@@ -47,6 +47,10 @@ class OrgServiceStub:
         corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserRequest,
         corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserResponse,
     ]
+    PurgeOrg: grpc.UnaryUnaryMultiCallable[
+        corvic_generated.platform.v1.platform_pb2.PurgeOrgRequest,
+        corvic_generated.platform.v1.platform_pb2.PurgeOrgResponse,
+    ]
 
 class OrgServiceAsyncStub:
     GetOrg: grpc.aio.UnaryUnaryMultiCallable[
@@ -77,6 +81,10 @@ class OrgServiceAsyncStub:
         corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserRequest,
         corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserResponse,
     ]
+    PurgeOrg: grpc.aio.UnaryUnaryMultiCallable[
+        corvic_generated.platform.v1.platform_pb2.PurgeOrgRequest,
+        corvic_generated.platform.v1.platform_pb2.PurgeOrgResponse,
+    ]
 
 class OrgServiceServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
@@ -121,5 +129,11 @@ class OrgServiceServicer(metaclass=abc.ABCMeta):
         request: corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserRequest,
         context: _ServicerContext,
     ) -> typing.Union[corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserResponse, collections.abc.Awaitable[corvic_generated.platform.v1.platform_pb2.DeactivateOrgUserResponse]]: ...
+    @abc.abstractmethod
+    def PurgeOrg(
+        self,
+        request: corvic_generated.platform.v1.platform_pb2.PurgeOrgRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[corvic_generated.platform.v1.platform_pb2.PurgeOrgResponse, collections.abc.Awaitable[corvic_generated.platform.v1.platform_pb2.PurgeOrgResponse]]: ...
 
 def add_OrgServiceServicer_to_server(servicer: OrgServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...