@absolutejs/ai 0.0.48 → 0.0.50

package/dist/src/ai/client/actions.d.ts

@@ -18,7 +18,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -45,7 +47,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -72,7 +76,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -99,7 +105,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -126,7 +134,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     isPartial?: undefined;
     revisedPrompt?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -153,7 +163,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     model?: undefined;
     sources?: undefined;
     usage?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -180,7 +192,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;
@@ -189,9 +203,11 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     trace?: undefined;
     message?: undefined;
 } | {
+    attachments: import("..").AIAttachment[] | undefined;
     content: string;
     fromMessageId: string;
     messageId: string;
+    mode: "append" | "replace" | undefined;
     newConversationId: string;
     oldConversationId: string;
     type: "branch";
@@ -235,7 +251,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalDurationMs?: undefined;
@@ -265,7 +283,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     model?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     message?: undefined;
@@ -289,7 +309,9 @@ export declare const serverMessageToAction: (message: AIServerMessage) => {
     sources?: undefined;
     usage?: undefined;
     position?: undefined;
+    attachments?: undefined;
     fromMessageId?: undefined;
+    mode?: undefined;
     newConversationId?: undefined;
     oldConversationId?: undefined;
     retrievalStartedAt?: undefined;

package/dist/src/ai/client/createAIStream.d.ts

@@ -3,6 +3,7 @@ export declare const createAIStream: (path: string, conversationId?: string) =>
     branch: (messageId: string, content: string) => void;
     cancel: () => void;
     destroy: () => void;
+    edit: (messageId: string, content: string) => void;
     send: (content: string, attachments?: AIAttachment[]) => void;
     subscribe: (callback: () => void) => () => void;
     readonly error: string | null;

package/dist/src/ai/client/index.d.ts

@@ -2,5 +2,4 @@ export { serverMessageToAction } from "./actions";
 export { createAIConnection } from "./connection";
 export { createAIStream } from "./createAIStream";
 export type { CreateAIStream } from "./createAIStream";
-export { isAIComposerSubmitKey, resizeAIComposerTextarea, type AIComposerSizingOptions, } from "./composer";
 export { ConversationTurnQueueFullError, createConversationTurnQueue, type ConversationTurnQueueItem, type ConversationTurnQueueOptions, type ConversationTurnQueueSnapshot, type ConversationTurnQueueStatus, } from "./turnQueue";

package/dist/src/ai/conversationManager.d.ts

@@ -3,6 +3,7 @@ export declare const createConversationManager: () => {
     abort: (conversationId: string) => void;
     appendMessage: (conversationId: string, message: AIMessage) => void;
     branch: (fromMessageId: string, sourceConversationId: string) => `${string}-${string}-${string}-${string}-${string}` | null;
+    edit: (messageId: string, sourceConversationId: string) => `${string}-${string}-${string}-${string}-${string}` | null;
     get: (conversationId: string) => AIConversation | undefined;
     getAbortController: (conversationId: string) => AbortController;
     getHistory: (conversationId: string) => AIProviderMessage[];

package/dist/src/ai/generateAI.d.ts

@@ -1,4 +1,4 @@
-import type { AIProviderConfig, AIProviderMessage, AIProviderResponseFormat, AIProviderToolChoice, AIProviderToolDefinition, AIToolMap, AIUsage, ReasoningConfig } from "../../types/ai";
+import type { AICitationChunk, AIProviderConfig, AIProviderMessage, AIProviderResponseFormat, AIProviderToolChoice, AIProviderToolDefinition, AIResponseMetadata, AIToolMap, AIUsage, ReasoningConfig } from "../../types/ai";
 export type GenerateAIToolCall = {
     id: string;
     name: string;
@@ -13,6 +13,7 @@ export type GenerateAIOptions = {
     cacheSystemPrompt?: boolean;
     /** Per-call override of the provider `promptCaching` default. See AIProviderStreamParams. */
     promptCaching?: boolean;
+    providerOptions?: Record<string, unknown>;
     maxTokens?: number;
     temperature?: number;
     topP?: number;
@@ -25,6 +26,8 @@ export type GenerateAIOptions = {
     signal?: AbortSignal;
 };
 export type GenerateAIResult = {
+    citations: AICitationChunk[];
+    metadata?: AIResponseMetadata;
     text: string;
     toolCalls: GenerateAIToolCall[];
     usage?: AIUsage;

package/dist/src/ai/index.d.ts

@@ -24,6 +24,8 @@ export { gemini } from "./providers/gemini";
 export { anthropic } from "./providers/anthropic";
 export { ollama } from "./providers/ollama";
 export { openai } from "./providers/openai";
+export { createOpenRouterClient, openrouter, openrouterResponses, openRouterModelMatchesRule, } from "./providers/openrouter";
+export type { OpenRouterClient, OpenRouterClientConfig, OpenRouterConfig, OpenRouterDataCollection, OpenRouterEmbeddingRequest, OpenRouterEmbeddingResponse, OpenRouterHttpRequestOptions, OpenRouterImageRequest, OpenRouterImageResponse, OpenRouterMaxPrice, OpenRouterModel, OpenRouterModelList, OpenRouterPerformancePreference, OpenRouterPlugin, OpenRouterProviderRouting, OpenRouterQuantization, OpenRouterRequestOptions, OpenRouterRerankRequest, OpenRouterRerankResponse, OpenRouterResponseCache, OpenRouterResponsesRequest, OpenRouterServerTool, OpenRouterServiceTier, OpenRouterSpeechRequest, OpenRouterSort, OpenRouterSortStrategy, OpenRouterTranscriptionRequest, OpenRouterVideoRequest, } from "./providers/openrouter";
 export { createOAuth2ClientCredentialsTokenSource } from "./providers/oauth2TokenSource";
 export type { OAuth2ClientCredentialsConfig } from "./providers/oauth2TokenSource";
 export { ProviderError, PROVIDER_STATUS_PAGES, providerStatusPage, } from "./errors/providerError";

package/dist/src/ai/providers/openai.d.ts

@@ -1,8 +1,12 @@
-import type { AIProviderConfig } from "../../../types/ai";
-type OpenAIConfig = {
+import type { AIProviderConfig, AIProviderStreamParams } from "../../../types/ai";
+export type OpenAIConfig = {
     apiKey?: string;
     baseUrl?: string;
+    fetch?: typeof globalThis.fetch;
+    headers?: HeadersInit | ((params: AIProviderStreamParams) => HeadersInit | Promise<HeadersInit>);
+    modelForCapabilities?: (model: string) => string;
+    providerName?: string;
     tokenSource?: () => Promise<string> | string;
+    transformRequestBody?: (body: Record<string, unknown>, params: AIProviderStreamParams) => Record<string, unknown>;
 };
 export declare const openai: (config: OpenAIConfig) => AIProviderConfig;
-export {};

package/dist/src/ai/providers/openaiResponses.d.ts

@@ -1,8 +1,13 @@
-import type { AIProviderConfig } from "../../../types/ai";
-type OpenAIResponsesConfig = {
-    apiKey: string;
+import type { AIProviderConfig, AIProviderStreamParams } from "../../../types/ai";
+export type OpenAIResponsesConfig = {
+    apiKey?: string;
     baseUrl?: string;
+    fetch?: typeof globalThis.fetch;
+    headers?: HeadersInit | ((params: AIProviderStreamParams) => HeadersInit | Promise<HeadersInit>);
     imageModels?: Set<string> | string[];
+    modelForCapabilities?: (model: string) => string;
+    providerName?: string;
+    tokenSource?: () => Promise<string> | string;
+    transformRequestBody?: (body: Record<string, unknown>, params: AIProviderStreamParams) => Record<string, unknown>;
 };
 export declare const openaiResponses: (config: OpenAIResponsesConfig) => AIProviderConfig;
-export {};

package/dist/src/ai/providers/openrouter.d.ts

@@ -0,0 +1,113 @@
+import type { AIProviderConfig } from "../../../types/ai";
+export type OpenRouterDataCollection = "allow" | "deny";
+export type OpenRouterQuantization = "bf16" | "fp4" | "fp6" | "fp8" | "fp16" | "fp32" | "int4" | "int8" | "unknown";
+export type OpenRouterSortStrategy = "latency" | "price" | "throughput";
+export type OpenRouterSort = OpenRouterSortStrategy | {
+    by: OpenRouterSortStrategy;
+    partition?: "model" | "none";
+};
+export type OpenRouterPerformancePreference = number | {
+    p50?: number;
+    p75?: number;
+    p90?: number;
+    p99?: number;
+};
+export type OpenRouterMaxPrice = {
+    /** Maximum completion-token price in USD per million tokens. */
+    completion?: number;
+    /** Maximum price in USD per image. */
+    image?: number;
+    /** Maximum prompt-token price in USD per million tokens. */
+    prompt?: number;
+    /** Maximum price in USD per request. */
+    request?: number;
+};
+export type OpenRouterProviderRouting = {
+    allowFallbacks?: boolean;
+    dataCollection?: OpenRouterDataCollection;
+    enforceDistillableText?: boolean;
+    ignore?: readonly string[];
+    maxPrice?: OpenRouterMaxPrice;
+    only?: readonly string[];
+    order?: readonly string[];
+    preferredMaxLatency?: OpenRouterPerformancePreference;
+    preferredMinThroughput?: OpenRouterPerformancePreference;
+    quantizations?: readonly OpenRouterQuantization[];
+    requireParameters?: boolean;
+    sort?: OpenRouterSort;
+    zdr?: boolean;
+};
+export type OpenRouterServiceTier = "auto" | "default" | "flex" | "priority" | "fast";
+export type OpenRouterResponseCache = {
+    clear?: boolean;
+    enabled: boolean;
+    /** OpenRouter response-cache TTL in seconds (1-86400). */
+    ttlSeconds?: number;
+};
+export type OpenRouterPlugin = {
+    id: string;
+    [option: string]: unknown;
+};
+export type OpenRouterServerTool = {
+    type: "openrouter:web_search" | "openrouter:web_fetch" | "openrouter:datetime" | "openrouter:image_generation" | "openrouter:apply_patch" | "openrouter:shell" | "openrouter:fusion" | "openrouter:advisor" | "openrouter:subagent" | "openrouter:experimental__search_models";
+    parameters?: Record<string, unknown>;
+};
+export type OpenRouterRequestOptions = {
+    /** Future OpenRouter fields. Security-sensitive routing fields are rejected. */
+    extraBody?: Record<string, unknown>;
+    fallbackModels?: readonly string[];
+    includeReasoning?: boolean;
+    maxToolCalls?: number;
+    plugins?: readonly OpenRouterPlugin[];
+    preset?: string;
+    responseCache?: OpenRouterResponseCache;
+    routerMetadata?: boolean;
+    routing?: OpenRouterProviderRouting;
+    serverTools?: readonly OpenRouterServerTool[];
+    serviceTier?: OpenRouterServiceTier;
+    sessionId?: string;
+    stopServerToolsWhen?: readonly Record<string, unknown>[];
+    transforms?: readonly string[];
+    user?: string;
+    verbosity?: "low" | "medium" | "high";
+};
+export type OpenRouterConfig = {
+    /**
+     * Local model policy. Entries are exact model IDs or namespace wildcards such
+     * as `anthropic/*`. Requests outside the list fail before network access.
+     */
+    allowedModels?: readonly string[];
+    /** Preset slugs deliberately approved for use with this provider. */
+    allowedPresets?: readonly string[];
+    /**
+     * Inference-provider policy sent as OpenRouter's `provider.only`. When
+     * `routing.only` is also set, every entry must be allowed by this policy.
+     */
+    allowedProviders?: readonly string[];
+    apiKey?: string;
+    /** Up to two OpenRouter marketplace categories. */
+    appCategories?: readonly string[];
+    /** Public application name used for optional OpenRouter attribution. */
+    appName?: string;
+    /** Public application URL used for optional OpenRouter attribution. */
+    appUrl?: string;
+    baseUrl?: string;
+    fetch?: typeof globalThis.fetch;
+    headers?: HeadersInit | (() => HeadersInit | Promise<HeadersInit>);
+    /** Default OpenRouter request options, overridable per call. */
+    requestOptions?: OpenRouterRequestOptions;
+    routing?: OpenRouterProviderRouting;
+    tokenSource?: () => Promise<string> | string;
+};
+/**
+ * OpenRouter provider for the shared AbsoluteJS AI contract.
+ *
+ * Uses the existing OpenAI-compatible streaming implementation while adding
+ * OpenRouter attribution, typed provider routing, local model policy, provider
+ * allowlists, cost metadata, and OpenRouter-specific error attribution.
+ */
+export declare const openrouter: (config: OpenRouterConfig) => AIProviderConfig;
+/** OpenRouter's stateless OpenAI-compatible Responses API provider skin. */
+export declare const openrouterResponses: (config: OpenRouterConfig) => AIProviderConfig;
+export { createOpenRouterClient, openRouterModelMatchesRule, } from "./openrouterClient";
+export type { OpenRouterClient, OpenRouterClientConfig, OpenRouterEmbeddingRequest, OpenRouterEmbeddingResponse, OpenRouterHttpRequestOptions, OpenRouterImageRequest, OpenRouterImageResponse, OpenRouterModel, OpenRouterModelList, OpenRouterRerankRequest, OpenRouterRerankResponse, OpenRouterResponsesRequest, OpenRouterSpeechRequest, OpenRouterTranscriptionRequest, OpenRouterVideoRequest, } from "./openrouterClient";

package/dist/src/ai/providers/openrouterClient.d.ts

@@ -0,0 +1,147 @@
+export type OpenRouterClientConfig = {
+    allowedModels?: readonly string[];
+    apiKey?: string;
+    baseUrl?: string;
+    fetch?: typeof globalThis.fetch;
+    headers?: HeadersInit | (() => HeadersInit | Promise<HeadersInit>);
+    tokenSource?: () => Promise<string> | string;
+};
+export type OpenRouterModel = {
+    id: string;
+    name?: string;
+    description?: string;
+    context_length?: number;
+    architecture?: Record<string, unknown>;
+    pricing?: Record<string, string>;
+    supported_parameters?: string[] | Record<string, unknown>;
+    [field: string]: unknown;
+};
+export type OpenRouterModelList = {
+    data: OpenRouterModel[];
+};
+export type OpenRouterEmbeddingRequest = {
+    model: string;
+    input: string | string[] | number[] | number[][];
+    dimensions?: number;
+    encoding_format?: "float" | "base64";
+    provider?: Record<string, unknown>;
+    user?: string;
+};
+export type OpenRouterEmbeddingResponse = {
+    data: Array<{
+        embedding: number[] | string;
+        index: number;
+        object: string;
+    }>;
+    model: string;
+    object: string;
+    usage?: {
+        prompt_tokens: number;
+        total_tokens: number;
+        cost?: number;
+    };
+};
+export type OpenRouterRerankRequest = {
+    model: string;
+    query: string;
+    documents: Array<string | {
+        text: string;
+    }>;
+    top_n?: number;
+    return_documents?: boolean;
+};
+export type OpenRouterRerankResponse = {
+    id?: string;
+    model?: string;
+    results: Array<{
+        document?: {
+            text: string;
+        };
+        index: number;
+        relevance_score: number;
+    }>;
+    usage?: Record<string, number>;
+};
+export type OpenRouterImageRequest = {
+    model: string;
+    prompt: string;
+    n?: number;
+    resolution?: string;
+    aspect_ratio?: string;
+    size?: string;
+    quality?: "auto" | "low" | "medium" | "high";
+    output_format?: "png" | "jpeg" | "webp" | "svg";
+    background?: "auto" | "transparent" | "opaque";
+    output_compression?: number;
+    seed?: number;
+    input_references?: Array<Record<string, unknown>>;
+    provider?: Record<string, unknown>;
+};
+export type OpenRouterImageResponse = {
+    created: number;
+    data: Array<{
+        b64_json: string;
+        media_type?: string;
+    }>;
+    usage?: Record<string, number>;
+};
+export type OpenRouterResponsesRequest = Record<string, unknown> & {
+    input: unknown;
+    model: string;
+    stream?: boolean;
+};
+export type OpenRouterSpeechRequest = Record<string, unknown> & {
+    input: string;
+    model: string;
+    voice: string;
+};
+export type OpenRouterTranscriptionRequest = Record<string, unknown> & {
+    model: string;
+};
+export type OpenRouterVideoRequest = Record<string, unknown> & {
+    model: string;
+    prompt: string;
+};
+export type OpenRouterHttpRequestOptions = Omit<RequestInit, "body" | "headers"> & {
+    body?: unknown;
+    headers?: HeadersInit;
+    query?: Record<string, boolean | number | string | undefined>;
+};
+export type OpenRouterClient = ReturnType<typeof createOpenRouterClient>;
+export declare const openRouterModelMatchesRule: (model: string, rule: string) => boolean;
+export declare const createOpenRouterClient: (config: OpenRouterClientConfig) => {
+    cancelBatch: (id: string) => Promise<Record<string, unknown>>;
+    createBatch: (body: Record<string, unknown>) => Promise<Record<string, unknown>>;
+    createEmbedding: (body: OpenRouterEmbeddingRequest) => Promise<OpenRouterEmbeddingResponse>;
+    generateImage: (body: OpenRouterImageRequest) => Promise<OpenRouterImageResponse>;
+    generateVideo: (body: OpenRouterVideoRequest) => Promise<Record<string, unknown>>;
+    getBatch: (id: string) => Promise<Record<string, unknown>>;
+    getCredits: () => Promise<{
+        data: Record<string, number>;
+    }>;
+    getCurrentKey: () => Promise<{
+        data: Record<string, unknown>;
+    }>;
+    getGeneration: (id: string) => Promise<{
+        data: Record<string, unknown>;
+    }>;
+    getModelEndpoints: (model: string) => Promise<Record<string, unknown>>;
+    getVideo: (id: string) => Promise<Record<string, unknown>>;
+    listImageModels: () => Promise<OpenRouterModelList>;
+    listModels: (query?: OpenRouterHttpRequestOptions["query"]) => Promise<OpenRouterModelList>;
+    listPresets: (offset?: number, limit?: number) => Promise<{
+        data: Record<string, unknown>[];
+        total_count: number;
+    }>;
+    listProviders: () => Promise<{
+        data: Record<string, unknown>[];
+    }>;
+    listRerankModels: () => Promise<OpenRouterModelList>;
+    listVideoModels: () => Promise<OpenRouterModelList>;
+    request: <T>(path: string, options?: OpenRouterHttpRequestOptions) => Promise<T>;
+    requestRaw: (path: string, options?: OpenRouterHttpRequestOptions) => Promise<Response>;
+    respond: (body: OpenRouterResponsesRequest) => Promise<Response> | Promise<Record<string, unknown>>;
+    rerank: (body: OpenRouterRerankRequest) => Promise<OpenRouterRerankResponse>;
+    speak: (body: OpenRouterSpeechRequest) => Promise<Response>;
+    transcribe: (body: OpenRouterTranscriptionRequest | FormData) => Promise<Record<string, unknown>>;
+};

package/dist/src/angular/ai/ai-stream.service.d.ts

@@ -5,6 +5,7 @@ export declare class AIStreamService implements OnDestroy {
     connect(path: string, conversationId?: string): {
         branch: (messageId: string, content: string) => void;
         cancel: () => void;
+        edit: (messageId: string, content: string) => void;
         error: import("@angular/core").Signal<string | null>;
         isStreaming: import("@angular/core").Signal<boolean>;
         messages: import("@angular/core").Signal<AIMessage[]>;

package/dist/src/react/ai/useAIStream.d.ts

@@ -2,6 +2,7 @@ import type { AIAttachment } from "../../../types/ai";
 export declare const useAIStream: (path?: string, conversationId?: string) => {
     branch: (messageId: string, content: string) => void;
     cancel: () => void;
+    edit: (messageId: string, content: string) => void;
     error: string | null;
     isStreaming: boolean;
     messages: import("../../ai").AIMessage[];

package/dist/src/svelte/ai/createAIStream.d.ts

@@ -3,6 +3,7 @@ export declare const createAIStream: (path: string, conversationId?: string) =>
     branch: (messageId: string, content: string) => void;
     cancel: () => void;
     destroy: () => void;
+    edit: (messageId: string, content: string) => void;
     send: (content: string, attachments?: AIAttachment[]) => void;
     readonly error: string | null;
     readonly isStreaming: boolean;

package/dist/src/vue/ai/useAIStream.d.ts

@@ -4,6 +4,7 @@ type AIStreamReturn = {
     branch: (messageId: string, content: string) => void;
     cancel: () => void;
     destroy: () => void;
+    edit: (messageId: string, content: string) => void;
     error: Ref<string | null>;
     isStreaming: Ref<boolean>;
     messages: Ref<AIMessage[]>;
@@ -14,6 +15,7 @@ export declare const useAIStream: (path: string, conversationId?: string) => {
     branch: (messageId: string, content: string) => void;
     cancel: () => void;
     destroy: () => void;
+    edit: (messageId: string, content: string) => void;
     error: Ref<string | null, string | null>;
     isStreaming: Ref<boolean, boolean>;
     messages: import("vue").ShallowRef<AIMessage[], AIMessage[]>;

package/dist/svelte/ai/index.js

@@ -131,9 +131,11 @@ var serverMessageToAction = (message) => {
       };
     case "branched":
       return {
+        attachments: message.attachments,
         content: message.content,
         fromMessageId: message.fromMessageId,
         messageId: message.messageId,
+        mode: message.mode,
         newConversationId: message.newConversationId,
         oldConversationId: message.oldConversationId,
         type: "branch"
@@ -186,7 +188,7 @@ var isValidAIServerMessage = (data) => {
     case "turn_started":
       return "conversationId" in data && typeof data.conversationId === "string" && "messageId" in data && typeof data.messageId === "string";
     case "branched":
-      return "content" in data && typeof data.content === "string" && "fromMessageId" in data && typeof data.fromMessageId === "string" && "messageId" in data && typeof data.messageId === "string" && "newConversationId" in data && typeof data.newConversationId === "string" && "oldConversationId" in data && typeof data.oldConversationId === "string";
+      return "content" in data && typeof data.content === "string" && "fromMessageId" in data && typeof data.fromMessageId === "string" && "messageId" in data && typeof data.messageId === "string" && "newConversationId" in data && typeof data.newConversationId === "string" && "oldConversationId" in data && typeof data.oldConversationId === "string" && (!("mode" in data) || data.mode === "append" || data.mode === "replace");
     case "rag_retrieved":
       return "conversationId" in data && "messageId" in data && "sources" in data && Array.isArray(data.sources);
     case "error":
@@ -572,13 +574,15 @@ var handleBranch = (state, action) => {
   if (cutoffIndex < 0) {
     return;
   }
-  const branchedMessages = source.messages.slice(0, cutoffIndex + 1).map((msg) => ({ ...msg, conversationId: action.newConversationId }));
+  const cutoffOffset = action.mode === "replace" ? 0 : 1;
+  const branchedMessages = source.messages.slice(0, cutoffIndex + cutoffOffset).map((msg) => ({ ...msg, conversationId: action.newConversationId }));
   const newConversation = {
     createdAt: Date.now(),
     id: action.newConversationId,
     messages: [
       ...branchedMessages,
       {
+        attachments: action.attachments,
         content: action.content,
         conversationId: action.newConversationId,
         id: action.messageId,
@@ -699,6 +703,16 @@ var createAIStream = (path, conversationId) => {
       });
     }
   };
+  const edit = (messageId, content) => {
+    if (activeConversationId) {
+      connection.send({
+        content,
+        conversationId: activeConversationId,
+        messageId,
+        type: "edit"
+      });
+    }
+  };
   const cancel = () => {
     if (activeConversationId) {
       store.dispatch({ type: "cancel" });
@@ -735,6 +749,7 @@ var createAIStream = (path, conversationId) => {
     branch,
     cancel,
     destroy,
+    edit,
     send,
     get error() {
       return currentError;
@@ -757,5 +772,5 @@ export {
   createAIStream
 };
 
-//# debugId=04909313392ABC9E64756E2164756E21
+//# debugId=1D557221A7953C4F64756E2164756E21
 //# sourceMappingURL=index.js.map