rag 0.1.1

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

use rmcp::{ServiceExt, transport::stdio};
use rag::mcp::RagMcpServer;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    tracing_subscriber::fmt()
        .with_env_filter(
            tracing_subscriber::EnvFilter::from_default_env()
                .add_directive(tracing::Level::DEBUG.into()),
        )
        .with_writer(std::io::stderr)
        .with_ansi(false)
        .init();

    tracing::info!("Starting RAG MCP Server");

    let api_key = std::env::var("OPENAI_API_KEY").ok();
    let ollama_url = std::env::var("OLLAMA_URL").ok();
    let ollama_model =
        std::env::var("OLLAMA_MODEL").unwrap_or_else(|_| "nomic-embed-text".to_string());

    let server = match api_key {
        Some(key) => {
            tracing::info!("Using OpenAI embedding model");
            RagMcpServer::new_openai(key)
        }
        None => {
            tracing::info!("Using Ollama embedding model: {}", ollama_model);
            RagMcpServer::new_ollama(ollama_model, ollama_url)
        }
    };

    let service = server.serve(stdio()).await.inspect_err(|e| {
        tracing::error!("Server error: {:?}", e);
    })?;

    tracing::info!("RAG MCP Server started, waiting for requests...");
    service.waiting().await?;
    Ok(())
}