corvic-engine 0.3.0rc45__cp38-abi3-win_amd64.whl → 0.3.0rc47__cp38-abi3-win_amd64.whl

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/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/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/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/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/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/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/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/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.0rc45.dist-info → corvic_engine-0.3.0rc47.dist-info}/METADATA

@@ -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.dist-info → corvic_engine-0.3.0rc47.dist-info}/RECORD

@@ -1,6 +1,6 @@
-corvic_engine-0.3.0rc45.dist-info/METADATA,sha256=yHt7aIKRrl0-UfwUTk3w-pG13PTQCdo2mz02nSPtcCA,1876
-corvic_engine-0.3.0rc45.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
-corvic_engine-0.3.0rc45.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
+corvic_engine-0.3.0rc47.dist-info/METADATA,sha256=DACl2w5jIjPNPHgd9k7VlUxQUC0fi2pN3UewAPoY4f8,1876
+corvic_engine-0.3.0rc47.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
+corvic_engine-0.3.0rc47.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
@@ -13,15 +13,15 @@ corvic/engine/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/engine/_native.pyi,sha256=KYMPtvXqHZ-jMgZohLf4se3rr-rBpCihmjANcr6s8ag,1390
 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=m4JkZu8QAoQEeHqzpMUlH08PGOjVZbqld4Rm0uW73QQ,8225
+corvic/model/_agent.py,sha256=8qmi1x1pBQCXgz7Zr2YdrWRHRZkbp63cyJw7j9OvqQY,4818
+corvic/model/_base_model.py,sha256=qRTZBBHdkqMH6xqX1FwtTm4cWFC4aPDktJeGyxyxtpk,8336
 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=kM8kMzb1oxAgIOMVmv6twZ-JLeP2NopdsKHCiKjuNIE,49700
 corvic/model/_pipeline.py,sha256=A_q_nWm6UBN-AKlbQkhWNMG2r-uW0IR6vGJbhYv7z3k,17578
-corvic/model/_proto_orm_convert.py,sha256=6CWe_b-L7JBS4Pb_SjUeuiuczWh-nPw0jElQIMOl_UA,25029
+corvic/model/_proto_orm_convert.py,sha256=rq4SJYsTCHyOkZswI9FYfj1Ne9Heed117ii89ChjPvc,26607
 corvic/model/_resource.py,sha256=O93gy2r4HjsZzVUahrGbLGmxI7sfSVQCvwvCLN3nSXQ,7097
 corvic/model/_room.py,sha256=57MiBfj8hZcmUfq2PeECrOWDpBZAOSjnVqNUIGXOy2Q,2898
 corvic/model/_source.py,sha256=JBCk1I6u_rUKPiB4Fvtl7uVm0Jx0LF1oWNd1-Wn_sbI,9412
@@ -50,13 +50,13 @@ 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=iDEAq3VGvJeR5rrQOeia728cM8pHogaKwyNGc5mmav0,17827
+corvic/orm/__init__.py,sha256=rOYy3hi3fVbAkXIoJ5NZLDz3uHdw3Bko2ZpRDtb3Fkg,14449
 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
 corvic/pa_scalar/_temporal.py,sha256=HPL1z5k4frIiGWM469YGj9j0hCPXNHtI_X1jk1y1c1Q,7842
-corvic/pa_scalar/_to_value.py,sha256=F11uEikSft9AahBqA7aHmvHObwsTqoZN90KC4ucbem0,11574
-corvic/pa_scalar/_types.py,sha256=73dusgldts4o8j-FgoKjER9malSlMbGVrD4hP3mflWE,1062
+corvic/pa_scalar/_to_value.py,sha256=C1l-BVsS4Rs_KL8hVPllpzjP0NVVwfaOb0WM2pjZhoA,13476
+corvic/pa_scalar/_types.py,sha256=shbytO0ji-H2rBOX_1fooVOshb22wwkVU1W99VBKz1A,1131
 corvic/pa_scalar/__init__.py,sha256=1nfc0MFGpw78RQEI13VE5hpHuyw_DoE7sJbmzqx5pws,1063
 corvic/proto_wrapper/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/proto_wrapper/_errors.py,sha256=0HFmBK9EGYi4lXJ3RFsktHoPZn2m3cMKroI0Eir0JcM,114
