21 lines
667 B
Rust
21 lines
667 B
Rust
use crate::appflowy_ai_client;
|
|
|
|
use appflowy_ai_client::dto::{
|
|
EmbeddingEncodingFormat, EmbeddingInput, EmbeddingModel, EmbeddingRequest,
|
|
};
|
|
|
|
#[tokio::test]
|
|
async fn embedding_test() {
|
|
let client = appflowy_ai_client();
|
|
let request = EmbeddingRequest {
|
|
input: EmbeddingInput::String("hello world".to_string()),
|
|
model: EmbeddingModel::TextEmbedding3Small.to_string(),
|
|
chunk_size: 1000,
|
|
encoding_format: EmbeddingEncodingFormat::Float,
|
|
dimensions: EmbeddingModel::TextEmbedding3Small.default_dimensions(),
|
|
};
|
|
let result = client.embeddings(request).await.unwrap();
|
|
assert!(result.total_tokens > 0);
|
|
assert!(!result.data.is_empty());
|
|
}
|