fix(answer): handle cases that no choices is returned from chat model. (

#3329) * fix(answer): handle cases that no choices is returned from chat model. * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
TabbyML · Oct 28, 2024 · 97f93df · 97f93df
1 parent d7b3c9b
commit 97f93df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ee/tabby-webserver/src/service/answer.rs b/ee/tabby-webserver/src/service/answer.rs
@@ -183,7 +183,8 @@ impl AnswerService {
                     }
                 };
 
-                if let Some(content) = chunk.choices[0].delta.content.as_deref() {
+                let content = chunk.choices.first().and_then(|x| x.delta.content.as_deref());
+                if let Some(content) = content {
                     yield Ok(ThreadRunItem::ThreadAssistantMessageContentDelta(ThreadAssistantMessageContentDelta {
                         delta: content.to_owned()
                     }));

diff --git a/ee/tabby-webserver/src/service/user_group.rs b/ee/tabby-webserver/src/service/user_group.rs
@@ -7,7 +7,7 @@ use tabby_schema::{
     AsID, AsRowid, Result,
 };
 
-use super::{UserGroupExt, UserSecuredExt};
+use super::UserGroupExt;
 
 struct UserGroupServiceImpl {
     db: DbConn,