From 9e0a76f58b18776f2e72c17eab5fae4eaa8999f7 Mon Sep 17 00:00:00 2001 From: Wei Zhang Date: Mon, 25 Nov 2024 12:05:46 +0800 Subject: [PATCH] test: fix fake chat stream --- ee/tabby-webserver/src/service/thread.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ee/tabby-webserver/src/service/thread.rs b/ee/tabby-webserver/src/service/thread.rs index 55f762842c5a..e7ebcbb6ccdf 100644 --- a/ee/tabby-webserver/src/service/thread.rs +++ b/ee/tabby-webserver/src/service/thread.rs @@ -548,7 +548,9 @@ mod tests { async fn test_create_run() { let db = DbConn::new_in_memory().await.unwrap(); let user_id = create_user(&db).await.as_id(); - let chat: Arc = Arc::new(FakeChatCompletionStream); + let chat: Arc = Arc::new(FakeChatCompletionStream { + return_error: false, + }); let code: Arc = Arc::new(FakeCodeSearch); let doc: Arc = Arc::new(FakeDocSearch); let context: Arc = Arc::new(FakeContextService);