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

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

use std::sync::Arc;

use serde_json::Value;

use super::EmbeddingResource;
use crate::core::exceptions::OperonError;
use crate::core::registry::{registry, ConfigDict, Factory, ResourceInstance};
use crate::providers::embeddings::{create_embedder, EmbeddingConfig};

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