operonx 0.6.2

High-performance Rust execution backend for Operon workflows
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

//! ONNX plugin registration.
//!
//! Mirrors Python [`operonx/providers/registry/onnx_plugin.py`](../../../../../operonx/providers/registry/onnx_plugin.py).

use std::sync::Arc;

use serde_json::Value;

use super::OnnxResource;
use crate::core::exceptions::OperonError;
use crate::core::registry::{registry, ConfigDict, Factory, ResourceInstance};
use crate::providers::onnx::{create_onnx_backend, OnnxInferenceConfig};

/// Register the onnx category factory. Idempotent.
pub fn register() -> Result<(), OperonError> {
    if registry().get_factory("onnx").is_some() {
        return Ok(());
    }
    let factory: Factory = Arc::new(|cfg: ConfigDict| {
        let typed: OnnxInferenceConfig = serde_json::from_value(Value::Object(cfg))?;
        let backend = create_onnx_backend(typed)?;
        Ok(Arc::new(OnnxResource(backend)) as ResourceInstance)
    });
    registry().register("onnx", factory, Some("OnnxInferenceConfig"))
}