corvic-engine 0.3.0rc71__cp38-abi3-win_amd64.whl → 0.3.0rc72__cp38-abi3-win_amd64.whl

corvic/model/_pipeline.py

@@ -293,6 +293,33 @@ class OcrPdfsPipeline(Pipeline):
                             op_graph.feature_type.categorical(),
                         ),
                         NewColumn("title", pl.String(), op_graph.feature_type.text()),
+                        NewColumn(
+                            "resource_id",
+                            pl.String(),
+                            op_graph.feature_type.identifier(),
+                        ),
+                        NewColumn(
+                            "page_number",
+                            pl.Int64(),
+                            op_graph.feature_type.numerical(),
+                        ),
+                        NewColumn(
+                            "bbox",
+                            pl.Struct(
+                                [
+                                    pl.Field("x1", pl.Float64()),
+                                    pl.Field("y1", pl.Float64()),
+                                    pl.Field("x2", pl.Float64()),
+                                    pl.Field("y2", pl.Float64()),
+                                ]
+                            ),
+                            op_graph.feature_type.embedding(),
+                        ),
+                        NewColumn(
+                            "created_at",
+                            pl.Datetime(),
+                            op_graph.feature_type.timestamp(),
+                        ),
                     ],
                 )
             )
@@ -360,6 +387,33 @@ class OcrPdfsPipeline(Pipeline):
                             pl.String(),
                             op_graph.feature_type.foreign_key(text_source.id),
                         ),
+                        NewColumn(
+                            "resource_id",
+                            pl.String(),
+                            op_graph.feature_type.identifier(),
+                        ),
+                        NewColumn(
+                            "page_number",
+                            pl.Int64(),
+                            op_graph.feature_type.numerical(),
+                        ),
+                        NewColumn(
+                            "bbox",
+                            pl.Struct(
+                                [
+                                    pl.Field("x1", pl.Float64()),
+                                    pl.Field("y1", pl.Float64()),
+                                    pl.Field("x2", pl.Float64()),
+                                    pl.Field("y2", pl.Float64()),
+                                ]
+                            ),
+                            op_graph.feature_type.embedding(),
+                        ),
+                        NewColumn(
+                            "created_at",
+                            pl.Datetime(),
+                            op_graph.feature_type.timestamp(),
+                        ),
                     ],
                 )
             )

corvic/model/_proto_orm_convert.py