@@ -72,7 +72,7 @@ corvic/system/in_memory_executor.py,sha256=dYgcxbA_O0mM1pI19t2OXs8q5B4TX-NFacR7T
 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/storage.py,sha256=N4lXnoCFPg8JbUaI9fbpPY0nt3CgXFUc8sZteki_nRk,5671
 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
@@ -80,9 +80,9 @@ corvic/system/_planner.py,sha256=HUf6UjCy1iHRrXfhU25w19TG4Ik3zVHhtzVcor0eTQY,788
 corvic/system/_text_embedder.py,sha256=LH79_4RxhvssySHpkeEoZFgM4Sa5XAYSjoytdsuwWK4,3269
 corvic/system/__init__.py,sha256=DBzMQ1QCO0OHo3w2LoibfgN1X3CLGBUxXjiExN2kyi0,2838
 corvic/system_sqlite/client.py,sha256=ntDJtCDG1SDGUKI-BhCju4sdqd59Y8RucGZ8VmRvHuI,7142
-corvic/system_sqlite/fs_blob_store.py,sha256=7CS2lq1WXNeB5f7lNU-eLNFM-D_B1hwMnCeq8nyIZ_E,8203
+corvic/system_sqlite/fs_blob_store.py,sha256=pYTMPiWYC6AUIdcgmRj8lvL7Chg82rf5dac6bKGaqL0,8461
 corvic/system_sqlite/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-corvic/system_sqlite/rdbms_blob_store.py,sha256=4eSEve0ee8wZYXtMIl1dIthUQ4iJ27WS6GBcrL4XPqk,10464
+corvic/system_sqlite/rdbms_blob_store.py,sha256=gTP_tQfTVb3wzZkzo8ys1zaz0rSrERzb57rqMHVpuBA,10563
 corvic/system_sqlite/staging.py,sha256=9chn6iXGmKAsurdiBzo0FSTyOmeDn7oCIq8fekbK67c,16878
 corvic/system_sqlite/__init__.py,sha256=MQCUux08NcxFLghpWcZ95lt5AMabr-6sJRH-eSsTnqI,229
 corvic/table/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -156,7 +156,7 @@ 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=PZhKYVfK1bgoYgKKoYwQ4KUbMTtmVlE1zhcQ8X8rR9w,10899
+corvic_generated/model/v1alpha/models_pb2.py,sha256=ncqP8kRV14sjRbL9KIjPB79J-eVFVULCNE1qq7RAMx0,8892
 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
@@ -220,7 +220,7 @@ 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=-HUawjooxKuzlkQTkIKzRIZN7F0qBNAl7azdqKlw60M,14386
+corvic_generated/model/v1alpha/models_pb2.pyi,sha256=eKV5BlocI0MRQuzPVN4eGFrjysyPco2dyszFo7s66ys,11462
 corvic_generated/model/v1alpha/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/orm/v1/agent_pb2.pyi,sha256=9AExLKFRvOJ1fSaOdZc3Src015uvNl-le2lSPatZjSQ,4575
 corvic_generated/orm/v1/agent_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
@@ -244,5 +244,5 @@ corvic_generated/status/v1/event_pb2.pyi,sha256=eU-ibrYpvEAJSIDlSa62-bC96AQU1ykF
 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=tqs1zEZzUSxpYJMFNawLRjHfuc7Mnb2BzcaW7bIgujg,438272
-corvic_engine-0.3.0rc45.dist-info/RECORD,,
+corvic/engine/_native.pyd,sha256=mlFXn3uT-o8-APpr9ihxwWATcd-mdiSGtZvkGcoJibs,438272
+corvic_engine-0.3.0rc47.dist-info/RECORD,,

