21 lines
573 B
Rust
21 lines
573 B
Rust
use appflowy_ai_client::dto::CompletionType;
|
|
use client_api_test::TestClient;
|
|
use shared_entity::dto::ai_dto::CompleteTextParams;
|
|
|
|
#[tokio::test]
|
|
async fn improve_writing_test() {
|
|
let test_client = TestClient::new_user().await;
|
|
let workspace_id = test_client.workspace_id().await;
|
|
let params = CompleteTextParams {
|
|
text: "I feel hungry".to_string(),
|
|
completion_type: CompletionType::ImproveWriting,
|
|
};
|
|
|
|
let resp = test_client
|
|
.api_client
|
|
.completion_text(&workspace_id, params)
|
|
.await
|
|
.unwrap();
|
|
assert!(resp.text.contains("hungry"));
|
|
}
|