add build_info to _pydantic_core (#749)

Author: samuelcolvin

python/pydantic_core/_pydantic_core.pyi

@@ -22,6 +22,7 @@ from _typeshed import SupportsAllComparisons
 __all__ = [
     '__version__',
     'build_profile',
+    'build_info',
     '_recursion_limit',
     'ArgsKwargs',
     'SchemaValidator',
@@ -45,6 +46,7 @@ __all__ = [
 ]
 __version__: str
 build_profile: str
+build_info: str
 _recursion_limit: int
 
 _T = TypeVar('_T', default=Any, covariant=True)

src/lib.rs

@@ -43,10 +43,20 @@ pub fn get_version() -> String {
     version.replace("-alpha", "a").replace("-beta", "b")
 }
 
+pub fn build_info() -> String {
+    format!(
+        "profile={} pgo={} mimalloc={}",
+        env!("PROFILE"),
+        option_env!("RUSTFLAGS").unwrap_or("").contains("-Cprofile-use="),
+        cfg!(feature = "mimalloc")
+    )
+}
+
 #[pymodule]
 fn _pydantic_core(py: Python, m: &PyModule) -> PyResult<()> {
     m.add("__version__", get_version())?;
     m.add("build_profile", env!("PROFILE"))?;
+    m.add("build_info", build_info())?;
     m.add("_recursion_limit", recursion_guard::RECURSION_GUARD_LIMIT)?;
     m.add("PydanticUndefined", PydanticUndefinedType::new(py))?;
     m.add_class::<PydanticUndefinedType>()?;

tests/test_misc.py

@@ -6,7 +6,14 @@
 from typing_extensions import get_args
 
 from pydantic_core import CoreSchema, CoreSchemaType, PydanticUndefined, core_schema
-from pydantic_core._pydantic_core import SchemaError, SchemaValidator, ValidationError, __version__, build_profile
+from pydantic_core._pydantic_core import (
+    SchemaError,
+    SchemaValidator,
+    ValidationError,
+    __version__,
+    build_info,
+    build_profile,
+)
 
 
 @pytest.mark.parametrize('obj', [ValidationError, SchemaValidator, SchemaError])
@@ -23,6 +30,10 @@ def test_build_profile():
     assert build_profile in ('debug', 'release')
 
 
+def test_build_info():
+    assert isinstance(build_info, str)
+
+
 def test_schema_error():
     err = SchemaError('test')
     assert isinstance(err, Exception)