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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

//! `LangfusePromptManager` — versioned prompt retrieval + caching.
//!
//! Mirrors Python [`operonx/telemetry/backends/langfuse/prompt_manager.py`](../../../../../../operonx/telemetry/backends/langfuse/prompt_manager.py).
//!
//! # Phase 7 scope
//! Stub. Per §6b.8 the Python side lazy-imports the `langfuse` SDK here —
//! the Rust port hits `{host}/api/public/v2/prompts/{name}` directly when
//! implemented in Phase 7b.

use std::sync::Arc;

use dashmap::DashMap;
use serde_json::Value;

use super::config::LangfuseConfig;
use crate::core::exceptions::OperonError;

/// Cached, versioned prompts keyed by `"{name}:{version}"`.
pub struct LangfusePromptManager {
    pub config: LangfuseConfig,
    cache: DashMap<String, Arc<Value>>,
}

impl std::fmt::Debug for LangfusePromptManager {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("LangfusePromptManager")
            .field("host", &self.config.host)
            .field("cached", &self.cache.len())
            .finish()
    }
}

impl LangfusePromptManager {
    pub fn new(config: LangfuseConfig) -> Self {
        Self {
            config,
            cache: DashMap::new(),
        }
    }

    /// Retrieve a named prompt (optionally at a specific version). Returns the
    /// cached value when present; otherwise fetches from the backend.
    pub async fn get(&self, name: &str, version: Option<&str>) -> Result<Arc<Value>, OperonError> {
        let key = match version {
            Some(v) => format!("{}:{}", name, v),
            None => format!("{}:latest", name),
        };
        if let Some(cached) = self.cache.get(&key) {
            return Ok(cached.clone());
        }
        Err(OperonError::Provider(format!(
            "LangfusePromptManager::get not yet implemented (Phase 7b) — asked for '{}'",
            key
        )))
    }

    /// Number of cached prompts.
    pub fn cache_len(&self) -> usize {
        self.cache.len()
    }
}