corvic-engine 0.3.0rc45__tar.gz → 0.3.0rc47__tar.gz

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/Cargo.lock

@@ -22,7 +22,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.45"
+version = "0.3.0-rc.47"
 dependencies = [
  "numpy",
  "pyo3",
@@ -63,7 +63,7 @@ checksum = "60b1af1c220855b6ceac025d3f6ecdd2b7c4894bfe9cd9bda4fbb4bc7c0d4cf0"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.45"
+version = "0.3.0-rc.47"
 dependencies = [
  "version_check",
 ]
@@ -172,9 +172,9 @@ dependencies = [
 
 [[package]]
 name = "once_cell"
-version = "1.20.2"
+version = "1.20.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "1261fe7e33c73b354eab43b1273a57c8f967d0391e80353e51f764ac02cf6775"
+checksum = "945462a4b81e43c4e3ba96bd7b49d834c6f61198356aa858733bc4acf3cbe62e"
 
 [[package]]
 name = "parking_lot"
@@ -335,15 +335,15 @@ checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49"
 
 [[package]]
 name = "smallvec"
-version = "1.13.2"
+version = "1.14.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "3c5e1a9a646d36c3599cd173a41282daf47c44583ad367b8e6837255952e5c67"
+checksum = "7fcf8323ef1faaee30a44a340193b1ac6814fd9b7b4e88e9d4519a3e4abe1cfd"
 
 [[package]]
 name = "syn"
-version = "2.0.96"
+version = "2.0.98"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d5d0adab1ae378d7f53bdebc67a39f1f151407ef230f0ce2883572f5d8985c80"
+checksum = "36147f1a48ae0ec2b5b3bc5b537d267457555a10dc06f3dbc8cb11ba3006d3b1"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -368,9 +368,9 @@ dependencies = [
 
 [[package]]
 name = "unicode-ident"
-version = "1.0.15"
+version = "1.0.17"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "11cd88e12b17c6494200a9c1b683a04fcac9573ed74cd1b62aeb2727c5592243"
+checksum = "00e2473a93778eb0bad35909dff6a10d28e63f792f16ed15e404fca9d5eeedbe"
 
 [[package]]
 name = "unindent"

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/Cargo.toml

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

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/PKG-INFO

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

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/pyproject.toml

@@ -20,7 +20,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.45" # sync this with tool.poetry.version below
+version = "0.3.0-rc.47" # 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.45"
+version = "0.3.0-rc.47"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/model/_agent.py

@@ -3,7 +3,8 @@
 from __future__ import annotations
 
 import copy
-from collections.abc import Sequence
+import datetime
+from collections.abc import Iterable, Sequence
 from typing import TypeAlias
 
 from sqlalchemy import orm as sa_orm
@@ -95,6 +96,39 @@ class Agent(BaseModel[AgentID, models_pb2.Agent, orm.Agent]):
             ),
         )
 
+    @classmethod
+    def from_proto(
+        cls, proto: models_pb2.Agent, client: system.Client | None = None
+    ) -> Agent:
+        client = client or Defaults.get_default_client()
+        return cls(client, proto)
+
+    @classmethod
+    def list(
+        cls,
+        *,
+        limit: int | None = None,
+        room_id: RoomID | None = None,
+        created_before: datetime.datetime | None = None,
+        client: system.Client | None = None,
+        ids: Iterable[AgentID] | None = None,
+        existing_session: sa_orm.Session | None = None,
+    ) -> Ok[list[Agent]] | NotFoundError | InvalidArgumentError:
+        """List agent models."""
+        client = client or Defaults.get_default_client()
+        match cls.list_as_proto(
+            client,
+            limit=limit,
+            room_id=room_id,
+            created_before=created_before,
+            ids=ids,
+            existing_session=existing_session,
+        ):
+            case NotFoundError() | InvalidArgumentError() as err:
+                return err
+            case Ok(protos):
+                return Ok([cls.from_proto(proto, client) for proto in protos])
+
     @property
     def name(self) -> str:
         return self.proto_self.name

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/model/_base_model.py

@@ -206,7 +206,8 @@ class BaseModel(Generic[ID, _ProtoObj, OrmObj], UsesOrmID[ID, _ProtoObj]):
         try:
             _ = self.proto_to_orm(self.proto_self, session).unwrap_or_raise()
             session.flush()
-        except sa.exc.DatabaseError as err:
+        # TODO(thunt): Possibly separate out DatabaseError into a precondition error
+        except (sa.exc.DatabaseError, sa.exc.StatementError) as err:
             return InvalidArgumentError.from_(err)
         return Ok(None)
 

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/model/_proto_orm_convert.py

@@ -386,6 +386,37 @@ def source_proto_to_orm(
     return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
 
 
+def _update_agent_associations(
+    agent: orm.Agent,
+    session: sa_orm.Session,
+) -> list[orm.SpaceID]:
+    associated_space_ids = list[orm.SpaceID]()
+    agent_id = agent.id
+    agent_parameters = agent.parameters
+    if not agent_id or not agent_parameters:
+        return associated_space_ids
+    for (
+        raw_space_id,
+        instruction,
+    ) in agent_parameters.orchestrator_parameters.space_instructions.items():
+        space_id = orm.SpaceID(raw_space_id)
+        space_run_id = (
+            orm.SpaceRunID(instruction.space_run_id)
+            if instruction.space_run_id
+            else None
+        )
+        associated_space_ids.append(space_id)
+        session.merge(
+            orm.AgentSpaceAssociation(
+                room_id=agent.room_id,
+                agent_id=agent_id,
+                space_id=space_id,
+                space_run_id=space_run_id,
+            )
+        )
+    return associated_space_ids
+
+
 def agent_proto_to_orm(
     proto_obj: models_pb2.Agent, session: sa_orm.Session
 ) -> Ok[orm.Agent] | orm.InvalidORMIdentifierError | InvalidArgumentError:
@@ -403,7 +434,18 @@ def agent_proto_to_orm(
         parameters=proto_obj.agent_parameters,
         room_id=room_id,
     )
-    return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
+    agent = _add_orm_to_session(orm_obj, proto_obj.org_id, session)
+    session.flush()
+
+    if not agent.id:
+        return InvalidArgumentError("failed to add agent to session")
+    associated_space_ids = _update_agent_associations(agent, session)
+    session.execute(
+        sa.delete(orm.AgentSpaceAssociation)
+        .where(orm.AgentSpaceAssociation.agent_id == agent.id)
+        .where(orm.AgentSpaceAssociation.space_id.not_in(associated_space_ids))
+    )
+    return Ok(agent)
 
 
 def space_proto_to_orm(
@@ -650,6 +692,11 @@ def resource_delete_orms(
 def agent_delete_orms(
     ids: Sequence[orm.AgentID], session: orm.Session
 ) -> Ok[None] | InvalidArgumentError:
+    session.execute(
+        sa.delete(orm.AgentSpaceAssociation).where(
+            orm.AgentSpaceAssociation.agent_id.in_(ids)
+        )
+    )
     session.execute(sa.delete(orm.Agent).where(orm.Agent.id.in_(ids)))
     return Ok(None)
 
@@ -718,12 +765,14 @@ def room_delete_orms(
     ids: Sequence[orm.RoomID], session: orm.Session
 ) -> Ok[None] | InvalidArgumentError:
     source_refs = list(
-        session.scalars(sa.select(orm.Source).where(orm.Source.room_id == ids).limit(1))
+        session.scalars(
+            sa.select(orm.Source).where(orm.Source.room_id.in_(ids)).limit(1)
+        )
     )
     if source_refs:
         return InvalidArgumentError("cannot delete a room that still has sources")
 
-    session.execute(sa.delete(orm.Room).where(orm.Room.id == ids))
+    session.execute(sa.delete(orm.Room).where(orm.Room.id.in_(ids)))
     return Ok(None)
 
 

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/orm/__init__.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import sqlalchemy as sa
 from sqlalchemy import orm as sa_orm
-from sqlalchemy.orm.collections import attribute_mapped_collection
 
 from corvic.orm._proto_columns import ProtoMessageDecorator
 from corvic.orm.base import Base, OrgBase
@@ -287,14 +286,6 @@ class Space(BelongsToOrgMixin, BelongsToRoomMixin, Base):
         viewonly=True,
     )
 
-    agent_associations: sa_orm.Mapped[list[AgentSpaceAssociation]] = (
-        sa_orm.relationship(
-            init=False,
-            viewonly=True,
-            default_factory=list,
-        )
-    )
-
     @property
     def space_key(self):
         return self.name
@@ -351,12 +342,6 @@ class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
     parameters: sa_orm.Mapped[agent_pb2.AgentParameters | None] = sa_orm.mapped_column(
         default=None
     )
-    messages: sa_orm.Mapped[dict[str, MessageEntry]] = sa_orm.relationship(
-        collection_class=attribute_mapped_collection("entry_id"),
-        default_factory=dict,
-        init=False,
-        viewonly=True,
-    )
 
     space_associations: sa_orm.Mapped[list[AgentSpaceAssociation]] = (
         sa_orm.relationship(default_factory=list, init=False, viewonly=True)
@@ -366,91 +351,15 @@ class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
 class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     __tablename__ = "agent_space_association"
 
+    space_run_id: sa_orm.Mapped[SpaceRunID | None] = sa_orm.mapped_column(
+        ForeignKey(SpaceRun).make()
+    )
     agent_id: sa_orm.Mapped[AgentID] = primary_key_foreign_column(
-        ForeignKey(Agent).make()
+        ForeignKey(Agent).make(ondelete="CASCADE")
     )
     space_id: sa_orm.Mapped[SpaceID] = primary_key_foreign_column(
         ForeignKey(Space).make()
     )
-    space: sa_orm.Mapped[Space] = sa_orm.relationship(
-        back_populates="agent_associations", init=False, viewonly=True
-    )
-
-
-class UserMessage(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
-    """A message sent by an user."""
-
-    __tablename__ = "user_message"
-
-    message_entry: sa_orm.Mapped[MessageEntry] = sa_orm.relationship(
-        init=True, default=None
-    )
-
-    id: sa_orm.Mapped[UserMessageID | None] = primary_key_identity_column()
-
-    message: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-
-
-class AgentMessage(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
-    """A message sent by an agent."""
-
-    __tablename__ = "agent_message"
-    message_metadata: sa_orm.Mapped[common_pb2.AgentMessageMetadata | None] = (
-        sa_orm.mapped_column(
-            default_factory=lambda: common_pb2.AgentMessageMetadata(
-                message_reaction=common_pb2.MessageReaction.MESSAGE_REACTION_UNSPECIFIED
-            )
-        )
-    )
-    id: sa_orm.Mapped[AgentMessageID | None] = primary_key_identity_column()
-
-    message_entry: sa_orm.Mapped[MessageEntry] = sa_orm.relationship(
-        init=True, default=None
-    )
-
-    user_message_id: sa_orm.Mapped[UserMessageID | None] = sa_orm.mapped_column(
-        ForeignKey(UserMessage).make(ondelete="CASCADE"), init=True, default=None
-    )
-    message: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    policy: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    context: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    retrieved_entities: sa_orm.Mapped[common_pb2.RetrievedEntities | None] = (
-        sa_orm.mapped_column(default=None)
-    )
-
-
-class MessageEntry(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
-    """A message either sent by an Agent or an User."""
-
-    __tablename__ = "message_entry"
-
-    id: sa_orm.Mapped[MessageEntryID | None] = primary_key_identity_column()
-
-    agent_id: sa_orm.Mapped[AgentID] = sa_orm.mapped_column(
-        ForeignKey(Agent).make(ondelete="CASCADE"),
-        nullable=True,
-        init=True,
-        default=None,
-    )
-
-    agent_message_id: sa_orm.Mapped[AgentMessageID | None] = sa_orm.mapped_column(
-        ForeignKey(AgentMessage).make(ondelete="CASCADE"), default=None
-    )
-    user_message_id: sa_orm.Mapped[UserMessageID | None] = sa_orm.mapped_column(
-        ForeignKey(UserMessage).make(ondelete="CASCADE"), default=None
-    )
-
-    agent_message: sa_orm.Mapped[AgentMessage | None] = sa_orm.relationship(
-        back_populates="message_entry", init=True, default=None
-    )
-
-    user_message: sa_orm.Mapped[UserMessage | None] = sa_orm.relationship(
-        back_populates="message_entry", init=True, default=None
-    )
-
-    @property
-    def entry_id(self):
-        return self.id
 
 
 class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base):
@@ -494,7 +403,6 @@ ID = (
 __all__ = [
     "Agent",
     "AgentID",
-    "AgentMessage",
     "AgentMessageID",
     "Base",
     "BaseID",
@@ -510,7 +418,6 @@ __all__ = [
     "FeatureViewSourceID",
     "ID",
     "InvalidORMIdentifierError",
-    "MessageEntry",
     "MessageEntryID",
     "Org",
     "OrgID",
@@ -530,7 +437,6 @@ __all__ = [
     "SpaceParametersID",
     "SpaceRun",
     "SpaceRunID",
-    "UserMessage",
     "UserMessageID",
     "primary_key_foreign_column",
     "primary_key_identity_column",

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/pa_scalar/_to_value.py

@@ -1,11 +1,13 @@
 import base64
 import datetime
 import decimal
+import math
 from collections.abc import Callable, Iterable, Mapping
-from typing import Final, TypeAlias, cast
+from typing import Any, Final, TypeAlias, cast
 
 import pyarrow as pa
 from google.protobuf import struct_pb2
+from pyarrow.lib import ArrowException
 
 from corvic.pa_scalar._temporal import (
     MonthDayNanoFrac,
@@ -118,11 +120,17 @@ def _visit_uint64(
     return struct_pb2.Value(string_value=str(value))
 
 
+def _is_invalid_float(value: float):
+    return math.isnan(value) or math.isinf(value)
+
+
 def _visit_float16(
     value: _PyValue,
     dtype: pa.DataType,
 ) -> struct_pb2.Value:
     value = cast(float, value)
+    if _is_invalid_float(value):
+        return _visit_null(value, pa.null())
     return struct_pb2.Value(number_value=value)
 
 
@@ -131,6 +139,8 @@ def _visit_float32(
     dtype: pa.DataType,
 ) -> struct_pb2.Value:
     value = cast(float, value)
+    if _is_invalid_float(value):
+        return _visit_null(value, pa.null())
     return struct_pb2.Value(number_value=value)
 
 
@@ -139,6 +149,8 @@ def _visit_float64(
     dtype: pa.DataType,
 ) -> struct_pb2.Value:
     value = cast(float, value)
+    if _is_invalid_float(value):
+        return _visit_null(value, pa.null())
     return struct_pb2.Value(number_value=value)
 
 
@@ -412,6 +424,8 @@ def to_value(
         case int():
             return struct_pb2.Value(string_value=str(scalar))
         case float():
+            if _is_invalid_float(scalar):
+                return to_value(None)
             return struct_pb2.Value(number_value=scalar)
         case str():
             return struct_pb2.Value(string_value=scalar)
@@ -424,6 +438,36 @@ def to_value(
             return _visit(value, scalar.type)
 
 
+def _to_py(scalar: Scalar) -> Any:
+    try:
+        # This scalar might have malformed binary data that's incorrectly typed
+        # as string (e.g. non-utf-8 strings), so skip it
+        scalar.validate(full=True)
+        return scalar.as_py()
+    except ArrowException:
+        # To create examples that hit this code path, one needs to create a
+        # pyarrow.BinaryArray and then cast it to a StringArray but also supply
+        # a CastOptions with allow_invalid_utf8 set to True.
+        # TODO(aneesh): also check pa.types.is_dictionary
+        if (
+            pa.types.is_list(scalar.type)
+            or pa.types.is_large_list(scalar.type)
+            or pa.types.is_fixed_size_list(scalar.type)
+        ):
+            lscalar = cast(pa.ListScalar[Any], scalar)
+            return [_to_py(v) for v in lscalar.values]  # noqa: PD011 # type: ignore[unknownMemberType]
+        if pa.types.is_map(scalar.type):
+            mscalar = cast(pa.MapScalar[Any, Any], scalar)
+            return {
+                res["key"]: res["value"]
+                for res in [_to_py(v) for v in mscalar.values]  # noqa: PD011 # type: ignore[unknownMemberType]
+            }
+        if pa.types.is_struct(scalar.type):
+            sscalar = cast(pa.StructScalar, scalar)
+            return {k: _to_py(sscalar[k]) for k in sscalar}
+        return None
+
+
 def batch_to_structs(
     batch: pa.RecordBatch,
 ) -> list[struct_pb2.Struct]:
@@ -432,7 +476,8 @@ def batch_to_structs(
     field_by_name = {field.name: field for field in schema}
 
     ret: list[struct_pb2.Struct] = []
-    for row in batch.to_pylist():
+    for pa_row in batch.to_struct_array():
+        row = _to_py(cast(Scalar, pa_row))
         fields = {
             col_name: _visit(v, field_by_name[col_name].type)
             for col_name, v in row.items()

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/pa_scalar/_types.py

@@ -39,3 +39,5 @@ class Scalar(Protocol):
     def equals(self, other: Any) -> bool: ...
 
     def as_py(self) -> Any: ...
+
+    def validate(self, *, full: bool = False) -> bool | None: ...

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/system/storage.py

@@ -102,7 +102,7 @@ class Bucket(Protocol):
 
     def create(self) -> None: ...
 
-    def list_blobs(self) -> Iterator[Blob]: ...
+    def list_blobs(self, prefix: str | None = None) -> Iterator[Blob]: ...
 
 
 class DataKindManager:

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/system_sqlite/fs_blob_store.py

@@ -228,10 +228,16 @@ class FSBucket:
     def create(self) -> None:
         return self.path().mkdir()
 
-    def list_blobs(self) -> Iterator[FSBlob]:
+    def list_blobs(self, prefix: str | None = None) -> Iterator[FSBlob]:
         for entry in self.path().glob("*"):
-            if not entry.name.endswith(".metadata"):
-                yield FSBlob.from_path(self, entry)
+            if entry.name.endswith(".metadata"):
+                continue
+            true_name = base64.urlsafe_b64decode(entry.name.encode("utf-8")).decode(
+                "utf-8"
+            )
+            if prefix and not true_name.startswith(prefix):
+                continue
+            yield FSBlob.from_path(self, entry)
 
     def path(self) -> Path:
         return self.client.root() / self.name

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc47}/python/corvic/system_sqlite/rdbms_blob_store.py

@@ -277,14 +277,15 @@ class RDBMSBucket:
             session.add(ORMBucket(name=self._name))
             session.commit()
 
-    def list_blobs(self) -> Iterator[RDBMSBlob]:
+    def list_blobs(self, prefix: str | None = None) -> Iterator[RDBMSBlob]:
         def get_blobs():
             with self._client.db_session() as session:
                 orm_self = self._to_orm(session)
                 if not orm_self:
                     raise ValueError("bucket does not exist")
                 for blob in orm_self.blobs:
-                    yield RDBMSBlob(self, blob.name)
+                    if not prefix or blob.name.startswith(prefix):
+                        yield RDBMSBlob(self, blob.name)
 
         return iter(list(get_blobs()))
 

corvic_engine-0.3.0rc47/python/corvic_generated/model/v1alpha/models_pb2.py

@@ -0,0 +1,58 @@
+# -*- 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 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 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$corvic/orm/v1/completion_model.proto\x1a corvic/orm/v1/feature_view.proto\x1a\x1c\x63orvic/orm/v1/pipeline.proto\x1a\x19\x63orvic/orm/v1/space.proto\x1a\x19\x63orvic/orm/v1/table.proto\x1a\x1c\x63orvic/status/v1/event.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x90\x01\n\x04Room\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x15\n\x06org_id\x18\x03 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xa8\x03\n\x08Resource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x1b\n\tmime_type\x18\x04 \x01(\tR\x08mimeType\x12\x10\n\x03url\x18\x05 \x01(\tR\x03url\x12\x12\n\x04size\x18\x06 \x01(\x04R\x04size\x12\x10\n\x03md5\x18\x07 \x01(\tR\x03md5\x12#\n\roriginal_path\x18\x08 \x01(\tR\x0coriginalPath\x12\x17\n\x07room_id\x18\t \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x0b \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\r \x01(\tR\npipelineId\x12<\n\rrecent_events\x18\x0e \x03(\x0b\x32\x17.corvic.status.v1.EventR\x0crecentEvents\x12>\n\ncreated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x91\x02\n\x06Source\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x06 \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\x08 \x01(\tR\npipelineId\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xa9\x05\n\x08Pipeline\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\t \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12[\n\x0fresource_inputs\x18\x04 \x03(\x0b\x32\x32.corvic.model.v1alpha.Pipeline.ResourceInputsEntryR\x0eresourceInputs\x12X\n\x0esource_outputs\x18\x05 \x03(\x0b\x32\x31.corvic.model.v1alpha.Pipeline.SourceOutputsEntryR\rsourceOutputs\x12^\n\x17pipeline_transformation\x18\x06 \x01(\x0b\x32%.corvic.orm.v1.PipelineTransformationR\x16pipelineTransformation\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x1a\x61\n\x13ResourceInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.corvic.model.v1alpha.ResourceR\x05value:\x02\x38\x01\x1a^\n\x12SourceOutputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x05value:\x02\x38\x01\x42\r\n\x0b_created_at\"\xca\x02\n\x11\x46\x65\x61tureViewSource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x34\n\x06source\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x06source\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12+\n\x11\x64rop_disconnected\x18\x04 \x01(\x08R\x10\x64ropDisconnected\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x07 \x01(\tR\x06roomIdB\r\n\x0b_created_at\"\x9c\x03\n\x0b\x46\x65\x61tureView\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12P\n\x13\x66\x65\x61ture_view_output\x18\x05 \x01(\x0b\x32 .corvic.orm.v1.FeatureViewOutputR\x11\x66\x65\x61tureViewOutput\x12Y\n\x14\x66\x65\x61ture_view_sources\x18\x06 \x03(\x0b\x32\'.corvic.model.v1alpha.FeatureViewSourceR\x12\x66\x65\x61tureViewSources\x12\x1b\n\tspace_ids\x18\x07 \x03(\tR\x08spaceIds\x12\x15\n\x06org_id\x18\x08 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xfa\x02\n\x05Space\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12I\n\x10space_parameters\x18\x05 \x01(\x0b\x32\x1e.corvic.orm.v1.SpaceParametersR\x0fspaceParameters\x12\x44\n\x0c\x66\x65\x61ture_view\x18\x06 \x01(\x0b\x32!.corvic.model.v1alpha.FeatureViewR\x0b\x66\x65\x61tureView\x12\x1b\n\tauto_sync\x18\t \x01(\x08R\x08\x61utoSync\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x85\x02\n\x05\x41gent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12I\n\x10\x61gent_parameters\x18\x04 \x01(\x0b\x32\x1e.corvic.orm.v1.AgentParametersR\x0f\x61gentParameters\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atJ\x04\x08\x06\x10\x07R\x08messages\"\xad\x02\n\x0f\x43ompletionModel\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x15\n\x06org_id\x18\x04 \x01(\tR\x05orgId\x12H\n\nparameters\x18\x05 \x01(\x0b\x32(.corvic.orm.v1.CompletionModelParametersR\nparameters\x12$\n\x0esecret_api_key\x18\x06 \x01(\tR\x0csecretApiKey\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atb\x06proto3')
+
+_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['_ROOM']._serialized_start=306
+  _globals['_ROOM']._serialized_end=450
+  _globals['_RESOURCE']._serialized_start=453
+  _globals['_RESOURCE']._serialized_end=877
+  _globals['_SOURCE']._serialized_start=880
+  _globals['_SOURCE']._serialized_end=1153
+  _globals['_PIPELINE']._serialized_start=1156
+  _globals['_PIPELINE']._serialized_end=1837
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1629
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1726
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1728
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1822
+  _globals['_FEATUREVIEWSOURCE']._serialized_start=1840
+  _globals['_FEATUREVIEWSOURCE']._serialized_end=2170
+  _globals['_FEATUREVIEW']._serialized_start=2173
+  _globals['_FEATUREVIEW']._serialized_end=2585
+  _globals['_SPACE']._serialized_start=2588
+  _globals['_SPACE']._serialized_end=2966
+  _globals['_AGENT']._serialized_start=2969
+  _globals['_AGENT']._serialized_end=3230
+  _globals['_COMPLETIONMODEL']._serialized_start=3233
+  _globals['_COMPLETIONMODEL']._serialized_end=3534
+# @@protoc_insertion_point(module_scope)