rust_llm 2.0.0

A 1:1 Rust port of RubyLLM 2.0: one API for chat, tools, agents, and embeddings across providers.
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

//! Port of `lib/ruby_llm/mcp/collection.rb`: the MCP servers connected to a chat, readable by
//! name (`chat.mcp().get("linear")`).

use super::Mcp;

#[derive(Clone, Default)]
pub struct Collection {
    servers: Vec<Mcp>,
}

impl Collection {
    /// `<<`: a server with the same name replaces the earlier one.
    pub(crate) fn push(&mut self, server: Mcp) {
        let name = server.name();
        self.servers.retain(|s| s.name() != name);
        self.servers.push(server);
    }

    /// `[name]`.
    pub fn get(&self, name: &str) -> Option<&Mcp> {
        self.servers.iter().find(|s| s.name() == name)
    }

    pub fn iter(&self) -> impl Iterator<Item = &Mcp> {
        self.servers.iter()
    }

    /// `empty?`.
    pub fn is_empty(&self) -> bool {
        self.servers.is_empty()
    }

    pub fn len(&self) -> usize {
        self.servers.len()
    }
}

impl std::fmt::Debug for Collection {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_list()
            .entries(self.servers.iter().map(|s| s.name()))
            .finish()
    }
}