From c642b3fe333c47e310fdf5fc60f672631aacb7bd Mon Sep 17 00:00:00 2001
From: Stephanie Yang <steph@goodcode.us>
Date: Sat, 10 Aug 2024 18:50:48 -0500
Subject: [PATCH] update integration stories

---
 stories/Integration.stories.tsx | 53 ++++-----------------------------
 1 file changed, 5 insertions(+), 48 deletions(-)

diff --git a/stories/Integration.stories.tsx b/stories/Integration.stories.tsx
index 10d0373..92c9279 100644
--- a/stories/Integration.stories.tsx
+++ b/stories/Integration.stories.tsx
@@ -5,15 +5,12 @@ import {
   Chat,
   Session,
   SessionsList,
-  SessionsGroup,
-  SessionListItem,
   NewSessionButton,
   SessionMessages,
   SessionGroups,
   ChatInput,
   SessionMessagePanel,
-  SessionMessagesHeader,
-  SessionMessage
+  SessionMessagesHeader
 } from '../src';
 import { Input } from 'reablocks';
 import { generateText } from 'ai';
@@ -152,32 +149,12 @@ export const _OpenAI = () => {
         >
           <SessionsList>
             <NewSessionButton />
-            <SessionGroups>
-              {groups =>
-                groups.map(({ heading, sessions }) => (
-                  <SessionsGroup heading={heading} key={heading}>
-                    {sessions.map(s => (
-                      <SessionListItem key={s.id} session={s} />
-                    ))}
-                  </SessionsGroup>
-                ))
-              }
-            </SessionGroups>
+            <SessionGroups />
           </SessionsList>
 
           <SessionMessagePanel>
             <SessionMessagesHeader />
-            <SessionMessages>
-              {conversations =>
-                conversations.map((conversation, index) => (
-                  <SessionMessage
-                    key={conversation.id}
-                    conversation={conversation}
-                    isLast={index === conversations.length - 1}
-                  />
-                ))
-              }
-            </SessionMessages>
+            <SessionMessages />
             <ChatInput />
           </SessionMessagePanel>
         </Chat>
@@ -320,32 +297,12 @@ export const VercelAI = () => {
         >
           <SessionsList>
             <NewSessionButton />
-            <SessionGroups>
-              {groups =>
-                groups.map(({ heading, sessions }) => (
-                  <SessionsGroup heading={heading} key={heading}>
-                    {sessions.map(s => (
-                      <SessionListItem key={s.id} session={s} />
-                    ))}
-                  </SessionsGroup>
-                ))
-              }
-            </SessionGroups>
+            <SessionGroups />
           </SessionsList>
 
           <SessionMessagePanel>
             <SessionMessagesHeader />
-            <SessionMessages>
-              {conversations =>
-                conversations.map((conversation, index) => (
-                  <SessionMessage
-                    key={conversation.id}
-                    conversation={conversation}
-                    isLast={index === conversations.length - 1}
-                  />
-                ))
-              }
-            </SessionMessages>
+            <SessionMessages />
             <ChatInput />
           </SessionMessagePanel>
         </Chat>