aether-mcp-utils 0.7.14

MCP client and server utilities for the Aether AI agent framework
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

use super::{Elicitation, ElicitationSink};
use aether_auth::OAuthCredentialStorage;
use rmcp::model::{ClientCapabilities, ClientConfig, Implementation};
use std::sync::Arc;
use tokio::sync::mpsc;

#[derive(Clone)]
pub struct ClientOptions {
    pub(super) implementation: Implementation,
    pub(super) elicitation: Option<ElicitationSink>,
    pub(super) oauth_store: Option<Arc<dyn OAuthCredentialStorage>>,
    tasks: bool,
}

impl Default for ClientOptions {
    fn default() -> Self {
        Self {
            implementation: Implementation::new(env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION")),
            elicitation: None,
            oauth_store: None,
            tasks: true,
        }
    }
}

impl ClientOptions {
    pub fn implementation(mut self, implementation: Implementation) -> Self {
        self.implementation = implementation;
        self
    }

    pub fn elicitation(mut self, sink: ElicitationSink) -> Self {
        self.elicitation = Some(sink);
        self
    }

    pub fn oauth_store(mut self, store: Arc<dyn OAuthCredentialStorage>) -> Self {
        self.oauth_store = Some(store);
        self
    }

    #[cfg(any(test, feature = "testing"))]
    pub fn without_tasks(mut self) -> Self {
        self.tasks = false;
        self
    }

    pub(super) fn client_config(&self) -> ClientConfig {
        let mut capabilities = if self.tasks {
            ClientCapabilities::builder().enable_tasks().build()
        } else {
            ClientCapabilities::default()
        };
        capabilities.elicitation = self
            .elicitation
            .as_ref()
            .map(|sink| sink.capability.clone())
            .filter(|capability| capability.form.is_some() || capability.url.is_some());
        ClientConfig::new(capabilities, self.implementation.clone())
    }

    pub(super) fn elicitations(&self) -> Option<&mpsc::Sender<Elicitation>> {
        self.elicitation.as_ref().map(|sink| &sink.events)
    }

    pub(super) fn oauth_prompts(&self) -> Option<&mpsc::Sender<Elicitation>> {
        self.elicitation.as_ref().filter(|sink| sink.capability.url.is_some()).map(|sink| &sink.events)
    }
}