26 lines
712 B
Rust
26 lines
712 B
Rust
use appflowy_ai_client::dto::{AIModel, CompletionType};
|
|
use client_api_test::{local_ai_test_enabled, TestClient};
|
|
use shared_entity::dto::ai_dto::CompleteTextParams;
|
|
|
|
#[tokio::test]
|
|
async fn improve_writing_test() {
|
|
if !local_ai_test_enabled() {
|
|
return;
|
|
}
|
|
let test_client = TestClient::new_user().await;
|
|
test_client.api_client.set_ai_model(AIModel::GPT4oMini);
|
|
|
|
let workspace_id = test_client.workspace_id().await;
|
|
let params = CompleteTextParams::new_with_completion_type(
|
|
"I feel hungry".to_string(),
|
|
CompletionType::ImproveWriting,
|
|
);
|
|
|
|
let resp = test_client
|
|
.api_client
|
|
.completion_text(&workspace_id, params)
|
|
.await
|
|
.unwrap();
|
|
assert!(!resp.text.is_empty());
|
|
}
|