uni-db 2.2.2__tar.gz → 2.2.4__tar.gz

{uni_db-2.2.2 → uni_db-2.2.4}/Cargo.lock

@@ -518,6 +518,12 @@ dependencies = [
  "stable_deref_trait",
 ]
 
+[[package]]
+name = "assoc"
+version = "0.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bfdc70193dadb9d7287fa4b633f15f90c876915b31f6af17da307fc59c9859a8"
+
 [[package]]
 name = "async-channel"
 version = "2.5.0"
@@ -2053,6 +2059,12 @@ dependencies = [
  "tiny-keccak",
 ]
 
+[[package]]
+name = "const-siphasher"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "03efed02df0504d71e44bfd51d3329f401b8303a2fd14254acf05c95a0af0153"
+
 [[package]]
 name = "const-str"
 version = "0.4.3"
@@ -2131,6 +2143,19 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "corosensei"
+version = "0.3.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6886a0c0f263965933c438626e7179139a62b978a33aa18281cbf0cd5a975f34"
+dependencies = [
+ "autocfg",
+ "cfg-if",
+ "libc",
+ "scopeguard",
+ "windows-sys 0.59.0",
+]
+
 [[package]]
 name = "cpp_demangle"
 version = "0.4.5"
@@ -8428,6 +8453,12 @@ version = "0.5.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1a80800c0488c3a21695ea981a54918fbb37abf04f4d0720c453632255e2ff0e"
 
+[[package]]
+name = "owo-colors"
+version = "3.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c1b04fb49957986fdce4d6ee7a65027d55d4b6d2265e5848bbb507b58ccfdb6f"
+
 [[package]]
 name = "packedvec"
 version = "1.2.5"
@@ -9566,6 +9597,15 @@ dependencies = [
  "rand_core 0.9.5",
 ]
 
+[[package]]
+name = "rand_pcg"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "59cad018caf63deb318e5a4586d99a24424a364f40f1e5778c29aca23f4fc73e"
+dependencies = [
+ "rand_core 0.6.4",
+]
+
 [[package]]
 name = "rand_xorshift"
 version = "0.4.0"
@@ -10929,6 +10969,27 @@ version = "2.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "f8fadd59c855ef2080decdef8ff161eb6661b86933c9d82e5ba29dc602a55aba"
 
+[[package]]
+name = "shuttle"
+version = "0.9.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ba93071c1b720be2505f4c8ce2863502cb9a26a3819e268df1932458a755152c"
+dependencies = [
+ "assoc",
+ "bitvec",
+ "cfg-if",
+ "const-siphasher",
+ "corosensei",
+ "hex",
+ "owo-colors",
+ "rand 0.8.6",
+ "rand_core 0.6.4",
+ "rand_pcg",
+ "scoped-tls",
+ "smallvec 1.15.2",
+ "tracing",
+]
+
 [[package]]
 name = "signal-hook"
 version = "0.3.18"
