update companion story with basic state

reaviz · Jul 31, 2024 · 85008fc · 85008fc
1 parent 96c32f9
commit 85008fc
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 9 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/stories/Companion.stories.tsx b/stories/Companion.stories.tsx
@@ -10,16 +10,28 @@ import {
   ChatInput,
   SessionMessagePanel,
   SessionMessagesHeader,
-  SessionMessage
+  SessionMessage,
+  Session
 } from '../src';
-import { fakeSessions, sessionWithSources, sessionsWithFiles } from './examples';
+import {
+  fakeSessions,
+  sessionWithSources,
+  sessionsWithFiles
+} from './examples';
+import { useState } from 'react';
 
 export default {
   title: 'Demos/Companion',
   component: Chat
 } as Meta;
 
 export const Basic = () => {
+  const [activeId, setActiveId] = useState<string | null>(null);
+  const [sessions, setSessions] = useState<Session[]>([
+    ...fakeSessions,
+    ...sessionsWithFiles,
+    ...sessionWithSources
+  ]);
   return (
     <div
       style={{
@@ -31,12 +43,24 @@ export const Basic = () => {
       }}
     >
       <Chat
-        sessions={[
-          ...fakeSessions,
-          ...sessionsWithFiles,
-          ...sessionWithSources
-        ]}
         viewType="companion"
+        sessions={sessions}
+        activeSessionId={activeId}
+        onNewSession={() => {
+          const newId = (sessions.length + 1).toLocaleString();
+          setSessions([
+            ...sessions,
+            {
+              id: newId,
+              title: `New Session #${newId}`,
+              createdAt: new Date(),
+              updatedAt: new Date(),
+              conversations: []
+            }
+          ]);
+          setActiveId(newId);
+        }}
+        onSelectSession={setActiveId}
         onDeleteSession={() => alert('delete!')}
       >
         <SessionsList>