@voltagent/core 1.0.1 → 1.1.1

package/dist/index.d.mts

@@ -1,9 +1,9 @@
+import { ModelMessage, DataContent as DataContent$1, UserContent, AssistantContent, ToolContent, ProviderOptions as ProviderOptions$1 } from '@ai-sdk/provider-utils';
+export { AssistantContent, FilePart, ImagePart, TextPart, ToolContent, UserContent } from '@ai-sdk/provider-utils';
 import { TextStreamPart, generateText, UIMessage, StreamTextResult, LanguageModel, CallSettings, ToolSet, GenerateTextResult, GenerateObjectResult, AsyncIterableStream as AsyncIterableStream$1, CallWarning, LanguageModelUsage, FinishReason, EmbeddingModel } from 'ai';
 export { hasToolCall, stepCountIs } from 'ai';
 import * as zod from 'zod';
 import { z } from 'zod';
-import { ModelMessage, DataContent as DataContent$1, UserContent, AssistantContent, ToolContent, ProviderOptions as ProviderOptions$1 } from '@ai-sdk/provider-utils';
-export { AssistantContent, FilePart, ImagePart, TextPart, ToolContent, UserContent } from '@ai-sdk/provider-utils';
 import { AsyncIterableStream } from '@voltagent/internal/utils';
 export { AsyncIterableStream, createAsyncIterableStream } from '@voltagent/internal/utils';
 import { Logger, LogFn, LogBuffer } from '@voltagent/internal';
