corvic-engine 0.3.0rc55__cp38-abi3-win_amd64.whl → 0.3.0rc56__cp38-abi3-win_amd64.whl

corvic/embed/node2vec.py

@@ -176,7 +176,7 @@ class Space:
                     left_on=start_id_column_names,
                     right_on=self._node_ids,
                     how="left",
-                    join_nulls=True,
+                    nulls_equal=True,
                 )
                 .select(["index"])
                 .rename({"index": "start"})
@@ -187,7 +187,7 @@ class Space:
                     left_on=end_id_column_names,
                     right_on=self._node_ids,
                     how="left",
-                    join_nulls=True,
+                    nulls_equal=True,
                 )
                 .select(["index"])
                 .rename({"index": "end"})

corvic/model/__init__.py

@@ -2,7 +2,13 @@
 
 import corvic.model._feature_type as feature_type
 from corvic.model._agent import Agent, AgentID
-from corvic.model._base_model import BaseModel, HasProtoSelf, UsesOrmID
+from corvic.model._base_model import (
+    BaseModel,
+    BelongsToOrgModel,
+    BelongsToRoomModel,
+    HasProtoSelf,
+    UsesOrmID,
+)
 from corvic.model._completion_model import (
     CompletionModel,
     CompletionModelID,
@@ -25,6 +31,14 @@ from corvic.model._pipeline import (
 )
 from corvic.model._proto_orm_convert import (
     UNCOMMITTED_ID_PREFIX,
+    OrmBelongsToOrgObj,
+    OrmBelongsToRoomObj,
+    OrmObj,
+    ProtoBelongsToOrgObj,
+    ProtoBelongsToRoomObj,
+    ProtoObj,
+    add_orm_org_mixin_to_session,
+    add_orm_room_mixin_to_session,
     space_orm_to_proto,
     timestamp_orm_to_proto,
 )
@@ -60,6 +74,8 @@ __all__ = [
     "Agent",
     "AgentID",
     "BaseModel",
+    "BelongsToOrgModel",
+    "BelongsToRoomModel",
     "ChunkPdfsPipeline",
     "Column",
     "CompletionModel",
@@ -76,8 +92,14 @@ __all__ = [
     "ImageSpace",
     "Node2VecParameters",
     "OcrPdfsPipeline",
+    "OrmObj",
+    "OrmBelongsToOrgObj",
+    "OrmBelongsToRoomObj",
     "Pipeline",
     "PipelineID",
+    "ProtoObj",
+    "ProtoBelongsToOrgObj",
+    "ProtoBelongsToRoomObj",
     "RelationalSpace",
     "Resource",
     "ResourceID",
@@ -96,6 +118,8 @@ __all__ = [
     "UnknownTransformationPipeline",
     "UnknownSpace",
     "UsesOrmID",
+    "add_orm_org_mixin_to_session",
+    "add_orm_room_mixin_to_session",
     "embedding_model_proto_to_name",
     "feature_type",
     "image_model_proto_to_name",

corvic/model/_agent.py

@@ -10,7 +10,7 @@ from typing import TypeAlias
 from sqlalchemy import orm as sa_orm
 
 from corvic import orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToRoomModel
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     agent_delete_orms,
@@ -27,7 +27,7 @@ FeatureViewID: TypeAlias = orm.FeatureViewID
 AgentID: TypeAlias = orm.AgentID
 
 
-class Agent(BaseModel[AgentID, models_pb2.Agent, orm.Agent]):
+class Agent(BelongsToRoomModel[AgentID, models_pb2.Agent, orm.Agent]):
     """A corvic agent represents a named agent that can produce embeddings."""
 
     @classmethod
@@ -133,10 +133,6 @@ class Agent(BaseModel[AgentID, models_pb2.Agent, orm.Agent]):
     def name(self) -> str:
         return self.proto_self.name
 
-    @property
-    def room_id(self) -> RoomID:
-        return RoomID(self.proto_self.room_id)
-
     @property
     def parameters(self) -> AgentParameters:
         return self.proto_self.agent_parameters

corvic/model/_base_model.py

@@ -5,29 +5,29 @@ import datetime
 import functools
 import uuid
 from collections.abc import Callable, Iterable, Iterator, Sequence
-from typing import Final, Generic, Protocol, TypeVar
+from typing import Final, Generic
 
 import sqlalchemy as sa
 import sqlalchemy.orm as sa_orm
 import structlog
-from google.protobuf import timestamp_pb2
 from typing_extensions import Self
 
 from corvic import orm, system
-from corvic.model._proto_orm_convert import ID, UNCOMMITTED_ID_PREFIX, OrmObj
+from corvic.model._proto_orm_convert import (
+    ID,
+    UNCOMMITTED_ID_PREFIX,
+    OrmBelongsToOrgObj,
+    OrmBelongsToRoomObj,
+    OrmObj,
+    ProtoBelongsToOrgObj,
+    ProtoBelongsToRoomObj,
+    ProtoObj,
+)
 from corvic.result import InvalidArgumentError, NotFoundError, Ok
 
 _logger = structlog.get_logger()
 
 
-class _ModelProto(Protocol):
-    id: str
-    created_at: timestamp_pb2.Timestamp
-
-
-_ProtoObj = TypeVar("_ProtoObj", bound=_ModelProto)
-
-
 def _generate_uncommitted_id_str():
     return f"{UNCOMMITTED_ID_PREFIX}{uuid.uuid4()}"
 
@@ -43,11 +43,11 @@ def _create_or_join_session(
             yield session
 
 
-class HasProtoSelf(Generic[_ProtoObj], abc.ABC):
+class HasProtoSelf(Generic[ProtoObj], abc.ABC):
     client: Final[system.Client]
-    proto_self: Final[_ProtoObj]
+    proto_self: Final[ProtoObj]
 
-    def __init__(self, client: system.Client, proto_self: _ProtoObj):
+    def __init__(self, client: system.Client, proto_self: ProtoObj):
         self.proto_self = proto_self
         self.client = client
 
@@ -58,8 +58,8 @@ class HasProtoSelf(Generic[_ProtoObj], abc.ABC):
         return None
 
 
-class UsesOrmID(Generic[ID, _ProtoObj], HasProtoSelf[_ProtoObj]):
-    def __init__(self, client: system.Client, proto_self: _ProtoObj):
+class UsesOrmID(Generic[ID, ProtoObj], HasProtoSelf[ProtoObj]):
+    def __init__(self, client: system.Client, proto_self: ProtoObj):
         if not proto_self.id:
             proto_self.id = _generate_uncommitted_id_str()
         super().__init__(client, proto_self)
@@ -73,7 +73,7 @@ class UsesOrmID(Generic[ID, _ProtoObj], HasProtoSelf[_ProtoObj]):
         return self.id_class().from_str(self.proto_self.id)
 
 
-class BaseModel(Generic[ID, _ProtoObj, OrmObj], UsesOrmID[ID, _ProtoObj]):
+class BaseModel(Generic[ID, ProtoObj, OrmObj], UsesOrmID[ID, ProtoObj]):
     """Base for orm wrappers providing a unified update mechanism."""
 
     @classmethod
@@ -82,12 +82,12 @@ class BaseModel(Generic[ID, _ProtoObj, OrmObj], UsesOrmID[ID, _ProtoObj]):
 
     @classmethod
     @abc.abstractmethod
-    def orm_to_proto(cls, orm_obj: OrmObj) -> _ProtoObj: ...
+    def orm_to_proto(cls, orm_obj: OrmObj) -> ProtoObj: ...
 
     @classmethod
     @abc.abstractmethod
     def proto_to_orm(
-        cls, proto_obj: _ProtoObj, session: orm.Session
+        cls, proto_obj: ProtoObj, session: orm.Session
     ) -> Ok[OrmObj] | InvalidArgumentError: ...
 
     @classmethod
@@ -102,7 +102,7 @@ class BaseModel(Generic[ID, _ProtoObj, OrmObj], UsesOrmID[ID, _ProtoObj]):
         obj_id: ID,
         client: system.Client,
         existing_session: sa_orm.Session | None = None,
-    ) -> Ok[_ProtoObj] | NotFoundError:
+    ) -> Ok[ProtoObj] | NotFoundError:
         """Create a model object by loading it from the database."""
         with _create_or_join_session(client, existing_session) as session:
             orm_self = session.get(cls.orm_class(), obj_id)
@@ -146,7 +146,7 @@ class BaseModel(Generic[ID, _ProtoObj, OrmObj], UsesOrmID[ID, _ProtoObj]):
         ]
         | None = None,
         existing_session: sa_orm.Session | None = None,
-    ) -> Ok[list[_ProtoObj]] | NotFoundError | InvalidArgumentError:
+    ) -> Ok[list[ProtoObj]] | NotFoundError | InvalidArgumentError:
         """List sources that exist in storage."""
         orm_class = cls.orm_class()
         with _create_or_join_session(client, existing_session) as session:
