fix button forward ref + ChatProvider level

Author: iamfaran

client/packages/lowcoder/src/comps/comps/chatComp/chatComp.tsx

@@ -12,13 +12,15 @@ import QuerySelectControl from "comps/controls/querySelectControl";
 import { eventHandlerControl, EventConfigType } from "comps/controls/eventHandlerControl";
 import { AutoHeightControl } from "comps/controls/autoHeightControl";
 import { ChatContainer } from "./components/ChatContainer";
+import { ChatProvider } from "./components/context/ChatContext";
 import { ChatPropertyView } from "./chatPropertyView";
 import { createChatStorage } from "./utils/storageFactory";
 import { QueryHandler, createMessageHandler } from "./handlers/messageHandlers";
 import { useMemo, useRef, useEffect } from "react";  
 import { changeChildAction } from "lowcoder-core";
 import { ChatMessage } from "./types/chatTypes";
 import { trans } from "i18n";
+import { TooltipProvider } from "@radix-ui/react-tooltip";
 import { styleControl } from "comps/controls/styleControl";
 import {
   ChatStyle,
@@ -302,17 +304,20 @@ const ChatTmpComp = new UICompBuilder(
     };
 
     return (
-      <ChatContainer
-        storage={storage}
-        messageHandler={messageHandler}
-        placeholder={props.placeholder}
-        autoHeight={props.autoHeight}
-        sidebarWidth={props.leftPanelWidth}
-        onMessageUpdate={handleMessageUpdate}
-        onConversationUpdate={handleConversationUpdate}
-        onEvent={props.onEvent}
-        {...styles}
-      />
+      <TooltipProvider>
+        <ChatProvider storage={storage}>
+          <ChatContainer
+            messageHandler={messageHandler}
+            placeholder={props.placeholder}
+            autoHeight={props.autoHeight}
+            sidebarWidth={props.leftPanelWidth}
+            onMessageUpdate={handleMessageUpdate}
+            onConversationUpdate={handleConversationUpdate}
+            onEvent={props.onEvent}
+            {...styles}
+          />
+        </ChatProvider>
+      </TooltipProvider>
     );
   }
 )

client/packages/lowcoder/src/comps/comps/chatComp/components/ChatContainer.tsx

@@ -14,7 +14,6 @@ import {
 import { Thread } from "./assistant-ui/thread";
 import { ThreadList } from "./assistant-ui/thread-list";
 import { 
-  ChatProvider,
   useChatContext, 
   RegularThreadData, 
   ArchivedThreadData 
@@ -23,7 +22,6 @@ import { MessageHandler, ChatMessage, ChatCoreProps } from "../types/chatTypes";
 import styled from "styled-components";
 import { trans } from "i18n";
 import { universalAttachmentAdapter } from "../utils/attachmentAdapter";
-import { TooltipProvider } from "@radix-ui/react-tooltip";
 
 // ============================================================================
 // STYLED CONTAINER - FOLLOWING LOWCODER PATTERNS
@@ -122,7 +120,7 @@ const StyledChatContainer = styled.div<ChatCoreProps>`
 `;
 
 // ============================================================================
-// CHAT CONTAINER - DIRECT RENDERING LIKE ButtonView
+// CHAT CONTAINER - USES CONTEXT FROM CHATPROVIDER
 // ============================================================================
 
 const generateId = () => Math.random().toString(36).substr(2, 9);
@@ -334,15 +332,7 @@ function ChatContainerView(props: ChatCoreProps) {
 }
 
 // ============================================================================
-// EXPORT - WITH PROVIDERS (LIKE BUTTON/INPUT PATTERN)
+// EXPORT - SIMPLIFIED (PROVIDERS MOVED UP ONE LEVEL)
 // ============================================================================
 
-export function ChatContainer(props: ChatCoreProps) {
-  return (
-    <TooltipProvider>
-      <ChatProvider storage={props.storage}>
-        <ChatContainerView {...props} />
-      </ChatProvider>
-    </TooltipProvider>
-  );
-}
+export const ChatContainer = ChatContainerView;

client/packages/lowcoder/src/comps/comps/chatComp/components/ui/button.tsx

@@ -21,25 +21,25 @@ const buttonVariants = cva("aui-button", {
   },
 });
 
-function Button({
-  className,
-  variant,
-  size,
-  asChild = false,
-  ...props
-}: React.ComponentProps<"button"> &
-  VariantProps<typeof buttonVariants> & {
-    asChild?: boolean;
-  }) {
+const Button = React.forwardRef<
+  HTMLButtonElement,
+  React.ComponentProps<"button"> &
+    VariantProps<typeof buttonVariants> & {
+      asChild?: boolean;
+    }
+>(({ className, variant, size, asChild = false, ...props }, ref) => {
   const Comp = asChild ? Slot : "button";
 
   return (
     <Comp
+      ref={ref}
       data-slot="button"
       className={cn(buttonVariants({ variant, size, className }))}
       {...props}
     />
   );
-}
+});
+
+Button.displayName = "Button";
 
 export { Button, buttonVariants };

client/packages/lowcoder/src/comps/comps/chatComp/types/chatTypes.ts

@@ -73,7 +73,6 @@ export interface ChatMessage {
 
 // Main Chat Component Props (with full styling support)
 export interface ChatCoreProps {
-  storage: ChatStorage;
   messageHandler: MessageHandler;
   placeholder?: string;
   autoHeight?: boolean;