feat: add session context and use it throughout new agents sdk components

Author: 1egoman

packages/react/src/components/RoomProvider.tsx

@@ -31,4 +31,4 @@ export {
   ChatEntry,
   formatChatMessageLinks,
 } from '../components/ChatEntry';
-export * from './RoomProvider';
+export * from './SessionProvider';

packages/react/src/components/index.ts

@@ -16,6 +16,7 @@ export {
 } from './participant-context';
 export {} from './pin-context';
 export { RoomContext, useEnsureRoom, useMaybeRoomContext, useRoomContext } from './room-context';
+export { SessionContext, useEnsureSession, useMaybeSessionContext, useSessionContext } from './session-context';
 export {
   TrackRefContext,
   useEnsureTrackRef,

packages/react/src/context/index.ts

@@ -16,6 +16,7 @@ import { ParticipantAgentAttributes, TrackReference } from '@livekit/components-
 import { useParticipantTracks } from './useParticipantTracks';
 import { useRemoteParticipants } from './useRemoteParticipants';
 import { UseSessionReturn } from './useSession';
+import { useMaybeSessionContext } from '../context';
 
 // FIXME: make this 10 seconds once room dispatch booting info is discoverable
 const DEFAULT_AGENT_CONNECT_TIMEOUT_MILLISECONDS = 20_000;
@@ -265,7 +266,15 @@ type SessionStub = Pick<UseSessionReturn, 'connectionState' | 'room' | 'internal
  * useAgent encapculates all agent state, normalizing some quirks around how LiveKit Agents work.
  * @public
  */
-export function useAgent(session: SessionStub): UseAgentReturn {
+export function useAgent(session?: SessionStub): UseAgentReturn {
+  const sessionFromContext = useMaybeSessionContext();
+  session = session ?? sessionFromContext;
+  if (!session) {
+    throw new Error(
+      'No session provided, make sure you are inside a Session context or pass the session explicitly',
+    );
+  }
+
   const {
     room,
     internal: { agentConnectTimeoutMilliseconds },

packages/react/src/hooks/useAgent.ts

@@ -14,6 +14,7 @@ import { useAgent } from './useAgent';
 import { useTranscriptions } from './useTranscriptions';
 import { useChat } from './useChat';
 import { UseSessionReturn } from './useSession';
+import { useEnsureSession } from '../context';
 
 /** @public */
 export type UseSessionMessagesReturn = {
@@ -42,8 +43,8 @@ export type MessagesCallbacks = {
 };
 
 /** @public */
-export function useSessionMessages(session: UseSessionReturn): UseSessionMessagesReturn {
-  const { room } = session;
+export function useSessionMessages(session?: UseSessionReturn): UseSessionMessagesReturn {
+  const { room } = useEnsureSession(session);
 
   const agent = useAgent(session);