@yourgpt/llm-sdk 2.1.4-alpha.0 → 2.1.4-alpha.2

package/dist/adapters/index.d.mts

@@ -1,5 +1,6 @@
-import { L as LLMAdapter, W as WebSearchConfig, Q as ChatCompletionRequest, g as StreamEvent, ap as CompletionResult, a3 as OllamaModelOptions } from '../types-COAOEe_y.mjs';
-export { V as AdapterFactory, aB as AnthropicContentBlock, aC as OpenAIContentBlock, az as attachmentToAnthropicDocument, ay as attachmentToAnthropicImage, aA as attachmentToOpenAIImage, aq as formatMessages, as as formatMessagesForAnthropic, at as formatMessagesForOpenAI, ar as formatTools, aw as hasImageAttachments, ax as hasMediaAttachments, au as messageToAnthropicContent, av as messageToOpenAIContent } from '../types-COAOEe_y.mjs';
+import { L as LLMAdapter, W as WebSearchConfig, C as ChatCompletionRequest, S as StreamEvent, N as CompletionResult, k as OllamaModelOptions } from '../types-DoZX2k3v.mjs';
+export { f as AdapterFactory, a4 as AnthropicContentBlock, a5 as OpenAIContentBlock, a2 as attachmentToAnthropicDocument, a1 as attachmentToAnthropicImage, a3 as attachmentToOpenAIImage, Q as formatMessages, V as formatMessagesForAnthropic, Y as formatMessagesForOpenAI, R as formatTools, $ as hasImageAttachments, a0 as hasMediaAttachments, Z as messageToAnthropicContent, _ as messageToOpenAIContent } from '../types-DoZX2k3v.mjs';
+import '../types-CR8mi9I0.mjs';
 import 'zod';
 
 /**

package/dist/adapters/index.d.ts

@@ -1,5 +1,6 @@
-import { L as LLMAdapter, W as WebSearchConfig, Q as ChatCompletionRequest, g as StreamEvent, ap as CompletionResult, a3 as OllamaModelOptions } from '../types-COAOEe_y.js';
-export { V as AdapterFactory, aB as AnthropicContentBlock, aC as OpenAIContentBlock, az as attachmentToAnthropicDocument, ay as attachmentToAnthropicImage, aA as attachmentToOpenAIImage, aq as formatMessages, as as formatMessagesForAnthropic, at as formatMessagesForOpenAI, ar as formatTools, aw as hasImageAttachments, ax as hasMediaAttachments, au as messageToAnthropicContent, av as messageToOpenAIContent } from '../types-COAOEe_y.js';
+import { L as LLMAdapter, W as WebSearchConfig, C as ChatCompletionRequest, S as StreamEvent, N as CompletionResult, k as OllamaModelOptions } from '../types-DtPQaoJi.js';
+export { f as AdapterFactory, a4 as AnthropicContentBlock, a5 as OpenAIContentBlock, a2 as attachmentToAnthropicDocument, a1 as attachmentToAnthropicImage, a3 as attachmentToOpenAIImage, Q as formatMessages, V as formatMessagesForAnthropic, Y as formatMessagesForOpenAI, R as formatTools, $ as hasImageAttachments, a0 as hasMediaAttachments, Z as messageToAnthropicContent, _ as messageToOpenAIContent } from '../types-DtPQaoJi.js';
+import '../types-CR8mi9I0.js';
 import 'zod';
 
 /**

package/dist/index.d.mts

@@ -1,6 +1,8 @@
-import { G as GenerateTextParams, a as GenerateTextResult, S as StreamTextParams, b as StreamTextResult, T as ToolContext, c as Tool, A as AIProvider, d as ActionDefinition, e as ToolDefinition, f as ToolProfile, K as KnowledgeBaseConfig, W as WebSearchConfig, L as LLMAdapter, D as DoneEventMessage, g as StreamEvent, h as ToolCallInfo, i as TokenUsageRaw, P as ProviderToolRuntimeOptions, M as Message, j as ToolResponse } from './types-COAOEe_y.mjs';
-export { V as AdapterFactory, _ as AnthropicProviderConfig, an as AnthropicProviderToolOptions, a5 as AnthropicTool, a7 as AnthropicToolResult, ak as AnthropicToolSelectionHints, a6 as AnthropicToolUse, p as AssistantMessage, a1 as AzureProviderConfig, Y as BaseProviderConfig, Q as ChatCompletionRequest, ao as Citation, C as CoreMessage, O as DEFAULT_CAPABILITIES, m as DoGenerateParams, n as DoGenerateResult, H as ErrorChunk, F as FilePart, E as FinishChunk, N as FinishReason, ac as GeminiFunctionCall, ab as GeminiFunctionDeclaration, ad as GeminiFunctionResponse, v as GenerateStep, $ as GoogleProviderConfig, I as ImagePart, ae as LLMConfig, k as LanguageModel, l as ModelCapabilities, a3 as OllamaModelOptions, a2 as OllamaProviderConfig, Z as OpenAIProviderConfig, am as OpenAIProviderToolOptions, a8 as OpenAITool, a9 as OpenAIToolCall, aa as OpenAIToolResult, aj as OpenAIToolSelectionHints, X as ProviderCapabilities, a4 as ProviderFormatter, R as ResponseOptions, x as StreamChunk, w as StreamPart, o as SystemMessage, y as TextDeltaChunk, s as TextPart, J as TokenUsage, t as ToolCall, z as ToolCallChunk, ai as ToolExecution, af as ToolLocation, q as ToolMessage, al as ToolNativeProviderHints, u as ToolResult, B as ToolResultChunk, ag as UnifiedToolCall, ah as UnifiedToolResult, r as UserContentPart, U as UserMessage, a0 as XAIProviderConfig } from './types-COAOEe_y.mjs';
+import { G as GenerateTextParams, a as GenerateTextResult, S as StreamTextParams, b as StreamTextResult, T as ToolContext, c as Tool, d as StorageAdapter, e as StorageMessage } from './types-CR8mi9I0.mjs';
+export { A as AssistantMessage, C as CoreMessage, w as DEFAULT_CAPABILITIES, D as DoGenerateParams, f as DoGenerateResult, E as ErrorChunk, F as FilePart, s as FinishChunk, u as FinishReason, m as GenerateStep, I as ImagePart, L as LanguageModel, M as ModelCapabilities, R as ResponseOptions, v as StorageFile, o as StreamChunk, n as StreamPart, g as SystemMessage, p as TextDeltaChunk, j as TextPart, t as TokenUsage, k as ToolCall, q as ToolCallChunk, h as ToolMessage, l as ToolResult, r as ToolResultChunk, i as UserContentPart, U as UserMessage } from './types-CR8mi9I0.mjs';
 import { z } from 'zod';
+import { A as AIProvider, a as ActionDefinition, T as ToolDefinition, b as ToolProfile, K as KnowledgeBaseConfig, W as WebSearchConfig, L as LLMAdapter, D as DoneEventMessage, S as StreamEvent, c as ToolCallInfo, d as TokenUsageRaw, P as ProviderToolRuntimeOptions, M as Message, e as ToolResponse } from './types-DoZX2k3v.mjs';
+export { f as AdapterFactory, h as AnthropicProviderConfig, I as AnthropicProviderToolOptions, m as AnthropicTool, o as AnthropicToolResult, E as AnthropicToolSelectionHints, n as AnthropicToolUse, i as AzureProviderConfig, B as BaseProviderConfig, C as ChatCompletionRequest, J as Citation, t as GeminiFunctionCall, s as GeminiFunctionDeclaration, u as GeminiFunctionResponse, G as GoogleProviderConfig, v as LLMConfig, k as OllamaModelOptions, j as OllamaProviderConfig, O as OpenAIProviderConfig, H as OpenAIProviderToolOptions, p as OpenAITool, q as OpenAIToolCall, r as OpenAIToolResult, z as OpenAIToolSelectionHints, g as ProviderCapabilities, l as ProviderFormatter, y as ToolExecution, w as ToolLocation, F as ToolNativeProviderHints, U as UnifiedToolCall, x as UnifiedToolResult, X as XAIProviderConfig } from './types-DoZX2k3v.mjs';
 import * as hono from 'hono';
 import { Hono } from 'hono';
 
@@ -257,6 +259,17 @@ interface RuntimeConfigWithAdapter {
      * Set to true for defaults, or pass WebSearchConfig for customization.
      */
     webSearch?: boolean | WebSearchConfig;
