chore: complete text endpoints

deploy.env

@@ -75,7 +75,7 @@ GOTRUE_EXTERNAL_DISCORD_REDIRECT_URI=http://your-host/gotrue/callback
 # This is where storage like images, files, etc. will be stored
 # By default, Minio is used as the default file storage which uses host's file system
 APPFLOWY_S3_USE_MINIO=true
-APPFLOWY_S3_MINIO_URL=http://minio:9000 # change this if you are using a different address for minio
+APPFLOWY_S3_MINIO_URL=http://minio:9000# change this if you are using a different address for minio
 APPFLOWY_S3_ACCESS_KEY=minioadmin
 APPFLOWY_S3_SECRET_KEY=minioadmin
 APPFLOWY_S3_BUCKET=appflowy
@@ -108,7 +108,6 @@ CLOUDFLARE_TUNNEL_TOKEN=
 
 # AppFlowy AI
 APPFLOWY_AI_OPENAI_API_KEY=
-APPFLOWY_AI_SERVER_HOST=ai
 APPFLOWY_AI_SERVER_PORT=5001
 
 # AppFlowy History

dev.env

@@ -100,7 +100,6 @@ CLOUDFLARE_TUNNEL_TOKEN=
 
 # AppFlowy AI
 APPFLOWY_AI_OPENAI_API_KEY=
-APPFLOWY_AI_SERVER_HOST=localhost
 APPFLOWY_AI_SERVER_PORT=5001
 
 # AppFlowy History

src/api/ai_tool.rs

@@ -1,10 +1,8 @@
-use crate::biz::chat::ops::create_chat;
 use crate::state::AppState;
 use actix_web::web::{Data, Json};
-use actix_web::{web, App, Scope};
+use actix_web::{web, Scope};
 use app_error::AppError;
 use appflowy_ai_client::dto::CompleteTextResponse;
-use database_entity::chat::CreateChatParams;
 use shared_entity::dto::ai_dto::{
   CompleteTextParams, SummarizeRowData, SummarizeRowParams, SummarizeRowResponse,
 };

src/api/workspace.rs

@@ -22,7 +22,6 @@ use collab_rt_protocol::validate_encode_collab;
 use database::collab::CollabStorage;
 use database::user::select_uid_from_email;
 use database_entity::dto::*;
-use shared_entity::dto::ai_dto::{SummarizeRowData, SummarizeRowParams, SummarizeRowResponse};
 use shared_entity::dto::workspace_dto::*;
 use shared_entity::response::AppResponseError;
 use shared_entity::response::{AppResponse, JsonAppResponse};

tests/ai_test/complete_text.rs

@@ -0,0 +1,20 @@
+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"));
+}

tests/ai_test/mod.rs

@@ -1 +1,2 @@
+mod complete_text;
 mod summarize_row;