From 5931d102a2ddf4d6e9e7cecf603d73267ef6d614 Mon Sep 17 00:00:00 2001
From: Meng Zhang <meng@tabbyml.com>
Date: Wed, 5 Mar 2025 15:11:06 +0800
Subject: [PATCH] fix(thread): add truncated field to
 MessageAttachmentCodeFileList (#3944)

* fix(thread): add truncated field to MessageAttachmentCodeFileList

* update

* update

---------

Co-authored-by: liangfung <jueliang@tabbyml.com>
---
 ee/tabby-schema/graphql/schema.graphql     | 1 +
 ee/tabby-schema/src/dao.rs                 | 1 +
 ee/tabby-schema/src/schema/thread/types.rs | 1 +
 ee/tabby-webserver/src/service/answer.rs   | 2 ++
 4 files changed, 5 insertions(+)

diff --git a/ee/tabby-schema/graphql/schema.graphql b/ee/tabby-schema/graphql/schema.graphql
index b896e9e99418..73eead3a67b2 100644
--- 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
index 6921aa5c0962..dab818615fd3 100644
--- 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
index a9ba69400521..7fce757f78c1 100644
--- 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
index 3aac90a9a59a..c38f0cf614e2 100644
--- 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,
             }),
         };