feat: support custom chat render (#3499)

Co-authored-by: 野声 <lijiacheng.ljc@antgroup.com>

packages/ai-native/src/browser/ai-core.contribution.ts

@@ -29,6 +29,7 @@ import { InlineChatIsVisible } from '@opensumi/ide-core-browser/lib/contextkey/a
 import { LayoutViewSizeConfig } from '@opensumi/ide-core-browser/lib/layout/constants';
 import {
   ChatFeatureRegistryToken,
+  ChatRenderRegistryToken,
   CommandService,
   InlineChatFeatureRegistryToken,
   RenameCandidatesProviderRegistryToken,
@@ -50,6 +51,7 @@ import { AIChatTabRenderer, AILeftTabRenderer, AIRightTabRenderer } from './layo
 import {
   AINativeCoreContribution,
   IChatFeatureRegistry,
+  IChatRenderRegistry,
   IInlineChatFeatureRegistry,
   IRenameCandidatesProviderRegistry,
   IResolveConflictRegistry,
@@ -89,6 +91,9 @@ export class AINativeBrowserContribution
   @Autowired(ChatFeatureRegistryToken)
   private readonly chatFeatureRegistry: IChatFeatureRegistry;
 
+  @Autowired(ChatRenderRegistryToken)
+  private readonly chatRenderRegistry: IChatRenderRegistry;
+
   @Autowired(ResolveConflictRegistryToken)
   private readonly resolveConflictRegistry: IResolveConflictRegistry;
 
@@ -162,6 +167,9 @@ export class AINativeBrowserContribution
       if (contribution.registerRenameProvider) {
         contribution.registerRenameProvider(this.renameCandidatesProviderRegistry);
       }
+      if (contribution.registerChatRender) {
+        contribution.registerChatRender(this.chatRenderRegistry);
+      }
     });
   }
 

packages/ai-native/src/browser/chat/chat.render.registry.ts

@@ -0,0 +1,34 @@
+import { Injectable } from '@opensumi/di';
+import { Disposable } from '@opensumi/ide-core-common';
+
+import {
+  ChatAIRoleRender,
+  ChatThinkingRender,
+  ChatUserRoleRender,
+  ChatWelcomeRender,
+  IChatRenderRegistry,
+} from '../types';
+
+@Injectable()
+export class ChatRenderRegistry extends Disposable implements IChatRenderRegistry {
+  public chatWelcomeRender?: ChatWelcomeRender;
+  public chatAIRoleRender?: ChatAIRoleRender;
+  public chatUserRoleRender?: ChatUserRoleRender;
+  public chatThinkingRender?: ChatThinkingRender;
+
+  registerWelcomeRender(render: ChatWelcomeRender): void {
+    this.chatWelcomeRender = render;
+  }
+
+  registerAIRoleRender(render: ChatAIRoleRender): void {
+    this.chatAIRoleRender = render;
+  }
+
+  registerUserRoleRender(render: ChatUserRoleRender): void {
+    this.chatUserRoleRender = render;
+  }
+
+  registerThinkingRender(render: ChatThinkingRender): void {
+    this.chatThinkingRender = render;
+  }
+}