corvic-engine 0.3.0rc49__tar.gz → 0.3.0rc51__tar.gz

{corvic_engine-0.3.0rc49 → corvic_engine-0.3.0rc51}/Cargo.lock

@@ -22,7 +22,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.49"
+version = "0.3.0-rc.51"
 dependencies = [
  "numpy",
  "pyo3",
@@ -63,7 +63,7 @@ checksum = "60b1af1c220855b6ceac025d3f6ecdd2b7c4894bfe9cd9bda4fbb4bc7c0d4cf0"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.49"
+version = "0.3.0-rc.51"
 dependencies = [
  "version_check",
 ]

{corvic_engine-0.3.0rc49 → corvic_engine-0.3.0rc51}/Cargo.toml

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

{corvic_engine-0.3.0rc49 → corvic_engine-0.3.0rc51}/PKG-INFO

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

{corvic_engine-0.3.0rc49 → corvic_engine-0.3.0rc51}/pyproject.toml

@@ -20,7 +20,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.49" # sync this with tool.poetry.version below
+version = "0.3.0-rc.51" # 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.49"
+version = "0.3.0-rc.51"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"
@@ -167,7 +167,7 @@ torch = [
 ]
 torch-geometric = "^2.5.3"
 deepgnn-torch = "^0.1.60"
-transformers = "^4.45.0"
+transformers = "^4.48.0"
 umap-learn = "^0.5.5"
 pillow = "10.4.0"
 

{corvic_engine-0.3.0rc49 → corvic_engine-0.3.0rc51}/python/corvic/system/storage.py

@@ -173,3 +173,12 @@ class StorageManager:
 
     def get_vector_blob_from_blob_name(self, blob_name: str) -> Blob:
         return self.bucket.blob(f"{self.vector_prefix}/{blob_name}")
+
+    def tabular_prefix_for_room_id(self, room_id: orm.RoomID) -> str:
+        return f"{self.tabular_prefix}/{self._render_room_id(room_id)}"
+
+    def unstructured_prefix_for_room_id(self, room_id: orm.RoomID) -> str:
+        return f"{self.unstructured_prefix}/{self._render_room_id(room_id)}"
+
+    def vector_prefix_for_room_id(self, room_id: orm.RoomID) -> str:
+        return f"{self.vector_prefix}/{self._render_room_id(room_id)}"

corvic_engine-0.3.0rc51/python/corvic/system_sqlite/__init__.py

@@ -0,0 +1,19 @@
+"""Single-machine, sqlite-backed implementation of corvic.system."""
+
+from corvic.system_sqlite.client import (
+    TABULAR_PREFIX,
+    UNSTRUCTURED_PREFIX,
+    VECTOR_PREFIX,
+    Client,
+    FSBlobClient,
+    RDBMSBlobClient,
+)
+
+__all__ = [
+    "Client",
+    "FSBlobClient",
+    "RDBMSBlobClient",
+    "TABULAR_PREFIX",
+    "UNSTRUCTURED_PREFIX",
+    "VECTOR_PREFIX",
+]

{corvic_engine-0.3.0rc49 → corvic_engine-0.3.0rc51}/python/corvic/system_sqlite/client.py

@@ -20,6 +20,10 @@ from corvic.system_sqlite.fs_blob_store import FSBlobClient
 from corvic.system_sqlite.rdbms_blob_store import RDBMSBlobClient
 from corvic.system_sqlite.staging import DuckDBStaging
 
+UNSTRUCTURED_PREFIX = "unstructured_data"
+TABULAR_PREFIX = "tabular_data"
+VECTOR_PREFIX = "vectors"
+
 
 @contextlib.contextmanager
 def _context_requester_org_id_is(value: str | corvic.orm.OrgID) -> Iterator[None]:
@@ -133,9 +137,9 @@ class Client(corvic.system.Client):
         self._storage_manager = corvic.system.StorageManager(
             self._blob_client,
             bucket_name=bucket.name,
-            unstructured_prefix="unstructured_data",
-            tabular_prefix="tabular_data",
-            vector_prefix="vectors",
+            unstructured_prefix=UNSTRUCTURED_PREFIX,
+            tabular_prefix=TABULAR_PREFIX,
+            vector_prefix=VECTOR_PREFIX,
         )
         corvic.orm.Base.metadata.create_all(self._sa_engine)
         duck_db_conn = duckdb.connect(":memory:")

corvic_engine-0.3.0rc49/python/corvic/system_sqlite/__init__.py

@@ -1,9 +0,0 @@
-"""Single-machine, sqlite-backed implementation of corvic.system."""
-
-from corvic.system_sqlite.client import Client, FSBlobClient, RDBMSBlobClient
-
-__all__ = [
-    "Client",
-    "FSBlobClient",
-    "RDBMSBlobClient",
-]