@@ -432,12 +432,14 @@ def feature_view_proto_to_orm(
             return err
     session.flush()
 
-    if not orm_obj.id:
+    if not orm_obj.id or not orm_obj.room_id:
         raise InternalError("internal assertion did not hold")
 
     new_fv_sources = list[orm.FeatureViewSource]()
     for fvs in proto_obj.feature_view_sources:
-        match _feature_view_source_proto_to_orm(fvs, orm_obj.id, session):
+        match _feature_view_source_proto_to_orm(
+            fvs, orm_obj.room_id, orm_obj.id, session
+        ):
             case orm.InvalidORMIdentifierError() | InvalidArgumentError() as err:
                 return err
             case Ok(fvs_orm):
@@ -481,6 +483,7 @@ def feature_view_proto_to_orm(
 
 def _feature_view_source_proto_to_orm(
     proto_obj: models_pb2.FeatureViewSource,
+    room_id: orm.RoomID,
     feature_view_id: orm.FeatureViewID,
     session: sa_orm.Session,
 ) -> Ok[orm.FeatureViewSource] | orm.InvalidORMIdentifierError | InvalidArgumentError:
@@ -491,6 +494,7 @@ def _feature_view_source_proto_to_orm(
             pass
 
     orm_obj = orm.FeatureViewSource(
+        room_id=room_id,
         table_op_graph=proto_obj.table_op_graph,
         drop_disconnected=proto_obj.drop_disconnected,
         source_id=source_id,

{corvic_engine-0.3.0rc71.dist-info → corvic_engine-0.3.0rc72.dist-info}/METADATA

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

{corvic_engine-0.3.0rc71.dist-info → corvic_engine-0.3.0rc72.dist-info}/RECORD

@@ -1,6 +1,6 @@
-corvic_engine-0.3.0rc71.dist-info/METADATA,sha256=cJkn4ymCzB4gGENGUthmUeQngNSeyO9nQ2zTzHrA5v0,1814
-corvic_engine-0.3.0rc71.dist-info/WHEEL,sha256=hKPP3BCTWtTwj6SFaSI--T5aOGqh_llYfbZ_BsqivwA,94
-corvic_engine-0.3.0rc71.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
+corvic_engine-0.3.0rc72.dist-info/METADATA,sha256=bPX0o4_5zdRiLQec28odvgxggU16H9wVesyR63TKMwk,1814
+corvic_engine-0.3.0rc72.dist-info/WHEEL,sha256=hKPP3BCTWtTwj6SFaSI--T5aOGqh_llYfbZ_BsqivwA,94
+corvic_engine-0.3.0rc72.dist-info/licenses/LICENSE,sha256=DSS1OD0oIgssKOmAzkMRBv5jvvVuZQbrIv8lpl9DXY8,1035
 corvic/context/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 corvic/context/__init__.py,sha256=J69SWL27p4euoS3e_Z1K-rqSFRzGdBT3ryuRfM9r9cM,1498
 corvic/embed/node2vec.py,sha256=Qep1lYMKN4nL4z6Ftylr0pj2aJ2LJmWRmnZV27xYJ-M,11251
@@ -19,8 +19,8 @@ corvic/model/_defaults.py,sha256=yoKPPSmYJCE5YAD5jLTEmT4XNf_zXoggNK-uyG8MfVs,152
 corvic/model/_errors.py,sha256=Ctlq04SDwHzJPvLaL1rzqzwVqf2b50EILfW3cH4vnh8,261
 corvic/model/_feature_type.py,sha256=Y-_-wa9fv7XaCAkxfjjoCLxxK2Ftfba-PMefD7bNXzs,917
 corvic/model/_feature_view.py,sha256=vRh9eVDlais8enZVymQtwPz8vd3QtwSRYR1CnlKtCnA,49698
-corvic/model/_pipeline.py,sha256=rAhs3gawRORJnnOtjtyWMz6HkTBkVJlyWAP1jB45q4c,16249
-corvic/model/_proto_orm_convert.py,sha256=OnpHJGfIH_J8RtdNZBL1_TYvd8FjkFTkoNfbVFt5vbM,23506
+corvic/model/_pipeline.py,sha256=X9b34eCTD4zr6IF61EZRLC89n_Fi9HrsukoghDKwhI8,18685
+corvic/model/_proto_orm_convert.py,sha256=eqYx7rJlshDRfoA199F1BEDSnCVQDC-xmEVqs6mU1pw,23622
 corvic/model/_resource.py,sha256=R973-POS5HDCo7hIoxsBNauH1YAPisZDFLrWIxjygbk,8495
 corvic/model/_room.py,sha256=36mXngZ38L4mr6_LgUm-QgsUUaoGMiYQRfvXLV_jd-4,2914
 corvic/model/_source.py,sha256=evgqs_6-IK2dl35EJpcc3rD5yTCUZudcQYL24vLTElg,9785
@@ -206,5 +206,5 @@ corvic_generated/status/v1/event_pb2.pyi,sha256=eU-ibrYpvEAJSIDlSa62-bC96AQU1ykF
 corvic_generated/status/v1/event_pb2_grpc.pyi,sha256=H9-ADaiKR9iyVZvmnXutZqWwRRCDxjUIktkfJrJFIHg,417
 corvic_generated/status/v1/service_pb2.pyi,sha256=iXLR2FOKQJpBgvBzpD2kVwcYOCksP2aRwK4JYaI9CBw,558
 corvic_generated/status/v1/service_pb2_grpc.pyi,sha256=OoAnaZ64FD0UTzPoRhYvQU8ecoilhHj3ySjSfHbVDaU,1501
-corvic/engine/_native.pyd,sha256=6Z2NRW6p7UWKoFEQqYEA5f6d8EpqdiWHB-mvmCgRIbI,438272
-corvic_engine-0.3.0rc71.dist-info/RECORD,,
+corvic/engine/_native.pyd,sha256=TrxxmpS21HXsRPgH__iDtGaYp2g9c24ImGODbfOeMUA,438272
+corvic_engine-0.3.0rc72.dist-info/RECORD,,