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

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

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

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

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

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

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

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

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

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

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import sqlalchemy as sa
 from sqlalchemy import orm as sa_orm
-from sqlalchemy.orm.collections import attribute_mapped_collection
 
 from corvic.orm._proto_columns import ProtoMessageDecorator
 from corvic.orm.base import Base, OrgBase
@@ -351,12 +350,6 @@ class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
     parameters: sa_orm.Mapped[agent_pb2.AgentParameters | None] = sa_orm.mapped_column(
         default=None
     )
-    messages: sa_orm.Mapped[dict[str, MessageEntry]] = sa_orm.relationship(
-        collection_class=attribute_mapped_collection("entry_id"),
-        default_factory=dict,
-        init=False,
-        viewonly=True,
-    )
 
     space_associations: sa_orm.Mapped[list[AgentSpaceAssociation]] = (
         sa_orm.relationship(default_factory=list, init=False, viewonly=True)
@@ -377,82 +370,6 @@ class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base):
     )
 
 
-class UserMessage(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
-    """A message sent by an user."""
-
-    __tablename__ = "user_message"
-
-    message_entry: sa_orm.Mapped[MessageEntry] = sa_orm.relationship(
-        init=True, default=None
-    )
-
-    id: sa_orm.Mapped[UserMessageID | None] = primary_key_identity_column()
-
-    message: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-
-
-class AgentMessage(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
-    """A message sent by an agent."""
-
-    __tablename__ = "agent_message"
-    message_metadata: sa_orm.Mapped[common_pb2.AgentMessageMetadata | None] = (
-        sa_orm.mapped_column(
-            default_factory=lambda: common_pb2.AgentMessageMetadata(
-                message_reaction=common_pb2.MessageReaction.MESSAGE_REACTION_UNSPECIFIED
-            )
-        )
-    )
-    id: sa_orm.Mapped[AgentMessageID | None] = primary_key_identity_column()
-
-    message_entry: sa_orm.Mapped[MessageEntry] = sa_orm.relationship(
-        init=True, default=None
-    )
-
-    user_message_id: sa_orm.Mapped[UserMessageID | None] = sa_orm.mapped_column(
-        ForeignKey(UserMessage).make(ondelete="CASCADE"), init=True, default=None
-    )
-    message: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    policy: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    context: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
-    retrieved_entities: sa_orm.Mapped[common_pb2.RetrievedEntities | None] = (
-        sa_orm.mapped_column(default=None)
-    )
-
-
-class MessageEntry(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
-    """A message either sent by an Agent or an User."""
-
-    __tablename__ = "message_entry"
-
-    id: sa_orm.Mapped[MessageEntryID | None] = primary_key_identity_column()
-
-    agent_id: sa_orm.Mapped[AgentID] = sa_orm.mapped_column(
-        ForeignKey(Agent).make(ondelete="CASCADE"),
-        nullable=True,
-        init=True,
-        default=None,
-    )
-
-    agent_message_id: sa_orm.Mapped[AgentMessageID | None] = sa_orm.mapped_column(
-        ForeignKey(AgentMessage).make(ondelete="CASCADE"), default=None
-    )
-    user_message_id: sa_orm.Mapped[UserMessageID | None] = sa_orm.mapped_column(
-        ForeignKey(UserMessage).make(ondelete="CASCADE"), default=None
-    )
-
-    agent_message: sa_orm.Mapped[AgentMessage | None] = sa_orm.relationship(
-        back_populates="message_entry", init=True, default=None
-    )
-
-    user_message: sa_orm.Mapped[UserMessage | None] = sa_orm.relationship(
-        back_populates="message_entry", init=True, default=None
-    )
-
-    @property
-    def entry_id(self):
-        return self.id
-
-
 class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base):
     """A customer's custom completion model definition."""
 
@@ -494,7 +411,6 @@ ID = (
 __all__ = [
     "Agent",
     "AgentID",
-    "AgentMessage",
     "AgentMessageID",
     "Base",
     "BaseID",
@@ -510,7 +426,6 @@ __all__ = [
     "FeatureViewSourceID",
     "ID",
     "InvalidORMIdentifierError",
-    "MessageEntry",
     "MessageEntryID",
     "Org",
     "OrgID",
@@ -530,7 +445,6 @@ __all__ = [
     "SpaceParametersID",
     "SpaceRun",
     "SpaceRunID",
-    "UserMessage",
     "UserMessageID",
     "primary_key_foreign_column",
     "primary_key_identity_column",

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

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

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

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

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

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

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

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: corvic/model/v1alpha/models.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from corvic_generated.orm.v1 import agent_pb2 as corvic_dot_orm_dot_v1_dot_agent__pb2
+from corvic_generated.orm.v1 import completion_model_pb2 as corvic_dot_orm_dot_v1_dot_completion__model__pb2
+from corvic_generated.orm.v1 import feature_view_pb2 as corvic_dot_orm_dot_v1_dot_feature__view__pb2
+from corvic_generated.orm.v1 import pipeline_pb2 as corvic_dot_orm_dot_v1_dot_pipeline__pb2
+from corvic_generated.orm.v1 import space_pb2 as corvic_dot_orm_dot_v1_dot_space__pb2
+from corvic_generated.orm.v1 import table_pb2 as corvic_dot_orm_dot_v1_dot_table__pb2
+from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot_event__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/model/v1alpha/models.proto\x12\x14\x63orvic.model.v1alpha\x1a\x19\x63orvic/orm/v1/agent.proto\x1a$corvic/orm/v1/completion_model.proto\x1a corvic/orm/v1/feature_view.proto\x1a\x1c\x63orvic/orm/v1/pipeline.proto\x1a\x19\x63orvic/orm/v1/space.proto\x1a\x19\x63orvic/orm/v1/table.proto\x1a\x1c\x63orvic/status/v1/event.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x90\x01\n\x04Room\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x15\n\x06org_id\x18\x03 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xa8\x03\n\x08Resource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x1b\n\tmime_type\x18\x04 \x01(\tR\x08mimeType\x12\x10\n\x03url\x18\x05 \x01(\tR\x03url\x12\x12\n\x04size\x18\x06 \x01(\x04R\x04size\x12\x10\n\x03md5\x18\x07 \x01(\tR\x03md5\x12#\n\roriginal_path\x18\x08 \x01(\tR\x0coriginalPath\x12\x17\n\x07room_id\x18\t \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x0b \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\r \x01(\tR\npipelineId\x12<\n\rrecent_events\x18\x0e \x03(\x0b\x32\x17.corvic.status.v1.EventR\x0crecentEvents\x12>\n\ncreated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x91\x02\n\x06Source\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x06 \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\x08 \x01(\tR\npipelineId\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xa9\x05\n\x08Pipeline\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\t \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12[\n\x0fresource_inputs\x18\x04 \x03(\x0b\x32\x32.corvic.model.v1alpha.Pipeline.ResourceInputsEntryR\x0eresourceInputs\x12X\n\x0esource_outputs\x18\x05 \x03(\x0b\x32\x31.corvic.model.v1alpha.Pipeline.SourceOutputsEntryR\rsourceOutputs\x12^\n\x17pipeline_transformation\x18\x06 \x01(\x0b\x32%.corvic.orm.v1.PipelineTransformationR\x16pipelineTransformation\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x1a\x61\n\x13ResourceInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.corvic.model.v1alpha.ResourceR\x05value:\x02\x38\x01\x1a^\n\x12SourceOutputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x05value:\x02\x38\x01\x42\r\n\x0b_created_at\"\xca\x02\n\x11\x46\x65\x61tureViewSource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x34\n\x06source\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x06source\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12+\n\x11\x64rop_disconnected\x18\x04 \x01(\x08R\x10\x64ropDisconnected\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x07 \x01(\tR\x06roomIdB\r\n\x0b_created_at\"\x9c\x03\n\x0b\x46\x65\x61tureView\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12P\n\x13\x66\x65\x61ture_view_output\x18\x05 \x01(\x0b\x32 .corvic.orm.v1.FeatureViewOutputR\x11\x66\x65\x61tureViewOutput\x12Y\n\x14\x66\x65\x61ture_view_sources\x18\x06 \x03(\x0b\x32\'.corvic.model.v1alpha.FeatureViewSourceR\x12\x66\x65\x61tureViewSources\x12\x1b\n\tspace_ids\x18\x07 \x03(\tR\x08spaceIds\x12\x15\n\x06org_id\x18\x08 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xfa\x02\n\x05Space\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12I\n\x10space_parameters\x18\x05 \x01(\x0b\x32\x1e.corvic.orm.v1.SpaceParametersR\x0fspaceParameters\x12\x44\n\x0c\x66\x65\x61ture_view\x18\x06 \x01(\x0b\x32!.corvic.model.v1alpha.FeatureViewR\x0b\x66\x65\x61tureView\x12\x1b\n\tauto_sync\x18\t \x01(\x08R\x08\x61utoSync\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x85\x02\n\x05\x41gent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12I\n\x10\x61gent_parameters\x18\x04 \x01(\x0b\x32\x1e.corvic.orm.v1.AgentParametersR\x0f\x61gentParameters\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atJ\x04\x08\x06\x10\x07R\x08messages\"\xad\x02\n\x0f\x43ompletionModel\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x15\n\x06org_id\x18\x04 \x01(\tR\x05orgId\x12H\n\nparameters\x18\x05 \x01(\x0b\x32(.corvic.orm.v1.CompletionModelParametersR\nparameters\x12$\n\x0esecret_api_key\x18\x06 \x01(\tR\x0csecretApiKey\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.model.v1alpha.models_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._options = None
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_options = b'8\001'
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
+  _globals['_ROOM']._serialized_start=306
+  _globals['_ROOM']._serialized_end=450
+  _globals['_RESOURCE']._serialized_start=453
+  _globals['_RESOURCE']._serialized_end=877
+  _globals['_SOURCE']._serialized_start=880
+  _globals['_SOURCE']._serialized_end=1153
+  _globals['_PIPELINE']._serialized_start=1156
+  _globals['_PIPELINE']._serialized_end=1837
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1629
+  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1726
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1728
+  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1822
+  _globals['_FEATUREVIEWSOURCE']._serialized_start=1840
+  _globals['_FEATUREVIEWSOURCE']._serialized_end=2170
+  _globals['_FEATUREVIEW']._serialized_start=2173
+  _globals['_FEATUREVIEW']._serialized_end=2585
+  _globals['_SPACE']._serialized_start=2588
+  _globals['_SPACE']._serialized_end=2966
+  _globals['_AGENT']._serialized_start=2969
+  _globals['_AGENT']._serialized_end=3230
+  _globals['_COMPLETIONMODEL']._serialized_start=3233
+  _globals['_COMPLETIONMODEL']._serialized_end=3534
+# @@protoc_insertion_point(module_scope)

{corvic_engine-0.3.0rc45 → corvic_engine-0.3.0rc46}/python/corvic_generated/model/v1alpha/models_pb2.pyi

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

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

@@ -1,68 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: corvic/model/v1alpha/models.proto
-# Protobuf Python Version: 4.25.3
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from corvic_generated.orm.v1 import agent_pb2 as corvic_dot_orm_dot_v1_dot_agent__pb2
-from corvic_generated.orm.v1 import common_pb2 as corvic_dot_orm_dot_v1_dot_common__pb2
-from corvic_generated.orm.v1 import completion_model_pb2 as corvic_dot_orm_dot_v1_dot_completion__model__pb2
-from corvic_generated.orm.v1 import feature_view_pb2 as corvic_dot_orm_dot_v1_dot_feature__view__pb2
-from corvic_generated.orm.v1 import pipeline_pb2 as corvic_dot_orm_dot_v1_dot_pipeline__pb2
-from corvic_generated.orm.v1 import space_pb2 as corvic_dot_orm_dot_v1_dot_space__pb2
-from corvic_generated.orm.v1 import table_pb2 as corvic_dot_orm_dot_v1_dot_table__pb2
-from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot_event__pb2
-from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
-from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/model/v1alpha/models.proto\x12\x14\x63orvic.model.v1alpha\x1a\x19\x63orvic/orm/v1/agent.proto\x1a\x1a\x63orvic/orm/v1/common.proto\x1a$corvic/orm/v1/completion_model.proto\x1a corvic/orm/v1/feature_view.proto\x1a\x1c\x63orvic/orm/v1/pipeline.proto\x1a\x19\x63orvic/orm/v1/space.proto\x1a\x19\x63orvic/orm/v1/table.proto\x1a\x1c\x63orvic/status/v1/event.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x90\x01\n\x04Room\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x15\n\x06org_id\x18\x03 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xa8\x03\n\x08Resource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x1b\n\tmime_type\x18\x04 \x01(\tR\x08mimeType\x12\x10\n\x03url\x18\x05 \x01(\tR\x03url\x12\x12\n\x04size\x18\x06 \x01(\x04R\x04size\x12\x10\n\x03md5\x18\x07 \x01(\tR\x03md5\x12#\n\roriginal_path\x18\x08 \x01(\tR\x0coriginalPath\x12\x17\n\x07room_id\x18\t \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x0b \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\r \x01(\tR\npipelineId\x12<\n\rrecent_events\x18\x0e \x03(\x0b\x32\x17.corvic.status.v1.EventR\x0crecentEvents\x12>\n\ncreated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\x91\x02\n\x06Source\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12\x15\n\x06org_id\x18\x06 \x01(\tR\x05orgId\x12\x1f\n\x0bpipeline_id\x18\x08 \x01(\tR\npipelineId\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xa9\x05\n\x08Pipeline\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\t \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12[\n\x0fresource_inputs\x18\x04 \x03(\x0b\x32\x32.corvic.model.v1alpha.Pipeline.ResourceInputsEntryR\x0eresourceInputs\x12X\n\x0esource_outputs\x18\x05 \x03(\x0b\x32\x31.corvic.model.v1alpha.Pipeline.SourceOutputsEntryR\rsourceOutputs\x12^\n\x17pipeline_transformation\x18\x06 \x01(\x0b\x32%.corvic.orm.v1.PipelineTransformationR\x16pipelineTransformation\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x1a\x61\n\x13ResourceInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.corvic.model.v1alpha.ResourceR\x05value:\x02\x38\x01\x1a^\n\x12SourceOutputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x05value:\x02\x38\x01\x42\r\n\x0b_created_at\"\xca\x02\n\x11\x46\x65\x61tureViewSource\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x34\n\x06source\x18\x02 \x01(\x0b\x32\x1c.corvic.model.v1alpha.SourceR\x06source\x12\x43\n\x0etable_op_graph\x18\x03 \x01(\x0b\x32\x1d.corvic.orm.v1.TableComputeOpR\x0ctableOpGraph\x12+\n\x11\x64rop_disconnected\x18\x04 \x01(\x08R\x10\x64ropDisconnected\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x07 \x01(\tR\x06roomIdB\r\n\x0b_created_at\"\x9c\x03\n\x0b\x46\x65\x61tureView\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12P\n\x13\x66\x65\x61ture_view_output\x18\x05 \x01(\x0b\x32 .corvic.orm.v1.FeatureViewOutputR\x11\x66\x65\x61tureViewOutput\x12Y\n\x14\x66\x65\x61ture_view_sources\x18\x06 \x03(\x0b\x32\'.corvic.model.v1alpha.FeatureViewSourceR\x12\x66\x65\x61tureViewSources\x12\x1b\n\tspace_ids\x18\x07 \x03(\tR\x08spaceIds\x12\x15\n\x06org_id\x18\x08 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"\xfa\x02\n\x05Space\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomId\x12I\n\x10space_parameters\x18\x05 \x01(\x0b\x32\x1e.corvic.orm.v1.SpaceParametersR\x0fspaceParameters\x12\x44\n\x0c\x66\x65\x61ture_view\x18\x06 \x01(\x0b\x32!.corvic.model.v1alpha.FeatureViewR\x0b\x66\x65\x61tureView\x12\x1b\n\tauto_sync\x18\t \x01(\x08R\x08\x61utoSync\x12\x15\n\x06org_id\x18\x07 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at\"P\n\x0bUserMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\"\xe7\x02\n\x0c\x41gentMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12/\n\x06policy\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x06policy\x12\x18\n\x07\x63ontext\x18\x04 \x01(\tR\x07\x63ontext\x12P\n\x10message_reaction\x18\x05 \x01(\x0e\x32%.corvic.model.v1alpha.MessageReactionR\x0fmessageReaction\x12&\n\x0fuser_message_id\x18\x06 \x01(\tR\ruserMessageId\x12O\n\x12retrieved_entities\x18\x07 \x01(\x0b\x32 .corvic.orm.v1.RetrievedEntitiesR\x11retrievedEntities\x12\x17\n\x07room_id\x18\x08 \x01(\tR\x06roomId\"\xa4\x02\n\x0cMessageEntry\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x46\n\x0cuser_message\x18\x02 \x01(\x0b\x32!.corvic.model.v1alpha.UserMessageH\x00R\x0buserMessage\x12I\n\ragent_message\x18\x03 \x01(\x0b\x32\".corvic.model.v1alpha.AgentMessageH\x00R\x0c\x61gentMessage\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tcreatedAt\x88\x01\x01\x12\x17\n\x07room_id\x18\x04 \x01(\tR\x06roomIdB\t\n\x07\x63ontentB\r\n\x0b_created_at\"\x85\x02\n\x05\x41gent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x17\n\x07room_id\x18\x03 \x01(\tR\x06roomId\x12I\n\x10\x61gent_parameters\x18\x04 \x01(\x0b\x32\x1e.corvic.orm.v1.AgentParametersR\x0f\x61gentParameters\x12\x15\n\x06org_id\x18\x05 \x01(\tR\x05orgId\x12>\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_atJ\x04\x08\x06\x10\x07R\x08messages\"\xad\x02\n\x0f\x43ompletionModel\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12\x15\n\x06org_id\x18\x04 \x01(\tR\x05orgId\x12H\n\nparameters\x18\x05 \x01(\x0b\x32(.corvic.orm.v1.CompletionModelParametersR\nparameters\x12$\n\x0esecret_api_key\x18\x06 \x01(\tR\x0csecretApiKey\x12>\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x42\r\n\x0b_created_at*u\n\x0fMessageReaction\x12 \n\x1cMESSAGE_REACTION_UNSPECIFIED\x10\x00\x12\x1e\n\x1aMESSAGE_REACTION_THUMBS_UP\x10\x01\x12 \n\x1cMESSAGE_REACTION_THUMBS_DOWN\x10\x02\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.model.v1alpha.models_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._options = None
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_options = b'8\001'
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
-  _globals['_MESSAGEREACTION']._serialized_start=4333
-  _globals['_MESSAGEREACTION']._serialized_end=4450
-  _globals['_ROOM']._serialized_start=364
-  _globals['_ROOM']._serialized_end=508
-  _globals['_RESOURCE']._serialized_start=511
-  _globals['_RESOURCE']._serialized_end=935
-  _globals['_SOURCE']._serialized_start=938
-  _globals['_SOURCE']._serialized_end=1211
-  _globals['_PIPELINE']._serialized_start=1214
-  _globals['_PIPELINE']._serialized_end=1895
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1687
-  _globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1784
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1786
-  _globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1880
-  _globals['_FEATUREVIEWSOURCE']._serialized_start=1898
-  _globals['_FEATUREVIEWSOURCE']._serialized_end=2228
-  _globals['_FEATUREVIEW']._serialized_start=2231
-  _globals['_FEATUREVIEW']._serialized_end=2643
-  _globals['_SPACE']._serialized_start=2646
-  _globals['_SPACE']._serialized_end=3024
-  _globals['_USERMESSAGE']._serialized_start=3026
-  _globals['_USERMESSAGE']._serialized_end=3106
-  _globals['_AGENTMESSAGE']._serialized_start=3109
-  _globals['_AGENTMESSAGE']._serialized_end=3468
-  _globals['_MESSAGEENTRY']._serialized_start=3471
-  _globals['_MESSAGEENTRY']._serialized_end=3763
-  _globals['_AGENT']._serialized_start=3766
-  _globals['_AGENT']._serialized_end=4027
-  _globals['_COMPLETIONMODEL']._serialized_start=4030
-  _globals['_COMPLETIONMODEL']._serialized_end=4331
-# @@protoc_insertion_point(module_scope)