corvic-engine 0.3.0rc42__tar.gz → 0.3.0rc43__tar.gz

{corvic_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/Cargo.lock

@@ -22,7 +22,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.42"
+version = "0.3.0-rc.43"
 dependencies = [
  "numpy",
  "pyo3",
@@ -63,7 +63,7 @@ checksum = "60b1af1c220855b6ceac025d3f6ecdd2b7c4894bfe9cd9bda4fbb4bc7c0d4cf0"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.42"
+version = "0.3.0-rc.43"
 dependencies = [
  "version_check",
 ]

{corvic_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/Cargo.toml

@@ -3,7 +3,7 @@ resolver = "2"
 members = ["python"]
 
 [workspace.package]
-version = "0.3.0-rc.42"
+version = "0.3.0-rc.43"
 edition = "2021"
 publish = false
 

{corvic_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/PKG-INFO

@@ -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 → corvic_engine-0.3.0rc43}/pyproject.toml

@@ -20,7 +20,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.42" # sync this with tool.poetry.version below
+version = "0.3.0-rc.43" # sync this with tool.poetry.version below
 
 # TODO(ddn): Pull dependencies from tool.poetry.dependencies. We use maturin as
 # the build system but poetry for the development environment.
@@ -74,7 +74,7 @@ packages = [
 ]
 description = ""
 authors = []
-version = "0.3.0-rc.42"
+version = "0.3.0-rc.43"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"

{corvic_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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.0rc43/python/corvic_generated/model/v1alpha/models_pb2.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: corvic/model/v1alpha/models.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()
+
+
+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
+from corvic_generated.orm.v1 import table_pb2 as corvic_dot_orm_dot_v1_dot_table__pb2
+from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot_event__pb2
+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/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)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.model.v1alpha.models_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._options = None
+  _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=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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc42 → corvic_engine-0.3.0rc43}/python/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_engine-0.3.0rc43/python/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_engine-0.3.0rc43/python/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: ...