elicitation 0.8.0

Conversational elicitation of strongly-typed Rust values via MCP
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

//! BTreeSet<T> implementation for ordered unique item collection.

use crate::{ElicitCommunicator, ElicitResult, Elicitation, Prompt};
use std::collections::BTreeSet;

// Default-only style for BTreeSet
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default)]
pub enum BTreeSetStyle {
    #[default]
    Default,
}

impl Prompt for BTreeSetStyle {
    fn prompt() -> Option<&'static str> {
        None
    }
}

impl Elicitation for BTreeSetStyle {
    type Style = BTreeSetStyle;

    #[tracing::instrument(skip(_communicator), level = "trace")]
    async fn elicit<C: ElicitCommunicator>(_communicator: &C) -> ElicitResult<Self> {
        Ok(Self::Default)
    }
}

impl<T> Prompt for BTreeSet<T>
where
    T: Elicitation + Ord + Send,
{
    fn prompt() -> Option<&'static str> {
        Some("Would you like to add items to this ordered set?")
    }
}

impl<T> Elicitation for BTreeSet<T>
where
    T: Elicitation + Ord + Send,
{
    type Style = BTreeSetStyle;

    #[tracing::instrument(skip(communicator), fields(item_type = std::any::type_name::<T>()))]
    async fn elicit<C: ElicitCommunicator>(communicator: &C) -> ElicitResult<Self> {
        let mut set = BTreeSet::new();
        tracing::debug!("Eliciting BTreeSet");

        loop {
            let add_more = if set.is_empty() {
                tracing::debug!("Prompting for first item");
                bool::elicit(communicator).await?
            } else {
                tracing::debug!(count = set.len(), "Prompting for additional item");
                bool::elicit(communicator).await?
            };

            if !add_more {
                tracing::debug!(final_count = set.len(), "Set complete");
                break;
            }

            tracing::debug!("Eliciting item");
            let item = T::elicit(communicator).await?;

            // Automatic duplicate handling - sets ignore duplicates
            if !set.insert(item) {
                tracing::debug!("Duplicate item ignored (already in set)");
            }
        }

        Ok(set)
    }
}