@@ -231,3 +231,25 @@ class BaseModel(Generic[ID, _ProtoObj, OrmObj], UsesOrmID[ID, _ProtoObj]):
                     proto_self=new_proto_self,
                 )
             )
+
+
+class BelongsToOrgModel(
+    Generic[ID, ProtoBelongsToOrgObj, OrmBelongsToOrgObj],
+    BaseModel[ID, ProtoBelongsToOrgObj, OrmBelongsToOrgObj],
+):
+    """Base for orm wrappers with org mixin providing a unified update mechanism."""
+
+    @property
+    def org_id(self) -> orm.OrgID:
+        return orm.OrgID().from_str(self.proto_self.org_id)
+
+
+class BelongsToRoomModel(
+    Generic[ID, ProtoBelongsToRoomObj, OrmBelongsToRoomObj],
+    BelongsToOrgModel[ID, ProtoBelongsToRoomObj, OrmBelongsToRoomObj],
+):
+    """Base for orm wrappers with room mixin providing a unified update mechanism."""
+
+    @property
+    def room_id(self) -> orm.RoomID:
+        return orm.RoomID().from_str(self.proto_self.room_id)

corvic/model/_completion_model.py

@@ -11,7 +11,7 @@ from google.protobuf import timestamp_pb2
 from sqlalchemy import orm as sa_orm
 
 from corvic import orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToOrgModel
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     completion_model_delete_orms,
@@ -29,7 +29,9 @@ UNIX_TIMESTAMP_START_DATETIME = timestamp_pb2.Timestamp(seconds=0, nanos=0)
 
 
 class CompletionModel(
-    BaseModel[CompletionModelID, models_pb2.CompletionModel, orm.CompletionModel]
+    BelongsToOrgModel[
+        CompletionModelID, models_pb2.CompletionModel, orm.CompletionModel
+    ]
 ):
     """Completion Models."""
 
