agentic 0.0.4

Support library for building Agentic MCP and Agent2Agent based systems
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

use agentic::mcp::client::{Client, ClientStdioTransportConfig};
use agentic::mcp::{CallToolParams, CreateMessageParams, CreateMessageResult, McpResponse, SamplingMessage};

type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>;

// doc: https://modelcontextprotocol.io/docs/concepts/sampling

#[tokio::main]
async fn main() -> Result<()> {
	tracing_subscriber::fmt()
		.with_max_level(tracing::Level::INFO)
		.without_time()
		.init();

	// -- Setup
	let mut client = Client::new("Demo Client", "0.1.0");

	// -- Register the sampling
	let some_stuff = 123;
	client.register_sampling_handler(
		async move |_params: CreateMessageParams| -> agentic::mcp::Result<CreateMessageResult> {
			let message_result = CreateMessageResult::new_assistant(
				//
				"The sky appears red, especially at sunrise or sunset, because sunlight passes through more of the Earth's atmosphere, scattering shorter blue wavelengths and allowing longer red wavelengths to dominate.",
				"mock-model-xp",
			);
			Ok(message_result)
		},
	);

	// -- Transport Config
	let transport_config = ClientStdioTransportConfig::new(
		// cmd and args (this MCP Server requires nodejs to be installed)
		"npx",
		["-y", "@modelcontextprotocol/server-everything"],
		None,
	);

	client.connect(transport_config).await?;

	let client = client;

	let client_for_req = client.clone();
	// -- Trigger the sampling
	let params = CallToolParams::new("sampleLLM").append_argument("prompt", "Why is the sky red?");

	let res = client_for_req.send_request(params).await?;

	let res = serde_json::to_string_pretty(&res)?;
	println!("Tool response (after sampling):\n{res}");

	// sleep 1 sec
	tokio::time::sleep(std::time::Duration::from_secs(1)).await;

	Ok(())
}