teodor-new-chat-ui 4.3.92 → 4.3.94

package/dist/context/services/useChatStreamingService.d.ts

@@ -1,8 +1,21 @@
-import type { ChatRequest, StreamEvent } from '@/types/api';
+import type { ChatRequest, HistoryPayload, StreamEvent } from '@/types/api';
 import { type ReducerStreamEvent } from '../reducers/MessageReducer';
 export type ThreadInfoEvent = Extract<StreamEvent, {
     type: 'thread_info';
 }>;
+type HeartbeatEvent = Extract<StreamEvent, {
+    type: 'heartbeat';
+}>;
+export interface StreamClosedInfo {
+    reason: 'complete' | 'aborted' | 'error';
+    lastSeq: number;
+    lastCheckpointId: string | null;
+    lastCheckpointNs: string | null;
+    lastStateHistory: HistoryPayload | null;
+    threadId: string | null;
+    manualStop: boolean;
+    error?: string | null;
+}
 interface UseChatStreamingServiceOptions {
     baseUrl: string;
     onThreadInfo: (event: ThreadInfoEvent) => void;
@@ -12,6 +25,9 @@ interface UseChatStreamingServiceOptions {
         from: number;
         to: number;
     }) => void;
+    onStreamClosed?: (info: StreamClosedInfo) => void;
+    onConnectionError?: (error: string) => void;
+    onHeartbeat?: (event: HeartbeatEvent) => void;
 }
 interface StartStreamCallbacks {
     onOpen?: (info: {
@@ -23,6 +39,9 @@ interface StartStreamCallbacks {
         from: number;
         to: number;
     }) => void;
+    onClosed?: (info: StreamClosedInfo) => void;
+    onConnectionError?: (error: string) => void;
+    onHeartbeat?: (event: HeartbeatEvent) => void;
 }
 /**
  * Thin integration layer that adapts the public `useChatStream` hook to the
@@ -30,11 +49,12 @@ interface StartStreamCallbacks {
  * here means we continue to rely on the well-tested streaming hook for all SSE
  * behaviour while the provider focuses purely on state updates.
  */
-export declare function useChatStreamingService({ baseUrl, onThreadInfo, onMessageEvent, onUnhandledEvent, onSequenceGap, }: UseChatStreamingServiceOptions): {
+export declare function useChatStreamingService({ baseUrl, onThreadInfo, onMessageEvent, onUnhandledEvent, onSequenceGap, onStreamClosed, onConnectionError, onHeartbeat, }: UseChatStreamingServiceOptions): {
     startStream: (request: ChatRequest, callbacks?: StartStreamCallbacks) => void;
     stop: () => void;
     isStreaming: boolean;
     error: string;
     setAuthToken: (t: string | null) => void;
+    replayBufferedEvents: (threadId?: string | null) => void;
 };
 export {};

package/dist/hooks/use-chat.d.ts

@@ -1,4 +1,4 @@
-import { ChatRequest, StreamEvent } from "@/types/api";
+import { ChatRequest, HistoryPayload, StreamEvent } from "@/types/api";
 export type UseChatStreamOptions = {
     baseUrl?: string;
     streamPath?: string;
@@ -24,6 +24,18 @@ export type UseChatStream = {
             from: number;
             to: number;
         }) => void;
+        onClosed?: (info: {
+            reason: "complete" | "aborted" | "error";
+            lastSeq: number;
+            lastCheckpointId: string | null;
+            lastCheckpointNs: string | null;
+            lastStateHistory: HistoryPayload | null;
+            error?: string | null;
+        }) => void;
+        onConnectionError?: (message: string) => void;
+        onHeartbeat?: (event: Extract<StreamEvent, {
+            type: "heartbeat";
+        }>) => void;
     }) => {
         close: () => void;
     };