xet-runtime 1.5.2

Async runtime, configuration, logging, and utility infrastructure for the Hugging Face Xet client tools.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use std::cell::RefCell;
use std::sync::Arc;

use super::XetRuntime;
use crate::config::XetConfig;

// Use thread-local references to the config that caches access.  This way, xet_config() will
// can be called outside of an existing runtime.
thread_local! {
    static THREAD_CONFIG_REF: RefCell<Option<Arc<XetConfig>>> = const { RefCell::new(None) };
}

pub fn xet_config() -> Arc<XetConfig> {
    if let Some(config) = THREAD_CONFIG_REF.with_borrow(|config| config.clone()) {
        return config;
    }

    let config = {
        if let Some(runtime) = XetRuntime::current_if_exists() {
            runtime.config().clone()
        } else {
            Arc::new(XetConfig::new())
        }
    };

    THREAD_CONFIG_REF.set(Some(config.clone()));

    config
}