@voltagent/core 0.1.46 → 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
@@ -1515,6 +1535,8 @@ interface StepWithContent {
     arguments?: Record<string, any>;
     result?: any;
     usage?: UsageInfo;
+    subAgentId?: string;
+    subAgentName?: string;
 }
 type StepFinishCallback = (step: StepWithContent) => void | Promise<void>;
 type StepChunkCallback = (chunk: any) => void | Promise<void>;
@@ -3093,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

@@ -2477,6 +2477,7 @@ var import_dev19 = require("@voltagent/internal/dev");
 
 // src/agent/index.ts
 var import_dev14 = require("@voltagent/internal/dev");
+var import_ts_pattern2 = require("ts-pattern");
 
 // src/memory/in-memory/index.ts
 var import_dev4 = require("@voltagent/internal/dev");
@@ -7721,7 +7722,14 @@ ${context}`;
     if (!context.conversationSteps) {
       context.conversationSteps = [];
     }
-    context.conversationSteps.push(step);
+    const finalStep = {
+      ...step,
+      ...(0, import_ts_pattern2.match)(context).with({ parentAgentId: import_ts_pattern2.P.not(import_ts_pattern2.P.nullish) }, () => ({
+        subAgentId: this.id,
+        subAgentName: this.name
+      })).otherwise(() => ({}))
+    };
+    context.conversationSteps.push(finalStep);
   }
   /**
    * Update history entry
@@ -8102,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
       });
@@ -8122,7 +8127,7 @@ ${context}`;
         endTime: /* @__PURE__ */ new Date(),
         status: "completed"
       });
-      return response;
+      return extendedResponse;
     } catch (error) {
       const voltagentError = error;
       const agentErrorStartInfo = {
@@ -8626,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;
   }
@@ -8784,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 = {
@@ -9114,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?.({