sz-orm-model-ops 6.2.0

AI model operations: local inference, routing, finetuning, evaluation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! 模型路由策略（TASK-011）

use crate::types::{ModelOpsError, ModelRouterConfig};

pub struct ModelRouter {
    pub config: ModelRouterConfig,
}

impl ModelRouter {
    pub fn new(config: ModelRouterConfig) -> Self {
        Self { config }
    }

    pub fn route(&self, complexity: f64) -> Result<&str, ModelOpsError> {
        if complexity < 0.3 {
            Ok(&self.config.small_model)
        } else if complexity < 0.7 {
            Ok(&self.config.medium_model)
        } else {
            Ok(&self.config.large_model)
        }
    }
}