rag 0.1.0

A Rust library and CLI for Retrieval-Augmented Generation
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
70
71
72
73
74
75
76
77
78
79
80
81
82

use rag::{
    mcp::{McpRequest, McpServer},
};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    println!("RAG MCP Server Example\n");

    let mcp_server = McpServer::new_ollama();

    println!("1. Initializing MCP server...");
    let init_req = McpRequest {
        jsonrpc: "2.0".to_string(),
        id: Some(serde_json::json!(1)),
        method: "initialize".to_string(),
        params: None,
    };

    let init_response = mcp_server.handle_request(init_req).await;
    println!("Initialize response: {}\n", serde_json::to_string_pretty(&init_response)?);

    println!("2. Listing available tools...");
    let list_req = McpRequest {
        jsonrpc: "2.0".to_string(),
        id: Some(serde_json::json!(2)),
        method: "tools/list".to_string(),
        params: None,
    };

    let list_response = mcp_server.handle_request(list_req).await;
    println!("Tools list response: {}\n", serde_json::to_string_pretty(&list_response)?);

    println!("3. Adding a document...");
    let add_req = McpRequest {
        jsonrpc: "2.0".to_string(),
        id: Some(serde_json::json!(3)),
        method: "tools/call".to_string(),
        params: Some(serde_json::json!({
            "name": "rag_add_document",
            "arguments": {
                "content": "Rust is a systems programming language that runs blazingly fast, prevents segfaults, and guarantees thread safety. Retrieval-Augmented Generation (RAG) is a technique that enhances large language models by providing them with relevant external context.",
                "source": "example-doc"
            }
        })),
    };

    let add_response = mcp_server.handle_request(add_req).await;
    println!("Add document response: {}\n", serde_json::to_string_pretty(&add_response)?);

    println!("4. Querying the document...");
    let query_req = McpRequest {
        jsonrpc: "2.0".to_string(),
        id: Some(serde_json::json!(4)),
        method: "tools/call".to_string(),
        params: Some(serde_json::json!({
            "name": "rag_query",
            "arguments": {
                "query": "What is Rust?",
                "top_k": 3
            }
        })),
    };

    let query_response = mcp_server.handle_request(query_req).await;
    println!("Query response: {}\n", serde_json::to_string_pretty(&query_response)?);

    println!("5. Counting documents...");
    let count_req = McpRequest {
        jsonrpc: "2.0".to_string(),
        id: Some(serde_json::json!(5)),
        method: "tools/call".to_string(),
        params: Some(serde_json::json!({
            "name": "rag_count",
            "arguments": {}
        })),
    };

    let count_response = mcp_server.handle_request(count_req).await;
    println!("Count response: {}\n", serde_json::to_string_pretty(&count_response)?);

    Ok(())
}