@@ -12585,7 +12646,7 @@ checksum = "eeba86d422ce181a719445e51872fa30f1f7413b62becb52e95ec91aa262d85c"
 
 [[package]]
 name = "uni-algo"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -12604,7 +12665,7 @@ dependencies = [
 
 [[package]]
 name = "uni-btic"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "chrono",
  "proptest",
@@ -12614,7 +12675,7 @@ dependencies = [
 
 [[package]]
 name = "uni-bulk"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -12634,7 +12695,7 @@ dependencies = [
 
 [[package]]
 name = "uni-cli"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "clap",
@@ -12654,7 +12715,7 @@ dependencies = [
 
 [[package]]
 name = "uni-common"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-schema",
@@ -12676,7 +12737,7 @@ dependencies = [
 
 [[package]]
 name = "uni-crdt"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "fxhash",
  "proptest",
@@ -12690,7 +12751,7 @@ dependencies = [
 
 [[package]]
 name = "uni-cypher"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "pest",
  "pest_derive",
@@ -12701,7 +12762,7 @@ dependencies = [
 
 [[package]]
 name = "uni-db"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow",
@@ -12771,7 +12832,7 @@ dependencies = [
 
 [[package]]
 name = "uni-fork"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -12787,7 +12848,7 @@ dependencies = [
 
 [[package]]
 name = "uni-locy"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "async-trait",
  "candle-core",
@@ -12801,9 +12862,20 @@ dependencies = [
  "uni-cypher",
 ]
 
+[[package]]
+name = "uni-locy-oracle"
+version = "2.2.4"
+dependencies = [
+ "anyhow",
+ "proptest",
+ "tokio",
+ "uni-common",
+ "uni-db",
+]
+
 [[package]]
 name = "uni-locy-tck"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "async-trait",
@@ -12830,7 +12902,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arc-swap",
  "arrow",
@@ -12868,7 +12940,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-apoc-core"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arrow",
  "arrow-array",
@@ -12888,7 +12960,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-builtin"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow",
@@ -12917,7 +12989,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-conformance"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "semver",
  "serde",
@@ -12929,7 +13001,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-custom"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arrow",
  "arrow-array",
@@ -12952,7 +13024,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-extism"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arrow",
  "arrow-array",
@@ -12977,7 +13049,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-host"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13012,7 +13084,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-pyo3"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arrow",
  "arrow-array",
@@ -13035,7 +13107,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-rhai"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arrow",
  "arrow-array",
@@ -13061,7 +13133,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-wasm"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow",
@@ -13087,7 +13159,7 @@ dependencies = [
 
 [[package]]
 name = "uni-plugin-wasm-rt"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "arrow",
  "arrow-array",
@@ -13100,7 +13172,7 @@ dependencies = [
 
 [[package]]
 name = "uni-python"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13125,7 +13197,7 @@ dependencies = [
 
 [[package]]
 name = "uni-python-cuda"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13148,7 +13220,7 @@ dependencies = [
 
 [[package]]
 name = "uni-python-metal"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13171,7 +13243,7 @@ dependencies = [
 
 [[package]]
 name = "uni-python-onnx"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13194,7 +13266,7 @@ dependencies = [
 
 [[package]]
 name = "uni-python-onnx-cuda"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13217,7 +13289,7 @@ dependencies = [
 
 [[package]]
 name = "uni-python-onnx-metal"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow-array",
@@ -13240,7 +13312,7 @@ dependencies = [
 
 [[package]]
 name = "uni-query"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow",
@@ -13289,7 +13361,7 @@ dependencies = [
 
 [[package]]
 name = "uni-query-functions"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arrow",
@@ -13319,7 +13391,7 @@ dependencies = [
 
 [[package]]
 name = "uni-sidecar"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "serde",
  "serde_json",
@@ -13329,7 +13401,7 @@ dependencies = [
 
 [[package]]
 name = "uni-store"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "arc-swap",
@@ -13348,6 +13420,7 @@ dependencies = [
  "lance-linalg",
  "lancedb",
  "log",
+ "loom",
  "lru",
  "metrics",
  "metrics-util",
@@ -13357,6 +13430,7 @@ dependencies = [
  "serde",
  "serde_json",
  "sha3",
+ "shuttle",
  "tempfile",
  "tokio",
  "tokio-util",
@@ -13372,7 +13446,7 @@ dependencies = [
 
 [[package]]
 name = "uni-tck"
-version = "2.2.2"
+version = "2.2.4"
 dependencies = [
  "anyhow",
  "async-trait",

{uni_db-2.2.2 → uni_db-2.2.4}/Cargo.toml

@@ -3,7 +3,7 @@ members = ["crates/uni-common", "crates/uni-btic", "crates/uni-store", "crates/u
 resolver = "2"
 
 [workspace.package]
-version = "2.2.2"
+version = "2.2.4"
 edition = "2024"
 authors = ["Dragonscale Industries Inc. <dev@dragonscale.ai>"]
 license = "Apache-2.0"

{uni_db-2.2.2 → uni_db-2.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uni-db
-Version: 2.2.2
+Version: 2.2.4
 Classifier: Development Status :: 4 - Beta
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Rust

{uni_db-2.2.2 → uni_db-2.2.4}/bindings/uni-db/src/async_api.rs

@@ -3566,6 +3566,54 @@ impl AsyncSessionLocyBuilder {
             Ok(convert::locy_explain_to_py_class(result))
         })
     }
+
+    /// Profile the Locy program: returns an awaitable `(LocyResult, profile)`.
+    ///
+    /// Async analog of the sync `SessionLocyBuilder.profile()` and the read-path
+    /// `AsyncSessionQueryBuilder.profile()`.
+    fn profile<'py>(&self, py: Python<'py>) -> PyResult<Bound<'py, PyAny>> {
+        let rust_params = convert::convert_params_ref(py, &self.params)?;
+        let inner = self.inner.clone();
+        let program = self.program.clone();
+        let timeout_secs = self.timeout_secs;
+        let max_iterations = self.max_iterations;
+        let locy_config = self.locy_config.clone();
+        let cancel_token = self.cancellation_token.as_ref().map(|ct| ct.inner.clone());
+        // Locy future is !Send — use spawn_blocking
+        pyo3_async_runtimes::tokio::future_into_py(py, async move {
+            let (result, profile) = tokio::task::spawn_blocking(move || {
+                tokio::runtime::Handle::current().block_on(async move {
+                    let guard = inner.lock().await;
+                    let mut builder = guard.locy_with(&program);
+                    for (k, v) in rust_params {
+                        builder = builder.param(k, v);
+                    }
+                    if let Some(t) = timeout_secs {
+                        builder = builder.timeout(std::time::Duration::from_secs_f64(t));
+                    }
+                    if let Some(n) = max_iterations {
+                        builder = builder.max_iterations(n);
+                    }
+                    if let Some(c) = locy_config {
+                        builder = builder.with_config(c);
+                    }
+                    if let Some(ct) = cancel_token {
+                        builder = builder.cancellation_token(ct);
+                    }
+                    builder.profile().await
+                })
+            })
+            .await
+            .map_err(|e| PyErr::new::<pyo3::exceptions::PyRuntimeError, _>(e.to_string()))?
+            .map_err(crate::exceptions::uni_error_to_pyerr)?;
+            Python::attach(|py| {
+                let locy_result = convert::locy_result_to_py_class(py, result)?;
+                let profile_output = convert::locy_profile_to_py_class(profile);
+                let tuple = (locy_result, profile_output);
+                tuple.into_py_any(py)
+            })
+        })
+    }
 }
 
 // ============================================================================
@@ -3886,6 +3934,56 @@ impl AsyncTxLocyBuilder {
             Python::attach(|py| convert::locy_result_to_py_class(py, result))
         })
     }
+
+    /// Profile the Locy program: returns an awaitable `(LocyResult, profile)`.
+    ///
+    /// Async analog of the sync `TxLocyBuilder.profile()`.
+    fn profile<'py>(&self, py: Python<'py>) -> PyResult<Bound<'py, PyAny>> {
+        let rust_params = convert::convert_params_ref(py, &self.params)?;
+        let inner = self.inner.clone();
+        let program = self.program.clone();
+        let timeout_secs = self.timeout_secs;
+        let max_iterations = self.max_iterations;
+        let locy_config = self.locy_config.clone();
+        let cancel_token = self.cancellation_token.as_ref().map(|ct| ct.inner.clone());
+        // Locy future is !Send — use spawn_blocking
+        pyo3_async_runtimes::tokio::future_into_py(py, async move {
+            let (result, profile) = tokio::task::spawn_blocking(move || {
+                tokio::runtime::Handle::current().block_on(async move {
+                    let guard = inner.lock().await;
+                    let tx = active_tx(&guard)?;
+                    let mut builder = tx.locy_with(&program);
+                    for (k, v) in rust_params {
+                        builder = builder.param(k, v);
+                    }
+                    if let Some(t) = timeout_secs {
+                        builder = builder.timeout(std::time::Duration::from_secs_f64(t));
+                    }
+                    if let Some(n) = max_iterations {
+                        builder = builder.max_iterations(n);
+                    }
+                    if let Some(c) = locy_config {
+                        builder = builder.with_config(c);
+                    }
+                    if let Some(ct) = cancel_token {
+                        builder = builder.cancellation_token(ct);
+                    }
+                    builder
+                        .profile()
+                        .await
+                        .map_err(crate::exceptions::uni_error_to_pyerr)
+                })
+            })
+            .await
+            .map_err(|e| PyErr::new::<pyo3::exceptions::PyRuntimeError, _>(e.to_string()))??;
+            Python::attach(|py| {
+                let locy_result = convert::locy_result_to_py_class(py, result)?;
+                let profile_output = convert::locy_profile_to_py_class(profile);
+                let tuple = (locy_result, profile_output);
+                tuple.into_py_any(py)
+            })
+        })
+    }
 }
 
 /// Async builder for applying a DerivedFactSet with staleness controls.

{uni_db-2.2.2 → uni_db-2.2.4}/bindings/uni-db/src/builders.rs

@@ -1577,6 +1577,44 @@ impl SessionLocyBuilder {
             .map_err(crate::exceptions::uni_error_to_pyerr)?;
         Ok(convert::locy_explain_to_py_class(result))
     }
+
+    /// Profile the Locy program: run it and return `(result, profile)`.
+    ///
+    /// The Locy analog of `query.profile()` — `profile` is a
+    /// [`LocyProfileOutput`](crate::types::PyLocyProfile) carrying the
+    /// stratum → rule → fixpoint-iteration timing and per-operator metrics.
+    fn profile(
+        &self,
+        py: Python,
+    ) -> PyResult<(crate::types::PyLocyResult, crate::types::PyLocyProfile)> {
+        let session = self.session.borrow(py);
+        let mut builder = session.inner.locy_with(&self.program);
+        for (k, v) in &self.params {
+            let val = convert::py_object_to_value(py, v)?;
+            builder = builder.param(k, val);
+        }
+        if let Some(t) = self.timeout_secs {
+            builder = builder.timeout(std::time::Duration::from_secs_f64(t));
+        }
+        if let Some(n) = self.max_iterations {
+            builder = builder.max_iterations(n);
+        }
+        if let Some(ref config) = self.locy_config {
+            builder = builder.with_config(config.clone());
+        }
+        if let Some(ref ct) = self.cancellation_token {
+            builder = builder.cancellation_token(ct.inner.clone());
+        }
+        // Release the GIL across `block_on` (the executor may call back into
+        // Python; see `run`).
+        let (result, profile) = py
+            .detach(|| pyo3_async_runtimes::tokio::get_runtime().block_on(builder.profile()))
+            .map_err(crate::exceptions::uni_error_to_pyerr)?;
+        Ok((
+            convert::locy_result_to_py_class(py, result)?,
+            convert::locy_profile_to_py_class(profile),
+        ))
+    }
 }
 
 /// Builder for transaction configuration.
@@ -1884,6 +1922,43 @@ impl PyTxLocyBuilder {
             .map_err(crate::exceptions::uni_error_to_pyerr)?;
         convert::locy_result_to_py_class(py, result)
     }
+
+    /// Profile the Locy program in this transaction: run it and return
+    /// `(result, profile)`. Transaction-level analog of
+    /// [`SessionLocyBuilder::profile`].
+    fn profile(
+        &self,
+        py: Python,
+    ) -> PyResult<(crate::types::PyLocyResult, crate::types::PyLocyProfile)> {
+        let tx_ref = self.tx.borrow(py);
+        let tx = tx_ref.inner.as_ref().ok_or_else(|| {
+            PyErr::new::<pyo3::exceptions::PyRuntimeError, _>("Transaction already completed")
+        })?;
+        let mut builder = tx.locy_with(&self.program);
+        for (k, v) in &self.params {
+            let val = convert::py_object_to_value(py, v)?;
+            builder = builder.param(k, val);
+        }
+        if let Some(t) = self.timeout_secs {
+            builder = builder.timeout(std::time::Duration::from_secs_f64(t));
+        }
+        if let Some(n) = self.max_iterations {
+            builder = builder.max_iterations(n);
+        }
+        if let Some(ref config) = self.locy_config {
+            builder = builder.with_config(config.clone());
+        }
+        if let Some(ref ct) = self.cancellation_token {
+            builder = builder.cancellation_token(ct.inner.clone());
+        }
+        let (result, profile) = py
+            .detach(|| pyo3_async_runtimes::tokio::get_runtime().block_on(builder.profile()))
+            .map_err(crate::exceptions::uni_error_to_pyerr)?;
+        Ok((
+            convert::locy_result_to_py_class(py, result)?,
+            convert::locy_profile_to_py_class(profile),
+        ))
+    }
 }
 
 /// Builder for applying a DerivedFactSet with options.

{uni_db-2.2.2 → uni_db-2.2.4}/bindings/uni-db/src/convert.rs

@@ -1275,6 +1275,15 @@ pub fn locy_explain_to_py_class(
     }
 }
 
+/// Wrap a Rust [`LocyProfileOutput`] in its Python class.
+///
+/// [`LocyProfileOutput`]: uni_db::api::locy_result::LocyProfileOutput
+pub fn locy_profile_to_py_class(
+    output: uni_db::api::locy_result::LocyProfileOutput,
+) -> crate::types::PyLocyProfile {
+    crate::types::PyLocyProfile { inner: output }
+}
+
 /// Extract a CloudStorageConfig from a Python dict.
 ///
 /// The dict must have a `"provider"` key: `"s3"`, `"gcs"`, or `"azure"`.

{uni_db-2.2.2 → uni_db-2.2.4}/bindings/uni-db/src/lib.rs

@@ -135,6 +135,7 @@ fn _uni_db(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<types::PyExplainOutput>()?;
     m.add_class::<types::PyProfileOutput>()?;
     m.add_class::<types::PyLocyExplainOutput>()?;
+    m.add_class::<types::PyLocyProfile>()?;
 
     // New result types
     m.add_class::<types::PyExecuteResult>()?;