omcp 0.2.0

Utility functions
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

use crate::{
    client::io::OMcpClientTrait,
    error::{Error, Result},
    types::{BakedMcpToolTrait, McpParams, McpTool},
};
use async_trait::async_trait;

pub struct BakedClient<E> {
    handler: Box<dyn BakedMcpToolTrait<Error = E>>,
}

impl<E: std::fmt::Display + 'static> BakedClient<E> {
    pub fn new<H>(handler: H) -> Box<dyn OMcpClientTrait>
    where
        H: BakedMcpToolTrait<Error = E> + 'static,
    {
        let b = Self {
            handler: Box::new(handler),
        };

        Box::new(b)
    }
}

#[async_trait(?Send)]
impl<E: std::fmt::Display> OMcpClientTrait for BakedClient<E> {
    async fn connect(&mut self) -> Result<()> {
        Ok(())
    }
    async fn disconnect(&mut self) -> Result<()> {
        Ok(())
    }
    async fn list_tools(&mut self) -> Result<Vec<McpTool>> {
        Err(Error::NotImplemented)
    }
    async fn call(&mut self, mcp_params: &McpParams) -> Result<String> {
        match self.handler.call(mcp_params).await {
            Ok(v) => Ok(v),
            Err(e) => {
                let err_msg = format!("{e}");
                Err(Error::FunctionCallFailure { error: err_msg })
            }
        }
    }
}