corvic_generated/model/v1alpha/models_pb2.py

@@ -13,18 +13,16 @@ _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\"\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\"P\n\x0bUserMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\"\xe7\x02\n\x0c\x41gentMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12/\n\x06policy\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x06policy\x12\x18\n\x07\x63ontext\x18\x04 \x01(\tR\x07\x63ontext\x12P\n\x10message_reaction\x18\x05 \x01(\x0e\x32%.corvic.model.v1alpha.MessageReactionR\x0fmessageReaction\x12&\n\x0fuser_message_id\x18\x06 \x01(\tR\ruserMessageId\x12O\n\x12retrieved_entities\x18\x07 \x01(\x0b\x32 .corvic.orm.v1.RetrievedEntitiesR\x11retrievedEntities\x12\x17\n\x07room_id\x18\x08 \x01(\tR\x06roomId\"\xa4\x02\n\x0cMessageEntry\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x46\n\x0cuser_message\x18\x02 \x01(\x0b\x32!.corvic.model.v1alpha.UserMessageH\x00R\x0buserMessage\x12I\n\ragent_message\x18\x03 \x01(\x0b\x32\".corvic.model.v1alpha.AgentMessageH\x00R\x0c\x61gentMessage\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomIdB\t\n\x07\x63ontentB\r\n\x0b_created_at\"\x85\x02\n\x05\x41gent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12I\n\x10\x61gent_parameters\x18\x04 \x01(\x0b\x32\x1e.corvic.orm.v1.AgentParametersR\x0f\x61gentParameters\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atJ\x04\x08\x06\x10\x07R\x08messages\"\xad\x02\n\x0f\x43ompletionModel\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x15\n\x06org_id\x18\x04 \x01(\tR\x05orgId\x12H\n\nparameters\x18\x05 \x01(\x0b\x32(.corvic.orm.v1.CompletionModelParametersR\nparameters\x12$\n\x0esecret_api_key\x18\x06 \x01(\tR\x0csecretApiKey\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at*u\n\x0fMessageReaction\x12 \n\x1cMESSAGE_REACTION_UNSPECIFIED\x10\x00\x12\x1e\n\x1aMESSAGE_REACTION_THUMBS_UP\x10\x01\x12 \n\x1cMESSAGE_REACTION_THUMBS_DOWN\x10\x02\x62\x06proto3')
+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)
@@ -35,34 +33,26 @@ 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=4333
-  _globals['_MESSAGEREACTION']._serialized_end=4450
-  _globals['_ROOM']._serialized_start=364
-  _globals['_ROOM']._serialized_end=508
-  _globals['_RESOURCE']._serialized_start=511
-  _globals['_RESOURCE']._serialized_end=935
-  _globals['_SOURCE']._serialized_start=938
-  _globals['_SOURCE']._serialized_end=1211
-  _globals['_PIPELINE']._serialized_start=1214
-  _globals['_PIPELINE']._serialized_end=1895
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1687
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1784
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1786
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1880
-  _globals['_FEATUREVIEWSOURCE']._serialized_start=1898
-  _globals['_FEATUREVIEWSOURCE']._serialized_end=2228
-  _globals['_FEATUREVIEW']._serialized_start=2231
-  _globals['_FEATUREVIEW']._serialized_end=2643
-  _globals['_SPACE']._serialized_start=2646
-  _globals['_SPACE']._serialized_end=3024
-  _globals['_USERMESSAGE']._serialized_start=3026
-  _globals['_USERMESSAGE']._serialized_end=3106
-  _globals['_AGENTMESSAGE']._serialized_start=3109
-  _globals['_AGENTMESSAGE']._serialized_end=3468
-  _globals['_MESSAGEENTRY']._serialized_start=3471
-  _globals['_MESSAGEENTRY']._serialized_end=3763
-  _globals['_AGENT']._serialized_start=3766
-  _globals['_AGENT']._serialized_end=4027
-  _globals['_COMPLETIONMODEL']._serialized_start=4030
-  _globals['_COMPLETIONMODEL']._serialized_end=4331
+  _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)

