dnrops
/
rig
mirror of https://github.com/0xplaygrounds/rig
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: make ci pass

This commit is contained in:
Joshua Mo 2025-04-18 12:43:31 +01:00
parent 0e2354977f
commit 12720160c2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rig-fastembed/examples/vector_search_local.rs
View File

@ -7,7 +7,7 @@ use rig::{
vector_store::{in_memory_store::InMemoryVectorStore, VectorStoreIndex},
Embed,
};
use rig_fastembed::EmbeddingModel;
use rig_fastembed::TextEmbeddingModel;
use serde::{Deserialize, Serialize};
use std::path::Path;
@ -51,7 +51,7 @@ async fn main() -> Result<(), anyhow::Error> {
UserDefinedEmbeddingModel::new(onnx_file, tokenizer_files).with_pooling(Pooling::Mean);
let embedding_model =
EmbeddingModel::new_from_user_defined(user_defined_model, 384, &test_model_info);
TextEmbeddingModel::new_from_user_defined(user_defined_model, 384, &test_model_info);
// Create documents
let documents = vec![