tea-context 0.1.0

Deterministic prompt modules, context providers, and compiler for tea-rs
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
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use std::str::FromStr;

use crate::{
    BudgetBehavior, CacheScope, ContextError, ContextProvider, ContextProviderFuture,
    ContextProviderId, ContextRequest, PromptAuthority, PromptModule, PromptModuleId,
    PromptPriority, PromptProvenance, PromptSegment, PromptSegmentId, TrustLevel,
};

/// Generates guidance from prompt hints on the active tool snapshot.
#[derive(Debug, Clone)]
pub struct ToolHintProvider {
    id: ContextProviderId,
}

impl ToolHintProvider {
    /// Creates the built-in active-tool provider.
    ///
    /// # Errors
    ///
    /// Returns an error only when the crate's static provider ID is invalid.
    pub fn new() -> Result<Self, crate::ContextIdentityError> {
        Ok(Self {
            id: ContextProviderId::from_str("builtin.tool_hints")?,
        })
    }
}

impl ContextProvider for ToolHintProvider {
    fn id(&self) -> &ContextProviderId {
        &self.id
    }

    fn provide(&self, request: ContextRequest) -> ContextProviderFuture<'_> {
        let provider_id = self.id.clone();
        Box::pin(async move {
            let snippets = request
                .active_tools()
                .iter()
                .filter_map(|tool| tool.prompt_snippet().map(|snippet| (tool, snippet)))
                .map(|(tool, snippet)| {
                    let segment_id = format!("tool.{}.snippet", tool.name().as_str());
                    let locator = format!("{}@{}", tool.name(), tool.version());
                    PromptSegment::new(
                        PromptSegmentId::from_str(&segment_id).map_err(value_error)?,
                        format!("Tool `{}`: {snippet}", tool.name()),
                        PromptProvenance::new(provider_id.clone(), "tool_spec", Some(locator))
                            .map_err(value_error)?,
                        TrustLevel::Delegated,
                        CacheScope::Profile,
                        BudgetBehavior::Omit,
                    )
                    .map_err(value_error)
                })
                .collect::<Result<Vec<_>, ContextError>>()?;
            let guidelines = request
                .active_tools()
                .iter()
                .filter(|tool| !tool.prompt_guidelines().is_empty())
                .map(|tool| {
                    let segment_id = format!("tool.{}.guidelines", tool.name().as_str());
                    let locator = format!("{}@{}", tool.name(), tool.version());
                    let guidelines = tool
                        .prompt_guidelines()
                        .iter()
                        .map(|guideline| format!("- {guideline}"))
                        .collect::<Vec<_>>()
                        .join("\n");
                    PromptSegment::new(
                        PromptSegmentId::from_str(&segment_id).map_err(value_error)?,
                        format!("Tool `{}` guidelines:\n{guidelines}", tool.name()),
                        PromptProvenance::new(provider_id.clone(), "tool_spec", Some(locator))
                            .map_err(value_error)?,
                        TrustLevel::Delegated,
                        CacheScope::Profile,
                        BudgetBehavior::Omit,
                    )
                    .map_err(value_error)
                })
                .collect::<Result<Vec<_>, ContextError>>()?;
            let mut modules = Vec::new();
            if !snippets.is_empty() {
                modules.push(
                    PromptModule::new(
                        PromptModuleId::from_str("tool.active_snippets").map_err(value_error)?,
                        PromptAuthority::Tool,
                        PromptPriority::new(1),
                        snippets,
                    )
                    .map_err(value_error)?,
                );
            }
            if !guidelines.is_empty() {
                modules.push(
                    PromptModule::new(
                        PromptModuleId::from_str("tool.active_guidelines").map_err(value_error)?,
                        PromptAuthority::Tool,
                        PromptPriority::new(0),
                        guidelines,
                    )
                    .map_err(value_error)?,
                );
            }
            Ok(modules)
        })
    }
}

fn value_error(error: impl std::fmt::Display) -> ContextError {
    ContextError::new(crate::ContextErrorCode::InvalidValue, error.to_string())
}