This commit is contained in:
19
e2e/rust/tests/embed_extra_test.rs
generated
Normal file
19
e2e/rust/tests/embed_extra_test.rs
generated
Normal file
@@ -0,0 +1,19 @@
|
||||
// This file is auto-generated by alef — DO NOT EDIT.
|
||||
// alef:hash:4e15143f4af1ae8bafbdb1506ef057da924484c66a19483966333558ad437e75
|
||||
// To regenerate: alef generate
|
||||
// To verify freshness: alef verify --exit-code
|
||||
// Issues & docs: https://github.com/kreuzberg-dev/alef
|
||||
//! E2e tests for category: embed_extra
|
||||
|
||||
use kreuzberg::embed_texts;
|
||||
|
||||
#[test]
|
||||
fn test_embed_texts_batch() {
|
||||
// Batch embed texts
|
||||
let texts_json: serde_json::Value = serde_json::from_str(r#"["Hello","World"]"#).unwrap();
|
||||
let texts = serde_json::from_value::<Vec<String>>(texts_json).unwrap();
|
||||
let config_json: serde_json::Value =
|
||||
serde_json::from_str(r#"{"model":{"name":"balanced","type":"preset"}}"#).unwrap();
|
||||
let config = serde_json::from_value(config_json).unwrap();
|
||||
let _ = embed_texts(texts, &config).expect("should succeed");
|
||||
}
|
||||
Reference in New Issue
Block a user