From 71d5bf46d8e630ad0f0324877cd51791fe7100ae Mon Sep 17 00:00:00 2001
From: nathan <nathan@appflowy.io>
Date: Mon, 12 Aug 2024 09:57:14 +0800
Subject: [PATCH] chore: v2 stream message

---
 appflowy-local-ai/src/chat_plugin.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/appflowy-local-ai/src/chat_plugin.rs b/appflowy-local-ai/src/chat_plugin.rs
index 8c603f5..42d446a 100644
--- a/appflowy-local-ai/src/chat_plugin.rs
+++ b/appflowy-local-ai/src/chat_plugin.rs
@@ -11,6 +11,7 @@ use appflowy_plugin::util::{get_operating_system, OperatingSystem};
 use bytes::Bytes;
 use serde::{Deserialize, Serialize};
 
+use serde_json::Value;
 use std::collections::HashMap;
 use std::fmt::Debug;
 use std::path::PathBuf;
@@ -117,7 +118,7 @@ impl AppFlowyLocalAI {
     chat_id: &str,
     message: &str,
     metadata: serde_json::Value,
-  ) -> Result<ReceiverStream<anyhow::Result<Bytes, PluginError>>, PluginError> {
+  ) -> Result<ReceiverStream<anyhow::Result<Value, PluginError>>, PluginError> {
     trace!("[AI Plugin] ask question: {}", message);
     self.wait_until_plugin_ready().await?;
     let plugin = self.get_ai_plugin().await?;