feat: support chat reasoning

Author: ensorrow

packages/ai-native/src/browser/chat/chat-model.ts

@@ -6,6 +6,7 @@ import {
   IChatComponent,
   IChatMarkdownContent,
   IChatProgress,
+  IChatReasoning,
   IChatToolContent,
   IChatTreeData,
   uuid,
@@ -33,7 +34,8 @@ export type IChatProgressResponseContent =
   | IChatAsyncContent
   | IChatTreeData
   | IChatComponent
-  | IChatToolContent;
+  | IChatToolContent
+  | IChatReasoning;
 
 export class ChatResponseModel extends Disposable {
   #responseParts: IChatProgressResponseContent[] = [];
@@ -131,6 +133,18 @@ export class ChatResponseModel extends Disposable {
         };
       }
 
+      this.#updateResponseText();
+    } else if (progress.kind === 'reasoning') {
+      const lastResponsePart = this.#responseParts[responsePartLength];
+      if (!lastResponsePart || lastResponsePart.kind !== 'reasoning') {
+        // 去掉开头的 <think> 标签
+        this.#responseParts.push({ content: progress.content.replace(/^<think>/, ''), kind: 'reasoning' });
+      } else {
+        this.#responseParts[responsePartLength] = {
+          content: lastResponsePart.content + progress.content,
+          kind: 'reasoning',
+        };
+      }
       this.#updateResponseText();
     } else if (progress.kind === 'asyncContent') {
       // Add a new resolving part
@@ -181,6 +195,9 @@ export class ChatResponseModel extends Disposable {
         if (part.kind === 'toolCall') {
           return part.content.function.name;
         }
+        if (part.kind === 'reasoning') {
+          return '';
+        }
         return part.content.value;
       })
       .join('\n\n');
@@ -387,7 +404,7 @@ export class ChatModel extends Disposable implements IChatModel {
 
     const { kind } = progress;
 
-    const basicKind = ['content', 'markdownContent', 'asyncContent', 'treeData', 'component', 'toolCall'];
+    const basicKind = ['content', 'markdownContent', 'asyncContent', 'treeData', 'component', 'toolCall', 'reasoning'];
 
     if (basicKind.includes(kind)) {
       request.response.updateContent(progress, quiet);

packages/ai-native/src/browser/components/ChatReply.tsx

@@ -1,3 +1,4 @@
+import cls from 'classnames';
 import React, {
   Fragment,
   ReactNode,
@@ -313,6 +314,12 @@ export const ChatReply = (props: IChatReplyProps) => {
           node = <ComponentRender component={item.component} value={item.value} messageId={msgId} />;
         } else if (item.kind === 'toolCall') {
           node = <ToolCallRender toolCall={item.content} messageId={msgId} />;
+        } else if (item.kind === 'reasoning') {
+          node = (
+            <div className={cls(styles.reasoning, { [styles.collapse]: request.response.isComplete })}>
+              {item.content}
+            </div>
+          );
         } else {
           node = renderMarkdown(item.content);
         }

packages/ai-native/src/browser/components/components.module.less

@@ -541,3 +541,9 @@
     }
   }
 }
+
+.reasoning {
+  padding-left: 12px;
+  border-left: 2px solid var(--design-text-foreground);
+  color: var(--design-text-foreground);
+}

packages/ai-native/src/node/base-language-model.ts

@@ -185,6 +185,11 @@ export abstract class BaseLanguageModel {
           });
         } else if (chunk.type === 'error') {
           chatReadableStream.emitError(new Error(chunk.error as string));
+        } else if (chunk.type === 'reasoning') {
+          chatReadableStream.emitData({
+            kind: 'reasoning',
+            content: chunk.textDelta,
+          });
         }
       }
 

packages/core-common/src/types/ai-native/index.ts

@@ -401,13 +401,19 @@ export interface IChatComponent {
   kind: 'component';
 }
 
+export interface IChatReasoning {
+  content: string;
+  kind: 'reasoning';
+}
+
 export type IChatProgress =
   | IChatContent
   | IChatMarkdownContent
   | IChatAsyncContent
   | IChatTreeData
   | IChatComponent
-  | IChatToolContent;
+  | IChatToolContent
+  | IChatReasoning;
 
 export interface IChatMessage {
   role: ChatMessageRole;