pub fn embed(model: &str, value: &str) -> Result<EmbeddingResult>Expand description
Embeds a single string with the given model id.
Examples found in repository?
examples/embeddings.rs (line 4)
3fn main() {
4 let query = embed("openai/text-embedding-3-small", "rust cargo").expect("embed failed");
5 let docs = vec![
6 (
7 "rust crates",
8 embed("openai/text-embedding-3-small", "rust crates")
9 .expect("embed failed")
10 .embedding,
11 ),
12 (
13 "garden flowers",
14 embed("openai/text-embedding-3-small", "garden flowers")
15 .expect("embed failed")
16 .embedding,
17 ),
18 (
19 "cargo workspace",
20 embed("openai/text-embedding-3-small", "cargo workspace")
21 .expect("embed failed")
22 .embedding,
23 ),
24 ];
25
26 let ranked = rank_by_similarity(
27 &query.embedding,
28 docs.iter()
29 .map(|(name, embedding)| (*name, embedding.as_slice())),
30 );
31
32 println!("best match: {}", ranked[0].0);
33}