@@ -2585,7 +2585,7 @@ interface TraceContextOptions {
     operationId: string;
     parentSpan?: Span;
     parentAgentId?: string;
-    input?: string | UIMessage[];
+    input?: string | UIMessage[] | BaseMessage[];
 }
 declare class AgentTraceContext {
     private rootSpan;
@@ -3335,19 +3335,19 @@ declare class Agent {
     /**
      * Generate text response
      */
-    generateText(input: string | UIMessage[], options?: GenerateTextOptions): Promise<GenerateTextResultWithContext>;
+    generateText(input: string | UIMessage[] | BaseMessage[], options?: GenerateTextOptions): Promise<GenerateTextResultWithContext>;
     /**
      * Stream text response
      */
-    streamText(input: string | UIMessage[], options?: StreamTextOptions): Promise<StreamTextResultWithContext>;
+    streamText(input: string | UIMessage[] | BaseMessage[], options?: StreamTextOptions): Promise<StreamTextResultWithContext>;
     /**
      * Generate structured object
      */
-    generateObject<T extends z.ZodType>(input: string | UIMessage[], schema: T, options?: GenerateObjectOptions): Promise<GenerateObjectResultWithContext<z.infer<T>>>;
+    generateObject<T extends z.ZodType>(input: string | UIMessage[] | BaseMessage[], schema: T, options?: GenerateObjectOptions): Promise<GenerateObjectResultWithContext<z.infer<T>>>;
     /**
      * Stream structured object
      */
-    streamObject<T extends z.ZodType>(input: string | UIMessage[], schema: T, options?: StreamObjectOptions): Promise<StreamObjectResultWithContext<z.infer<T>>>;
+    streamObject<T extends z.ZodType>(input: string | UIMessage[] | BaseMessage[], schema: T, options?: StreamObjectOptions): Promise<StreamObjectResultWithContext<z.infer<T>>>;
     /**
      * Common preparation for all execution methods
      */
@@ -3788,7 +3788,7 @@ type WorkflowHooks<DATA, RESULT> = {
      */
     onEnd?: (state: WorkflowState<DATA, RESULT>) => Promise<void>;
 };
-type WorkflowInput<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema> = TF.IsUnknown<INPUT_SCHEMA> extends true ? BaseMessage | BaseMessage[] | string : INPUT_SCHEMA extends z.ZodTypeAny ? z.infer<INPUT_SCHEMA> : undefined;
+type WorkflowInput<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema> = TF.IsUnknown<INPUT_SCHEMA> extends true ? BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string : INPUT_SCHEMA extends z.ZodTypeAny ? z.infer<INPUT_SCHEMA> : undefined;
 type WorkflowResult<RESULT_SCHEMA extends z.ZodTypeAny> = RESULT_SCHEMA extends z.ZodTypeAny ? z.infer<RESULT_SCHEMA> : RESULT_SCHEMA;
 type WorkflowConfig<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema, RESULT_SCHEMA extends z.ZodTypeAny, SUSPEND_SCHEMA extends z.ZodTypeAny = z.ZodAny, RESUME_SCHEMA extends z.ZodTypeAny = z.ZodAny> = {
     /**
@@ -4226,7 +4226,7 @@ interface WorkflowStepContext {
  * The base input type for the workflow
  * @private - INTERNAL USE ONLY
  */
-type InternalBaseWorkflowInputSchema = z.ZodTypeAny | BaseMessage | BaseMessage[] | string;
+type InternalBaseWorkflowInputSchema = z.ZodTypeAny | BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string;
 /**
  * The state parameter for the workflow, used to pass the state to a step or other function (i.e. hooks)
  * @private - INTERNAL USE ONLY
@@ -4338,7 +4338,7 @@ type InternalAnyWorkflowStep<INPUT = DangerouslyAllowAny, DATA = DangerouslyAllo
 type InternalInferWorkflowStepsResult<STEPS extends ReadonlyArray<InternalAnyWorkflowStep<any, any, any, any, any>>> = {
     [K in keyof STEPS]: ExtractExecuteResult<STEPS[K]>;
 };
-type InternalExtractWorkflowInputData<T> = TF.IsUnknown<T> extends true ? BaseMessage | BaseMessage[] | string : TF.IsAny<T> extends true ? BaseMessage | BaseMessage[] | string : T extends z.ZodType ? z.infer<T> : T;
+type InternalExtractWorkflowInputData<T> = TF.IsUnknown<T> extends true ? BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string : TF.IsAny<T> extends true ? BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string : T extends z.ZodType ? z.infer<T> : T;
 type ExtractExecuteResult<T> = T extends {
     execute: (...args: any[]) => infer R;
 } ? R extends Promise<infer U> ? U : R : never;
@@ -4369,7 +4369,7 @@ type AgentConfig$1<SCHEMA extends z.ZodTypeAny> = BaseGenerationOptions & {
  * @param config - The config for the agent (schema) `generateObject` call
  * @returns A workflow step that executes the agent with the task
  */
-declare function andAgent<INPUT, DATA, SCHEMA extends z.ZodTypeAny>(task: UIMessage[] | string | InternalWorkflowFunc<INPUT, DATA, UIMessage[] | string, any, any>, agent: Agent, config: AgentConfig$1<SCHEMA>): {
+declare function andAgent<INPUT, DATA, SCHEMA extends z.ZodTypeAny>(task: UIMessage[] | ModelMessage[] | string | InternalWorkflowFunc<INPUT, DATA, UIMessage[] | ModelMessage[] | string, any, any>, agent: Agent, config: AgentConfig$1<SCHEMA>): {
     type: "agent";
     id: string;
     name: string;
@@ -4856,7 +4856,7 @@ declare class WorkflowChain<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema
      * @param config - The config for the agent (schema) `generateObject` call
      * @returns A workflow step that executes the agent with the task
      */
-    andAgent<SCHEMA extends z.ZodTypeAny>(task: string | InternalWorkflowFunc<INPUT_SCHEMA, CURRENT_DATA, string, any, any>, agent: Agent, config: AgentConfig<SCHEMA>): WorkflowChain<INPUT_SCHEMA, RESULT_SCHEMA, z.infer<SCHEMA>, SUSPEND_SCHEMA, RESUME_SCHEMA>;
+    andAgent<SCHEMA extends z.ZodTypeAny>(task: string | UIMessage[] | ModelMessage[] | InternalWorkflowFunc<INPUT_SCHEMA, CURRENT_DATA, string | UIMessage[] | ModelMessage[], any, any>, agent: Agent, config: AgentConfig<SCHEMA>): WorkflowChain<INPUT_SCHEMA, RESULT_SCHEMA, z.infer<SCHEMA>, SUSPEND_SCHEMA, RESUME_SCHEMA>;
     /**
      * Add a function step to the workflow with both input and output schemas
      * @param config - Step configuration with schemas
@@ -6943,6 +6943,12 @@ declare class VoltAgent {
      * Shutdown telemetry (delegates to VoltAgentObservability)
      */
     shutdownTelemetry(): Promise<void>;
+    /**
+     * Gracefully shutdown all VoltAgent resources
+     * This includes stopping the server, suspending workflows, and shutting down telemetry
+     * Useful for programmatic cleanup or when integrating with other frameworks
+     */
+    shutdown(): Promise<void>;
 }
 
 /**

package/dist/index.d.ts

@@ -1,9 +1,9 @@
+import { ModelMessage, DataContent as DataContent$1, UserContent, AssistantContent, ToolContent, ProviderOptions as ProviderOptions$1 } from '@ai-sdk/provider-utils';
+export { AssistantContent, FilePart, ImagePart, TextPart, ToolContent, UserContent } from '@ai-sdk/provider-utils';
 import { TextStreamPart, generateText, UIMessage, StreamTextResult, LanguageModel, CallSettings, ToolSet, GenerateTextResult, GenerateObjectResult, AsyncIterableStream as AsyncIterableStream$1, CallWarning, LanguageModelUsage, FinishReason, EmbeddingModel } from 'ai';
 export { hasToolCall, stepCountIs } from 'ai';
 import * as zod from 'zod';
 import { z } from 'zod';
-import { ModelMessage, DataContent as DataContent$1, UserContent, AssistantContent, ToolContent, ProviderOptions as ProviderOptions$1 } from '@ai-sdk/provider-utils';
-export { AssistantContent, FilePart, ImagePart, TextPart, ToolContent, UserContent } from '@ai-sdk/provider-utils';
 import { AsyncIterableStream } from '@voltagent/internal/utils';
 export { AsyncIterableStream, createAsyncIterableStream } from '@voltagent/internal/utils';
 import { Logger, LogFn, LogBuffer } from '@voltagent/internal';
@@ -2585,7 +2585,7 @@ interface TraceContextOptions {
     operationId: string;
     parentSpan?: Span;
     parentAgentId?: string;
-    input?: string | UIMessage[];
+    input?: string | UIMessage[] | BaseMessage[];
 }
 declare class AgentTraceContext {
     private rootSpan;
@@ -3335,19 +3335,19 @@ declare class Agent {
     /**
      * Generate text response
      */
-    generateText(input: string | UIMessage[], options?: GenerateTextOptions): Promise<GenerateTextResultWithContext>;
+    generateText(input: string | UIMessage[] | BaseMessage[], options?: GenerateTextOptions): Promise<GenerateTextResultWithContext>;
     /**
      * Stream text response
      */
-    streamText(input: string | UIMessage[], options?: StreamTextOptions): Promise<StreamTextResultWithContext>;
+    streamText(input: string | UIMessage[] | BaseMessage[], options?: StreamTextOptions): Promise<StreamTextResultWithContext>;
     /**
      * Generate structured object
      */
-    generateObject<T extends z.ZodType>(input: string | UIMessage[], schema: T, options?: GenerateObjectOptions): Promise<GenerateObjectResultWithContext<z.infer<T>>>;
+    generateObject<T extends z.ZodType>(input: string | UIMessage[] | BaseMessage[], schema: T, options?: GenerateObjectOptions): Promise<GenerateObjectResultWithContext<z.infer<T>>>;
     /**
      * Stream structured object
      */
-    streamObject<T extends z.ZodType>(input: string | UIMessage[], schema: T, options?: StreamObjectOptions): Promise<StreamObjectResultWithContext<z.infer<T>>>;
+    streamObject<T extends z.ZodType>(input: string | UIMessage[] | BaseMessage[], schema: T, options?: StreamObjectOptions): Promise<StreamObjectResultWithContext<z.infer<T>>>;
     /**
      * Common preparation for all execution methods
      */
@@ -3788,7 +3788,7 @@ type WorkflowHooks<DATA, RESULT> = {
      */
     onEnd?: (state: WorkflowState<DATA, RESULT>) => Promise<void>;
 };
-type WorkflowInput<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema> = TF.IsUnknown<INPUT_SCHEMA> extends true ? BaseMessage | BaseMessage[] | string : INPUT_SCHEMA extends z.ZodTypeAny ? z.infer<INPUT_SCHEMA> : undefined;
+type WorkflowInput<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema> = TF.IsUnknown<INPUT_SCHEMA> extends true ? BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string : INPUT_SCHEMA extends z.ZodTypeAny ? z.infer<INPUT_SCHEMA> : undefined;
 type WorkflowResult<RESULT_SCHEMA extends z.ZodTypeAny> = RESULT_SCHEMA extends z.ZodTypeAny ? z.infer<RESULT_SCHEMA> : RESULT_SCHEMA;
 type WorkflowConfig<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema, RESULT_SCHEMA extends z.ZodTypeAny, SUSPEND_SCHEMA extends z.ZodTypeAny = z.ZodAny, RESUME_SCHEMA extends z.ZodTypeAny = z.ZodAny> = {
     /**
@@ -4226,7 +4226,7 @@ interface WorkflowStepContext {
  * The base input type for the workflow
  * @private - INTERNAL USE ONLY
  */
-type InternalBaseWorkflowInputSchema = z.ZodTypeAny | BaseMessage | BaseMessage[] | string;
+type InternalBaseWorkflowInputSchema = z.ZodTypeAny | BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string;
 /**
  * The state parameter for the workflow, used to pass the state to a step or other function (i.e. hooks)
  * @private - INTERNAL USE ONLY
@@ -4338,7 +4338,7 @@ type InternalAnyWorkflowStep<INPUT = DangerouslyAllowAny, DATA = DangerouslyAllo
 type InternalInferWorkflowStepsResult<STEPS extends ReadonlyArray<InternalAnyWorkflowStep<any, any, any, any, any>>> = {
     [K in keyof STEPS]: ExtractExecuteResult<STEPS[K]>;
 };
-type InternalExtractWorkflowInputData<T> = TF.IsUnknown<T> extends true ? BaseMessage | BaseMessage[] | string : TF.IsAny<T> extends true ? BaseMessage | BaseMessage[] | string : T extends z.ZodType ? z.infer<T> : T;
+type InternalExtractWorkflowInputData<T> = TF.IsUnknown<T> extends true ? BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string : TF.IsAny<T> extends true ? BaseMessage | BaseMessage[] | UIMessage | UIMessage[] | string : T extends z.ZodType ? z.infer<T> : T;
 type ExtractExecuteResult<T> = T extends {
     execute: (...args: any[]) => infer R;
 } ? R extends Promise<infer U> ? U : R : never;
@@ -4369,7 +4369,7 @@ type AgentConfig$1<SCHEMA extends z.ZodTypeAny> = BaseGenerationOptions & {
  * @param config - The config for the agent (schema) `generateObject` call
  * @returns A workflow step that executes the agent with the task
  */
-declare function andAgent<INPUT, DATA, SCHEMA extends z.ZodTypeAny>(task: UIMessage[] | string | InternalWorkflowFunc<INPUT, DATA, UIMessage[] | string, any, any>, agent: Agent, config: AgentConfig$1<SCHEMA>): {
+declare function andAgent<INPUT, DATA, SCHEMA extends z.ZodTypeAny>(task: UIMessage[] | ModelMessage[] | string | InternalWorkflowFunc<INPUT, DATA, UIMessage[] | ModelMessage[] | string, any, any>, agent: Agent, config: AgentConfig$1<SCHEMA>): {
     type: "agent";
     id: string;
     name: string;
@@ -4856,7 +4856,7 @@ declare class WorkflowChain<INPUT_SCHEMA extends InternalBaseWorkflowInputSchema
      * @param config - The config for the agent (schema) `generateObject` call
      * @returns A workflow step that executes the agent with the task
      */
-    andAgent<SCHEMA extends z.ZodTypeAny>(task: string | InternalWorkflowFunc<INPUT_SCHEMA, CURRENT_DATA, string, any, any>, agent: Agent, config: AgentConfig<SCHEMA>): WorkflowChain<INPUT_SCHEMA, RESULT_SCHEMA, z.infer<SCHEMA>, SUSPEND_SCHEMA, RESUME_SCHEMA>;
+    andAgent<SCHEMA extends z.ZodTypeAny>(task: string | UIMessage[] | ModelMessage[] | InternalWorkflowFunc<INPUT_SCHEMA, CURRENT_DATA, string | UIMessage[] | ModelMessage[], any, any>, agent: Agent, config: AgentConfig<SCHEMA>): WorkflowChain<INPUT_SCHEMA, RESULT_SCHEMA, z.infer<SCHEMA>, SUSPEND_SCHEMA, RESUME_SCHEMA>;
     /**
      * Add a function step to the workflow with both input and output schemas
      * @param config - Step configuration with schemas
@@ -6943,6 +6943,12 @@ declare class VoltAgent {
      * Shutdown telemetry (delegates to VoltAgentObservability)
      */
     shutdownTelemetry(): Promise<void>;
+    /**
+     * Gracefully shutdown all VoltAgent resources
+     * This includes stopping the server, suspending workflows, and shutting down telemetry
+     * Useful for programmatic cleanup or when integrating with other frameworks
+     */
+    shutdown(): Promise<void>;
 }
 
 /**

package/dist/index.js

@@ -7197,6 +7197,148 @@ async function convertResponseMessagesToUIMessages(responseMessages) {
   return uiMessage.parts.length > 0 ? [uiMessage] : [];
 }
 __name(convertResponseMessagesToUIMessages, "convertResponseMessagesToUIMessages");
+function convertModelMessagesToUIMessages(messages) {
+  const uiMessages = [];
+  for (const message of messages) {
+    if (message.role === "tool") {
+      if (Array.isArray(message.content)) {
+        for (const part of message.content) {
+          if (part.type === "tool-result") {
+            uiMessages.push({
+              id: crypto3.randomUUID(),
+              role: "assistant",
+              parts: [
+                {
+                  type: `tool-${part.toolName}`,
+                  toolCallId: part.toolCallId,
+                  state: "output-available",
+                  input: {},
+                  output: part.output,
+                  providerExecuted: true
+                }
+              ]
+            });
+          }
+        }
+      }
+      continue;
+    }
+    const ui = {
+      id: crypto3.randomUUID(),
+      role: message.role,
+      parts: []
+    };
+    if (typeof message.content === "string") {
+      if (message.content.trim().length > 0) {
+        ui.parts.push({ type: "text", text: message.content });
+      }
+      uiMessages.push(ui);
+      continue;
+    }
+    for (const contentPart of message.content || []) {
+      switch (contentPart.type) {
+        case "text": {
+          if (contentPart.text && contentPart.text.length > 0) {
+            const prev = ui.parts.at(-1);
+            if (message.role === "assistant" && prev && typeof prev?.type === "string" && prev.type.startsWith("tool-") && prev.state === "output-available") {
+              ui.parts.push({ type: "step-start" });
+            }
+            ui.parts.push({
+              type: "text",
+              text: contentPart.text,
+              ...contentPart.providerOptions ? { providerMetadata: contentPart.providerOptions } : {}
+            });
+          }
+          break;
+        }
+        case "reasoning": {
+          if (contentPart.text && contentPart.text.length > 0) {
+            ui.parts.push({
+              type: "reasoning",
+              text: contentPart.text,
+              ...contentPart.providerOptions ? { providerMetadata: contentPart.providerOptions } : {}
+            });
+          }
+          break;
+        }
+        case "tool-call": {
+          ui.parts.push({
+            type: `tool-${contentPart.toolName}`,
+            toolCallId: contentPart.toolCallId,
+            state: "input-available",
+            input: contentPart.input || {},
+            ...contentPart.providerOptions ? { callProviderMetadata: contentPart.providerOptions } : {},
+            ...contentPart.providerExecuted != null ? { providerExecuted: contentPart.providerExecuted } : {}
+          });
+          break;
+        }
+        case "tool-result": {
+          ui.parts.push({
+            type: `tool-${contentPart.toolName}`,
+            toolCallId: contentPart.toolCallId,
+            state: "output-available",
+            input: {},
+            output: contentPart.output,
+            providerExecuted: true
+          });
+          break;
+        }
+        case "image": {
+          let url;
+          const mediaType = contentPart.mediaType || "image/png";
+          const img = contentPart.image;
+          if (img instanceof URL) {
+            url = img.toString();
+          } else if (typeof img === "string") {
+            if (/^(https?:\/\/|data:)/i.test(img)) {
+              url = img;
+            } else {
+              url = `data:${mediaType};base64,${img}`;
+            }
+          } else {
+            const uint8 = img;
+            const base64 = Buffer.from(uint8).toString("base64");
+            url = `data:${mediaType};base64,${base64}`;
+          }
+          ui.parts.push({
+            type: "file",
+            mediaType,
+            url,
+            ...contentPart.providerOptions ? { providerMetadata: contentPart.providerOptions } : {}
+          });
+          break;
+        }
+        case "file": {
+          let url;
+          if (contentPart.data instanceof URL) {
+            url = contentPart.data.toString();
+          } else if (typeof contentPart.data === "string") {
+            if (/^(https?:\/\/|data:)/i.test(contentPart.data)) {
+              url = contentPart.data;
+            } else {
+              url = `data:${contentPart.mediaType};base64,${contentPart.data}`;
+            }
+          } else {
+            const base64 = Buffer.from(contentPart.data).toString("base64");
+            url = `data:${contentPart.mediaType};base64,${base64}`;
+          }
+          ui.parts.push({
+            type: "file",
+            mediaType: contentPart.mediaType,
+            url,
+            ...contentPart.providerOptions ? { providerMetadata: contentPart.providerOptions } : {}
+          });
+          break;
+        }
+        default:
+          break;
+      }
+    }
+    uiMessages.push(ui);
+  }
+  return uiMessages;
+}
+__name(convertModelMessagesToUIMessages, "convertModelMessagesToUIMessages");
 
 // src/voltops/prompt-api-client.ts
 var VoltOpsPromptApiClient = class {
@@ -9620,16 +9762,24 @@ var Agent = class {
     if (typeof input === "string") {
       return input;
     }
+    if (!Array.isArray(input) || input.length === 0) return void 0;
+    const isUI = /* @__PURE__ */ __name((msg) => Array.isArray(msg?.parts), "isUI");
     const userMessages = input.filter((msg) => msg.role === "user");
     const lastUserMessage = userMessages.at(-1);
-    if (lastUserMessage?.parts) {
-      const textParts = lastUserMessage.parts.filter((part) => part.type === "text" && "text" in part).map((part) => part.text).filter((text) => text).map((text) => text.trim());
+    if (!lastUserMessage) return void 0;
+    if (isUI(lastUserMessage)) {
+      const textParts = lastUserMessage.parts.filter((part) => part.type === "text" && typeof part.text === "string").map((part) => part.text.trim()).filter(Boolean);
       if (textParts.length > 0) return textParts.join(" ");
+      return void 0;
     }
-    if (lastUserMessage?.content && typeof lastUserMessage.content === "string") {
+    if (typeof lastUserMessage.content === "string") {
       const content = lastUserMessage.content.trim();
       return content.length > 0 ? content : void 0;
     }
+    if (Array.isArray(lastUserMessage.content)) {
+      const textParts = lastUserMessage.content.filter((part) => part.type === "text" && typeof part.text === "string").map((part) => part.text.trim()).filter(Boolean);
+      if (textParts.length > 0) return textParts.join(" ");
+    }
     return void 0;
   }
   /**
@@ -9701,9 +9851,10 @@ var Agent = class {
                 }
               );
             }
+            const inputForMemory = typeof input === "string" ? input : Array.isArray(input) && input[0]?.parts ? input : convertModelMessagesToUIMessages(input);
             const result = await this.memoryManager.prepareConversationContext(
               oc,
-              input,
+              inputForMemory,
               oc.userId,
               oc.conversationId,
               options?.contextLimit
@@ -9723,7 +9874,8 @@ var Agent = class {
           messages.push(...memoryResult);
           if (isSemanticSearch && oc.userId && oc.conversationId) {
             try {
-              this.memoryManager.queueSaveInput(oc, input, oc.userId, oc.conversationId);
+              const inputForMemory = typeof input === "string" ? input : Array.isArray(input) && input[0]?.parts ? input : convertModelMessagesToUIMessages(input);
+              this.memoryManager.queueSaveInput(oc, inputForMemory, oc.userId, oc.conversationId);
             } catch (_e) {
             }
           }
@@ -9741,8 +9893,13 @@ var Agent = class {
         role: "user",
         parts: [{ type: "text", text: input }]
       });
-    } else {
-      messages.push(...input);
+    } else if (Array.isArray(input)) {
+      const first = input[0];
+      if (first && Array.isArray(first.parts)) {
+        messages.push(...input);
+      } else {
+        messages.push(...convertModelMessagesToUIMessages(input));
+      }
     }
     const hooks = this.getMergedHooks(options);
     if (hooks.onPrepareMessages) {
@@ -9957,7 +10114,7 @@ ${toolkit.instructions}`;
       }
     });
     try {
-      const retrieverInput = typeof input === "string" ? input : (0, import_ai.convertToModelMessages)(input);
+      const retrieverInput = typeof input === "string" ? input : Array.isArray(input) && input[0]?.content !== void 0 ? input : (0, import_ai.convertToModelMessages)(input);
       const retrievedContent = await oc.traceContext.withSpan(retrieverSpan, async () => {
         if (!this.retriever) return null;
         return await this.retriever.retrieve(retrieverInput, {
@@ -12722,11 +12879,10 @@ var VoltAgent = class {
    * Setup graceful shutdown handlers
    */
   setupShutdownHandlers() {
-    const shutdown = /* @__PURE__ */ __name(async (signal) => {
-      this.logger.info(`[VoltAgent] Received ${signal}, starting graceful shutdown...`);
+    const handleSignal = /* @__PURE__ */ __name(async (signal) => {
+      this.logger.info(`[VoltAgent] Received ${signal}...`);
       try {
-        await this.workflowRegistry.suspendAllActiveWorkflows();
-        this.logger.info("[VoltAgent] All workflows suspended, exiting...");
+        await this.shutdown();
         if (this.isSoleSignalHandler(signal)) {
           process.exit(0);
         }
@@ -12736,9 +12892,9 @@ var VoltAgent = class {
           process.exit(1);
         }
       }
-    }, "shutdown");
-    process.on("SIGTERM", () => shutdown("SIGTERM"));
-    process.on("SIGINT", () => shutdown("SIGINT"));
+    }, "handleSignal");
+    process.once("SIGTERM", () => handleSignal("SIGTERM"));
+    process.once("SIGINT", () => handleSignal("SIGINT"));
     process.on("unhandledRejection", (reason) => {
       this.logger.error("[VoltAgent] Unhandled Promise Rejection:", {
         reason: reason instanceof Error ? reason.message : reason,
@@ -12895,6 +13051,30 @@ var VoltAgent = class {
       await this.observability.shutdown();
     }
   }
+  /**
+   * Gracefully shutdown all VoltAgent resources
+   * This includes stopping the server, suspending workflows, and shutting down telemetry
+   * Useful for programmatic cleanup or when integrating with other frameworks
+   */
+  async shutdown() {
+    this.logger.info("[VoltAgent] Starting graceful shutdown...");
+    try {
+      if (this.serverInstance?.isRunning()) {
+        this.logger.info("[VoltAgent] Stopping server...");
+        await this.stopServer();
+      }
+      this.logger.info("[VoltAgent] Suspending active workflows...");
+      await this.workflowRegistry.suspendAllActiveWorkflows();
+      if (this.observability) {
+        this.logger.info("[VoltAgent] Shutting down telemetry...");
+        await this.shutdownTelemetry();
+      }
+      this.logger.info("[VoltAgent] Graceful shutdown complete");
+    } catch (error) {
+      this.logger.error("[VoltAgent] Error during shutdown:", { error });
+      throw error;
+    }
+  }
 };
 
 // src/index.ts