sachaarbonel

fn main() -> anyhow::Result<()> {

let model = SentenceEmbeddingsBuilder::remote(SentenceEmbeddingsModelType::AllMiniLmL12V2).create_model()?;
let json = fs::read_to_string("data/books.json")?;
let library: Library = serde_json::from_str(&json)?;
 let mut embeddedbooks = Vec::new();
for book in library.books.clone() {
    println!("Embedding book: {}", book.title);
    let embeddings = model.encode(&[book.clone().summary])?;
    let embedding = to_array(embeddings[0].as_slice());

sachaarbonel

impl EmbeddedBook {
    fn topic(embeddings: [f32; 384]) -> Self {
        Self {
            title: None,
            author: None,
            summary: None,
            embeddings: embeddings,
        }
    }
}

sachaarbonel

fn main() -> anyhow::Result<()> {

let model = SentenceEmbeddingsBuilder::remote(SentenceEmbeddingsModelType::AllMiniLmL12V2).create_model()?;
let json = fs::read_to_string("data/books.json")?;
let library: Library = serde_json::from_str(&json)?;
 let mut embeddedbooks = Vec::new();
for book in library.books.clone() {
    println!("Embedding book: {}", book.title);
    let embeddings = model.encode(&[book.clone().summary])?;
    let embedding = to_array(embeddings[0].as_slice());

sachaarbonel

impl KdPoint for EmbeddedBook {
    type Scalar = f32;
    type Dim = typenum::U2; // 2 dimensional tree.
    fn at(&self, k: usize) -> f32 {
        self.embeddings[k]
    }
}

sachaarbonel

fn main() -> anyhow::Result<()> {

+ let model = SentenceEmbeddingsBuilder::remote(SentenceEmbeddingsModelType::AllMiniLmL12V2).create_model()?;
let json = fs::read_to_string("data/books.json")?;
let library: Library = serde_json::from_str(&json)?;
+ let mut embeddedbooks = Vec::new();
for book in library.books.clone() {
-    println!("Embedding book: {}", book.title);
+    let embeddings = model.encode(&[book.clone().summary])?;
+    let embedding = to_array(embeddings[0].as_slice());

sachaarbonel

#[derive(Debug)]
pub struct EmbeddedBook {
    pub title: String,

    pub author: String,

    pub summary: String,

    pub embeddings: [f32; 384],
}

sachaarbonel

fn main() -> anyhow::Result<()> {

let json = fs::read_to_string("data/books.json")?;
let library: Library = serde_json::from_str(&json)?;
for book in library.books.clone() {
    println!("Embedding book: {}", book.title);
}

    Ok(())
}

sachaarbonel

use serde::{Deserialize};

#[derive(Debug, Deserialize)]
pub struct Library {
    pub books: Vec<Book>,
}

#[derive(Debug, Deserialize, Clone)]
pub struct Book {
    pub title: String,

sachaarbonel

{
    "books": [
        {
            "title": "The Great Gatsby",
            "author": "F. Scott Fitzgerald",
            "summary": "The story primarily concerns the young and mysterious millionaire Jay Gatsby and his quixotic passion and obsession with the beautiful former debutante Daisy Buchanan."
        },
        {
            "title": "The Catcher in the Rye",
            "author": "J. D. Salinger",

sachaarbonel

let kdtree = kd_tree::KdTree::build_by_ordered_float(vec![
    [1.0, 2.0, 3.0],
    [3.0, 1.0, 2.0],
    [2.0, 3.0, 1.0],
]);

// search the nearest neighbor
let found = kdtree.nearest(&[3.1, 0.9, 2.1]).unwrap();