@@ -61,10 +63,6 @@ class CompletionModel(
     def name(self) -> str:
         return self.proto_self.name
 
-    @property
-    def org_id(self) -> OrgID:
-        return OrgID(self.proto_self.org_id)
-
     @property
     def provider(self) -> Literal["openai-generic", "azure-openai"] | None:
         match self.proto_self.parameters.WhichOneof("params"):

corvic/model/_feature_view.py

@@ -18,7 +18,7 @@ from sqlalchemy.orm.interfaces import LoaderOption
 from typing_extensions import Self
 
 from corvic import op_graph, orm, system
-from corvic.model._base_model import BaseModel, UsesOrmID
+from corvic.model._base_model import BelongsToRoomModel, UsesOrmID
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     feature_view_delete_orms,
@@ -642,7 +642,9 @@ class DeepGnnCsvUrlMetadata(DataclassAsTypedMetadataMixin):
     output_csv_url: str
 
 
-class FeatureView(BaseModel[FeatureViewID, models_pb2.FeatureView, orm.FeatureView]):
+class FeatureView(
+    BelongsToRoomModel[FeatureViewID, models_pb2.FeatureView, orm.FeatureView]
+):
     """FeatureViews describe how Sources should be modeled to create a feature space.
 
     Example:
@@ -738,10 +740,6 @@ class FeatureView(BaseModel[FeatureViewID, models_pb2.FeatureView, orm.FeatureVi
                 pass
         return Ok([cls.from_proto(proto, client) for proto in protos])
 
-    @property
-    def room_id(self):
-        return RoomID(self.proto_self.room_id)
-
     @property
     def source_ids(self):
         return [fv_source.source.id for fv_source in self.feature_view_sources]

corvic/model/_pipeline.py

@@ -15,7 +15,7 @@ from typing_extensions import Self
 
 import corvic.table
 from corvic import op_graph, orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToRoomModel
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     pipeline_delete_orms,
@@ -31,7 +31,7 @@ PipelineID: TypeAlias = orm.PipelineID
 RoomID: TypeAlias = orm.RoomID
 
 
-class Pipeline(BaseModel[PipelineID, models_pb2.Pipeline, orm.Pipeline]):
+class Pipeline(BelongsToRoomModel[PipelineID, models_pb2.Pipeline, orm.Pipeline]):
     """Pipelines map resources to sources."""
 
     @classmethod
@@ -152,10 +152,6 @@ class Pipeline(BaseModel[PipelineID, models_pb2.Pipeline, orm.Pipeline]):
             case Ok(protos):
                 return Ok([cls.from_proto(proto, client) for proto in protos])
 
-    @property
-    def room_id(self):
-        return RoomID(self.proto_self.room_id)
-
     @property
     def name(self):
         return self.proto_self.name

corvic/model/_proto_orm_convert.py

@@ -1,6 +1,6 @@
 import datetime
 from collections.abc import Callable, Sequence
-from typing import Any, Protocol, TypeVar, cast
+from typing import Any, Protocol, TypeVar
 
 import sqlalchemy as sa
 import sqlalchemy.orm as sa_orm
@@ -14,26 +14,30 @@ from corvic_generated.orm.v1 import feature_view_pb2
 
 UNCOMMITTED_ID_PREFIX = "__uncommitted_object-"
 
-_Proto = TypeVar(
-    "_Proto",
-    models_pb2.Resource,
-    models_pb2.Source,
-    models_pb2.FeatureView,
-    models_pb2.Space,
-    models_pb2.FeatureViewSource,
-    models_pb2.Agent,
-    models_pb2.Pipeline,
-    models_pb2.Room,
-    models_pb2.CompletionModel,
-)
 ID = TypeVar("ID", bound=orm.BaseID[Any])
 
 
+class _ModelProto(Protocol):
+    id: str
+    created_at: timestamp_pb2.Timestamp
+
+
+class _ModelBelongsToOrgProto(_ModelProto, Protocol):
+    org_id: str
+
+
+class _ModelBelongsToRoomProto(_ModelBelongsToOrgProto, Protocol):
+    room_id: str
+
+
+ProtoObj = TypeVar("ProtoObj", bound=_ModelProto)
+ProtoBelongsToOrgObj = TypeVar("ProtoBelongsToOrgObj", bound=_ModelBelongsToOrgProto)
+ProtoBelongsToRoomObj = TypeVar("ProtoBelongsToRoomObj", bound=_ModelBelongsToRoomProto)
+
+
 class _OrmModel(Protocol[ID]):
     id: sa_orm.Mapped[ID | None]
 
-    org_id: sa_orm.Mapped[orm.OrgID | None]
-
     @sa.ext.hybrid.hybrid_property
     def created_at(self) -> datetime.datetime | None: ...
 
@@ -42,7 +46,17 @@ class _OrmModel(Protocol[ID]):
     def _created_at_expression(cls): ...
 
 
+class _OrmBelongsToOrgModel(_OrmModel[ID], Protocol):
+    org_id: sa_orm.Mapped[orm.OrgID | None]
+
+
+class _OrmBelongsToRoomModel(_OrmBelongsToOrgModel[ID], Protocol):
+    room_id: sa_orm.Mapped[orm.RoomID | None]
+
+
 OrmObj = TypeVar("OrmObj", bound=_OrmModel[Any])
+OrmBelongsToOrgObj = TypeVar("OrmBelongsToOrgObj", bound=_OrmBelongsToOrgModel[Any])
+OrmBelongsToRoomObj = TypeVar("OrmBelongsToRoomObj", bound=_OrmBelongsToRoomModel[Any])
 
 
 def _translate_orm_id(
@@ -58,39 +72,6 @@ def _translate_orm_id(
             return Ok(parsed_obj_id)
 
 
-def _translate_orm_ids(
-    proto_obj: _Proto, obj_id_class: type[ID]
-) -> Ok[tuple[ID | None, orm.RoomID | None]] | orm.InvalidORMIdentifierError:
-    match _translate_orm_id(proto_obj.id, obj_id_class):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok(obj_id):
-            pass
-
-    match proto_obj:
-        case (
-            models_pb2.Resource()
-            | models_pb2.Source()
-            | models_pb2.FeatureView()
-            | models_pb2.Space()
-            | models_pb2.Agent()
-            | models_pb2.Pipeline()
-            | models_pb2.FeatureViewSource()
-        ):
-            room_id = orm.RoomID(proto_obj.room_id)
-            match room_id.to_db():
-                case orm.InvalidORMIdentifierError() as err:
-                    return err
-                case Ok():
-                    pass
-        case models_pb2.CompletionModel():
-            room_id = None
-        case models_pb2.Room():
-            room_id = cast(orm.RoomID, obj_id)
-
-    return Ok((obj_id, room_id))
-
-
 def timestamp_orm_to_proto(
     timestamp_orm: datetime.datetime | None,
 ) -> timestamp_pb2.Timestamp | None:
@@ -247,43 +228,51 @@ def completion_model_orm_to_proto(
     )
 
 
-def _add_orm_to_session(
-    orm_obj: OrmObj, org_id: str, session: sa_orm.Session
-) -> OrmObj:
-    if org_id:
-        orm_obj.org_id = orm.OrgID(org_id)
+def add_orm_org_mixin_to_session(
+    orm_obj: OrmBelongsToOrgObj,
+    proto_obj: _ModelBelongsToOrgProto,
+    id_class: type[ID],
+    session: sa_orm.Session,
+) -> Ok[OrmBelongsToOrgObj] | orm.InvalidORMIdentifierError:
+    match _translate_orm_id(proto_obj.id, id_class):
+        case Ok(orm_id):
+            orm_obj.id = orm_id
+        case orm.InvalidORMIdentifierError() as err:
+            return err
+    if proto_obj.org_id:
+        org_id = orm.OrgID(proto_obj.org_id)
+        match org_id.to_db():
+            case Ok():
+                orm_obj.org_id = org_id
+            case orm.InvalidORMIdentifierError() as err:
+                return err
+        orm_obj.org_id = org_id
     if not orm_obj.id:
         session.add(orm_obj)
     else:
         orm_obj = session.merge(orm_obj)
-    return orm_obj
+    return Ok(orm_obj)
 
 
-def resource_proto_to_orm(
-    proto_obj: models_pb2.Resource, session: sa_orm.Session
-) -> Ok[orm.Resource] | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.ResourceID):
+def add_orm_room_mixin_to_session(
+    orm_obj: OrmBelongsToRoomObj,
+    proto_obj: _ModelBelongsToRoomProto,
+    id_class: type[ID],
+    session: sa_orm.Session,
+) -> Ok[OrmBelongsToRoomObj] | orm.InvalidORMIdentifierError:
+    room_id = orm.RoomID(proto_obj.room_id)
+    match room_id.to_db():
+        case Ok():
+            pass
         case orm.InvalidORMIdentifierError() as err:
             return err
-        case Ok((obj_id, room_id)):
-            pass
-    if not room_id:
-        return InvalidArgumentError("room id required to commit")
+    orm_obj.room_id = orm.RoomID(proto_obj.room_id)
+    return add_orm_org_mixin_to_session(orm_obj, proto_obj, id_class, session)
 
-    orm_obj = orm.Resource(
-        id=obj_id,
-        name=proto_obj.name,
-        description=proto_obj.description,
-        mime_type=proto_obj.mime_type,
-        md5=proto_obj.md5,
-        url=proto_obj.url,
-        size=proto_obj.size,
-        original_path=proto_obj.original_path,
-        latest_event=proto_obj.recent_events[-1] if proto_obj.recent_events else None,
-        room_id=room_id,
-    )
-    _add_orm_to_session(orm_obj, proto_obj.org_id, session)
 
+def _resource_pipeline_to_orm(
+    proto_obj: models_pb2.Resource, orm_obj: orm.Resource, session: sa_orm.Session
+) -> Ok[None] | InvalidArgumentError:
     if proto_obj.pipeline_id:
         match _translate_orm_id(proto_obj.pipeline_id, orm.PipelineID):
             case orm.InvalidORMIdentifierError() as err:
@@ -299,18 +288,41 @@ def resource_proto_to_orm(
             resource_id=orm_obj.id,
             name=proto_obj.pipeline_input_name,
             pipeline_id=pipeline_id,
-            room_id=room_id,
+            room_id=orm_obj.room_id,
         )
         if orm_obj.org_id:
             pipeline_input.org_id = orm_obj.org_id
         orm_obj.pipeline_ref = session.merge(pipeline_input)
+    return Ok(None)
 
-    return Ok(orm_obj)
+
+def resource_proto_to_orm(
+    proto_obj: models_pb2.Resource, session: sa_orm.Session
+) -> Ok[orm.Resource] | InvalidArgumentError:
+    orm_obj = orm.Resource(
+        name=proto_obj.name,
+        description=proto_obj.description,
+        mime_type=proto_obj.mime_type,
+        md5=proto_obj.md5,
+        url=proto_obj.url,
+        size=proto_obj.size,
+        original_path=proto_obj.original_path,
+        latest_event=proto_obj.recent_events[-1] if proto_obj.recent_events else None,
+    )
+    add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.ResourceID, session)
+
+    match _resource_pipeline_to_orm(proto_obj, orm_obj, session):
+        case Ok(None):
+            return Ok(orm_obj)
+        case InvalidArgumentError() as err:
+            return err
 
 
 def _ensure_id(
-    proto_obj: _Proto,
-    proto_to_orm: Callable[[_Proto, sa_orm.Session], Ok[OrmObj] | InvalidArgumentError],
+    proto_obj: ProtoObj,
+    proto_to_orm: Callable[
+        [ProtoObj, sa_orm.Session], Ok[OrmObj] | InvalidArgumentError
+    ],
     id_type: type[ID],
     session: sa_orm.Session,
 ) -> Ok[ID] | orm.InvalidORMIdentifierError | InvalidArgumentError:
@@ -333,23 +345,19 @@ def _ensure_id(
 def pipeline_proto_to_orm(  # noqa: C901
     proto_obj: models_pb2.Pipeline, session: sa_orm.Session
 ) -> Ok[orm.Pipeline] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.PipelineID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, room_id)):
-            pass
-    if not room_id:
-        return InvalidArgumentError("room id required to commit")
-
     inputs = list[orm.PipelineInput]()
     orm_obj = orm.Pipeline(
-        id=obj_id,
         name=proto_obj.name,
-        room_id=room_id,
         transformation=proto_obj.pipeline_transformation,
         description=proto_obj.description,
     )
-    orm_obj = _add_orm_to_session(orm_obj, proto_obj.org_id, session)
+    if proto_obj.org_id:
+        orm_obj.org_id = orm.OrgID(proto_obj.org_id)
+    match add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.PipelineID, session):
+        case Ok(orm_obj):
+            pass
+        case orm.InvalidORMIdentifierError() as err:
+            return err
     session.flush()
 
     if not orm_obj.id:
@@ -364,7 +372,10 @@ def pipeline_proto_to_orm(  # noqa: C901
                 pass
         outputs.append(
             orm.PipelineOutput(
-                source_id=source_id, name=name, pipeline_id=orm_obj.id, room_id=room_id
+                source_id=source_id,
+                name=name,
+                pipeline_id=orm_obj.id,
+                room_id=orm_obj.room_id,
             )
         )
 
@@ -380,21 +391,11 @@ def pipeline_proto_to_orm(  # noqa: C901
 def source_proto_to_orm(
     proto_obj: models_pb2.Source, session: sa_orm.Session
 ) -> Ok[orm.Source] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.SourceID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, room_id)):
-            pass
-    if not room_id:
-        return InvalidArgumentError("room id required to commit")
-
     orm_obj = orm.Source(
-        id=obj_id,
         name=proto_obj.name,
         table_op_graph=proto_obj.table_op_graph,
-        room_id=room_id,
     )
-    return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
+    return add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.SourceID, session)
 
 
 def _update_agent_associations(
@@ -431,21 +432,15 @@ def _update_agent_associations(
 def agent_proto_to_orm(
     proto_obj: models_pb2.Agent, session: sa_orm.Session
 ) -> Ok[orm.Agent] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.AgentID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, room_id)):
-            pass
-    if not room_id:
-        return InvalidArgumentError("room id required to commit")
-
     orm_obj = orm.Agent(
-        id=obj_id,
         name=proto_obj.name,
         parameters=proto_obj.agent_parameters,
-        room_id=room_id,
     )
-    agent = _add_orm_to_session(orm_obj, proto_obj.org_id, session)
+    match add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.AgentID, session):
+        case Ok(agent):
+            pass
+        case orm.InvalidORMIdentifierError() as err:
+            return err
     session.flush()
 
     if not agent.id:
@@ -462,14 +457,6 @@ def agent_proto_to_orm(
 def space_proto_to_orm(
     proto_obj: models_pb2.Space, session: sa_orm.Session
 ) -> Ok[orm.Space] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.SpaceID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, room_id)):
-            pass
-    if not room_id:
-        return InvalidArgumentError("room id required to commit")
-
     match _ensure_id(
         proto_obj.feature_view, feature_view_proto_to_orm, orm.FeatureViewID, session
     ):
@@ -482,38 +469,29 @@ def space_proto_to_orm(
         raise InternalError("internal assertion did not hold")
 
     orm_obj = orm.Space(
-        id=obj_id,
         name=proto_obj.name,
         description=proto_obj.description,
-        room_id=room_id,
         feature_view_id=feature_view_id,
         parameters=proto_obj.space_parameters,
         auto_sync=proto_obj.auto_sync,
     )
-    return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
+    return add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.SpaceID, session)
 
 
-def feature_view_proto_to_orm(  # noqa: C901
+def feature_view_proto_to_orm(
     proto_obj: models_pb2.FeatureView, session: sa_orm.Session
 ) -> Ok[orm.FeatureView] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.FeatureViewID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, room_id)):
-            pass
-    if not room_id:
-        return InvalidArgumentError("room id required to commit")
-
     orm_obj = orm.FeatureView(
-        id=obj_id,
         name=proto_obj.name,
         description=proto_obj.description,
-        room_id=room_id,
     )
     if proto_obj.org_id:
         orm_obj.org_id = orm.OrgID(proto_obj.org_id)
-
-    orm_obj = _add_orm_to_session(orm_obj, proto_obj.org_id, session)
+    match add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.FeatureViewID, session):
+        case Ok(orm_obj):
+            pass
+        case orm.InvalidORMIdentifierError() as err:
+            return err
     session.flush()
 
     if not orm_obj.id:
@@ -573,57 +551,38 @@ def _feature_view_source_proto_to_orm(
             return err
         case Ok(source_id):
             pass
-    match _translate_orm_id(proto_obj.id, orm.FeatureViewSourceID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok(obj_id):
-            pass
 
     orm_obj = orm.FeatureViewSource(
-        room_id=orm.RoomID(proto_obj.room_id),
-        id=obj_id,
         table_op_graph=proto_obj.table_op_graph,
         drop_disconnected=proto_obj.drop_disconnected,
         source_id=source_id,
         feature_view_id=feature_view_id,
     )
-    return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
+    return add_orm_room_mixin_to_session(
+        orm_obj, proto_obj, orm.FeatureViewSourceID, session
+    )
 
 
 def room_proto_to_orm(
     proto_obj: models_pb2.Room, session: sa_orm.Session
 ) -> Ok[orm.Room] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.RoomID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, _)):
-            pass
-
-    orm_obj = orm.Room(
-        id=obj_id,
-        name=proto_obj.name,
-    )
-    return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
+    orm_obj = orm.Room(name=proto_obj.name)
+    return add_orm_org_mixin_to_session(orm_obj, proto_obj, orm.RoomID, session)
 
 
 def completion_model_proto_to_orm(
     proto_obj: models_pb2.CompletionModel, session: sa_orm.Session
 ) -> Ok[orm.CompletionModel] | InvalidArgumentError:
-    match _translate_orm_ids(proto_obj, orm.CompletionModelID):
-        case orm.InvalidORMIdentifierError() as err:
-            return err
-        case Ok((obj_id, _)):
-            pass
-
     orm_obj = orm.CompletionModel(
-        id=obj_id,
         name=proto_obj.name,
         description=proto_obj.description,
         parameters=proto_obj.parameters,
         secret_api_key=proto_obj.secret_api_key,
         last_validation_time=proto_obj.last_validation_time.ToDatetime(),
     )
-    return Ok(_add_orm_to_session(orm_obj, proto_obj.org_id, session))
+    return add_orm_org_mixin_to_session(
+        orm_obj, proto_obj, orm.CompletionModelID, session
+    )
 
 
 def source_delete_orms(

corvic/model/_resource.py

@@ -15,7 +15,7 @@ from sqlalchemy.orm.interfaces import LoaderOption
 from typing_extensions import Self
 
 from corvic import orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToRoomModel
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     resource_delete_orms,
@@ -32,7 +32,7 @@ RoomID: TypeAlias = orm.RoomID
 PipelineID: TypeAlias = orm.PipelineID
 
 
-class Resource(BaseModel[ResourceID, models_pb2.Resource, orm.Resource]):
+class Resource(BelongsToRoomModel[ResourceID, models_pb2.Resource, orm.Resource]):
     """Resources represent import data."""
 
     @classmethod
@@ -67,10 +67,6 @@ class Resource(BaseModel[ResourceID, models_pb2.Resource, orm.Resource]):
     def name(self) -> str:
         return self.proto_self.name
 
-    @property
-    def room_id(self) -> RoomID:
-        return RoomID(self.proto_self.room_id)
-
     @property
     def pipeline_id(self) -> PipelineID | None:
         return PipelineID(self.proto_self.pipeline_id) or None

corvic/model/_room.py

@@ -8,7 +8,7 @@ from typing import TypeAlias
 import structlog
 
 from corvic import orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToOrgModel
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     room_delete_orms,
@@ -25,7 +25,7 @@ RoomID: TypeAlias = orm.RoomID
 FeatureViewID: TypeAlias = orm.FeatureViewID
 
 
-class Room(BaseModel[RoomID, models_pb2.Room, orm.Room]):
+class Room(BelongsToOrgModel[RoomID, models_pb2.Room, orm.Room]):
     """Rooms contain conversations and tables."""
 
     @classmethod

corvic/model/_source.py

@@ -15,7 +15,7 @@ from sqlalchemy.orm.interfaces import LoaderOption
 from typing_extensions import Self
 
 from corvic import op_graph, orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToRoomModel
 from corvic.model._defaults import Defaults
 from corvic.model._proto_orm_convert import (
     source_delete_orms,
@@ -46,7 +46,7 @@ def foreign_key(
             )
 
 
-class Source(BaseModel[SourceID, models_pb2.Source, orm.Source]):
+class Source(BelongsToRoomModel[SourceID, models_pb2.Source, orm.Source]):
     """Sources describe how resources should be treated.
 
     Example:
@@ -269,14 +269,18 @@ class Source(BaseModel[SourceID, models_pb2.Source, orm.Source]):
             self.client, op_graph.op.from_proto(self.proto_self.table_op_graph)
         )
 
