chore: fix test

appflowy-local-ai/tests/chat_test/mod.rs

@@ -5,6 +5,7 @@ use std::collections::HashMap;
 
 use appflowy_local_ai::ai_ops::{CompleteTextType, LocalAITranslateItem, LocalAITranslateRowData};
 use appflowy_plugin::manager::PluginManager;
+use serde_json::Value;
 use std::env::temp_dir;
 use std::path::PathBuf;
 use std::sync::Arc;
@@ -51,7 +52,10 @@ async fn ci_chat_stream_test() {
   let mut resp = test.stream_chat_message(&chat_id, "what is banana?").await;
   let mut list = vec![];
   while let Some(s) = resp.next().await {
-    list.push(String::from_utf8(s.unwrap().to_vec()).unwrap());
+    if let Value::Object(mut map) = s.unwrap() {
+      let s = map.remove("1").unwrap().as_str().unwrap().to_string();
+      list.push(s);
+    }
   }
 
   let answer = list.join("");

appflowy-local-ai/tests/util.rs

@@ -3,8 +3,8 @@ use appflowy_local_ai::chat_plugin::{AIPluginConfig, AppFlowyLocalAI};
 use appflowy_local_ai::embedding_plugin::{EmbeddingPluginConfig, LocalEmbedding};
 use appflowy_plugin::error::PluginError;
 use appflowy_plugin::manager::PluginManager;
-use bytes::Bytes;
-use serde_json::json;
+
+use serde_json::{json, Value};
 use simsimd::SpatialSimilarity;
 use std::f64;
 use std::path::{Path, PathBuf};
@@ -76,7 +76,7 @@ impl LocalAITest {
     &self,
     chat_id: &str,
     message: &str,
-  ) -> ReceiverStream<Result<Bytes, PluginError>> {
+  ) -> ReceiverStream<Result<Value, PluginError>> {
     self
       .local_ai
       .stream_question(chat_id, message, json!([]))