@voltagent/core 0.1.47 → 0.1.48

package/dist/index.d.ts

@@ -1261,6 +1261,26 @@ type InferResponseFromProvider<TProvider extends {
 type InferOriginalResponseFromProvider<TProvider extends {
     llm: LLMProvider<any>;
 }, TMethod extends "generateText" | "streamText" | "generateObject" | "streamObject"> = InferResponseFromProvider<TProvider, TMethod>["provider"];
+type GenerateTextResponse<TProvider extends {
+    llm: LLMProvider<any>;
+}> = InferGenerateTextResponseFromProvider<TProvider> & {
+    userContext: Map<string | symbol, unknown>;
+};
+type StreamTextResponse<TProvider extends {
+    llm: LLMProvider<any>;
+}> = InferStreamTextResponseFromProvider<TProvider> & {
+    userContext?: Map<string | symbol, unknown>;
+};
+type GenerateObjectResponse<TProvider extends {
+    llm: LLMProvider<any>;
+}, TSchema extends z.ZodType> = InferGenerateObjectResponseFromProvider<TProvider, TSchema> & {
+    userContext: Map<string | symbol, unknown>;
+};
+type StreamObjectResponse<TProvider extends {
+    llm: LLMProvider<any>;
+}, TSchema extends z.ZodType> = InferStreamObjectResponseFromProvider<TProvider, TSchema> & {
+    userContext?: Map<string | symbol, unknown>;
+};
 
 /**
  * Token usage information
@@ -3095,19 +3115,19 @@ declare class Agent<TProvider extends {
     /**
      * Generate a text response without streaming
      */
-    generateText(input: string | BaseMessage[], options?: PublicGenerateOptions): Promise<InferGenerateTextResponseFromProvider<TProvider>>;
+    generateText(input: string | BaseMessage[], options?: PublicGenerateOptions): Promise<GenerateTextResponse<TProvider>>;
     /**
      * Stream a text response
      */
-    streamText(input: string | BaseMessage[], options?: PublicGenerateOptions): Promise<InferStreamTextResponseFromProvider<TProvider>>;
+    streamText(input: string | BaseMessage[], options?: PublicGenerateOptions): Promise<StreamTextResponse<TProvider>>;
     /**
      * Generate a structured object response
      */
-    generateObject<TSchema extends z.ZodType>(input: string | BaseMessage[], schema: TSchema, options?: PublicGenerateOptions): Promise<InferGenerateObjectResponseFromProvider<TProvider, TSchema>>;
+    generateObject<TSchema extends z.ZodType>(input: string | BaseMessage[], schema: TSchema, options?: PublicGenerateOptions): Promise<GenerateObjectResponse<TProvider, TSchema>>;
     /**
      * Stream a structured object response
      */
-    streamObject<TSchema extends z.ZodType>(input: string | BaseMessage[], schema: TSchema, options?: PublicGenerateOptions): Promise<InferStreamObjectResponseFromProvider<TProvider, TSchema>>;
+    streamObject<TSchema extends z.ZodType>(input: string | BaseMessage[], schema: TSchema, options?: PublicGenerateOptions): Promise<StreamObjectResponse<TProvider, TSchema>>;
     /**
      * Add a sub-agent that this agent can delegate tasks to
      */

package/dist/index.js

@@ -8110,17 +8110,14 @@ ${context}`;
         status: "completed"
       });
       operationContext.isActive = false;
-      const standardizedOutput = {
-        text: response.text,
-        usage: response.usage,
-        finishReason: response.finishReason,
-        providerResponse: response,
+      const extendedResponse = {
+        ...response,
         userContext: new Map(operationContext.userContext)
       };
       await this.hooks.onEnd?.({
         conversationId: finalConversationId,
         agent: this,
-        output: standardizedOutput,
+        output: extendedResponse,
         error: void 0,
         context: operationContext
       });
@@ -8130,7 +8127,7 @@ ${context}`;
         endTime: /* @__PURE__ */ new Date(),
         status: "completed"
       });
-      return response;
+      return extendedResponse;
     } catch (error) {
       const voltagentError = error;
       const agentErrorStartInfo = {
@@ -8634,7 +8631,8 @@ ${context}`;
     });
     const wrappedResponse = {
       ...response,
-      fullStream: response.fullStream ? this.createEnhancedFullStream(response.fullStream, streamController, subAgentStatus) : void 0
+      fullStream: response.fullStream ? this.createEnhancedFullStream(response.fullStream, streamController, subAgentStatus) : void 0,
+      userContext: new Map(operationContext.userContext)
     };
     return wrappedResponse;
   }
@@ -8792,21 +8790,18 @@ ${context}`;
         endTime: /* @__PURE__ */ new Date(),
         status: "completed"
       });
-      const standardizedOutput = {
-        object: response.object,
-        usage: response.usage,
-        finishReason: response.finishReason,
-        providerResponse: response,
+      const extendedResponse = {
+        ...response,
         userContext: new Map(operationContext.userContext)
       };
       await this.hooks.onEnd?.({
         agent: this,
-        output: standardizedOutput,
+        output: extendedResponse,
         error: void 0,
         conversationId: finalConversationId,
         context: operationContext
       });
-      return response;
+      return extendedResponse;
     } catch (error) {
       const voltagentError = error;
       const agentErrorStartInfo = {
@@ -9122,7 +9117,11 @@ ${context}`;
           });
         }
       });
-      return response;
+      const extendedResponse = {
+        ...response,
+        userContext: new Map(operationContext.userContext)
+      };
+      return extendedResponse;
     } catch (error) {
       operationContext.isActive = false;
       await this.hooks.onEnd?.({