@tanstack/ai-client 0.6.0 → 0.7.2

package/src/types.ts

@@ -36,6 +36,15 @@ export type ToolResultState =
  */
 export type ChatClientState = 'ready' | 'submitted' | 'streaming' | 'error'
 
+/**
+ * Connection lifecycle state for the subscription loop.
+ */
+export type ConnectionStatus =
+  | 'disconnected'
+  | 'connecting'
+  | 'connected'
+  | 'error'
+
 /**
  * Multimodal content input for sending messages with rich media.
  * Allows sending text, images, audio, video, and documents to the LLM.
@@ -178,8 +187,9 @@ export interface ChatClientOptions<
   TTools extends ReadonlyArray<AnyClientTool> = any,
 > {
   /**
-   * Connection adapter for streaming
-   * Use fetchServerSentEvents(), fetchHttpStream(), or stream() to create adapters
+   * Connection adapter for streaming.
+   * Supports mutually exclusive modes: request-response via `connect()`, or
+   * subscribe/send mode via `subscribe()` + `send()`.
    */
   connection: ConnectionAdapter
 
@@ -239,6 +249,25 @@ export interface ChatClientOptions<
    */
   onStatusChange?: (status: ChatClientState) => void
 
+  /**
+   * Callback when subscription lifecycle changes.
+   * This is independent from request lifecycle (`isLoading`, `status`).
+   */
+  onSubscriptionChange?: (isSubscribed: boolean) => void
+
+  /**
+   * Callback when connection lifecycle changes.
+   */
+  onConnectionStatusChange?: (status: ConnectionStatus) => void
+
+  /**
+   * Callback when session generation activity changes.
+   * Derived from stream run events (RUN_STARTED / RUN_FINISHED / RUN_ERROR).
+   * Unlike `onLoadingChange` (request-local), this reflects shared generation
+   * activity visible to all subscribers (e.g. across tabs/devices).
+   */
+  onSessionGeneratingChange?: (isGenerating: boolean) => void
+
   /**
    * Callback when a custom event is received from a server-side tool.
    * Custom events are emitted by tools using `context.emitCustomEvent()` during execution.

package/src/video-generation-client.ts

@@ -1,7 +1,7 @@
 import { GENERATION_EVENTS } from './generation-types'
 import { parseSSEResponse } from './sse-parser'
 import type { StreamChunk } from '@tanstack/ai'
-import type { ConnectionAdapter } from './connection-adapters'
+import type { ConnectConnectionAdapter } from './connection-adapters'
 import type {
   GenerationClientState,
   GenerationFetcher,
@@ -36,14 +36,14 @@ interface VideoCallbacks<TOutput> {
  * until completion. This client handles the full lifecycle.
  *
  * Supports two transport modes:
- * - **ConnectionAdapter** — Server handles the polling loop internally and
+ * - **ConnectConnectionAdapter** — Server handles the polling loop internally and
  *   streams status updates via CUSTOM events.
  * - **Fetcher** — Direct async function that returns a completed
  *   `VideoGenerateResult`.
  *
  * @example
  * ```typescript
- * // With ConnectionAdapter (server-driven polling)
+ * // With streaming connection adapter (server-driven polling)
  * const client = new VideoGenerationClient({
  *   connection: fetchServerSentEvents('/api/generate/video'),
  *   onResultChange: setResult,
@@ -65,7 +65,7 @@ interface VideoCallbacks<TOutput> {
  * ```
  */
 export class VideoGenerationClient<TOutput = VideoGenerateResult> {
-  private connection: ConnectionAdapter | undefined
+  private connection: ConnectConnectionAdapter | undefined
   private fetcher:
     | GenerationFetcher<VideoGenerateInput, VideoGenerateResult>
     | undefined
@@ -83,7 +83,7 @@ export class VideoGenerationClient<TOutput = VideoGenerateResult> {
   constructor(
     options: VideoGenerationClientOptions<TOutput> &
       (
-        | { connection: ConnectionAdapter; fetcher?: never }
+        | { connection: ConnectConnectionAdapter; fetcher?: never }
         | {
             fetcher: GenerationFetcher<VideoGenerateInput, VideoGenerateResult>
             connection?: never
@@ -174,7 +174,7 @@ export class VideoGenerationClient<TOutput = VideoGenerateResult> {
   }
 
   /**
-   * Process a stream of AG-UI events from the ConnectionAdapter.
+   * Process a stream of AG-UI events from the streaming connection adapter.
    * The server handles the polling loop and streams status updates.
    */
   private async processStream(