chore: v2 stream message

AppFlowy-IO · Aug 12, 2024 · 4d9d47b · 4d9d47b
1 parent 71d5bf4
commit 4d9d47b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/appflowy-local-ai/src/ai_ops.rs b/appflowy-local-ai/src/ai_ops.rs
@@ -229,7 +229,10 @@ impl ResponseParser for ChatStreamResponseV2Parser {
   type ValueType = serde_json::Value;
 
   fn parse_json(json: JsonValue) -> Result<Self::ValueType, RemoteError> {
-    Ok(json)
+    json
+      .as_str()
+      .and_then(|s| serde_json::from_str(s).ok())
+      .ok_or(RemoteError::ParseResponse(json))
   }
 }