+    /**
+     * Storage adapter for automatic session creation and message persistence.
+     * When provided, runtime.chat() and runtime.stream() auto-save messages.
+     *
+     * @example
+     * ```ts
+     * import { createYourGPT } from '@yourgpt/llm-sdk/yourgpt'
+     * storage: createYourGPT({ apiKey, widgetUid })
+     * ```
+     */
+    storage?: StorageAdapter;
 }
 /**
  * Runtime configuration with AIProvider
@@ -303,6 +316,17 @@ interface RuntimeConfigWithProvider {
      * Set to true for defaults, or pass WebSearchConfig for customization.
      */
     webSearch?: boolean | WebSearchConfig;
+    /**
+     * Storage adapter for automatic session creation and message persistence.
+     * When provided, runtime.chat() and runtime.stream() auto-save messages.
+     *
+     * @example
+     * ```ts
+     * import { createYourGPT } from '@yourgpt/llm-sdk/yourgpt'
+     * storage: createYourGPT({ apiKey, widgetUid })
+     * ```
+     */
+    storage?: StorageAdapter;
 }
 /**
  * Runtime configuration - provide either a provider instance or a custom adapter
@@ -537,6 +561,8 @@ interface OnFinishResult {
         completionTokens: number;
         totalTokens: number;
     };
+    /** Session ID — present when storage adapter created/resolved a session */
+    threadId?: string;
 }
 /**
  * Options for StreamResult constructor
@@ -562,6 +588,8 @@ interface CollectedResult {
     requiresAction: boolean;
     /** Token usage for billing/tracking */
     usage?: TokenUsageRaw;
