corvic-engine 0.3.0rc59__tar.gz → 0.3.0rc61__tar.gz

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/Cargo.lock

@@ -22,7 +22,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.59"
+version = "0.3.0-rc.61"
 dependencies = [
  "numpy",
  "pyo3",
@@ -63,7 +63,7 @@ checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.59"
+version = "0.3.0-rc.61"
 dependencies = [
  "version_check",
 ]
@@ -82,9 +82,9 @@ checksum = "f4c7245a08504955605670dbf141fceab975f15ca21570696aebe9d2e71576bd"
 
 [[package]]
 name = "libc"
-version = "0.2.170"
+version = "0.2.171"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "875b3680cb2f8f71bdcf9a30f38d48282f5d3c95cbf9b3fa57269bb5d5c06828"
+checksum = "c19937216e9d3aa9956d9bb8dfc0b0c8beb6058fc4f7a4dc4d850edf86a237d6"
 
 [[package]]
 name = "lock_api"
@@ -172,9 +172,9 @@ dependencies = [
 
 [[package]]
 name = "once_cell"
-version = "1.20.3"
+version = "1.21.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "945462a4b81e43c4e3ba96bd7b49d834c6f61198356aa858733bc4acf3cbe62e"
+checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
 
 [[package]]
 name = "parking_lot"
@@ -279,9 +279,9 @@ dependencies = [
 
 [[package]]
 name = "quote"
-version = "1.0.39"
+version = "1.0.40"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c1f1914ce909e1658d9907913b4b91947430c7d9be598b15a1912935b8c04801"
+checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d"
 dependencies = [
  "proc-macro2",
 ]
@@ -341,9 +341,9 @@ checksum = "7fcf8323ef1faaee30a44a340193b1ac6814fd9b7b4e88e9d4519a3e4abe1cfd"
 
 [[package]]
 name = "syn"
-version = "2.0.99"
+version = "2.0.100"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e02e925281e18ffd9d640e234264753c43edc62d64b2d4cf898f1bc5e75f3fc2"
+checksum = "b09a44accad81e1ba1cd74a32461ba89dee89095ba17b32f5d03683b1b1fc2a0"
 dependencies = [
  "proc-macro2",
  "quote",

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/Cargo.toml

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

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/PKG-INFO

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

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/pyproject.toml

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

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/model/__init__.py

@@ -1,7 +1,6 @@
 """Data modeling objects for creating corvic pipelines."""
 
 import corvic.model._feature_type as feature_type
-from corvic.model._agent import Agent, AgentID
 from corvic.model._base_model import (
     BaseModel,
     BelongsToOrgModel,
@@ -71,8 +70,6 @@ from corvic.model._space import (
 FeatureType = feature_type.FeatureType
 
 __all__ = [
-    "Agent",
-    "AgentID",
     "BaseModel",
     "BelongsToOrgModel",
     "BelongsToRoomModel",

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/model/_base_model.py

@@ -215,12 +215,15 @@ class BaseModel(Generic[ID, ProtoObj, OrmObj], UsesOrmID[ID, ProtoObj]):
         with orm.Session(
             self.client.sa_engine, expire_on_commit=False, autoflush=False
         ) as session:
-            match self.delete_by_ids([self.id], session):
-                case InvalidArgumentError() as err:
-                    return err
-                case Ok(None):
-                    pass
-            session.commit()
+            try:
+                match self.delete_by_ids([self.id], session):
+                    case InvalidArgumentError() as err:
+                        return err
+                    case Ok(None):
+                        pass
+                session.commit()
+            except sa.exc.IntegrityError as exc:
+                return InvalidArgumentError.from_(exc)
 
             new_proto_self = copy.copy(self.proto_self)
             new_proto_self.id = ""

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/model/_proto_orm_convert.py

@@ -121,17 +121,6 @@ def source_orm_to_proto(source_orm: orm.Source) -> models_pb2.Source:
     )
 
 
-def agent_orm_to_proto(agent_orm: orm.Agent) -> models_pb2.Agent:
-    return models_pb2.Agent(
-        id=str(agent_orm.id),
-        name=agent_orm.name,
-        room_id=str(agent_orm.room_id),
-        agent_parameters=agent_orm.parameters,
-        org_id=str(agent_orm.org_id),
-        created_at=timestamp_orm_to_proto(agent_orm.created_at),
-    )
-
-
 def feature_view_source_orm_to_proto(
     feature_view_source_orm: orm.FeatureViewSource,
 ) -> models_pb2.FeatureViewSource:
@@ -401,62 +390,6 @@ def source_proto_to_orm(
     return add_orm_room_mixin_to_session(orm_obj, proto_obj, orm.SourceID, 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:
-    orm_obj = orm.Agent(
-        name=proto_obj.name,
-        parameters=proto_obj.agent_parameters,
-    )
-    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:
-        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(
     proto_obj: models_pb2.Space, session: sa_orm.Session
 ) -> Ok[orm.Space] | orm.InvalidORMIdentifierError | InvalidArgumentError:
@@ -664,18 +597,6 @@ def resource_delete_orms(
     return Ok(None)
 
 
-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)
-
-
 def feature_view_source_delete_orms(
     ids: Sequence[orm.FeatureViewSourceID], session: orm.Session
 ) -> Ok[None] | InvalidArgumentError:
@@ -722,16 +643,6 @@ def feature_view_delete_orms(
 def space_delete_orms(
     ids: Sequence[orm.SpaceID], session: orm.Session
 ) -> Ok[None] | InvalidArgumentError:
-    existing_agents = list(
-        session.scalars(
-            sa.select(orm.AgentSpaceAssociation)
-            .where(orm.AgentSpaceAssociation.space_id.in_(ids))
-            .limit(1)
-        )
-    )
-    if existing_agents:
-        return InvalidArgumentError("cannot delete a space that still has agents")
-
     session.execute(sa.delete(orm.Space).where(orm.Space.id.in_(ids)))
     return Ok(None)
 

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/orm/__init__.py

@@ -48,7 +48,6 @@ from corvic.orm.mixins import (
     live_unique_constraint,
 )
 from corvic_generated.orm.v1 import (
-    agent_pb2,
     common_pb2,
     completion_model_pb2,
     feature_view_pb2,
@@ -292,77 +291,6 @@ class Space(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
         return self.name
 
 
-class SpaceRun(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
-    """A Space run."""
-
-    __tablename__ = "space_run"
-
-    id: sa_orm.Mapped[SpaceRunID | None] = primary_key_identity_column()
-    table_op_graph: sa_orm.Mapped[table_pb2.TableComputeOp] = sa_orm.mapped_column(
-        default_factory=table_pb2.TableComputeOp
-    )
-    space_id: sa_orm.Mapped[SpaceID] = sa_orm.mapped_column(
-        ForeignKey(Space).make(ondelete="CASCADE"), nullable=False
-    )
-    space: sa_orm.Mapped[Space] = sa_orm.relationship(init=False, viewonly=True)
-    result_url: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    coordinates_urls: sa_orm.Mapped[common_pb2.BlobUrlList | None] = (
-        sa_orm.mapped_column(default=None)
-    )
-    latest_event: sa_orm.Mapped[event_pb2.Event | None] = sa_orm.mapped_column(
-        default=None, nullable=True
-    )
-    vector_urls: sa_orm.Mapped[common_pb2.BlobUrlList | None] = sa_orm.mapped_column(
-        default=None
-    )
-
-    embedding_metrics: sa_orm.Mapped[common_pb2.EmbeddingMetrics | None] = (
-        sa_orm.mapped_column(default=None)
-    )
-    insight_tools: sa_orm.Mapped[table_pb2.NamedTables | None] = sa_orm.mapped_column(
-        default=None
-    )
-    combine_embeddings_from: sa_orm.Mapped[SpaceRunID | None] = sa_orm.mapped_column(
-        sa.ForeignKey("space_run.id"), nullable=True, default=None
-    )
-
-
-class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
-    """An Agent."""
-
-    __tablename__ = "agent"
-    __table_args__ = (live_unique_constraint("name", "room_id"),)
-
-    id: sa_orm.Mapped[AgentID | None] = primary_key_identity_column()
-    name: sa_orm.Mapped[str] = sa_orm.mapped_column(sa.Text, default=None)
-
-    @property
-    def agent_key(self):
-        return self.name
-
-    parameters: sa_orm.Mapped[agent_pb2.AgentParameters | None] = sa_orm.mapped_column(
-        default=None
-    )
-
-    space_associations: sa_orm.Mapped[list[AgentSpaceAssociation]] = (
-        sa_orm.relationship(default_factory=list, init=False, viewonly=True)
-    )
-
-
-class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base, kw_only=True):
-    __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(ondelete="CASCADE")
-    )
-    space_id: sa_orm.Mapped[SpaceID] = primary_key_foreign_column(
-        ForeignKey(Space).make()
-    )
-
-
 class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base, kw_only=True):
     """A customer's custom completion model definition."""
 
@@ -412,7 +340,6 @@ ID = (
 
 
 __all__ = [
-    "Agent",
     "AgentID",
     "AgentMessageID",
     "Base",
@@ -446,7 +373,6 @@ __all__ = [
     "Space",
     "SpaceID",
     "SpaceParametersID",
-    "SpaceRun",
     "SpaceRunID",
     "UserMessageID",
     "primary_key_foreign_column",

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/orm/base.py

@@ -167,7 +167,7 @@ class Base(sa_orm.MappedAsDataclass, sa_orm.DeclarativeBase):
     @updated_at.inplace.expression
     @classmethod
     def _updated_at_expression(cls):
-        return cls._created_at
+        return cls._updated_at
 
 
 class OrgBase(Base):

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/system/_embedder.py

@@ -6,6 +6,7 @@ import numpy as np
 import polars as pl
 from typing_extensions import Protocol
 
+from corvic import orm
 from corvic.result import InternalError, InvalidArgumentError, Ok
 
 
@@ -18,6 +19,7 @@ class EmbedTextContext:
     tokenizer_name: str
     expected_vector_length: int
     expected_coordinate_bitwidth: Literal[32, 64]
+    room_id: orm.RoomID
 
 
 @dataclasses.dataclass

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/system/in_memory_executor.py

@@ -807,6 +807,7 @@ class InMemoryExecutor(OpGraphExecutor):
             tokenizer_name=op.tokenizer_name,
             expected_vector_length=op.expected_vector_length,
             expected_coordinate_bitwidth=op.expected_coordinate_bitwidth,
+            room_id=context.exec_context.room_id,
         )
         match self._text_embedder.embed(embed_context):
             case Ok(result):

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/system/op_graph_executor.py

@@ -8,7 +8,7 @@ from typing import Any, Protocol
 
 import pyarrow as pa
 
-from corvic import op_graph
+from corvic import op_graph, orm
 from corvic.result import (
     InternalError,
     InvalidArgumentError,
@@ -55,9 +55,10 @@ class ExecutionContext:
     """Description of the computation to be completed."""
 
     tables_to_compute: list[TableComputeContext]
+    room_id: orm.RoomID
     """A list of tables that the caller wants in addition to table_to_compute.
 
-    This has advantages over multimple invocations of OpGraphExecutor when those
+    This has advantages over multiple invocations of OpGraphExecutor when those
     additional tables would be computed to compute tables_to_compute anyway; i.e.,
     they are nodes in the tables_to_compute op graph.
     """

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic/table/table.py

@@ -456,6 +456,7 @@ class Table:
     ):
         """Convert every row to a dictionary of Python-native values."""
         context = ExecutionContext(
+            room_id=room_id,
             tables_to_compute=[
                 TableComputeContext(
                     self.op_graph,

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

@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: corvic/model/v1alpha/models.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from corvic_generated.orm.v1 import completion_model_pb2 as corvic_dot_orm_dot_v1_dot_completion__model__pb2
+from corvic_generated.orm.v1 import feature_view_pb2 as corvic_dot_orm_dot_v1_dot_feature__view__pb2
+from corvic_generated.orm.v1 import pipeline_pb2 as corvic_dot_orm_dot_v1_dot_pipeline__pb2
+from corvic_generated.orm.v1 import space_pb2 as corvic_dot_orm_dot_v1_dot_space__pb2
+from corvic_generated.orm.v1 import table_pb2 as corvic_dot_orm_dot_v1_dot_table__pb2
+from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot_event__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/model/v1alpha/models.proto\x12\x14\x63orvic.model.v1alpha\x1a$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\"\x97\x04\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\x12Q\n\x14last_validation_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x12lastValidationTime\x88\x01\x01\x12]\n\x1alast_successful_validation\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x18lastSuccessfulValidation\x88\x01\x01\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\tcreatedAt\x88\x01\x01\x42\x17\n\x15_last_validation_timeB\x1d\n\x1b_last_successful_validationB\r\n\x0b_created_atb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.model.v1alpha.models_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
+  _globals['_ROOM']._serialized_start=279
+  _globals['_ROOM']._serialized_end=423
+  _globals['_RESOURCE']._serialized_start=426
+  _globals['_RESOURCE']._serialized_end=898
+  _globals['_SOURCE']._serialized_start=901
+  _globals['_SOURCE']._serialized_end=1281
+  _globals['_PIPELINE']._serialized_start=1284
+  _globals['_PIPELINE']._serialized_end=1773
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1664
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1758
+  _globals['_FEATUREVIEWSOURCE']._serialized_start=1776
+  _globals['_FEATUREVIEWSOURCE']._serialized_end=2106
+  _globals['_FEATUREVIEW']._serialized_start=2109
+  _globals['_FEATUREVIEW']._serialized_end=2521
+  _globals['_SPACE']._serialized_start=2524
+  _globals['_SPACE']._serialized_end=2902
+  _globals['_COMPLETIONMODEL']._serialized_start=2905
+  _globals['_COMPLETIONMODEL']._serialized_end=3440
+# @@protoc_insertion_point(module_scope)

{corvic_engine-0.3.0rc59 → corvic_engine-0.3.0rc61}/python/corvic_generated/model/v1alpha/models_pb2.pyi

@@ -1,4 +1,3 @@
-from corvic_generated.orm.v1 import agent_pb2 as _agent_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
@@ -166,22 +165,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 Agent(_message.Message):
-    __slots__ = ("id", "name", "room_id", "agent_parameters", "org_id", "created_at")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    ROOM_ID_FIELD_NUMBER: _ClassVar[int]
-    AGENT_PARAMETERS_FIELD_NUMBER: _ClassVar[int]
-    ORG_ID_FIELD_NUMBER: _ClassVar[int]
-    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    name: str
-    room_id: str
-    agent_parameters: _agent_pb2.AgentParameters
-    org_id: str
-    created_at: _timestamp_pb2.Timestamp
-    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., room_id: _Optional[str] = ..., agent_parameters: _Optional[_Union[_agent_pb2.AgentParameters, _Mapping]] = ..., org_id: _Optional[str] = ..., created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
-
 class CompletionModel(_message.Message):
     __slots__ = ("id", "name", "description", "org_id", "parameters", "secret_api_key", "last_validation_time", "last_successful_validation", "created_at")
     ID_FIELD_NUMBER: _ClassVar[int]

corvic_engine-0.3.0rc59/python/corvic/model/_agent.py

@@ -1,154 +0,0 @@
-"""Corvic Agents."""
-
-from __future__ import annotations
-
-import copy
-import datetime
-from collections.abc import Iterable, Sequence
-from typing import TypeAlias
-
-from sqlalchemy import orm as sa_orm
-
-from corvic import orm, system
-from corvic.model._base_model import BelongsToRoomModel
-from corvic.model._defaults import Defaults
-from corvic.model._proto_orm_convert import (
-    agent_delete_orms,
-    agent_orm_to_proto,
-    agent_proto_to_orm,
-)
-from corvic.result import InvalidArgumentError, NotFoundError, Ok
-from corvic_generated.model.v1alpha import models_pb2
-from corvic_generated.orm.v1.agent_pb2 import AgentParameters
-
-OrgID: TypeAlias = orm.OrgID
-RoomID: TypeAlias = orm.RoomID
-FeatureViewID: TypeAlias = orm.FeatureViewID
-AgentID: TypeAlias = orm.AgentID
-
-
-class Agent(BelongsToRoomModel[AgentID, models_pb2.Agent, orm.Agent]):
-    """A corvic agent represents a named agent that can produce embeddings."""
-
-    @classmethod
-    def orm_class(cls):
-        return orm.Agent
-
-    @classmethod
-    def id_class(cls):
-        return AgentID
-
-    @classmethod
-    def orm_to_proto(cls, orm_obj: orm.Agent) -> models_pb2.Agent:
-        return agent_orm_to_proto(orm_obj)
-
-    @classmethod
-    def proto_to_orm(
-        cls, proto_obj: models_pb2.Agent, session: orm.Session
-    ) -> Ok[orm.Agent] | InvalidArgumentError:
-        return agent_proto_to_orm(proto_obj, session)
-
-    @classmethod
-    def delete_by_ids(
-        cls, ids: Sequence[AgentID], session: orm.Session
-    ) -> Ok[None] | InvalidArgumentError:
-        return agent_delete_orms(ids, session)
-
-    @classmethod
-    def from_id(
-        cls,
-        agent_id: AgentID,
-        client: system.Client | None = None,
-        session: sa_orm.Session | None = None,
-    ) -> Ok[Agent] | NotFoundError:
-        client = client or Defaults.get_default_client()
-        return cls.load_proto_for(agent_id, client, existing_session=session).map(
-            lambda proto_self: cls(client, proto_self)
-        )
-
-    @classmethod
-    def from_orm(
-        cls,
-        agent: orm.Agent,
-        client: system.Client | None = None,
-    ):
-        client = client or Defaults.get_default_client()
-        return cls(
-            client,
-            cls.orm_to_proto(agent),
-        )
-
-    @classmethod
-    def create(
-        cls,
-        name: str,
-        parameters: AgentParameters,
-        room_id: RoomID | None = None,
-        client: system.Client | None = None,
-    ):
-        client = client or Defaults.get_default_client()
-        return cls(
-            client,
-            models_pb2.Agent(
-                name=name,
-                agent_parameters=parameters,
-                room_id=str(room_id or Defaults.get_default_room_id(client)),
-            ),
-        )
-
-    @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
-
-    @property
-    def parameters(self) -> AgentParameters:
-        return self.proto_self.agent_parameters
-
-    def with_name(self, name: str) -> Agent:
-        proto_self = copy.deepcopy(self.proto_self)
-        proto_self.name = name
-        return self.__class__(
-            self.client,
-            proto_self=proto_self,
-        )
-
-    def with_parameters(self, parameters: AgentParameters) -> Agent:
-        proto_self = copy.deepcopy(self.proto_self)
-        proto_self.agent_parameters.CopyFrom(parameters)
-        return self.__class__(
-            self.client,
-            proto_self=proto_self,
-        )