chore: making embeddings code compile

services/appflowy-collaborate/src/collab/queue.rs

@@ -122,6 +122,7 @@ impl StorageQueue {
       workspace_id: workspace_id.to_string(),
       object_id: params.object_id.clone(),
       collab_type: params.collab_type.clone(),
+      embeddings: params.embeddings.clone(),
     };
 
     self
@@ -419,7 +420,7 @@ async fn write_pending_to_disk(
           object_id: meta.object_id.clone(),
           collab_type: meta.collab_type.clone(),
           encode_collab_v1,
-          embeddings,
+          embeddings: meta.embeddings.clone(),
         })
       } else {
         None
@@ -526,6 +527,8 @@ pub struct PendingWriteMeta {
   pub workspace_id: String,
   pub object_id: String,
   pub collab_type: CollabType,
+  #[serde(default)]
+  pub embeddings: Option<AFCollabEmbeddings>,
 }
 
 impl From<&PendingWriteMeta> for QueryCollab {

src/api/workspace.rs

@@ -609,7 +609,7 @@ async fn batch_create_collab_handler(
     {
       state
         .collab_access_control_storage
-        .insert_new_collab(&workspace_id, &uid, params, None)
+        .insert_new_collab(&workspace_id, &uid, params)
         .await?;
 
       state
@@ -854,7 +854,7 @@ async fn update_collab_handler(
   let (params, workspace_id) = create_params.split();
   state
     .collab_access_control_storage
-    .insert_or_update_collab(&workspace_id, &uid, params, None, false)
+    .insert_or_update_collab(&workspace_id, &uid, params, false)
     .await?;
   Ok(AppResponse::Ok().into())
 }

src/biz/user/user_init.rs

@@ -66,6 +66,7 @@ where
           object_id,
           encoded_collab_v1,
           collab_type: template.object_type,
+          embeddings: None,
         },
         txn,
       )
@@ -104,6 +105,7 @@ async fn create_workspace_database_collab(
         object_id: object_id.to_string(),
         encoded_collab_v1,
         collab_type,
+        embeddings: None,
       },
       txn,
     )

tests/collab/collab_curd_test.rs

@@ -87,6 +87,7 @@ async fn batch_insert_collab_success_test() {
       object_id: Uuid::new_v4().to_string(),
       encoded_collab_v1: mock_encoded_collab_v1[i].encode_to_bytes().unwrap(),
       collab_type: CollabType::Unknown,
+      embeddings: None,
     })
     .collect::<Vec<_>>();
 
@@ -182,6 +183,7 @@ async fn create_collab_compatibility_with_json_params_test() {
       object_id: object_id.clone(),
       encoded_collab_v1: encoded_collab.encode_to_bytes().unwrap(),
       collab_type: CollabType::Unknown,
+      embeddings: None,
     },
     workspace_id: workspace_id.clone(),
   };
@@ -238,6 +240,7 @@ async fn batch_create_collab_compatibility_with_uncompress_params_test() {
       object_id: object_id.clone(),
       encoded_collab_v1: encoded_collab.encode_to_bytes().unwrap(),
       collab_type: CollabType::Unknown,
+      embeddings: None,
     }],
   }
   .to_bytes()

tests/collab/storage_test.rs

@@ -468,6 +468,7 @@ async fn simulate_small_data_set_write(pool: PgPool) {
       object_id: format!("object_id_{}", i),
       collab_type: CollabType::Unknown,
       encoded_collab_v1: encode_collab.encode_to_bytes().unwrap(),
+      embeddings: None,
     };
     cloned_storage_queue
       .push(
@@ -543,6 +544,7 @@ async fn simulate_large_data_set_write(pool: PgPool) {
     object_id: uuid::Uuid::new_v4().to_string(),
     collab_type: CollabType::Unknown,
     encoded_collab_v1: origin_encode_collab.encode_to_bytes().unwrap(),
+    embeddings: None,
   };
   storage_queue
     .push(&user.workspace_id, &user.uid, &params, WritePriority::Low)

tests/sql_test/workspace_test.rs

@@ -29,6 +29,7 @@ async fn insert_collab_sql_test(pool: PgPool) {
       object_id,
       collab_type: CollabType::Unknown,
       encoded_collab_v1,
+      embeddings: None,
     };
     insert_into_af_collab(&mut txn, &user.uid, &user.workspace_id, &params)
       .await