corvic-engine 0.3.0rc45__cp38-abi3-win_amd64.whl → 0.3.0rc46__cp38-abi3-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- corvic/engine/_native.pyd +0 -0
- corvic/orm/__init__.py +0 -86
- corvic/system/storage.py +1 -1
- corvic/system_sqlite/fs_blob_store.py +9 -3
- corvic/system_sqlite/rdbms_blob_store.py +3 -2
- {corvic_engine-0.3.0rc45.dist-info → corvic_engine-0.3.0rc46.dist-info}/METADATA +1 -1
- {corvic_engine-0.3.0rc45.dist-info → corvic_engine-0.3.0rc46.dist-info}/RECORD +11 -11
- corvic_generated/model/v1alpha/models_pb2.py +23 -33
- corvic_generated/model/v1alpha/models_pb2.pyi +0 -56
- {corvic_engine-0.3.0rc45.dist-info → corvic_engine-0.3.0rc46.dist-info}/WHEEL +0 -0
- {corvic_engine-0.3.0rc45.dist-info → corvic_engine-0.3.0rc46.dist-info}/licenses/LICENSE +0 -0
corvic/engine/_native.pyd
CHANGED
|
Binary file
|
corvic/orm/__init__.py
CHANGED
|
@@ -4,7 +4,6 @@ from __future__ import annotations
|
|
|
4
4
|
|
|
5
5
|
import sqlalchemy as sa
|
|
6
6
|
from sqlalchemy import orm as sa_orm
|
|
7
|
-
from sqlalchemy.orm.collections import attribute_mapped_collection
|
|
8
7
|
|
|
9
8
|
from corvic.orm._proto_columns import ProtoMessageDecorator
|
|
10
9
|
from corvic.orm.base import Base, OrgBase
|
|
@@ -351,12 +350,6 @@ class Agent(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
|
|
|
351
350
|
parameters: sa_orm.Mapped[agent_pb2.AgentParameters | None] = sa_orm.mapped_column(
|
|
352
351
|
default=None
|
|
353
352
|
)
|
|
354
|
-
messages: sa_orm.Mapped[dict[str, MessageEntry]] = sa_orm.relationship(
|
|
355
|
-
collection_class=attribute_mapped_collection("entry_id"),
|
|
356
|
-
default_factory=dict,
|
|
357
|
-
init=False,
|
|
358
|
-
viewonly=True,
|
|
359
|
-
)
|
|
360
353
|
|
|
361
354
|
space_associations: sa_orm.Mapped[list[AgentSpaceAssociation]] = (
|
|
362
355
|
sa_orm.relationship(default_factory=list, init=False, viewonly=True)
|
|
@@ -377,82 +370,6 @@ class AgentSpaceAssociation(BelongsToOrgMixin, BelongsToRoomMixin, Base):
|
|
|
377
370
|
)
|
|
378
371
|
|
|
379
372
|
|
|
380
|
-
class UserMessage(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
|
|
381
|
-
"""A message sent by an user."""
|
|
382
|
-
|
|
383
|
-
__tablename__ = "user_message"
|
|
384
|
-
|
|
385
|
-
message_entry: sa_orm.Mapped[MessageEntry] = sa_orm.relationship(
|
|
386
|
-
init=True, default=None
|
|
387
|
-
)
|
|
388
|
-
|
|
389
|
-
id: sa_orm.Mapped[UserMessageID | None] = primary_key_identity_column()
|
|
390
|
-
|
|
391
|
-
message: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
|
|
392
|
-
|
|
393
|
-
|
|
394
|
-
class AgentMessage(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
|
|
395
|
-
"""A message sent by an agent."""
|
|
396
|
-
|
|
397
|
-
__tablename__ = "agent_message"
|
|
398
|
-
message_metadata: sa_orm.Mapped[common_pb2.AgentMessageMetadata | None] = (
|
|
399
|
-
sa_orm.mapped_column(
|
|
400
|
-
default_factory=lambda: common_pb2.AgentMessageMetadata(
|
|
401
|
-
message_reaction=common_pb2.MessageReaction.MESSAGE_REACTION_UNSPECIFIED
|
|
402
|
-
)
|
|
403
|
-
)
|
|
404
|
-
)
|
|
405
|
-
id: sa_orm.Mapped[AgentMessageID | None] = primary_key_identity_column()
|
|
406
|
-
|
|
407
|
-
message_entry: sa_orm.Mapped[MessageEntry] = sa_orm.relationship(
|
|
408
|
-
init=True, default=None
|
|
409
|
-
)
|
|
410
|
-
|
|
411
|
-
user_message_id: sa_orm.Mapped[UserMessageID | None] = sa_orm.mapped_column(
|
|
412
|
-
ForeignKey(UserMessage).make(ondelete="CASCADE"), init=True, default=None
|
|
413
|
-
)
|
|
414
|
-
message: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
|
|
415
|
-
policy: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
|
|
416
|
-
context: sa_orm.Mapped[str | None] = sa_orm.mapped_column(sa.Text, default=None)
|
|
417
|
-
retrieved_entities: sa_orm.Mapped[common_pb2.RetrievedEntities | None] = (
|
|
418
|
-
sa_orm.mapped_column(default=None)
|
|
419
|
-
)
|
|
420
|
-
|
|
421
|
-
|
|
422
|
-
class MessageEntry(SoftDeleteMixin, BelongsToOrgMixin, BelongsToRoomMixin, Base):
|
|
423
|
-
"""A message either sent by an Agent or an User."""
|
|
424
|
-
|
|
425
|
-
__tablename__ = "message_entry"
|
|
426
|
-
|
|
427
|
-
id: sa_orm.Mapped[MessageEntryID | None] = primary_key_identity_column()
|
|
428
|
-
|
|
429
|
-
agent_id: sa_orm.Mapped[AgentID] = sa_orm.mapped_column(
|
|
430
|
-
ForeignKey(Agent).make(ondelete="CASCADE"),
|
|
431
|
-
nullable=True,
|
|
432
|
-
init=True,
|
|
433
|
-
default=None,
|
|
434
|
-
)
|
|
435
|
-
|
|
436
|
-
agent_message_id: sa_orm.Mapped[AgentMessageID | None] = sa_orm.mapped_column(
|
|
437
|
-
ForeignKey(AgentMessage).make(ondelete="CASCADE"), default=None
|
|
438
|
-
)
|
|
439
|
-
user_message_id: sa_orm.Mapped[UserMessageID | None] = sa_orm.mapped_column(
|
|
440
|
-
ForeignKey(UserMessage).make(ondelete="CASCADE"), default=None
|
|
441
|
-
)
|
|
442
|
-
|
|
443
|
-
agent_message: sa_orm.Mapped[AgentMessage | None] = sa_orm.relationship(
|
|
444
|
-
back_populates="message_entry", init=True, default=None
|
|
445
|
-
)
|
|
446
|
-
|
|
447
|
-
user_message: sa_orm.Mapped[UserMessage | None] = sa_orm.relationship(
|
|
448
|
-
back_populates="message_entry", init=True, default=None
|
|
449
|
-
)
|
|
450
|
-
|
|
451
|
-
@property
|
|
452
|
-
def entry_id(self):
|
|
453
|
-
return self.id
|
|
454
|
-
|
|
455
|
-
|
|
456
373
|
class CompletionModel(SoftDeleteMixin, BelongsToOrgMixin, Base):
|
|
457
374
|
"""A customer's custom completion model definition."""
|
|
458
375
|
|
|
@@ -494,7 +411,6 @@ ID = (
|
|
|
494
411
|
__all__ = [
|
|
495
412
|
"Agent",
|
|
496
413
|
"AgentID",
|
|
497
|
-
"AgentMessage",
|
|
498
414
|
"AgentMessageID",
|
|
499
415
|
"Base",
|
|
500
416
|
"BaseID",
|
|
@@ -510,7 +426,6 @@ __all__ = [
|
|
|
510
426
|
"FeatureViewSourceID",
|
|
511
427
|
"ID",
|
|
512
428
|
"InvalidORMIdentifierError",
|
|
513
|
-
"MessageEntry",
|
|
514
429
|
"MessageEntryID",
|
|
515
430
|
"Org",
|
|
516
431
|
"OrgID",
|
|
@@ -530,7 +445,6 @@ __all__ = [
|
|
|
530
445
|
"SpaceParametersID",
|
|
531
446
|
"SpaceRun",
|
|
532
447
|
"SpaceRunID",
|
|
533
|
-
"UserMessage",
|
|
534
448
|
"UserMessageID",
|
|
535
449
|
"primary_key_foreign_column",
|
|
536
450
|
"primary_key_identity_column",
|
corvic/system/storage.py
CHANGED
|
@@ -228,10 +228,16 @@ class FSBucket:
|
|
|
228
228
|
def create(self) -> None:
|
|
229
229
|
return self.path().mkdir()
|
|
230
230
|
|
|
231
|
-
def list_blobs(self) -> Iterator[FSBlob]:
|
|
231
|
+
def list_blobs(self, prefix: str | None = None) -> Iterator[FSBlob]:
|
|
232
232
|
for entry in self.path().glob("*"):
|
|
233
|
-
if
|
|
234
|
-
|
|
233
|
+
if entry.name.endswith(".metadata"):
|
|
234
|
+
continue
|
|
235
|
+
true_name = base64.urlsafe_b64decode(entry.name.encode("utf-8")).decode(
|
|
236
|
+
"utf-8"
|
|
237
|
+
)
|
|
238
|
+
if prefix and not true_name.startswith(prefix):
|
|
239
|
+
continue
|
|
240
|
+
yield FSBlob.from_path(self, entry)
|
|
235
241
|
|
|
236
242
|
def path(self) -> Path:
|
|
237
243
|
return self.client.root() / self.name
|
|
@@ -277,14 +277,15 @@ class RDBMSBucket:
|
|
|
277
277
|
session.add(ORMBucket(name=self._name))
|
|
278
278
|
session.commit()
|
|
279
279
|
|
|
280
|
-
def list_blobs(self) -> Iterator[RDBMSBlob]:
|
|
280
|
+
def list_blobs(self, prefix: str | None = None) -> Iterator[RDBMSBlob]:
|
|
281
281
|
def get_blobs():
|
|
282
282
|
with self._client.db_session() as session:
|
|
283
283
|
orm_self = self._to_orm(session)
|
|
284
284
|
if not orm_self:
|
|
285
285
|
raise ValueError("bucket does not exist")
|
|
286
286
|
for blob in orm_self.blobs:
|
|
287
|
-
|
|
287
|
+
if not prefix or blob.name.startswith(prefix):
|
|
288
|
+
yield RDBMSBlob(self, blob.name)
|
|
288
289
|
|
|
289
290
|
return iter(list(get_blobs()))
|
|
290
291
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
corvic_engine-0.3.
|
|
2
|
-
corvic_engine-0.3.
|
|
3
|
-
corvic_engine-0.3.
|
|
1
|
+
corvic_engine-0.3.0rc46.dist-info/METADATA,sha256=_LLt2PwaSBy5Jy3CuT9BHywURG_SkNOW0g4yJ9yahC8,1876
|
|
2
|
+
corvic_engine-0.3.0rc46.dist-info/WHEEL,sha256=_g1M2QM3kt1Ssm_sHOg_3TUY7GxNE2Ueyslb9ZDtPwk,94
|
|
3
|
+
corvic_engine-0.3.0rc46.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
|
|
4
4
|
corvic/context/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
5
5
|
corvic/context/__init__.py,sha256=zBnPiP-tStGSVMG_0-G_0ay6-yIX2aerW_oYRzAex74,1702
|
|
6
6
|
corvic/embed/node2vec.py,sha256=JnYb8f2g4XhF6LL2TjpMxLfKhn_Yp1AzptsWwrKQWgc,11146
|
|
@@ -50,7 +50,7 @@ corvic/orm/keys.py,sha256=Ag6Xbpvxev-VByT1KJ8ChUn9vKVEzkkMXxrjvtADCtY,2182
|
|
|
50
50
|
corvic/orm/mixins.py,sha256=HfmzJ7LblHtddbbkDmv7nNWURL87Bnj8NeOnNbfmSN4,17794
|
|
51
51
|
corvic/orm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
52
52
|
corvic/orm/_proto_columns.py,sha256=tcOu92UjFJFYZLasS6sWJQBDRK26yrnmpTii_LDY4iw,913
|
|
53
|
-
corvic/orm/__init__.py,sha256=
|
|
53
|
+
corvic/orm/__init__.py,sha256=3HLMICNB1iMyMS70tlvaap8WDrAfxqSxp7W1O-cnHc4,14656
|
|
54
54
|
corvic/pa_scalar/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
55
55
|
corvic/pa_scalar/_const.py,sha256=1nk6w3Y7crd3J5jSCq7DRVa1lcGk4H1RUr1l4NjnlzE,868
|
|
56
56
|
corvic/pa_scalar/_from_value.py,sha256=fS3TNPcPI3jAKGmcUIhn8rdqdQEAwgTLEneVxFUeK6M,27531
|
|
@@ -72,7 +72,7 @@ corvic/system/in_memory_executor.py,sha256=dYgcxbA_O0mM1pI19t2OXs8q5B4TX-NFacR7T
|
|
|
72
72
|
corvic/system/op_graph_executor.py,sha256=gXFnVkemS5EwNegJdU-xVAfMLPULqMFPF7d3EG3AD_U,3482
|
|
73
73
|
corvic/system/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
74
74
|
corvic/system/staging.py,sha256=K5P5moiuAMfPx7lxK4mArxeURBwKoyB6x9HGu9JJ16E,1846
|
|
75
|
-
corvic/system/storage.py,sha256=
|
|
75
|
+
corvic/system/storage.py,sha256=N4lXnoCFPg8JbUaI9fbpPY0nt3CgXFUc8sZteki_nRk,5671
|
|
76
76
|
corvic/system/_dimension_reduction.py,sha256=vyD8wOs0vE-hlVnCrBTjStTowAPWYREqnQ_bVuGYvis,2907
|
|
77
77
|
corvic/system/_embedder.py,sha256=0WO24IKi8VC8jsFdvNuzDsgNejyacQf6r9Q34jxhHc4,3844
|
|
78
78
|
corvic/system/_image_embedder.py,sha256=iQc3KlLcqrhP6K84hncHutThAN8Qd6K7K5dceHyU1TU,8373
|
|
@@ -80,9 +80,9 @@ corvic/system/_planner.py,sha256=HUf6UjCy1iHRrXfhU25w19TG4Ik3zVHhtzVcor0eTQY,788
|
|
|
80
80
|
corvic/system/_text_embedder.py,sha256=LH79_4RxhvssySHpkeEoZFgM4Sa5XAYSjoytdsuwWK4,3269
|
|
81
81
|
corvic/system/__init__.py,sha256=DBzMQ1QCO0OHo3w2LoibfgN1X3CLGBUxXjiExN2kyi0,2838
|
|
82
82
|
corvic/system_sqlite/client.py,sha256=ntDJtCDG1SDGUKI-BhCju4sdqd59Y8RucGZ8VmRvHuI,7142
|
|
83
|
-
corvic/system_sqlite/fs_blob_store.py,sha256=
|
|
83
|
+
corvic/system_sqlite/fs_blob_store.py,sha256=pYTMPiWYC6AUIdcgmRj8lvL7Chg82rf5dac6bKGaqL0,8461
|
|
84
84
|
corvic/system_sqlite/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
85
|
-
corvic/system_sqlite/rdbms_blob_store.py,sha256=
|
|
85
|
+
corvic/system_sqlite/rdbms_blob_store.py,sha256=gTP_tQfTVb3wzZkzo8ys1zaz0rSrERzb57rqMHVpuBA,10563
|
|
86
86
|
corvic/system_sqlite/staging.py,sha256=9chn6iXGmKAsurdiBzo0FSTyOmeDn7oCIq8fekbK67c,16878
|
|
87
87
|
corvic/system_sqlite/__init__.py,sha256=MQCUux08NcxFLghpWcZ95lt5AMabr-6sJRH-eSsTnqI,229
|
|
88
88
|
corvic/table/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
@@ -156,7 +156,7 @@ corvic_generated/ingest/v2/table_pb2.py,sha256=aTJHaliZm5DMtp7gslNxyn9uDagz-2-_e
|
|
|
156
156
|
corvic_generated/ingest/v2/table_pb2_grpc.py,sha256=tVs7wMWyAfvHcCQEiUOHLwaptKxgMFG6E7Ki9vNmmvQ,8151
|
|
157
157
|
corvic_generated/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
158
158
|
corvic_generated/model/v1alpha/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
159
|
-
corvic_generated/model/v1alpha/models_pb2.py,sha256=
|
|
159
|
+
corvic_generated/model/v1alpha/models_pb2.py,sha256=ncqP8kRV14sjRbL9KIjPB79J-eVFVULCNE1qq7RAMx0,8892
|
|
160
160
|
corvic_generated/model/v1alpha/models_pb2_grpc.py,sha256=_bXoS025FcWrXR1E_3Mh4GHB1RMvgz8lIpit-Awnf-s,163
|
|
161
161
|
corvic_generated/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
162
162
|
corvic_generated/orm/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
@@ -220,7 +220,7 @@ corvic_generated/ingest/v2/source_pb2.pyi,sha256=k7FdbgurQLk0JA1WiTUerznzxLv8b50
|
|
|
220
220
|
corvic_generated/ingest/v2/source_pb2_grpc.pyi,sha256=VG5gpql2SREHgqMC_ycT-QJBVpPeSYKOYS2COgGrZa4,6195
|
|
221
221
|
corvic_generated/ingest/v2/table_pb2.pyi,sha256=p22F8kv0HfM-9OzGP88bLofxmUtxfLR5eVN0HOxXiEo,4382
|
|
222
222
|
corvic_generated/ingest/v2/table_pb2_grpc.pyi,sha256=AEXYNtrU4xyENumcCrkD2FmFV7T1UVidxxeZ5pyE4Qc,4554
|
|
223
|
-
corvic_generated/model/v1alpha/models_pb2.pyi,sha256
|
|
223
|
+
corvic_generated/model/v1alpha/models_pb2.pyi,sha256=eKV5BlocI0MRQuzPVN4eGFrjysyPco2dyszFo7s66ys,11462
|
|
224
224
|
corvic_generated/model/v1alpha/models_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
|
|
225
225
|
corvic_generated/orm/v1/agent_pb2.pyi,sha256=9AExLKFRvOJ1fSaOdZc3Src015uvNl-le2lSPatZjSQ,4575
|
|
226
226
|
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
|
|
|
244
244
|
corvic_generated/status/v1/event_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
|
|
245
245
|
corvic_generated/status/v1/service_pb2.pyi,sha256=iXLR2FOKQJpBgvBzpD2kVwcYOCksP2aRwK4JYaI9CBw,558
|
|
246
246
|
corvic_generated/status/v1/service_pb2_grpc.pyi,sha256=OoAnaZ64FD0UTzPoRhYvQU8ecoilhHj3ySjSfHbVDaU,1501
|
|
247
|
-
corvic/engine/_native.pyd,sha256=
|
|
248
|
-
corvic_engine-0.3.
|
|
247
|
+
corvic/engine/_native.pyd,sha256=6ts0z2Hw7EtkPpwjGCBvZqq9Pg282NoTZk5EwiiLZtc,438272
|
|
248
|
+
corvic_engine-0.3.0rc46.dist-info/RECORD,,
|
|
@@ -13,18 +13,16 @@ _sym_db = _symbol_database.Default()
|
|
|
13
13
|
|
|
14
14
|
|
|
15
15
|
from corvic_generated.orm.v1 import agent_pb2 as corvic_dot_orm_dot_v1_dot_agent__pb2
|
|
16
|
-
from corvic_generated.orm.v1 import common_pb2 as corvic_dot_orm_dot_v1_dot_common__pb2
|
|
17
16
|
from corvic_generated.orm.v1 import completion_model_pb2 as corvic_dot_orm_dot_v1_dot_completion__model__pb2
|
|
18
17
|
from corvic_generated.orm.v1 import feature_view_pb2 as corvic_dot_orm_dot_v1_dot_feature__view__pb2
|
|
19
18
|
from corvic_generated.orm.v1 import pipeline_pb2 as corvic_dot_orm_dot_v1_dot_pipeline__pb2
|
|
20
19
|
from corvic_generated.orm.v1 import space_pb2 as corvic_dot_orm_dot_v1_dot_space__pb2
|
|
21
20
|
from corvic_generated.orm.v1 import table_pb2 as corvic_dot_orm_dot_v1_dot_table__pb2
|
|
22
21
|
from corvic_generated.status.v1 import event_pb2 as corvic_dot_status_dot_v1_dot_event__pb2
|
|
23
|
-
from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
|
|
24
22
|
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
25
23
|
|
|
26
24
|
|
|
27
|
-
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
|
|
25
|
+
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')
|
|
28
26
|
|
|
29
27
|
_globals = globals()
|
|
30
28
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
@@ -35,34 +33,26 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
35
33
|
_globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_options = b'8\001'
|
|
36
34
|
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._options = None
|
|
37
35
|
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_options = b'8\001'
|
|
38
|
-
_globals['
|
|
39
|
-
_globals['
|
|
40
|
-
_globals['
|
|
41
|
-
_globals['
|
|
42
|
-
_globals['
|
|
43
|
-
_globals['
|
|
44
|
-
_globals['
|
|
45
|
-
_globals['
|
|
46
|
-
_globals['
|
|
47
|
-
_globals['
|
|
48
|
-
_globals['
|
|
49
|
-
_globals['
|
|
50
|
-
_globals['
|
|
51
|
-
_globals['
|
|
52
|
-
_globals['
|
|
53
|
-
_globals['
|
|
54
|
-
_globals['
|
|
55
|
-
_globals['
|
|
56
|
-
_globals['
|
|
57
|
-
_globals['
|
|
58
|
-
_globals['
|
|
59
|
-
_globals['
|
|
60
|
-
_globals['_AGENTMESSAGE']._serialized_start=3109
|
|
61
|
-
_globals['_AGENTMESSAGE']._serialized_end=3468
|
|
62
|
-
_globals['_MESSAGEENTRY']._serialized_start=3471
|
|
63
|
-
_globals['_MESSAGEENTRY']._serialized_end=3763
|
|
64
|
-
_globals['_AGENT']._serialized_start=3766
|
|
65
|
-
_globals['_AGENT']._serialized_end=4027
|
|
66
|
-
_globals['_COMPLETIONMODEL']._serialized_start=4030
|
|
67
|
-
_globals['_COMPLETIONMODEL']._serialized_end=4331
|
|
36
|
+
_globals['_ROOM']._serialized_start=306
|
|
37
|
+
_globals['_ROOM']._serialized_end=450
|
|
38
|
+
_globals['_RESOURCE']._serialized_start=453
|
|
39
|
+
_globals['_RESOURCE']._serialized_end=877
|
|
40
|
+
_globals['_SOURCE']._serialized_start=880
|
|
41
|
+
_globals['_SOURCE']._serialized_end=1153
|
|
42
|
+
_globals['_PIPELINE']._serialized_start=1156
|
|
43
|
+
_globals['_PIPELINE']._serialized_end=1837
|
|
44
|
+
_globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_start=1629
|
|
45
|
+
_globals['_PIPELINE_RESOURCEINPUTSENTRY']._serialized_end=1726
|
|
46
|
+
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_start=1728
|
|
47
|
+
_globals['_PIPELINE_SOURCEOUTPUTSENTRY']._serialized_end=1822
|
|
48
|
+
_globals['_FEATUREVIEWSOURCE']._serialized_start=1840
|
|
49
|
+
_globals['_FEATUREVIEWSOURCE']._serialized_end=2170
|
|
50
|
+
_globals['_FEATUREVIEW']._serialized_start=2173
|
|
51
|
+
_globals['_FEATUREVIEW']._serialized_end=2585
|
|
52
|
+
_globals['_SPACE']._serialized_start=2588
|
|
53
|
+
_globals['_SPACE']._serialized_end=2966
|
|
54
|
+
_globals['_AGENT']._serialized_start=2969
|
|
55
|
+
_globals['_AGENT']._serialized_end=3230
|
|
56
|
+
_globals['_COMPLETIONMODEL']._serialized_start=3233
|
|
57
|
+
_globals['_COMPLETIONMODEL']._serialized_end=3534
|
|
68
58
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -1,30 +1,18 @@
|
|
|
1
1
|
from corvic_generated.orm.v1 import agent_pb2 as _agent_pb2
|
|
2
|
-
from corvic_generated.orm.v1 import common_pb2 as _common_pb2
|
|
3
2
|
from corvic_generated.orm.v1 import completion_model_pb2 as _completion_model_pb2
|
|
4
3
|
from corvic_generated.orm.v1 import feature_view_pb2 as _feature_view_pb2
|
|
5
4
|
from corvic_generated.orm.v1 import pipeline_pb2 as _pipeline_pb2
|
|
6
5
|
from corvic_generated.orm.v1 import space_pb2 as _space_pb2
|
|
7
6
|
from corvic_generated.orm.v1 import table_pb2 as _table_pb2
|
|
8
7
|
from corvic_generated.status.v1 import event_pb2 as _event_pb2
|
|
9
|
-
from google.protobuf import struct_pb2 as _struct_pb2
|
|
10
8
|
from google.protobuf import timestamp_pb2 as _timestamp_pb2
|
|
11
9
|
from google.protobuf.internal import containers as _containers
|
|
12
|
-
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
|
|
13
10
|
from google.protobuf import descriptor as _descriptor
|
|
14
11
|
from google.protobuf import message as _message
|
|
15
12
|
from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
16
13
|
|
|
17
14
|
DESCRIPTOR: _descriptor.FileDescriptor
|
|
18
15
|
|
|
19
|
-
class MessageReaction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
|
20
|
-
__slots__ = ()
|
|
21
|
-
MESSAGE_REACTION_UNSPECIFIED: _ClassVar[MessageReaction]
|
|
22
|
-
MESSAGE_REACTION_THUMBS_UP: _ClassVar[MessageReaction]
|
|
23
|
-
MESSAGE_REACTION_THUMBS_DOWN: _ClassVar[MessageReaction]
|
|
24
|
-
MESSAGE_REACTION_UNSPECIFIED: MessageReaction
|
|
25
|
-
MESSAGE_REACTION_THUMBS_UP: MessageReaction
|
|
26
|
-
MESSAGE_REACTION_THUMBS_DOWN: MessageReaction
|
|
27
|
-
|
|
28
16
|
class Room(_message.Message):
|
|
29
17
|
__slots__ = ("id", "name", "org_id", "created_at")
|
|
30
18
|
ID_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -183,50 +171,6 @@ class Space(_message.Message):
|
|
|
183
171
|
created_at: _timestamp_pb2.Timestamp
|
|
184
172
|
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: ...
|
|
185
173
|
|
|
186
|
-
class UserMessage(_message.Message):
|
|
187
|
-
__slots__ = ("id", "message", "room_id")
|
|
188
|
-
ID_FIELD_NUMBER: _ClassVar[int]
|
|
189
|
-
MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
190
|
-
ROOM_ID_FIELD_NUMBER: _ClassVar[int]
|
|
191
|
-
id: str
|
|
192
|
-
message: str
|
|
193
|
-
room_id: str
|
|
194
|
-
def __init__(self, id: _Optional[str] = ..., message: _Optional[str] = ..., room_id: _Optional[str] = ...) -> None: ...
|
|
195
|
-
|
|
196
|
-
class AgentMessage(_message.Message):
|
|
197
|
-
__slots__ = ("id", "message", "policy", "context", "message_reaction", "user_message_id", "retrieved_entities", "room_id")
|
|
198
|
-
ID_FIELD_NUMBER: _ClassVar[int]
|
|
199
|
-
MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
200
|
-
POLICY_FIELD_NUMBER: _ClassVar[int]
|
|
201
|
-
CONTEXT_FIELD_NUMBER: _ClassVar[int]
|
|
202
|
-
MESSAGE_REACTION_FIELD_NUMBER: _ClassVar[int]
|
|
203
|
-
USER_MESSAGE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
204
|
-
RETRIEVED_ENTITIES_FIELD_NUMBER: _ClassVar[int]
|
|
205
|
-
ROOM_ID_FIELD_NUMBER: _ClassVar[int]
|
|
206
|
-
id: str
|
|
207
|
-
message: str
|
|
208
|
-
policy: _struct_pb2.Struct
|
|
209
|
-
context: str
|
|
210
|
-
message_reaction: MessageReaction
|
|
211
|
-
user_message_id: str
|
|
212
|
-
retrieved_entities: _common_pb2.RetrievedEntities
|
|
213
|
-
room_id: str
|
|
214
|
-
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: ...
|
|
215
|
-
|
|
216
|
-
class MessageEntry(_message.Message):
|
|
217
|
-
__slots__ = ("id", "user_message", "agent_message", "created_at", "room_id")
|
|
218
|
-
ID_FIELD_NUMBER: _ClassVar[int]
|
|
219
|
-
USER_MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
220
|
-
AGENT_MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
221
|
-
CREATED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
222
|
-
ROOM_ID_FIELD_NUMBER: _ClassVar[int]
|
|
223
|
-
id: str
|
|
224
|
-
user_message: UserMessage
|
|
225
|
-
agent_message: AgentMessage
|
|
226
|
-
created_at: _timestamp_pb2.Timestamp
|
|
227
|
-
room_id: str
|
|
228
|
-
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: ...
|
|
229
|
-
|
|
230
174
|
class Agent(_message.Message):
|
|
231
175
|
__slots__ = ("id", "name", "room_id", "agent_parameters", "org_id", "created_at")
|
|
232
176
|
ID_FIELD_NUMBER: _ClassVar[int]
|
|
File without changes
|
|
File without changes
|