rmcp-agent 0.1.1

A library that extends langchain-rust with support for Model Context Protocol (MCP) tool integration and streaming tool execution capabilities.
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

use std::pin::Pin;

use async_trait::async_trait;
use futures_util::Stream;
use langchain_rust::agent::{Agent, AgentError};
use langchain_rust::chain::ChainError;
use langchain_rust::prompt::PromptArgs;
use langchain_rust::schemas::{AgentAction, AgentEvent};

use crate::agent::intermediate::IntermediateStep;

#[async_trait]
pub trait AgentExt: Agent {
    async fn plan_with_steps(
        &self,
        intermediate_steps: &[impl IntermediateStep],
        inputs: PromptArgs,
    ) -> Result<AgentEvent, AgentError>;

    async fn plan_stream(
        &self,
        steps: &[impl IntermediateStep],
        inputs: PromptArgs,
    ) -> Result<AgentStream, AgentError>;
}

pub type AgentStream = Pin<Box<dyn Stream<Item = Result<AgentEventChunk, ChainError>> + Send>>;

pub enum AgentEventChunk {
    Delta(DeltaEvent),
    Final(AgentEvent),
}

pub enum DeltaEvent {
    Action(AgentAction),
    Content(String),
}