corvic_generated/model/v1alpha/models_pb2.pyi

@@ -1,30 +1,18 @@
 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
 from corvic_generated.orm.v1 import table_pb2 as _table_pb2
 from corvic_generated.status.v1 import event_pb2 as _event_pb2
-from google.protobuf import struct_pb2 as _struct_pb2
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from google.protobuf.internal import containers as _containers
-from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
-class MessageReaction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    MESSAGE_REACTION_UNSPECIFIED: _ClassVar[MessageReaction]
-    MESSAGE_REACTION_THUMBS_UP: _ClassVar[MessageReaction]
-    MESSAGE_REACTION_THUMBS_DOWN: _ClassVar[MessageReaction]
-MESSAGE_REACTION_UNSPECIFIED: MessageReaction
-MESSAGE_REACTION_THUMBS_UP: MessageReaction
-MESSAGE_REACTION_THUMBS_DOWN: MessageReaction
-
 class Room(_message.Message):
     __slots__ = ("id", "name", "org_id", "created_at")
     ID_FIELD_NUMBER: _ClassVar[int]
@@ -183,50 +171,6 @@ class Space(_message.Message):
     created_at: _timestamp_pb2.Timestamp
     def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., room_id: _Optional[str] = ..., space_parameters: _Optional[_Union[_space_pb2.SpaceParameters, _Mapping]] = ..., feature_view: _Optional[_Union[FeatureView, _Mapping]] = ..., auto_sync: bool = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
-class UserMessage(_message.Message):
-    __slots__ = ("id", "message", "room_id")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    ROOM_ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    message: str
-    room_id: str
-    def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ..., room_id: _Optional[str] = ...) -> None: ...
-
-class AgentMessage(_message.Message):
-    __slots__ = ("id", "message", "policy", "context", "message_reaction", "user_message_id", "retrieved_entities", "room_id")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    POLICY_FIELD_NUMBER: _ClassVar[int]
-    CONTEXT_FIELD_NUMBER: _ClassVar[int]
-    MESSAGE_REACTION_FIELD_NUMBER: _ClassVar[int]
-    USER_MESSAGE_ID_FIELD_NUMBER: _ClassVar[int]
-    RETRIEVED_ENTITIES_FIELD_NUMBER: _ClassVar[int]
-    ROOM_ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    message: str
-    policy: _struct_pb2.Struct
-    context: str
-    message_reaction: MessageReaction
-    user_message_id: str
-    retrieved_entities: _common_pb2.RetrievedEntities
-    room_id: str
-    def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ..., policy: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., context: _Optional[str] = ..., message_reaction: _Optional[_Union[MessageReaction, str]] = ..., user_message_id: _Optional[str] = ..., retrieved_entities: _Optional[_Union[_common_pb2.RetrievedEntities, _Mapping]] = ..., room_id: _Optional[str] = ...) -> None: ...
-
-class MessageEntry(_message.Message):
-    __slots__ = ("id", "user_message", "agent_message", "created_at", "room_id")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    USER_MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    AGENT_MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
-    ROOM_ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    user_message: UserMessage
-    agent_message: AgentMessage
-    created_at: _timestamp_pb2.Timestamp
-    room_id: str
-    def __init__(self, id: _Optional[str] = ..., user_message: _Optional[_Union[UserMessage, _Mapping]] = ..., agent_message: _Optional[_Union[AgentMessage, _Mapping]] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., room_id: _Optional[str] = ...) -> None: ...
-
 class Agent(_message.Message):
     __slots__ = ("id", "name", "room_id", "agent_parameters", "org_id", "created_at")
     ID_FIELD_NUMBER: _ClassVar[int]