add think segment

sdcb · Feb 7, 2025 · c92e959 · c92e959
1 parent d0db52f
commit c92e959
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/BE/Controllers/Chats/Chats/Dtos/SseResponseKind.cs b/src/BE/Controllers/Chats/Chats/Dtos/SseResponseKind.cs
@@ -11,4 +11,5 @@ public enum SseResponseKind
     TitleSegment = 5,
     ResponseMessage = 6,
     ChatLeafMessageId = 7,
+    ThinkSegment = 8,
 }
diff --git a/src/BE/Controllers/Chats/Chats/Dtos/SseResponseLine.cs b/src/BE/Controllers/Chats/Chats/Dtos/SseResponseLine.cs
@@ -28,6 +28,16 @@ public static SseResponseLine Segment(byte spanId, string segment)
         };
     }
 
+    public static SseResponseLine ThinkSegment(byte spanId, string segment)
+    {
+        return new SseResponseLine
+        {
+            SpanId = spanId,
+            Result = segment,
+            Kind = SseResponseKind.ThinkSegment,
+        };
+    }
+
     public static SseResponseLine Error(byte spanId, string error)
     {
         return new SseResponseLine