rullst-orm 12.0.0

Fast, un-opinionated, database-agnostic ORM and query builder for Rust
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

use rullst_orm::{Orm, RagContext};
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, Serialize, Deserialize, Orm, rullst_orm::FromRow)]
#[orm(table = "documents")]
pub struct Document {
    pub id: i32,

    #[orm(rag_context)]
    pub title: String,

    #[orm(rag_context)]
    pub body: String,
    // We can't easily test pgvector + rullst_ai in a pure unit test without a DB and API key,
    // so we just test the parsing and trait generation for RagContext here.
}

#[test]
fn test_rag_context_generation() {
    let doc = Document {
        id: 1,
        title: "Hello World".to_string(),
        body: "This is a test document for RAG.".to_string(),
    };

    let context = doc.get_context();
    assert!(context.contains("title: Hello World"));
    assert!(context.contains("body: This is a test document for RAG."));
}