langchainrust 0.16.0

A LangChain-inspired framework for building LLM applications in Rust. Supports OpenAI, Agents, Tools, Memory, Chains, RAG, BM25, Hybrid Retrieval, LangGraph, HyDE, Reranking, MultiQuery, and native Function Calling.
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

//! Buffer memory example
//!
//! Shows ConversationBufferMemory storing a multi-turn conversation (no API key required).
//!
//! # Run
//! ```bash
//! cargo run --example memory_buffer_memory
//! ```

use langchainrust::memory::BaseMemory;
use langchainrust::ConversationBufferMemory;
use std::collections::HashMap;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut memory = ConversationBufferMemory::new();

    for i in 1..=3 {
        let inputs = HashMap::from([("input".to_string(), format!("question {}", i))]);
        let outputs = HashMap::from([("output".to_string(), format!("answer {}", i))]);
        memory.save_context(&inputs, &outputs).await?;
        println!(
            "round {} saved, {} messages in total",
            i,
            memory.chat_memory().len()
        );
    }

    let loaded = memory.load_memory_variables(&HashMap::new()).await?;
    println!("\nLoaded history:\n{}", loaded.get("history").unwrap());
    Ok(())
}