chk
/
AppFlowy-Cloud
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: fix ai client test (#589)

This commit is contained in:
Nathan.fooo 2024-05-30 07:50:25 +08:00 committed by GitHub
parent 128e978016
commit 9b4bfce616
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Cargo.lock generated
View File

@ -576,6 +576,7 @@ dependencies = [
"tokio", "tokio",
"tracing", "tracing",
"tracing-subscriber", "tracing-subscriber",
"uuid",
] ]
[[package]] [[package]]

1
libs/appflowy-ai-client/Cargo.toml
View File

@ -17,6 +17,7 @@ serde_repr = { version = "0.1", optional = true }
[dev-dependencies] [dev-dependencies]
tokio = { version = "1.37.0", features = ["macros", "test-util"] } tokio = { version = "1.37.0", features = ["macros", "test-util"] }
tracing-subscriber = { version = "0.3.18", features = ["registry", "env-filter", "ansi", "json"] } tracing-subscriber = { version = "0.3.18", features = ["registry", "env-filter", "ansi", "json"] }
uuid = { version = "1.6", features = ["v4"] }
[features] [features]
default = ["client-api"] default = ["client-api"]

3
libs/appflowy-ai-client/tests/chat_test/qa_test.rs
View File

@ -4,8 +4,9 @@ use crate::appflowy_ai_client;
async fn qa_test() { async fn qa_test() {
let client = appflowy_ai_client(); let client = appflowy_ai_client();
client.health_check().await.unwrap(); client.health_check().await.unwrap();
let chat_id = uuid::Uuid::new_v4().to_string();
let resp = client let resp = client
.send_question("fake_chat_id", "I feel hungry") .send_question(&chat_id, "I feel hungry")
.await .await
.unwrap(); .unwrap();
assert!(!resp.content.is_empty()); assert!(!resp.content.is_empty());

4
services/appflowy-history/src/core/manager.rs
View File

@ -97,6 +97,10 @@ async fn spawn_control_group(
.await .await
{ {
if let Some(handles) = weak_handles.upgrade() { if let Some(handles) = weak_handles.upgrade() {
if messages.is_empty() {
continue;
}
trace!("[History] received {} control messages", messages.len()); trace!("[History] received {} control messages", messages.len());
for message in &messages { for message in &messages {
if let Ok(event) = CollabControlEvent::decode(&message.data) { if let Ok(event) = CollabControlEvent::decode(&message.data) {