sim-lib-openai-server 0.1.2

OpenAI-compatible gateway skeleton for SIM.
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
30
31
32
33

use sim_kernel::{Error, Result};

/// Generates sequential, prefixed gateway ids of the form `prefix_NNNNNN`.
///
/// Counting is deterministic: ids advance monotonically from a fixed start,
/// which makes generated ids reproducible across runs.
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct GatewayIdGenerator {
    prefix: String,
    next: u64,
}

impl GatewayIdGenerator {
    /// Builds a generator that emits ids with the given `prefix`, starting the
    /// counter at `start`.
    pub fn deterministic(prefix: impl Into<String>, start: u64) -> Self {
        Self {
            prefix: prefix.into(),
            next: start,
        }
    }

    /// Returns the next id and advances the counter, erroring if the counter
    /// would overflow `u64`.
    pub fn next_id(&mut self) -> Result<String> {
        let value = self.next;
        self.next = self
            .next
            .checked_add(1)
            .ok_or_else(|| Error::Eval("gateway id counter overflow".to_owned()))?;
        Ok(format!("{}_{value:06}", self.prefix))
    }
}