A simple example to group documents to embedded, increase the group size to reduce the size of the groups

group-embeddings.rhai.rs

// This is set by the engine for every invocation of this function.
let context = #{ group_size: 3, group_id: 1 };
// The _vectors field doesn't even exists.
let doc = #{ id: "abc1234", metadata: #{ title: "super title" } };

print(doc);

// ---- start -----

fn hash(s) {
    s.to_chars().reduce(|sum| sum + this.to_int(), 0)
}

if hash(doc.id) % context.group_size == context.group_id {
    // Force embeddings regeneration
    doc._vectors = #{ openai: #{ regenerate: true } };
    // Change the template output and trigger the embedder
    doc.metadata.title = " " + doc.metadata.title;
}


// ----- end ------

print(doc);