Struct Queryable 

pub struct Queryable {
    pub query_type: QueryType,
    pub text: String,
}

A single query with its target backend type.

Fields

query_type: QueryType

Query type.

text: String

Query text.

Implementations

impl Queryable

pub fn new(query_type: QueryType, text: impl Into<String>) -> Self

Create a new queryable.

pub fn lex(text: impl Into<String>) -> Self

Create a lexical query.

Examples found in repository

examples/query_expansion.rs (line 63)

fn main() -> Result<()> {
    let db_path = std::env::temp_dir().join("qmd_expansion.db");
    let _ = std::fs::remove_file(&db_path);
    let store = Store::open(&db_path)?;

    let now = chrono::Utc::now().to_rfc3339();
    for (name, content) in SAMPLE_DOCS {
        let hash = Store::hash_content(content);
        let title = Store::extract_title(content);
        store.insert_content(&hash, content, &now)?;
        store.insert_document("samples", name, &title, &hash, &now, &now)?;
    }

    let query = "rust error handling";
    println!("Query: '{}'\n", query);

    // Simple expansion
    println!("Simple expansion:");
    for q in expand_query_simple(query) {
        let t = match q.query_type {
            QueryType::Lex => "LEX",
            QueryType::Vec => "VEC",
            QueryType::Hyde => "HYD",
        };
        println!("  [{}] {}", t, q.text);
    }

    // Search with expanded queries
    println!("\nSearch results:");
    for q in expand_query_simple(query) {
        if q.query_type == QueryType::Lex {
            let n = store.search_fts(&q.text, 5, None)?.len();
            println!("  '{}': {} results", q.text, n);
        }
    }

    // LLM expansion
    println!("\nLLM expansion:");
    if GenerationEngine::is_available() {
        let engine = GenerationEngine::load_default()?;
        for q in engine.expand_query(query, true)? {
            println!("  [{:?}] {}", q.query_type, q.text);
        }
    } else {
        println!("  (not available)");
    }

    // Manual construction
    println!("\nManual:");
    for q in [
        Queryable::lex("rust error"),
        Queryable::vec("exception handling"),
    ] {
        println!("  [{:?}] {}", q.query_type, q.text);
    }

    let _ = std::fs::remove_file(&db_path);
    Ok(())
}

pub fn vec(text: impl Into<String>) -> Self

Create a vector query.

Examples found in repository

examples/query_expansion.rs (line 64)

fn main() -> Result<()> {
    let db_path = std::env::temp_dir().join("qmd_expansion.db");
    let _ = std::fs::remove_file(&db_path);
    let store = Store::open(&db_path)?;

    let now = chrono::Utc::now().to_rfc3339();
    for (name, content) in SAMPLE_DOCS {
        let hash = Store::hash_content(content);
        let title = Store::extract_title(content);
        store.insert_content(&hash, content, &now)?;
        store.insert_document("samples", name, &title, &hash, &now, &now)?;
    }

    let query = "rust error handling";
    println!("Query: '{}'\n", query);

    // Simple expansion
    println!("Simple expansion:");
    for q in expand_query_simple(query) {
        let t = match q.query_type {
            QueryType::Lex => "LEX",
            QueryType::Vec => "VEC",
            QueryType::Hyde => "HYD",
        };
        println!("  [{}] {}", t, q.text);
    }

    // Search with expanded queries
    println!("\nSearch results:");
    for q in expand_query_simple(query) {
        if q.query_type == QueryType::Lex {
            let n = store.search_fts(&q.text, 5, None)?.len();
            println!("  '{}': {} results", q.text, n);
        }
    }

    // LLM expansion
    println!("\nLLM expansion:");
    if GenerationEngine::is_available() {
        let engine = GenerationEngine::load_default()?;
        for q in engine.expand_query(query, true)? {
            println!("  [{:?}] {}", q.query_type, q.text);
        }
    } else {
        println!("  (not available)");
    }

    // Manual construction
    println!("\nManual:");
    for q in [
        Queryable::lex("rust error"),
        Queryable::vec("exception handling"),
    ] {
        println!("  [{:?}] {}", q.query_type, q.text);
    }

    let _ = std::fs::remove_file(&db_path);
    Ok(())
}

pub fn hyde(text: impl Into<String>) -> Self

Create a HyDE query.

Trait Implementations

impl Clone for Queryable

fn clone(&self) -> Queryable

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Queryable

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.