oai-statsig-python-core 0.28.0rc2__tar.gz → 0.29.0__tar.gz

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/Cargo.lock

@@ -1738,7 +1738,7 @@ dependencies = [
 
 [[package]]
 name = "oai-statsig-grpc"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 dependencies = [
  "async-trait",
  "chrono",
@@ -1753,7 +1753,7 @@ dependencies = [
 
 [[package]]
 name = "oai-statsig-rust"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 dependencies = [
  "ahash",
  "arc-swap",
@@ -3043,7 +3043,7 @@ checksum = "a2eb9349b6444b326872e140eb1cf5e7c522154d69e7a0ffb0fb81c06b37543f"
 
 [[package]]
 name = "statsig-node"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 dependencies = [
  "async-trait",
  "napi",
@@ -3058,7 +3058,7 @@ dependencies = [
 
 [[package]]
 name = "statsig-pyo3"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 dependencies = [
  "async-trait",
  "lazy_static",
@@ -3073,7 +3073,7 @@ dependencies = [
 
 [[package]]
 name = "statsig_elixir"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 dependencies = [
  "async-trait",
  "oai-statsig-rust",
@@ -3086,7 +3086,7 @@ dependencies = [
 
 [[package]]
 name = "statsig_ffi"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 dependencies = [
  "async-trait",
  "cbindgen",

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/Cargo.toml

@@ -26,7 +26,7 @@ exclude = [
 [workspace.package]
 edition = "2021"
 license = "ISC"
-version = "0.28.0-rc.2"
+version = "0.29.0"
 homepage = "https://statsig.com/"
 authors = ["Statsig", "Daniel Loomb <daniel@statsig.com>"]
 readme = "README.md"

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oai_statsig_python_core
-Version: 0.28.0rc2
+Version: 0.29.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/py_src/statsig_python_core/statsig_python_core.pyi

@@ -22,6 +22,7 @@ __all__ = [
     "LayerEvaluationOptions",
     "LayerParamExposureData",
     "LogEventRequest",
+    "MmapPreloadReport",
     "MmapReaderMemorySnapshot",
     "ObservabilityClientBase",
     "OutputLoggerProviderBase",
@@ -122,6 +123,8 @@ class InternedStore:
     def preload_mmap(cls, sdk_key: builtins.str) -> None: ...
     @classmethod
     def mmap_reader_memory_snapshot(cls) -> typing.Optional[MmapReaderMemorySnapshot]: ...
+    @classmethod
+    def preload_mmap_multi(cls, required_sdk_keys: typing.Sequence[builtins.str], optional_sdk_keys: typing.Optional[typing.Sequence[builtins.str]] = None) -> MmapPreloadReport: ...
 
 @typing.final
 class LayerEvaluationOptions:
@@ -147,6 +150,13 @@ class LogEventRequest:
     def retries(self) -> builtins.int: ...
     def __new__(cls, payload: typing.Any, event_count: builtins.int, retries: builtins.int) -> LogEventRequest: ...
 
+@typing.final
+class MmapPreloadReport:
+    @property
+    def loaded(self) -> builtins.int: ...
+    @property
+    def skipped_optional_indexes(self) -> builtins.list[builtins.int]: ...
+
 @typing.final
 class MmapReaderMemorySnapshot:
     @property
@@ -245,7 +255,7 @@ class StatsigBasePy:
     def unsubscribe(self, event_name: builtins.str) -> None: ...
     def unsubscribe_by_id(self, subscription_id: builtins.str) -> None: ...
     def unsubscribe_all(self) -> None: ...
-    def log_event(self, user: StatsigUser, event_name: builtins.str, value: typing.Optional[typing.Any] = None, metadata: typing.Optional[dict] = None) -> None: ...
+    def log_event(self, user: StatsigUser, event_name: builtins.str, value: typing.Optional[typing.Any] = None, metadata: typing.Optional[dict] = None, timestamp_ms: typing.Optional[builtins.int] = None) -> None: ...
     def check_gate(self, user: StatsigUser, name: builtins.str, options: typing.Optional[FeatureGateEvaluationOptions] = None) -> builtins.bool: ...
     def _INTERNAL_get_feature_gate(self, user: StatsigUser, name: builtins.str, options: typing.Optional[FeatureGateEvaluationOptions] = None) -> dict: ...
     def manually_log_gate_exposure(self, user: StatsigUser, name: builtins.str) -> None: ...

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/data_store_base_py.rs

@@ -1,14 +1,15 @@
 use async_trait::async_trait;
 use pyo3::exceptions::PyTypeError;
 use pyo3::types::{PyAny, PyAnyMethods, PyBytes, PyDict, PyModule};
-use pyo3::{prelude::Bound, pyclass, pymethods, FromPyObject, Py, PyResult};
+use pyo3::{FromPyObject, Py, PyResult, prelude::Bound, pyclass, pymethods};
 use pyo3_stub_gen::derive::*;
 use statsig_rust::{
+    StatsigErr,
     data_store_interface::{
         DataStoreBytesResponse, DataStoreGetBytesRequest, DataStoreResponse, DataStoreTrait,
         RequestPath,
     },
-    log_e, log_w, StatsigErr,
+    log_e, log_w,
 };
 
 use crate::safe_gil::SafeGil;
@@ -88,7 +89,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "Python interpreter has been shutdown".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -97,7 +98,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "No 'get' function provided".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -177,7 +178,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "Python interpreter has been shutdown".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -186,7 +187,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "No 'set' function provided".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -223,7 +224,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "Python interpreter has been shutdown".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -232,7 +233,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "No 'get_bytes' function provided".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -372,7 +373,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "Python interpreter has been shutdown".to_string(),
-                    ))
+                    ));
                 }
             };
 
@@ -381,7 +382,7 @@ impl DataStoreTrait for DataStoreBasePy {
                 None => {
                     return Err(StatsigErr::DataStoreFailure(
                         "No 'set_bytes' function provided".to_string(),
-                    ))
+                    ));
                 }
             };
 

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/event_logging_adapter_base_py.rs

@@ -1,8 +1,8 @@
 use async_trait::async_trait;
-use pyo3::{prelude::*, pyclass, pymethods, types::PyDict, FromPyObject};
+use pyo3::{FromPyObject, prelude::*, pyclass, pymethods, types::PyDict};
 use pyo3_stub_gen::derive::*;
 use statsig_rust::{
-    log_e, log_event_payload::LogEventRequest, EventLoggingAdapter, StatsigErr, StatsigRuntime,
+    EventLoggingAdapter, StatsigErr, StatsigRuntime, log_e, log_event_payload::LogEventRequest,
 };
 use std::sync::Arc;
 

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/interned_store_py.rs

@@ -8,9 +8,11 @@ use pyo3::types::PyModule;
 use pyo3::types::PyType;
 use pyo3_stub_gen::derive::*;
 
-use statsig_rust::interned_values::{InternedStore, MmapReaderMemorySnapshot};
-use statsig_rust::log_e;
 use statsig_rust::StatsigRuntime;
+use statsig_rust::interned_values::{
+    InternedStore, MmapPreloadReport as RustMmapPreloadReport, MmapReaderMemorySnapshot,
+};
+use statsig_rust::log_e;
 
 const TAG: &str = stringify!(InternStorePy);
 
@@ -50,6 +52,29 @@ impl From<MmapReaderMemorySnapshot> for MmapReaderMemorySnapshotPy {
     }
 }
 
+#[gen_stub_pyclass]
+#[pyclass(name = "MmapPreloadReport", module = "statsig_python_core")]
+pub struct MmapPreloadReportPy {
+    #[pyo3(get)]
+    pub loaded: usize,
+
+    #[pyo3(get)]
+    pub skipped_optional_indexes: Vec<usize>,
+}
+
+impl From<RustMmapPreloadReport> for MmapPreloadReportPy {
+    fn from(report: RustMmapPreloadReport) -> Self {
+        Self {
+            loaded: report.loaded,
+            skipped_optional_indexes: report
+                .skipped_optional
+                .into_iter()
+                .map(|failure| failure.index)
+                .collect(),
+        }
+    }
+}
+
 #[gen_stub_pyclass]
 #[pyclass(name = "InternedStore", module = "statsig_python_core")]
 #[derive(Default)]
@@ -135,7 +160,6 @@ impl InternedStorePy {
 
         Ok(())
     }
-
     #[classmethod]
     pub fn mmap_reader_memory_snapshot(
         _cls: &Bound<'_, PyType>,
@@ -150,6 +174,32 @@ impl InternedStorePy {
             PyRuntimeError::new_err(error)
         })
     }
+
+    #[classmethod]
+    #[pyo3(signature = (required_sdk_keys, optional_sdk_keys=None))]
+    pub fn preload_mmap_multi(
+        _cls: &Bound<'_, PyType>,
+        required_sdk_keys: Vec<String>,
+        optional_sdk_keys: Option<Vec<String>>,
+    ) -> PyResult<MmapPreloadReportPy> {
+        let optional_sdk_keys = optional_sdk_keys.unwrap_or_default();
+        let required = required_sdk_keys
+            .iter()
+            .map(String::as_str)
+            .collect::<Vec<_>>();
+        let optional = optional_sdk_keys
+            .iter()
+            .map(String::as_str)
+            .collect::<Vec<_>>();
+
+        match InternedStore::preload_mmap_multi(&required, &optional) {
+            Ok(report) => Ok(report.into()),
+            Err(error) => {
+                log_e!(TAG, "Failed to load mmap data: {}", error);
+                Err(PyRuntimeError::new_err(error.to_string()))
+            }
+        }
+    }
 }
 
 fn get_completion_event(py: Python) -> PyResult<(Py<PyAny>, Py<PyAny>)> {

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/lib.rs

@@ -44,6 +44,7 @@ fn statsig_python_core(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<output_logger_provider_base_py::OutputLoggerProviderBasePy>()?;
     m.add_class::<statsig_options_py::SpecAdapterConfigPy>()?;
     m.add_class::<interned_store_py::MmapReaderMemorySnapshotPy>()?;
+    m.add_class::<interned_store_py::MmapPreloadReportPy>()?;
     m.add_class::<interned_store_py::InternedStorePy>()?;
     m.add_function(wrap_pyfunction!(notify_python_shutdown, m)?)?;
     m.add_function(wrap_pyfunction!(notify_python_fork, m)?)?;

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/observability_client_base_py.rs

@@ -1,6 +1,6 @@
-use pyo3::{prelude::*, pyclass, pymethods, FromPyObject};
+use pyo3::{FromPyObject, prelude::*, pyclass, pymethods};
 use pyo3_stub_gen::derive::*;
-use statsig_rust::{log_e, ObservabilityClient, OpsStatsEventObserver};
+use statsig_rust::{ObservabilityClient, OpsStatsEventObserver, log_e};
 use std::collections::HashMap;
 use std::sync::Arc;
 
@@ -138,7 +138,11 @@ impl ObservabilityClient for ObservabilityClientBasePy {
             let value = match func.as_ref().call1(py, (tag,)) {
                 Ok(value) => value,
                 Err(e) => {
-                    log_e!(TAG, "Failed to call ObservabilityClient.should_enable_high_cardinality_for_this_tag: {:?}", e);
+                    log_e!(
+                        TAG,
+                        "Failed to call ObservabilityClient.should_enable_high_cardinality_for_this_tag: {:?}",
+                        e
+                    );
                     return None;
                 }
             };
@@ -146,7 +150,11 @@ impl ObservabilityClient for ObservabilityClientBasePy {
             match value.extract(py) {
                 Ok(value) => Some(value),
                 Err(e) => {
-                    log_e!(TAG, "Failed to extract ObservabilityClient.should_enable_high_cardinality_for_this_tag: {:?}", e);
+                    log_e!(
+                        TAG,
+                        "Failed to extract ObservabilityClient.should_enable_high_cardinality_for_this_tag: {:?}",
+                        e
+                    );
                     None
                 }
             }

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/output_logger_provider_base_py.rs

@@ -1,4 +1,4 @@
-use pyo3::{prelude::*, pyclass, pymethods, FromPyObject};
+use pyo3::{FromPyObject, prelude::*, pyclass, pymethods};
 use pyo3_stub_gen::derive::*;
 use statsig_rust::output_logger::OutputLogProvider;
 

{oai_statsig_python_core-0.28.0rc2 → oai_statsig_python_core-0.29.0}/statsig-pyo3/src/pyo_utils.rs

@@ -4,10 +4,10 @@ use pyo3::types::{
     PyString, PyStringMethods, PyTypeMethods,
 };
 use pyo3::{Bound, IntoPyObject, Py, PyAny, PyErr, PyResult, PyTypeCheck, Python};
-use serde_json::{json, Number, Value};
+use serde_json::{Number, Value, json};
 use statsig_rust::evaluation::dynamic_string::DynamicString;
 use statsig_rust::user::fast_statsig_user::{FastUserCustomMap, FastUserUnitIDMap};
-use statsig_rust::{log_e, log_w, DynamicValue, StatsigUserDataMap, StatsigUserValue};
+use statsig_rust::{DynamicValue, StatsigUserDataMap, StatsigUserValue, log_e, log_w};
 use std::borrow::Cow;
 use std::collections::HashMap;
 
@@ -228,10 +228,12 @@ pub fn py_any_to_dynamic_value(value: &Bound<PyAny>) -> PyResult<DynamicValue> {
             })?;
             hashmap.insert(key_str, py_any_to_dynamic_value(&val)?);
         }
-        let json_value = json!(hashmap
-            .iter()
-            .map(|(k, v)| (k, &v.json_value))
-            .collect::<HashMap<_, _>>());
+        let json_value = json!(
+            hashmap
+                .iter()
+                .map(|(k, v)| (k, &v.json_value))
+                .collect::<HashMap<_, _>>()
+        );
         return Ok(DynamicValue {
             object_value: Some(hashmap.clone()),
             json_value,