+    @functools.cached_property
+    def prop_table(self):
+        if self.proto_self.prop_table_op_graph is None:
+            return None
+        return Table.from_ops(
+            self.client, op_graph.op.from_proto(self.proto_self.prop_table_op_graph)
+        )
+
     @property
     def name(self) -> str:
         return self.proto_self.name
 
-    @property
-    def room_id(self) -> RoomID:
-        return RoomID(self.proto_self.room_id)
-
     @property
     def pipeline_id(self) -> PipelineID | None:
         return PipelineID(self.proto_self.pipeline_id) or None

corvic/model/_space.py

@@ -14,7 +14,7 @@ from sqlalchemy import orm as sa_orm
 from typing_extensions import Self
 
 from corvic import op_graph, orm, system
-from corvic.model._base_model import BaseModel
+from corvic.model._base_model import BelongsToRoomModel
 from corvic.model._defaults import Defaults
 from corvic.model._feature_view import FeatureView, FeatureViewEdgeTableMetadata
 from corvic.model._proto_orm_convert import (
@@ -61,7 +61,7 @@ name_to_proto_image_model = {
 }
 
 
-class Space(BaseModel[SpaceID, models_pb2.Space, orm.Space]):
+class Space(BelongsToRoomModel[SpaceID, models_pb2.Space, orm.Space]):
     """Spaces apply embedding methods to FeatureViews.
 
     Example:
@@ -105,10 +105,6 @@ class Space(BaseModel[SpaceID, models_pb2.Space, orm.Space]):
     def name(self):
         return self.proto_self.name
 
-    @property
-    def room_id(self):
-        return RoomID(self.proto_self.room_id)
-
     @property
     def description(self):
         return self.proto_self.description

corvic/orm/__init__.py

@@ -39,6 +39,7 @@ from corvic.orm.keys import (
     ForeignKey,
     primary_key_foreign_column,
     primary_key_identity_column,
+    primary_key_uuid_column,
 )
 from corvic.orm.mixins import (
     BelongsToOrgMixin,
@@ -66,11 +67,11 @@ from corvic_generated.status.v1 import event_pb2
 # and if sub-orm-model updates are required they are explicit.
 
 
-class Org(SoftDeleteMixin, OrgBase):
+class Org(SoftDeleteMixin, OrgBase, kw_only=True):
     """An organization it a top level grouping of resources."""
 
 
-class Room(BelongsToOrgMixin, SoftDeleteMixin, Base):
+class Room(BelongsToOrgMixin, SoftDeleteMixin, Base, kw_only=True):
     """A Room is a logical collection of Documents."""
 
     __tablename__ = "room"
@@ -88,15 +89,17 @@ class BelongsToRoomMixin(sa_orm.MappedAsDataclass):
     room_id: sa_orm.Mapped[RoomID | None] = sa_orm.mapped_column(
         ForeignKey(Room).make(ondelete="CASCADE"),
         nullable=True,
+        default=None,
     )
 
 
-class DefaultObjects(Base):
+class DefaultObjects(Base, kw_only=True):
     """Holds the identifiers for default objects."""
 
     __tablename__ = "default_objects"
-    default_org: sa_orm.Mapped[OrgID] = sa_orm.mapped_column(
-        ForeignKey(Org).make(ondelete="CASCADE")
+    default_org: sa_orm.Mapped[OrgID | None] = sa_orm.mapped_column(
+        ForeignKey(Org).make(ondelete="CASCADE"),
+        nullable=False,
     )
     default_room: sa_orm.Mapped[RoomID | None] = sa_orm.mapped_column(
         ForeignKey(Room).make(ondelete="CASCADE"), nullable=True, default=None
@@ -104,7 +107,7 @@ class DefaultObjects(Base):
     version: sa_orm.Mapped[int | None] = primary_key_identity_column(type_=INT_PK_TYPE)
 
 
-class Resource(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class Resource(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """A Resource is a reference to some durably stored file.
 
     E.g., a document could be a PDF file, an image, or a text transcript of a
@@ -129,7 +132,7 @@ class Resource(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class Source(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class Source(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """A source."""
 
     __tablename__ = "source"
@@ -152,7 +155,7 @@ class Source(BelongsToOrgMixin, BelongsToRoomMixin, Base):
         return self.name
 
 
-class Pipeline(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class Pipeline(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """A resource to source pipeline."""
 
     __tablename__ = "pipeline"
@@ -172,7 +175,7 @@ class Pipeline(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class PipelineInput(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class PipelineInput(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """Pipeline input resources."""
 
     __tablename__ = "pipeline_input"
@@ -190,7 +193,7 @@ class PipelineInput(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class PipelineOutput(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class PipelineOutput(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """Objects for tracking pipeline output sources."""
 
     __tablename__ = "pipeline_output"
@@ -208,7 +211,9 @@ class PipelineOutput(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class FeatureView(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class FeatureView(
+    SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True
+):
     """A FeatureView is a logical collection of sources used by various spaces."""
 
     __tablename__ = "feature_view"
@@ -233,7 +238,7 @@ class FeatureView(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class FeatureViewSource(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class FeatureViewSource(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """A source inside of a feature view."""
 
     __tablename__ = "feature_view_source"
@@ -257,7 +262,7 @@ class FeatureViewSource(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class Space(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class Space(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """A space is a named evaluation of space parameters."""
 
     __tablename__ = "space"
@@ -322,7 +327,7 @@ class SpaceRun(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     )
 
 
-class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     """An Agent."""
 
     __tablename__ = "agent"
@@ -344,7 +349,7 @@ class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base):
+class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
     __tablename__ = "agent_space_association"
 
     space_run_id: sa_orm.Mapped[SpaceRunID | None] = sa_orm.mapped_column(
@@ -358,7 +363,7 @@ class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base):
+class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base, kw_only=True):
     """A customer's custom completion model definition."""
 
     __tablename__ = "completion_model"
@@ -441,6 +446,7 @@ __all__ = [
     "UserMessageID",
     "primary_key_foreign_column",
     "primary_key_identity_column",
+    "primary_key_uuid_column",
     "ProtoMessageDecorator",
     "IntIDDecorator",
 ]

corvic/orm/base.py

@@ -178,7 +178,7 @@ class OrgBase(Base):
     # overriding table_args is the recommending way of defining these base model types
     __table_args__: ClassVar[Any] = ({"extend_existing": True},)
 
-    id: sa_orm.Mapped[OrgID] = primary_key_uuid_column()
+    id: sa_orm.Mapped[OrgID | None] = primary_key_uuid_column()
 
     @property
     def name(self) -> str:

{corvic_engine-0.3.0rc55.dist-info → corvic_engine-0.3.0rc56.dist-info}/METADATA

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

{corvic_engine-0.3.0rc55.dist-info → corvic_engine-0.3.0rc56.dist-info}/RECORD

@@ -1,9 +1,9 @@
-corvic_engine-0.3.0rc55.dist-info/METADATA,sha256=UGe1WYtobGOVsljbDSR0j_HfG7yeY43axe7-fOXqoEM,1876
-corvic_engine-0.3.0rc55.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
-corvic_engine-0.3.0rc55.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
+corvic_engine-0.3.0rc56.dist-info/METADATA,sha256=kGcTUywSpPxHTYF5tu-LPySqBXMZgKxvNVgkOssAn0A,1876
+corvic_engine-0.3.0rc56.dist-info/WHEEL,sha256=hKPP3BCTWtTwj6SFaSI--T5aOGqh_llYfbZ_BsqivwA,94
+corvic_engine-0.3.0rc56.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
+corvic/embed/node2vec.py,sha256=XIJjFDdT-JnmZ43lgP-K-dLgnR17L_uaJqBPAYlsPsk,11148
 corvic/embed/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/embed/__init__.py,sha256=cZZSrRXmezJuTafcQgrB1rbitqXZTVY1B5ryRzAlvgs,144
 corvic/embedding_metric/embeddings.py,sha256=5jvSY0cg5P-Wg_KN7DsrcPo5AfJ_1-XKdErx_dNN5B8,14082
@@ -13,20 +13,20 @@ 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=8qmi1x1pBQCXgz7Zr2YdrWRHRZkbp63cyJw7j9OvqQY,4818
-corvic/model/_base_model.py,sha256=qRTZBBHdkqMH6xqX1FwtTm4cWFC4aPDktJeGyxyxtpk,8336
-corvic/model/_completion_model.py,sha256=fjNP7h9Uu6nVAwy5awj3ILDF150gTuH2j3soAqKd08o,7400
+corvic/model/_agent.py,sha256=8tle_IGhy0LTPd1nNXDfBypnzF3CI7S9fhZbDsVxnZc,4737
+corvic/model/_base_model.py,sha256=WYBBPa1TeU9wchh-UBFlMwuQMDY5cKHBlDznhLbnSHA,8989
+corvic/model/_completion_model.py,sha256=f_ud3xW1iFXSijGMo0WYDmLfmM6mQayAcjTW37AM3q8,7337
 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=YThcU0T4pK_W6IOJ8uQUUsc3NP7JMBWqYjU_37UjN2o,49757
-corvic/model/_pipeline.py,sha256=LtF6uq3VkXGEDTotNEpiPyWJLQMJSrJ8ttXIAHsj3RU,16350
-corvic/model/_proto_orm_convert.py,sha256=UhfzFmkp_-Isi4dHD3T0LUQYh02V0uvM4cZbbQco2Hk,27138
-corvic/model/_resource.py,sha256=qR9iQQgxRO7c_6VGMH8gXbmA-6PCigne2ycNIfEskZg,7854
-corvic/model/_room.py,sha256=57MiBfj8hZcmUfq2PeECrOWDpBZAOSjnVqNUIGXOy2Q,2898
-corvic/model/_source.py,sha256=JBCk1I6u_rUKPiB4Fvtl7uVm0Jx0LF1oWNd1-Wn_sbI,9412
-corvic/model/_space.py,sha256=YMm6gLYk8ii16toAle93D8-cf_javloVAEVSM5mHW5o,35661
-corvic/model/__init__.py,sha256=IzYwjTcZE8MgmCESoj7W518N2CmZgcUFbzBhnpvNers,2526
+corvic/model/_feature_view.py,sha256=gdcXzsMuxpJ7vwbIGYgZlLYNxi2zvdZXvFsb36x6lKg,49694
+corvic/model/_pipeline.py,sha256=c16ap3yHQXqBmjG_2bMzz8hBYJCr14V2WxwlAYOw5Zw,16279
+corvic/model/_proto_orm_convert.py,sha256=jmzmaaUkSxeHB5OMef92AyGw7sorJ6pP4ylbeKXoHvA,26120
+corvic/model/_resource.py,sha256=w5m6mmD8KrHJ8efPTfRV0JKaCmkDRaxlGeuRMmVbw10,7773
+corvic/model/_room.py,sha256=36mXngZ38L4mr6_LgUm-QgsUUaoGMiYQRfvXLV_jd-4,2914
+corvic/model/_source.py,sha256=A1Jk4r5mB0f-Y3L8esaQFCUAu7CCTlwAm7f4qSnvjsM,9603
+corvic/model/_space.py,sha256=13ggLTCQMNTYYpP5PldMqtJiKp3sWOVRhQcktmoHefA,35590
+corvic/model/__init__.py,sha256=Lb-yC04t17Hr2TlnGfn5Ewzd2h1nH4hb9tKdMNAak9s,3075
 corvic/op_graph/aggregation.py,sha256=8X6vqXD7dLHrhYJU0BqmhUsWGbzD1zSP5Db5VHdIru4,6187
 corvic/op_graph/encoders.py,sha256=93wYoBCn_us5lRCkqvjaP0LTg3LBB3yEfhzICv06bB0,10460
 corvic/op_graph/errors.py,sha256=I4NE5053d0deGm5xx5EmyP4f98qx42xnIsW1IA-2hy4,163
@@ -40,7 +40,7 @@ corvic/op_graph/sample_strategy.py,sha256=DrbtJ3ORkIRfyIE_FdlOh_UMnCW_K9jL1LeonV
 corvic/op_graph/_schema.py,sha256=STbxY5PIqIA6xkSDeK8k72Nutsxq5jGe7e_aT35aznI,5733
 corvic/op_graph/_transformations.py,sha256=L9Au_GcciPynww4ZXojMtNdPJ36Qboc9gn0bVzXLifU,9445
 corvic/op_graph/__init__.py,sha256=1DMrQfuuS3FkLa9DXYDjSDLurdxxpG5H1jB2ctaa9xo,1444
-corvic/orm/base.py,sha256=2LJh6gRZXpQwV50hquGIGkgqRsIq_pew1djgGd7umW0,8817
+corvic/orm/base.py,sha256=95nkqycCZ1FaWAhTsa7zbZ0YuwNFkMUW7Wk8yhtYau8,8824
 corvic/orm/errors.py,sha256=uFhFXpVG6pby1lndJZHGHxv3Y0Fbt0RiaZ-CqDfuY1o,545
 corvic/orm/func/utc_func.py,sha256=-FC6w9wBWXejMv1AICT2Gg7tdkSo7gqL2dFT-YKPGQ4,4518
 corvic/orm/func/uuid_func.py,sha256=oXPjDGAl3mvlNtvcvBrLmRRHPJgtKffShIPbHm-EswA,1152
@@ -50,7 +50,7 @@ corvic/orm/keys.py,sha256=Ag6Xbpvxev-VByT1KJ8ChUn9vKVEzkkMXxrjvtADCtY,2182
 corvic/orm/mixins.py,sha256=HfmzJ7LblHtddbbkDmv7nNWURL87Bnj8NeOnNbfmSN4,17794
 corvic/orm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/orm/_proto_columns.py,sha256=tcOu92UjFJFYZLasS6sWJQBDRK26yrnmpTii_LDY4iw,913
-corvic/orm/__init__.py,sha256=Cp50x14iAMcPkE4S25JctmydVEVP1iFg41ZY9kyVBKc,14505
+corvic/orm/__init__.py,sha256=Yzfn_GyCGHzf-wt-CmtamW15PyuZ7tHI7IqQw-3aPmQ,14827
 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
@@ -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=WNpKsDJpoayuoeAxNpjg3CE8fS9ULQETw5fFyi1RFWk,8551
+corvic_generated/model/v1alpha/models_pb2.py,sha256=Jvw4rYuekrbjI7sx0QPcLnTDL5aXI3l0drMiM7dy4ac,8703
 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=HX-6ED-FNhlY4ZAhbFP8t-ikNbfSMH5s0dc6DmaGcV8,11301
+corvic_generated/model/v1alpha/models_pb2.pyi,sha256=K8clNf_M36tu0DEOb4Lo4l_fh4DA2IOD3c1uTI07Wgo,11513
 corvic_generated/model/v1alpha/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/orm/v1/agent_pb2.pyi,sha256=AxcZC0AJqiOyu_5quSMR-E0MjVhDY7b5ym4uZa7WFug,4670
 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=DDDClUHAN_qLIwjp4njDisVErHObCAmHEyFWmBywE8c,438272
-corvic_engine-0.3.0rc55.dist-info/RECORD,,
+corvic/engine/_native.pyd,sha256=XkU3bVVXAk3up15IfaE0ih1d0_Lo8jRl_mJp1ZwbBls,438272
+corvic_engine-0.3.0rc56.dist-info/RECORD,,

{corvic_engine-0.3.0rc55.dist-info → corvic_engine-0.3.0rc56.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.2)
+Generator: maturin (1.8.3)
 Root-Is-Purelib: false
 Tag: cp38-abi3-win_amd64

corvic_generated/model/v1alpha/models_pb2.py

@@ -22,7 +22,7 @@ from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot
 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\"\xd8\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\x13pipeline_input_name\x18\x0f \x01(\tR\x11pipelineInputName\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\"\xe9\x03\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\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^\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\"\x99\x03\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\x12Q\n\x14last_validation_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x12lastValidationTime\x88\x01\x01\x42\r\n\x0b_created_atB\x17\n\x15_last_validation_timeb\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\"\xd8\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\x13pipeline_input_name\x18\x0f \x01(\tR\x11pipelineInputName\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\"\xfc\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\x12Q\n\x13prop_table_op_graph\x18\t \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpH\x01R\x10propTableOpGraph\x88\x01\x01\x42\r\n\x0b_created_atB\x16\n\x14_prop_table_op_graph\"\xe9\x03\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\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^\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\"\x99\x03\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\x12Q\n\x14last_validation_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x12lastValidationTime\x88\x01\x01\x42\r\n\x0b_created_atB\x17\n\x15_last_validation_timeb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -36,19 +36,19 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_RESOURCE']._serialized_start=453
   _globals['_RESOURCE']._serialized_end=925
   _globals['_SOURCE']._serialized_start=928
-  _globals['_SOURCE']._serialized_end=1201
-  _globals['_PIPELINE']._serialized_start=1204
-  _globals['_PIPELINE']._serialized_end=1693
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1584
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1678
-  _globals['_FEATUREVIEWSOURCE']._serialized_start=1696
-  _globals['_FEATUREVIEWSOURCE']._serialized_end=2026
-  _globals['_FEATUREVIEW']._serialized_start=2029
-  _globals['_FEATUREVIEW']._serialized_end=2441
-  _globals['_SPACE']._serialized_start=2444
-  _globals['_SPACE']._serialized_end=2822
-  _globals['_AGENT']._serialized_start=2825
-  _globals['_AGENT']._serialized_end=3086
-  _globals['_COMPLETIONMODEL']._serialized_start=3089
-  _globals['_COMPLETIONMODEL']._serialized_end=3498
+  _globals['_SOURCE']._serialized_end=1308
+  _globals['_PIPELINE']._serialized_start=1311
+  _globals['_PIPELINE']._serialized_end=1800
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1691
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1785
+  _globals['_FEATUREVIEWSOURCE']._serialized_start=1803
+  _globals['_FEATUREVIEWSOURCE']._serialized_end=2133
+  _globals['_FEATUREVIEW']._serialized_start=2136
+  _globals['_FEATUREVIEW']._serialized_end=2548
+  _globals['_SPACE']._serialized_start=2551
+  _globals['_SPACE']._serialized_end=2929
+  _globals['_AGENT']._serialized_start=2932
+  _globals['_AGENT']._serialized_end=3193
+  _globals['_COMPLETIONMODEL']._serialized_start=3196
+  _globals['_COMPLETIONMODEL']._serialized_end=3605
 # @@protoc_insertion_point(module_scope)

corvic_generated/model/v1alpha/models_pb2.pyi

@@ -58,7 +58,7 @@ class Resource(_message.Message):
     def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., mime_type: _Optional[str] = ..., url: _Optional[str] = ..., size: _Optional[int] = ..., md5: _Optional[str] = ..., original_path: _Optional[str] = ..., room_id: _Optional[str] = ..., org_id: _Optional[str] = ..., pipeline_id: _Optional[str] = ..., pipeline_input_name: _Optional[str] = ..., recent_events: _Optional[_Iterable[_Union[_event_pb2.Event, _Mapping]]] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
 class Source(_message.Message):
-    __slots__ = ("id", "name", "table_op_graph", "room_id", "org_id", "pipeline_id", "created_at")
+    __slots__ = ("id", "name", "table_op_graph", "room_id", "org_id", "pipeline_id", "created_at", "prop_table_op_graph")
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     TABLE_OP_GRAPH_FIELD_NUMBER: _ClassVar[int]
@@ -66,6 +66,7 @@ class Source(_message.Message):
     ORG_ID_FIELD_NUMBER: _ClassVar[int]
     PIPELINE_ID_FIELD_NUMBER: _ClassVar[int]
     CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    PROP_TABLE_OP_GRAPH_FIELD_NUMBER: _ClassVar[int]
     id: str
     name: str
     table_op_graph: _table_pb2.TableComputeOp
@@ -73,7 +74,8 @@ class Source(_message.Message):
     org_id: str
     pipeline_id: str
     created_at: _timestamp_pb2.Timestamp
-    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., table_op_graph: _Optional[_Union[_table_pb2.TableComputeOp, _Mapping]] = ..., room_id: _Optional[str] = ..., org_id: _Optional[str] = ..., pipeline_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+    prop_table_op_graph: _table_pb2.TableComputeOp
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., table_op_graph: _Optional[_Union[_table_pb2.TableComputeOp, _Mapping]] = ..., room_id: _Optional[str] = ..., org_id: _Optional[str] = ..., pipeline_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., prop_table_op_graph: _Optional[_Union[_table_pb2.TableComputeOp, _Mapping]] = ...) -> None: ...
 
 class Pipeline(_message.Message):
     __slots__ = ("id", "name", "description", "room_id", "source_outputs", "pipeline_transformation", "org_id", "created_at")