fix(thread): add truncated field to MessageAttachmentCodeFileList (#3944

) * fix(thread): add truncated field to MessageAttachmentCodeFileList * update * update --------- Co-authored-by: liangfung <[email protected]>
TabbyML · Mar 5, 2025 · 5931d10 · 5931d10
1 parent 9cb0a07
commit 5931d10
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/ee/tabby-schema/graphql/schema.graphql b/ee/tabby-schema/graphql/schema.graphql
@@ -574,6 +574,7 @@ type MessageAttachmentCode {
 
 type MessageAttachmentCodeFileList {
   fileList: [String!]!
+  truncated: Boolean!
 }
 
 type MessageAttachmentCodeScores {

diff --git a/ee/tabby-schema/src/dao.rs b/ee/tabby-schema/src/dao.rs
@@ -270,6 +270,7 @@ impl From<AttachmentCodeFileList> for thread::MessageAttachmentCodeFileList {
     fn from(value: AttachmentCodeFileList) -> Self {
         Self {
             file_list: value.file_list,
+            truncated: value.truncated,
         }
     }
 }

diff --git a/ee/tabby-schema/src/schema/thread/types.rs b/ee/tabby-schema/src/schema/thread/types.rs
@@ -81,6 +81,7 @@ pub struct MessageAttachment {
 #[derive(GraphQLObject, Clone)]
 pub struct MessageAttachmentCodeFileList {
     pub file_list: Vec<String>,
+    pub truncated: bool,
 }
 
 #[derive(GraphQLObject, Clone)]

diff --git a/ee/tabby-webserver/src/service/answer.rs b/ee/tabby-webserver/src/service/answer.rs
@@ -94,6 +94,7 @@ impl AnswerService {
                             Ok((file_list, truncated)) => {
                                 attachment.code_file_list = Some(MessageAttachmentCodeFileList {
                                     file_list: file_list.clone(),
+                                    truncated,
                                 });
                                 yield Ok(ThreadRunItem::ThreadAssistantMessageAttachmentsCodeFileList(ThreadAssistantMessageAttachmentsCodeFileList {
                                     file_list,
@@ -474,6 +475,7 @@ mod tests {
             }],
             code_file_list: Some(MessageAttachmentCodeFileList {
                 file_list: vec!["client.py".to_owned(), "server.py".to_owned()],
+                truncated: false,
             }),
         };
-Original file line number
+Diff line change
@@ Expand Up @@
         fn from(value: AttachmentCodeFileList) -> Self {
             Self {
                 file_list: value.file_list,
+                truncated: value.truncated,
             }
         }
     }
@@ Expand Down @@