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
26
27

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

use std::sync::Arc;

use serde_json::Value;

use super::RerankerResource;
use crate::core::exceptions::OperonError;
use crate::core::registry::{registry, ConfigDict, Factory, ResourceInstance};
use crate::providers::rerankers::{create_reranker, RerankingConfig};

/// Register the reranking category factory. Idempotent.
///
/// Note the category key — Python uses `"reranking"`, not `"rerank"`.
pub fn register() -> Result<(), OperonError> {
    if registry().get_factory("reranking").is_some() {
        return Ok(());
    }
    let factory: Factory = Arc::new(|cfg: ConfigDict| {
        let typed: RerankingConfig = serde_json::from_value(Value::Object(cfg))?;
        let reranker = create_reranker(typed)?;
        Ok(Arc::new(RerankerResource(reranker)) as ResourceInstance)
    });
    registry().register("reranking", factory, Some("RerankingConfig"))
}