+    /** Session ID — present when storage adapter created/resolved a session */
+    threadId?: string;
     /** Raw events (for debugging) */
     events: StreamEvent[];
 }
@@ -870,6 +898,7 @@ type ToolSearchState = {
 declare class Runtime {
     private adapter;
     private config;
+    private storage;
     private actions;
     private tools;
     constructor(config: RuntimeConfig);
@@ -1034,6 +1063,7 @@ declare class Runtime {
          */
         onFinish?: (result: {
             messages: DoneEventMessage[];
+            threadId?: string;
             usage?: {
                 promptTokens: number;
                 completionTokens: number;
@@ -1464,6 +1494,32 @@ interface AgentLoopOptions {
  */
 declare function runAgentLoop(options: AgentLoopOptions): AsyncGenerator<StreamEvent>;
 
+/**
+ * Storage Helpers
+ *
+ * Extract input/output messages from request/response for storage adapters.
+ * Used internally by Runtime when `storage` is configured.
+ */
+
+/**
+ * Extract new INPUT messages from a request's message array.
+ *
+ * The SDK sends the full conversation history on each request. This function
+ * picks only the NEW messages the user just added:
+ * - New user turn (last meaningful msg = "user"): returns [userMsg]
+ * - Tool continuation (last meaningful msg = "tool"): returns tool results after last assistant
+ * - Otherwise: returns []
+ *
+ * Skips empty assistant placeholders the SDK pushes before sending.
+ */
+declare function extractInputMessages(reqMessages: unknown[]): StorageMessage[];
+/**
+ * Map LLM output messages (DoneEventMessage format) to StorageMessage format.
+ *
+ * Converts from snake_case API format to camelCase storage format.
+ */
+declare function mapOutputMessages(resultMessages: DoneEventMessage[]): StorageMessage[];
+
 /**
  * Generate a message ID
  */
@@ -1477,4 +1533,4 @@ declare function generateThreadId(): string;
  */
 declare function generateToolCallId(): string;
 
-export { AIProvider, ActionDefinition, type ActionRequest, type AgentLoopOptions, type ChatRequest, type CollectedResult, type CopilotChatResponse, DEFAULT_MAX_ITERATIONS, DoneEventMessage, GenerateResult, type GenerateResultData, GenerateTextParams, GenerateTextResult, LLMAdapter, Message, ProviderToolRuntimeOptions, type RequestContext, Runtime, type RuntimeConfig, StreamEvent, StreamResult, type StreamResultOptions, StreamTextParams, StreamTextResult, TokenUsageRaw, Tool, ToolCallInfo, ToolContext, ToolDefinition, ToolProfile, ToolResponse, type ToolSearchMatch, WebSearchConfig, buildProviderToolOptions, createEventStream, createExpressHandler, createExpressMiddleware, createHonoApp, createNextHandler, createNodeHandler, createRuntime, createSSEHeaders, createSSEResponse, createStreamResult, createTextStreamHeaders, createTextStreamResponse, formatSSEData, formatToolsForAnthropic, formatToolsForGoogle, formatToolsForOpenAI, generateMessageId, generateText, generateThreadId, generateToolCallId, pipeSSEToResponse, pipeTextToResponse, runAgentLoop, searchTools, selectTools, shouldExposeToolSearch, streamText, tool };
+export { AIProvider, ActionDefinition, type ActionRequest, type AgentLoopOptions, type ChatRequest, type CollectedResult, type CopilotChatResponse, DEFAULT_MAX_ITERATIONS, DoneEventMessage, GenerateResult, type GenerateResultData, GenerateTextParams, GenerateTextResult, LLMAdapter, Message, ProviderToolRuntimeOptions, type RequestContext, Runtime, type RuntimeConfig, StorageAdapter, StorageMessage, StreamEvent, StreamResult, type StreamResultOptions, StreamTextParams, StreamTextResult, TokenUsageRaw, Tool, ToolCallInfo, ToolContext, ToolDefinition, ToolProfile, ToolResponse, type ToolSearchMatch, WebSearchConfig, buildProviderToolOptions, createEventStream, createExpressHandler, createExpressMiddleware, createHonoApp, createNextHandler, createNodeHandler, createRuntime, createSSEHeaders, createSSEResponse, createStreamResult, createTextStreamHeaders, createTextStreamResponse, extractInputMessages, formatSSEData, formatToolsForAnthropic, formatToolsForGoogle, formatToolsForOpenAI, generateMessageId, generateText, generateThreadId, generateToolCallId, mapOutputMessages, pipeSSEToResponse, pipeTextToResponse, runAgentLoop, searchTools, selectTools, shouldExposeToolSearch, streamText, tool };

package/dist/index.d.ts

@@ -1,6 +1,8 @@
-import { G as GenerateTextParams, a as GenerateTextResult, S as StreamTextParams, b as StreamTextResult, T as ToolContext, c as Tool, A as AIProvider, d as ActionDefinition, e as ToolDefinition, f as ToolProfile, K as KnowledgeBaseConfig, W as WebSearchConfig, L as LLMAdapter, D as DoneEventMessage, g as StreamEvent, h as ToolCallInfo, i as TokenUsageRaw, P as ProviderToolRuntimeOptions, M as Message, j as ToolResponse } from './types-COAOEe_y.js';
-export { V as AdapterFactory, _ as AnthropicProviderConfig, an as AnthropicProviderToolOptions, a5 as AnthropicTool, a7 as AnthropicToolResult, ak as AnthropicToolSelectionHints, a6 as AnthropicToolUse, p as AssistantMessage, a1 as AzureProviderConfig, Y as BaseProviderConfig, Q as ChatCompletionRequest, ao as Citation, C as CoreMessage, O as DEFAULT_CAPABILITIES, m as DoGenerateParams, n as DoGenerateResult, H as ErrorChunk, F as FilePart, E as FinishChunk, N as FinishReason, ac as GeminiFunctionCall, ab as GeminiFunctionDeclaration, ad as GeminiFunctionResponse, v as GenerateStep, $ as GoogleProviderConfig, I as ImagePart, ae as LLMConfig, k as LanguageModel, l as ModelCapabilities, a3 as OllamaModelOptions, a2 as OllamaProviderConfig, Z as OpenAIProviderConfig, am as OpenAIProviderToolOptions, a8 as OpenAITool, a9 as OpenAIToolCall, aa as OpenAIToolResult, aj as OpenAIToolSelectionHints, X as ProviderCapabilities, a4 as ProviderFormatter, R as ResponseOptions, x as StreamChunk, w as StreamPart, o as SystemMessage, y as TextDeltaChunk, s as TextPart, J as TokenUsage, t as ToolCall, z as ToolCallChunk, ai as ToolExecution, af as ToolLocation, q as ToolMessage, al as ToolNativeProviderHints, u as ToolResult, B as ToolResultChunk, ag as UnifiedToolCall, ah as UnifiedToolResult, r as UserContentPart, U as UserMessage, a0 as XAIProviderConfig } from './types-COAOEe_y.js';
+import { G as GenerateTextParams, a as GenerateTextResult, S as StreamTextParams, b as StreamTextResult, T as ToolContext, c as Tool, d as StorageAdapter, e as StorageMessage } from './types-CR8mi9I0.js';
+export { A as AssistantMessage, C as CoreMessage, w as DEFAULT_CAPABILITIES, D as DoGenerateParams, f as DoGenerateResult, E as ErrorChunk, F as FilePart, s as FinishChunk, u as FinishReason, m as GenerateStep, I as ImagePart, L as LanguageModel, M as ModelCapabilities, R as ResponseOptions, v as StorageFile, o as StreamChunk, n as StreamPart, g as SystemMessage, p as TextDeltaChunk, j as TextPart, t as TokenUsage, k as ToolCall, q as ToolCallChunk, h as ToolMessage, l as ToolResult, r as ToolResultChunk, i as UserContentPart, U as UserMessage } from './types-CR8mi9I0.js';
 import { z } from 'zod';
+import { A as AIProvider, a as ActionDefinition, T as ToolDefinition, b as ToolProfile, K as KnowledgeBaseConfig, W as WebSearchConfig, L as LLMAdapter, D as DoneEventMessage, S as StreamEvent, c as ToolCallInfo, d as TokenUsageRaw, P as ProviderToolRuntimeOptions, M as Message, e as ToolResponse } from './types-DtPQaoJi.js';
+export { f as AdapterFactory, h as AnthropicProviderConfig, I as AnthropicProviderToolOptions, m as AnthropicTool, o as AnthropicToolResult, E as AnthropicToolSelectionHints, n as AnthropicToolUse, i as AzureProviderConfig, B as BaseProviderConfig, C as ChatCompletionRequest, J as Citation, t as GeminiFunctionCall, s as GeminiFunctionDeclaration, u as GeminiFunctionResponse, G as GoogleProviderConfig, v as LLMConfig, k as OllamaModelOptions, j as OllamaProviderConfig, O as OpenAIProviderConfig, H as OpenAIProviderToolOptions, p as OpenAITool, q as OpenAIToolCall, r as OpenAIToolResult, z as OpenAIToolSelectionHints, g as ProviderCapabilities, l as ProviderFormatter, y as ToolExecution, w as ToolLocation, F as ToolNativeProviderHints, U as UnifiedToolCall, x as UnifiedToolResult, X as XAIProviderConfig } from './types-DtPQaoJi.js';
 import * as hono from 'hono';
 import { Hono } from 'hono';
 
@@ -257,6 +259,17 @@ interface RuntimeConfigWithAdapter {
      * Set to true for defaults, or pass WebSearchConfig for customization.
      */
     webSearch?: boolean | WebSearchConfig;
+    /**
+     * Storage adapter for automatic session creation and message persistence.
+     * When provided, runtime.chat() and runtime.stream() auto-save messages.
+     *
+     * @example
+     * ```ts
+     * import { createYourGPT } from '@yourgpt/llm-sdk/yourgpt'
+     * storage: createYourGPT({ apiKey, widgetUid })
+     * ```
+     */
+    storage?: StorageAdapter;
 }
 /**
  * Runtime configuration with AIProvider
@@ -303,6 +316,17 @@ interface RuntimeConfigWithProvider {
      * Set to true for defaults, or pass WebSearchConfig for customization.
      */
     webSearch?: boolean | WebSearchConfig;
+    /**
+     * Storage adapter for automatic session creation and message persistence.
+     * When provided, runtime.chat() and runtime.stream() auto-save messages.
+     *
+     * @example
+     * ```ts
+     * import { createYourGPT } from '@yourgpt/llm-sdk/yourgpt'
+     * storage: createYourGPT({ apiKey, widgetUid })
+     * ```
+     */
+    storage?: StorageAdapter;
 }
 /**
  * Runtime configuration - provide either a provider instance or a custom adapter
@@ -537,6 +561,8 @@ interface OnFinishResult {
         completionTokens: number;
         totalTokens: number;
     };
+    /** Session ID — present when storage adapter created/resolved a session */
+    threadId?: string;
 }
 /**
  * Options for StreamResult constructor
@@ -562,6 +588,8 @@ interface CollectedResult {
     requiresAction: boolean;
     /** Token usage for billing/tracking */
     usage?: TokenUsageRaw;
+    /** Session ID — present when storage adapter created/resolved a session */
+    threadId?: string;
     /** Raw events (for debugging) */
     events: StreamEvent[];
 }
@@ -870,6 +898,7 @@ type ToolSearchState = {
 declare class Runtime {
     private adapter;
     private config;
+    private storage;
     private actions;
     private tools;
     constructor(config: RuntimeConfig);
@@ -1034,6 +1063,7 @@ declare class Runtime {
          */
         onFinish?: (result: {
             messages: DoneEventMessage[];
+            threadId?: string;
             usage?: {
                 promptTokens: number;
                 completionTokens: number;
@@ -1464,6 +1494,32 @@ interface AgentLoopOptions {
  */
 declare function runAgentLoop(options: AgentLoopOptions): AsyncGenerator<StreamEvent>;
 
+/**
+ * Storage Helpers
+ *
+ * Extract input/output messages from request/response for storage adapters.
+ * Used internally by Runtime when `storage` is configured.
+ */
+
+/**
+ * Extract new INPUT messages from a request's message array.
+ *
+ * The SDK sends the full conversation history on each request. This function
+ * picks only the NEW messages the user just added:
+ * - New user turn (last meaningful msg = "user"): returns [userMsg]
+ * - Tool continuation (last meaningful msg = "tool"): returns tool results after last assistant
+ * - Otherwise: returns []
+ *
+ * Skips empty assistant placeholders the SDK pushes before sending.
+ */
+declare function extractInputMessages(reqMessages: unknown[]): StorageMessage[];
+/**
+ * Map LLM output messages (DoneEventMessage format) to StorageMessage format.
+ *
+ * Converts from snake_case API format to camelCase storage format.
+ */
+declare function mapOutputMessages(resultMessages: DoneEventMessage[]): StorageMessage[];
+
 /**
  * Generate a message ID
  */
@@ -1477,4 +1533,4 @@ declare function generateThreadId(): string;
  */
 declare function generateToolCallId(): string;
 
-export { AIProvider, ActionDefinition, type ActionRequest, type AgentLoopOptions, type ChatRequest, type CollectedResult, type CopilotChatResponse, DEFAULT_MAX_ITERATIONS, DoneEventMessage, GenerateResult, type GenerateResultData, GenerateTextParams, GenerateTextResult, LLMAdapter, Message, ProviderToolRuntimeOptions, type RequestContext, Runtime, type RuntimeConfig, StreamEvent, StreamResult, type StreamResultOptions, StreamTextParams, StreamTextResult, TokenUsageRaw, Tool, ToolCallInfo, ToolContext, ToolDefinition, ToolProfile, ToolResponse, type ToolSearchMatch, WebSearchConfig, buildProviderToolOptions, createEventStream, createExpressHandler, createExpressMiddleware, createHonoApp, createNextHandler, createNodeHandler, createRuntime, createSSEHeaders, createSSEResponse, createStreamResult, createTextStreamHeaders, createTextStreamResponse, formatSSEData, formatToolsForAnthropic, formatToolsForGoogle, formatToolsForOpenAI, generateMessageId, generateText, generateThreadId, generateToolCallId, pipeSSEToResponse, pipeTextToResponse, runAgentLoop, searchTools, selectTools, shouldExposeToolSearch, streamText, tool };
+export { AIProvider, ActionDefinition, type ActionRequest, type AgentLoopOptions, type ChatRequest, type CollectedResult, type CopilotChatResponse, DEFAULT_MAX_ITERATIONS, DoneEventMessage, GenerateResult, type GenerateResultData, GenerateTextParams, GenerateTextResult, LLMAdapter, Message, ProviderToolRuntimeOptions, type RequestContext, Runtime, type RuntimeConfig, StorageAdapter, StorageMessage, StreamEvent, StreamResult, type StreamResultOptions, StreamTextParams, StreamTextResult, TokenUsageRaw, Tool, ToolCallInfo, ToolContext, ToolDefinition, ToolProfile, ToolResponse, type ToolSearchMatch, WebSearchConfig, buildProviderToolOptions, createEventStream, createExpressHandler, createExpressMiddleware, createHonoApp, createNextHandler, createNodeHandler, createRuntime, createSSEHeaders, createSSEResponse, createStreamResult, createTextStreamHeaders, createTextStreamResponse, extractInputMessages, formatSSEData, formatToolsForAnthropic, formatToolsForGoogle, formatToolsForOpenAI, generateMessageId, generateText, generateThreadId, generateToolCallId, mapOutputMessages, pipeSSEToResponse, pipeTextToResponse, runAgentLoop, searchTools, selectTools, shouldExposeToolSearch, streamText, tool };

package/dist/index.js

@@ -553,6 +553,54 @@ function createMessage(partial) {
   };
 }
 
+// src/server/storage-helpers.ts
+function extractInputMessages(reqMessages) {
+  if (!reqMessages?.length) return [];
+  let lastMeaningful = null;
+  for (let i = reqMessages.length - 1; i >= 0; i--) {
+    const m = reqMessages[i];
+    if (m.role === "assistant" && (!m.content || m.content === "")) continue;
+    lastMeaningful = m;
+    break;
+  }
+  if (!lastMeaningful) return [];
+  if (lastMeaningful.role === "user") {
+    const textContent = typeof lastMeaningful.content === "string" ? lastMeaningful.content : JSON.stringify(lastMeaningful.content);
+    const attachments = lastMeaningful.attachments;
+    const imageAtt = attachments?.find((a) => a.type === "image" && a.url);
+    const fileAtt = attachments?.find((a) => a.type === "file" && a.url);
+    const msg = { role: "user", content: textContent || "" };
+    if (imageAtt?.url) {
+      msg.contentType = "image";
+      msg.url = imageAtt.url;
+    } else if (fileAtt?.url) {
+      msg.contentType = "file";
+      msg.url = fileAtt.url;
+    }
+    return [msg];
+  }
+  if (lastMeaningful.role === "tool" || lastMeaningful.role === "function") {
+    const msgs = reqMessages;
+    const lastAssistantIdx = msgs.map((m) => m.role).lastIndexOf("assistant");
+    return msgs.slice(lastAssistantIdx + 1).filter(
+      (m) => !(m.role === "assistant" && (!m.content || m.content === ""))
+    ).map((m) => ({
+      role: m.role,
+      content: typeof m.content === "string" ? m.content : JSON.stringify(m.content),
+      toolCallId: m.tool_call_id
+    }));
+  }
+  return [];
+}
+function mapOutputMessages(resultMessages) {
+  return resultMessages.map((m) => ({
+    role: m.role,
+    content: m.content ?? "",
+    toolCalls: m.tool_calls,
+    toolCallId: m.tool_call_id
+  }));
+}
+
 // src/server/streaming.ts
 function createSSEHeaders() {
   return {
@@ -951,6 +999,7 @@ var StreamResult = class {
       toolCalls: [],
       requiresAction: false,
       usage: void 0,
+      threadId: void 0,
       events: []
     };
   }
@@ -996,6 +1045,9 @@ var StreamResult = class {
           this.capturedUsage = event.usage;
           collected.usage = event.usage;
         }
+        if (event.threadId) {
+          collected.threadId = event.threadId;
+        }
         break;
     }
   }
@@ -1008,6 +1060,7 @@ var StreamResult = class {
         const usage = this.capturedUsage;
         await this.onFinishCallback({
           messages: collected.messages,
+          threadId: collected.threadId,
           usage: usage ? {
             promptTokens: usage.prompt_tokens,
             completionTokens: usage.completion_tokens,
@@ -1495,6 +1548,7 @@ var Runtime = class {
     this.actions = /* @__PURE__ */ new Map();
     this.tools = /* @__PURE__ */ new Map();
     this.config = config;
+    this.storage = config.storage;
     if ("provider" in config && config.provider) {
       this.adapter = config.provider.languageModel(config.model);
     } else if ("adapter" in config && config.adapter) {
@@ -2744,8 +2798,67 @@ var Runtime = class {
    * ```
    */
   stream(request, options) {
-    const generator = this.processChatWithLoop(request, options?.signal);
-    return new StreamResult(generator, { onFinish: options?.onFinish });
+    const storage = this.storage;
+    if (!storage) {
+      const generator = this.processChatWithLoop(request, options?.signal);
+      return new StreamResult(generator, { onFinish: options?.onFinish });
+    }
+    let resolvedThreadId = request.threadId;
+    const self = this;
+    let storageHealthy = true;
+    async function* storageWrappedGenerator() {
+      if (!resolvedThreadId) {
+        try {
+          const session = await storage.createSession();
+          resolvedThreadId = session.id;
+        } catch (err) {
+          console.error(
+            "[Runtime] storage.createSession failed \u2014 generating fallback threadId:",
+            err
+          );
+          resolvedThreadId = `local_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+          storageHealthy = false;
+        }
+      }
+      if (resolvedThreadId && storageHealthy) {
+        try {
+          const inputMsgs = extractInputMessages(request.messages);
+          if (inputMsgs.length) {
+            await storage.saveMessages(resolvedThreadId, inputMsgs);
+          }
+        } catch (err) {
+          console.error("[Runtime] storage.saveMessages (input) failed:", err);
+        }
+      }
+      for await (const event of self.processChatWithLoop(
+        request,
+        options?.signal
+      )) {
+        if (event.type === "done" && resolvedThreadId) {
+          yield { ...event, threadId: resolvedThreadId };
+        } else {
+          yield event;
+        }
+      }
+    }
+    return new StreamResult(storageWrappedGenerator(), {
+      onFinish: async (result) => {
+        if (resolvedThreadId && storageHealthy && result.messages.length > 0) {
+          try {
+            const outputMsgs = mapOutputMessages(result.messages);
+            await storage.saveMessages(resolvedThreadId, outputMsgs);
+          } catch (err) {
+            console.error(
+              "[Runtime] storage.saveMessages (output) failed:",
+              err
+            );
+          }
+        }
+        if (options?.onFinish) {
+          await options.onFinish({ ...result, threadId: resolvedThreadId });
+        }
+      }
+    });
   }
   /**
    * Chat and collect the full response (non-streaming)
@@ -3703,6 +3816,7 @@ exports.createSSEResponse = createSSEResponse;
 exports.createStreamResult = createStreamResult;
 exports.createTextStreamHeaders = createTextStreamHeaders;
 exports.createTextStreamResponse = createTextStreamResponse;
+exports.extractInputMessages = extractInputMessages;
 exports.formatSSEData = formatSSEData;
 exports.formatToolsForAnthropic = formatToolsForAnthropic;
 exports.formatToolsForGoogle = formatToolsForGoogle;
@@ -3711,6 +3825,7 @@ exports.generateMessageId = generateMessageId;
 exports.generateText = generateText;
 exports.generateThreadId = generateThreadId;
 exports.generateToolCallId = generateToolCallId;
+exports.mapOutputMessages = mapOutputMessages;
 exports.pipeSSEToResponse = pipeSSEToResponse;
 exports.pipeTextToResponse = pipeTextToResponse;
 exports.runAgentLoop = runAgentLoop;